1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479 |
- /* SPDX-License-Identifier: GPL-2.0 */
- /* Copyright (c) 2018 Facebook */
- #include <linux/bpf.h>
- #include <linux/btf.h>
- #include <linux/err.h>
- #include <bpf/bpf.h>
- #include <sys/resource.h>
- #include <libelf.h>
- #include <gelf.h>
- #include <string.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include <stdarg.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <errno.h>
- #include <bpf/libbpf.h>
- #include <bpf/btf.h>
- #include "bpf_rlimit.h"
- #include "bpf_util.h"
- static uint32_t pass_cnt;
- static uint32_t error_cnt;
- static uint32_t skip_cnt;
- #define CHECK(condition, format...) ({ \
- int __ret = !!(condition); \
- if (__ret) { \
- fprintf(stderr, "%s:%d:FAIL ", __func__, __LINE__); \
- fprintf(stderr, format); \
- } \
- __ret; \
- })
- static int count_result(int err)
- {
- if (err)
- error_cnt++;
- else
- pass_cnt++;
- fprintf(stderr, "\n");
- return err;
- }
- #define min(a, b) ((a) < (b) ? (a) : (b))
- #define __printf(a, b) __attribute__((format(printf, a, b)))
- __printf(1, 2)
- static int __base_pr(const char *format, ...)
- {
- va_list args;
- int err;
- va_start(args, format);
- err = vfprintf(stderr, format, args);
- va_end(args);
- return err;
- }
- #define BTF_INFO_ENC(kind, root, vlen) \
- ((!!(root) << 31) | ((kind) << 24) | ((vlen) & BTF_MAX_VLEN))
- #define BTF_TYPE_ENC(name, info, size_or_type) \
- (name), (info), (size_or_type)
- #define BTF_INT_ENC(encoding, bits_offset, nr_bits) \
- ((encoding) << 24 | (bits_offset) << 16 | (nr_bits))
- #define BTF_TYPE_INT_ENC(name, encoding, bits_offset, bits, sz) \
- BTF_TYPE_ENC(name, BTF_INFO_ENC(BTF_KIND_INT, 0, 0), sz), \
- BTF_INT_ENC(encoding, bits_offset, bits)
- #define BTF_ARRAY_ENC(type, index_type, nr_elems) \
- (type), (index_type), (nr_elems)
- #define BTF_TYPE_ARRAY_ENC(type, index_type, nr_elems) \
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_ARRAY, 0, 0), 0), \
- BTF_ARRAY_ENC(type, index_type, nr_elems)
- #define BTF_MEMBER_ENC(name, type, bits_offset) \
- (name), (type), (bits_offset)
- #define BTF_ENUM_ENC(name, val) (name), (val)
- #define BTF_TYPEDEF_ENC(name, type) \
- BTF_TYPE_ENC(name, BTF_INFO_ENC(BTF_KIND_TYPEDEF, 0, 0), type)
- #define BTF_PTR_ENC(name, type) \
- BTF_TYPE_ENC(name, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), type)
- #define BTF_END_RAW 0xdeadbeef
- #define NAME_TBD 0xdeadb33f
- #define MAX_NR_RAW_TYPES 1024
- #define BTF_LOG_BUF_SIZE 65535
- static struct args {
- unsigned int raw_test_num;
- unsigned int file_test_num;
- unsigned int get_info_test_num;
- bool raw_test;
- bool file_test;
- bool get_info_test;
- bool pprint_test;
- bool always_log;
- } args;
- static char btf_log_buf[BTF_LOG_BUF_SIZE];
- static struct btf_header hdr_tmpl = {
- .magic = BTF_MAGIC,
- .version = BTF_VERSION,
- .hdr_len = sizeof(struct btf_header),
- };
- struct btf_raw_test {
- const char *descr;
- const char *str_sec;
- const char *map_name;
- const char *err_str;
- __u32 raw_types[MAX_NR_RAW_TYPES];
- __u32 str_sec_size;
- enum bpf_map_type map_type;
- __u32 key_size;
- __u32 value_size;
- __u32 key_type_id;
- __u32 value_type_id;
- __u32 max_entries;
- bool btf_load_err;
- bool map_create_err;
- bool ordered_map;
- bool lossless_map;
- int hdr_len_delta;
- int type_off_delta;
- int str_off_delta;
- int str_len_delta;
- };
- static struct btf_raw_test raw_tests[] = {
- /* enum E {
- * E0,
- * E1,
- * };
- *
- * struct A {
- * unsigned long long m;
- * int n;
- * char o;
- * [3 bytes hole]
- * int p[8];
- * int q[4][8];
- * enum E r;
- * };
- */
- {
- .descr = "struct test #1",
- .raw_types = {
- /* int */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4), /* [1] */
- /* unsigned long long */
- BTF_TYPE_INT_ENC(0, 0, 0, 64, 8), /* [2] */
- /* char */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 8, 1), /* [3] */
- /* int[8] */
- BTF_TYPE_ARRAY_ENC(1, 1, 8), /* [4] */
- /* struct A { */ /* [5] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 6), 180),
- BTF_MEMBER_ENC(NAME_TBD, 2, 0), /* unsigned long long m;*/
- BTF_MEMBER_ENC(NAME_TBD, 1, 64),/* int n; */
- BTF_MEMBER_ENC(NAME_TBD, 3, 96),/* char o; */
- BTF_MEMBER_ENC(NAME_TBD, 4, 128),/* int p[8] */
- BTF_MEMBER_ENC(NAME_TBD, 6, 384),/* int q[4][8] */
- BTF_MEMBER_ENC(NAME_TBD, 7, 1408), /* enum E r */
- /* } */
- /* int[4][8] */
- BTF_TYPE_ARRAY_ENC(4, 1, 4), /* [6] */
- /* enum E */ /* [7] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_ENUM, 0, 2), sizeof(int)),
- BTF_ENUM_ENC(NAME_TBD, 0),
- BTF_ENUM_ENC(NAME_TBD, 1),
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0n\0o\0p\0q\0r\0E\0E0\0E1",
- .str_sec_size = sizeof("\0A\0m\0n\0o\0p\0q\0r\0E\0E0\0E1"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "struct_test1_map",
- .key_size = sizeof(int),
- .value_size = 180,
- .key_type_id = 1,
- .value_type_id = 5,
- .max_entries = 4,
- },
- /* typedef struct b Struct_B;
- *
- * struct A {
- * int m;
- * struct b n[4];
- * const Struct_B o[4];
- * };
- *
- * struct B {
- * int m;
- * int n;
- * };
- */
- {
- .descr = "struct test #2",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* struct b [4] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(4, 1, 4),
- /* struct A { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 3), 68),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 32),/* struct B n[4] */
- BTF_MEMBER_ENC(NAME_TBD, 8, 288),/* const Struct_B o[4];*/
- /* } */
- /* struct B { */ /* [4] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), 8),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 1, 32),/* int n; */
- /* } */
- /* const int */ /* [5] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 1),
- /* typedef struct b Struct_B */ /* [6] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_TYPEDEF, 0, 0), 4),
- /* const Struct_B */ /* [7] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 6),
- /* const Struct_B [4] */ /* [8] */
- BTF_TYPE_ARRAY_ENC(7, 1, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0n\0o\0B\0m\0n\0Struct_B",
- .str_sec_size = sizeof("\0A\0m\0n\0o\0B\0m\0n\0Struct_B"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "struct_test2_map",
- .key_size = sizeof(int),
- .value_size = 68,
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- },
- {
- .descr = "struct test #3 Invalid member offset",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int64 */ /* [2] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 64, 8),
- /* struct A { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), 16),
- BTF_MEMBER_ENC(NAME_TBD, 1, 64), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 0), /* int64 n; */
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0n\0",
- .str_sec_size = sizeof("\0A\0m\0n\0"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "struct_test3_map",
- .key_size = sizeof(int),
- .value_size = 16,
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid member bits_offset",
- },
- /* Test member exceeds the size of struct.
- *
- * struct A {
- * int m;
- * int n;
- * };
- */
- {
- .descr = "size check test #1",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* struct A { */ /* [2] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), sizeof(int) * 2 - 1),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 1, 32),/* int n; */
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0n",
- .str_sec_size = sizeof("\0A\0m\0n"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "size_check1_map",
- .key_size = sizeof(int),
- .value_size = 1,
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Member exceeds struct_size",
- },
- /* Test member exeeds the size of struct
- *
- * struct A {
- * int m;
- * int n[2];
- * };
- */
- {
- .descr = "size check test #2",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, sizeof(int)),
- /* int[2] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(1, 1, 2),
- /* struct A { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), sizeof(int) * 3 - 1),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 32),/* int n[2]; */
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0n",
- .str_sec_size = sizeof("\0A\0m\0n"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "size_check2_map",
- .key_size = sizeof(int),
- .value_size = 1,
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Member exceeds struct_size",
- },
- /* Test member exeeds the size of struct
- *
- * struct A {
- * int m;
- * void *n;
- * };
- */
- {
- .descr = "size check test #3",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, sizeof(int)),
- /* void* */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 0),
- /* struct A { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), sizeof(int) + sizeof(void *) - 1),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 32),/* void *n; */
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0n",
- .str_sec_size = sizeof("\0A\0m\0n"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "size_check3_map",
- .key_size = sizeof(int),
- .value_size = 1,
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Member exceeds struct_size",
- },
- /* Test member exceeds the size of struct
- *
- * enum E {
- * E0,
- * E1,
- * };
- *
- * struct A {
- * int m;
- * enum E n;
- * };
- */
- {
- .descr = "size check test #4",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, sizeof(int)),
- /* enum E { */ /* [2] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_ENUM, 0, 2), sizeof(int)),
- BTF_ENUM_ENC(NAME_TBD, 0),
- BTF_ENUM_ENC(NAME_TBD, 1),
- /* } */
- /* struct A { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), sizeof(int) * 2 - 1),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int m; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 32),/* enum E n; */
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0E\0E0\0E1\0A\0m\0n",
- .str_sec_size = sizeof("\0E\0E0\0E1\0A\0m\0n"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "size_check4_map",
- .key_size = sizeof(int),
- .value_size = 1,
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Member exceeds struct_size",
- },
- /* typedef const void * const_void_ptr;
- * struct A {
- * const_void_ptr m;
- * };
- */
- {
- .descr = "void test #1",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* const void */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 0),
- /* const void* */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 2),
- /* typedef const void * const_void_ptr */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 3),
- /* struct A { */ /* [4] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 1), sizeof(void *)),
- /* const_void_ptr m; */
- BTF_MEMBER_ENC(NAME_TBD, 3, 0),
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0const_void_ptr\0A\0m",
- .str_sec_size = sizeof("\0const_void_ptr\0A\0m"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "void_test1_map",
- .key_size = sizeof(int),
- .value_size = sizeof(void *),
- .key_type_id = 1,
- .value_type_id = 4,
- .max_entries = 4,
- },
- /* struct A {
- * const void m;
- * };
- */
- {
- .descr = "void test #2",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* const void */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 0),
- /* struct A { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 1), 8),
- /* const void m; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 0),
- /* } */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m",
- .str_sec_size = sizeof("\0A\0m"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "void_test2_map",
- .key_size = sizeof(int),
- .value_size = sizeof(void *),
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid member",
- },
- /* typedef const void * const_void_ptr;
- * const_void_ptr[4]
- */
- {
- .descr = "void test #3",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* const void */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 0),
- /* const void* */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 2),
- /* typedef const void * const_void_ptr */ /* [4] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 3),
- /* const_void_ptr[4] */ /* [5] */
- BTF_TYPE_ARRAY_ENC(3, 1, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0const_void_ptr",
- .str_sec_size = sizeof("\0const_void_ptr"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "void_test3_map",
- .key_size = sizeof(int),
- .value_size = sizeof(void *) * 4,
- .key_type_id = 1,
- .value_type_id = 5,
- .max_entries = 4,
- },
- /* const void[4] */
- {
- .descr = "void test #4",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* const void */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 0),
- /* const void[4] */ /* [3] */
- BTF_TYPE_ARRAY_ENC(2, 1, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m",
- .str_sec_size = sizeof("\0A\0m"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "void_test4_map",
- .key_size = sizeof(int),
- .value_size = sizeof(void *) * 4,
- .key_type_id = 1,
- .value_type_id = 3,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid elem",
- },
- /* Array_A <------------------+
- * elem_type == Array_B |
- * | |
- * | |
- * Array_B <-------- + |
- * elem_type == Array A --+
- */
- {
- .descr = "loop test #1",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* Array_A */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 1, 8),
- /* Array_B */ /* [3] */
- BTF_TYPE_ARRAY_ENC(2, 1, 8),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test1_map",
- .key_size = sizeof(int),
- .value_size = sizeof(sizeof(int) * 8),
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- /* typedef is _before_ the BTF type of Array_A and Array_B
- *
- * typedef Array_B int_array;
- *
- * Array_A <------------------+
- * elem_type == int_array |
- * | |
- * | |
- * Array_B <-------- + |
- * elem_type == Array_A --+
- */
- {
- .descr = "loop test #2",
- .raw_types = {
- /* int */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4), /* [1] */
- /* typedef Array_B int_array */
- BTF_TYPEDEF_ENC(1, 4), /* [2] */
- /* Array_A */
- BTF_TYPE_ARRAY_ENC(2, 1, 8), /* [3] */
- /* Array_B */
- BTF_TYPE_ARRAY_ENC(3, 1, 8), /* [4] */
- BTF_END_RAW,
- },
- .str_sec = "\0int_array\0",
- .str_sec_size = sizeof("\0int_array"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test2_map",
- .key_size = sizeof(int),
- .value_size = sizeof(sizeof(int) * 8),
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- /* Array_A <------------------+
- * elem_type == Array_B |
- * | |
- * | |
- * Array_B <-------- + |
- * elem_type == Array_A --+
- */
- {
- .descr = "loop test #3",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* Array_A */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 1, 8),
- /* Array_B */ /* [3] */
- BTF_TYPE_ARRAY_ENC(2, 1, 8),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test3_map",
- .key_size = sizeof(int),
- .value_size = sizeof(sizeof(int) * 8),
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- /* typedef is _between_ the BTF type of Array_A and Array_B
- *
- * typedef Array_B int_array;
- *
- * Array_A <------------------+
- * elem_type == int_array |
- * | |
- * | |
- * Array_B <-------- + |
- * elem_type == Array_A --+
- */
- {
- .descr = "loop test #4",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* Array_A */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 1, 8),
- /* typedef Array_B int_array */ /* [3] */
- BTF_TYPEDEF_ENC(NAME_TBD, 4),
- /* Array_B */ /* [4] */
- BTF_TYPE_ARRAY_ENC(2, 1, 8),
- BTF_END_RAW,
- },
- .str_sec = "\0int_array\0",
- .str_sec_size = sizeof("\0int_array"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test4_map",
- .key_size = sizeof(int),
- .value_size = sizeof(sizeof(int) * 8),
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- /* typedef struct B Struct_B
- *
- * struct A {
- * int x;
- * Struct_B y;
- * };
- *
- * struct B {
- * int x;
- * struct A y;
- * };
- */
- {
- .descr = "loop test #5",
- .raw_types = {
- /* int */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4), /* [1] */
- /* struct A */ /* [2] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), 8),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int x; */
- BTF_MEMBER_ENC(NAME_TBD, 3, 32),/* Struct_B y; */
- /* typedef struct B Struct_B */
- BTF_TYPEDEF_ENC(NAME_TBD, 4), /* [3] */
- /* struct B */ /* [4] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), 8),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int x; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 32),/* struct A y; */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0x\0y\0Struct_B\0B\0x\0y",
- .str_sec_size = sizeof("\0A\0x\0y\0Struct_B\0B\0x\0y"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test5_map",
- .key_size = sizeof(int),
- .value_size = 8,
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- /* struct A {
- * int x;
- * struct A array_a[4];
- * };
- */
- {
- .descr = "loop test #6",
- .raw_types = {
- /* int */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4), /* [1] */
- BTF_TYPE_ARRAY_ENC(3, 1, 4), /* [2] */
- /* struct A */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 2), 8),
- BTF_MEMBER_ENC(NAME_TBD, 1, 0), /* int x; */
- BTF_MEMBER_ENC(NAME_TBD, 2, 32),/* struct A array_a[4]; */
- BTF_END_RAW,
- },
- .str_sec = "\0A\0x\0y",
- .str_sec_size = sizeof("\0A\0x\0y"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test6_map",
- .key_size = sizeof(int),
- .value_size = 8,
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- {
- .descr = "loop test #7",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* struct A { */ /* [2] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 1), sizeof(void *)),
- /* const void *m; */
- BTF_MEMBER_ENC(NAME_TBD, 3, 0),
- /* CONST type_id=3 */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 4),
- /* PTR type_id=2 */ /* [4] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 3),
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m",
- .str_sec_size = sizeof("\0A\0m"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test7_map",
- .key_size = sizeof(int),
- .value_size = sizeof(void *),
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- {
- .descr = "loop test #8",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* struct A { */ /* [2] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 1), sizeof(void *)),
- /* const void *m; */
- BTF_MEMBER_ENC(NAME_TBD, 4, 0),
- /* struct B { */ /* [3] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 1), sizeof(void *)),
- /* const void *n; */
- BTF_MEMBER_ENC(NAME_TBD, 6, 0),
- /* CONST type_id=5 */ /* [4] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 5),
- /* PTR type_id=6 */ /* [5] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 6),
- /* CONST type_id=7 */ /* [6] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 7),
- /* PTR type_id=4 */ /* [7] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 4),
- BTF_END_RAW,
- },
- .str_sec = "\0A\0m\0B\0n",
- .str_sec_size = sizeof("\0A\0m\0B\0n"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "loop_test8_map",
- .key_size = sizeof(int),
- .value_size = sizeof(void *),
- .key_type_id = 1,
- .value_type_id = 2,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Loop detected",
- },
- {
- .descr = "string section does not end with null",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int") - 1,
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid string section",
- },
- {
- .descr = "empty string section",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = 0,
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid string section",
- },
- {
- .descr = "empty type section",
- .raw_types = {
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "No type found",
- },
- {
- .descr = "btf_header test. Longer hdr_len",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .hdr_len_delta = 4,
- .err_str = "Unsupported btf_header",
- },
- {
- .descr = "btf_header test. Gap between hdr and type",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .type_off_delta = 4,
- .err_str = "Unsupported section found",
- },
- {
- .descr = "btf_header test. Gap between type and str",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .str_off_delta = 4,
- .err_str = "Unsupported section found",
- },
- {
- .descr = "btf_header test. Overlap between type and str",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .str_off_delta = -4,
- .err_str = "Section overlap found",
- },
- {
- .descr = "btf_header test. Larger BTF size",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .str_len_delta = -4,
- .err_str = "Unsupported section found",
- },
- {
- .descr = "btf_header test. Smaller BTF size",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "\0int",
- .str_sec_size = sizeof("\0int"),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "hdr_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .str_len_delta = 4,
- .err_str = "Total section length too long",
- },
- {
- .descr = "array test. index_type/elem_type \"int\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(1, 1, 16),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- },
- {
- .descr = "array test. index_type/elem_type \"const int\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 3, 16),
- /* CONST type_id=1 */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 1),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- },
- {
- .descr = "array test. index_type \"const int:31\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int:31 */ /* [2] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 31, 4),
- /* int[16] */ /* [3] */
- BTF_TYPE_ARRAY_ENC(1, 4, 16),
- /* CONST type_id=2 */ /* [4] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 2),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid index",
- },
- {
- .descr = "array test. elem_type \"const int:31\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int:31 */ /* [2] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 31, 4),
- /* int[16] */ /* [3] */
- BTF_TYPE_ARRAY_ENC(4, 1, 16),
- /* CONST type_id=2 */ /* [4] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 2),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid array of int",
- },
- {
- .descr = "array test. index_type \"void\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(1, 0, 16),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid index",
- },
- {
- .descr = "array test. index_type \"const void\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(1, 3, 16),
- /* CONST type_id=0 (void) */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 0),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid index",
- },
- {
- .descr = "array test. elem_type \"const void\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 1, 16),
- /* CONST type_id=0 (void) */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 0),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid elem",
- },
- {
- .descr = "array test. elem_type \"const void *\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* const void *[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 1, 16),
- /* CONST type_id=4 */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 4),
- /* void* */ /* [4] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 0),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- },
- {
- .descr = "array test. index_type \"const void *\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* const void *[16] */ /* [2] */
- BTF_TYPE_ARRAY_ENC(3, 3, 16),
- /* CONST type_id=4 */ /* [3] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_CONST, 0, 0), 4),
- /* void* */ /* [4] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_PTR, 0, 0), 0),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid index",
- },
- {
- .descr = "array test. t->size != 0\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* int[16] */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_ARRAY, 0, 0), 1),
- BTF_ARRAY_ENC(1, 1, 16),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "size != 0",
- },
- {
- .descr = "int test. invalid int_data",
- .raw_types = {
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_INT, 0, 0), 4),
- 0x10000000,
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid int_data",
- },
- {
- .descr = "invalid BTF_INFO",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_TYPE_ENC(0, 0x10000000, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "Invalid btf_info",
- },
- {
- .descr = "fwd test. t->type != 0\"",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* fwd type */ /* [2] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_FWD, 0, 0), 1),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "fwd_test_map",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .btf_load_err = true,
- .err_str = "type != 0",
- },
- {
- .descr = "arraymap invalid btf key (a bit field)",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* 32 bit int with 32 bit offset */ /* [2] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 32, 32, 8),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_map_check_btf",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 2,
- .value_type_id = 1,
- .max_entries = 4,
- .map_create_err = true,
- },
- {
- .descr = "arraymap invalid btf key (!= 32 bits)",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* 16 bit int with 0 bit offset */ /* [2] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 16, 2),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_map_check_btf",
- .key_size = sizeof(int),
- .value_size = sizeof(int),
- .key_type_id = 2,
- .value_type_id = 1,
- .max_entries = 4,
- .map_create_err = true,
- },
- {
- .descr = "arraymap invalid btf value (too small)",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_map_check_btf",
- .key_size = sizeof(int),
- /* btf_value_size < map->value_size */
- .value_size = sizeof(__u64),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .map_create_err = true,
- },
- {
- .descr = "arraymap invalid btf value (too big)",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "array_map_check_btf",
- .key_size = sizeof(int),
- /* btf_value_size > map->value_size */
- .value_size = sizeof(__u16),
- .key_type_id = 1,
- .value_type_id = 1,
- .max_entries = 4,
- .map_create_err = true,
- },
- }; /* struct btf_raw_test raw_tests[] */
- static const char *get_next_str(const char *start, const char *end)
- {
- return start < end - 1 ? start + 1 : NULL;
- }
- static int get_type_sec_size(const __u32 *raw_types)
- {
- int i;
- for (i = MAX_NR_RAW_TYPES - 1;
- i >= 0 && raw_types[i] != BTF_END_RAW;
- i--)
- ;
- return i < 0 ? i : i * sizeof(raw_types[0]);
- }
- static void *btf_raw_create(const struct btf_header *hdr,
- const __u32 *raw_types,
- const char *str,
- unsigned int str_sec_size,
- unsigned int *btf_size)
- {
- const char *next_str = str, *end_str = str + str_sec_size;
- unsigned int size_needed, offset;
- struct btf_header *ret_hdr;
- int i, type_sec_size;
- uint32_t *ret_types;
- void *raw_btf;
- type_sec_size = get_type_sec_size(raw_types);
- if (CHECK(type_sec_size < 0, "Cannot get nr_raw_types"))
- return NULL;
- size_needed = sizeof(*hdr) + type_sec_size + str_sec_size;
- raw_btf = malloc(size_needed);
- if (CHECK(!raw_btf, "Cannot allocate memory for raw_btf"))
- return NULL;
- /* Copy header */
- memcpy(raw_btf, hdr, sizeof(*hdr));
- offset = sizeof(*hdr);
- /* Copy type section */
- ret_types = raw_btf + offset;
- for (i = 0; i < type_sec_size / sizeof(raw_types[0]); i++) {
- if (raw_types[i] == NAME_TBD) {
- next_str = get_next_str(next_str, end_str);
- if (CHECK(!next_str, "Error in getting next_str")) {
- free(raw_btf);
- return NULL;
- }
- ret_types[i] = next_str - str;
- next_str += strlen(next_str);
- } else {
- ret_types[i] = raw_types[i];
- }
- }
- offset += type_sec_size;
- /* Copy string section */
- memcpy(raw_btf + offset, str, str_sec_size);
- ret_hdr = (struct btf_header *)raw_btf;
- ret_hdr->type_len = type_sec_size;
- ret_hdr->str_off = type_sec_size;
- ret_hdr->str_len = str_sec_size;
- *btf_size = size_needed;
- return raw_btf;
- }
- static int do_test_raw(unsigned int test_num)
- {
- struct btf_raw_test *test = &raw_tests[test_num - 1];
- struct bpf_create_map_attr create_attr = {};
- int map_fd = -1, btf_fd = -1;
- unsigned int raw_btf_size;
- struct btf_header *hdr;
- void *raw_btf;
- int err;
- fprintf(stderr, "BTF raw test[%u] (%s): ", test_num, test->descr);
- raw_btf = btf_raw_create(&hdr_tmpl,
- test->raw_types,
- test->str_sec,
- test->str_sec_size,
- &raw_btf_size);
- if (!raw_btf)
- return -1;
- hdr = raw_btf;
- hdr->hdr_len = (int)hdr->hdr_len + test->hdr_len_delta;
- hdr->type_off = (int)hdr->type_off + test->type_off_delta;
- hdr->str_off = (int)hdr->str_off + test->str_off_delta;
- hdr->str_len = (int)hdr->str_len + test->str_len_delta;
- *btf_log_buf = '\0';
- btf_fd = bpf_load_btf(raw_btf, raw_btf_size,
- btf_log_buf, BTF_LOG_BUF_SIZE,
- args.always_log);
- free(raw_btf);
- err = ((btf_fd == -1) != test->btf_load_err);
- if (CHECK(err, "btf_fd:%d test->btf_load_err:%u",
- btf_fd, test->btf_load_err) ||
- CHECK(test->err_str && !strstr(btf_log_buf, test->err_str),
- "expected err_str:%s", test->err_str)) {
- err = -1;
- goto done;
- }
- if (err || btf_fd == -1)
- goto done;
- create_attr.name = test->map_name;
- create_attr.map_type = test->map_type;
- create_attr.key_size = test->key_size;
- create_attr.value_size = test->value_size;
- create_attr.max_entries = test->max_entries;
- create_attr.btf_fd = btf_fd;
- create_attr.btf_key_type_id = test->key_type_id;
- create_attr.btf_value_type_id = test->value_type_id;
- map_fd = bpf_create_map_xattr(&create_attr);
- err = ((map_fd == -1) != test->map_create_err);
- CHECK(err, "map_fd:%d test->map_create_err:%u",
- map_fd, test->map_create_err);
- done:
- if (!err)
- fprintf(stderr, "OK");
- if (*btf_log_buf && (err || args.always_log))
- fprintf(stderr, "\n%s", btf_log_buf);
- if (btf_fd != -1)
- close(btf_fd);
- if (map_fd != -1)
- close(map_fd);
- return err;
- }
- static int test_raw(void)
- {
- unsigned int i;
- int err = 0;
- if (args.raw_test_num)
- return count_result(do_test_raw(args.raw_test_num));
- for (i = 1; i <= ARRAY_SIZE(raw_tests); i++)
- err |= count_result(do_test_raw(i));
- return err;
- }
- struct btf_get_info_test {
- const char *descr;
- const char *str_sec;
- __u32 raw_types[MAX_NR_RAW_TYPES];
- __u32 str_sec_size;
- int btf_size_delta;
- int (*special_test)(unsigned int test_num);
- };
- static int test_big_btf_info(unsigned int test_num);
- static int test_btf_id(unsigned int test_num);
- const struct btf_get_info_test get_info_tests[] = {
- {
- .descr = "== raw_btf_size+1",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .btf_size_delta = 1,
- },
- {
- .descr = "== raw_btf_size-3",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .btf_size_delta = -3,
- },
- {
- .descr = "Large bpf_btf_info",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .special_test = test_big_btf_info,
- },
- {
- .descr = "BTF ID",
- .raw_types = {
- /* int */ /* [1] */
- BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),
- /* unsigned int */ /* [2] */
- BTF_TYPE_INT_ENC(0, 0, 0, 32, 4),
- BTF_END_RAW,
- },
- .str_sec = "",
- .str_sec_size = sizeof(""),
- .special_test = test_btf_id,
- },
- };
- static inline __u64 ptr_to_u64(const void *ptr)
- {
- return (__u64)(unsigned long)ptr;
- }
- static int test_big_btf_info(unsigned int test_num)
- {
- const struct btf_get_info_test *test = &get_info_tests[test_num - 1];
- uint8_t *raw_btf = NULL, *user_btf = NULL;
- unsigned int raw_btf_size;
- struct {
- struct bpf_btf_info info;
- uint64_t garbage;
- } info_garbage;
- struct bpf_btf_info *info;
- int btf_fd = -1, err;
- uint32_t info_len;
- raw_btf = btf_raw_create(&hdr_tmpl,
- test->raw_types,
- test->str_sec,
- test->str_sec_size,
- &raw_btf_size);
- if (!raw_btf)
- return -1;
- *btf_log_buf = '\0';
- user_btf = malloc(raw_btf_size);
- if (CHECK(!user_btf, "!user_btf")) {
- err = -1;
- goto done;
- }
- btf_fd = bpf_load_btf(raw_btf, raw_btf_size,
- btf_log_buf, BTF_LOG_BUF_SIZE,
- args.always_log);
- if (CHECK(btf_fd == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- /*
- * GET_INFO should error out if the userspace info
- * has non zero tailing bytes.
- */
- info = &info_garbage.info;
- memset(info, 0, sizeof(*info));
- info_garbage.garbage = 0xdeadbeef;
- info_len = sizeof(info_garbage);
- info->btf = ptr_to_u64(user_btf);
- info->btf_size = raw_btf_size;
- err = bpf_obj_get_info_by_fd(btf_fd, info, &info_len);
- if (CHECK(!err, "!err")) {
- err = -1;
- goto done;
- }
- /*
- * GET_INFO should succeed even info_len is larger than
- * the kernel supported as long as tailing bytes are zero.
- * The kernel supported info len should also be returned
- * to userspace.
- */
- info_garbage.garbage = 0;
- err = bpf_obj_get_info_by_fd(btf_fd, info, &info_len);
- if (CHECK(err || info_len != sizeof(*info),
- "err:%d errno:%d info_len:%u sizeof(*info):%lu",
- err, errno, info_len, sizeof(*info))) {
- err = -1;
- goto done;
- }
- fprintf(stderr, "OK");
- done:
- if (*btf_log_buf && (err || args.always_log))
- fprintf(stderr, "\n%s", btf_log_buf);
- free(raw_btf);
- free(user_btf);
- if (btf_fd != -1)
- close(btf_fd);
- return err;
- }
- static int test_btf_id(unsigned int test_num)
- {
- const struct btf_get_info_test *test = &get_info_tests[test_num - 1];
- struct bpf_create_map_attr create_attr = {};
- uint8_t *raw_btf = NULL, *user_btf[2] = {};
- int btf_fd[2] = {-1, -1}, map_fd = -1;
- struct bpf_map_info map_info = {};
- struct bpf_btf_info info[2] = {};
- unsigned int raw_btf_size;
- uint32_t info_len;
- int err, i, ret;
- raw_btf = btf_raw_create(&hdr_tmpl,
- test->raw_types,
- test->str_sec,
- test->str_sec_size,
- &raw_btf_size);
- if (!raw_btf)
- return -1;
- *btf_log_buf = '\0';
- for (i = 0; i < 2; i++) {
- user_btf[i] = malloc(raw_btf_size);
- if (CHECK(!user_btf[i], "!user_btf[%d]", i)) {
- err = -1;
- goto done;
- }
- info[i].btf = ptr_to_u64(user_btf[i]);
- info[i].btf_size = raw_btf_size;
- }
- btf_fd[0] = bpf_load_btf(raw_btf, raw_btf_size,
- btf_log_buf, BTF_LOG_BUF_SIZE,
- args.always_log);
- if (CHECK(btf_fd[0] == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- /* Test BPF_OBJ_GET_INFO_BY_ID on btf_id */
- info_len = sizeof(info[0]);
- err = bpf_obj_get_info_by_fd(btf_fd[0], &info[0], &info_len);
- if (CHECK(err, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- btf_fd[1] = bpf_btf_get_fd_by_id(info[0].id);
- if (CHECK(btf_fd[1] == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- ret = 0;
- err = bpf_obj_get_info_by_fd(btf_fd[1], &info[1], &info_len);
- if (CHECK(err || info[0].id != info[1].id ||
- info[0].btf_size != info[1].btf_size ||
- (ret = memcmp(user_btf[0], user_btf[1], info[0].btf_size)),
- "err:%d errno:%d id0:%u id1:%u btf_size0:%u btf_size1:%u memcmp:%d",
- err, errno, info[0].id, info[1].id,
- info[0].btf_size, info[1].btf_size, ret)) {
- err = -1;
- goto done;
- }
- /* Test btf members in struct bpf_map_info */
- create_attr.name = "test_btf_id";
- create_attr.map_type = BPF_MAP_TYPE_ARRAY;
- create_attr.key_size = sizeof(int);
- create_attr.value_size = sizeof(unsigned int);
- create_attr.max_entries = 4;
- create_attr.btf_fd = btf_fd[0];
- create_attr.btf_key_type_id = 1;
- create_attr.btf_value_type_id = 2;
- map_fd = bpf_create_map_xattr(&create_attr);
- if (CHECK(map_fd == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- info_len = sizeof(map_info);
- err = bpf_obj_get_info_by_fd(map_fd, &map_info, &info_len);
- if (CHECK(err || map_info.btf_id != info[0].id ||
- map_info.btf_key_type_id != 1 || map_info.btf_value_type_id != 2,
- "err:%d errno:%d info.id:%u btf_id:%u btf_key_type_id:%u btf_value_type_id:%u",
- err, errno, info[0].id, map_info.btf_id, map_info.btf_key_type_id,
- map_info.btf_value_type_id)) {
- err = -1;
- goto done;
- }
- for (i = 0; i < 2; i++) {
- close(btf_fd[i]);
- btf_fd[i] = -1;
- }
- /* Test BTF ID is removed from the kernel */
- btf_fd[0] = bpf_btf_get_fd_by_id(map_info.btf_id);
- if (CHECK(btf_fd[0] == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- close(btf_fd[0]);
- btf_fd[0] = -1;
- /* The map holds the last ref to BTF and its btf_id */
- close(map_fd);
- map_fd = -1;
- btf_fd[0] = bpf_btf_get_fd_by_id(map_info.btf_id);
- if (CHECK(btf_fd[0] != -1, "BTF lingers")) {
- err = -1;
- goto done;
- }
- fprintf(stderr, "OK");
- done:
- if (*btf_log_buf && (err || args.always_log))
- fprintf(stderr, "\n%s", btf_log_buf);
- free(raw_btf);
- if (map_fd != -1)
- close(map_fd);
- for (i = 0; i < 2; i++) {
- free(user_btf[i]);
- if (btf_fd[i] != -1)
- close(btf_fd[i]);
- }
- return err;
- }
- static int do_test_get_info(unsigned int test_num)
- {
- const struct btf_get_info_test *test = &get_info_tests[test_num - 1];
- unsigned int raw_btf_size, user_btf_size, expected_nbytes;
- uint8_t *raw_btf = NULL, *user_btf = NULL;
- struct bpf_btf_info info = {};
- int btf_fd = -1, err, ret;
- uint32_t info_len;
- fprintf(stderr, "BTF GET_INFO test[%u] (%s): ",
- test_num, test->descr);
- if (test->special_test)
- return test->special_test(test_num);
- raw_btf = btf_raw_create(&hdr_tmpl,
- test->raw_types,
- test->str_sec,
- test->str_sec_size,
- &raw_btf_size);
- if (!raw_btf)
- return -1;
- *btf_log_buf = '\0';
- user_btf = malloc(raw_btf_size);
- if (CHECK(!user_btf, "!user_btf")) {
- err = -1;
- goto done;
- }
- btf_fd = bpf_load_btf(raw_btf, raw_btf_size,
- btf_log_buf, BTF_LOG_BUF_SIZE,
- args.always_log);
- if (CHECK(btf_fd == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- user_btf_size = (int)raw_btf_size + test->btf_size_delta;
- expected_nbytes = min(raw_btf_size, user_btf_size);
- if (raw_btf_size > expected_nbytes)
- memset(user_btf + expected_nbytes, 0xff,
- raw_btf_size - expected_nbytes);
- info_len = sizeof(info);
- info.btf = ptr_to_u64(user_btf);
- info.btf_size = user_btf_size;
- ret = 0;
- err = bpf_obj_get_info_by_fd(btf_fd, &info, &info_len);
- if (CHECK(err || !info.id || info_len != sizeof(info) ||
- info.btf_size != raw_btf_size ||
- (ret = memcmp(raw_btf, user_btf, expected_nbytes)),
- "err:%d errno:%d info.id:%u info_len:%u sizeof(info):%lu raw_btf_size:%u info.btf_size:%u expected_nbytes:%u memcmp:%d",
- err, errno, info.id, info_len, sizeof(info),
- raw_btf_size, info.btf_size, expected_nbytes, ret)) {
- err = -1;
- goto done;
- }
- while (expected_nbytes < raw_btf_size) {
- fprintf(stderr, "%u...", expected_nbytes);
- if (CHECK(user_btf[expected_nbytes++] != 0xff,
- "user_btf[%u]:%x != 0xff", expected_nbytes - 1,
- user_btf[expected_nbytes - 1])) {
- err = -1;
- goto done;
- }
- }
- fprintf(stderr, "OK");
- done:
- if (*btf_log_buf && (err || args.always_log))
- fprintf(stderr, "\n%s", btf_log_buf);
- free(raw_btf);
- free(user_btf);
- if (btf_fd != -1)
- close(btf_fd);
- return err;
- }
- static int test_get_info(void)
- {
- unsigned int i;
- int err = 0;
- if (args.get_info_test_num)
- return count_result(do_test_get_info(args.get_info_test_num));
- for (i = 1; i <= ARRAY_SIZE(get_info_tests); i++)
- err |= count_result(do_test_get_info(i));
- return err;
- }
- struct btf_file_test {
- const char *file;
- bool btf_kv_notfound;
- };
- static struct btf_file_test file_tests[] = {
- {
- .file = "test_btf_haskv.o",
- },
- {
- .file = "test_btf_nokv.o",
- .btf_kv_notfound = true,
- },
- };
- static int file_has_btf_elf(const char *fn)
- {
- Elf_Scn *scn = NULL;
- GElf_Ehdr ehdr;
- int elf_fd;
- Elf *elf;
- int ret;
- if (CHECK(elf_version(EV_CURRENT) == EV_NONE,
- "elf_version(EV_CURRENT) == EV_NONE"))
- return -1;
- elf_fd = open(fn, O_RDONLY);
- if (CHECK(elf_fd == -1, "open(%s): errno:%d", fn, errno))
- return -1;
- elf = elf_begin(elf_fd, ELF_C_READ, NULL);
- if (CHECK(!elf, "elf_begin(%s): %s", fn, elf_errmsg(elf_errno()))) {
- ret = -1;
- goto done;
- }
- if (CHECK(!gelf_getehdr(elf, &ehdr), "!gelf_getehdr(%s)", fn)) {
- ret = -1;
- goto done;
- }
- while ((scn = elf_nextscn(elf, scn))) {
- const char *sh_name;
- GElf_Shdr sh;
- if (CHECK(gelf_getshdr(scn, &sh) != &sh,
- "file:%s gelf_getshdr != &sh", fn)) {
- ret = -1;
- goto done;
- }
- sh_name = elf_strptr(elf, ehdr.e_shstrndx, sh.sh_name);
- if (!strcmp(sh_name, BTF_ELF_SEC)) {
- ret = 1;
- goto done;
- }
- }
- ret = 0;
- done:
- close(elf_fd);
- elf_end(elf);
- return ret;
- }
- static int do_test_file(unsigned int test_num)
- {
- const struct btf_file_test *test = &file_tests[test_num - 1];
- struct bpf_object *obj = NULL;
- struct bpf_program *prog;
- struct bpf_map *map;
- int err;
- fprintf(stderr, "BTF libbpf test[%u] (%s): ", test_num,
- test->file);
- err = file_has_btf_elf(test->file);
- if (err == -1)
- return err;
- if (err == 0) {
- fprintf(stderr, "SKIP. No ELF %s found", BTF_ELF_SEC);
- skip_cnt++;
- return 0;
- }
- obj = bpf_object__open(test->file);
- if (CHECK(IS_ERR(obj), "obj: %ld", PTR_ERR(obj)))
- return PTR_ERR(obj);
- err = bpf_object__btf_fd(obj);
- if (CHECK(err == -1, "bpf_object__btf_fd: -1"))
- goto done;
- prog = bpf_program__next(NULL, obj);
- if (CHECK(!prog, "Cannot find bpf_prog")) {
- err = -1;
- goto done;
- }
- bpf_program__set_type(prog, BPF_PROG_TYPE_TRACEPOINT);
- err = bpf_object__load(obj);
- if (CHECK(err < 0, "bpf_object__load: %d", err))
- goto done;
- map = bpf_object__find_map_by_name(obj, "btf_map");
- if (CHECK(!map, "btf_map not found")) {
- err = -1;
- goto done;
- }
- err = (bpf_map__btf_key_type_id(map) == 0 || bpf_map__btf_value_type_id(map) == 0)
- != test->btf_kv_notfound;
- if (CHECK(err, "btf_key_type_id:%u btf_value_type_id:%u test->btf_kv_notfound:%u",
- bpf_map__btf_key_type_id(map), bpf_map__btf_value_type_id(map),
- test->btf_kv_notfound))
- goto done;
- fprintf(stderr, "OK");
- done:
- bpf_object__close(obj);
- return err;
- }
- static int test_file(void)
- {
- unsigned int i;
- int err = 0;
- if (args.file_test_num)
- return count_result(do_test_file(args.file_test_num));
- for (i = 1; i <= ARRAY_SIZE(file_tests); i++)
- err |= count_result(do_test_file(i));
- return err;
- }
- const char *pprint_enum_str[] = {
- "ENUM_ZERO",
- "ENUM_ONE",
- "ENUM_TWO",
- "ENUM_THREE",
- };
- struct pprint_mapv {
- uint32_t ui32;
- uint16_t ui16;
- /* 2 bytes hole */
- int32_t si32;
- uint32_t unused_bits2a:2,
- bits28:28,
- unused_bits2b:2;
- union {
- uint64_t ui64;
- uint8_t ui8a[8];
- };
- enum {
- ENUM_ZERO,
- ENUM_ONE,
- ENUM_TWO,
- ENUM_THREE,
- } aenum;
- };
- static struct btf_raw_test pprint_test_template = {
- .raw_types = {
- /* unsighed char */ /* [1] */
- BTF_TYPE_INT_ENC(NAME_TBD, 0, 0, 8, 1),
- /* unsigned short */ /* [2] */
- BTF_TYPE_INT_ENC(NAME_TBD, 0, 0, 16, 2),
- /* unsigned int */ /* [3] */
- BTF_TYPE_INT_ENC(NAME_TBD, 0, 0, 32, 4),
- /* int */ /* [4] */
- BTF_TYPE_INT_ENC(NAME_TBD, BTF_INT_SIGNED, 0, 32, 4),
- /* unsigned long long */ /* [5] */
- BTF_TYPE_INT_ENC(NAME_TBD, 0, 0, 64, 8),
- /* 2 bits */ /* [6] */
- BTF_TYPE_INT_ENC(0, 0, 0, 2, 2),
- /* 28 bits */ /* [7] */
- BTF_TYPE_INT_ENC(0, 0, 0, 28, 4),
- /* uint8_t[8] */ /* [8] */
- BTF_TYPE_ARRAY_ENC(9, 1, 8),
- /* typedef unsigned char uint8_t */ /* [9] */
- BTF_TYPEDEF_ENC(NAME_TBD, 1),
- /* typedef unsigned short uint16_t */ /* [10] */
- BTF_TYPEDEF_ENC(NAME_TBD, 2),
- /* typedef unsigned int uint32_t */ /* [11] */
- BTF_TYPEDEF_ENC(NAME_TBD, 3),
- /* typedef int int32_t */ /* [12] */
- BTF_TYPEDEF_ENC(NAME_TBD, 4),
- /* typedef unsigned long long uint64_t *//* [13] */
- BTF_TYPEDEF_ENC(NAME_TBD, 5),
- /* union (anon) */ /* [14] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_UNION, 0, 2), 8),
- BTF_MEMBER_ENC(NAME_TBD, 13, 0),/* uint64_t ui64; */
- BTF_MEMBER_ENC(NAME_TBD, 8, 0), /* uint8_t ui8a[8]; */
- /* enum (anon) */ /* [15] */
- BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_ENUM, 0, 4), 4),
- BTF_ENUM_ENC(NAME_TBD, 0),
- BTF_ENUM_ENC(NAME_TBD, 1),
- BTF_ENUM_ENC(NAME_TBD, 2),
- BTF_ENUM_ENC(NAME_TBD, 3),
- /* struct pprint_mapv */ /* [16] */
- BTF_TYPE_ENC(NAME_TBD, BTF_INFO_ENC(BTF_KIND_STRUCT, 0, 8), 32),
- BTF_MEMBER_ENC(NAME_TBD, 11, 0), /* uint32_t ui32 */
- BTF_MEMBER_ENC(NAME_TBD, 10, 32), /* uint16_t ui16 */
- BTF_MEMBER_ENC(NAME_TBD, 12, 64), /* int32_t si32 */
- BTF_MEMBER_ENC(NAME_TBD, 6, 96), /* unused_bits2a */
- BTF_MEMBER_ENC(NAME_TBD, 7, 98), /* bits28 */
- BTF_MEMBER_ENC(NAME_TBD, 6, 126), /* unused_bits2b */
- BTF_MEMBER_ENC(0, 14, 128), /* union (anon) */
- BTF_MEMBER_ENC(NAME_TBD, 15, 192), /* aenum */
- BTF_END_RAW,
- },
- .str_sec = "\0unsigned char\0unsigned short\0unsigned int\0int\0unsigned long long\0uint8_t\0uint16_t\0uint32_t\0int32_t\0uint64_t\0ui64\0ui8a\0ENUM_ZERO\0ENUM_ONE\0ENUM_TWO\0ENUM_THREE\0pprint_mapv\0ui32\0ui16\0si32\0unused_bits2a\0bits28\0unused_bits2b\0aenum",
- .str_sec_size = sizeof("\0unsigned char\0unsigned short\0unsigned int\0int\0unsigned long long\0uint8_t\0uint16_t\0uint32_t\0int32_t\0uint64_t\0ui64\0ui8a\0ENUM_ZERO\0ENUM_ONE\0ENUM_TWO\0ENUM_THREE\0pprint_mapv\0ui32\0ui16\0si32\0unused_bits2a\0bits28\0unused_bits2b\0aenum"),
- .key_size = sizeof(unsigned int),
- .value_size = sizeof(struct pprint_mapv),
- .key_type_id = 3, /* unsigned int */
- .value_type_id = 16, /* struct pprint_mapv */
- .max_entries = 128 * 1024,
- };
- static struct btf_pprint_test_meta {
- const char *descr;
- enum bpf_map_type map_type;
- const char *map_name;
- bool ordered_map;
- bool lossless_map;
- } pprint_tests_meta[] = {
- {
- .descr = "BTF pretty print array",
- .map_type = BPF_MAP_TYPE_ARRAY,
- .map_name = "pprint_test_array",
- .ordered_map = true,
- .lossless_map = true,
- },
- {
- .descr = "BTF pretty print hash",
- .map_type = BPF_MAP_TYPE_HASH,
- .map_name = "pprint_test_hash",
- .ordered_map = false,
- .lossless_map = true,
- },
- {
- .descr = "BTF pretty print lru hash",
- .map_type = BPF_MAP_TYPE_LRU_HASH,
- .map_name = "pprint_test_lru_hash",
- .ordered_map = false,
- .lossless_map = false,
- },
- };
- static void set_pprint_mapv(struct pprint_mapv *v, uint32_t i)
- {
- v->ui32 = i;
- v->si32 = -i;
- v->unused_bits2a = 3;
- v->bits28 = i;
- v->unused_bits2b = 3;
- v->ui64 = i;
- v->aenum = i & 0x03;
- }
- static int do_test_pprint(void)
- {
- const struct btf_raw_test *test = &pprint_test_template;
- struct bpf_create_map_attr create_attr = {};
- unsigned int key, nr_read_elems;
- bool ordered_map, lossless_map;
- int map_fd = -1, btf_fd = -1;
- struct pprint_mapv mapv = {};
- unsigned int raw_btf_size;
- char expected_line[255];
- FILE *pin_file = NULL;
- char pin_path[255];
- size_t line_len = 0;
- char *line = NULL;
- uint8_t *raw_btf;
- ssize_t nread;
- int err, ret;
- fprintf(stderr, "%s......", test->descr);
- raw_btf = btf_raw_create(&hdr_tmpl, test->raw_types,
- test->str_sec, test->str_sec_size,
- &raw_btf_size);
- if (!raw_btf)
- return -1;
- *btf_log_buf = '\0';
- btf_fd = bpf_load_btf(raw_btf, raw_btf_size,
- btf_log_buf, BTF_LOG_BUF_SIZE,
- args.always_log);
- free(raw_btf);
- if (CHECK(btf_fd == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- create_attr.name = test->map_name;
- create_attr.map_type = test->map_type;
- create_attr.key_size = test->key_size;
- create_attr.value_size = test->value_size;
- create_attr.max_entries = test->max_entries;
- create_attr.btf_fd = btf_fd;
- create_attr.btf_key_type_id = test->key_type_id;
- create_attr.btf_value_type_id = test->value_type_id;
- map_fd = bpf_create_map_xattr(&create_attr);
- if (CHECK(map_fd == -1, "errno:%d", errno)) {
- err = -1;
- goto done;
- }
- ret = snprintf(pin_path, sizeof(pin_path), "%s/%s",
- "/sys/fs/bpf", test->map_name);
- if (CHECK(ret == sizeof(pin_path), "pin_path %s/%s is too long",
- "/sys/fs/bpf", test->map_name)) {
- err = -1;
- goto done;
- }
- err = bpf_obj_pin(map_fd, pin_path);
- if (CHECK(err, "bpf_obj_pin(%s): errno:%d.", pin_path, errno))
- goto done;
- for (key = 0; key < test->max_entries; key++) {
- set_pprint_mapv(&mapv, key);
- bpf_map_update_elem(map_fd, &key, &mapv, 0);
- }
- pin_file = fopen(pin_path, "r");
- if (CHECK(!pin_file, "fopen(%s): errno:%d", pin_path, errno)) {
- err = -1;
- goto done;
- }
- /* Skip lines start with '#' */
- while ((nread = getline(&line, &line_len, pin_file)) > 0 &&
- *line == '#')
- ;
- if (CHECK(nread <= 0, "Unexpected EOF")) {
- err = -1;
- goto done;
- }
- nr_read_elems = 0;
- ordered_map = test->ordered_map;
- lossless_map = test->lossless_map;
- do {
- ssize_t nexpected_line;
- unsigned int next_key;
- next_key = ordered_map ? nr_read_elems : atoi(line);
- set_pprint_mapv(&mapv, next_key);
- nexpected_line = snprintf(expected_line, sizeof(expected_line),
- "%u: {%u,0,%d,0x%x,0x%x,0x%x,{%lu|[%u,%u,%u,%u,%u,%u,%u,%u]},%s}\n",
- next_key,
- mapv.ui32, mapv.si32,
- mapv.unused_bits2a, mapv.bits28, mapv.unused_bits2b,
- mapv.ui64,
- mapv.ui8a[0], mapv.ui8a[1], mapv.ui8a[2], mapv.ui8a[3],
- mapv.ui8a[4], mapv.ui8a[5], mapv.ui8a[6], mapv.ui8a[7],
- pprint_enum_str[mapv.aenum]);
- if (CHECK(nexpected_line == sizeof(expected_line),
- "expected_line is too long")) {
- err = -1;
- goto done;
- }
- if (strcmp(expected_line, line)) {
- err = -1;
- fprintf(stderr, "unexpected pprint output\n");
- fprintf(stderr, "expected: %s", expected_line);
- fprintf(stderr, " read: %s", line);
- goto done;
- }
- nread = getline(&line, &line_len, pin_file);
- } while (++nr_read_elems < test->max_entries && nread > 0);
- if (lossless_map &&
- CHECK(nr_read_elems < test->max_entries,
- "Unexpected EOF. nr_read_elems:%u test->max_entries:%u",
- nr_read_elems, test->max_entries)) {
- err = -1;
- goto done;
- }
- if (CHECK(nread > 0, "Unexpected extra pprint output: %s", line)) {
- err = -1;
- goto done;
- }
- err = 0;
- done:
- if (!err)
- fprintf(stderr, "OK");
- if (*btf_log_buf && (err || args.always_log))
- fprintf(stderr, "\n%s", btf_log_buf);
- if (btf_fd != -1)
- close(btf_fd);
- if (map_fd != -1)
- close(map_fd);
- if (pin_file)
- fclose(pin_file);
- unlink(pin_path);
- free(line);
- return err;
- }
- static int test_pprint(void)
- {
- unsigned int i;
- int err = 0;
- for (i = 0; i < ARRAY_SIZE(pprint_tests_meta); i++) {
- pprint_test_template.descr = pprint_tests_meta[i].descr;
- pprint_test_template.map_type = pprint_tests_meta[i].map_type;
- pprint_test_template.map_name = pprint_tests_meta[i].map_name;
- pprint_test_template.ordered_map = pprint_tests_meta[i].ordered_map;
- pprint_test_template.lossless_map = pprint_tests_meta[i].lossless_map;
- err |= count_result(do_test_pprint());
- }
- return err;
- }
- static void usage(const char *cmd)
- {
- fprintf(stderr, "Usage: %s [-l] [[-r test_num (1 - %zu)] | [-g test_num (1 - %zu)] | [-f test_num (1 - %zu)] | [-p]]\n",
- cmd, ARRAY_SIZE(raw_tests), ARRAY_SIZE(get_info_tests),
- ARRAY_SIZE(file_tests));
- }
- static int parse_args(int argc, char **argv)
- {
- const char *optstr = "lpf:r:g:";
- int opt;
- while ((opt = getopt(argc, argv, optstr)) != -1) {
- switch (opt) {
- case 'l':
- args.always_log = true;
- break;
- case 'f':
- args.file_test_num = atoi(optarg);
- args.file_test = true;
- break;
- case 'r':
- args.raw_test_num = atoi(optarg);
- args.raw_test = true;
- break;
- case 'g':
- args.get_info_test_num = atoi(optarg);
- args.get_info_test = true;
- break;
- case 'p':
- args.pprint_test = true;
- break;
- case 'h':
- usage(argv[0]);
- exit(0);
- default:
- usage(argv[0]);
- return -1;
- }
- }
- if (args.raw_test_num &&
- (args.raw_test_num < 1 ||
- args.raw_test_num > ARRAY_SIZE(raw_tests))) {
- fprintf(stderr, "BTF raw test number must be [1 - %zu]\n",
- ARRAY_SIZE(raw_tests));
- return -1;
- }
- if (args.file_test_num &&
- (args.file_test_num < 1 ||
- args.file_test_num > ARRAY_SIZE(file_tests))) {
- fprintf(stderr, "BTF file test number must be [1 - %zu]\n",
- ARRAY_SIZE(file_tests));
- return -1;
- }
- if (args.get_info_test_num &&
- (args.get_info_test_num < 1 ||
- args.get_info_test_num > ARRAY_SIZE(get_info_tests))) {
- fprintf(stderr, "BTF get info test number must be [1 - %zu]\n",
- ARRAY_SIZE(get_info_tests));
- return -1;
- }
- return 0;
- }
- static void print_summary(void)
- {
- fprintf(stderr, "PASS:%u SKIP:%u FAIL:%u\n",
- pass_cnt - skip_cnt, skip_cnt, error_cnt);
- }
- int main(int argc, char **argv)
- {
- int err = 0;
- err = parse_args(argc, argv);
- if (err)
- return err;
- if (args.always_log)
- libbpf_set_print(__base_pr, __base_pr, __base_pr);
- if (args.raw_test)
- err |= test_raw();
- if (args.get_info_test)
- err |= test_get_info();
- if (args.file_test)
- err |= test_file();
- if (args.pprint_test)
- err |= test_pprint();
- if (args.raw_test || args.get_info_test || args.file_test ||
- args.pprint_test)
- goto done;
- err |= test_raw();
- err |= test_get_info();
- err |= test_file();
- done:
- print_summary();
- return err;
- }
|