| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618 |
- // SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause)
- /* Copyright (C) 2017-2018 Netronome Systems, Inc. */
- #ifndef _GNU_SOURCE
- #define _GNU_SOURCE
- #endif
- #include <errno.h>
- #include <fcntl.h>
- #include <signal.h>
- #include <stdarg.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <time.h>
- #include <unistd.h>
- #include <net/if.h>
- #include <sys/ioctl.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <sys/syscall.h>
- #include <dirent.h>
- #include <linux/err.h>
- #include <linux/perf_event.h>
- #include <linux/sizes.h>
- #include <linux/keyctl.h>
- #include <bpf/bpf.h>
- #include <bpf/btf.h>
- #include <bpf/hashmap.h>
- #include <bpf/libbpf.h>
- #include <bpf/libbpf_internal.h>
- #include <bpf/skel_internal.h>
- #include "cfg.h"
- #include "main.h"
- #include "xlated_dumper.h"
- #define BPF_METADATA_PREFIX "bpf_metadata_"
- #define BPF_METADATA_PREFIX_LEN (sizeof(BPF_METADATA_PREFIX) - 1)
- enum dump_mode {
- DUMP_JITED,
- DUMP_XLATED,
- };
- static const bool attach_types[] = {
- [BPF_SK_SKB_STREAM_PARSER] = true,
- [BPF_SK_SKB_STREAM_VERDICT] = true,
- [BPF_SK_SKB_VERDICT] = true,
- [BPF_SK_MSG_VERDICT] = true,
- [BPF_FLOW_DISSECTOR] = true,
- [__MAX_BPF_ATTACH_TYPE] = false,
- };
- /* Textual representations traditionally used by the program and kept around
- * for the sake of backwards compatibility.
- */
- static const char * const attach_type_strings[] = {
- [BPF_SK_SKB_STREAM_PARSER] = "stream_parser",
- [BPF_SK_SKB_STREAM_VERDICT] = "stream_verdict",
- [BPF_SK_SKB_VERDICT] = "skb_verdict",
- [BPF_SK_MSG_VERDICT] = "msg_verdict",
- [__MAX_BPF_ATTACH_TYPE] = NULL,
- };
- static struct hashmap *prog_table;
- static enum bpf_attach_type parse_attach_type(const char *str)
- {
- enum bpf_attach_type type;
- for (type = 0; type < __MAX_BPF_ATTACH_TYPE; type++) {
- if (attach_types[type]) {
- const char *attach_type_str;
- attach_type_str = libbpf_bpf_attach_type_str(type);
- if (!strcmp(str, attach_type_str))
- return type;
- }
- if (attach_type_strings[type] &&
- is_prefix(str, attach_type_strings[type]))
- return type;
- }
- return __MAX_BPF_ATTACH_TYPE;
- }
- static int prep_prog_info(struct bpf_prog_info *const info, enum dump_mode mode,
- void **info_data, size_t *const info_data_sz)
- {
- struct bpf_prog_info holder = {};
- size_t needed = 0;
- void *ptr;
- if (mode == DUMP_JITED) {
- holder.jited_prog_len = info->jited_prog_len;
- needed += info->jited_prog_len;
- } else {
- holder.xlated_prog_len = info->xlated_prog_len;
- needed += info->xlated_prog_len;
- }
- holder.nr_jited_ksyms = info->nr_jited_ksyms;
- needed += info->nr_jited_ksyms * sizeof(__u64);
- holder.nr_jited_func_lens = info->nr_jited_func_lens;
- needed += info->nr_jited_func_lens * sizeof(__u32);
- holder.nr_func_info = info->nr_func_info;
- holder.func_info_rec_size = info->func_info_rec_size;
- needed += info->nr_func_info * info->func_info_rec_size;
- holder.nr_line_info = info->nr_line_info;
- holder.line_info_rec_size = info->line_info_rec_size;
- needed += info->nr_line_info * info->line_info_rec_size;
- holder.nr_jited_line_info = info->nr_jited_line_info;
- holder.jited_line_info_rec_size = info->jited_line_info_rec_size;
- needed += info->nr_jited_line_info * info->jited_line_info_rec_size;
- if (needed > *info_data_sz) {
- ptr = realloc(*info_data, needed);
- if (!ptr)
- return -1;
- *info_data = ptr;
- *info_data_sz = needed;
- }
- ptr = *info_data;
- if (mode == DUMP_JITED) {
- holder.jited_prog_insns = ptr_to_u64(ptr);
- ptr += holder.jited_prog_len;
- } else {
- holder.xlated_prog_insns = ptr_to_u64(ptr);
- ptr += holder.xlated_prog_len;
- }
- holder.jited_ksyms = ptr_to_u64(ptr);
- ptr += holder.nr_jited_ksyms * sizeof(__u64);
- holder.jited_func_lens = ptr_to_u64(ptr);
- ptr += holder.nr_jited_func_lens * sizeof(__u32);
- holder.func_info = ptr_to_u64(ptr);
- ptr += holder.nr_func_info * holder.func_info_rec_size;
- holder.line_info = ptr_to_u64(ptr);
- ptr += holder.nr_line_info * holder.line_info_rec_size;
- holder.jited_line_info = ptr_to_u64(ptr);
- ptr += holder.nr_jited_line_info * holder.jited_line_info_rec_size;
- *info = holder;
- return 0;
- }
- static void print_boot_time(__u64 nsecs, char *buf, unsigned int size)
- {
- struct timespec real_time_ts, boot_time_ts;
- time_t wallclock_secs;
- struct tm load_tm;
- buf[--size] = '\0';
- if (clock_gettime(CLOCK_REALTIME, &real_time_ts) ||
- clock_gettime(CLOCK_BOOTTIME, &boot_time_ts)) {
- perror("Can't read clocks");
- snprintf(buf, size, "%llu", nsecs / 1000000000);
- return;
- }
- wallclock_secs = (real_time_ts.tv_sec - boot_time_ts.tv_sec) +
- (real_time_ts.tv_nsec - boot_time_ts.tv_nsec + nsecs) /
- 1000000000;
- if (!localtime_r(&wallclock_secs, &load_tm)) {
- snprintf(buf, size, "%llu", nsecs / 1000000000);
- return;
- }
- if (json_output)
- strftime(buf, size, "%s", &load_tm);
- else
- strftime(buf, size, "%FT%T%z", &load_tm);
- }
- static void show_prog_maps(int fd, __u32 num_maps)
- {
- struct bpf_prog_info info = {};
- __u32 len = sizeof(info);
- __u32 map_ids[num_maps];
- unsigned int i;
- int err;
- info.nr_map_ids = num_maps;
- info.map_ids = ptr_to_u64(map_ids);
- err = bpf_prog_get_info_by_fd(fd, &info, &len);
- if (err || !info.nr_map_ids)
- return;
- if (json_output) {
- jsonw_name(json_wtr, "map_ids");
- jsonw_start_array(json_wtr);
- for (i = 0; i < info.nr_map_ids; i++)
- jsonw_uint(json_wtr, map_ids[i]);
- jsonw_end_array(json_wtr);
- } else {
- printf(" map_ids ");
- for (i = 0; i < info.nr_map_ids; i++)
- printf("%u%s", map_ids[i],
- i == info.nr_map_ids - 1 ? "" : ",");
- }
- }
- static void *find_metadata(int prog_fd, struct bpf_map_info *map_info)
- {
- struct bpf_prog_info prog_info;
- __u32 prog_info_len;
- __u32 map_info_len;
- void *value = NULL;
- __u32 *map_ids;
- int nr_maps;
- int key = 0;
- int map_fd;
- int ret;
- __u32 i;
- memset(&prog_info, 0, sizeof(prog_info));
- prog_info_len = sizeof(prog_info);
- ret = bpf_prog_get_info_by_fd(prog_fd, &prog_info, &prog_info_len);
- if (ret)
- return NULL;
- if (!prog_info.nr_map_ids)
- return NULL;
- map_ids = calloc(prog_info.nr_map_ids, sizeof(__u32));
- if (!map_ids)
- return NULL;
- nr_maps = prog_info.nr_map_ids;
- memset(&prog_info, 0, sizeof(prog_info));
- prog_info.nr_map_ids = nr_maps;
- prog_info.map_ids = ptr_to_u64(map_ids);
- prog_info_len = sizeof(prog_info);
- ret = bpf_prog_get_info_by_fd(prog_fd, &prog_info, &prog_info_len);
- if (ret)
- goto free_map_ids;
- for (i = 0; i < prog_info.nr_map_ids; i++) {
- map_fd = bpf_map_get_fd_by_id(map_ids[i]);
- if (map_fd < 0)
- goto free_map_ids;
- memset(map_info, 0, sizeof(*map_info));
- map_info_len = sizeof(*map_info);
- ret = bpf_map_get_info_by_fd(map_fd, map_info, &map_info_len);
- if (ret < 0) {
- close(map_fd);
- goto free_map_ids;
- }
- if (map_info->type != BPF_MAP_TYPE_ARRAY ||
- map_info->key_size != sizeof(int) ||
- map_info->max_entries != 1 ||
- !map_info->btf_value_type_id ||
- !strstr(map_info->name, ".rodata")) {
- close(map_fd);
- continue;
- }
- value = malloc(map_info->value_size);
- if (!value) {
- close(map_fd);
- goto free_map_ids;
- }
- if (bpf_map_lookup_elem(map_fd, &key, value)) {
- close(map_fd);
- free(value);
- value = NULL;
- goto free_map_ids;
- }
- close(map_fd);
- break;
- }
- free_map_ids:
- free(map_ids);
- return value;
- }
- static bool has_metadata_prefix(const char *s)
- {
- return strncmp(s, BPF_METADATA_PREFIX, BPF_METADATA_PREFIX_LEN) == 0;
- }
- static void show_prog_metadata(int fd, __u32 num_maps)
- {
- const struct btf_type *t_datasec, *t_var;
- struct bpf_map_info map_info;
- struct btf_var_secinfo *vsi;
- bool printed_header = false;
- unsigned int i, vlen;
- void *value = NULL;
- const char *name;
- struct btf *btf;
- int err;
- if (!num_maps)
- return;
- memset(&map_info, 0, sizeof(map_info));
- value = find_metadata(fd, &map_info);
- if (!value)
- return;
- btf = btf__load_from_kernel_by_id(map_info.btf_id);
- if (!btf)
- goto out_free;
- t_datasec = btf__type_by_id(btf, map_info.btf_value_type_id);
- if (!btf_is_datasec(t_datasec))
- goto out_free;
- vlen = btf_vlen(t_datasec);
- vsi = btf_var_secinfos(t_datasec);
- /* We don't proceed to check the kinds of the elements of the DATASEC.
- * The verifier enforces them to be BTF_KIND_VAR.
- */
- if (json_output) {
- struct btf_dumper d = {
- .btf = btf,
- .jw = json_wtr,
- .is_plain_text = false,
- };
- for (i = 0; i < vlen; i++, vsi++) {
- t_var = btf__type_by_id(btf, vsi->type);
- name = btf__name_by_offset(btf, t_var->name_off);
- if (!has_metadata_prefix(name))
- continue;
- if (!printed_header) {
- jsonw_name(json_wtr, "metadata");
- jsonw_start_object(json_wtr);
- printed_header = true;
- }
- jsonw_name(json_wtr, name + BPF_METADATA_PREFIX_LEN);
- err = btf_dumper_type(&d, t_var->type, value + vsi->offset);
- if (err) {
- p_err("btf dump failed: %d", err);
- break;
- }
- }
- if (printed_header)
- jsonw_end_object(json_wtr);
- } else {
- json_writer_t *btf_wtr;
- struct btf_dumper d = {
- .btf = btf,
- .is_plain_text = true,
- };
- for (i = 0; i < vlen; i++, vsi++) {
- t_var = btf__type_by_id(btf, vsi->type);
- name = btf__name_by_offset(btf, t_var->name_off);
- if (!has_metadata_prefix(name))
- continue;
- if (!printed_header) {
- printf("\tmetadata:");
- btf_wtr = jsonw_new(stdout);
- if (!btf_wtr) {
- p_err("jsonw alloc failed");
- goto out_free;
- }
- d.jw = btf_wtr,
- printed_header = true;
- }
- printf("\n\t\t%s = ", name + BPF_METADATA_PREFIX_LEN);
- jsonw_reset(btf_wtr);
- err = btf_dumper_type(&d, t_var->type, value + vsi->offset);
- if (err) {
- p_err("btf dump failed: %d", err);
- break;
- }
- }
- if (printed_header)
- jsonw_destroy(&btf_wtr);
- }
- out_free:
- btf__free(btf);
- free(value);
- }
- static void print_prog_header_json(struct bpf_prog_info *info, int fd)
- {
- const char *prog_type_str;
- char prog_name[MAX_PROG_FULL_NAME];
- jsonw_uint_field(json_wtr, "id", info->id);
- prog_type_str = libbpf_bpf_prog_type_str(info->type);
- if (prog_type_str)
- jsonw_string_field(json_wtr, "type", prog_type_str);
- else
- jsonw_uint_field(json_wtr, "type", info->type);
- if (*info->name) {
- get_prog_full_name(info, fd, prog_name, sizeof(prog_name));
- jsonw_string_field(json_wtr, "name", prog_name);
- }
- jsonw_name(json_wtr, "tag");
- jsonw_printf(json_wtr, "\"" BPF_TAG_FMT "\"",
- info->tag[0], info->tag[1], info->tag[2], info->tag[3],
- info->tag[4], info->tag[5], info->tag[6], info->tag[7]);
- jsonw_bool_field(json_wtr, "gpl_compatible", info->gpl_compatible);
- if (info->run_time_ns) {
- jsonw_uint_field(json_wtr, "run_time_ns", info->run_time_ns);
- jsonw_uint_field(json_wtr, "run_cnt", info->run_cnt);
- }
- if (info->recursion_misses)
- jsonw_uint_field(json_wtr, "recursion_misses", info->recursion_misses);
- }
- static void print_prog_json(struct bpf_prog_info *info, int fd, bool orphaned)
- {
- char *memlock;
- jsonw_start_object(json_wtr);
- print_prog_header_json(info, fd);
- print_dev_json(info->ifindex, info->netns_dev, info->netns_ino);
- if (info->load_time) {
- char buf[32];
- print_boot_time(info->load_time, buf, sizeof(buf));
- /* Piggy back on load_time, since 0 uid is a valid one */
- jsonw_name(json_wtr, "loaded_at");
- jsonw_printf(json_wtr, "%s", buf);
- jsonw_uint_field(json_wtr, "uid", info->created_by_uid);
- }
- jsonw_bool_field(json_wtr, "orphaned", orphaned);
- jsonw_uint_field(json_wtr, "bytes_xlated", info->xlated_prog_len);
- if (info->jited_prog_len) {
- jsonw_bool_field(json_wtr, "jited", true);
- jsonw_uint_field(json_wtr, "bytes_jited", info->jited_prog_len);
- } else {
- jsonw_bool_field(json_wtr, "jited", false);
- }
- memlock = get_fdinfo(fd, "memlock");
- if (memlock)
- jsonw_int_field(json_wtr, "bytes_memlock", atoll(memlock));
- free(memlock);
- if (info->nr_map_ids)
- show_prog_maps(fd, info->nr_map_ids);
- if (info->btf_id)
- jsonw_int_field(json_wtr, "btf_id", info->btf_id);
- if (!hashmap__empty(prog_table)) {
- struct hashmap_entry *entry;
- jsonw_name(json_wtr, "pinned");
- jsonw_start_array(json_wtr);
- hashmap__for_each_key_entry(prog_table, entry, info->id)
- jsonw_string(json_wtr, entry->pvalue);
- jsonw_end_array(json_wtr);
- }
- emit_obj_refs_json(refs_table, info->id, json_wtr);
- show_prog_metadata(fd, info->nr_map_ids);
- jsonw_end_object(json_wtr);
- }
- static void print_prog_header_plain(struct bpf_prog_info *info, int fd)
- {
- const char *prog_type_str;
- char prog_name[MAX_PROG_FULL_NAME];
- printf("%u: ", info->id);
- prog_type_str = libbpf_bpf_prog_type_str(info->type);
- if (prog_type_str)
- printf("%s ", prog_type_str);
- else
- printf("type %u ", info->type);
- if (*info->name) {
- get_prog_full_name(info, fd, prog_name, sizeof(prog_name));
- printf("name %s ", prog_name);
- }
- printf("tag ");
- fprint_hex(stdout, info->tag, BPF_TAG_SIZE, "");
- print_dev_plain(info->ifindex, info->netns_dev, info->netns_ino);
- printf("%s", info->gpl_compatible ? " gpl" : "");
- if (info->run_time_ns)
- printf(" run_time_ns %llu run_cnt %llu",
- info->run_time_ns, info->run_cnt);
- if (info->recursion_misses)
- printf(" recursion_misses %llu", info->recursion_misses);
- printf("\n");
- }
- static void print_prog_plain(struct bpf_prog_info *info, int fd, bool orphaned)
- {
- char *memlock;
- print_prog_header_plain(info, fd);
- if (info->load_time) {
- char buf[32];
- print_boot_time(info->load_time, buf, sizeof(buf));
- /* Piggy back on load_time, since 0 uid is a valid one */
- printf("\tloaded_at %s uid %u\n", buf, info->created_by_uid);
- }
- printf("\txlated %uB", info->xlated_prog_len);
- if (info->jited_prog_len)
- printf(" jited %uB", info->jited_prog_len);
- else
- printf(" not jited");
- memlock = get_fdinfo(fd, "memlock");
- if (memlock)
- printf(" memlock %sB", memlock);
- free(memlock);
- if (orphaned)
- printf(" orphaned");
- if (info->nr_map_ids)
- show_prog_maps(fd, info->nr_map_ids);
- if (!hashmap__empty(prog_table)) {
- struct hashmap_entry *entry;
- hashmap__for_each_key_entry(prog_table, entry, info->id)
- printf("\n\tpinned %s", (char *)entry->pvalue);
- }
- if (info->btf_id)
- printf("\n\tbtf_id %u", info->btf_id);
- emit_obj_refs_plain(refs_table, info->id, "\n\tpids ");
- printf("\n");
- show_prog_metadata(fd, info->nr_map_ids);
- }
- static int show_prog(int fd)
- {
- struct bpf_prog_info info = {};
- __u32 len = sizeof(info);
- int err;
- err = bpf_prog_get_info_by_fd(fd, &info, &len);
- if (err && err != -ENODEV) {
- p_err("can't get prog info: %s", strerror(errno));
- return -1;
- }
- if (json_output)
- print_prog_json(&info, fd, err == -ENODEV);
- else
- print_prog_plain(&info, fd, err == -ENODEV);
- return 0;
- }
- static int do_show_subset(int argc, char **argv)
- {
- int *fds = NULL;
- int nb_fds, i;
- int err = -1;
- fds = malloc(sizeof(int));
- if (!fds) {
- p_err("mem alloc failed");
- return -1;
- }
- nb_fds = prog_parse_fds(&argc, &argv, &fds);
- if (nb_fds < 1)
- goto exit_free;
- if (json_output && nb_fds > 1)
- jsonw_start_array(json_wtr); /* root array */
- for (i = 0; i < nb_fds; i++) {
- err = show_prog(fds[i]);
- if (err) {
- for (; i < nb_fds; i++)
- close(fds[i]);
- break;
- }
- close(fds[i]);
- }
- if (json_output && nb_fds > 1)
- jsonw_end_array(json_wtr); /* root array */
- exit_free:
- free(fds);
- return err;
- }
- static int do_show(int argc, char **argv)
- {
- __u32 id = 0;
- int err;
- int fd;
- if (show_pinned) {
- prog_table = hashmap__new(hash_fn_for_key_as_id,
- equal_fn_for_key_as_id, NULL);
- if (IS_ERR(prog_table)) {
- p_err("failed to create hashmap for pinned paths");
- return -1;
- }
- build_pinned_obj_table(prog_table, BPF_OBJ_PROG);
- }
- build_obj_refs_table(&refs_table, BPF_OBJ_PROG);
- if (argc == 2)
- return do_show_subset(argc, argv);
- if (argc)
- return BAD_ARG();
- if (json_output)
- jsonw_start_array(json_wtr);
- while (true) {
- err = bpf_prog_get_next_id(id, &id);
- if (err) {
- if (errno == ENOENT) {
- err = 0;
- break;
- }
- p_err("can't get next program: %s%s", strerror(errno),
- errno == EINVAL ? " -- kernel too old?" : "");
- err = -1;
- break;
- }
- fd = bpf_prog_get_fd_by_id(id);
- if (fd < 0) {
- if (errno == ENOENT)
- continue;
- p_err("can't get prog by id (%u): %s",
- id, strerror(errno));
- err = -1;
- break;
- }
- err = show_prog(fd);
- close(fd);
- if (err)
- break;
- }
- if (json_output)
- jsonw_end_array(json_wtr);
- delete_obj_refs_table(refs_table);
- if (show_pinned)
- delete_pinned_obj_table(prog_table);
- return err;
- }
- static int
- prog_dump(struct bpf_prog_info *info, enum dump_mode mode,
- char *filepath, bool opcodes, bool visual, bool linum)
- {
- struct bpf_prog_linfo *prog_linfo = NULL;
- const char *disasm_opt = NULL;
- struct dump_data dd = {};
- void *func_info = NULL;
- struct btf *btf = NULL;
- char func_sig[1024];
- unsigned char *buf;
- __u32 member_len;
- int fd, err = -1;
- ssize_t n;
- if (mode == DUMP_JITED) {
- if (info->jited_prog_len == 0 || !info->jited_prog_insns) {
- p_err("error retrieving jit dump: no instructions returned or kernel.kptr_restrict set?");
- return -1;
- }
- buf = u64_to_ptr(info->jited_prog_insns);
- member_len = info->jited_prog_len;
- } else { /* DUMP_XLATED */
- if (info->xlated_prog_len == 0 || !info->xlated_prog_insns) {
- p_err("error retrieving insn dump: kernel.kptr_restrict set?");
- return -1;
- }
- buf = u64_to_ptr(info->xlated_prog_insns);
- member_len = info->xlated_prog_len;
- }
- if (info->btf_id) {
- btf = btf__load_from_kernel_by_id(info->btf_id);
- if (!btf) {
- p_err("failed to get btf");
- return -1;
- }
- }
- func_info = u64_to_ptr(info->func_info);
- if (info->nr_line_info) {
- prog_linfo = bpf_prog_linfo__new(info);
- if (!prog_linfo)
- p_info("error in processing bpf_line_info. continue without it.");
- }
- if (filepath) {
- fd = open(filepath, O_WRONLY | O_CREAT | O_TRUNC, 0600);
- if (fd < 0) {
- p_err("can't open file %s: %s", filepath,
- strerror(errno));
- goto exit_free;
- }
- n = write(fd, buf, member_len);
- close(fd);
- if (n != (ssize_t)member_len) {
- p_err("error writing output file: %s",
- n < 0 ? strerror(errno) : "short write");
- goto exit_free;
- }
- if (json_output)
- jsonw_null(json_wtr);
- } else if (mode == DUMP_JITED) {
- const char *name = NULL;
- if (info->ifindex) {
- name = ifindex_to_arch(info->ifindex, info->netns_dev,
- info->netns_ino, &disasm_opt);
- if (!name)
- goto exit_free;
- }
- if (info->nr_jited_func_lens && info->jited_func_lens) {
- struct kernel_sym *sym = NULL;
- struct bpf_func_info *record;
- char sym_name[SYM_MAX_NAME];
- unsigned char *img = buf;
- __u64 *ksyms = NULL;
- __u32 *lens;
- __u32 i;
- if (info->nr_jited_ksyms) {
- kernel_syms_load(&dd);
- ksyms = u64_to_ptr(info->jited_ksyms);
- }
- if (json_output)
- jsonw_start_array(json_wtr);
- lens = u64_to_ptr(info->jited_func_lens);
- for (i = 0; i < info->nr_jited_func_lens; i++) {
- if (ksyms) {
- sym = kernel_syms_search(&dd, ksyms[i]);
- if (sym)
- sprintf(sym_name, "%s", sym->name);
- else
- sprintf(sym_name, "0x%016llx", ksyms[i]);
- } else {
- strcpy(sym_name, "unknown");
- }
- if (func_info) {
- record = func_info + i * info->func_info_rec_size;
- btf_dumper_type_only(btf, record->type_id,
- func_sig,
- sizeof(func_sig));
- }
- if (json_output) {
- jsonw_start_object(json_wtr);
- if (func_info && func_sig[0] != '\0') {
- jsonw_name(json_wtr, "proto");
- jsonw_string(json_wtr, func_sig);
- }
- jsonw_name(json_wtr, "name");
- jsonw_string(json_wtr, sym_name);
- jsonw_name(json_wtr, "insns");
- } else {
- if (func_info && func_sig[0] != '\0')
- printf("%s:\n", func_sig);
- printf("%s:\n", sym_name);
- }
- if (ksyms) {
- if (disasm_print_insn(img, lens[i], opcodes,
- name, disasm_opt, btf,
- prog_linfo, ksyms[i], i,
- linum))
- goto exit_free;
- } else {
- if (disasm_print_insn(img, lens[i], opcodes,
- name, disasm_opt, btf,
- NULL, 0, 0, false))
- goto exit_free;
- }
- img += lens[i];
- if (json_output)
- jsonw_end_object(json_wtr);
- else
- printf("\n");
- }
- if (json_output)
- jsonw_end_array(json_wtr);
- } else {
- if (disasm_print_insn(buf, member_len, opcodes, name,
- disasm_opt, btf, NULL, 0, 0,
- false))
- goto exit_free;
- }
- } else {
- kernel_syms_load(&dd);
- dd.nr_jited_ksyms = info->nr_jited_ksyms;
- dd.jited_ksyms = u64_to_ptr(info->jited_ksyms);
- dd.btf = btf;
- dd.func_info = func_info;
- dd.finfo_rec_size = info->func_info_rec_size;
- dd.prog_linfo = prog_linfo;
- if (json_output)
- dump_xlated_json(&dd, buf, member_len, opcodes, linum);
- else if (visual)
- dump_xlated_cfg(&dd, buf, member_len, opcodes, linum);
- else
- dump_xlated_plain(&dd, buf, member_len, opcodes, linum);
- kernel_syms_destroy(&dd);
- }
- err = 0;
- exit_free:
- btf__free(btf);
- bpf_prog_linfo__free(prog_linfo);
- return err;
- }
- static int do_dump(int argc, char **argv)
- {
- struct bpf_prog_info info;
- __u32 info_len = sizeof(info);
- size_t info_data_sz = 0;
- void *info_data = NULL;
- char *filepath = NULL;
- bool opcodes = false;
- bool visual = false;
- enum dump_mode mode;
- bool linum = false;
- int nb_fds, i = 0;
- int *fds = NULL;
- int err = -1;
- if (is_prefix(*argv, "jited")) {
- if (disasm_init())
- return -1;
- mode = DUMP_JITED;
- } else if (is_prefix(*argv, "xlated")) {
- mode = DUMP_XLATED;
- } else {
- p_err("expected 'xlated' or 'jited', got: %s", *argv);
- return -1;
- }
- NEXT_ARG();
- if (argc < 2)
- usage();
- fds = malloc(sizeof(int));
- if (!fds) {
- p_err("mem alloc failed");
- return -1;
- }
- nb_fds = prog_parse_fds(&argc, &argv, &fds);
- if (nb_fds < 1)
- goto exit_free;
- while (argc) {
- if (is_prefix(*argv, "file")) {
- NEXT_ARG();
- if (!argc) {
- p_err("expected file path");
- goto exit_close;
- }
- if (nb_fds > 1) {
- p_err("several programs matched");
- goto exit_close;
- }
- filepath = *argv;
- NEXT_ARG();
- } else if (is_prefix(*argv, "opcodes")) {
- opcodes = true;
- NEXT_ARG();
- } else if (is_prefix(*argv, "visual")) {
- if (nb_fds > 1) {
- p_err("several programs matched");
- goto exit_close;
- }
- visual = true;
- NEXT_ARG();
- } else if (is_prefix(*argv, "linum")) {
- linum = true;
- NEXT_ARG();
- } else {
- usage();
- goto exit_close;
- }
- }
- if (filepath && (opcodes || visual || linum)) {
- p_err("'file' is not compatible with 'opcodes', 'visual', or 'linum'");
- goto exit_close;
- }
- if (json_output && visual) {
- p_err("'visual' is not compatible with JSON output");
- goto exit_close;
- }
- if (json_output && nb_fds > 1)
- jsonw_start_array(json_wtr); /* root array */
- for (i = 0; i < nb_fds; i++) {
- memset(&info, 0, sizeof(info));
- err = bpf_prog_get_info_by_fd(fds[i], &info, &info_len);
- if (err) {
- p_err("can't get prog info: %s", strerror(errno));
- break;
- }
- err = prep_prog_info(&info, mode, &info_data, &info_data_sz);
- if (err) {
- p_err("can't grow prog info_data");
- break;
- }
- err = bpf_prog_get_info_by_fd(fds[i], &info, &info_len);
- if (err) {
- p_err("can't get prog info: %s", strerror(errno));
- break;
- }
- if (json_output && nb_fds > 1) {
- jsonw_start_object(json_wtr); /* prog object */
- print_prog_header_json(&info, fds[i]);
- jsonw_name(json_wtr, "insns");
- } else if (nb_fds > 1) {
- print_prog_header_plain(&info, fds[i]);
- }
- err = prog_dump(&info, mode, filepath, opcodes, visual, linum);
- if (json_output && nb_fds > 1)
- jsonw_end_object(json_wtr); /* prog object */
- else if (i != nb_fds - 1 && nb_fds > 1)
- printf("\n");
- if (err)
- break;
- close(fds[i]);
- }
- if (json_output && nb_fds > 1)
- jsonw_end_array(json_wtr); /* root array */
- exit_close:
- for (; i < nb_fds; i++)
- close(fds[i]);
- exit_free:
- free(info_data);
- free(fds);
- return err;
- }
- static int do_pin(int argc, char **argv)
- {
- int err;
- err = do_pin_any(argc, argv, prog_parse_fd);
- if (!err && json_output)
- jsonw_null(json_wtr);
- return err;
- }
- struct map_replace {
- int idx;
- int fd;
- char *name;
- };
- static int map_replace_compar(const void *p1, const void *p2)
- {
- const struct map_replace *a = p1, *b = p2;
- return a->idx - b->idx;
- }
- static int parse_attach_detach_args(int argc, char **argv, int *progfd,
- enum bpf_attach_type *attach_type,
- int *mapfd)
- {
- if (!REQ_ARGS(3))
- return -EINVAL;
- *progfd = prog_parse_fd(&argc, &argv);
- if (*progfd < 0)
- return *progfd;
- *attach_type = parse_attach_type(*argv);
- if (*attach_type == __MAX_BPF_ATTACH_TYPE) {
- p_err("invalid attach/detach type");
- return -EINVAL;
- }
- if (*attach_type == BPF_FLOW_DISSECTOR) {
- *mapfd = 0;
- return 0;
- }
- NEXT_ARG();
- if (!REQ_ARGS(2))
- return -EINVAL;
- *mapfd = map_parse_fd(&argc, &argv, 0);
- if (*mapfd < 0)
- return *mapfd;
- return 0;
- }
- static int do_attach(int argc, char **argv)
- {
- enum bpf_attach_type attach_type;
- int err, progfd;
- int mapfd;
- err = parse_attach_detach_args(argc, argv,
- &progfd, &attach_type, &mapfd);
- if (err)
- return err;
- err = bpf_prog_attach(progfd, mapfd, attach_type, 0);
- if (err) {
- p_err("failed prog attach to map");
- return -EINVAL;
- }
- if (json_output)
- jsonw_null(json_wtr);
- return 0;
- }
- static int do_detach(int argc, char **argv)
- {
- enum bpf_attach_type attach_type;
- int err, progfd;
- int mapfd;
- err = parse_attach_detach_args(argc, argv,
- &progfd, &attach_type, &mapfd);
- if (err)
- return err;
- err = bpf_prog_detach2(progfd, mapfd, attach_type);
- if (err) {
- p_err("failed prog detach from map");
- return -EINVAL;
- }
- if (json_output)
- jsonw_null(json_wtr);
- return 0;
- }
- enum prog_tracelog_mode {
- TRACE_STDOUT,
- TRACE_STDERR,
- };
- static int
- prog_tracelog_stream(int prog_fd, enum prog_tracelog_mode mode)
- {
- FILE *file = mode == TRACE_STDOUT ? stdout : stderr;
- int stream_id = mode == TRACE_STDOUT ? 1 : 2;
- char buf[512];
- int ret;
- ret = 0;
- do {
- ret = bpf_prog_stream_read(prog_fd, stream_id, buf, sizeof(buf), NULL);
- if (ret > 0)
- fwrite(buf, sizeof(buf[0]), ret, file);
- } while (ret > 0);
- fflush(file);
- return ret ? -1 : 0;
- }
- static int do_tracelog_any(int argc, char **argv)
- {
- enum prog_tracelog_mode mode;
- int fd;
- if (argc == 0)
- return do_tracelog(argc, argv);
- if (!is_prefix(*argv, "stdout") && !is_prefix(*argv, "stderr"))
- usage();
- mode = is_prefix(*argv, "stdout") ? TRACE_STDOUT : TRACE_STDERR;
- NEXT_ARG();
- if (!REQ_ARGS(2))
- return -1;
- fd = prog_parse_fd(&argc, &argv);
- if (fd < 0)
- return -1;
- return prog_tracelog_stream(fd, mode);
- }
- static int check_single_stdin(char *file_data_in, char *file_ctx_in)
- {
- if (file_data_in && file_ctx_in &&
- !strcmp(file_data_in, "-") && !strcmp(file_ctx_in, "-")) {
- p_err("cannot use standard input for both data_in and ctx_in");
- return -1;
- }
- return 0;
- }
- static int get_run_data(const char *fname, void **data_ptr, unsigned int *size)
- {
- size_t block_size = 256;
- size_t buf_size = block_size;
- size_t nb_read = 0;
- void *tmp;
- FILE *f;
- if (!fname) {
- *data_ptr = NULL;
- *size = 0;
- return 0;
- }
- if (!strcmp(fname, "-"))
- f = stdin;
- else
- f = fopen(fname, "r");
- if (!f) {
- p_err("failed to open %s: %s", fname, strerror(errno));
- return -1;
- }
- *data_ptr = malloc(block_size);
- if (!*data_ptr) {
- p_err("failed to allocate memory for data_in/ctx_in: %s",
- strerror(errno));
- goto err_fclose;
- }
- while ((nb_read += fread(*data_ptr + nb_read, 1, block_size, f))) {
- if (feof(f))
- break;
- if (ferror(f)) {
- p_err("failed to read data_in/ctx_in from %s: %s",
- fname, strerror(errno));
- goto err_free;
- }
- if (nb_read > buf_size - block_size) {
- if (buf_size == UINT32_MAX) {
- p_err("data_in/ctx_in is too long (max: %u)",
- UINT32_MAX);
- goto err_free;
- }
- /* No space for fread()-ing next chunk; realloc() */
- buf_size *= 2;
- tmp = realloc(*data_ptr, buf_size);
- if (!tmp) {
- p_err("failed to reallocate data_in/ctx_in: %s",
- strerror(errno));
- goto err_free;
- }
- *data_ptr = tmp;
- }
- }
- if (f != stdin)
- fclose(f);
- *size = nb_read;
- return 0;
- err_free:
- free(*data_ptr);
- *data_ptr = NULL;
- err_fclose:
- if (f != stdin)
- fclose(f);
- return -1;
- }
- static void hex_print(void *data, unsigned int size, FILE *f)
- {
- size_t i, j;
- char c;
- for (i = 0; i < size; i += 16) {
- /* Row offset */
- fprintf(f, "%07zx\t", i);
- /* Hexadecimal values */
- for (j = i; j < i + 16 && j < size; j++)
- fprintf(f, "%02x%s", *(uint8_t *)(data + j),
- j % 2 ? " " : "");
- for (; j < i + 16; j++)
- fprintf(f, " %s", j % 2 ? " " : "");
- /* ASCII values (if relevant), '.' otherwise */
- fprintf(f, "| ");
- for (j = i; j < i + 16 && j < size; j++) {
- c = *(char *)(data + j);
- if (c < ' ' || c > '~')
- c = '.';
- fprintf(f, "%c%s", c, j == i + 7 ? " " : "");
- }
- fprintf(f, "\n");
- }
- }
- static int
- print_run_output(void *data, unsigned int size, const char *fname,
- const char *json_key)
- {
- size_t nb_written;
- FILE *f;
- if (!fname)
- return 0;
- if (!strcmp(fname, "-")) {
- f = stdout;
- if (json_output) {
- jsonw_name(json_wtr, json_key);
- print_data_json(data, size);
- } else {
- hex_print(data, size, f);
- }
- return 0;
- }
- f = fopen(fname, "w");
- if (!f) {
- p_err("failed to open %s: %s", fname, strerror(errno));
- return -1;
- }
- nb_written = fwrite(data, 1, size, f);
- fclose(f);
- if (nb_written != size) {
- p_err("failed to write output data/ctx: %s", strerror(errno));
- return -1;
- }
- return 0;
- }
- static int alloc_run_data(void **data_ptr, unsigned int size_out)
- {
- *data_ptr = calloc(size_out, 1);
- if (!*data_ptr) {
- p_err("failed to allocate memory for output data/ctx: %s",
- strerror(errno));
- return -1;
- }
- return 0;
- }
- static int do_run(int argc, char **argv)
- {
- char *data_fname_in = NULL, *data_fname_out = NULL;
- char *ctx_fname_in = NULL, *ctx_fname_out = NULL;
- const unsigned int default_size = SZ_32K;
- void *data_in = NULL, *data_out = NULL;
- void *ctx_in = NULL, *ctx_out = NULL;
- unsigned int repeat = 1;
- int fd, err;
- LIBBPF_OPTS(bpf_test_run_opts, test_attr);
- if (!REQ_ARGS(4))
- return -1;
- fd = prog_parse_fd(&argc, &argv);
- if (fd < 0)
- return -1;
- while (argc) {
- if (detect_common_prefix(*argv, "data_in", "data_out",
- "data_size_out", NULL))
- return -1;
- if (detect_common_prefix(*argv, "ctx_in", "ctx_out",
- "ctx_size_out", NULL))
- return -1;
- if (is_prefix(*argv, "data_in")) {
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- data_fname_in = GET_ARG();
- if (check_single_stdin(data_fname_in, ctx_fname_in))
- return -1;
- } else if (is_prefix(*argv, "data_out")) {
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- data_fname_out = GET_ARG();
- } else if (is_prefix(*argv, "data_size_out")) {
- char *endptr;
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- test_attr.data_size_out = strtoul(*argv, &endptr, 0);
- if (*endptr) {
- p_err("can't parse %s as output data size",
- *argv);
- return -1;
- }
- NEXT_ARG();
- } else if (is_prefix(*argv, "ctx_in")) {
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- ctx_fname_in = GET_ARG();
- if (check_single_stdin(data_fname_in, ctx_fname_in))
- return -1;
- } else if (is_prefix(*argv, "ctx_out")) {
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- ctx_fname_out = GET_ARG();
- } else if (is_prefix(*argv, "ctx_size_out")) {
- char *endptr;
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- test_attr.ctx_size_out = strtoul(*argv, &endptr, 0);
- if (*endptr) {
- p_err("can't parse %s as output context size",
- *argv);
- return -1;
- }
- NEXT_ARG();
- } else if (is_prefix(*argv, "repeat")) {
- char *endptr;
- NEXT_ARG();
- if (!REQ_ARGS(1))
- return -1;
- repeat = strtoul(*argv, &endptr, 0);
- if (*endptr) {
- p_err("can't parse %s as repeat number",
- *argv);
- return -1;
- }
- NEXT_ARG();
- } else {
- p_err("expected no more arguments, 'data_in', 'data_out', 'data_size_out', 'ctx_in', 'ctx_out', 'ctx_size_out' or 'repeat', got: '%s'?",
- *argv);
- return -1;
- }
- }
- err = get_run_data(data_fname_in, &data_in, &test_attr.data_size_in);
- if (err)
- return -1;
- if (data_in) {
- if (!test_attr.data_size_out)
- test_attr.data_size_out = default_size;
- err = alloc_run_data(&data_out, test_attr.data_size_out);
- if (err)
- goto free_data_in;
- }
- err = get_run_data(ctx_fname_in, &ctx_in, &test_attr.ctx_size_in);
- if (err)
- goto free_data_out;
- if (ctx_in) {
- if (!test_attr.ctx_size_out)
- test_attr.ctx_size_out = default_size;
- err = alloc_run_data(&ctx_out, test_attr.ctx_size_out);
- if (err)
- goto free_ctx_in;
- }
- test_attr.repeat = repeat;
- test_attr.data_in = data_in;
- test_attr.data_out = data_out;
- test_attr.ctx_in = ctx_in;
- test_attr.ctx_out = ctx_out;
- err = bpf_prog_test_run_opts(fd, &test_attr);
- if (err) {
- p_err("failed to run program: %s", strerror(errno));
- goto free_ctx_out;
- }
- err = 0;
- if (json_output)
- jsonw_start_object(json_wtr); /* root */
- /* Do not exit on errors occurring when printing output data/context,
- * we still want to print return value and duration for program run.
- */
- if (test_attr.data_size_out)
- err += print_run_output(test_attr.data_out,
- test_attr.data_size_out,
- data_fname_out, "data_out");
- if (test_attr.ctx_size_out)
- err += print_run_output(test_attr.ctx_out,
- test_attr.ctx_size_out,
- ctx_fname_out, "ctx_out");
- if (json_output) {
- jsonw_uint_field(json_wtr, "retval", test_attr.retval);
- jsonw_uint_field(json_wtr, "duration", test_attr.duration);
- jsonw_end_object(json_wtr); /* root */
- } else {
- fprintf(stdout, "Return value: %u, duration%s: %uns\n",
- test_attr.retval,
- repeat > 1 ? " (average)" : "", test_attr.duration);
- }
- free_ctx_out:
- free(ctx_out);
- free_ctx_in:
- free(ctx_in);
- free_data_out:
- free(data_out);
- free_data_in:
- free(data_in);
- return err;
- }
- static int
- get_prog_type_by_name(const char *name, enum bpf_prog_type *prog_type,
- enum bpf_attach_type *expected_attach_type)
- {
- libbpf_print_fn_t print_backup;
- int ret;
- ret = libbpf_prog_type_by_name(name, prog_type, expected_attach_type);
- if (!ret)
- return ret;
- /* libbpf_prog_type_by_name() failed, let's re-run with debug level */
- print_backup = libbpf_set_print(print_all_levels);
- ret = libbpf_prog_type_by_name(name, prog_type, expected_attach_type);
- libbpf_set_print(print_backup);
- return ret;
- }
- static int
- auto_attach_program(struct bpf_program *prog, const char *path)
- {
- struct bpf_link *link;
- int err;
- link = bpf_program__attach(prog);
- if (!link) {
- p_info("Program %s does not support autoattach, falling back to pinning",
- bpf_program__name(prog));
- return bpf_obj_pin(bpf_program__fd(prog), path);
- }
- err = bpf_link__pin(link, path);
- bpf_link__destroy(link);
- return err;
- }
- static int
- auto_attach_programs(struct bpf_object *obj, const char *path)
- {
- struct bpf_program *prog;
- char buf[PATH_MAX];
- int err;
- bpf_object__for_each_program(prog, obj) {
- err = pathname_concat(buf, sizeof(buf), path, bpf_program__name(prog));
- if (err)
- goto err_unpin_programs;
- err = auto_attach_program(prog, buf);
- if (err)
- goto err_unpin_programs;
- }
- return 0;
- err_unpin_programs:
- while ((prog = bpf_object__prev_program(obj, prog))) {
- if (pathname_concat(buf, sizeof(buf), path, bpf_program__name(prog)))
- continue;
- bpf_program__unpin(prog, buf);
- }
- return err;
- }
- static int load_with_options(int argc, char **argv, bool first_prog_only)
- {
- enum bpf_prog_type common_prog_type = BPF_PROG_TYPE_UNSPEC;
- DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts,
- .relaxed_maps = relaxed_maps,
- );
- enum bpf_attach_type expected_attach_type;
- struct map_replace *map_replace = NULL;
- struct bpf_program *prog = NULL, *pos;
- unsigned int old_map_fds = 0;
- const char *pinmaps = NULL;
- __u32 xdpmeta_ifindex = 0;
- __u32 offload_ifindex = 0;
- bool auto_attach = false;
- struct bpf_object *obj;
- struct bpf_map *map;
- const char *pinfile;
- unsigned int i, j;
- const char *file;
- int idx, err;
- if (!REQ_ARGS(2))
- return -1;
- file = GET_ARG();
- pinfile = GET_ARG();
- while (argc) {
- if (is_prefix(*argv, "type")) {
- NEXT_ARG();
- if (common_prog_type != BPF_PROG_TYPE_UNSPEC) {
- p_err("program type already specified");
- goto err_free_reuse_maps;
- }
- if (!REQ_ARGS(1))
- goto err_free_reuse_maps;
- err = libbpf_prog_type_by_name(*argv, &common_prog_type,
- &expected_attach_type);
- if (err < 0) {
- /* Put a '/' at the end of type to appease libbpf */
- char *type = malloc(strlen(*argv) + 2);
- if (!type) {
- p_err("mem alloc failed");
- goto err_free_reuse_maps;
- }
- *type = 0;
- strcat(type, *argv);
- strcat(type, "/");
- err = get_prog_type_by_name(type, &common_prog_type,
- &expected_attach_type);
- free(type);
- if (err < 0)
- goto err_free_reuse_maps;
- }
- NEXT_ARG();
- } else if (is_prefix(*argv, "map")) {
- void *new_map_replace;
- char *endptr, *name;
- int fd;
- NEXT_ARG();
- if (!REQ_ARGS(4))
- goto err_free_reuse_maps;
- if (is_prefix(*argv, "idx")) {
- NEXT_ARG();
- idx = strtoul(*argv, &endptr, 0);
- if (*endptr) {
- p_err("can't parse %s as IDX", *argv);
- goto err_free_reuse_maps;
- }
- name = NULL;
- } else if (is_prefix(*argv, "name")) {
- NEXT_ARG();
- name = *argv;
- idx = -1;
- } else {
- p_err("expected 'idx' or 'name', got: '%s'?",
- *argv);
- goto err_free_reuse_maps;
- }
- NEXT_ARG();
- fd = map_parse_fd(&argc, &argv, 0);
- if (fd < 0)
- goto err_free_reuse_maps;
- new_map_replace = libbpf_reallocarray(map_replace,
- old_map_fds + 1,
- sizeof(*map_replace));
- if (!new_map_replace) {
- p_err("mem alloc failed");
- goto err_free_reuse_maps;
- }
- map_replace = new_map_replace;
- map_replace[old_map_fds].idx = idx;
- map_replace[old_map_fds].name = name;
- map_replace[old_map_fds].fd = fd;
- old_map_fds++;
- } else if (is_prefix(*argv, "dev")) {
- p_info("Warning: 'bpftool prog load [...] dev <ifname>' syntax is deprecated.\n"
- "Going further, please use 'offload_dev <ifname>' to offload program to device.\n"
- "For applications using XDP hints only, use 'xdpmeta_dev <ifname>'.");
- goto offload_dev;
- } else if (is_prefix(*argv, "offload_dev")) {
- offload_dev:
- NEXT_ARG();
- if (offload_ifindex) {
- p_err("offload_dev already specified");
- goto err_free_reuse_maps;
- } else if (xdpmeta_ifindex) {
- p_err("xdpmeta_dev and offload_dev are mutually exclusive");
- goto err_free_reuse_maps;
- }
- if (!REQ_ARGS(1))
- goto err_free_reuse_maps;
- offload_ifindex = if_nametoindex(*argv);
- if (!offload_ifindex) {
- p_err("unrecognized netdevice '%s': %s",
- *argv, strerror(errno));
- goto err_free_reuse_maps;
- }
- NEXT_ARG();
- } else if (is_prefix(*argv, "xdpmeta_dev")) {
- NEXT_ARG();
- if (xdpmeta_ifindex) {
- p_err("xdpmeta_dev already specified");
- goto err_free_reuse_maps;
- } else if (offload_ifindex) {
- p_err("xdpmeta_dev and offload_dev are mutually exclusive");
- goto err_free_reuse_maps;
- }
- if (!REQ_ARGS(1))
- goto err_free_reuse_maps;
- xdpmeta_ifindex = if_nametoindex(*argv);
- if (!xdpmeta_ifindex) {
- p_err("unrecognized netdevice '%s': %s",
- *argv, strerror(errno));
- goto err_free_reuse_maps;
- }
- NEXT_ARG();
- } else if (is_prefix(*argv, "pinmaps")) {
- NEXT_ARG();
- if (!REQ_ARGS(1))
- goto err_free_reuse_maps;
- pinmaps = GET_ARG();
- } else if (is_prefix(*argv, "autoattach")) {
- auto_attach = true;
- NEXT_ARG();
- } else if (is_prefix(*argv, "kernel_btf")) {
- NEXT_ARG();
- if (!REQ_ARGS(1))
- goto err_free_reuse_maps;
- open_opts.btf_custom_path = GET_ARG();
- } else {
- p_err("expected no more arguments, "
- "'type', 'map', 'offload_dev', 'xdpmeta_dev', 'pinmaps', "
- "'autoattach', or 'kernel_btf', got: '%s'?",
- *argv);
- goto err_free_reuse_maps;
- }
- }
- set_max_rlimit();
- if (verifier_logs)
- /* log_level1 + log_level2 + stats, but not stable UAPI */
- open_opts.kernel_log_level = 1 + 2 + 4;
- obj = bpf_object__open_file(file, &open_opts);
- if (!obj) {
- p_err("failed to open object file");
- goto err_free_reuse_maps;
- }
- bpf_object__for_each_program(pos, obj) {
- enum bpf_prog_type prog_type = common_prog_type;
- if (prog_type == BPF_PROG_TYPE_UNSPEC) {
- const char *sec_name = bpf_program__section_name(pos);
- err = get_prog_type_by_name(sec_name, &prog_type,
- &expected_attach_type);
- if (err < 0)
- goto err_close_obj;
- }
- if (prog_type == BPF_PROG_TYPE_XDP && xdpmeta_ifindex) {
- bpf_program__set_flags(pos, BPF_F_XDP_DEV_BOUND_ONLY);
- bpf_program__set_ifindex(pos, xdpmeta_ifindex);
- } else {
- bpf_program__set_ifindex(pos, offload_ifindex);
- }
- if (bpf_program__type(pos) != prog_type)
- bpf_program__set_type(pos, prog_type);
- bpf_program__set_expected_attach_type(pos, expected_attach_type);
- }
- qsort(map_replace, old_map_fds, sizeof(*map_replace),
- map_replace_compar);
- /* After the sort maps by name will be first on the list, because they
- * have idx == -1. Resolve them.
- */
- j = 0;
- while (j < old_map_fds && map_replace[j].name) {
- i = 0;
- bpf_object__for_each_map(map, obj) {
- if (!strcmp(bpf_map__name(map), map_replace[j].name)) {
- map_replace[j].idx = i;
- break;
- }
- i++;
- }
- if (map_replace[j].idx == -1) {
- p_err("unable to find map '%s'", map_replace[j].name);
- goto err_close_obj;
- }
- j++;
- }
- /* Resort if any names were resolved */
- if (j)
- qsort(map_replace, old_map_fds, sizeof(*map_replace),
- map_replace_compar);
- /* Set ifindex and name reuse */
- j = 0;
- idx = 0;
- bpf_object__for_each_map(map, obj) {
- if (bpf_map__type(map) != BPF_MAP_TYPE_PERF_EVENT_ARRAY)
- bpf_map__set_ifindex(map, offload_ifindex);
- if (j < old_map_fds && idx == map_replace[j].idx) {
- err = bpf_map__reuse_fd(map, map_replace[j++].fd);
- if (err) {
- p_err("unable to set up map reuse: %d", err);
- goto err_close_obj;
- }
- /* Next reuse wants to apply to the same map */
- if (j < old_map_fds && map_replace[j].idx == idx) {
- p_err("replacement for map idx %d specified more than once",
- idx);
- goto err_close_obj;
- }
- }
- idx++;
- }
- if (j < old_map_fds) {
- p_err("map idx '%d' not used", map_replace[j].idx);
- goto err_close_obj;
- }
- err = bpf_object__load(obj);
- if (err) {
- p_err("failed to load object file");
- goto err_close_obj;
- }
- if (first_prog_only)
- err = mount_bpffs_for_file(pinfile);
- else
- err = create_and_mount_bpffs_dir(pinfile);
- if (err)
- goto err_close_obj;
- if (first_prog_only) {
- prog = bpf_object__next_program(obj, NULL);
- if (!prog) {
- p_err("object file doesn't contain any bpf program");
- goto err_close_obj;
- }
- if (auto_attach)
- err = auto_attach_program(prog, pinfile);
- else
- err = bpf_obj_pin(bpf_program__fd(prog), pinfile);
- if (err) {
- p_err("failed to pin program %s",
- bpf_program__section_name(prog));
- goto err_close_obj;
- }
- } else {
- if (auto_attach)
- err = auto_attach_programs(obj, pinfile);
- else
- err = bpf_object__pin_programs(obj, pinfile);
- if (err) {
- p_err("failed to pin all programs");
- goto err_close_obj;
- }
- }
- if (pinmaps) {
- err = create_and_mount_bpffs_dir(pinmaps);
- if (err)
- goto err_unpin;
- err = bpf_object__pin_maps(obj, pinmaps);
- if (err) {
- p_err("failed to pin all maps");
- goto err_unpin;
- }
- }
- if (json_output)
- jsonw_null(json_wtr);
- bpf_object__close(obj);
- for (i = 0; i < old_map_fds; i++)
- close(map_replace[i].fd);
- free(map_replace);
- return 0;
- err_unpin:
- if (first_prog_only)
- unlink(pinfile);
- else
- bpf_object__unpin_programs(obj, pinfile);
- err_close_obj:
- bpf_object__close(obj);
- err_free_reuse_maps:
- for (i = 0; i < old_map_fds; i++)
- close(map_replace[i].fd);
- free(map_replace);
- return -1;
- }
- static int count_open_fds(void)
- {
- DIR *dp = opendir("/proc/self/fd");
- struct dirent *de;
- int cnt = -3;
- if (!dp)
- return -1;
- while ((de = readdir(dp)))
- cnt++;
- closedir(dp);
- return cnt;
- }
- static int try_loader(struct gen_loader_opts *gen)
- {
- struct bpf_load_and_run_opts opts = {};
- struct bpf_loader_ctx *ctx;
- char sig_buf[MAX_SIG_SIZE];
- __u8 prog_sha[SHA256_DIGEST_LENGTH];
- int ctx_sz = sizeof(*ctx) + 64 * max(sizeof(struct bpf_map_desc),
- sizeof(struct bpf_prog_desc));
- int log_buf_sz = (1u << 24) - 1;
- int err, fds_before, fd_delta;
- char *log_buf = NULL;
- ctx = alloca(ctx_sz);
- memset(ctx, 0, ctx_sz);
- ctx->sz = ctx_sz;
- if (verifier_logs) {
- ctx->log_level = 1 + 2 + 4;
- ctx->log_size = log_buf_sz;
- log_buf = malloc(log_buf_sz);
- if (!log_buf)
- return -ENOMEM;
- ctx->log_buf = (long) log_buf;
- }
- opts.ctx = ctx;
- opts.data = gen->data;
- opts.data_sz = gen->data_sz;
- opts.insns = gen->insns;
- opts.insns_sz = gen->insns_sz;
- fds_before = count_open_fds();
- if (sign_progs) {
- opts.excl_prog_hash = prog_sha;
- opts.excl_prog_hash_sz = sizeof(prog_sha);
- opts.signature = sig_buf;
- opts.signature_sz = MAX_SIG_SIZE;
- opts.keyring_id = KEY_SPEC_SESSION_KEYRING;
- err = bpftool_prog_sign(&opts);
- if (err < 0) {
- p_err("failed to sign program");
- goto out;
- }
- err = register_session_key(cert_path);
- if (err < 0) {
- p_err("failed to add session key");
- goto out;
- }
- }
- err = bpf_load_and_run(&opts);
- fd_delta = count_open_fds() - fds_before;
- if (err < 0 || verifier_logs) {
- fprintf(stderr, "err %d\n%s\n%s", err, opts.errstr, log_buf);
- if (fd_delta && err < 0)
- fprintf(stderr, "loader prog leaked %d FDs\n",
- fd_delta);
- }
- out:
- free(log_buf);
- return err;
- }
- static int do_loader(int argc, char **argv)
- {
- DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts);
- DECLARE_LIBBPF_OPTS(gen_loader_opts, gen);
- struct bpf_object *obj;
- const char *file;
- int err = 0;
- if (!REQ_ARGS(1))
- return -1;
- file = GET_ARG();
- if (verifier_logs)
- /* log_level1 + log_level2 + stats, but not stable UAPI */
- open_opts.kernel_log_level = 1 + 2 + 4;
- obj = bpf_object__open_file(file, &open_opts);
- if (!obj) {
- err = -1;
- p_err("failed to open object file");
- goto err_close_obj;
- }
- if (sign_progs)
- gen.gen_hash = true;
- err = bpf_object__gen_loader(obj, &gen);
- if (err)
- goto err_close_obj;
- err = bpf_object__load(obj);
- if (err) {
- p_err("failed to load object file");
- goto err_close_obj;
- }
- if (verifier_logs) {
- struct dump_data dd = {};
- kernel_syms_load(&dd);
- dump_xlated_plain(&dd, (void *)gen.insns, gen.insns_sz, false, false);
- kernel_syms_destroy(&dd);
- }
- err = try_loader(&gen);
- err_close_obj:
- bpf_object__close(obj);
- return err;
- }
- static int do_load(int argc, char **argv)
- {
- if (use_loader)
- return do_loader(argc, argv);
- return load_with_options(argc, argv, true);
- }
- static int do_loadall(int argc, char **argv)
- {
- return load_with_options(argc, argv, false);
- }
- #ifdef BPFTOOL_WITHOUT_SKELETONS
- static int do_profile(int argc, char **argv)
- {
- p_err("bpftool prog profile command is not supported. Please build bpftool with clang >= 10.0.0");
- return 0;
- }
- #else /* BPFTOOL_WITHOUT_SKELETONS */
- #include "profiler.skel.h"
- struct profile_metric {
- const char *name;
- struct bpf_perf_event_value val;
- struct perf_event_attr attr;
- bool selected;
- /* calculate ratios like instructions per cycle */
- const int ratio_metric; /* 0 for N/A, 1 for index 0 (cycles) */
- const char *ratio_desc;
- const float ratio_mul;
- } metrics[] = {
- {
- .name = "cycles",
- .attr = {
- .type = PERF_TYPE_HARDWARE,
- .config = PERF_COUNT_HW_CPU_CYCLES,
- .exclude_user = 1,
- },
- },
- {
- .name = "instructions",
- .attr = {
- .type = PERF_TYPE_HARDWARE,
- .config = PERF_COUNT_HW_INSTRUCTIONS,
- .exclude_user = 1,
- },
- .ratio_metric = 1,
- .ratio_desc = "insns per cycle",
- .ratio_mul = 1.0,
- },
- {
- .name = "l1d_loads",
- .attr = {
- .type = PERF_TYPE_HW_CACHE,
- .config =
- PERF_COUNT_HW_CACHE_L1D |
- (PERF_COUNT_HW_CACHE_OP_READ << 8) |
- (PERF_COUNT_HW_CACHE_RESULT_ACCESS << 16),
- .exclude_user = 1,
- },
- },
- {
- .name = "llc_misses",
- .attr = {
- .type = PERF_TYPE_HW_CACHE,
- .config =
- PERF_COUNT_HW_CACHE_LL |
- (PERF_COUNT_HW_CACHE_OP_READ << 8) |
- (PERF_COUNT_HW_CACHE_RESULT_MISS << 16),
- .exclude_user = 1
- },
- .ratio_metric = 2,
- .ratio_desc = "LLC misses per million insns",
- .ratio_mul = 1e6,
- },
- {
- .name = "itlb_misses",
- .attr = {
- .type = PERF_TYPE_HW_CACHE,
- .config =
- PERF_COUNT_HW_CACHE_ITLB |
- (PERF_COUNT_HW_CACHE_OP_READ << 8) |
- (PERF_COUNT_HW_CACHE_RESULT_MISS << 16),
- .exclude_user = 1
- },
- .ratio_metric = 2,
- .ratio_desc = "itlb misses per million insns",
- .ratio_mul = 1e6,
- },
- {
- .name = "dtlb_misses",
- .attr = {
- .type = PERF_TYPE_HW_CACHE,
- .config =
- PERF_COUNT_HW_CACHE_DTLB |
- (PERF_COUNT_HW_CACHE_OP_READ << 8) |
- (PERF_COUNT_HW_CACHE_RESULT_MISS << 16),
- .exclude_user = 1
- },
- .ratio_metric = 2,
- .ratio_desc = "dtlb misses per million insns",
- .ratio_mul = 1e6,
- },
- };
- static __u64 profile_total_count;
- #define MAX_NUM_PROFILE_METRICS 4
- static int profile_parse_metrics(int argc, char **argv)
- {
- unsigned int metric_cnt;
- int selected_cnt = 0;
- unsigned int i;
- metric_cnt = ARRAY_SIZE(metrics);
- while (argc > 0) {
- for (i = 0; i < metric_cnt; i++) {
- if (is_prefix(argv[0], metrics[i].name)) {
- if (!metrics[i].selected)
- selected_cnt++;
- metrics[i].selected = true;
- break;
- }
- }
- if (i == metric_cnt) {
- p_err("unknown metric %s", argv[0]);
- return -1;
- }
- NEXT_ARG();
- }
- if (selected_cnt > MAX_NUM_PROFILE_METRICS) {
- p_err("too many (%d) metrics, please specify no more than %d metrics at a time",
- selected_cnt, MAX_NUM_PROFILE_METRICS);
- return -1;
- }
- return selected_cnt;
- }
- static void profile_read_values(struct profiler_bpf *obj)
- {
- __u32 m, cpu, num_cpu = obj->rodata->num_cpu;
- int reading_map_fd, count_map_fd;
- __u64 counts[num_cpu];
- __u32 key = 0;
- int err;
- reading_map_fd = bpf_map__fd(obj->maps.accum_readings);
- count_map_fd = bpf_map__fd(obj->maps.counts);
- if (reading_map_fd < 0 || count_map_fd < 0) {
- p_err("failed to get fd for map");
- return;
- }
- err = bpf_map_lookup_elem(count_map_fd, &key, counts);
- if (err) {
- p_err("failed to read count_map: %s", strerror(errno));
- return;
- }
- profile_total_count = 0;
- for (cpu = 0; cpu < num_cpu; cpu++)
- profile_total_count += counts[cpu];
- for (m = 0; m < ARRAY_SIZE(metrics); m++) {
- struct bpf_perf_event_value values[num_cpu];
- if (!metrics[m].selected)
- continue;
- err = bpf_map_lookup_elem(reading_map_fd, &key, values);
- if (err) {
- p_err("failed to read reading_map: %s",
- strerror(errno));
- return;
- }
- for (cpu = 0; cpu < num_cpu; cpu++) {
- metrics[m].val.counter += values[cpu].counter;
- metrics[m].val.enabled += values[cpu].enabled;
- metrics[m].val.running += values[cpu].running;
- }
- key++;
- }
- }
- static void profile_print_readings_json(void)
- {
- __u32 m;
- jsonw_start_array(json_wtr);
- for (m = 0; m < ARRAY_SIZE(metrics); m++) {
- if (!metrics[m].selected)
- continue;
- jsonw_start_object(json_wtr);
- jsonw_string_field(json_wtr, "metric", metrics[m].name);
- jsonw_lluint_field(json_wtr, "run_cnt", profile_total_count);
- jsonw_lluint_field(json_wtr, "value", metrics[m].val.counter);
- jsonw_lluint_field(json_wtr, "enabled", metrics[m].val.enabled);
- jsonw_lluint_field(json_wtr, "running", metrics[m].val.running);
- jsonw_end_object(json_wtr);
- }
- jsonw_end_array(json_wtr);
- }
- static void profile_print_readings_plain(void)
- {
- __u32 m;
- printf("\n%18llu %-20s\n", profile_total_count, "run_cnt");
- for (m = 0; m < ARRAY_SIZE(metrics); m++) {
- struct bpf_perf_event_value *val = &metrics[m].val;
- int r;
- if (!metrics[m].selected)
- continue;
- printf("%18llu %-20s", val->counter, metrics[m].name);
- r = metrics[m].ratio_metric - 1;
- if (r >= 0 && metrics[r].selected &&
- metrics[r].val.counter > 0) {
- printf("# %8.2f %-30s",
- val->counter * metrics[m].ratio_mul /
- metrics[r].val.counter,
- metrics[m].ratio_desc);
- } else {
- printf("%-41s", "");
- }
- if (val->enabled > val->running)
- printf("(%4.2f%%)",
- val->running * 100.0 / val->enabled);
- printf("\n");
- }
- }
- static void profile_print_readings(void)
- {
- if (json_output)
- profile_print_readings_json();
- else
- profile_print_readings_plain();
- }
- static char *profile_target_name(int tgt_fd)
- {
- struct bpf_func_info func_info = {};
- struct bpf_prog_info info = {};
- __u32 info_len = sizeof(info);
- const struct btf_type *t;
- __u32 func_info_rec_size;
- struct btf *btf = NULL;
- char *name = NULL;
- int err;
- err = bpf_prog_get_info_by_fd(tgt_fd, &info, &info_len);
- if (err) {
- p_err("failed to get info for prog FD %d", tgt_fd);
- goto out;
- }
- if (info.btf_id == 0) {
- p_err("prog FD %d doesn't have valid btf", tgt_fd);
- goto out;
- }
- func_info_rec_size = info.func_info_rec_size;
- if (info.nr_func_info == 0) {
- p_err("found 0 func_info for prog FD %d", tgt_fd);
- goto out;
- }
- memset(&info, 0, sizeof(info));
- info.nr_func_info = 1;
- info.func_info_rec_size = func_info_rec_size;
- info.func_info = ptr_to_u64(&func_info);
- err = bpf_prog_get_info_by_fd(tgt_fd, &info, &info_len);
- if (err) {
- p_err("failed to get func_info for prog FD %d", tgt_fd);
- goto out;
- }
- btf = btf__load_from_kernel_by_id(info.btf_id);
- if (!btf) {
- p_err("failed to load btf for prog FD %d", tgt_fd);
- goto out;
- }
- t = btf__type_by_id(btf, func_info.type_id);
- if (!t) {
- p_err("btf %u doesn't have type %u",
- info.btf_id, func_info.type_id);
- goto out;
- }
- name = strdup(btf__name_by_offset(btf, t->name_off));
- out:
- btf__free(btf);
- return name;
- }
- static struct profiler_bpf *profile_obj;
- static int profile_tgt_fd = -1;
- static char *profile_tgt_name;
- static int *profile_perf_events;
- static int profile_perf_event_cnt;
- static void profile_close_perf_events(struct profiler_bpf *obj)
- {
- int i;
- for (i = profile_perf_event_cnt - 1; i >= 0; i--)
- close(profile_perf_events[i]);
- free(profile_perf_events);
- profile_perf_event_cnt = 0;
- }
- static int profile_open_perf_event(int mid, int cpu, int map_fd)
- {
- int pmu_fd;
- pmu_fd = syscall(__NR_perf_event_open, &metrics[mid].attr,
- -1 /*pid*/, cpu, -1 /*group_fd*/, 0);
- if (pmu_fd < 0) {
- if (errno == ENODEV) {
- p_info("cpu %d may be offline, skip %s profiling.",
- cpu, metrics[mid].name);
- profile_perf_event_cnt++;
- return 0;
- }
- return -1;
- }
- if (bpf_map_update_elem(map_fd,
- &profile_perf_event_cnt,
- &pmu_fd, BPF_ANY) ||
- ioctl(pmu_fd, PERF_EVENT_IOC_ENABLE, 0)) {
- close(pmu_fd);
- return -1;
- }
- profile_perf_events[profile_perf_event_cnt++] = pmu_fd;
- return 0;
- }
- static int profile_open_perf_events(struct profiler_bpf *obj)
- {
- unsigned int cpu, m;
- int map_fd;
- profile_perf_events = calloc(
- obj->rodata->num_cpu * obj->rodata->num_metric, sizeof(int));
- if (!profile_perf_events) {
- p_err("failed to allocate memory for perf_event array: %s",
- strerror(errno));
- return -1;
- }
- map_fd = bpf_map__fd(obj->maps.events);
- if (map_fd < 0) {
- p_err("failed to get fd for events map");
- return -1;
- }
- for (m = 0; m < ARRAY_SIZE(metrics); m++) {
- if (!metrics[m].selected)
- continue;
- for (cpu = 0; cpu < obj->rodata->num_cpu; cpu++) {
- if (profile_open_perf_event(m, cpu, map_fd)) {
- p_err("failed to create event %s on cpu %u",
- metrics[m].name, cpu);
- return -1;
- }
- }
- }
- return 0;
- }
- static void profile_print_and_cleanup(void)
- {
- profile_close_perf_events(profile_obj);
- profile_read_values(profile_obj);
- profile_print_readings();
- profiler_bpf__destroy(profile_obj);
- close(profile_tgt_fd);
- free(profile_tgt_name);
- }
- static void int_exit(int signo)
- {
- profile_print_and_cleanup();
- exit(0);
- }
- static int do_profile(int argc, char **argv)
- {
- int num_metric, num_cpu, err = -1;
- struct bpf_program *prog;
- unsigned long duration;
- char *endptr;
- /* we at least need two args for the prog and one metric */
- if (!REQ_ARGS(3))
- return -EINVAL;
- /* parse target fd */
- profile_tgt_fd = prog_parse_fd(&argc, &argv);
- if (profile_tgt_fd < 0) {
- p_err("failed to parse fd");
- return -1;
- }
- /* parse profiling optional duration */
- if (argc > 2 && is_prefix(argv[0], "duration")) {
- NEXT_ARG();
- duration = strtoul(*argv, &endptr, 0);
- if (*endptr)
- usage();
- NEXT_ARG();
- } else {
- duration = UINT_MAX;
- }
- num_metric = profile_parse_metrics(argc, argv);
- if (num_metric <= 0)
- goto out;
- num_cpu = libbpf_num_possible_cpus();
- if (num_cpu <= 0) {
- p_err("failed to identify number of CPUs");
- goto out;
- }
- profile_obj = profiler_bpf__open();
- if (!profile_obj) {
- p_err("failed to open and/or load BPF object");
- goto out;
- }
- profile_obj->rodata->num_cpu = num_cpu;
- profile_obj->rodata->num_metric = num_metric;
- /* adjust map sizes */
- bpf_map__set_max_entries(profile_obj->maps.events, num_metric * num_cpu);
- bpf_map__set_max_entries(profile_obj->maps.fentry_readings, num_metric);
- bpf_map__set_max_entries(profile_obj->maps.accum_readings, num_metric);
- bpf_map__set_max_entries(profile_obj->maps.counts, 1);
- /* change target name */
- profile_tgt_name = profile_target_name(profile_tgt_fd);
- if (!profile_tgt_name)
- goto out;
- bpf_object__for_each_program(prog, profile_obj->obj) {
- err = bpf_program__set_attach_target(prog, profile_tgt_fd,
- profile_tgt_name);
- if (err) {
- p_err("failed to set attach target\n");
- goto out;
- }
- }
- set_max_rlimit();
- err = profiler_bpf__load(profile_obj);
- if (err) {
- p_err("failed to load profile_obj");
- goto out;
- }
- err = profile_open_perf_events(profile_obj);
- if (err)
- goto out;
- err = profiler_bpf__attach(profile_obj);
- if (err) {
- p_err("failed to attach profile_obj");
- goto out;
- }
- signal(SIGINT, int_exit);
- sleep(duration);
- profile_print_and_cleanup();
- return 0;
- out:
- profile_close_perf_events(profile_obj);
- if (profile_obj)
- profiler_bpf__destroy(profile_obj);
- close(profile_tgt_fd);
- free(profile_tgt_name);
- return err;
- }
- #endif /* BPFTOOL_WITHOUT_SKELETONS */
- static int do_help(int argc, char **argv)
- {
- if (json_output) {
- jsonw_null(json_wtr);
- return 0;
- }
- fprintf(stderr,
- "Usage: %1$s %2$s { show | list } [PROG]\n"
- " %1$s %2$s dump xlated PROG [{ file FILE | [opcodes] [linum] [visual] }]\n"
- " %1$s %2$s dump jited PROG [{ file FILE | [opcodes] [linum] }]\n"
- " %1$s %2$s pin PROG FILE\n"
- " %1$s %2$s { load | loadall } OBJ PATH \\\n"
- " [type TYPE] [{ offload_dev | xdpmeta_dev } NAME] \\\n"
- " [map { idx IDX | name NAME } MAP]\\\n"
- " [pinmaps MAP_DIR]\n"
- " [autoattach]\n"
- " [kernel_btf BTF_FILE]\n"
- " %1$s %2$s attach PROG ATTACH_TYPE [MAP]\n"
- " %1$s %2$s detach PROG ATTACH_TYPE [MAP]\n"
- " %1$s %2$s run PROG \\\n"
- " data_in FILE \\\n"
- " [data_out FILE [data_size_out L]] \\\n"
- " [ctx_in FILE [ctx_out FILE [ctx_size_out M]]] \\\n"
- " [repeat N]\n"
- " %1$s %2$s profile PROG [duration DURATION] METRICs\n"
- " %1$s %2$s tracelog\n"
- " %1$s %2$s tracelog { stdout | stderr } PROG\n"
- " %1$s %2$s help\n"
- "\n"
- " " HELP_SPEC_MAP "\n"
- " " HELP_SPEC_PROGRAM "\n"
- " TYPE := { socket | kprobe | kretprobe | classifier | action |\n"
- " tracepoint | raw_tracepoint | xdp | perf_event | cgroup/skb |\n"
- " cgroup/sock | cgroup/dev | lwt_in | lwt_out | lwt_xmit |\n"
- " lwt_seg6local | sockops | sk_skb | sk_msg | lirc_mode2 |\n"
- " sk_reuseport | flow_dissector | cgroup/sysctl |\n"
- " cgroup/bind4 | cgroup/bind6 | cgroup/post_bind4 |\n"
- " cgroup/post_bind6 | cgroup/connect4 | cgroup/connect6 |\n"
- " cgroup/connect_unix | cgroup/getpeername4 | cgroup/getpeername6 |\n"
- " cgroup/getpeername_unix | cgroup/getsockname4 | cgroup/getsockname6 |\n"
- " cgroup/getsockname_unix | cgroup/sendmsg4 | cgroup/sendmsg6 |\n"
- " cgroup/sendmsg_unix | cgroup/recvmsg4 | cgroup/recvmsg6 | cgroup/recvmsg_unix |\n"
- " cgroup/getsockopt | cgroup/setsockopt | cgroup/sock_release |\n"
- " struct_ops | fentry | fexit | freplace | sk_lookup }\n"
- " ATTACH_TYPE := { sk_msg_verdict | sk_skb_verdict | sk_skb_stream_verdict |\n"
- " sk_skb_stream_parser | flow_dissector }\n"
- " METRIC := { cycles | instructions | l1d_loads | llc_misses | itlb_misses | dtlb_misses }\n"
- " " HELP_SPEC_OPTIONS " |\n"
- " {-f|--bpffs} | {-m|--mapcompat} | {-n|--nomount} |\n"
- " {-L|--use-loader} | [ {-S|--sign } {-k} <private_key.pem> {-i} <certificate.x509> ] \n"
- "",
- bin_name, argv[-2]);
- return 0;
- }
- static const struct cmd cmds[] = {
- { "show", do_show },
- { "list", do_show },
- { "help", do_help },
- { "dump", do_dump },
- { "pin", do_pin },
- { "load", do_load },
- { "loadall", do_loadall },
- { "attach", do_attach },
- { "detach", do_detach },
- { "tracelog", do_tracelog_any },
- { "run", do_run },
- { "profile", do_profile },
- { 0 }
- };
- int do_prog(int argc, char **argv)
- {
- return cmd_select(cmds, argc, argv, do_help);
- }
|