prog.c 60 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618
  1. // SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause)
  2. /* Copyright (C) 2017-2018 Netronome Systems, Inc. */
  3. #ifndef _GNU_SOURCE
  4. #define _GNU_SOURCE
  5. #endif
  6. #include <errno.h>
  7. #include <fcntl.h>
  8. #include <signal.h>
  9. #include <stdarg.h>
  10. #include <stdio.h>
  11. #include <stdlib.h>
  12. #include <string.h>
  13. #include <time.h>
  14. #include <unistd.h>
  15. #include <net/if.h>
  16. #include <sys/ioctl.h>
  17. #include <sys/types.h>
  18. #include <sys/stat.h>
  19. #include <sys/syscall.h>
  20. #include <dirent.h>
  21. #include <linux/err.h>
  22. #include <linux/perf_event.h>
  23. #include <linux/sizes.h>
  24. #include <linux/keyctl.h>
  25. #include <bpf/bpf.h>
  26. #include <bpf/btf.h>
  27. #include <bpf/hashmap.h>
  28. #include <bpf/libbpf.h>
  29. #include <bpf/libbpf_internal.h>
  30. #include <bpf/skel_internal.h>
  31. #include "cfg.h"
  32. #include "main.h"
  33. #include "xlated_dumper.h"
  34. #define BPF_METADATA_PREFIX "bpf_metadata_"
  35. #define BPF_METADATA_PREFIX_LEN (sizeof(BPF_METADATA_PREFIX) - 1)
  36. enum dump_mode {
  37. DUMP_JITED,
  38. DUMP_XLATED,
  39. };
  40. static const bool attach_types[] = {
  41. [BPF_SK_SKB_STREAM_PARSER] = true,
  42. [BPF_SK_SKB_STREAM_VERDICT] = true,
  43. [BPF_SK_SKB_VERDICT] = true,
  44. [BPF_SK_MSG_VERDICT] = true,
  45. [BPF_FLOW_DISSECTOR] = true,
  46. [__MAX_BPF_ATTACH_TYPE] = false,
  47. };
  48. /* Textual representations traditionally used by the program and kept around
  49. * for the sake of backwards compatibility.
  50. */
  51. static const char * const attach_type_strings[] = {
  52. [BPF_SK_SKB_STREAM_PARSER] = "stream_parser",
  53. [BPF_SK_SKB_STREAM_VERDICT] = "stream_verdict",
  54. [BPF_SK_SKB_VERDICT] = "skb_verdict",
  55. [BPF_SK_MSG_VERDICT] = "msg_verdict",
  56. [__MAX_BPF_ATTACH_TYPE] = NULL,
  57. };
  58. static struct hashmap *prog_table;
  59. static enum bpf_attach_type parse_attach_type(const char *str)
  60. {
  61. enum bpf_attach_type type;
  62. for (type = 0; type < __MAX_BPF_ATTACH_TYPE; type++) {
  63. if (attach_types[type]) {
  64. const char *attach_type_str;
  65. attach_type_str = libbpf_bpf_attach_type_str(type);
  66. if (!strcmp(str, attach_type_str))
  67. return type;
  68. }
  69. if (attach_type_strings[type] &&
  70. is_prefix(str, attach_type_strings[type]))
  71. return type;
  72. }
  73. return __MAX_BPF_ATTACH_TYPE;
  74. }
  75. static int prep_prog_info(struct bpf_prog_info *const info, enum dump_mode mode,
  76. void **info_data, size_t *const info_data_sz)
  77. {
  78. struct bpf_prog_info holder = {};
  79. size_t needed = 0;
  80. void *ptr;
  81. if (mode == DUMP_JITED) {
  82. holder.jited_prog_len = info->jited_prog_len;
  83. needed += info->jited_prog_len;
  84. } else {
  85. holder.xlated_prog_len = info->xlated_prog_len;
  86. needed += info->xlated_prog_len;
  87. }
  88. holder.nr_jited_ksyms = info->nr_jited_ksyms;
  89. needed += info->nr_jited_ksyms * sizeof(__u64);
  90. holder.nr_jited_func_lens = info->nr_jited_func_lens;
  91. needed += info->nr_jited_func_lens * sizeof(__u32);
  92. holder.nr_func_info = info->nr_func_info;
  93. holder.func_info_rec_size = info->func_info_rec_size;
  94. needed += info->nr_func_info * info->func_info_rec_size;
  95. holder.nr_line_info = info->nr_line_info;
  96. holder.line_info_rec_size = info->line_info_rec_size;
  97. needed += info->nr_line_info * info->line_info_rec_size;
  98. holder.nr_jited_line_info = info->nr_jited_line_info;
  99. holder.jited_line_info_rec_size = info->jited_line_info_rec_size;
  100. needed += info->nr_jited_line_info * info->jited_line_info_rec_size;
  101. if (needed > *info_data_sz) {
  102. ptr = realloc(*info_data, needed);
  103. if (!ptr)
  104. return -1;
  105. *info_data = ptr;
  106. *info_data_sz = needed;
  107. }
  108. ptr = *info_data;
  109. if (mode == DUMP_JITED) {
  110. holder.jited_prog_insns = ptr_to_u64(ptr);
  111. ptr += holder.jited_prog_len;
  112. } else {
  113. holder.xlated_prog_insns = ptr_to_u64(ptr);
  114. ptr += holder.xlated_prog_len;
  115. }
  116. holder.jited_ksyms = ptr_to_u64(ptr);
  117. ptr += holder.nr_jited_ksyms * sizeof(__u64);
  118. holder.jited_func_lens = ptr_to_u64(ptr);
  119. ptr += holder.nr_jited_func_lens * sizeof(__u32);
  120. holder.func_info = ptr_to_u64(ptr);
  121. ptr += holder.nr_func_info * holder.func_info_rec_size;
  122. holder.line_info = ptr_to_u64(ptr);
  123. ptr += holder.nr_line_info * holder.line_info_rec_size;
  124. holder.jited_line_info = ptr_to_u64(ptr);
  125. ptr += holder.nr_jited_line_info * holder.jited_line_info_rec_size;
  126. *info = holder;
  127. return 0;
  128. }
  129. static void print_boot_time(__u64 nsecs, char *buf, unsigned int size)
  130. {
  131. struct timespec real_time_ts, boot_time_ts;
  132. time_t wallclock_secs;
  133. struct tm load_tm;
  134. buf[--size] = '\0';
  135. if (clock_gettime(CLOCK_REALTIME, &real_time_ts) ||
  136. clock_gettime(CLOCK_BOOTTIME, &boot_time_ts)) {
  137. perror("Can't read clocks");
  138. snprintf(buf, size, "%llu", nsecs / 1000000000);
  139. return;
  140. }
  141. wallclock_secs = (real_time_ts.tv_sec - boot_time_ts.tv_sec) +
  142. (real_time_ts.tv_nsec - boot_time_ts.tv_nsec + nsecs) /
  143. 1000000000;
  144. if (!localtime_r(&wallclock_secs, &load_tm)) {
  145. snprintf(buf, size, "%llu", nsecs / 1000000000);
  146. return;
  147. }
  148. if (json_output)
  149. strftime(buf, size, "%s", &load_tm);
  150. else
  151. strftime(buf, size, "%FT%T%z", &load_tm);
  152. }
  153. static void show_prog_maps(int fd, __u32 num_maps)
  154. {
  155. struct bpf_prog_info info = {};
  156. __u32 len = sizeof(info);
  157. __u32 map_ids[num_maps];
  158. unsigned int i;
  159. int err;
  160. info.nr_map_ids = num_maps;
  161. info.map_ids = ptr_to_u64(map_ids);
  162. err = bpf_prog_get_info_by_fd(fd, &info, &len);
  163. if (err || !info.nr_map_ids)
  164. return;
  165. if (json_output) {
  166. jsonw_name(json_wtr, "map_ids");
  167. jsonw_start_array(json_wtr);
  168. for (i = 0; i < info.nr_map_ids; i++)
  169. jsonw_uint(json_wtr, map_ids[i]);
  170. jsonw_end_array(json_wtr);
  171. } else {
  172. printf(" map_ids ");
  173. for (i = 0; i < info.nr_map_ids; i++)
  174. printf("%u%s", map_ids[i],
  175. i == info.nr_map_ids - 1 ? "" : ",");
  176. }
  177. }
  178. static void *find_metadata(int prog_fd, struct bpf_map_info *map_info)
  179. {
  180. struct bpf_prog_info prog_info;
  181. __u32 prog_info_len;
  182. __u32 map_info_len;
  183. void *value = NULL;
  184. __u32 *map_ids;
  185. int nr_maps;
  186. int key = 0;
  187. int map_fd;
  188. int ret;
  189. __u32 i;
  190. memset(&prog_info, 0, sizeof(prog_info));
  191. prog_info_len = sizeof(prog_info);
  192. ret = bpf_prog_get_info_by_fd(prog_fd, &prog_info, &prog_info_len);
  193. if (ret)
  194. return NULL;
  195. if (!prog_info.nr_map_ids)
  196. return NULL;
  197. map_ids = calloc(prog_info.nr_map_ids, sizeof(__u32));
  198. if (!map_ids)
  199. return NULL;
  200. nr_maps = prog_info.nr_map_ids;
  201. memset(&prog_info, 0, sizeof(prog_info));
  202. prog_info.nr_map_ids = nr_maps;
  203. prog_info.map_ids = ptr_to_u64(map_ids);
  204. prog_info_len = sizeof(prog_info);
  205. ret = bpf_prog_get_info_by_fd(prog_fd, &prog_info, &prog_info_len);
  206. if (ret)
  207. goto free_map_ids;
  208. for (i = 0; i < prog_info.nr_map_ids; i++) {
  209. map_fd = bpf_map_get_fd_by_id(map_ids[i]);
  210. if (map_fd < 0)
  211. goto free_map_ids;
  212. memset(map_info, 0, sizeof(*map_info));
  213. map_info_len = sizeof(*map_info);
  214. ret = bpf_map_get_info_by_fd(map_fd, map_info, &map_info_len);
  215. if (ret < 0) {
  216. close(map_fd);
  217. goto free_map_ids;
  218. }
  219. if (map_info->type != BPF_MAP_TYPE_ARRAY ||
  220. map_info->key_size != sizeof(int) ||
  221. map_info->max_entries != 1 ||
  222. !map_info->btf_value_type_id ||
  223. !strstr(map_info->name, ".rodata")) {
  224. close(map_fd);
  225. continue;
  226. }
  227. value = malloc(map_info->value_size);
  228. if (!value) {
  229. close(map_fd);
  230. goto free_map_ids;
  231. }
  232. if (bpf_map_lookup_elem(map_fd, &key, value)) {
  233. close(map_fd);
  234. free(value);
  235. value = NULL;
  236. goto free_map_ids;
  237. }
  238. close(map_fd);
  239. break;
  240. }
  241. free_map_ids:
  242. free(map_ids);
  243. return value;
  244. }
  245. static bool has_metadata_prefix(const char *s)
  246. {
  247. return strncmp(s, BPF_METADATA_PREFIX, BPF_METADATA_PREFIX_LEN) == 0;
  248. }
  249. static void show_prog_metadata(int fd, __u32 num_maps)
  250. {
  251. const struct btf_type *t_datasec, *t_var;
  252. struct bpf_map_info map_info;
  253. struct btf_var_secinfo *vsi;
  254. bool printed_header = false;
  255. unsigned int i, vlen;
  256. void *value = NULL;
  257. const char *name;
  258. struct btf *btf;
  259. int err;
  260. if (!num_maps)
  261. return;
  262. memset(&map_info, 0, sizeof(map_info));
  263. value = find_metadata(fd, &map_info);
  264. if (!value)
  265. return;
  266. btf = btf__load_from_kernel_by_id(map_info.btf_id);
  267. if (!btf)
  268. goto out_free;
  269. t_datasec = btf__type_by_id(btf, map_info.btf_value_type_id);
  270. if (!btf_is_datasec(t_datasec))
  271. goto out_free;
  272. vlen = btf_vlen(t_datasec);
  273. vsi = btf_var_secinfos(t_datasec);
  274. /* We don't proceed to check the kinds of the elements of the DATASEC.
  275. * The verifier enforces them to be BTF_KIND_VAR.
  276. */
  277. if (json_output) {
  278. struct btf_dumper d = {
  279. .btf = btf,
  280. .jw = json_wtr,
  281. .is_plain_text = false,
  282. };
  283. for (i = 0; i < vlen; i++, vsi++) {
  284. t_var = btf__type_by_id(btf, vsi->type);
  285. name = btf__name_by_offset(btf, t_var->name_off);
  286. if (!has_metadata_prefix(name))
  287. continue;
  288. if (!printed_header) {
  289. jsonw_name(json_wtr, "metadata");
  290. jsonw_start_object(json_wtr);
  291. printed_header = true;
  292. }
  293. jsonw_name(json_wtr, name + BPF_METADATA_PREFIX_LEN);
  294. err = btf_dumper_type(&d, t_var->type, value + vsi->offset);
  295. if (err) {
  296. p_err("btf dump failed: %d", err);
  297. break;
  298. }
  299. }
  300. if (printed_header)
  301. jsonw_end_object(json_wtr);
  302. } else {
  303. json_writer_t *btf_wtr;
  304. struct btf_dumper d = {
  305. .btf = btf,
  306. .is_plain_text = true,
  307. };
  308. for (i = 0; i < vlen; i++, vsi++) {
  309. t_var = btf__type_by_id(btf, vsi->type);
  310. name = btf__name_by_offset(btf, t_var->name_off);
  311. if (!has_metadata_prefix(name))
  312. continue;
  313. if (!printed_header) {
  314. printf("\tmetadata:");
  315. btf_wtr = jsonw_new(stdout);
  316. if (!btf_wtr) {
  317. p_err("jsonw alloc failed");
  318. goto out_free;
  319. }
  320. d.jw = btf_wtr,
  321. printed_header = true;
  322. }
  323. printf("\n\t\t%s = ", name + BPF_METADATA_PREFIX_LEN);
  324. jsonw_reset(btf_wtr);
  325. err = btf_dumper_type(&d, t_var->type, value + vsi->offset);
  326. if (err) {
  327. p_err("btf dump failed: %d", err);
  328. break;
  329. }
  330. }
  331. if (printed_header)
  332. jsonw_destroy(&btf_wtr);
  333. }
  334. out_free:
  335. btf__free(btf);
  336. free(value);
  337. }
  338. static void print_prog_header_json(struct bpf_prog_info *info, int fd)
  339. {
  340. const char *prog_type_str;
  341. char prog_name[MAX_PROG_FULL_NAME];
  342. jsonw_uint_field(json_wtr, "id", info->id);
  343. prog_type_str = libbpf_bpf_prog_type_str(info->type);
  344. if (prog_type_str)
  345. jsonw_string_field(json_wtr, "type", prog_type_str);
  346. else
  347. jsonw_uint_field(json_wtr, "type", info->type);
  348. if (*info->name) {
  349. get_prog_full_name(info, fd, prog_name, sizeof(prog_name));
  350. jsonw_string_field(json_wtr, "name", prog_name);
  351. }
  352. jsonw_name(json_wtr, "tag");
  353. jsonw_printf(json_wtr, "\"" BPF_TAG_FMT "\"",
  354. info->tag[0], info->tag[1], info->tag[2], info->tag[3],
  355. info->tag[4], info->tag[5], info->tag[6], info->tag[7]);
  356. jsonw_bool_field(json_wtr, "gpl_compatible", info->gpl_compatible);
  357. if (info->run_time_ns) {
  358. jsonw_uint_field(json_wtr, "run_time_ns", info->run_time_ns);
  359. jsonw_uint_field(json_wtr, "run_cnt", info->run_cnt);
  360. }
  361. if (info->recursion_misses)
  362. jsonw_uint_field(json_wtr, "recursion_misses", info->recursion_misses);
  363. }
  364. static void print_prog_json(struct bpf_prog_info *info, int fd, bool orphaned)
  365. {
  366. char *memlock;
  367. jsonw_start_object(json_wtr);
  368. print_prog_header_json(info, fd);
  369. print_dev_json(info->ifindex, info->netns_dev, info->netns_ino);
  370. if (info->load_time) {
  371. char buf[32];
  372. print_boot_time(info->load_time, buf, sizeof(buf));
  373. /* Piggy back on load_time, since 0 uid is a valid one */
  374. jsonw_name(json_wtr, "loaded_at");
  375. jsonw_printf(json_wtr, "%s", buf);
  376. jsonw_uint_field(json_wtr, "uid", info->created_by_uid);
  377. }
  378. jsonw_bool_field(json_wtr, "orphaned", orphaned);
  379. jsonw_uint_field(json_wtr, "bytes_xlated", info->xlated_prog_len);
  380. if (info->jited_prog_len) {
  381. jsonw_bool_field(json_wtr, "jited", true);
  382. jsonw_uint_field(json_wtr, "bytes_jited", info->jited_prog_len);
  383. } else {
  384. jsonw_bool_field(json_wtr, "jited", false);
  385. }
  386. memlock = get_fdinfo(fd, "memlock");
  387. if (memlock)
  388. jsonw_int_field(json_wtr, "bytes_memlock", atoll(memlock));
  389. free(memlock);
  390. if (info->nr_map_ids)
  391. show_prog_maps(fd, info->nr_map_ids);
  392. if (info->btf_id)
  393. jsonw_int_field(json_wtr, "btf_id", info->btf_id);
  394. if (!hashmap__empty(prog_table)) {
  395. struct hashmap_entry *entry;
  396. jsonw_name(json_wtr, "pinned");
  397. jsonw_start_array(json_wtr);
  398. hashmap__for_each_key_entry(prog_table, entry, info->id)
  399. jsonw_string(json_wtr, entry->pvalue);
  400. jsonw_end_array(json_wtr);
  401. }
  402. emit_obj_refs_json(refs_table, info->id, json_wtr);
  403. show_prog_metadata(fd, info->nr_map_ids);
  404. jsonw_end_object(json_wtr);
  405. }
  406. static void print_prog_header_plain(struct bpf_prog_info *info, int fd)
  407. {
  408. const char *prog_type_str;
  409. char prog_name[MAX_PROG_FULL_NAME];
  410. printf("%u: ", info->id);
  411. prog_type_str = libbpf_bpf_prog_type_str(info->type);
  412. if (prog_type_str)
  413. printf("%s ", prog_type_str);
  414. else
  415. printf("type %u ", info->type);
  416. if (*info->name) {
  417. get_prog_full_name(info, fd, prog_name, sizeof(prog_name));
  418. printf("name %s ", prog_name);
  419. }
  420. printf("tag ");
  421. fprint_hex(stdout, info->tag, BPF_TAG_SIZE, "");
  422. print_dev_plain(info->ifindex, info->netns_dev, info->netns_ino);
  423. printf("%s", info->gpl_compatible ? " gpl" : "");
  424. if (info->run_time_ns)
  425. printf(" run_time_ns %llu run_cnt %llu",
  426. info->run_time_ns, info->run_cnt);
  427. if (info->recursion_misses)
  428. printf(" recursion_misses %llu", info->recursion_misses);
  429. printf("\n");
  430. }
  431. static void print_prog_plain(struct bpf_prog_info *info, int fd, bool orphaned)
  432. {
  433. char *memlock;
  434. print_prog_header_plain(info, fd);
  435. if (info->load_time) {
  436. char buf[32];
  437. print_boot_time(info->load_time, buf, sizeof(buf));
  438. /* Piggy back on load_time, since 0 uid is a valid one */
  439. printf("\tloaded_at %s uid %u\n", buf, info->created_by_uid);
  440. }
  441. printf("\txlated %uB", info->xlated_prog_len);
  442. if (info->jited_prog_len)
  443. printf(" jited %uB", info->jited_prog_len);
  444. else
  445. printf(" not jited");
  446. memlock = get_fdinfo(fd, "memlock");
  447. if (memlock)
  448. printf(" memlock %sB", memlock);
  449. free(memlock);
  450. if (orphaned)
  451. printf(" orphaned");
  452. if (info->nr_map_ids)
  453. show_prog_maps(fd, info->nr_map_ids);
  454. if (!hashmap__empty(prog_table)) {
  455. struct hashmap_entry *entry;
  456. hashmap__for_each_key_entry(prog_table, entry, info->id)
  457. printf("\n\tpinned %s", (char *)entry->pvalue);
  458. }
  459. if (info->btf_id)
  460. printf("\n\tbtf_id %u", info->btf_id);
  461. emit_obj_refs_plain(refs_table, info->id, "\n\tpids ");
  462. printf("\n");
  463. show_prog_metadata(fd, info->nr_map_ids);
  464. }
  465. static int show_prog(int fd)
  466. {
  467. struct bpf_prog_info info = {};
  468. __u32 len = sizeof(info);
  469. int err;
  470. err = bpf_prog_get_info_by_fd(fd, &info, &len);
  471. if (err && err != -ENODEV) {
  472. p_err("can't get prog info: %s", strerror(errno));
  473. return -1;
  474. }
  475. if (json_output)
  476. print_prog_json(&info, fd, err == -ENODEV);
  477. else
  478. print_prog_plain(&info, fd, err == -ENODEV);
  479. return 0;
  480. }
  481. static int do_show_subset(int argc, char **argv)
  482. {
  483. int *fds = NULL;
  484. int nb_fds, i;
  485. int err = -1;
  486. fds = malloc(sizeof(int));
  487. if (!fds) {
  488. p_err("mem alloc failed");
  489. return -1;
  490. }
  491. nb_fds = prog_parse_fds(&argc, &argv, &fds);
  492. if (nb_fds < 1)
  493. goto exit_free;
  494. if (json_output && nb_fds > 1)
  495. jsonw_start_array(json_wtr); /* root array */
  496. for (i = 0; i < nb_fds; i++) {
  497. err = show_prog(fds[i]);
  498. if (err) {
  499. for (; i < nb_fds; i++)
  500. close(fds[i]);
  501. break;
  502. }
  503. close(fds[i]);
  504. }
  505. if (json_output && nb_fds > 1)
  506. jsonw_end_array(json_wtr); /* root array */
  507. exit_free:
  508. free(fds);
  509. return err;
  510. }
  511. static int do_show(int argc, char **argv)
  512. {
  513. __u32 id = 0;
  514. int err;
  515. int fd;
  516. if (show_pinned) {
  517. prog_table = hashmap__new(hash_fn_for_key_as_id,
  518. equal_fn_for_key_as_id, NULL);
  519. if (IS_ERR(prog_table)) {
  520. p_err("failed to create hashmap for pinned paths");
  521. return -1;
  522. }
  523. build_pinned_obj_table(prog_table, BPF_OBJ_PROG);
  524. }
  525. build_obj_refs_table(&refs_table, BPF_OBJ_PROG);
  526. if (argc == 2)
  527. return do_show_subset(argc, argv);
  528. if (argc)
  529. return BAD_ARG();
  530. if (json_output)
  531. jsonw_start_array(json_wtr);
  532. while (true) {
  533. err = bpf_prog_get_next_id(id, &id);
  534. if (err) {
  535. if (errno == ENOENT) {
  536. err = 0;
  537. break;
  538. }
  539. p_err("can't get next program: %s%s", strerror(errno),
  540. errno == EINVAL ? " -- kernel too old?" : "");
  541. err = -1;
  542. break;
  543. }
  544. fd = bpf_prog_get_fd_by_id(id);
  545. if (fd < 0) {
  546. if (errno == ENOENT)
  547. continue;
  548. p_err("can't get prog by id (%u): %s",
  549. id, strerror(errno));
  550. err = -1;
  551. break;
  552. }
  553. err = show_prog(fd);
  554. close(fd);
  555. if (err)
  556. break;
  557. }
  558. if (json_output)
  559. jsonw_end_array(json_wtr);
  560. delete_obj_refs_table(refs_table);
  561. if (show_pinned)
  562. delete_pinned_obj_table(prog_table);
  563. return err;
  564. }
  565. static int
  566. prog_dump(struct bpf_prog_info *info, enum dump_mode mode,
  567. char *filepath, bool opcodes, bool visual, bool linum)
  568. {
  569. struct bpf_prog_linfo *prog_linfo = NULL;
  570. const char *disasm_opt = NULL;
  571. struct dump_data dd = {};
  572. void *func_info = NULL;
  573. struct btf *btf = NULL;
  574. char func_sig[1024];
  575. unsigned char *buf;
  576. __u32 member_len;
  577. int fd, err = -1;
  578. ssize_t n;
  579. if (mode == DUMP_JITED) {
  580. if (info->jited_prog_len == 0 || !info->jited_prog_insns) {
  581. p_err("error retrieving jit dump: no instructions returned or kernel.kptr_restrict set?");
  582. return -1;
  583. }
  584. buf = u64_to_ptr(info->jited_prog_insns);
  585. member_len = info->jited_prog_len;
  586. } else { /* DUMP_XLATED */
  587. if (info->xlated_prog_len == 0 || !info->xlated_prog_insns) {
  588. p_err("error retrieving insn dump: kernel.kptr_restrict set?");
  589. return -1;
  590. }
  591. buf = u64_to_ptr(info->xlated_prog_insns);
  592. member_len = info->xlated_prog_len;
  593. }
  594. if (info->btf_id) {
  595. btf = btf__load_from_kernel_by_id(info->btf_id);
  596. if (!btf) {
  597. p_err("failed to get btf");
  598. return -1;
  599. }
  600. }
  601. func_info = u64_to_ptr(info->func_info);
  602. if (info->nr_line_info) {
  603. prog_linfo = bpf_prog_linfo__new(info);
  604. if (!prog_linfo)
  605. p_info("error in processing bpf_line_info. continue without it.");
  606. }
  607. if (filepath) {
  608. fd = open(filepath, O_WRONLY | O_CREAT | O_TRUNC, 0600);
  609. if (fd < 0) {
  610. p_err("can't open file %s: %s", filepath,
  611. strerror(errno));
  612. goto exit_free;
  613. }
  614. n = write(fd, buf, member_len);
  615. close(fd);
  616. if (n != (ssize_t)member_len) {
  617. p_err("error writing output file: %s",
  618. n < 0 ? strerror(errno) : "short write");
  619. goto exit_free;
  620. }
  621. if (json_output)
  622. jsonw_null(json_wtr);
  623. } else if (mode == DUMP_JITED) {
  624. const char *name = NULL;
  625. if (info->ifindex) {
  626. name = ifindex_to_arch(info->ifindex, info->netns_dev,
  627. info->netns_ino, &disasm_opt);
  628. if (!name)
  629. goto exit_free;
  630. }
  631. if (info->nr_jited_func_lens && info->jited_func_lens) {
  632. struct kernel_sym *sym = NULL;
  633. struct bpf_func_info *record;
  634. char sym_name[SYM_MAX_NAME];
  635. unsigned char *img = buf;
  636. __u64 *ksyms = NULL;
  637. __u32 *lens;
  638. __u32 i;
  639. if (info->nr_jited_ksyms) {
  640. kernel_syms_load(&dd);
  641. ksyms = u64_to_ptr(info->jited_ksyms);
  642. }
  643. if (json_output)
  644. jsonw_start_array(json_wtr);
  645. lens = u64_to_ptr(info->jited_func_lens);
  646. for (i = 0; i < info->nr_jited_func_lens; i++) {
  647. if (ksyms) {
  648. sym = kernel_syms_search(&dd, ksyms[i]);
  649. if (sym)
  650. sprintf(sym_name, "%s", sym->name);
  651. else
  652. sprintf(sym_name, "0x%016llx", ksyms[i]);
  653. } else {
  654. strcpy(sym_name, "unknown");
  655. }
  656. if (func_info) {
  657. record = func_info + i * info->func_info_rec_size;
  658. btf_dumper_type_only(btf, record->type_id,
  659. func_sig,
  660. sizeof(func_sig));
  661. }
  662. if (json_output) {
  663. jsonw_start_object(json_wtr);
  664. if (func_info && func_sig[0] != '\0') {
  665. jsonw_name(json_wtr, "proto");
  666. jsonw_string(json_wtr, func_sig);
  667. }
  668. jsonw_name(json_wtr, "name");
  669. jsonw_string(json_wtr, sym_name);
  670. jsonw_name(json_wtr, "insns");
  671. } else {
  672. if (func_info && func_sig[0] != '\0')
  673. printf("%s:\n", func_sig);
  674. printf("%s:\n", sym_name);
  675. }
  676. if (ksyms) {
  677. if (disasm_print_insn(img, lens[i], opcodes,
  678. name, disasm_opt, btf,
  679. prog_linfo, ksyms[i], i,
  680. linum))
  681. goto exit_free;
  682. } else {
  683. if (disasm_print_insn(img, lens[i], opcodes,
  684. name, disasm_opt, btf,
  685. NULL, 0, 0, false))
  686. goto exit_free;
  687. }
  688. img += lens[i];
  689. if (json_output)
  690. jsonw_end_object(json_wtr);
  691. else
  692. printf("\n");
  693. }
  694. if (json_output)
  695. jsonw_end_array(json_wtr);
  696. } else {
  697. if (disasm_print_insn(buf, member_len, opcodes, name,
  698. disasm_opt, btf, NULL, 0, 0,
  699. false))
  700. goto exit_free;
  701. }
  702. } else {
  703. kernel_syms_load(&dd);
  704. dd.nr_jited_ksyms = info->nr_jited_ksyms;
  705. dd.jited_ksyms = u64_to_ptr(info->jited_ksyms);
  706. dd.btf = btf;
  707. dd.func_info = func_info;
  708. dd.finfo_rec_size = info->func_info_rec_size;
  709. dd.prog_linfo = prog_linfo;
  710. if (json_output)
  711. dump_xlated_json(&dd, buf, member_len, opcodes, linum);
  712. else if (visual)
  713. dump_xlated_cfg(&dd, buf, member_len, opcodes, linum);
  714. else
  715. dump_xlated_plain(&dd, buf, member_len, opcodes, linum);
  716. kernel_syms_destroy(&dd);
  717. }
  718. err = 0;
  719. exit_free:
  720. btf__free(btf);
  721. bpf_prog_linfo__free(prog_linfo);
  722. return err;
  723. }
  724. static int do_dump(int argc, char **argv)
  725. {
  726. struct bpf_prog_info info;
  727. __u32 info_len = sizeof(info);
  728. size_t info_data_sz = 0;
  729. void *info_data = NULL;
  730. char *filepath = NULL;
  731. bool opcodes = false;
  732. bool visual = false;
  733. enum dump_mode mode;
  734. bool linum = false;
  735. int nb_fds, i = 0;
  736. int *fds = NULL;
  737. int err = -1;
  738. if (is_prefix(*argv, "jited")) {
  739. if (disasm_init())
  740. return -1;
  741. mode = DUMP_JITED;
  742. } else if (is_prefix(*argv, "xlated")) {
  743. mode = DUMP_XLATED;
  744. } else {
  745. p_err("expected 'xlated' or 'jited', got: %s", *argv);
  746. return -1;
  747. }
  748. NEXT_ARG();
  749. if (argc < 2)
  750. usage();
  751. fds = malloc(sizeof(int));
  752. if (!fds) {
  753. p_err("mem alloc failed");
  754. return -1;
  755. }
  756. nb_fds = prog_parse_fds(&argc, &argv, &fds);
  757. if (nb_fds < 1)
  758. goto exit_free;
  759. while (argc) {
  760. if (is_prefix(*argv, "file")) {
  761. NEXT_ARG();
  762. if (!argc) {
  763. p_err("expected file path");
  764. goto exit_close;
  765. }
  766. if (nb_fds > 1) {
  767. p_err("several programs matched");
  768. goto exit_close;
  769. }
  770. filepath = *argv;
  771. NEXT_ARG();
  772. } else if (is_prefix(*argv, "opcodes")) {
  773. opcodes = true;
  774. NEXT_ARG();
  775. } else if (is_prefix(*argv, "visual")) {
  776. if (nb_fds > 1) {
  777. p_err("several programs matched");
  778. goto exit_close;
  779. }
  780. visual = true;
  781. NEXT_ARG();
  782. } else if (is_prefix(*argv, "linum")) {
  783. linum = true;
  784. NEXT_ARG();
  785. } else {
  786. usage();
  787. goto exit_close;
  788. }
  789. }
  790. if (filepath && (opcodes || visual || linum)) {
  791. p_err("'file' is not compatible with 'opcodes', 'visual', or 'linum'");
  792. goto exit_close;
  793. }
  794. if (json_output && visual) {
  795. p_err("'visual' is not compatible with JSON output");
  796. goto exit_close;
  797. }
  798. if (json_output && nb_fds > 1)
  799. jsonw_start_array(json_wtr); /* root array */
  800. for (i = 0; i < nb_fds; i++) {
  801. memset(&info, 0, sizeof(info));
  802. err = bpf_prog_get_info_by_fd(fds[i], &info, &info_len);
  803. if (err) {
  804. p_err("can't get prog info: %s", strerror(errno));
  805. break;
  806. }
  807. err = prep_prog_info(&info, mode, &info_data, &info_data_sz);
  808. if (err) {
  809. p_err("can't grow prog info_data");
  810. break;
  811. }
  812. err = bpf_prog_get_info_by_fd(fds[i], &info, &info_len);
  813. if (err) {
  814. p_err("can't get prog info: %s", strerror(errno));
  815. break;
  816. }
  817. if (json_output && nb_fds > 1) {
  818. jsonw_start_object(json_wtr); /* prog object */
  819. print_prog_header_json(&info, fds[i]);
  820. jsonw_name(json_wtr, "insns");
  821. } else if (nb_fds > 1) {
  822. print_prog_header_plain(&info, fds[i]);
  823. }
  824. err = prog_dump(&info, mode, filepath, opcodes, visual, linum);
  825. if (json_output && nb_fds > 1)
  826. jsonw_end_object(json_wtr); /* prog object */
  827. else if (i != nb_fds - 1 && nb_fds > 1)
  828. printf("\n");
  829. if (err)
  830. break;
  831. close(fds[i]);
  832. }
  833. if (json_output && nb_fds > 1)
  834. jsonw_end_array(json_wtr); /* root array */
  835. exit_close:
  836. for (; i < nb_fds; i++)
  837. close(fds[i]);
  838. exit_free:
  839. free(info_data);
  840. free(fds);
  841. return err;
  842. }
  843. static int do_pin(int argc, char **argv)
  844. {
  845. int err;
  846. err = do_pin_any(argc, argv, prog_parse_fd);
  847. if (!err && json_output)
  848. jsonw_null(json_wtr);
  849. return err;
  850. }
  851. struct map_replace {
  852. int idx;
  853. int fd;
  854. char *name;
  855. };
  856. static int map_replace_compar(const void *p1, const void *p2)
  857. {
  858. const struct map_replace *a = p1, *b = p2;
  859. return a->idx - b->idx;
  860. }
  861. static int parse_attach_detach_args(int argc, char **argv, int *progfd,
  862. enum bpf_attach_type *attach_type,
  863. int *mapfd)
  864. {
  865. if (!REQ_ARGS(3))
  866. return -EINVAL;
  867. *progfd = prog_parse_fd(&argc, &argv);
  868. if (*progfd < 0)
  869. return *progfd;
  870. *attach_type = parse_attach_type(*argv);
  871. if (*attach_type == __MAX_BPF_ATTACH_TYPE) {
  872. p_err("invalid attach/detach type");
  873. return -EINVAL;
  874. }
  875. if (*attach_type == BPF_FLOW_DISSECTOR) {
  876. *mapfd = 0;
  877. return 0;
  878. }
  879. NEXT_ARG();
  880. if (!REQ_ARGS(2))
  881. return -EINVAL;
  882. *mapfd = map_parse_fd(&argc, &argv, 0);
  883. if (*mapfd < 0)
  884. return *mapfd;
  885. return 0;
  886. }
  887. static int do_attach(int argc, char **argv)
  888. {
  889. enum bpf_attach_type attach_type;
  890. int err, progfd;
  891. int mapfd;
  892. err = parse_attach_detach_args(argc, argv,
  893. &progfd, &attach_type, &mapfd);
  894. if (err)
  895. return err;
  896. err = bpf_prog_attach(progfd, mapfd, attach_type, 0);
  897. if (err) {
  898. p_err("failed prog attach to map");
  899. return -EINVAL;
  900. }
  901. if (json_output)
  902. jsonw_null(json_wtr);
  903. return 0;
  904. }
  905. static int do_detach(int argc, char **argv)
  906. {
  907. enum bpf_attach_type attach_type;
  908. int err, progfd;
  909. int mapfd;
  910. err = parse_attach_detach_args(argc, argv,
  911. &progfd, &attach_type, &mapfd);
  912. if (err)
  913. return err;
  914. err = bpf_prog_detach2(progfd, mapfd, attach_type);
  915. if (err) {
  916. p_err("failed prog detach from map");
  917. return -EINVAL;
  918. }
  919. if (json_output)
  920. jsonw_null(json_wtr);
  921. return 0;
  922. }
  923. enum prog_tracelog_mode {
  924. TRACE_STDOUT,
  925. TRACE_STDERR,
  926. };
  927. static int
  928. prog_tracelog_stream(int prog_fd, enum prog_tracelog_mode mode)
  929. {
  930. FILE *file = mode == TRACE_STDOUT ? stdout : stderr;
  931. int stream_id = mode == TRACE_STDOUT ? 1 : 2;
  932. char buf[512];
  933. int ret;
  934. ret = 0;
  935. do {
  936. ret = bpf_prog_stream_read(prog_fd, stream_id, buf, sizeof(buf), NULL);
  937. if (ret > 0)
  938. fwrite(buf, sizeof(buf[0]), ret, file);
  939. } while (ret > 0);
  940. fflush(file);
  941. return ret ? -1 : 0;
  942. }
  943. static int do_tracelog_any(int argc, char **argv)
  944. {
  945. enum prog_tracelog_mode mode;
  946. int fd;
  947. if (argc == 0)
  948. return do_tracelog(argc, argv);
  949. if (!is_prefix(*argv, "stdout") && !is_prefix(*argv, "stderr"))
  950. usage();
  951. mode = is_prefix(*argv, "stdout") ? TRACE_STDOUT : TRACE_STDERR;
  952. NEXT_ARG();
  953. if (!REQ_ARGS(2))
  954. return -1;
  955. fd = prog_parse_fd(&argc, &argv);
  956. if (fd < 0)
  957. return -1;
  958. return prog_tracelog_stream(fd, mode);
  959. }
  960. static int check_single_stdin(char *file_data_in, char *file_ctx_in)
  961. {
  962. if (file_data_in && file_ctx_in &&
  963. !strcmp(file_data_in, "-") && !strcmp(file_ctx_in, "-")) {
  964. p_err("cannot use standard input for both data_in and ctx_in");
  965. return -1;
  966. }
  967. return 0;
  968. }
  969. static int get_run_data(const char *fname, void **data_ptr, unsigned int *size)
  970. {
  971. size_t block_size = 256;
  972. size_t buf_size = block_size;
  973. size_t nb_read = 0;
  974. void *tmp;
  975. FILE *f;
  976. if (!fname) {
  977. *data_ptr = NULL;
  978. *size = 0;
  979. return 0;
  980. }
  981. if (!strcmp(fname, "-"))
  982. f = stdin;
  983. else
  984. f = fopen(fname, "r");
  985. if (!f) {
  986. p_err("failed to open %s: %s", fname, strerror(errno));
  987. return -1;
  988. }
  989. *data_ptr = malloc(block_size);
  990. if (!*data_ptr) {
  991. p_err("failed to allocate memory for data_in/ctx_in: %s",
  992. strerror(errno));
  993. goto err_fclose;
  994. }
  995. while ((nb_read += fread(*data_ptr + nb_read, 1, block_size, f))) {
  996. if (feof(f))
  997. break;
  998. if (ferror(f)) {
  999. p_err("failed to read data_in/ctx_in from %s: %s",
  1000. fname, strerror(errno));
  1001. goto err_free;
  1002. }
  1003. if (nb_read > buf_size - block_size) {
  1004. if (buf_size == UINT32_MAX) {
  1005. p_err("data_in/ctx_in is too long (max: %u)",
  1006. UINT32_MAX);
  1007. goto err_free;
  1008. }
  1009. /* No space for fread()-ing next chunk; realloc() */
  1010. buf_size *= 2;
  1011. tmp = realloc(*data_ptr, buf_size);
  1012. if (!tmp) {
  1013. p_err("failed to reallocate data_in/ctx_in: %s",
  1014. strerror(errno));
  1015. goto err_free;
  1016. }
  1017. *data_ptr = tmp;
  1018. }
  1019. }
  1020. if (f != stdin)
  1021. fclose(f);
  1022. *size = nb_read;
  1023. return 0;
  1024. err_free:
  1025. free(*data_ptr);
  1026. *data_ptr = NULL;
  1027. err_fclose:
  1028. if (f != stdin)
  1029. fclose(f);
  1030. return -1;
  1031. }
  1032. static void hex_print(void *data, unsigned int size, FILE *f)
  1033. {
  1034. size_t i, j;
  1035. char c;
  1036. for (i = 0; i < size; i += 16) {
  1037. /* Row offset */
  1038. fprintf(f, "%07zx\t", i);
  1039. /* Hexadecimal values */
  1040. for (j = i; j < i + 16 && j < size; j++)
  1041. fprintf(f, "%02x%s", *(uint8_t *)(data + j),
  1042. j % 2 ? " " : "");
  1043. for (; j < i + 16; j++)
  1044. fprintf(f, " %s", j % 2 ? " " : "");
  1045. /* ASCII values (if relevant), '.' otherwise */
  1046. fprintf(f, "| ");
  1047. for (j = i; j < i + 16 && j < size; j++) {
  1048. c = *(char *)(data + j);
  1049. if (c < ' ' || c > '~')
  1050. c = '.';
  1051. fprintf(f, "%c%s", c, j == i + 7 ? " " : "");
  1052. }
  1053. fprintf(f, "\n");
  1054. }
  1055. }
  1056. static int
  1057. print_run_output(void *data, unsigned int size, const char *fname,
  1058. const char *json_key)
  1059. {
  1060. size_t nb_written;
  1061. FILE *f;
  1062. if (!fname)
  1063. return 0;
  1064. if (!strcmp(fname, "-")) {
  1065. f = stdout;
  1066. if (json_output) {
  1067. jsonw_name(json_wtr, json_key);
  1068. print_data_json(data, size);
  1069. } else {
  1070. hex_print(data, size, f);
  1071. }
  1072. return 0;
  1073. }
  1074. f = fopen(fname, "w");
  1075. if (!f) {
  1076. p_err("failed to open %s: %s", fname, strerror(errno));
  1077. return -1;
  1078. }
  1079. nb_written = fwrite(data, 1, size, f);
  1080. fclose(f);
  1081. if (nb_written != size) {
  1082. p_err("failed to write output data/ctx: %s", strerror(errno));
  1083. return -1;
  1084. }
  1085. return 0;
  1086. }
  1087. static int alloc_run_data(void **data_ptr, unsigned int size_out)
  1088. {
  1089. *data_ptr = calloc(size_out, 1);
  1090. if (!*data_ptr) {
  1091. p_err("failed to allocate memory for output data/ctx: %s",
  1092. strerror(errno));
  1093. return -1;
  1094. }
  1095. return 0;
  1096. }
  1097. static int do_run(int argc, char **argv)
  1098. {
  1099. char *data_fname_in = NULL, *data_fname_out = NULL;
  1100. char *ctx_fname_in = NULL, *ctx_fname_out = NULL;
  1101. const unsigned int default_size = SZ_32K;
  1102. void *data_in = NULL, *data_out = NULL;
  1103. void *ctx_in = NULL, *ctx_out = NULL;
  1104. unsigned int repeat = 1;
  1105. int fd, err;
  1106. LIBBPF_OPTS(bpf_test_run_opts, test_attr);
  1107. if (!REQ_ARGS(4))
  1108. return -1;
  1109. fd = prog_parse_fd(&argc, &argv);
  1110. if (fd < 0)
  1111. return -1;
  1112. while (argc) {
  1113. if (detect_common_prefix(*argv, "data_in", "data_out",
  1114. "data_size_out", NULL))
  1115. return -1;
  1116. if (detect_common_prefix(*argv, "ctx_in", "ctx_out",
  1117. "ctx_size_out", NULL))
  1118. return -1;
  1119. if (is_prefix(*argv, "data_in")) {
  1120. NEXT_ARG();
  1121. if (!REQ_ARGS(1))
  1122. return -1;
  1123. data_fname_in = GET_ARG();
  1124. if (check_single_stdin(data_fname_in, ctx_fname_in))
  1125. return -1;
  1126. } else if (is_prefix(*argv, "data_out")) {
  1127. NEXT_ARG();
  1128. if (!REQ_ARGS(1))
  1129. return -1;
  1130. data_fname_out = GET_ARG();
  1131. } else if (is_prefix(*argv, "data_size_out")) {
  1132. char *endptr;
  1133. NEXT_ARG();
  1134. if (!REQ_ARGS(1))
  1135. return -1;
  1136. test_attr.data_size_out = strtoul(*argv, &endptr, 0);
  1137. if (*endptr) {
  1138. p_err("can't parse %s as output data size",
  1139. *argv);
  1140. return -1;
  1141. }
  1142. NEXT_ARG();
  1143. } else if (is_prefix(*argv, "ctx_in")) {
  1144. NEXT_ARG();
  1145. if (!REQ_ARGS(1))
  1146. return -1;
  1147. ctx_fname_in = GET_ARG();
  1148. if (check_single_stdin(data_fname_in, ctx_fname_in))
  1149. return -1;
  1150. } else if (is_prefix(*argv, "ctx_out")) {
  1151. NEXT_ARG();
  1152. if (!REQ_ARGS(1))
  1153. return -1;
  1154. ctx_fname_out = GET_ARG();
  1155. } else if (is_prefix(*argv, "ctx_size_out")) {
  1156. char *endptr;
  1157. NEXT_ARG();
  1158. if (!REQ_ARGS(1))
  1159. return -1;
  1160. test_attr.ctx_size_out = strtoul(*argv, &endptr, 0);
  1161. if (*endptr) {
  1162. p_err("can't parse %s as output context size",
  1163. *argv);
  1164. return -1;
  1165. }
  1166. NEXT_ARG();
  1167. } else if (is_prefix(*argv, "repeat")) {
  1168. char *endptr;
  1169. NEXT_ARG();
  1170. if (!REQ_ARGS(1))
  1171. return -1;
  1172. repeat = strtoul(*argv, &endptr, 0);
  1173. if (*endptr) {
  1174. p_err("can't parse %s as repeat number",
  1175. *argv);
  1176. return -1;
  1177. }
  1178. NEXT_ARG();
  1179. } else {
  1180. p_err("expected no more arguments, 'data_in', 'data_out', 'data_size_out', 'ctx_in', 'ctx_out', 'ctx_size_out' or 'repeat', got: '%s'?",
  1181. *argv);
  1182. return -1;
  1183. }
  1184. }
  1185. err = get_run_data(data_fname_in, &data_in, &test_attr.data_size_in);
  1186. if (err)
  1187. return -1;
  1188. if (data_in) {
  1189. if (!test_attr.data_size_out)
  1190. test_attr.data_size_out = default_size;
  1191. err = alloc_run_data(&data_out, test_attr.data_size_out);
  1192. if (err)
  1193. goto free_data_in;
  1194. }
  1195. err = get_run_data(ctx_fname_in, &ctx_in, &test_attr.ctx_size_in);
  1196. if (err)
  1197. goto free_data_out;
  1198. if (ctx_in) {
  1199. if (!test_attr.ctx_size_out)
  1200. test_attr.ctx_size_out = default_size;
  1201. err = alloc_run_data(&ctx_out, test_attr.ctx_size_out);
  1202. if (err)
  1203. goto free_ctx_in;
  1204. }
  1205. test_attr.repeat = repeat;
  1206. test_attr.data_in = data_in;
  1207. test_attr.data_out = data_out;
  1208. test_attr.ctx_in = ctx_in;
  1209. test_attr.ctx_out = ctx_out;
  1210. err = bpf_prog_test_run_opts(fd, &test_attr);
  1211. if (err) {
  1212. p_err("failed to run program: %s", strerror(errno));
  1213. goto free_ctx_out;
  1214. }
  1215. err = 0;
  1216. if (json_output)
  1217. jsonw_start_object(json_wtr); /* root */
  1218. /* Do not exit on errors occurring when printing output data/context,
  1219. * we still want to print return value and duration for program run.
  1220. */
  1221. if (test_attr.data_size_out)
  1222. err += print_run_output(test_attr.data_out,
  1223. test_attr.data_size_out,
  1224. data_fname_out, "data_out");
  1225. if (test_attr.ctx_size_out)
  1226. err += print_run_output(test_attr.ctx_out,
  1227. test_attr.ctx_size_out,
  1228. ctx_fname_out, "ctx_out");
  1229. if (json_output) {
  1230. jsonw_uint_field(json_wtr, "retval", test_attr.retval);
  1231. jsonw_uint_field(json_wtr, "duration", test_attr.duration);
  1232. jsonw_end_object(json_wtr); /* root */
  1233. } else {
  1234. fprintf(stdout, "Return value: %u, duration%s: %uns\n",
  1235. test_attr.retval,
  1236. repeat > 1 ? " (average)" : "", test_attr.duration);
  1237. }
  1238. free_ctx_out:
  1239. free(ctx_out);
  1240. free_ctx_in:
  1241. free(ctx_in);
  1242. free_data_out:
  1243. free(data_out);
  1244. free_data_in:
  1245. free(data_in);
  1246. return err;
  1247. }
  1248. static int
  1249. get_prog_type_by_name(const char *name, enum bpf_prog_type *prog_type,
  1250. enum bpf_attach_type *expected_attach_type)
  1251. {
  1252. libbpf_print_fn_t print_backup;
  1253. int ret;
  1254. ret = libbpf_prog_type_by_name(name, prog_type, expected_attach_type);
  1255. if (!ret)
  1256. return ret;
  1257. /* libbpf_prog_type_by_name() failed, let's re-run with debug level */
  1258. print_backup = libbpf_set_print(print_all_levels);
  1259. ret = libbpf_prog_type_by_name(name, prog_type, expected_attach_type);
  1260. libbpf_set_print(print_backup);
  1261. return ret;
  1262. }
  1263. static int
  1264. auto_attach_program(struct bpf_program *prog, const char *path)
  1265. {
  1266. struct bpf_link *link;
  1267. int err;
  1268. link = bpf_program__attach(prog);
  1269. if (!link) {
  1270. p_info("Program %s does not support autoattach, falling back to pinning",
  1271. bpf_program__name(prog));
  1272. return bpf_obj_pin(bpf_program__fd(prog), path);
  1273. }
  1274. err = bpf_link__pin(link, path);
  1275. bpf_link__destroy(link);
  1276. return err;
  1277. }
  1278. static int
  1279. auto_attach_programs(struct bpf_object *obj, const char *path)
  1280. {
  1281. struct bpf_program *prog;
  1282. char buf[PATH_MAX];
  1283. int err;
  1284. bpf_object__for_each_program(prog, obj) {
  1285. err = pathname_concat(buf, sizeof(buf), path, bpf_program__name(prog));
  1286. if (err)
  1287. goto err_unpin_programs;
  1288. err = auto_attach_program(prog, buf);
  1289. if (err)
  1290. goto err_unpin_programs;
  1291. }
  1292. return 0;
  1293. err_unpin_programs:
  1294. while ((prog = bpf_object__prev_program(obj, prog))) {
  1295. if (pathname_concat(buf, sizeof(buf), path, bpf_program__name(prog)))
  1296. continue;
  1297. bpf_program__unpin(prog, buf);
  1298. }
  1299. return err;
  1300. }
  1301. static int load_with_options(int argc, char **argv, bool first_prog_only)
  1302. {
  1303. enum bpf_prog_type common_prog_type = BPF_PROG_TYPE_UNSPEC;
  1304. DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts,
  1305. .relaxed_maps = relaxed_maps,
  1306. );
  1307. enum bpf_attach_type expected_attach_type;
  1308. struct map_replace *map_replace = NULL;
  1309. struct bpf_program *prog = NULL, *pos;
  1310. unsigned int old_map_fds = 0;
  1311. const char *pinmaps = NULL;
  1312. __u32 xdpmeta_ifindex = 0;
  1313. __u32 offload_ifindex = 0;
  1314. bool auto_attach = false;
  1315. struct bpf_object *obj;
  1316. struct bpf_map *map;
  1317. const char *pinfile;
  1318. unsigned int i, j;
  1319. const char *file;
  1320. int idx, err;
  1321. if (!REQ_ARGS(2))
  1322. return -1;
  1323. file = GET_ARG();
  1324. pinfile = GET_ARG();
  1325. while (argc) {
  1326. if (is_prefix(*argv, "type")) {
  1327. NEXT_ARG();
  1328. if (common_prog_type != BPF_PROG_TYPE_UNSPEC) {
  1329. p_err("program type already specified");
  1330. goto err_free_reuse_maps;
  1331. }
  1332. if (!REQ_ARGS(1))
  1333. goto err_free_reuse_maps;
  1334. err = libbpf_prog_type_by_name(*argv, &common_prog_type,
  1335. &expected_attach_type);
  1336. if (err < 0) {
  1337. /* Put a '/' at the end of type to appease libbpf */
  1338. char *type = malloc(strlen(*argv) + 2);
  1339. if (!type) {
  1340. p_err("mem alloc failed");
  1341. goto err_free_reuse_maps;
  1342. }
  1343. *type = 0;
  1344. strcat(type, *argv);
  1345. strcat(type, "/");
  1346. err = get_prog_type_by_name(type, &common_prog_type,
  1347. &expected_attach_type);
  1348. free(type);
  1349. if (err < 0)
  1350. goto err_free_reuse_maps;
  1351. }
  1352. NEXT_ARG();
  1353. } else if (is_prefix(*argv, "map")) {
  1354. void *new_map_replace;
  1355. char *endptr, *name;
  1356. int fd;
  1357. NEXT_ARG();
  1358. if (!REQ_ARGS(4))
  1359. goto err_free_reuse_maps;
  1360. if (is_prefix(*argv, "idx")) {
  1361. NEXT_ARG();
  1362. idx = strtoul(*argv, &endptr, 0);
  1363. if (*endptr) {
  1364. p_err("can't parse %s as IDX", *argv);
  1365. goto err_free_reuse_maps;
  1366. }
  1367. name = NULL;
  1368. } else if (is_prefix(*argv, "name")) {
  1369. NEXT_ARG();
  1370. name = *argv;
  1371. idx = -1;
  1372. } else {
  1373. p_err("expected 'idx' or 'name', got: '%s'?",
  1374. *argv);
  1375. goto err_free_reuse_maps;
  1376. }
  1377. NEXT_ARG();
  1378. fd = map_parse_fd(&argc, &argv, 0);
  1379. if (fd < 0)
  1380. goto err_free_reuse_maps;
  1381. new_map_replace = libbpf_reallocarray(map_replace,
  1382. old_map_fds + 1,
  1383. sizeof(*map_replace));
  1384. if (!new_map_replace) {
  1385. p_err("mem alloc failed");
  1386. goto err_free_reuse_maps;
  1387. }
  1388. map_replace = new_map_replace;
  1389. map_replace[old_map_fds].idx = idx;
  1390. map_replace[old_map_fds].name = name;
  1391. map_replace[old_map_fds].fd = fd;
  1392. old_map_fds++;
  1393. } else if (is_prefix(*argv, "dev")) {
  1394. p_info("Warning: 'bpftool prog load [...] dev <ifname>' syntax is deprecated.\n"
  1395. "Going further, please use 'offload_dev <ifname>' to offload program to device.\n"
  1396. "For applications using XDP hints only, use 'xdpmeta_dev <ifname>'.");
  1397. goto offload_dev;
  1398. } else if (is_prefix(*argv, "offload_dev")) {
  1399. offload_dev:
  1400. NEXT_ARG();
  1401. if (offload_ifindex) {
  1402. p_err("offload_dev already specified");
  1403. goto err_free_reuse_maps;
  1404. } else if (xdpmeta_ifindex) {
  1405. p_err("xdpmeta_dev and offload_dev are mutually exclusive");
  1406. goto err_free_reuse_maps;
  1407. }
  1408. if (!REQ_ARGS(1))
  1409. goto err_free_reuse_maps;
  1410. offload_ifindex = if_nametoindex(*argv);
  1411. if (!offload_ifindex) {
  1412. p_err("unrecognized netdevice '%s': %s",
  1413. *argv, strerror(errno));
  1414. goto err_free_reuse_maps;
  1415. }
  1416. NEXT_ARG();
  1417. } else if (is_prefix(*argv, "xdpmeta_dev")) {
  1418. NEXT_ARG();
  1419. if (xdpmeta_ifindex) {
  1420. p_err("xdpmeta_dev already specified");
  1421. goto err_free_reuse_maps;
  1422. } else if (offload_ifindex) {
  1423. p_err("xdpmeta_dev and offload_dev are mutually exclusive");
  1424. goto err_free_reuse_maps;
  1425. }
  1426. if (!REQ_ARGS(1))
  1427. goto err_free_reuse_maps;
  1428. xdpmeta_ifindex = if_nametoindex(*argv);
  1429. if (!xdpmeta_ifindex) {
  1430. p_err("unrecognized netdevice '%s': %s",
  1431. *argv, strerror(errno));
  1432. goto err_free_reuse_maps;
  1433. }
  1434. NEXT_ARG();
  1435. } else if (is_prefix(*argv, "pinmaps")) {
  1436. NEXT_ARG();
  1437. if (!REQ_ARGS(1))
  1438. goto err_free_reuse_maps;
  1439. pinmaps = GET_ARG();
  1440. } else if (is_prefix(*argv, "autoattach")) {
  1441. auto_attach = true;
  1442. NEXT_ARG();
  1443. } else if (is_prefix(*argv, "kernel_btf")) {
  1444. NEXT_ARG();
  1445. if (!REQ_ARGS(1))
  1446. goto err_free_reuse_maps;
  1447. open_opts.btf_custom_path = GET_ARG();
  1448. } else {
  1449. p_err("expected no more arguments, "
  1450. "'type', 'map', 'offload_dev', 'xdpmeta_dev', 'pinmaps', "
  1451. "'autoattach', or 'kernel_btf', got: '%s'?",
  1452. *argv);
  1453. goto err_free_reuse_maps;
  1454. }
  1455. }
  1456. set_max_rlimit();
  1457. if (verifier_logs)
  1458. /* log_level1 + log_level2 + stats, but not stable UAPI */
  1459. open_opts.kernel_log_level = 1 + 2 + 4;
  1460. obj = bpf_object__open_file(file, &open_opts);
  1461. if (!obj) {
  1462. p_err("failed to open object file");
  1463. goto err_free_reuse_maps;
  1464. }
  1465. bpf_object__for_each_program(pos, obj) {
  1466. enum bpf_prog_type prog_type = common_prog_type;
  1467. if (prog_type == BPF_PROG_TYPE_UNSPEC) {
  1468. const char *sec_name = bpf_program__section_name(pos);
  1469. err = get_prog_type_by_name(sec_name, &prog_type,
  1470. &expected_attach_type);
  1471. if (err < 0)
  1472. goto err_close_obj;
  1473. }
  1474. if (prog_type == BPF_PROG_TYPE_XDP && xdpmeta_ifindex) {
  1475. bpf_program__set_flags(pos, BPF_F_XDP_DEV_BOUND_ONLY);
  1476. bpf_program__set_ifindex(pos, xdpmeta_ifindex);
  1477. } else {
  1478. bpf_program__set_ifindex(pos, offload_ifindex);
  1479. }
  1480. if (bpf_program__type(pos) != prog_type)
  1481. bpf_program__set_type(pos, prog_type);
  1482. bpf_program__set_expected_attach_type(pos, expected_attach_type);
  1483. }
  1484. qsort(map_replace, old_map_fds, sizeof(*map_replace),
  1485. map_replace_compar);
  1486. /* After the sort maps by name will be first on the list, because they
  1487. * have idx == -1. Resolve them.
  1488. */
  1489. j = 0;
  1490. while (j < old_map_fds && map_replace[j].name) {
  1491. i = 0;
  1492. bpf_object__for_each_map(map, obj) {
  1493. if (!strcmp(bpf_map__name(map), map_replace[j].name)) {
  1494. map_replace[j].idx = i;
  1495. break;
  1496. }
  1497. i++;
  1498. }
  1499. if (map_replace[j].idx == -1) {
  1500. p_err("unable to find map '%s'", map_replace[j].name);
  1501. goto err_close_obj;
  1502. }
  1503. j++;
  1504. }
  1505. /* Resort if any names were resolved */
  1506. if (j)
  1507. qsort(map_replace, old_map_fds, sizeof(*map_replace),
  1508. map_replace_compar);
  1509. /* Set ifindex and name reuse */
  1510. j = 0;
  1511. idx = 0;
  1512. bpf_object__for_each_map(map, obj) {
  1513. if (bpf_map__type(map) != BPF_MAP_TYPE_PERF_EVENT_ARRAY)
  1514. bpf_map__set_ifindex(map, offload_ifindex);
  1515. if (j < old_map_fds && idx == map_replace[j].idx) {
  1516. err = bpf_map__reuse_fd(map, map_replace[j++].fd);
  1517. if (err) {
  1518. p_err("unable to set up map reuse: %d", err);
  1519. goto err_close_obj;
  1520. }
  1521. /* Next reuse wants to apply to the same map */
  1522. if (j < old_map_fds && map_replace[j].idx == idx) {
  1523. p_err("replacement for map idx %d specified more than once",
  1524. idx);
  1525. goto err_close_obj;
  1526. }
  1527. }
  1528. idx++;
  1529. }
  1530. if (j < old_map_fds) {
  1531. p_err("map idx '%d' not used", map_replace[j].idx);
  1532. goto err_close_obj;
  1533. }
  1534. err = bpf_object__load(obj);
  1535. if (err) {
  1536. p_err("failed to load object file");
  1537. goto err_close_obj;
  1538. }
  1539. if (first_prog_only)
  1540. err = mount_bpffs_for_file(pinfile);
  1541. else
  1542. err = create_and_mount_bpffs_dir(pinfile);
  1543. if (err)
  1544. goto err_close_obj;
  1545. if (first_prog_only) {
  1546. prog = bpf_object__next_program(obj, NULL);
  1547. if (!prog) {
  1548. p_err("object file doesn't contain any bpf program");
  1549. goto err_close_obj;
  1550. }
  1551. if (auto_attach)
  1552. err = auto_attach_program(prog, pinfile);
  1553. else
  1554. err = bpf_obj_pin(bpf_program__fd(prog), pinfile);
  1555. if (err) {
  1556. p_err("failed to pin program %s",
  1557. bpf_program__section_name(prog));
  1558. goto err_close_obj;
  1559. }
  1560. } else {
  1561. if (auto_attach)
  1562. err = auto_attach_programs(obj, pinfile);
  1563. else
  1564. err = bpf_object__pin_programs(obj, pinfile);
  1565. if (err) {
  1566. p_err("failed to pin all programs");
  1567. goto err_close_obj;
  1568. }
  1569. }
  1570. if (pinmaps) {
  1571. err = create_and_mount_bpffs_dir(pinmaps);
  1572. if (err)
  1573. goto err_unpin;
  1574. err = bpf_object__pin_maps(obj, pinmaps);
  1575. if (err) {
  1576. p_err("failed to pin all maps");
  1577. goto err_unpin;
  1578. }
  1579. }
  1580. if (json_output)
  1581. jsonw_null(json_wtr);
  1582. bpf_object__close(obj);
  1583. for (i = 0; i < old_map_fds; i++)
  1584. close(map_replace[i].fd);
  1585. free(map_replace);
  1586. return 0;
  1587. err_unpin:
  1588. if (first_prog_only)
  1589. unlink(pinfile);
  1590. else
  1591. bpf_object__unpin_programs(obj, pinfile);
  1592. err_close_obj:
  1593. bpf_object__close(obj);
  1594. err_free_reuse_maps:
  1595. for (i = 0; i < old_map_fds; i++)
  1596. close(map_replace[i].fd);
  1597. free(map_replace);
  1598. return -1;
  1599. }
  1600. static int count_open_fds(void)
  1601. {
  1602. DIR *dp = opendir("/proc/self/fd");
  1603. struct dirent *de;
  1604. int cnt = -3;
  1605. if (!dp)
  1606. return -1;
  1607. while ((de = readdir(dp)))
  1608. cnt++;
  1609. closedir(dp);
  1610. return cnt;
  1611. }
  1612. static int try_loader(struct gen_loader_opts *gen)
  1613. {
  1614. struct bpf_load_and_run_opts opts = {};
  1615. struct bpf_loader_ctx *ctx;
  1616. char sig_buf[MAX_SIG_SIZE];
  1617. __u8 prog_sha[SHA256_DIGEST_LENGTH];
  1618. int ctx_sz = sizeof(*ctx) + 64 * max(sizeof(struct bpf_map_desc),
  1619. sizeof(struct bpf_prog_desc));
  1620. int log_buf_sz = (1u << 24) - 1;
  1621. int err, fds_before, fd_delta;
  1622. char *log_buf = NULL;
  1623. ctx = alloca(ctx_sz);
  1624. memset(ctx, 0, ctx_sz);
  1625. ctx->sz = ctx_sz;
  1626. if (verifier_logs) {
  1627. ctx->log_level = 1 + 2 + 4;
  1628. ctx->log_size = log_buf_sz;
  1629. log_buf = malloc(log_buf_sz);
  1630. if (!log_buf)
  1631. return -ENOMEM;
  1632. ctx->log_buf = (long) log_buf;
  1633. }
  1634. opts.ctx = ctx;
  1635. opts.data = gen->data;
  1636. opts.data_sz = gen->data_sz;
  1637. opts.insns = gen->insns;
  1638. opts.insns_sz = gen->insns_sz;
  1639. fds_before = count_open_fds();
  1640. if (sign_progs) {
  1641. opts.excl_prog_hash = prog_sha;
  1642. opts.excl_prog_hash_sz = sizeof(prog_sha);
  1643. opts.signature = sig_buf;
  1644. opts.signature_sz = MAX_SIG_SIZE;
  1645. opts.keyring_id = KEY_SPEC_SESSION_KEYRING;
  1646. err = bpftool_prog_sign(&opts);
  1647. if (err < 0) {
  1648. p_err("failed to sign program");
  1649. goto out;
  1650. }
  1651. err = register_session_key(cert_path);
  1652. if (err < 0) {
  1653. p_err("failed to add session key");
  1654. goto out;
  1655. }
  1656. }
  1657. err = bpf_load_and_run(&opts);
  1658. fd_delta = count_open_fds() - fds_before;
  1659. if (err < 0 || verifier_logs) {
  1660. fprintf(stderr, "err %d\n%s\n%s", err, opts.errstr, log_buf);
  1661. if (fd_delta && err < 0)
  1662. fprintf(stderr, "loader prog leaked %d FDs\n",
  1663. fd_delta);
  1664. }
  1665. out:
  1666. free(log_buf);
  1667. return err;
  1668. }
  1669. static int do_loader(int argc, char **argv)
  1670. {
  1671. DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts);
  1672. DECLARE_LIBBPF_OPTS(gen_loader_opts, gen);
  1673. struct bpf_object *obj;
  1674. const char *file;
  1675. int err = 0;
  1676. if (!REQ_ARGS(1))
  1677. return -1;
  1678. file = GET_ARG();
  1679. if (verifier_logs)
  1680. /* log_level1 + log_level2 + stats, but not stable UAPI */
  1681. open_opts.kernel_log_level = 1 + 2 + 4;
  1682. obj = bpf_object__open_file(file, &open_opts);
  1683. if (!obj) {
  1684. err = -1;
  1685. p_err("failed to open object file");
  1686. goto err_close_obj;
  1687. }
  1688. if (sign_progs)
  1689. gen.gen_hash = true;
  1690. err = bpf_object__gen_loader(obj, &gen);
  1691. if (err)
  1692. goto err_close_obj;
  1693. err = bpf_object__load(obj);
  1694. if (err) {
  1695. p_err("failed to load object file");
  1696. goto err_close_obj;
  1697. }
  1698. if (verifier_logs) {
  1699. struct dump_data dd = {};
  1700. kernel_syms_load(&dd);
  1701. dump_xlated_plain(&dd, (void *)gen.insns, gen.insns_sz, false, false);
  1702. kernel_syms_destroy(&dd);
  1703. }
  1704. err = try_loader(&gen);
  1705. err_close_obj:
  1706. bpf_object__close(obj);
  1707. return err;
  1708. }
  1709. static int do_load(int argc, char **argv)
  1710. {
  1711. if (use_loader)
  1712. return do_loader(argc, argv);
  1713. return load_with_options(argc, argv, true);
  1714. }
  1715. static int do_loadall(int argc, char **argv)
  1716. {
  1717. return load_with_options(argc, argv, false);
  1718. }
  1719. #ifdef BPFTOOL_WITHOUT_SKELETONS
  1720. static int do_profile(int argc, char **argv)
  1721. {
  1722. p_err("bpftool prog profile command is not supported. Please build bpftool with clang >= 10.0.0");
  1723. return 0;
  1724. }
  1725. #else /* BPFTOOL_WITHOUT_SKELETONS */
  1726. #include "profiler.skel.h"
  1727. struct profile_metric {
  1728. const char *name;
  1729. struct bpf_perf_event_value val;
  1730. struct perf_event_attr attr;
  1731. bool selected;
  1732. /* calculate ratios like instructions per cycle */
  1733. const int ratio_metric; /* 0 for N/A, 1 for index 0 (cycles) */
  1734. const char *ratio_desc;
  1735. const float ratio_mul;
  1736. } metrics[] = {
  1737. {
  1738. .name = "cycles",
  1739. .attr = {
  1740. .type = PERF_TYPE_HARDWARE,
  1741. .config = PERF_COUNT_HW_CPU_CYCLES,
  1742. .exclude_user = 1,
  1743. },
  1744. },
  1745. {
  1746. .name = "instructions",
  1747. .attr = {
  1748. .type = PERF_TYPE_HARDWARE,
  1749. .config = PERF_COUNT_HW_INSTRUCTIONS,
  1750. .exclude_user = 1,
  1751. },
  1752. .ratio_metric = 1,
  1753. .ratio_desc = "insns per cycle",
  1754. .ratio_mul = 1.0,
  1755. },
  1756. {
  1757. .name = "l1d_loads",
  1758. .attr = {
  1759. .type = PERF_TYPE_HW_CACHE,
  1760. .config =
  1761. PERF_COUNT_HW_CACHE_L1D |
  1762. (PERF_COUNT_HW_CACHE_OP_READ << 8) |
  1763. (PERF_COUNT_HW_CACHE_RESULT_ACCESS << 16),
  1764. .exclude_user = 1,
  1765. },
  1766. },
  1767. {
  1768. .name = "llc_misses",
  1769. .attr = {
  1770. .type = PERF_TYPE_HW_CACHE,
  1771. .config =
  1772. PERF_COUNT_HW_CACHE_LL |
  1773. (PERF_COUNT_HW_CACHE_OP_READ << 8) |
  1774. (PERF_COUNT_HW_CACHE_RESULT_MISS << 16),
  1775. .exclude_user = 1
  1776. },
  1777. .ratio_metric = 2,
  1778. .ratio_desc = "LLC misses per million insns",
  1779. .ratio_mul = 1e6,
  1780. },
  1781. {
  1782. .name = "itlb_misses",
  1783. .attr = {
  1784. .type = PERF_TYPE_HW_CACHE,
  1785. .config =
  1786. PERF_COUNT_HW_CACHE_ITLB |
  1787. (PERF_COUNT_HW_CACHE_OP_READ << 8) |
  1788. (PERF_COUNT_HW_CACHE_RESULT_MISS << 16),
  1789. .exclude_user = 1
  1790. },
  1791. .ratio_metric = 2,
  1792. .ratio_desc = "itlb misses per million insns",
  1793. .ratio_mul = 1e6,
  1794. },
  1795. {
  1796. .name = "dtlb_misses",
  1797. .attr = {
  1798. .type = PERF_TYPE_HW_CACHE,
  1799. .config =
  1800. PERF_COUNT_HW_CACHE_DTLB |
  1801. (PERF_COUNT_HW_CACHE_OP_READ << 8) |
  1802. (PERF_COUNT_HW_CACHE_RESULT_MISS << 16),
  1803. .exclude_user = 1
  1804. },
  1805. .ratio_metric = 2,
  1806. .ratio_desc = "dtlb misses per million insns",
  1807. .ratio_mul = 1e6,
  1808. },
  1809. };
  1810. static __u64 profile_total_count;
  1811. #define MAX_NUM_PROFILE_METRICS 4
  1812. static int profile_parse_metrics(int argc, char **argv)
  1813. {
  1814. unsigned int metric_cnt;
  1815. int selected_cnt = 0;
  1816. unsigned int i;
  1817. metric_cnt = ARRAY_SIZE(metrics);
  1818. while (argc > 0) {
  1819. for (i = 0; i < metric_cnt; i++) {
  1820. if (is_prefix(argv[0], metrics[i].name)) {
  1821. if (!metrics[i].selected)
  1822. selected_cnt++;
  1823. metrics[i].selected = true;
  1824. break;
  1825. }
  1826. }
  1827. if (i == metric_cnt) {
  1828. p_err("unknown metric %s", argv[0]);
  1829. return -1;
  1830. }
  1831. NEXT_ARG();
  1832. }
  1833. if (selected_cnt > MAX_NUM_PROFILE_METRICS) {
  1834. p_err("too many (%d) metrics, please specify no more than %d metrics at a time",
  1835. selected_cnt, MAX_NUM_PROFILE_METRICS);
  1836. return -1;
  1837. }
  1838. return selected_cnt;
  1839. }
  1840. static void profile_read_values(struct profiler_bpf *obj)
  1841. {
  1842. __u32 m, cpu, num_cpu = obj->rodata->num_cpu;
  1843. int reading_map_fd, count_map_fd;
  1844. __u64 counts[num_cpu];
  1845. __u32 key = 0;
  1846. int err;
  1847. reading_map_fd = bpf_map__fd(obj->maps.accum_readings);
  1848. count_map_fd = bpf_map__fd(obj->maps.counts);
  1849. if (reading_map_fd < 0 || count_map_fd < 0) {
  1850. p_err("failed to get fd for map");
  1851. return;
  1852. }
  1853. err = bpf_map_lookup_elem(count_map_fd, &key, counts);
  1854. if (err) {
  1855. p_err("failed to read count_map: %s", strerror(errno));
  1856. return;
  1857. }
  1858. profile_total_count = 0;
  1859. for (cpu = 0; cpu < num_cpu; cpu++)
  1860. profile_total_count += counts[cpu];
  1861. for (m = 0; m < ARRAY_SIZE(metrics); m++) {
  1862. struct bpf_perf_event_value values[num_cpu];
  1863. if (!metrics[m].selected)
  1864. continue;
  1865. err = bpf_map_lookup_elem(reading_map_fd, &key, values);
  1866. if (err) {
  1867. p_err("failed to read reading_map: %s",
  1868. strerror(errno));
  1869. return;
  1870. }
  1871. for (cpu = 0; cpu < num_cpu; cpu++) {
  1872. metrics[m].val.counter += values[cpu].counter;
  1873. metrics[m].val.enabled += values[cpu].enabled;
  1874. metrics[m].val.running += values[cpu].running;
  1875. }
  1876. key++;
  1877. }
  1878. }
  1879. static void profile_print_readings_json(void)
  1880. {
  1881. __u32 m;
  1882. jsonw_start_array(json_wtr);
  1883. for (m = 0; m < ARRAY_SIZE(metrics); m++) {
  1884. if (!metrics[m].selected)
  1885. continue;
  1886. jsonw_start_object(json_wtr);
  1887. jsonw_string_field(json_wtr, "metric", metrics[m].name);
  1888. jsonw_lluint_field(json_wtr, "run_cnt", profile_total_count);
  1889. jsonw_lluint_field(json_wtr, "value", metrics[m].val.counter);
  1890. jsonw_lluint_field(json_wtr, "enabled", metrics[m].val.enabled);
  1891. jsonw_lluint_field(json_wtr, "running", metrics[m].val.running);
  1892. jsonw_end_object(json_wtr);
  1893. }
  1894. jsonw_end_array(json_wtr);
  1895. }
  1896. static void profile_print_readings_plain(void)
  1897. {
  1898. __u32 m;
  1899. printf("\n%18llu %-20s\n", profile_total_count, "run_cnt");
  1900. for (m = 0; m < ARRAY_SIZE(metrics); m++) {
  1901. struct bpf_perf_event_value *val = &metrics[m].val;
  1902. int r;
  1903. if (!metrics[m].selected)
  1904. continue;
  1905. printf("%18llu %-20s", val->counter, metrics[m].name);
  1906. r = metrics[m].ratio_metric - 1;
  1907. if (r >= 0 && metrics[r].selected &&
  1908. metrics[r].val.counter > 0) {
  1909. printf("# %8.2f %-30s",
  1910. val->counter * metrics[m].ratio_mul /
  1911. metrics[r].val.counter,
  1912. metrics[m].ratio_desc);
  1913. } else {
  1914. printf("%-41s", "");
  1915. }
  1916. if (val->enabled > val->running)
  1917. printf("(%4.2f%%)",
  1918. val->running * 100.0 / val->enabled);
  1919. printf("\n");
  1920. }
  1921. }
  1922. static void profile_print_readings(void)
  1923. {
  1924. if (json_output)
  1925. profile_print_readings_json();
  1926. else
  1927. profile_print_readings_plain();
  1928. }
  1929. static char *profile_target_name(int tgt_fd)
  1930. {
  1931. struct bpf_func_info func_info = {};
  1932. struct bpf_prog_info info = {};
  1933. __u32 info_len = sizeof(info);
  1934. const struct btf_type *t;
  1935. __u32 func_info_rec_size;
  1936. struct btf *btf = NULL;
  1937. char *name = NULL;
  1938. int err;
  1939. err = bpf_prog_get_info_by_fd(tgt_fd, &info, &info_len);
  1940. if (err) {
  1941. p_err("failed to get info for prog FD %d", tgt_fd);
  1942. goto out;
  1943. }
  1944. if (info.btf_id == 0) {
  1945. p_err("prog FD %d doesn't have valid btf", tgt_fd);
  1946. goto out;
  1947. }
  1948. func_info_rec_size = info.func_info_rec_size;
  1949. if (info.nr_func_info == 0) {
  1950. p_err("found 0 func_info for prog FD %d", tgt_fd);
  1951. goto out;
  1952. }
  1953. memset(&info, 0, sizeof(info));
  1954. info.nr_func_info = 1;
  1955. info.func_info_rec_size = func_info_rec_size;
  1956. info.func_info = ptr_to_u64(&func_info);
  1957. err = bpf_prog_get_info_by_fd(tgt_fd, &info, &info_len);
  1958. if (err) {
  1959. p_err("failed to get func_info for prog FD %d", tgt_fd);
  1960. goto out;
  1961. }
  1962. btf = btf__load_from_kernel_by_id(info.btf_id);
  1963. if (!btf) {
  1964. p_err("failed to load btf for prog FD %d", tgt_fd);
  1965. goto out;
  1966. }
  1967. t = btf__type_by_id(btf, func_info.type_id);
  1968. if (!t) {
  1969. p_err("btf %u doesn't have type %u",
  1970. info.btf_id, func_info.type_id);
  1971. goto out;
  1972. }
  1973. name = strdup(btf__name_by_offset(btf, t->name_off));
  1974. out:
  1975. btf__free(btf);
  1976. return name;
  1977. }
  1978. static struct profiler_bpf *profile_obj;
  1979. static int profile_tgt_fd = -1;
  1980. static char *profile_tgt_name;
  1981. static int *profile_perf_events;
  1982. static int profile_perf_event_cnt;
  1983. static void profile_close_perf_events(struct profiler_bpf *obj)
  1984. {
  1985. int i;
  1986. for (i = profile_perf_event_cnt - 1; i >= 0; i--)
  1987. close(profile_perf_events[i]);
  1988. free(profile_perf_events);
  1989. profile_perf_event_cnt = 0;
  1990. }
  1991. static int profile_open_perf_event(int mid, int cpu, int map_fd)
  1992. {
  1993. int pmu_fd;
  1994. pmu_fd = syscall(__NR_perf_event_open, &metrics[mid].attr,
  1995. -1 /*pid*/, cpu, -1 /*group_fd*/, 0);
  1996. if (pmu_fd < 0) {
  1997. if (errno == ENODEV) {
  1998. p_info("cpu %d may be offline, skip %s profiling.",
  1999. cpu, metrics[mid].name);
  2000. profile_perf_event_cnt++;
  2001. return 0;
  2002. }
  2003. return -1;
  2004. }
  2005. if (bpf_map_update_elem(map_fd,
  2006. &profile_perf_event_cnt,
  2007. &pmu_fd, BPF_ANY) ||
  2008. ioctl(pmu_fd, PERF_EVENT_IOC_ENABLE, 0)) {
  2009. close(pmu_fd);
  2010. return -1;
  2011. }
  2012. profile_perf_events[profile_perf_event_cnt++] = pmu_fd;
  2013. return 0;
  2014. }
  2015. static int profile_open_perf_events(struct profiler_bpf *obj)
  2016. {
  2017. unsigned int cpu, m;
  2018. int map_fd;
  2019. profile_perf_events = calloc(
  2020. obj->rodata->num_cpu * obj->rodata->num_metric, sizeof(int));
  2021. if (!profile_perf_events) {
  2022. p_err("failed to allocate memory for perf_event array: %s",
  2023. strerror(errno));
  2024. return -1;
  2025. }
  2026. map_fd = bpf_map__fd(obj->maps.events);
  2027. if (map_fd < 0) {
  2028. p_err("failed to get fd for events map");
  2029. return -1;
  2030. }
  2031. for (m = 0; m < ARRAY_SIZE(metrics); m++) {
  2032. if (!metrics[m].selected)
  2033. continue;
  2034. for (cpu = 0; cpu < obj->rodata->num_cpu; cpu++) {
  2035. if (profile_open_perf_event(m, cpu, map_fd)) {
  2036. p_err("failed to create event %s on cpu %u",
  2037. metrics[m].name, cpu);
  2038. return -1;
  2039. }
  2040. }
  2041. }
  2042. return 0;
  2043. }
  2044. static void profile_print_and_cleanup(void)
  2045. {
  2046. profile_close_perf_events(profile_obj);
  2047. profile_read_values(profile_obj);
  2048. profile_print_readings();
  2049. profiler_bpf__destroy(profile_obj);
  2050. close(profile_tgt_fd);
  2051. free(profile_tgt_name);
  2052. }
  2053. static void int_exit(int signo)
  2054. {
  2055. profile_print_and_cleanup();
  2056. exit(0);
  2057. }
  2058. static int do_profile(int argc, char **argv)
  2059. {
  2060. int num_metric, num_cpu, err = -1;
  2061. struct bpf_program *prog;
  2062. unsigned long duration;
  2063. char *endptr;
  2064. /* we at least need two args for the prog and one metric */
  2065. if (!REQ_ARGS(3))
  2066. return -EINVAL;
  2067. /* parse target fd */
  2068. profile_tgt_fd = prog_parse_fd(&argc, &argv);
  2069. if (profile_tgt_fd < 0) {
  2070. p_err("failed to parse fd");
  2071. return -1;
  2072. }
  2073. /* parse profiling optional duration */
  2074. if (argc > 2 && is_prefix(argv[0], "duration")) {
  2075. NEXT_ARG();
  2076. duration = strtoul(*argv, &endptr, 0);
  2077. if (*endptr)
  2078. usage();
  2079. NEXT_ARG();
  2080. } else {
  2081. duration = UINT_MAX;
  2082. }
  2083. num_metric = profile_parse_metrics(argc, argv);
  2084. if (num_metric <= 0)
  2085. goto out;
  2086. num_cpu = libbpf_num_possible_cpus();
  2087. if (num_cpu <= 0) {
  2088. p_err("failed to identify number of CPUs");
  2089. goto out;
  2090. }
  2091. profile_obj = profiler_bpf__open();
  2092. if (!profile_obj) {
  2093. p_err("failed to open and/or load BPF object");
  2094. goto out;
  2095. }
  2096. profile_obj->rodata->num_cpu = num_cpu;
  2097. profile_obj->rodata->num_metric = num_metric;
  2098. /* adjust map sizes */
  2099. bpf_map__set_max_entries(profile_obj->maps.events, num_metric * num_cpu);
  2100. bpf_map__set_max_entries(profile_obj->maps.fentry_readings, num_metric);
  2101. bpf_map__set_max_entries(profile_obj->maps.accum_readings, num_metric);
  2102. bpf_map__set_max_entries(profile_obj->maps.counts, 1);
  2103. /* change target name */
  2104. profile_tgt_name = profile_target_name(profile_tgt_fd);
  2105. if (!profile_tgt_name)
  2106. goto out;
  2107. bpf_object__for_each_program(prog, profile_obj->obj) {
  2108. err = bpf_program__set_attach_target(prog, profile_tgt_fd,
  2109. profile_tgt_name);
  2110. if (err) {
  2111. p_err("failed to set attach target\n");
  2112. goto out;
  2113. }
  2114. }
  2115. set_max_rlimit();
  2116. err = profiler_bpf__load(profile_obj);
  2117. if (err) {
  2118. p_err("failed to load profile_obj");
  2119. goto out;
  2120. }
  2121. err = profile_open_perf_events(profile_obj);
  2122. if (err)
  2123. goto out;
  2124. err = profiler_bpf__attach(profile_obj);
  2125. if (err) {
  2126. p_err("failed to attach profile_obj");
  2127. goto out;
  2128. }
  2129. signal(SIGINT, int_exit);
  2130. sleep(duration);
  2131. profile_print_and_cleanup();
  2132. return 0;
  2133. out:
  2134. profile_close_perf_events(profile_obj);
  2135. if (profile_obj)
  2136. profiler_bpf__destroy(profile_obj);
  2137. close(profile_tgt_fd);
  2138. free(profile_tgt_name);
  2139. return err;
  2140. }
  2141. #endif /* BPFTOOL_WITHOUT_SKELETONS */
  2142. static int do_help(int argc, char **argv)
  2143. {
  2144. if (json_output) {
  2145. jsonw_null(json_wtr);
  2146. return 0;
  2147. }
  2148. fprintf(stderr,
  2149. "Usage: %1$s %2$s { show | list } [PROG]\n"
  2150. " %1$s %2$s dump xlated PROG [{ file FILE | [opcodes] [linum] [visual] }]\n"
  2151. " %1$s %2$s dump jited PROG [{ file FILE | [opcodes] [linum] }]\n"
  2152. " %1$s %2$s pin PROG FILE\n"
  2153. " %1$s %2$s { load | loadall } OBJ PATH \\\n"
  2154. " [type TYPE] [{ offload_dev | xdpmeta_dev } NAME] \\\n"
  2155. " [map { idx IDX | name NAME } MAP]\\\n"
  2156. " [pinmaps MAP_DIR]\n"
  2157. " [autoattach]\n"
  2158. " [kernel_btf BTF_FILE]\n"
  2159. " %1$s %2$s attach PROG ATTACH_TYPE [MAP]\n"
  2160. " %1$s %2$s detach PROG ATTACH_TYPE [MAP]\n"
  2161. " %1$s %2$s run PROG \\\n"
  2162. " data_in FILE \\\n"
  2163. " [data_out FILE [data_size_out L]] \\\n"
  2164. " [ctx_in FILE [ctx_out FILE [ctx_size_out M]]] \\\n"
  2165. " [repeat N]\n"
  2166. " %1$s %2$s profile PROG [duration DURATION] METRICs\n"
  2167. " %1$s %2$s tracelog\n"
  2168. " %1$s %2$s tracelog { stdout | stderr } PROG\n"
  2169. " %1$s %2$s help\n"
  2170. "\n"
  2171. " " HELP_SPEC_MAP "\n"
  2172. " " HELP_SPEC_PROGRAM "\n"
  2173. " TYPE := { socket | kprobe | kretprobe | classifier | action |\n"
  2174. " tracepoint | raw_tracepoint | xdp | perf_event | cgroup/skb |\n"
  2175. " cgroup/sock | cgroup/dev | lwt_in | lwt_out | lwt_xmit |\n"
  2176. " lwt_seg6local | sockops | sk_skb | sk_msg | lirc_mode2 |\n"
  2177. " sk_reuseport | flow_dissector | cgroup/sysctl |\n"
  2178. " cgroup/bind4 | cgroup/bind6 | cgroup/post_bind4 |\n"
  2179. " cgroup/post_bind6 | cgroup/connect4 | cgroup/connect6 |\n"
  2180. " cgroup/connect_unix | cgroup/getpeername4 | cgroup/getpeername6 |\n"
  2181. " cgroup/getpeername_unix | cgroup/getsockname4 | cgroup/getsockname6 |\n"
  2182. " cgroup/getsockname_unix | cgroup/sendmsg4 | cgroup/sendmsg6 |\n"
  2183. " cgroup/sendmsg_unix | cgroup/recvmsg4 | cgroup/recvmsg6 | cgroup/recvmsg_unix |\n"
  2184. " cgroup/getsockopt | cgroup/setsockopt | cgroup/sock_release |\n"
  2185. " struct_ops | fentry | fexit | freplace | sk_lookup }\n"
  2186. " ATTACH_TYPE := { sk_msg_verdict | sk_skb_verdict | sk_skb_stream_verdict |\n"
  2187. " sk_skb_stream_parser | flow_dissector }\n"
  2188. " METRIC := { cycles | instructions | l1d_loads | llc_misses | itlb_misses | dtlb_misses }\n"
  2189. " " HELP_SPEC_OPTIONS " |\n"
  2190. " {-f|--bpffs} | {-m|--mapcompat} | {-n|--nomount} |\n"
  2191. " {-L|--use-loader} | [ {-S|--sign } {-k} <private_key.pem> {-i} <certificate.x509> ] \n"
  2192. "",
  2193. bin_name, argv[-2]);
  2194. return 0;
  2195. }
  2196. static const struct cmd cmds[] = {
  2197. { "show", do_show },
  2198. { "list", do_show },
  2199. { "help", do_help },
  2200. { "dump", do_dump },
  2201. { "pin", do_pin },
  2202. { "load", do_load },
  2203. { "loadall", do_loadall },
  2204. { "attach", do_attach },
  2205. { "detach", do_detach },
  2206. { "tracelog", do_tracelog_any },
  2207. { "run", do_run },
  2208. { "profile", do_profile },
  2209. { 0 }
  2210. };
  2211. int do_prog(int argc, char **argv)
  2212. {
  2213. return cmd_select(cmds, argc, argv, do_help);
  2214. }