1076a95f5SAlan Maguire // SPDX-License-Identifier: GPL-2.0 2076a95f5SAlan Maguire /* Copyright (c) 2020, Oracle and/or its affiliates. */ 3076a95f5SAlan Maguire 4076a95f5SAlan Maguire #include "vmlinux.h" 5076a95f5SAlan Maguire #include <bpf/bpf_helpers.h> 6076a95f5SAlan Maguire #include <bpf/bpf_tracing.h> 7076a95f5SAlan Maguire #include <bpf/bpf_core_read.h> 8076a95f5SAlan Maguire 9076a95f5SAlan Maguire #include <errno.h> 10076a95f5SAlan Maguire 11076a95f5SAlan Maguire long ret = 0; 12076a95f5SAlan Maguire int num_subtests = 0; 13076a95f5SAlan Maguire int ran_subtests = 0; 14076a95f5SAlan Maguire bool skip = false; 15076a95f5SAlan Maguire 16076a95f5SAlan Maguire #define STRSIZE 2048 17076a95f5SAlan Maguire #define EXPECTED_STRSIZE 256 18076a95f5SAlan Maguire 19076a95f5SAlan Maguire #ifndef ARRAY_SIZE 20076a95f5SAlan Maguire #define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0])) 21076a95f5SAlan Maguire #endif 22076a95f5SAlan Maguire 23076a95f5SAlan Maguire struct { 24076a95f5SAlan Maguire __uint(type, BPF_MAP_TYPE_PERCPU_ARRAY); 25076a95f5SAlan Maguire __uint(max_entries, 1); 26076a95f5SAlan Maguire __type(key, __u32); 27076a95f5SAlan Maguire __type(value, char[STRSIZE]); 28076a95f5SAlan Maguire } strdata SEC(".maps"); 29076a95f5SAlan Maguire 30076a95f5SAlan Maguire static int __strncmp(const void *m1, const void *m2, size_t len) 31076a95f5SAlan Maguire { 32076a95f5SAlan Maguire const unsigned char *s1 = m1; 33076a95f5SAlan Maguire const unsigned char *s2 = m2; 34076a95f5SAlan Maguire int i, delta = 0; 35076a95f5SAlan Maguire 36076a95f5SAlan Maguire for (i = 0; i < len; i++) { 37076a95f5SAlan Maguire delta = s1[i] - s2[i]; 38076a95f5SAlan Maguire if (delta || s1[i] == 0 || s2[i] == 0) 39076a95f5SAlan Maguire break; 40076a95f5SAlan Maguire } 41076a95f5SAlan Maguire return delta; 42076a95f5SAlan Maguire } 43076a95f5SAlan Maguire 44076a95f5SAlan Maguire #if __has_builtin(__builtin_btf_type_id) 45076a95f5SAlan Maguire #define TEST_BTF(_str, _type, _flags, _expected, ...) \ 46076a95f5SAlan Maguire do { \ 47076a95f5SAlan Maguire static const char _expectedval[EXPECTED_STRSIZE] = \ 48076a95f5SAlan Maguire _expected; \ 49076a95f5SAlan Maguire static const char _ptrtype[64] = #_type; \ 50076a95f5SAlan Maguire __u64 _hflags = _flags | BTF_F_COMPACT; \ 51076a95f5SAlan Maguire static _type _ptrdata = __VA_ARGS__; \ 52076a95f5SAlan Maguire static struct btf_ptr _ptr = { }; \ 53076a95f5SAlan Maguire int _cmp; \ 54076a95f5SAlan Maguire \ 55076a95f5SAlan Maguire ++num_subtests; \ 56076a95f5SAlan Maguire if (ret < 0) \ 57076a95f5SAlan Maguire break; \ 58076a95f5SAlan Maguire ++ran_subtests; \ 59076a95f5SAlan Maguire _ptr.ptr = &_ptrdata; \ 60076a95f5SAlan Maguire _ptr.type_id = bpf_core_type_id_kernel(_type); \ 61076a95f5SAlan Maguire if (_ptr.type_id <= 0) { \ 62076a95f5SAlan Maguire ret = -EINVAL; \ 63076a95f5SAlan Maguire break; \ 64076a95f5SAlan Maguire } \ 65076a95f5SAlan Maguire ret = bpf_snprintf_btf(_str, STRSIZE, \ 66076a95f5SAlan Maguire &_ptr, sizeof(_ptr), _hflags); \ 67076a95f5SAlan Maguire if (ret) \ 68076a95f5SAlan Maguire break; \ 69076a95f5SAlan Maguire _cmp = __strncmp(_str, _expectedval, EXPECTED_STRSIZE); \ 70076a95f5SAlan Maguire if (_cmp != 0) { \ 71076a95f5SAlan Maguire bpf_printk("(%d) got %s", _cmp, _str); \ 72076a95f5SAlan Maguire bpf_printk("(%d) expected %s", _cmp, \ 73076a95f5SAlan Maguire _expectedval); \ 74076a95f5SAlan Maguire ret = -EBADMSG; \ 75076a95f5SAlan Maguire break; \ 76076a95f5SAlan Maguire } \ 77076a95f5SAlan Maguire } while (0) 78076a95f5SAlan Maguire #endif 79076a95f5SAlan Maguire 80076a95f5SAlan Maguire /* Use where expected data string matches its stringified declaration */ 81076a95f5SAlan Maguire #define TEST_BTF_C(_str, _type, _flags, ...) \ 82076a95f5SAlan Maguire TEST_BTF(_str, _type, _flags, "(" #_type ")" #__VA_ARGS__, \ 83076a95f5SAlan Maguire __VA_ARGS__) 84076a95f5SAlan Maguire 85076a95f5SAlan Maguire /* TRACE_EVENT(netif_receive_skb, 86076a95f5SAlan Maguire * TP_PROTO(struct sk_buff *skb), 87076a95f5SAlan Maguire */ 88076a95f5SAlan Maguire SEC("tp_btf/netif_receive_skb") 89076a95f5SAlan Maguire int BPF_PROG(trace_netif_receive_skb, struct sk_buff *skb) 90076a95f5SAlan Maguire { 91076a95f5SAlan Maguire static __u64 flags[] = { 0, BTF_F_COMPACT, BTF_F_ZERO, BTF_F_PTR_RAW, 92076a95f5SAlan Maguire BTF_F_NONAME, BTF_F_COMPACT | BTF_F_ZERO | 93076a95f5SAlan Maguire BTF_F_PTR_RAW | BTF_F_NONAME }; 94076a95f5SAlan Maguire static struct btf_ptr p = { }; 95076a95f5SAlan Maguire __u32 key = 0; 96076a95f5SAlan Maguire int i, __ret; 97076a95f5SAlan Maguire char *str; 98076a95f5SAlan Maguire 99076a95f5SAlan Maguire #if __has_builtin(__builtin_btf_type_id) 100076a95f5SAlan Maguire str = bpf_map_lookup_elem(&strdata, &key); 101076a95f5SAlan Maguire if (!str) 102076a95f5SAlan Maguire return 0; 103076a95f5SAlan Maguire 104076a95f5SAlan Maguire /* Ensure we can write skb string representation */ 105076a95f5SAlan Maguire p.type_id = bpf_core_type_id_kernel(struct sk_buff); 106076a95f5SAlan Maguire p.ptr = skb; 107076a95f5SAlan Maguire for (i = 0; i < ARRAY_SIZE(flags); i++) { 108076a95f5SAlan Maguire ++num_subtests; 109076a95f5SAlan Maguire ret = bpf_snprintf_btf(str, STRSIZE, &p, sizeof(p), 0); 110076a95f5SAlan Maguire if (ret < 0) 111076a95f5SAlan Maguire bpf_printk("returned %d when writing skb", ret); 112076a95f5SAlan Maguire ++ran_subtests; 113076a95f5SAlan Maguire } 114076a95f5SAlan Maguire 115076a95f5SAlan Maguire /* Check invalid ptr value */ 116076a95f5SAlan Maguire p.ptr = 0; 117076a95f5SAlan Maguire __ret = bpf_snprintf_btf(str, STRSIZE, &p, sizeof(p), 0); 118076a95f5SAlan Maguire if (__ret >= 0) { 119076a95f5SAlan Maguire bpf_printk("printing NULL should generate error, got (%d)", 120076a95f5SAlan Maguire __ret); 121076a95f5SAlan Maguire ret = -ERANGE; 122076a95f5SAlan Maguire } 123076a95f5SAlan Maguire 124076a95f5SAlan Maguire /* Verify type display for various types. */ 125076a95f5SAlan Maguire 126076a95f5SAlan Maguire /* simple int */ 127076a95f5SAlan Maguire TEST_BTF_C(str, int, 0, 1234); 128076a95f5SAlan Maguire TEST_BTF(str, int, BTF_F_NONAME, "1234", 1234); 129076a95f5SAlan Maguire /* zero value should be printed at toplevel */ 130076a95f5SAlan Maguire TEST_BTF(str, int, 0, "(int)0", 0); 131076a95f5SAlan Maguire TEST_BTF(str, int, BTF_F_NONAME, "0", 0); 132076a95f5SAlan Maguire TEST_BTF(str, int, BTF_F_ZERO, "(int)0", 0); 133076a95f5SAlan Maguire TEST_BTF(str, int, BTF_F_NONAME | BTF_F_ZERO, "0", 0); 134076a95f5SAlan Maguire TEST_BTF_C(str, int, 0, -4567); 135076a95f5SAlan Maguire TEST_BTF(str, int, BTF_F_NONAME, "-4567", -4567); 136076a95f5SAlan Maguire 137076a95f5SAlan Maguire /* simple char */ 138076a95f5SAlan Maguire TEST_BTF_C(str, char, 0, 100); 139076a95f5SAlan Maguire TEST_BTF(str, char, BTF_F_NONAME, "100", 100); 140076a95f5SAlan Maguire /* zero value should be printed at toplevel */ 141076a95f5SAlan Maguire TEST_BTF(str, char, 0, "(char)0", 0); 142076a95f5SAlan Maguire TEST_BTF(str, char, BTF_F_NONAME, "0", 0); 143076a95f5SAlan Maguire TEST_BTF(str, char, BTF_F_ZERO, "(char)0", 0); 144076a95f5SAlan Maguire TEST_BTF(str, char, BTF_F_NONAME | BTF_F_ZERO, "0", 0); 145076a95f5SAlan Maguire 146076a95f5SAlan Maguire /* simple typedef */ 147076a95f5SAlan Maguire TEST_BTF_C(str, uint64_t, 0, 100); 148076a95f5SAlan Maguire TEST_BTF(str, u64, BTF_F_NONAME, "1", 1); 149076a95f5SAlan Maguire /* zero value should be printed at toplevel */ 150076a95f5SAlan Maguire TEST_BTF(str, u64, 0, "(u64)0", 0); 151076a95f5SAlan Maguire TEST_BTF(str, u64, BTF_F_NONAME, "0", 0); 152076a95f5SAlan Maguire TEST_BTF(str, u64, BTF_F_ZERO, "(u64)0", 0); 153076a95f5SAlan Maguire TEST_BTF(str, u64, BTF_F_NONAME|BTF_F_ZERO, "0", 0); 154076a95f5SAlan Maguire 155076a95f5SAlan Maguire /* typedef struct */ 156076a95f5SAlan Maguire TEST_BTF_C(str, atomic_t, 0, {.counter = (int)1,}); 157076a95f5SAlan Maguire TEST_BTF(str, atomic_t, BTF_F_NONAME, "{1,}", {.counter = 1,}); 158076a95f5SAlan Maguire /* typedef with 0 value should be printed at toplevel */ 159076a95f5SAlan Maguire TEST_BTF(str, atomic_t, 0, "(atomic_t){}", {.counter = 0,}); 160076a95f5SAlan Maguire TEST_BTF(str, atomic_t, BTF_F_NONAME, "{}", {.counter = 0,}); 161076a95f5SAlan Maguire TEST_BTF(str, atomic_t, BTF_F_ZERO, "(atomic_t){.counter = (int)0,}", 162076a95f5SAlan Maguire {.counter = 0,}); 163076a95f5SAlan Maguire TEST_BTF(str, atomic_t, BTF_F_NONAME|BTF_F_ZERO, 164076a95f5SAlan Maguire "{0,}", {.counter = 0,}); 165076a95f5SAlan Maguire 166076a95f5SAlan Maguire /* enum where enum value does (and does not) exist */ 167076a95f5SAlan Maguire TEST_BTF_C(str, enum bpf_cmd, 0, BPF_MAP_CREATE); 168076a95f5SAlan Maguire TEST_BTF(str, enum bpf_cmd, 0, "(enum bpf_cmd)BPF_MAP_CREATE", 0); 169076a95f5SAlan Maguire TEST_BTF(str, enum bpf_cmd, BTF_F_NONAME, "BPF_MAP_CREATE", 170076a95f5SAlan Maguire BPF_MAP_CREATE); 171076a95f5SAlan Maguire TEST_BTF(str, enum bpf_cmd, BTF_F_NONAME|BTF_F_ZERO, 172076a95f5SAlan Maguire "BPF_MAP_CREATE", 0); 173076a95f5SAlan Maguire 174076a95f5SAlan Maguire TEST_BTF(str, enum bpf_cmd, BTF_F_ZERO, "(enum bpf_cmd)BPF_MAP_CREATE", 175076a95f5SAlan Maguire BPF_MAP_CREATE); 176076a95f5SAlan Maguire TEST_BTF(str, enum bpf_cmd, BTF_F_NONAME|BTF_F_ZERO, 177076a95f5SAlan Maguire "BPF_MAP_CREATE", BPF_MAP_CREATE); 178076a95f5SAlan Maguire TEST_BTF_C(str, enum bpf_cmd, 0, 2000); 179076a95f5SAlan Maguire TEST_BTF(str, enum bpf_cmd, BTF_F_NONAME, "2000", 2000); 180076a95f5SAlan Maguire 181076a95f5SAlan Maguire /* simple struct */ 182076a95f5SAlan Maguire TEST_BTF_C(str, struct btf_enum, 0, 183076a95f5SAlan Maguire {.name_off = (__u32)3,.val = (__s32)-1,}); 184076a95f5SAlan Maguire TEST_BTF(str, struct btf_enum, BTF_F_NONAME, "{3,-1,}", 185076a95f5SAlan Maguire { .name_off = 3, .val = -1,}); 186076a95f5SAlan Maguire TEST_BTF(str, struct btf_enum, BTF_F_NONAME, "{-1,}", 187076a95f5SAlan Maguire { .name_off = 0, .val = -1,}); 188076a95f5SAlan Maguire TEST_BTF(str, struct btf_enum, BTF_F_NONAME|BTF_F_ZERO, "{0,-1,}", 189076a95f5SAlan Maguire { .name_off = 0, .val = -1,}); 190076a95f5SAlan Maguire /* empty struct should be printed */ 191076a95f5SAlan Maguire TEST_BTF(str, struct btf_enum, 0, "(struct btf_enum){}", 192076a95f5SAlan Maguire { .name_off = 0, .val = 0,}); 193076a95f5SAlan Maguire TEST_BTF(str, struct btf_enum, BTF_F_NONAME, "{}", 194076a95f5SAlan Maguire { .name_off = 0, .val = 0,}); 195076a95f5SAlan Maguire TEST_BTF(str, struct btf_enum, BTF_F_ZERO, 196076a95f5SAlan Maguire "(struct btf_enum){.name_off = (__u32)0,.val = (__s32)0,}", 197076a95f5SAlan Maguire { .name_off = 0, .val = 0,}); 198076a95f5SAlan Maguire 199076a95f5SAlan Maguire /* struct with pointers */ 200076a95f5SAlan Maguire TEST_BTF(str, struct list_head, BTF_F_PTR_RAW, 201076a95f5SAlan Maguire "(struct list_head){.next = (struct list_head *)0x0000000000000001,}", 202076a95f5SAlan Maguire { .next = (struct list_head *)1 }); 203076a95f5SAlan Maguire /* NULL pointer should not be displayed */ 204076a95f5SAlan Maguire TEST_BTF(str, struct list_head, BTF_F_PTR_RAW, 205076a95f5SAlan Maguire "(struct list_head){}", 206076a95f5SAlan Maguire { .next = (struct list_head *)0 }); 207076a95f5SAlan Maguire 208076a95f5SAlan Maguire /* struct with char array */ 209076a95f5SAlan Maguire TEST_BTF(str, struct bpf_prog_info, 0, 210076a95f5SAlan Maguire "(struct bpf_prog_info){.name = (char[])['f','o','o',],}", 211076a95f5SAlan Maguire { .name = "foo",}); 212076a95f5SAlan Maguire TEST_BTF(str, struct bpf_prog_info, BTF_F_NONAME, 213076a95f5SAlan Maguire "{['f','o','o',],}", 214076a95f5SAlan Maguire {.name = "foo",}); 215076a95f5SAlan Maguire /* leading null char means do not display string */ 216076a95f5SAlan Maguire TEST_BTF(str, struct bpf_prog_info, 0, 217076a95f5SAlan Maguire "(struct bpf_prog_info){}", 218076a95f5SAlan Maguire {.name = {'\0', 'f', 'o', 'o'}}); 219076a95f5SAlan Maguire /* handle non-printable characters */ 220076a95f5SAlan Maguire TEST_BTF(str, struct bpf_prog_info, 0, 221076a95f5SAlan Maguire "(struct bpf_prog_info){.name = (char[])[1,2,3,],}", 222076a95f5SAlan Maguire { .name = {1, 2, 3, 0}}); 223076a95f5SAlan Maguire 224076a95f5SAlan Maguire /* struct with non-char array */ 225076a95f5SAlan Maguire TEST_BTF(str, struct __sk_buff, 0, 226076a95f5SAlan Maguire "(struct __sk_buff){.cb = (__u32[])[1,2,3,4,5,],}", 227076a95f5SAlan Maguire { .cb = {1, 2, 3, 4, 5,},}); 228076a95f5SAlan Maguire TEST_BTF(str, struct __sk_buff, BTF_F_NONAME, 229076a95f5SAlan Maguire "{[1,2,3,4,5,],}", 230076a95f5SAlan Maguire { .cb = { 1, 2, 3, 4, 5},}); 231076a95f5SAlan Maguire /* For non-char, arrays, show non-zero values only */ 232076a95f5SAlan Maguire TEST_BTF(str, struct __sk_buff, 0, 233076a95f5SAlan Maguire "(struct __sk_buff){.cb = (__u32[])[1,],}", 234076a95f5SAlan Maguire { .cb = { 0, 0, 1, 0, 0},}); 235076a95f5SAlan Maguire 236076a95f5SAlan Maguire /* struct with bitfields */ 237076a95f5SAlan Maguire TEST_BTF_C(str, struct bpf_insn, 0, 238076a95f5SAlan Maguire {.code = (__u8)1,.dst_reg = (__u8)0x2,.src_reg = (__u8)0x3,.off = (__s16)4,.imm = (__s32)5,}); 239076a95f5SAlan Maguire TEST_BTF(str, struct bpf_insn, BTF_F_NONAME, "{1,0x2,0x3,4,5,}", 240076a95f5SAlan Maguire {.code = 1, .dst_reg = 0x2, .src_reg = 0x3, .off = 4, 241076a95f5SAlan Maguire .imm = 5,}); 242076a95f5SAlan Maguire #else 243076a95f5SAlan Maguire skip = true; 244076a95f5SAlan Maguire #endif 245076a95f5SAlan Maguire 246076a95f5SAlan Maguire return 0; 247076a95f5SAlan Maguire } 248076a95f5SAlan Maguire 249076a95f5SAlan Maguire char _license[] SEC("license") = "GPL"; 250