linux/tools/testing/selftests/bpf/prog_tests/obj_name.c
<<
>>
Prefs
   1// SPDX-License-Identifier: GPL-2.0
   2#include <test_progs.h>
   3
   4void test_obj_name(void)
   5{
   6        struct {
   7                const char *name;
   8                int success;
   9                int expected_errno;
  10        } tests[] = {
  11                { "", 1, 0 },
  12                { "_123456789ABCDE", 1, 0 },
  13                { "_123456789ABCDEF", 0, EINVAL },
  14                { "_123456789ABCD\n", 0, EINVAL },
  15        };
  16        struct bpf_insn prog[] = {
  17                BPF_ALU64_IMM(BPF_MOV, BPF_REG_0, 0),
  18                BPF_EXIT_INSN(),
  19        };
  20        __u32 duration = 0;
  21        int i;
  22
  23        for (i = 0; i < sizeof(tests) / sizeof(tests[0]); i++) {
  24                size_t name_len = strlen(tests[i].name) + 1;
  25                union bpf_attr attr;
  26                size_t ncopy;
  27                int fd;
  28
  29                /* test different attr.prog_name during BPF_PROG_LOAD */
  30                ncopy = name_len < sizeof(attr.prog_name) ?
  31                        name_len : sizeof(attr.prog_name);
  32                bzero(&attr, sizeof(attr));
  33                attr.prog_type = BPF_PROG_TYPE_SCHED_CLS;
  34                attr.insn_cnt = 2;
  35                attr.insns = ptr_to_u64(prog);
  36                attr.license = ptr_to_u64("");
  37                memcpy(attr.prog_name, tests[i].name, ncopy);
  38
  39                fd = syscall(__NR_bpf, BPF_PROG_LOAD, &attr, sizeof(attr));
  40                CHECK((tests[i].success && fd < 0) ||
  41                      (!tests[i].success && fd != -1) ||
  42                      (!tests[i].success && errno != tests[i].expected_errno),
  43                      "check-bpf-prog-name",
  44                      "fd %d(%d) errno %d(%d)\n",
  45                       fd, tests[i].success, errno, tests[i].expected_errno);
  46
  47                if (fd != -1)
  48                        close(fd);
  49
  50                /* test different attr.map_name during BPF_MAP_CREATE */
  51                ncopy = name_len < sizeof(attr.map_name) ?
  52                        name_len : sizeof(attr.map_name);
  53                bzero(&attr, sizeof(attr));
  54                attr.map_type = BPF_MAP_TYPE_ARRAY;
  55                attr.key_size = 4;
  56                attr.value_size = 4;
  57                attr.max_entries = 1;
  58                attr.map_flags = 0;
  59                memcpy(attr.map_name, tests[i].name, ncopy);
  60                fd = syscall(__NR_bpf, BPF_MAP_CREATE, &attr, sizeof(attr));
  61                CHECK((tests[i].success && fd < 0) ||
  62                      (!tests[i].success && fd != -1) ||
  63                      (!tests[i].success && errno != tests[i].expected_errno),
  64                      "check-bpf-map-name",
  65                      "fd %d(%d) errno %d(%d)\n",
  66                      fd, tests[i].success, errno, tests[i].expected_errno);
  67
  68                if (fd != -1)
  69                        close(fd);
  70        }
  71}
  72