Re: [PATCH bpf-next 0/3] bpf: introduce BPF_MAP_TYPE_RELAY

2023-12-22 Thread Philo Lu




On 2023/12/22 22:45, Jiri Olsa wrote:

On Fri, Dec 22, 2023 at 08:21:43PM +0800, Philo Lu wrote:

The patch set introduce a new type of map, BPF_MAP_TYPE_RELAY, based on
relay interface [0]. It provides a way for persistent and overwritable data
transfer.

As stated in [0], relay is a efficient method for log and data transfer.
And the interface is simple enough so that we can implement and use this
type of map with current map interfaces. Besides we need a new helper
bpf_relay_output to output data to user, similar with bpf_ringbuf_output.

We need this map because currently neither ringbuf nor perfbuf satisfies
the requirements of relatively long-term consistent tracing, where the bpf
program keeps writing into the buffer without any bundled reader, and the
buffer supports overwriting. For users, they just run the bpf program to
collect data, and are able to read as need. The detailed discussion can be
found at [1].

The buffer is exposed to users as per-cpu files in debugfs, supporting read
and mmap, and it is up to users how to formulate and read it, either
through a program with mmap or just `cat`. Specifically, the files are
created as "/sys/kerenl/debug//#cpu", where the 
is defined with map_update_elem (Note that we do not need to implement
actual elem operators for relay_map).

If this map is acceptable, other parts including docs, libbpf support,
selftests, and benchmarks (if need) will be added in the following version.


looks useful, selftests might be already helpful to see the usage



Ok, I will add selftests in the next version.

Thanks.


jirka



[0]
https://github.com/torvalds/linux/blob/master/Documentation/filesystems/relay.rst
[1]
https://lore.kernel.org/bpf/20231219122850.433be...@gandalf.local.home/T/

Philo Lu (3):
   bpf: implement relay map basis
   bpf: implement map_update_elem to init relay file
   bpf: introduce bpf_relay_output helper

  include/linux/bpf.h   |   1 +
  include/linux/bpf_types.h |   3 +
  include/uapi/linux/bpf.h  |  17 +++
  kernel/bpf/Makefile   |   3 +
  kernel/bpf/helpers.c  |   4 +
  kernel/bpf/relaymap.c | 213 ++
  kernel/bpf/syscall.c  |   1 +
  kernel/bpf/verifier.c |   8 ++
  kernel/trace/bpf_trace.c  |   4 +
  9 files changed, 254 insertions(+)
  create mode 100644 kernel/bpf/relaymap.c

--
2.32.0.3.g01195cf9f





Re: [PATCH bpf-next 3/3] bpf: introduce bpf_relay_output helper

2023-12-22 Thread Philo Lu




On 2023/12/22 22:46, Jiri Olsa wrote:

On Fri, Dec 22, 2023 at 08:21:46PM +0800, Philo Lu wrote:

Like perfbuf/ringbuf, a helper is needed to write into the buffer, named
bpf_relay_output, whose usage is same as ringbuf. Note that it works only
after relay files are set, i.e., after calling map_update_elem for the
created relay map.


we can't add helpers anymore, this will need to be kfunc
check functions marked with __bpf_kfunc as examples



Got it, I will change it to kfunc in the next version.

Thanks.


jirka



Signed-off-by: Philo Lu 
---
  include/linux/bpf.h  |  1 +
  include/uapi/linux/bpf.h | 10 ++
  kernel/bpf/helpers.c |  4 
  kernel/bpf/relaymap.c| 26 ++
  kernel/bpf/verifier.c|  8 
  kernel/trace/bpf_trace.c |  4 
  6 files changed, 53 insertions(+)

diff --git a/include/linux/bpf.h b/include/linux/bpf.h
index 7671530d6e4e..b177122369e6 100644
--- a/include/linux/bpf.h
+++ b/include/linux/bpf.h
@@ -3095,6 +3095,7 @@ extern const struct bpf_func_proto bpf_get_retval_proto;
  extern const struct bpf_func_proto bpf_user_ringbuf_drain_proto;
  extern const struct bpf_func_proto bpf_cgrp_storage_get_proto;
  extern const struct bpf_func_proto bpf_cgrp_storage_delete_proto;
+extern const struct bpf_func_proto bpf_relay_output_proto;
  
  const struct bpf_func_proto *tracing_prog_func_proto(

enum bpf_func_id func_id, const struct bpf_prog *prog);
diff --git a/include/uapi/linux/bpf.h b/include/uapi/linux/bpf.h
index 143b75676bd3..03c0c1953ba1 100644
--- a/include/uapi/linux/bpf.h
+++ b/include/uapi/linux/bpf.h
@@ -5686,6 +5686,15 @@ union bpf_attr {
   *0 on success.
   *
   ***-ENOENT** if the bpf_local_storage cannot be found.
+ *
+ * long bpf_relay_output(void *map, void *data, u64 size, u64 flags)
+ * Description
+ * Copy *size* bytes from *data* into *map* of type 
BPF_MAP_TYPE_RELAY.
+ * Currently, the *flags* must be 0.
+ * Return
+ * 0 on success.
+ *
+ * **-ENOENT** if the relay base_file in debugfs cannot be found.
   */
  #define ___BPF_FUNC_MAPPER(FN, ctx...)\
FN(unspec, 0, ##ctx)\
@@ -5900,6 +5909,7 @@ union bpf_attr {
FN(user_ringbuf_drain, 209, ##ctx)  \
FN(cgrp_storage_get, 210, ##ctx)\
FN(cgrp_storage_delete, 211, ##ctx) \
+   FN(relay_output, 212, ##ctx)\
/* */
  
  /* backwards-compatibility macros for users of __BPF_FUNC_MAPPER that don't

diff --git a/kernel/bpf/helpers.c b/kernel/bpf/helpers.c
index be72824f32b2..0c26e87ce729 100644
--- a/kernel/bpf/helpers.c
+++ b/kernel/bpf/helpers.c
@@ -1720,6 +1720,10 @@ bpf_base_func_proto(enum bpf_func_id func_id)
return _ringbuf_discard_proto;
case BPF_FUNC_ringbuf_query:
return _ringbuf_query_proto;
+#ifdef CONFIG_RELAY
+   case BPF_FUNC_relay_output:
+   return _relay_output_proto;
+#endif
case BPF_FUNC_strncmp:
return _strncmp_proto;
case BPF_FUNC_strtol:
diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
index 588c8de0a4bd..f9e2e4a780df 100644
--- a/kernel/bpf/relaymap.c
+++ b/kernel/bpf/relaymap.c
@@ -173,6 +173,32 @@ static u64 relay_map_mem_usage(const struct bpf_map *map)
return usage;
  }
  
+BPF_CALL_4(bpf_relay_output, struct bpf_map *, map, void *, data, u64, size,

+  u64, flags)
+{
+   struct bpf_relay_map *rmap;
+
+   /* not support any flag now */
+   if (unlikely(flags))
+   return -EINVAL;
+
+   rmap = container_of(map, struct bpf_relay_map, map);
+   if (!rmap->relay_chan->has_base_filename)
+   return -ENOENT;
+
+   relay_write(rmap->relay_chan, data, size);
+   return 0;
+}
+
+const struct bpf_func_proto bpf_relay_output_proto = {
+   .func   = bpf_relay_output,
+   .ret_type   = RET_INTEGER,
+   .arg1_type  = ARG_CONST_MAP_PTR,
+   .arg2_type  = ARG_PTR_TO_MEM | MEM_RDONLY,
+   .arg3_type  = ARG_CONST_SIZE_OR_ZERO,
+   .arg4_type  = ARG_ANYTHING,
+};
+
  BTF_ID_LIST_SINGLE(relay_map_btf_ids, struct, bpf_relay_map)
  const struct bpf_map_ops relay_map_ops = {
.map_meta_equal = bpf_map_meta_equal,
diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c
index f13008d27f35..8c8287d6ae18 100644
--- a/kernel/bpf/verifier.c
+++ b/kernel/bpf/verifier.c
@@ -8800,6 +8800,10 @@ static int check_map_func_compatibility(struct 
bpf_verifier_env *env,
if (func_id != BPF_FUNC_user_ringbuf_drain)
goto error;
break;
+   case BPF_MAP_TYPE_RELAY:
+   if (func_id != BPF_FUNC_relay_output)
+   goto error;
+   break;
case BPF_MAP_TYPE_STACK_TRACE:
if (func_id != BPF_FUNC_get_stackid)
 

Re: [PATCH bpf-next 2/3] bpf: implement map_update_elem to init relay file

2023-12-22 Thread Philo Lu




On 2023/12/22 22:45, Jiri Olsa wrote:

On Fri, Dec 22, 2023 at 08:21:45PM +0800, Philo Lu wrote:

map_update_elem is used to create relay files and bind them with the
relay channel, which is created with BPF_MAP_CREATE. This allows users
to set a custom directory name. It must be used with key=NULL and
flag=0.

Here is an example:
```
struct {
__uint(type, BPF_MAP_TYPE_RELAY);
__uint(max_entries, 4096);
} my_relay SEC(".maps");
...
char dir_name[] = "relay_test";
bpf_map_update_elem(map_fd, NULL, dir_name, 0);
```

Then, directory `/sys/kerenl/debug/relay_test` will be created, which
includes files of my_relay0...my_relay[#cpu]. Each represents a per-cpu
buffer with size 8 * 4096 B (there are 8 subbufs by default, each with
size 4096B).

Signed-off-by: Philo Lu 
---
  kernel/bpf/relaymap.c | 32 +++-
  1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
index d0adc7f67758..588c8de0a4bd 100644
--- a/kernel/bpf/relaymap.c
+++ b/kernel/bpf/relaymap.c
@@ -117,7 +117,37 @@ static void *relay_map_lookup_elem(struct bpf_map *map, 
void *key)
  static long relay_map_update_elem(struct bpf_map *map, void *key, void *value,
   u64 flags)
  {
-   return -EOPNOTSUPP;
+   struct bpf_relay_map *rmap;
+   struct dentry *parent;
+   int err;
+
+   if (unlikely(flags))
+   return -EINVAL;
+
+   if (unlikely(key))
+   return -EINVAL;
+
+   rmap = container_of(map, struct bpf_relay_map, map);
+
+   /* The directory already exists */
+   if (rmap->relay_chan->has_base_filename)
+   return -EEXIST;
+
+   /* Setup relay files. Note that the directory name passed as value 
should
+* not be longer than map->value_size, including the '\0' at the end.
+*/
+   ((char *)value)[map->value_size - 1] = '\0';
+   parent = debugfs_create_dir(value, NULL);
+   if (IS_ERR_OR_NULL(parent))
+   return PTR_ERR(parent);
+
+   err = relay_late_setup_files(rmap->relay_chan, map->name, parent);
+   if (err) {
+   debugfs_remove_recursive(parent);
+   return err;
+   }


looks like this patch could be moved to the previous one?



OK, will do it.


jirka


+
+   return 0;
  }
  
  static long relay_map_delete_elem(struct bpf_map *map, void *key)

--
2.32.0.3.g01195cf9f






Re: [PATCH bpf-next 1/3] bpf: implement relay map basis

2023-12-22 Thread Philo Lu




On 2023/12/22 22:45, Jiri Olsa wrote:

On Fri, Dec 22, 2023 at 08:21:44PM +0800, Philo Lu wrote:

SNIP


+/* bpf_attr is used as follows:
+ * - key size: must be 0
+ * - value size: value will be used as directory name by map_update_elem
+ *   (to create relay files). If passed as 0, it will be set to NAME_MAX as
+ *   default
+ *
+ * - max_entries: subbuf size
+ * - map_extra: subbuf num, default as 8
+ *
+ * When alloc, we do not set up relay files considering dir_name conflicts.
+ * Instead we use relay_late_setup_files() in map_update_elem(), and thus the
+ * value is used as dir_name, and map->name is used as base_filename.
+ */
+static struct bpf_map *relay_map_alloc(union bpf_attr *attr)
+{
+   struct bpf_relay_map *rmap;
+
+   if (unlikely(attr->map_flags & ~RELAY_CREATE_FLAG_MASK))
+   return ERR_PTR(-EINVAL);
+
+   /* key size must be 0 in relay map */
+   if (unlikely(attr->key_size))
+   return ERR_PTR(-EINVAL);
+
+   if (unlikely(attr->value_size > NAME_MAX)) {
+   pr_warn("value_size should be no more than %d\n", NAME_MAX);
+   return ERR_PTR(-EINVAL);
+   } else if (attr->value_size == 0)
+   attr->value_size = NAME_MAX;


the concept of no key with just value seems strange.. I never worked
with relay channels, so perhaps stupid question: but why not have one
relay channel for given key? having the debugfs like:

   /sys/kernel/debug/my_rmap/mychannel/

Here, a relay map is actually a relay channel, which includes buffers 
for all cpus. And I think 2 levels is enough when we use relay map in 
`/sys/kernel/debug/`: /[#cpu]. The `dir_name` is 
necessary because user could use the same `map_name` in different bpf 
programs, and we can use it as an additional tag to distinguish them. 
The `dir_name` is set by user with relay_map_update_elem.


Here is an example. Assume we have 2 relay maps (rmap_a and rmap_b) and 
2 cpus, the debugfs will be like:

```
/sys/kernel/debug//rmap_a0
/sys/kernel/debug//rmap_a1
/sys/kernel/debug//rmap_b0
/sys/kernel/debug//rmap_b1
```

So I think the key point here is that we just need one field to set the 
`dir_name`, either key or value. I chose key as NULL because I think it 
suggests "Normally map_update_elem should be invoked just once for a 
relay map". But I think it okay to use key instead, and value as NULL.



+
+   /* set default subbuf num */
+   attr->map_extra = attr->map_extra & UINT_MAX;
+   if (!attr->map_extra)
+   attr->map_extra = 8;
+
+   if (!attr->map_name || strlen(attr->map_name) == 0)


attr->map_name is allways != NULL


+   return ERR_PTR(-EINVAL);
+
+   rmap = bpf_map_area_alloc(sizeof(*rmap), NUMA_NO_NODE);
+   if (!rmap)
+   return ERR_PTR(-ENOMEM);
+
+   bpf_map_init_from_attr(>map, attr);
+
+   rmap->relay_cb.create_buf_file = create_buf_file_handler;
+   rmap->relay_cb.remove_buf_file = remove_buf_file_handler;
+   if (attr->map_flags & BPF_F_OVERWRITE)
+   rmap->relay_cb.subbuf_start = subbuf_start_overwrite;
+
+   rmap->relay_chan = relay_open(NULL, NULL,
+   attr->max_entries, 
attr->map_extra,
+   >relay_cb, NULL);


wrong indentation


Got it. I will adjust it.


+   if (!rmap->relay_chan)
+   return ERR_PTR(-EINVAL);
+
+   return >map;
+}
+
+static void relay_map_free(struct bpf_map *map)
+{
+   struct bpf_relay_map *rmap;
+
+   rmap = container_of(map, struct bpf_relay_map, map);
+   relay_close(rmap->relay_chan);
+   debugfs_remove_recursive(rmap->relay_chan->parent);
+   kfree(rmap);


should you use bpf_map_area_free instead?


Thanks for catching. Will fix it.


jirka


+}
+
+static void *relay_map_lookup_elem(struct bpf_map *map, void *key)
+{
+   return ERR_PTR(-EOPNOTSUPP);
+}
+
+static long relay_map_update_elem(struct bpf_map *map, void *key, void *value,
+  u64 flags)
+{
+   return -EOPNOTSUPP;
+}
+
+static long relay_map_delete_elem(struct bpf_map *map, void *key)
+{
+   return -EOPNOTSUPP;
+}
+
+static int relay_map_get_next_key(struct bpf_map *map, void *key,
+   void *next_key)
+{
+   return -EOPNOTSUPP;
+}
+
+static u64 relay_map_mem_usage(const struct bpf_map *map)
+{
+   struct bpf_relay_map *rmap;
+   u64 usage = sizeof(struct bpf_relay_map);
+
+   rmap = container_of(map, struct bpf_relay_map, map);
+   usage += sizeof(struct rchan);
+   usage += (sizeof(struct rchan_buf) + rmap->relay_chan->alloc_size)
+* num_online_cpus();
+   return usage;
+}
+
+BTF_ID_LIST_SINGLE(relay_map_btf_ids, struct, bpf_relay_map)
+const struct bpf_map_ops relay_map_ops = {
+   .map_meta_equal = bpf_map_meta_equal,
+   .map_alloc = relay_map_alloc,
+   .map_free = 

Re: [PATCH bpf-next 1/3] bpf: implement relay map basis

2023-12-22 Thread kernel test robot
Hi Philo,

kernel test robot noticed the following build warnings:

[auto build test WARNING on bpf-next/master]

url:
https://github.com/intel-lab-lkp/linux/commits/Philo-Lu/bpf-implement-relay-map-basis/20231222-204545
base:   https://git.kernel.org/pub/scm/linux/kernel/git/bpf/bpf-next.git master
patch link:
https://lore.kernel.org/r/20231222122146.65519-2-lulie%40linux.alibaba.com
patch subject: [PATCH bpf-next 1/3] bpf: implement relay map basis
config: x86_64-kexec 
(https://download.01.org/0day-ci/archive/20231223/202312230850.nqeizxj6-...@intel.com/config)
compiler: gcc-12 (Debian 12.2.0-14) 12.2.0
reproduce (this is a W=1 build): 
(https://download.01.org/0day-ci/archive/20231223/202312230850.nqeizxj6-...@intel.com/reproduce)

If you fix the issue in a separate patch/commit (i.e. not just a new version of
the same patch/commit), kindly add following tags
| Reported-by: kernel test robot 
| Closes: 
https://lore.kernel.org/oe-kbuild-all/202312230850.nqeizxj6-...@intel.com/

All warnings (new ones prefixed by >>):

   kernel/bpf/relaymap.c: In function 'relay_map_alloc':
>> kernel/bpf/relaymap.c:79:13: warning: the comparison will always evaluate as 
>> 'true' for the address of 'map_name' will never be NULL [-Waddress]
  79 | if (!attr->map_name || strlen(attr->map_name) == 0)
 | ^
   In file included from include/linux/bpf.h:7,
from include/linux/filter.h:9,
from kernel/bpf/relaymap.c:3:
   include/uapi/linux/bpf.h:1394:25: note: 'map_name' declared here
1394 | charmap_name[BPF_OBJ_NAME_LEN];
 | ^~~~


vim +79 kernel/bpf/relaymap.c

43  
44  /* bpf_attr is used as follows:
45   * - key size: must be 0
46   * - value size: value will be used as directory name by map_update_elem
47   *   (to create relay files). If passed as 0, it will be set to 
NAME_MAX as
48   *   default
49   *
50   * - max_entries: subbuf size
51   * - map_extra: subbuf num, default as 8
52   *
53   * When alloc, we do not set up relay files considering dir_name 
conflicts.
54   * Instead we use relay_late_setup_files() in map_update_elem(), and 
thus the
55   * value is used as dir_name, and map->name is used as base_filename.
56   */
57  static struct bpf_map *relay_map_alloc(union bpf_attr *attr)
58  {
59  struct bpf_relay_map *rmap;
60  
61  if (unlikely(attr->map_flags & ~RELAY_CREATE_FLAG_MASK))
62  return ERR_PTR(-EINVAL);
63  
64  /* key size must be 0 in relay map */
65  if (unlikely(attr->key_size))
66  return ERR_PTR(-EINVAL);
67  
68  if (unlikely(attr->value_size > NAME_MAX)) {
69  pr_warn("value_size should be no more than %d\n", 
NAME_MAX);
70  return ERR_PTR(-EINVAL);
71  } else if (attr->value_size == 0)
72  attr->value_size = NAME_MAX;
73  
74  /* set default subbuf num */
75  attr->map_extra = attr->map_extra & UINT_MAX;
76  if (!attr->map_extra)
77  attr->map_extra = 8;
78  
  > 79  if (!attr->map_name || strlen(attr->map_name) == 0)
80  return ERR_PTR(-EINVAL);
81  
82  rmap = bpf_map_area_alloc(sizeof(*rmap), NUMA_NO_NODE);
83  if (!rmap)
84  return ERR_PTR(-ENOMEM);
85  
86  bpf_map_init_from_attr(>map, attr);
87  
88  rmap->relay_cb.create_buf_file = create_buf_file_handler;
89  rmap->relay_cb.remove_buf_file = remove_buf_file_handler;
90  if (attr->map_flags & BPF_F_OVERWRITE)
91  rmap->relay_cb.subbuf_start = subbuf_start_overwrite;
92  
93  rmap->relay_chan = relay_open(NULL, NULL,
94  
attr->max_entries, attr->map_extra,
95  
>relay_cb, NULL);
96  if (!rmap->relay_chan)
97  return ERR_PTR(-EINVAL);
98  
99  return >map;
   100  }
   101  

-- 
0-DAY CI Kernel Test Service
https://github.com/intel/lkp-tests/wiki



Re: [PATCH bpf-next 1/3] bpf: implement relay map basis

2023-12-22 Thread kernel test robot
Hi Philo,

kernel test robot noticed the following build warnings:

[auto build test WARNING on bpf-next/master]

url:
https://github.com/intel-lab-lkp/linux/commits/Philo-Lu/bpf-implement-relay-map-basis/20231222-204545
base:   https://git.kernel.org/pub/scm/linux/kernel/git/bpf/bpf-next.git master
patch link:
https://lore.kernel.org/r/20231222122146.65519-2-lulie%40linux.alibaba.com
patch subject: [PATCH bpf-next 1/3] bpf: implement relay map basis
config: arm-randconfig-001-20231223 
(https://download.01.org/0day-ci/archive/20231223/202312230638.iydd8joz-...@intel.com/config)
compiler: clang version 18.0.0git (https://github.com/llvm/llvm-project 
d3ef86708241a3bee902615c190dead1638c4e09)
reproduce (this is a W=1 build): 
(https://download.01.org/0day-ci/archive/20231223/202312230638.iydd8joz-...@intel.com/reproduce)

If you fix the issue in a separate patch/commit (i.e. not just a new version of
the same patch/commit), kindly add following tags
| Reported-by: kernel test robot 
| Closes: 
https://lore.kernel.org/oe-kbuild-all/202312230638.iydd8joz-...@intel.com/

All warnings (new ones prefixed by >>):

>> kernel/bpf/relaymap.c:79:13: warning: address of array 'attr->map_name' will 
>> always evaluate to 'true' [-Wpointer-bool-conversion]
  79 | if (!attr->map_name || strlen(attr->map_name) == 0)
 | ~~~^~~~
   1 warning generated.


vim +79 kernel/bpf/relaymap.c

43  
44  /* bpf_attr is used as follows:
45   * - key size: must be 0
46   * - value size: value will be used as directory name by map_update_elem
47   *   (to create relay files). If passed as 0, it will be set to 
NAME_MAX as
48   *   default
49   *
50   * - max_entries: subbuf size
51   * - map_extra: subbuf num, default as 8
52   *
53   * When alloc, we do not set up relay files considering dir_name 
conflicts.
54   * Instead we use relay_late_setup_files() in map_update_elem(), and 
thus the
55   * value is used as dir_name, and map->name is used as base_filename.
56   */
57  static struct bpf_map *relay_map_alloc(union bpf_attr *attr)
58  {
59  struct bpf_relay_map *rmap;
60  
61  if (unlikely(attr->map_flags & ~RELAY_CREATE_FLAG_MASK))
62  return ERR_PTR(-EINVAL);
63  
64  /* key size must be 0 in relay map */
65  if (unlikely(attr->key_size))
66  return ERR_PTR(-EINVAL);
67  
68  if (unlikely(attr->value_size > NAME_MAX)) {
69  pr_warn("value_size should be no more than %d\n", 
NAME_MAX);
70  return ERR_PTR(-EINVAL);
71  } else if (attr->value_size == 0)
72  attr->value_size = NAME_MAX;
73  
74  /* set default subbuf num */
75  attr->map_extra = attr->map_extra & UINT_MAX;
76  if (!attr->map_extra)
77  attr->map_extra = 8;
78  
  > 79  if (!attr->map_name || strlen(attr->map_name) == 0)
80  return ERR_PTR(-EINVAL);
81  
82  rmap = bpf_map_area_alloc(sizeof(*rmap), NUMA_NO_NODE);
83  if (!rmap)
84  return ERR_PTR(-ENOMEM);
85  
86  bpf_map_init_from_attr(>map, attr);
87  
88  rmap->relay_cb.create_buf_file = create_buf_file_handler;
89  rmap->relay_cb.remove_buf_file = remove_buf_file_handler;
90  if (attr->map_flags & BPF_F_OVERWRITE)
91  rmap->relay_cb.subbuf_start = subbuf_start_overwrite;
92  
93  rmap->relay_chan = relay_open(NULL, NULL,
94  
attr->max_entries, attr->map_extra,
95  
>relay_cb, NULL);
96  if (!rmap->relay_chan)
97  return ERR_PTR(-EINVAL);
98  
99  return >map;
   100  }
   101  

-- 
0-DAY CI Kernel Test Service
https://github.com/intel/lkp-tests/wiki



Re: [PATCH bpf-next 3/3] bpf: introduce bpf_relay_output helper

2023-12-22 Thread Jiri Olsa
On Fri, Dec 22, 2023 at 08:21:46PM +0800, Philo Lu wrote:
> Like perfbuf/ringbuf, a helper is needed to write into the buffer, named
> bpf_relay_output, whose usage is same as ringbuf. Note that it works only
> after relay files are set, i.e., after calling map_update_elem for the
> created relay map.

we can't add helpers anymore, this will need to be kfunc
check functions marked with __bpf_kfunc as examples

jirka

> 
> Signed-off-by: Philo Lu 
> ---
>  include/linux/bpf.h  |  1 +
>  include/uapi/linux/bpf.h | 10 ++
>  kernel/bpf/helpers.c |  4 
>  kernel/bpf/relaymap.c| 26 ++
>  kernel/bpf/verifier.c|  8 
>  kernel/trace/bpf_trace.c |  4 
>  6 files changed, 53 insertions(+)
> 
> diff --git a/include/linux/bpf.h b/include/linux/bpf.h
> index 7671530d6e4e..b177122369e6 100644
> --- a/include/linux/bpf.h
> +++ b/include/linux/bpf.h
> @@ -3095,6 +3095,7 @@ extern const struct bpf_func_proto bpf_get_retval_proto;
>  extern const struct bpf_func_proto bpf_user_ringbuf_drain_proto;
>  extern const struct bpf_func_proto bpf_cgrp_storage_get_proto;
>  extern const struct bpf_func_proto bpf_cgrp_storage_delete_proto;
> +extern const struct bpf_func_proto bpf_relay_output_proto;
>  
>  const struct bpf_func_proto *tracing_prog_func_proto(
>enum bpf_func_id func_id, const struct bpf_prog *prog);
> diff --git a/include/uapi/linux/bpf.h b/include/uapi/linux/bpf.h
> index 143b75676bd3..03c0c1953ba1 100644
> --- a/include/uapi/linux/bpf.h
> +++ b/include/uapi/linux/bpf.h
> @@ -5686,6 +5686,15 @@ union bpf_attr {
>   *   0 on success.
>   *
>   *   **-ENOENT** if the bpf_local_storage cannot be found.
> + *
> + * long bpf_relay_output(void *map, void *data, u64 size, u64 flags)
> + *   Description
> + *   Copy *size* bytes from *data* into *map* of type 
> BPF_MAP_TYPE_RELAY.
> + *   Currently, the *flags* must be 0.
> + *   Return
> + *   0 on success.
> + *
> + *   **-ENOENT** if the relay base_file in debugfs cannot be found.
>   */
>  #define ___BPF_FUNC_MAPPER(FN, ctx...)   \
>   FN(unspec, 0, ##ctx)\
> @@ -5900,6 +5909,7 @@ union bpf_attr {
>   FN(user_ringbuf_drain, 209, ##ctx)  \
>   FN(cgrp_storage_get, 210, ##ctx)\
>   FN(cgrp_storage_delete, 211, ##ctx) \
> + FN(relay_output, 212, ##ctx)\
>   /* */
>  
>  /* backwards-compatibility macros for users of __BPF_FUNC_MAPPER that don't
> diff --git a/kernel/bpf/helpers.c b/kernel/bpf/helpers.c
> index be72824f32b2..0c26e87ce729 100644
> --- a/kernel/bpf/helpers.c
> +++ b/kernel/bpf/helpers.c
> @@ -1720,6 +1720,10 @@ bpf_base_func_proto(enum bpf_func_id func_id)
>   return _ringbuf_discard_proto;
>   case BPF_FUNC_ringbuf_query:
>   return _ringbuf_query_proto;
> +#ifdef CONFIG_RELAY
> + case BPF_FUNC_relay_output:
> + return _relay_output_proto;
> +#endif
>   case BPF_FUNC_strncmp:
>   return _strncmp_proto;
>   case BPF_FUNC_strtol:
> diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
> index 588c8de0a4bd..f9e2e4a780df 100644
> --- a/kernel/bpf/relaymap.c
> +++ b/kernel/bpf/relaymap.c
> @@ -173,6 +173,32 @@ static u64 relay_map_mem_usage(const struct bpf_map *map)
>   return usage;
>  }
>  
> +BPF_CALL_4(bpf_relay_output, struct bpf_map *, map, void *, data, u64, size,
> +u64, flags)
> +{
> + struct bpf_relay_map *rmap;
> +
> + /* not support any flag now */
> + if (unlikely(flags))
> + return -EINVAL;
> +
> + rmap = container_of(map, struct bpf_relay_map, map);
> + if (!rmap->relay_chan->has_base_filename)
> + return -ENOENT;
> +
> + relay_write(rmap->relay_chan, data, size);
> + return 0;
> +}
> +
> +const struct bpf_func_proto bpf_relay_output_proto = {
> + .func   = bpf_relay_output,
> + .ret_type   = RET_INTEGER,
> + .arg1_type  = ARG_CONST_MAP_PTR,
> + .arg2_type  = ARG_PTR_TO_MEM | MEM_RDONLY,
> + .arg3_type  = ARG_CONST_SIZE_OR_ZERO,
> + .arg4_type  = ARG_ANYTHING,
> +};
> +
>  BTF_ID_LIST_SINGLE(relay_map_btf_ids, struct, bpf_relay_map)
>  const struct bpf_map_ops relay_map_ops = {
>   .map_meta_equal = bpf_map_meta_equal,
> diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c
> index f13008d27f35..8c8287d6ae18 100644
> --- a/kernel/bpf/verifier.c
> +++ b/kernel/bpf/verifier.c
> @@ -8800,6 +8800,10 @@ static int check_map_func_compatibility(struct 
> bpf_verifier_env *env,
>   if (func_id != BPF_FUNC_user_ringbuf_drain)
>   goto error;
>   break;
> + case BPF_MAP_TYPE_RELAY:
> + if (func_id != BPF_FUNC_relay_output)
> + goto error;
> + break;
>   case BPF_MAP_TYPE_STACK_TRACE:
>   if (func_id != BPF_FUNC_get_stackid)

Re: [PATCH bpf-next 2/3] bpf: implement map_update_elem to init relay file

2023-12-22 Thread Jiri Olsa
On Fri, Dec 22, 2023 at 08:21:45PM +0800, Philo Lu wrote:
> map_update_elem is used to create relay files and bind them with the
> relay channel, which is created with BPF_MAP_CREATE. This allows users
> to set a custom directory name. It must be used with key=NULL and
> flag=0.
> 
> Here is an example:
> ```
> struct {
> __uint(type, BPF_MAP_TYPE_RELAY);
> __uint(max_entries, 4096);
> } my_relay SEC(".maps");
> ...
> char dir_name[] = "relay_test";
> bpf_map_update_elem(map_fd, NULL, dir_name, 0);
> ```
> 
> Then, directory `/sys/kerenl/debug/relay_test` will be created, which
> includes files of my_relay0...my_relay[#cpu]. Each represents a per-cpu
> buffer with size 8 * 4096 B (there are 8 subbufs by default, each with
> size 4096B).
> 
> Signed-off-by: Philo Lu 
> ---
>  kernel/bpf/relaymap.c | 32 +++-
>  1 file changed, 31 insertions(+), 1 deletion(-)
> 
> diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
> index d0adc7f67758..588c8de0a4bd 100644
> --- a/kernel/bpf/relaymap.c
> +++ b/kernel/bpf/relaymap.c
> @@ -117,7 +117,37 @@ static void *relay_map_lookup_elem(struct bpf_map *map, 
> void *key)
>  static long relay_map_update_elem(struct bpf_map *map, void *key, void 
> *value,
>  u64 flags)
>  {
> - return -EOPNOTSUPP;
> + struct bpf_relay_map *rmap;
> + struct dentry *parent;
> + int err;
> +
> + if (unlikely(flags))
> + return -EINVAL;
> +
> + if (unlikely(key))
> + return -EINVAL;
> +
> + rmap = container_of(map, struct bpf_relay_map, map);
> +
> + /* The directory already exists */
> + if (rmap->relay_chan->has_base_filename)
> + return -EEXIST;
> +
> + /* Setup relay files. Note that the directory name passed as value 
> should
> +  * not be longer than map->value_size, including the '\0' at the end.
> +  */
> + ((char *)value)[map->value_size - 1] = '\0';
> + parent = debugfs_create_dir(value, NULL);
> + if (IS_ERR_OR_NULL(parent))
> + return PTR_ERR(parent);
> +
> + err = relay_late_setup_files(rmap->relay_chan, map->name, parent);
> + if (err) {
> + debugfs_remove_recursive(parent);
> + return err;
> + }

looks like this patch could be moved to the previous one?

jirka

> +
> + return 0;
>  }
>  
>  static long relay_map_delete_elem(struct bpf_map *map, void *key)
> -- 
> 2.32.0.3.g01195cf9f
> 
> 



Re: [PATCH bpf-next 1/3] bpf: implement relay map basis

2023-12-22 Thread Jiri Olsa
On Fri, Dec 22, 2023 at 08:21:44PM +0800, Philo Lu wrote:

SNIP

> +/* bpf_attr is used as follows:
> + * - key size: must be 0
> + * - value size: value will be used as directory name by map_update_elem
> + *   (to create relay files). If passed as 0, it will be set to NAME_MAX as
> + *   default
> + *
> + * - max_entries: subbuf size
> + * - map_extra: subbuf num, default as 8
> + *
> + * When alloc, we do not set up relay files considering dir_name conflicts.
> + * Instead we use relay_late_setup_files() in map_update_elem(), and thus the
> + * value is used as dir_name, and map->name is used as base_filename.
> + */
> +static struct bpf_map *relay_map_alloc(union bpf_attr *attr)
> +{
> + struct bpf_relay_map *rmap;
> +
> + if (unlikely(attr->map_flags & ~RELAY_CREATE_FLAG_MASK))
> + return ERR_PTR(-EINVAL);
> +
> + /* key size must be 0 in relay map */
> + if (unlikely(attr->key_size))
> + return ERR_PTR(-EINVAL);
> +
> + if (unlikely(attr->value_size > NAME_MAX)) {
> + pr_warn("value_size should be no more than %d\n", NAME_MAX);
> + return ERR_PTR(-EINVAL);
> + } else if (attr->value_size == 0)
> + attr->value_size = NAME_MAX;

the concept of no key with just value seems strange.. I never worked
with relay channels, so perhaps stupid question: but why not have one
relay channel for given key? having the debugfs like:

  /sys/kernel/debug/my_rmap/mychannel/

> +
> + /* set default subbuf num */
> + attr->map_extra = attr->map_extra & UINT_MAX;
> + if (!attr->map_extra)
> + attr->map_extra = 8;
> +
> + if (!attr->map_name || strlen(attr->map_name) == 0)

attr->map_name is allways != NULL

> + return ERR_PTR(-EINVAL);
> +
> + rmap = bpf_map_area_alloc(sizeof(*rmap), NUMA_NO_NODE);
> + if (!rmap)
> + return ERR_PTR(-ENOMEM);
> +
> + bpf_map_init_from_attr(>map, attr);
> +
> + rmap->relay_cb.create_buf_file = create_buf_file_handler;
> + rmap->relay_cb.remove_buf_file = remove_buf_file_handler;
> + if (attr->map_flags & BPF_F_OVERWRITE)
> + rmap->relay_cb.subbuf_start = subbuf_start_overwrite;
> +
> + rmap->relay_chan = relay_open(NULL, NULL,
> + attr->max_entries, 
> attr->map_extra,
> + >relay_cb, NULL);

wrong indentation

> + if (!rmap->relay_chan)
> + return ERR_PTR(-EINVAL);
> +
> + return >map;
> +}
> +
> +static void relay_map_free(struct bpf_map *map)
> +{
> + struct bpf_relay_map *rmap;
> +
> + rmap = container_of(map, struct bpf_relay_map, map);
> + relay_close(rmap->relay_chan);
> + debugfs_remove_recursive(rmap->relay_chan->parent);
> + kfree(rmap);

should you use bpf_map_area_free instead?

jirka

> +}
> +
> +static void *relay_map_lookup_elem(struct bpf_map *map, void *key)
> +{
> + return ERR_PTR(-EOPNOTSUPP);
> +}
> +
> +static long relay_map_update_elem(struct bpf_map *map, void *key, void 
> *value,
> +u64 flags)
> +{
> + return -EOPNOTSUPP;
> +}
> +
> +static long relay_map_delete_elem(struct bpf_map *map, void *key)
> +{
> + return -EOPNOTSUPP;
> +}
> +
> +static int relay_map_get_next_key(struct bpf_map *map, void *key,
> + void *next_key)
> +{
> + return -EOPNOTSUPP;
> +}
> +
> +static u64 relay_map_mem_usage(const struct bpf_map *map)
> +{
> + struct bpf_relay_map *rmap;
> + u64 usage = sizeof(struct bpf_relay_map);
> +
> + rmap = container_of(map, struct bpf_relay_map, map);
> + usage += sizeof(struct rchan);
> + usage += (sizeof(struct rchan_buf) + rmap->relay_chan->alloc_size)
> +  * num_online_cpus();
> + return usage;
> +}
> +
> +BTF_ID_LIST_SINGLE(relay_map_btf_ids, struct, bpf_relay_map)
> +const struct bpf_map_ops relay_map_ops = {
> + .map_meta_equal = bpf_map_meta_equal,
> + .map_alloc = relay_map_alloc,
> + .map_free = relay_map_free,
> + .map_lookup_elem = relay_map_lookup_elem,
> + .map_update_elem = relay_map_update_elem,
> + .map_delete_elem = relay_map_delete_elem,
> + .map_get_next_key = relay_map_get_next_key,
> + .map_mem_usage = relay_map_mem_usage,
> + .map_btf_id = _map_btf_ids[0],
> +};
> diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
> index 1bf9805ee185..35ae54ac6736 100644
> --- a/kernel/bpf/syscall.c
> +++ b/kernel/bpf/syscall.c
> @@ -1147,6 +1147,7 @@ static int map_create(union bpf_attr *attr)
>   }
>  
>   if (attr->map_type != BPF_MAP_TYPE_BLOOM_FILTER &&
> + attr->map_type != BPF_MAP_TYPE_RELAY &&
>   attr->map_extra != 0)
>   return -EINVAL;
>  
> -- 
> 2.32.0.3.g01195cf9f
> 



Re: [PATCH bpf-next 0/3] bpf: introduce BPF_MAP_TYPE_RELAY

2023-12-22 Thread Jiri Olsa
On Fri, Dec 22, 2023 at 08:21:43PM +0800, Philo Lu wrote:
> The patch set introduce a new type of map, BPF_MAP_TYPE_RELAY, based on
> relay interface [0]. It provides a way for persistent and overwritable data
> transfer.
> 
> As stated in [0], relay is a efficient method for log and data transfer.
> And the interface is simple enough so that we can implement and use this
> type of map with current map interfaces. Besides we need a new helper
> bpf_relay_output to output data to user, similar with bpf_ringbuf_output.
> 
> We need this map because currently neither ringbuf nor perfbuf satisfies
> the requirements of relatively long-term consistent tracing, where the bpf
> program keeps writing into the buffer without any bundled reader, and the
> buffer supports overwriting. For users, they just run the bpf program to
> collect data, and are able to read as need. The detailed discussion can be
> found at [1].
> 
> The buffer is exposed to users as per-cpu files in debugfs, supporting read
> and mmap, and it is up to users how to formulate and read it, either
> through a program with mmap or just `cat`. Specifically, the files are
> created as "/sys/kerenl/debug//#cpu", where the 
> is defined with map_update_elem (Note that we do not need to implement
> actual elem operators for relay_map).
> 
> If this map is acceptable, other parts including docs, libbpf support,
> selftests, and benchmarks (if need) will be added in the following version.

looks useful, selftests might be already helpful to see the usage

jirka

> 
> [0]
> https://github.com/torvalds/linux/blob/master/Documentation/filesystems/relay.rst
> [1]
> https://lore.kernel.org/bpf/20231219122850.433be...@gandalf.local.home/T/
> 
> Philo Lu (3):
>   bpf: implement relay map basis
>   bpf: implement map_update_elem to init relay file
>   bpf: introduce bpf_relay_output helper
> 
>  include/linux/bpf.h   |   1 +
>  include/linux/bpf_types.h |   3 +
>  include/uapi/linux/bpf.h  |  17 +++
>  kernel/bpf/Makefile   |   3 +
>  kernel/bpf/helpers.c  |   4 +
>  kernel/bpf/relaymap.c | 213 ++
>  kernel/bpf/syscall.c  |   1 +
>  kernel/bpf/verifier.c |   8 ++
>  kernel/trace/bpf_trace.c  |   4 +
>  9 files changed, 254 insertions(+)
>  create mode 100644 kernel/bpf/relaymap.c
> 
> --
> 2.32.0.3.g01195cf9f
> 



[PATCH bpf-next 3/3] bpf: introduce bpf_relay_output helper

2023-12-22 Thread Philo Lu
Like perfbuf/ringbuf, a helper is needed to write into the buffer, named
bpf_relay_output, whose usage is same as ringbuf. Note that it works only
after relay files are set, i.e., after calling map_update_elem for the
created relay map.

Signed-off-by: Philo Lu 
---
 include/linux/bpf.h  |  1 +
 include/uapi/linux/bpf.h | 10 ++
 kernel/bpf/helpers.c |  4 
 kernel/bpf/relaymap.c| 26 ++
 kernel/bpf/verifier.c|  8 
 kernel/trace/bpf_trace.c |  4 
 6 files changed, 53 insertions(+)

diff --git a/include/linux/bpf.h b/include/linux/bpf.h
index 7671530d6e4e..b177122369e6 100644
--- a/include/linux/bpf.h
+++ b/include/linux/bpf.h
@@ -3095,6 +3095,7 @@ extern const struct bpf_func_proto bpf_get_retval_proto;
 extern const struct bpf_func_proto bpf_user_ringbuf_drain_proto;
 extern const struct bpf_func_proto bpf_cgrp_storage_get_proto;
 extern const struct bpf_func_proto bpf_cgrp_storage_delete_proto;
+extern const struct bpf_func_proto bpf_relay_output_proto;
 
 const struct bpf_func_proto *tracing_prog_func_proto(
   enum bpf_func_id func_id, const struct bpf_prog *prog);
diff --git a/include/uapi/linux/bpf.h b/include/uapi/linux/bpf.h
index 143b75676bd3..03c0c1953ba1 100644
--- a/include/uapi/linux/bpf.h
+++ b/include/uapi/linux/bpf.h
@@ -5686,6 +5686,15 @@ union bpf_attr {
  * 0 on success.
  *
  * **-ENOENT** if the bpf_local_storage cannot be found.
+ *
+ * long bpf_relay_output(void *map, void *data, u64 size, u64 flags)
+ * Description
+ * Copy *size* bytes from *data* into *map* of type 
BPF_MAP_TYPE_RELAY.
+ * Currently, the *flags* must be 0.
+ * Return
+ * 0 on success.
+ *
+ * **-ENOENT** if the relay base_file in debugfs cannot be found.
  */
 #define ___BPF_FUNC_MAPPER(FN, ctx...) \
FN(unspec, 0, ##ctx)\
@@ -5900,6 +5909,7 @@ union bpf_attr {
FN(user_ringbuf_drain, 209, ##ctx)  \
FN(cgrp_storage_get, 210, ##ctx)\
FN(cgrp_storage_delete, 211, ##ctx) \
+   FN(relay_output, 212, ##ctx)\
/* */
 
 /* backwards-compatibility macros for users of __BPF_FUNC_MAPPER that don't
diff --git a/kernel/bpf/helpers.c b/kernel/bpf/helpers.c
index be72824f32b2..0c26e87ce729 100644
--- a/kernel/bpf/helpers.c
+++ b/kernel/bpf/helpers.c
@@ -1720,6 +1720,10 @@ bpf_base_func_proto(enum bpf_func_id func_id)
return _ringbuf_discard_proto;
case BPF_FUNC_ringbuf_query:
return _ringbuf_query_proto;
+#ifdef CONFIG_RELAY
+   case BPF_FUNC_relay_output:
+   return _relay_output_proto;
+#endif
case BPF_FUNC_strncmp:
return _strncmp_proto;
case BPF_FUNC_strtol:
diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
index 588c8de0a4bd..f9e2e4a780df 100644
--- a/kernel/bpf/relaymap.c
+++ b/kernel/bpf/relaymap.c
@@ -173,6 +173,32 @@ static u64 relay_map_mem_usage(const struct bpf_map *map)
return usage;
 }
 
+BPF_CALL_4(bpf_relay_output, struct bpf_map *, map, void *, data, u64, size,
+  u64, flags)
+{
+   struct bpf_relay_map *rmap;
+
+   /* not support any flag now */
+   if (unlikely(flags))
+   return -EINVAL;
+
+   rmap = container_of(map, struct bpf_relay_map, map);
+   if (!rmap->relay_chan->has_base_filename)
+   return -ENOENT;
+
+   relay_write(rmap->relay_chan, data, size);
+   return 0;
+}
+
+const struct bpf_func_proto bpf_relay_output_proto = {
+   .func   = bpf_relay_output,
+   .ret_type   = RET_INTEGER,
+   .arg1_type  = ARG_CONST_MAP_PTR,
+   .arg2_type  = ARG_PTR_TO_MEM | MEM_RDONLY,
+   .arg3_type  = ARG_CONST_SIZE_OR_ZERO,
+   .arg4_type  = ARG_ANYTHING,
+};
+
 BTF_ID_LIST_SINGLE(relay_map_btf_ids, struct, bpf_relay_map)
 const struct bpf_map_ops relay_map_ops = {
.map_meta_equal = bpf_map_meta_equal,
diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c
index f13008d27f35..8c8287d6ae18 100644
--- a/kernel/bpf/verifier.c
+++ b/kernel/bpf/verifier.c
@@ -8800,6 +8800,10 @@ static int check_map_func_compatibility(struct 
bpf_verifier_env *env,
if (func_id != BPF_FUNC_user_ringbuf_drain)
goto error;
break;
+   case BPF_MAP_TYPE_RELAY:
+   if (func_id != BPF_FUNC_relay_output)
+   goto error;
+   break;
case BPF_MAP_TYPE_STACK_TRACE:
if (func_id != BPF_FUNC_get_stackid)
goto error;
@@ -8932,6 +8936,10 @@ static int check_map_func_compatibility(struct 
bpf_verifier_env *env,
if (map->map_type != BPF_MAP_TYPE_USER_RINGBUF)
goto error;
break;
+   case BPF_FUNC_relay_output:
+   if (map->map_type != 

[PATCH bpf-next 0/3] bpf: introduce BPF_MAP_TYPE_RELAY

2023-12-22 Thread Philo Lu
The patch set introduce a new type of map, BPF_MAP_TYPE_RELAY, based on
relay interface [0]. It provides a way for persistent and overwritable data
transfer.

As stated in [0], relay is a efficient method for log and data transfer.
And the interface is simple enough so that we can implement and use this
type of map with current map interfaces. Besides we need a new helper
bpf_relay_output to output data to user, similar with bpf_ringbuf_output.

We need this map because currently neither ringbuf nor perfbuf satisfies
the requirements of relatively long-term consistent tracing, where the bpf
program keeps writing into the buffer without any bundled reader, and the
buffer supports overwriting. For users, they just run the bpf program to
collect data, and are able to read as need. The detailed discussion can be
found at [1].

The buffer is exposed to users as per-cpu files in debugfs, supporting read
and mmap, and it is up to users how to formulate and read it, either
through a program with mmap or just `cat`. Specifically, the files are
created as "/sys/kerenl/debug//#cpu", where the 
is defined with map_update_elem (Note that we do not need to implement
actual elem operators for relay_map).

If this map is acceptable, other parts including docs, libbpf support,
selftests, and benchmarks (if need) will be added in the following version.

[0]
https://github.com/torvalds/linux/blob/master/Documentation/filesystems/relay.rst
[1]
https://lore.kernel.org/bpf/20231219122850.433be...@gandalf.local.home/T/

Philo Lu (3):
  bpf: implement relay map basis
  bpf: implement map_update_elem to init relay file
  bpf: introduce bpf_relay_output helper

 include/linux/bpf.h   |   1 +
 include/linux/bpf_types.h |   3 +
 include/uapi/linux/bpf.h  |  17 +++
 kernel/bpf/Makefile   |   3 +
 kernel/bpf/helpers.c  |   4 +
 kernel/bpf/relaymap.c | 213 ++
 kernel/bpf/syscall.c  |   1 +
 kernel/bpf/verifier.c |   8 ++
 kernel/trace/bpf_trace.c  |   4 +
 9 files changed, 254 insertions(+)
 create mode 100644 kernel/bpf/relaymap.c

--
2.32.0.3.g01195cf9f




[PATCH bpf-next 2/3] bpf: implement map_update_elem to init relay file

2023-12-22 Thread Philo Lu
map_update_elem is used to create relay files and bind them with the
relay channel, which is created with BPF_MAP_CREATE. This allows users
to set a custom directory name. It must be used with key=NULL and
flag=0.

Here is an example:
```
struct {
__uint(type, BPF_MAP_TYPE_RELAY);
__uint(max_entries, 4096);
} my_relay SEC(".maps");
...
char dir_name[] = "relay_test";
bpf_map_update_elem(map_fd, NULL, dir_name, 0);
```

Then, directory `/sys/kerenl/debug/relay_test` will be created, which
includes files of my_relay0...my_relay[#cpu]. Each represents a per-cpu
buffer with size 8 * 4096 B (there are 8 subbufs by default, each with
size 4096B).

Signed-off-by: Philo Lu 
---
 kernel/bpf/relaymap.c | 32 +++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
index d0adc7f67758..588c8de0a4bd 100644
--- a/kernel/bpf/relaymap.c
+++ b/kernel/bpf/relaymap.c
@@ -117,7 +117,37 @@ static void *relay_map_lookup_elem(struct bpf_map *map, 
void *key)
 static long relay_map_update_elem(struct bpf_map *map, void *key, void *value,
   u64 flags)
 {
-   return -EOPNOTSUPP;
+   struct bpf_relay_map *rmap;
+   struct dentry *parent;
+   int err;
+
+   if (unlikely(flags))
+   return -EINVAL;
+
+   if (unlikely(key))
+   return -EINVAL;
+
+   rmap = container_of(map, struct bpf_relay_map, map);
+
+   /* The directory already exists */
+   if (rmap->relay_chan->has_base_filename)
+   return -EEXIST;
+
+   /* Setup relay files. Note that the directory name passed as value 
should
+* not be longer than map->value_size, including the '\0' at the end.
+*/
+   ((char *)value)[map->value_size - 1] = '\0';
+   parent = debugfs_create_dir(value, NULL);
+   if (IS_ERR_OR_NULL(parent))
+   return PTR_ERR(parent);
+
+   err = relay_late_setup_files(rmap->relay_chan, map->name, parent);
+   if (err) {
+   debugfs_remove_recursive(parent);
+   return err;
+   }
+
+   return 0;
 }
 
 static long relay_map_delete_elem(struct bpf_map *map, void *key)
-- 
2.32.0.3.g01195cf9f




[PATCH bpf-next 1/3] bpf: implement relay map basis

2023-12-22 Thread Philo Lu
BPF_MAP_TYPE_RELAY is implemented based on relay interface, which
creates per-cpu buffer to transfer data. Each buffer is essentially a
list of fix-sized sub-buffers, and is exposed to user space as files in
debugfs. attr->max_entries is used as subbuf size and attr->map_extra is
used as subbuf num. Currently, the default value of subbuf num is 8.

The data can be accessed by read or mmap via these files. For example,
if there are 2 cpus, files could be `/sys/kernel/debug/mydir/my_rmap0`
and `/sys/kernel/debug/mydir/my_rmap1`.

Buffer-only mode is used to create the relay map, which just allocates
the buffer without creating user-space files. Then user can setup the
files with map_update_elem, thus allowing user to define the directory
name in debugfs. map_update_elem is implemented in the following patch.

A new map flag named BPF_F_OVERWRITE is introduced to set overwrite mode
of relay map.

Signed-off-by: Philo Lu 
---
 include/linux/bpf_types.h |   3 +
 include/uapi/linux/bpf.h  |   7 ++
 kernel/bpf/Makefile   |   3 +
 kernel/bpf/relaymap.c | 157 ++
 kernel/bpf/syscall.c  |   1 +
 5 files changed, 171 insertions(+)
 create mode 100644 kernel/bpf/relaymap.c

diff --git a/include/linux/bpf_types.h b/include/linux/bpf_types.h
index fc0d6f32c687..c122d7b494c5 100644
--- a/include/linux/bpf_types.h
+++ b/include/linux/bpf_types.h
@@ -132,6 +132,9 @@ BPF_MAP_TYPE(BPF_MAP_TYPE_STRUCT_OPS, 
bpf_struct_ops_map_ops)
 BPF_MAP_TYPE(BPF_MAP_TYPE_RINGBUF, ringbuf_map_ops)
 BPF_MAP_TYPE(BPF_MAP_TYPE_BLOOM_FILTER, bloom_filter_map_ops)
 BPF_MAP_TYPE(BPF_MAP_TYPE_USER_RINGBUF, user_ringbuf_map_ops)
+#ifdef CONFIG_RELAY
+BPF_MAP_TYPE(BPF_MAP_TYPE_RELAY, relay_map_ops)
+#endif
 
 BPF_LINK_TYPE(BPF_LINK_TYPE_RAW_TRACEPOINT, raw_tracepoint)
 BPF_LINK_TYPE(BPF_LINK_TYPE_TRACING, tracing)
diff --git a/include/uapi/linux/bpf.h b/include/uapi/linux/bpf.h
index 754e68ca8744..143b75676bd3 100644
--- a/include/uapi/linux/bpf.h
+++ b/include/uapi/linux/bpf.h
@@ -951,6 +951,7 @@ enum bpf_map_type {
BPF_MAP_TYPE_BLOOM_FILTER,
BPF_MAP_TYPE_USER_RINGBUF,
BPF_MAP_TYPE_CGRP_STORAGE,
+   BPF_MAP_TYPE_RELAY,
 };
 
 /* Note that tracing related programs such as
@@ -1330,6 +1331,9 @@ enum {
 
 /* Get path from provided FD in BPF_OBJ_PIN/BPF_OBJ_GET commands */
BPF_F_PATH_FD   = (1U << 14),
+
+/* Enable overwrite for relay map */
+   BPF_F_OVERWRITE = (1U << 15),
 };
 
 /* Flags for BPF_PROG_QUERY. */
@@ -1401,6 +1405,9 @@ union bpf_attr {
 * BPF_MAP_TYPE_BLOOM_FILTER - the lowest 4 bits indicate the
 * number of hash functions (if 0, the bloom filter will default
 * to using 5 hash functions).
+*
+* BPF_MAP_TYPE_RELAY - the lowest 32 bits indicate the number 
of
+* relay subbufs (if 0, the number will be set to 8 by default).
 */
__u64   map_extra;
};
diff --git a/kernel/bpf/Makefile b/kernel/bpf/Makefile
index f526b7573e97..45b35bb0e572 100644
--- a/kernel/bpf/Makefile
+++ b/kernel/bpf/Makefile
@@ -10,6 +10,9 @@ obj-$(CONFIG_BPF_SYSCALL) += syscall.o verifier.o inode.o 
helpers.o tnum.o log.o
 obj-$(CONFIG_BPF_SYSCALL) += bpf_iter.o map_iter.o task_iter.o prog_iter.o 
link_iter.o
 obj-$(CONFIG_BPF_SYSCALL) += hashtab.o arraymap.o percpu_freelist.o 
bpf_lru_list.o lpm_trie.o map_in_map.o bloom_filter.o
 obj-$(CONFIG_BPF_SYSCALL) += local_storage.o queue_stack_maps.o ringbuf.o
+ifeq ($(CONFIG_RELAY),y)
+obj-$(CONFIG_BPF_SYSCALL) += relaymap.o
+endif
 obj-$(CONFIG_BPF_SYSCALL) += bpf_local_storage.o bpf_task_storage.o
 obj-${CONFIG_BPF_LSM}+= bpf_inode_storage.o
 obj-$(CONFIG_BPF_SYSCALL) += disasm.o mprog.o
diff --git a/kernel/bpf/relaymap.c b/kernel/bpf/relaymap.c
new file mode 100644
index ..d0adc7f67758
--- /dev/null
+++ b/kernel/bpf/relaymap.c
@@ -0,0 +1,157 @@
+#include 
+#include 
+#include 
+#include 
+#include 
+#include 
+#include 
+
+#define RELAY_CREATE_FLAG_MASK (BPF_F_OVERWRITE)
+
+struct bpf_relay_map {
+   struct bpf_map map;
+   struct rchan *relay_chan;
+   struct rchan_callbacks relay_cb;
+};
+
+static struct dentry *create_buf_file_handler(const char *filename,
+  struct dentry 
*parent, umode_t mode,
+  struct rchan_buf 
*buf, int *is_global)
+{
+   /* Because we do relay_late_setup_files(), create_buf_file(NULL, NULL, 
...)
+* will be called by relay_open.
+*/
+   if (!filename)
+   return NULL;
+
+   return debugfs_create_file(filename, mode, parent, buf,
+  _file_operations);
+}
+
+static int remove_buf_file_handler(struct dentry *dentry)
+{
+   debugfs_remove(dentry);
+   return 0;
+}
+
+/* For non-overwrite, use default subbuf_start cb */
+static int