bpf-loader.c 40 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787
  1. // SPDX-License-Identifier: GPL-2.0
  2. /*
  3. * bpf-loader.c
  4. *
  5. * Copyright (C) 2015 Wang Nan <wangnan0@huawei.com>
  6. * Copyright (C) 2015 Huawei Inc.
  7. */
  8. #include <linux/bpf.h>
  9. #include <bpf/libbpf.h>
  10. #include <bpf/bpf.h>
  11. #include <linux/err.h>
  12. #include <linux/kernel.h>
  13. #include <linux/string.h>
  14. #include <errno.h>
  15. #include "perf.h"
  16. #include "debug.h"
  17. #include "bpf-loader.h"
  18. #include "bpf-prologue.h"
  19. #include "probe-event.h"
  20. #include "probe-finder.h" // for MAX_PROBES
  21. #include "parse-events.h"
  22. #include "strfilter.h"
  23. #include "llvm-utils.h"
  24. #include "c++/clang-c.h"
  25. #define DEFINE_PRINT_FN(name, level) \
  26. static int libbpf_##name(const char *fmt, ...) \
  27. { \
  28. va_list args; \
  29. int ret; \
  30. \
  31. va_start(args, fmt); \
  32. ret = veprintf(level, verbose, pr_fmt(fmt), args);\
  33. va_end(args); \
  34. return ret; \
  35. }
  36. DEFINE_PRINT_FN(warning, 1)
  37. DEFINE_PRINT_FN(info, 1)
  38. DEFINE_PRINT_FN(debug, 1)
  39. struct bpf_prog_priv {
  40. bool is_tp;
  41. char *sys_name;
  42. char *evt_name;
  43. struct perf_probe_event pev;
  44. bool need_prologue;
  45. struct bpf_insn *insns_buf;
  46. int nr_types;
  47. int *type_mapping;
  48. };
  49. static bool libbpf_initialized;
  50. struct bpf_object *
  51. bpf__prepare_load_buffer(void *obj_buf, size_t obj_buf_sz, const char *name)
  52. {
  53. struct bpf_object *obj;
  54. if (!libbpf_initialized) {
  55. libbpf_set_print(libbpf_warning,
  56. libbpf_info,
  57. libbpf_debug);
  58. libbpf_initialized = true;
  59. }
  60. obj = bpf_object__open_buffer(obj_buf, obj_buf_sz, name);
  61. if (IS_ERR_OR_NULL(obj)) {
  62. pr_debug("bpf: failed to load buffer\n");
  63. return ERR_PTR(-EINVAL);
  64. }
  65. return obj;
  66. }
  67. struct bpf_object *bpf__prepare_load(const char *filename, bool source)
  68. {
  69. struct bpf_object *obj;
  70. if (!libbpf_initialized) {
  71. libbpf_set_print(libbpf_warning,
  72. libbpf_info,
  73. libbpf_debug);
  74. libbpf_initialized = true;
  75. }
  76. if (source) {
  77. int err;
  78. void *obj_buf;
  79. size_t obj_buf_sz;
  80. perf_clang__init();
  81. err = perf_clang__compile_bpf(filename, &obj_buf, &obj_buf_sz);
  82. perf_clang__cleanup();
  83. if (err) {
  84. pr_debug("bpf: builtin compilation failed: %d, try external compiler\n", err);
  85. err = llvm__compile_bpf(filename, &obj_buf, &obj_buf_sz);
  86. if (err)
  87. return ERR_PTR(-BPF_LOADER_ERRNO__COMPILE);
  88. } else
  89. pr_debug("bpf: successfull builtin compilation\n");
  90. obj = bpf_object__open_buffer(obj_buf, obj_buf_sz, filename);
  91. if (!IS_ERR_OR_NULL(obj) && llvm_param.dump_obj)
  92. llvm__dump_obj(filename, obj_buf, obj_buf_sz);
  93. free(obj_buf);
  94. } else
  95. obj = bpf_object__open(filename);
  96. if (IS_ERR_OR_NULL(obj)) {
  97. pr_debug("bpf: failed to load %s\n", filename);
  98. return obj;
  99. }
  100. return obj;
  101. }
  102. void bpf__clear(void)
  103. {
  104. struct bpf_object *obj, *tmp;
  105. bpf_object__for_each_safe(obj, tmp) {
  106. bpf__unprobe(obj);
  107. bpf_object__close(obj);
  108. }
  109. }
  110. static void
  111. clear_prog_priv(struct bpf_program *prog __maybe_unused,
  112. void *_priv)
  113. {
  114. struct bpf_prog_priv *priv = _priv;
  115. cleanup_perf_probe_events(&priv->pev, 1);
  116. zfree(&priv->insns_buf);
  117. zfree(&priv->type_mapping);
  118. zfree(&priv->sys_name);
  119. zfree(&priv->evt_name);
  120. free(priv);
  121. }
  122. static int
  123. prog_config__exec(const char *value, struct perf_probe_event *pev)
  124. {
  125. pev->uprobes = true;
  126. pev->target = strdup(value);
  127. if (!pev->target)
  128. return -ENOMEM;
  129. return 0;
  130. }
  131. static int
  132. prog_config__module(const char *value, struct perf_probe_event *pev)
  133. {
  134. pev->uprobes = false;
  135. pev->target = strdup(value);
  136. if (!pev->target)
  137. return -ENOMEM;
  138. return 0;
  139. }
  140. static int
  141. prog_config__bool(const char *value, bool *pbool, bool invert)
  142. {
  143. int err;
  144. bool bool_value;
  145. if (!pbool)
  146. return -EINVAL;
  147. err = strtobool(value, &bool_value);
  148. if (err)
  149. return err;
  150. *pbool = invert ? !bool_value : bool_value;
  151. return 0;
  152. }
  153. static int
  154. prog_config__inlines(const char *value,
  155. struct perf_probe_event *pev __maybe_unused)
  156. {
  157. return prog_config__bool(value, &probe_conf.no_inlines, true);
  158. }
  159. static int
  160. prog_config__force(const char *value,
  161. struct perf_probe_event *pev __maybe_unused)
  162. {
  163. return prog_config__bool(value, &probe_conf.force_add, false);
  164. }
  165. static struct {
  166. const char *key;
  167. const char *usage;
  168. const char *desc;
  169. int (*func)(const char *, struct perf_probe_event *);
  170. } bpf_prog_config_terms[] = {
  171. {
  172. .key = "exec",
  173. .usage = "exec=<full path of file>",
  174. .desc = "Set uprobe target",
  175. .func = prog_config__exec,
  176. },
  177. {
  178. .key = "module",
  179. .usage = "module=<module name> ",
  180. .desc = "Set kprobe module",
  181. .func = prog_config__module,
  182. },
  183. {
  184. .key = "inlines",
  185. .usage = "inlines=[yes|no] ",
  186. .desc = "Probe at inline symbol",
  187. .func = prog_config__inlines,
  188. },
  189. {
  190. .key = "force",
  191. .usage = "force=[yes|no] ",
  192. .desc = "Forcibly add events with existing name",
  193. .func = prog_config__force,
  194. },
  195. };
  196. static int
  197. do_prog_config(const char *key, const char *value,
  198. struct perf_probe_event *pev)
  199. {
  200. unsigned int i;
  201. pr_debug("config bpf program: %s=%s\n", key, value);
  202. for (i = 0; i < ARRAY_SIZE(bpf_prog_config_terms); i++)
  203. if (strcmp(key, bpf_prog_config_terms[i].key) == 0)
  204. return bpf_prog_config_terms[i].func(value, pev);
  205. pr_debug("BPF: ERROR: invalid program config option: %s=%s\n",
  206. key, value);
  207. pr_debug("\nHint: Valid options are:\n");
  208. for (i = 0; i < ARRAY_SIZE(bpf_prog_config_terms); i++)
  209. pr_debug("\t%s:\t%s\n", bpf_prog_config_terms[i].usage,
  210. bpf_prog_config_terms[i].desc);
  211. pr_debug("\n");
  212. return -BPF_LOADER_ERRNO__PROGCONF_TERM;
  213. }
  214. static const char *
  215. parse_prog_config_kvpair(const char *config_str, struct perf_probe_event *pev)
  216. {
  217. char *text = strdup(config_str);
  218. char *sep, *line;
  219. const char *main_str = NULL;
  220. int err = 0;
  221. if (!text) {
  222. pr_debug("Not enough memory: dup config_str failed\n");
  223. return ERR_PTR(-ENOMEM);
  224. }
  225. line = text;
  226. while ((sep = strchr(line, ';'))) {
  227. char *equ;
  228. *sep = '\0';
  229. equ = strchr(line, '=');
  230. if (!equ) {
  231. pr_warning("WARNING: invalid config in BPF object: %s\n",
  232. line);
  233. pr_warning("\tShould be 'key=value'.\n");
  234. goto nextline;
  235. }
  236. *equ = '\0';
  237. err = do_prog_config(line, equ + 1, pev);
  238. if (err)
  239. break;
  240. nextline:
  241. line = sep + 1;
  242. }
  243. if (!err)
  244. main_str = config_str + (line - text);
  245. free(text);
  246. return err ? ERR_PTR(err) : main_str;
  247. }
  248. static int
  249. parse_prog_config(const char *config_str, const char **p_main_str,
  250. bool *is_tp, struct perf_probe_event *pev)
  251. {
  252. int err;
  253. const char *main_str = parse_prog_config_kvpair(config_str, pev);
  254. if (IS_ERR(main_str))
  255. return PTR_ERR(main_str);
  256. *p_main_str = main_str;
  257. if (!strchr(main_str, '=')) {
  258. /* Is a tracepoint event? */
  259. const char *s = strchr(main_str, ':');
  260. if (!s) {
  261. pr_debug("bpf: '%s' is not a valid tracepoint\n",
  262. config_str);
  263. return -BPF_LOADER_ERRNO__CONFIG;
  264. }
  265. *is_tp = true;
  266. return 0;
  267. }
  268. *is_tp = false;
  269. err = parse_perf_probe_command(main_str, pev);
  270. if (err < 0) {
  271. pr_debug("bpf: '%s' is not a valid config string\n",
  272. config_str);
  273. /* parse failed, don't need clear pev. */
  274. return -BPF_LOADER_ERRNO__CONFIG;
  275. }
  276. return 0;
  277. }
  278. static int
  279. config_bpf_program(struct bpf_program *prog)
  280. {
  281. struct perf_probe_event *pev = NULL;
  282. struct bpf_prog_priv *priv = NULL;
  283. const char *config_str, *main_str;
  284. bool is_tp = false;
  285. int err;
  286. /* Initialize per-program probing setting */
  287. probe_conf.no_inlines = false;
  288. probe_conf.force_add = false;
  289. config_str = bpf_program__title(prog, false);
  290. if (IS_ERR(config_str)) {
  291. pr_debug("bpf: unable to get title for program\n");
  292. return PTR_ERR(config_str);
  293. }
  294. priv = calloc(sizeof(*priv), 1);
  295. if (!priv) {
  296. pr_debug("bpf: failed to alloc priv\n");
  297. return -ENOMEM;
  298. }
  299. pev = &priv->pev;
  300. pr_debug("bpf: config program '%s'\n", config_str);
  301. err = parse_prog_config(config_str, &main_str, &is_tp, pev);
  302. if (err)
  303. goto errout;
  304. if (is_tp) {
  305. char *s = strchr(main_str, ':');
  306. priv->is_tp = true;
  307. priv->sys_name = strndup(main_str, s - main_str);
  308. priv->evt_name = strdup(s + 1);
  309. goto set_priv;
  310. }
  311. if (pev->group && strcmp(pev->group, PERF_BPF_PROBE_GROUP)) {
  312. pr_debug("bpf: '%s': group for event is set and not '%s'.\n",
  313. config_str, PERF_BPF_PROBE_GROUP);
  314. err = -BPF_LOADER_ERRNO__GROUP;
  315. goto errout;
  316. } else if (!pev->group)
  317. pev->group = strdup(PERF_BPF_PROBE_GROUP);
  318. if (!pev->group) {
  319. pr_debug("bpf: strdup failed\n");
  320. err = -ENOMEM;
  321. goto errout;
  322. }
  323. if (!pev->event) {
  324. pr_debug("bpf: '%s': event name is missing. Section name should be 'key=value'\n",
  325. config_str);
  326. err = -BPF_LOADER_ERRNO__EVENTNAME;
  327. goto errout;
  328. }
  329. pr_debug("bpf: config '%s' is ok\n", config_str);
  330. set_priv:
  331. err = bpf_program__set_priv(prog, priv, clear_prog_priv);
  332. if (err) {
  333. pr_debug("Failed to set priv for program '%s'\n", config_str);
  334. goto errout;
  335. }
  336. return 0;
  337. errout:
  338. if (pev)
  339. clear_perf_probe_event(pev);
  340. free(priv);
  341. return err;
  342. }
  343. static int bpf__prepare_probe(void)
  344. {
  345. static int err = 0;
  346. static bool initialized = false;
  347. /*
  348. * Make err static, so if init failed the first, bpf__prepare_probe()
  349. * fails each time without calling init_probe_symbol_maps multiple
  350. * times.
  351. */
  352. if (initialized)
  353. return err;
  354. initialized = true;
  355. err = init_probe_symbol_maps(false);
  356. if (err < 0)
  357. pr_debug("Failed to init_probe_symbol_maps\n");
  358. probe_conf.max_probes = MAX_PROBES;
  359. return err;
  360. }
  361. static int
  362. preproc_gen_prologue(struct bpf_program *prog, int n,
  363. struct bpf_insn *orig_insns, int orig_insns_cnt,
  364. struct bpf_prog_prep_result *res)
  365. {
  366. struct bpf_prog_priv *priv = bpf_program__priv(prog);
  367. struct probe_trace_event *tev;
  368. struct perf_probe_event *pev;
  369. struct bpf_insn *buf;
  370. size_t prologue_cnt = 0;
  371. int i, err;
  372. if (IS_ERR(priv) || !priv || priv->is_tp)
  373. goto errout;
  374. pev = &priv->pev;
  375. if (n < 0 || n >= priv->nr_types)
  376. goto errout;
  377. /* Find a tev belongs to that type */
  378. for (i = 0; i < pev->ntevs; i++) {
  379. if (priv->type_mapping[i] == n)
  380. break;
  381. }
  382. if (i >= pev->ntevs) {
  383. pr_debug("Internal error: prologue type %d not found\n", n);
  384. return -BPF_LOADER_ERRNO__PROLOGUE;
  385. }
  386. tev = &pev->tevs[i];
  387. buf = priv->insns_buf;
  388. err = bpf__gen_prologue(tev->args, tev->nargs,
  389. buf, &prologue_cnt,
  390. BPF_MAXINSNS - orig_insns_cnt);
  391. if (err) {
  392. const char *title;
  393. title = bpf_program__title(prog, false);
  394. if (!title)
  395. title = "[unknown]";
  396. pr_debug("Failed to generate prologue for program %s\n",
  397. title);
  398. return err;
  399. }
  400. memcpy(&buf[prologue_cnt], orig_insns,
  401. sizeof(struct bpf_insn) * orig_insns_cnt);
  402. res->new_insn_ptr = buf;
  403. res->new_insn_cnt = prologue_cnt + orig_insns_cnt;
  404. res->pfd = NULL;
  405. return 0;
  406. errout:
  407. pr_debug("Internal error in preproc_gen_prologue\n");
  408. return -BPF_LOADER_ERRNO__PROLOGUE;
  409. }
  410. /*
  411. * compare_tev_args is reflexive, transitive and antisymmetric.
  412. * I can proof it but this margin is too narrow to contain.
  413. */
  414. static int compare_tev_args(const void *ptev1, const void *ptev2)
  415. {
  416. int i, ret;
  417. const struct probe_trace_event *tev1 =
  418. *(const struct probe_trace_event **)ptev1;
  419. const struct probe_trace_event *tev2 =
  420. *(const struct probe_trace_event **)ptev2;
  421. ret = tev2->nargs - tev1->nargs;
  422. if (ret)
  423. return ret;
  424. for (i = 0; i < tev1->nargs; i++) {
  425. struct probe_trace_arg *arg1, *arg2;
  426. struct probe_trace_arg_ref *ref1, *ref2;
  427. arg1 = &tev1->args[i];
  428. arg2 = &tev2->args[i];
  429. ret = strcmp(arg1->value, arg2->value);
  430. if (ret)
  431. return ret;
  432. ref1 = arg1->ref;
  433. ref2 = arg2->ref;
  434. while (ref1 && ref2) {
  435. ret = ref2->offset - ref1->offset;
  436. if (ret)
  437. return ret;
  438. ref1 = ref1->next;
  439. ref2 = ref2->next;
  440. }
  441. if (ref1 || ref2)
  442. return ref2 ? 1 : -1;
  443. }
  444. return 0;
  445. }
  446. /*
  447. * Assign a type number to each tevs in a pev.
  448. * mapping is an array with same slots as tevs in that pev.
  449. * nr_types will be set to number of types.
  450. */
  451. static int map_prologue(struct perf_probe_event *pev, int *mapping,
  452. int *nr_types)
  453. {
  454. int i, type = 0;
  455. struct probe_trace_event **ptevs;
  456. size_t array_sz = sizeof(*ptevs) * pev->ntevs;
  457. ptevs = malloc(array_sz);
  458. if (!ptevs) {
  459. pr_debug("Not enough memory: alloc ptevs failed\n");
  460. return -ENOMEM;
  461. }
  462. pr_debug("In map_prologue, ntevs=%d\n", pev->ntevs);
  463. for (i = 0; i < pev->ntevs; i++)
  464. ptevs[i] = &pev->tevs[i];
  465. qsort(ptevs, pev->ntevs, sizeof(*ptevs),
  466. compare_tev_args);
  467. for (i = 0; i < pev->ntevs; i++) {
  468. int n;
  469. n = ptevs[i] - pev->tevs;
  470. if (i == 0) {
  471. mapping[n] = type;
  472. pr_debug("mapping[%d]=%d\n", n, type);
  473. continue;
  474. }
  475. if (compare_tev_args(ptevs + i, ptevs + i - 1) == 0)
  476. mapping[n] = type;
  477. else
  478. mapping[n] = ++type;
  479. pr_debug("mapping[%d]=%d\n", n, mapping[n]);
  480. }
  481. free(ptevs);
  482. *nr_types = type + 1;
  483. return 0;
  484. }
  485. static int hook_load_preprocessor(struct bpf_program *prog)
  486. {
  487. struct bpf_prog_priv *priv = bpf_program__priv(prog);
  488. struct perf_probe_event *pev;
  489. bool need_prologue = false;
  490. int err, i;
  491. if (IS_ERR(priv) || !priv) {
  492. pr_debug("Internal error when hook preprocessor\n");
  493. return -BPF_LOADER_ERRNO__INTERNAL;
  494. }
  495. if (priv->is_tp) {
  496. priv->need_prologue = false;
  497. return 0;
  498. }
  499. pev = &priv->pev;
  500. for (i = 0; i < pev->ntevs; i++) {
  501. struct probe_trace_event *tev = &pev->tevs[i];
  502. if (tev->nargs > 0) {
  503. need_prologue = true;
  504. break;
  505. }
  506. }
  507. /*
  508. * Since all tevs don't have argument, we don't need generate
  509. * prologue.
  510. */
  511. if (!need_prologue) {
  512. priv->need_prologue = false;
  513. return 0;
  514. }
  515. priv->need_prologue = true;
  516. priv->insns_buf = malloc(sizeof(struct bpf_insn) * BPF_MAXINSNS);
  517. if (!priv->insns_buf) {
  518. pr_debug("Not enough memory: alloc insns_buf failed\n");
  519. return -ENOMEM;
  520. }
  521. priv->type_mapping = malloc(sizeof(int) * pev->ntevs);
  522. if (!priv->type_mapping) {
  523. pr_debug("Not enough memory: alloc type_mapping failed\n");
  524. return -ENOMEM;
  525. }
  526. memset(priv->type_mapping, -1,
  527. sizeof(int) * pev->ntevs);
  528. err = map_prologue(pev, priv->type_mapping, &priv->nr_types);
  529. if (err)
  530. return err;
  531. err = bpf_program__set_prep(prog, priv->nr_types,
  532. preproc_gen_prologue);
  533. return err;
  534. }
  535. int bpf__probe(struct bpf_object *obj)
  536. {
  537. int err = 0;
  538. struct bpf_program *prog;
  539. struct bpf_prog_priv *priv;
  540. struct perf_probe_event *pev;
  541. err = bpf__prepare_probe();
  542. if (err) {
  543. pr_debug("bpf__prepare_probe failed\n");
  544. return err;
  545. }
  546. bpf_object__for_each_program(prog, obj) {
  547. err = config_bpf_program(prog);
  548. if (err)
  549. goto out;
  550. priv = bpf_program__priv(prog);
  551. if (IS_ERR(priv) || !priv) {
  552. err = PTR_ERR(priv);
  553. goto out;
  554. }
  555. if (priv->is_tp) {
  556. bpf_program__set_tracepoint(prog);
  557. continue;
  558. }
  559. bpf_program__set_kprobe(prog);
  560. pev = &priv->pev;
  561. err = convert_perf_probe_events(pev, 1);
  562. if (err < 0) {
  563. pr_debug("bpf_probe: failed to convert perf probe events\n");
  564. goto out;
  565. }
  566. err = apply_perf_probe_events(pev, 1);
  567. if (err < 0) {
  568. pr_debug("bpf_probe: failed to apply perf probe events\n");
  569. goto out;
  570. }
  571. /*
  572. * After probing, let's consider prologue, which
  573. * adds program fetcher to BPF programs.
  574. *
  575. * hook_load_preprocessorr() hooks pre-processor
  576. * to bpf_program, let it generate prologue
  577. * dynamically during loading.
  578. */
  579. err = hook_load_preprocessor(prog);
  580. if (err)
  581. goto out;
  582. }
  583. out:
  584. return err < 0 ? err : 0;
  585. }
  586. #define EVENTS_WRITE_BUFSIZE 4096
  587. int bpf__unprobe(struct bpf_object *obj)
  588. {
  589. int err, ret = 0;
  590. struct bpf_program *prog;
  591. bpf_object__for_each_program(prog, obj) {
  592. struct bpf_prog_priv *priv = bpf_program__priv(prog);
  593. int i;
  594. if (IS_ERR(priv) || !priv || priv->is_tp)
  595. continue;
  596. for (i = 0; i < priv->pev.ntevs; i++) {
  597. struct probe_trace_event *tev = &priv->pev.tevs[i];
  598. char name_buf[EVENTS_WRITE_BUFSIZE];
  599. struct strfilter *delfilter;
  600. snprintf(name_buf, EVENTS_WRITE_BUFSIZE,
  601. "%s:%s", tev->group, tev->event);
  602. name_buf[EVENTS_WRITE_BUFSIZE - 1] = '\0';
  603. delfilter = strfilter__new(name_buf, NULL);
  604. if (!delfilter) {
  605. pr_debug("Failed to create filter for unprobing\n");
  606. ret = -ENOMEM;
  607. continue;
  608. }
  609. err = del_perf_probe_events(delfilter);
  610. strfilter__delete(delfilter);
  611. if (err) {
  612. pr_debug("Failed to delete %s\n", name_buf);
  613. ret = err;
  614. continue;
  615. }
  616. }
  617. }
  618. return ret;
  619. }
  620. int bpf__load(struct bpf_object *obj)
  621. {
  622. int err;
  623. err = bpf_object__load(obj);
  624. if (err) {
  625. pr_debug("bpf: load objects failed\n");
  626. return err;
  627. }
  628. return 0;
  629. }
  630. int bpf__foreach_event(struct bpf_object *obj,
  631. bpf_prog_iter_callback_t func,
  632. void *arg)
  633. {
  634. struct bpf_program *prog;
  635. int err;
  636. bpf_object__for_each_program(prog, obj) {
  637. struct bpf_prog_priv *priv = bpf_program__priv(prog);
  638. struct probe_trace_event *tev;
  639. struct perf_probe_event *pev;
  640. int i, fd;
  641. if (IS_ERR(priv) || !priv) {
  642. pr_debug("bpf: failed to get private field\n");
  643. return -BPF_LOADER_ERRNO__INTERNAL;
  644. }
  645. if (priv->is_tp) {
  646. fd = bpf_program__fd(prog);
  647. err = (*func)(priv->sys_name, priv->evt_name, fd, arg);
  648. if (err) {
  649. pr_debug("bpf: tracepoint call back failed, stop iterate\n");
  650. return err;
  651. }
  652. continue;
  653. }
  654. pev = &priv->pev;
  655. for (i = 0; i < pev->ntevs; i++) {
  656. tev = &pev->tevs[i];
  657. if (priv->need_prologue) {
  658. int type = priv->type_mapping[i];
  659. fd = bpf_program__nth_fd(prog, type);
  660. } else {
  661. fd = bpf_program__fd(prog);
  662. }
  663. if (fd < 0) {
  664. pr_debug("bpf: failed to get file descriptor\n");
  665. return fd;
  666. }
  667. err = (*func)(tev->group, tev->event, fd, arg);
  668. if (err) {
  669. pr_debug("bpf: call back failed, stop iterate\n");
  670. return err;
  671. }
  672. }
  673. }
  674. return 0;
  675. }
  676. enum bpf_map_op_type {
  677. BPF_MAP_OP_SET_VALUE,
  678. BPF_MAP_OP_SET_EVSEL,
  679. };
  680. enum bpf_map_key_type {
  681. BPF_MAP_KEY_ALL,
  682. BPF_MAP_KEY_RANGES,
  683. };
  684. struct bpf_map_op {
  685. struct list_head list;
  686. enum bpf_map_op_type op_type;
  687. enum bpf_map_key_type key_type;
  688. union {
  689. struct parse_events_array array;
  690. } k;
  691. union {
  692. u64 value;
  693. struct perf_evsel *evsel;
  694. } v;
  695. };
  696. struct bpf_map_priv {
  697. struct list_head ops_list;
  698. };
  699. static void
  700. bpf_map_op__delete(struct bpf_map_op *op)
  701. {
  702. if (!list_empty(&op->list))
  703. list_del(&op->list);
  704. if (op->key_type == BPF_MAP_KEY_RANGES)
  705. parse_events__clear_array(&op->k.array);
  706. free(op);
  707. }
  708. static void
  709. bpf_map_priv__purge(struct bpf_map_priv *priv)
  710. {
  711. struct bpf_map_op *pos, *n;
  712. list_for_each_entry_safe(pos, n, &priv->ops_list, list) {
  713. list_del_init(&pos->list);
  714. bpf_map_op__delete(pos);
  715. }
  716. }
  717. static void
  718. bpf_map_priv__clear(struct bpf_map *map __maybe_unused,
  719. void *_priv)
  720. {
  721. struct bpf_map_priv *priv = _priv;
  722. bpf_map_priv__purge(priv);
  723. free(priv);
  724. }
  725. static int
  726. bpf_map_op_setkey(struct bpf_map_op *op, struct parse_events_term *term)
  727. {
  728. op->key_type = BPF_MAP_KEY_ALL;
  729. if (!term)
  730. return 0;
  731. if (term->array.nr_ranges) {
  732. size_t memsz = term->array.nr_ranges *
  733. sizeof(op->k.array.ranges[0]);
  734. op->k.array.ranges = memdup(term->array.ranges, memsz);
  735. if (!op->k.array.ranges) {
  736. pr_debug("Not enough memory to alloc indices for map\n");
  737. return -ENOMEM;
  738. }
  739. op->key_type = BPF_MAP_KEY_RANGES;
  740. op->k.array.nr_ranges = term->array.nr_ranges;
  741. }
  742. return 0;
  743. }
  744. static struct bpf_map_op *
  745. bpf_map_op__new(struct parse_events_term *term)
  746. {
  747. struct bpf_map_op *op;
  748. int err;
  749. op = zalloc(sizeof(*op));
  750. if (!op) {
  751. pr_debug("Failed to alloc bpf_map_op\n");
  752. return ERR_PTR(-ENOMEM);
  753. }
  754. INIT_LIST_HEAD(&op->list);
  755. err = bpf_map_op_setkey(op, term);
  756. if (err) {
  757. free(op);
  758. return ERR_PTR(err);
  759. }
  760. return op;
  761. }
  762. static struct bpf_map_op *
  763. bpf_map_op__clone(struct bpf_map_op *op)
  764. {
  765. struct bpf_map_op *newop;
  766. newop = memdup(op, sizeof(*op));
  767. if (!newop) {
  768. pr_debug("Failed to alloc bpf_map_op\n");
  769. return NULL;
  770. }
  771. INIT_LIST_HEAD(&newop->list);
  772. if (op->key_type == BPF_MAP_KEY_RANGES) {
  773. size_t memsz = op->k.array.nr_ranges *
  774. sizeof(op->k.array.ranges[0]);
  775. newop->k.array.ranges = memdup(op->k.array.ranges, memsz);
  776. if (!newop->k.array.ranges) {
  777. pr_debug("Failed to alloc indices for map\n");
  778. free(newop);
  779. return NULL;
  780. }
  781. }
  782. return newop;
  783. }
  784. static struct bpf_map_priv *
  785. bpf_map_priv__clone(struct bpf_map_priv *priv)
  786. {
  787. struct bpf_map_priv *newpriv;
  788. struct bpf_map_op *pos, *newop;
  789. newpriv = zalloc(sizeof(*newpriv));
  790. if (!newpriv) {
  791. pr_debug("Not enough memory to alloc map private\n");
  792. return NULL;
  793. }
  794. INIT_LIST_HEAD(&newpriv->ops_list);
  795. list_for_each_entry(pos, &priv->ops_list, list) {
  796. newop = bpf_map_op__clone(pos);
  797. if (!newop) {
  798. bpf_map_priv__purge(newpriv);
  799. return NULL;
  800. }
  801. list_add_tail(&newop->list, &newpriv->ops_list);
  802. }
  803. return newpriv;
  804. }
  805. static int
  806. bpf_map__add_op(struct bpf_map *map, struct bpf_map_op *op)
  807. {
  808. const char *map_name = bpf_map__name(map);
  809. struct bpf_map_priv *priv = bpf_map__priv(map);
  810. if (IS_ERR(priv)) {
  811. pr_debug("Failed to get private from map %s\n", map_name);
  812. return PTR_ERR(priv);
  813. }
  814. if (!priv) {
  815. priv = zalloc(sizeof(*priv));
  816. if (!priv) {
  817. pr_debug("Not enough memory to alloc map private\n");
  818. return -ENOMEM;
  819. }
  820. INIT_LIST_HEAD(&priv->ops_list);
  821. if (bpf_map__set_priv(map, priv, bpf_map_priv__clear)) {
  822. free(priv);
  823. return -BPF_LOADER_ERRNO__INTERNAL;
  824. }
  825. }
  826. list_add_tail(&op->list, &priv->ops_list);
  827. return 0;
  828. }
  829. static struct bpf_map_op *
  830. bpf_map__add_newop(struct bpf_map *map, struct parse_events_term *term)
  831. {
  832. struct bpf_map_op *op;
  833. int err;
  834. op = bpf_map_op__new(term);
  835. if (IS_ERR(op))
  836. return op;
  837. err = bpf_map__add_op(map, op);
  838. if (err) {
  839. bpf_map_op__delete(op);
  840. return ERR_PTR(err);
  841. }
  842. return op;
  843. }
  844. static int
  845. __bpf_map__config_value(struct bpf_map *map,
  846. struct parse_events_term *term)
  847. {
  848. struct bpf_map_op *op;
  849. const char *map_name = bpf_map__name(map);
  850. const struct bpf_map_def *def = bpf_map__def(map);
  851. if (IS_ERR(def)) {
  852. pr_debug("Unable to get map definition from '%s'\n",
  853. map_name);
  854. return -BPF_LOADER_ERRNO__INTERNAL;
  855. }
  856. if (def->type != BPF_MAP_TYPE_ARRAY) {
  857. pr_debug("Map %s type is not BPF_MAP_TYPE_ARRAY\n",
  858. map_name);
  859. return -BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE;
  860. }
  861. if (def->key_size < sizeof(unsigned int)) {
  862. pr_debug("Map %s has incorrect key size\n", map_name);
  863. return -BPF_LOADER_ERRNO__OBJCONF_MAP_KEYSIZE;
  864. }
  865. switch (def->value_size) {
  866. case 1:
  867. case 2:
  868. case 4:
  869. case 8:
  870. break;
  871. default:
  872. pr_debug("Map %s has incorrect value size\n", map_name);
  873. return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUESIZE;
  874. }
  875. op = bpf_map__add_newop(map, term);
  876. if (IS_ERR(op))
  877. return PTR_ERR(op);
  878. op->op_type = BPF_MAP_OP_SET_VALUE;
  879. op->v.value = term->val.num;
  880. return 0;
  881. }
  882. static int
  883. bpf_map__config_value(struct bpf_map *map,
  884. struct parse_events_term *term,
  885. struct perf_evlist *evlist __maybe_unused)
  886. {
  887. if (!term->err_val) {
  888. pr_debug("Config value not set\n");
  889. return -BPF_LOADER_ERRNO__OBJCONF_CONF;
  890. }
  891. if (term->type_val != PARSE_EVENTS__TERM_TYPE_NUM) {
  892. pr_debug("ERROR: wrong value type for 'value'\n");
  893. return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUE;
  894. }
  895. return __bpf_map__config_value(map, term);
  896. }
  897. static int
  898. __bpf_map__config_event(struct bpf_map *map,
  899. struct parse_events_term *term,
  900. struct perf_evlist *evlist)
  901. {
  902. struct perf_evsel *evsel;
  903. const struct bpf_map_def *def;
  904. struct bpf_map_op *op;
  905. const char *map_name = bpf_map__name(map);
  906. evsel = perf_evlist__find_evsel_by_str(evlist, term->val.str);
  907. if (!evsel) {
  908. pr_debug("Event (for '%s') '%s' doesn't exist\n",
  909. map_name, term->val.str);
  910. return -BPF_LOADER_ERRNO__OBJCONF_MAP_NOEVT;
  911. }
  912. def = bpf_map__def(map);
  913. if (IS_ERR(def)) {
  914. pr_debug("Unable to get map definition from '%s'\n",
  915. map_name);
  916. return PTR_ERR(def);
  917. }
  918. /*
  919. * No need to check key_size and value_size:
  920. * kernel has already checked them.
  921. */
  922. if (def->type != BPF_MAP_TYPE_PERF_EVENT_ARRAY) {
  923. pr_debug("Map %s type is not BPF_MAP_TYPE_PERF_EVENT_ARRAY\n",
  924. map_name);
  925. return -BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE;
  926. }
  927. op = bpf_map__add_newop(map, term);
  928. if (IS_ERR(op))
  929. return PTR_ERR(op);
  930. op->op_type = BPF_MAP_OP_SET_EVSEL;
  931. op->v.evsel = evsel;
  932. return 0;
  933. }
  934. static int
  935. bpf_map__config_event(struct bpf_map *map,
  936. struct parse_events_term *term,
  937. struct perf_evlist *evlist)
  938. {
  939. if (!term->err_val) {
  940. pr_debug("Config value not set\n");
  941. return -BPF_LOADER_ERRNO__OBJCONF_CONF;
  942. }
  943. if (term->type_val != PARSE_EVENTS__TERM_TYPE_STR) {
  944. pr_debug("ERROR: wrong value type for 'event'\n");
  945. return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUE;
  946. }
  947. return __bpf_map__config_event(map, term, evlist);
  948. }
  949. struct bpf_obj_config__map_func {
  950. const char *config_opt;
  951. int (*config_func)(struct bpf_map *, struct parse_events_term *,
  952. struct perf_evlist *);
  953. };
  954. struct bpf_obj_config__map_func bpf_obj_config__map_funcs[] = {
  955. {"value", bpf_map__config_value},
  956. {"event", bpf_map__config_event},
  957. };
  958. static int
  959. config_map_indices_range_check(struct parse_events_term *term,
  960. struct bpf_map *map,
  961. const char *map_name)
  962. {
  963. struct parse_events_array *array = &term->array;
  964. const struct bpf_map_def *def;
  965. unsigned int i;
  966. if (!array->nr_ranges)
  967. return 0;
  968. if (!array->ranges) {
  969. pr_debug("ERROR: map %s: array->nr_ranges is %d but range array is NULL\n",
  970. map_name, (int)array->nr_ranges);
  971. return -BPF_LOADER_ERRNO__INTERNAL;
  972. }
  973. def = bpf_map__def(map);
  974. if (IS_ERR(def)) {
  975. pr_debug("ERROR: Unable to get map definition from '%s'\n",
  976. map_name);
  977. return -BPF_LOADER_ERRNO__INTERNAL;
  978. }
  979. for (i = 0; i < array->nr_ranges; i++) {
  980. unsigned int start = array->ranges[i].start;
  981. size_t length = array->ranges[i].length;
  982. unsigned int idx = start + length - 1;
  983. if (idx >= def->max_entries) {
  984. pr_debug("ERROR: index %d too large\n", idx);
  985. return -BPF_LOADER_ERRNO__OBJCONF_MAP_IDX2BIG;
  986. }
  987. }
  988. return 0;
  989. }
  990. static int
  991. bpf__obj_config_map(struct bpf_object *obj,
  992. struct parse_events_term *term,
  993. struct perf_evlist *evlist,
  994. int *key_scan_pos)
  995. {
  996. /* key is "map:<mapname>.<config opt>" */
  997. char *map_name = strdup(term->config + sizeof("map:") - 1);
  998. struct bpf_map *map;
  999. int err = -BPF_LOADER_ERRNO__OBJCONF_OPT;
  1000. char *map_opt;
  1001. size_t i;
  1002. if (!map_name)
  1003. return -ENOMEM;
  1004. map_opt = strchr(map_name, '.');
  1005. if (!map_opt) {
  1006. pr_debug("ERROR: Invalid map config: %s\n", map_name);
  1007. goto out;
  1008. }
  1009. *map_opt++ = '\0';
  1010. if (*map_opt == '\0') {
  1011. pr_debug("ERROR: Invalid map option: %s\n", term->config);
  1012. goto out;
  1013. }
  1014. map = bpf_object__find_map_by_name(obj, map_name);
  1015. if (!map) {
  1016. pr_debug("ERROR: Map %s doesn't exist\n", map_name);
  1017. err = -BPF_LOADER_ERRNO__OBJCONF_MAP_NOTEXIST;
  1018. goto out;
  1019. }
  1020. *key_scan_pos += strlen(map_opt);
  1021. err = config_map_indices_range_check(term, map, map_name);
  1022. if (err)
  1023. goto out;
  1024. *key_scan_pos -= strlen(map_opt);
  1025. for (i = 0; i < ARRAY_SIZE(bpf_obj_config__map_funcs); i++) {
  1026. struct bpf_obj_config__map_func *func =
  1027. &bpf_obj_config__map_funcs[i];
  1028. if (strcmp(map_opt, func->config_opt) == 0) {
  1029. err = func->config_func(map, term, evlist);
  1030. goto out;
  1031. }
  1032. }
  1033. pr_debug("ERROR: Invalid map config option '%s'\n", map_opt);
  1034. err = -BPF_LOADER_ERRNO__OBJCONF_MAP_OPT;
  1035. out:
  1036. free(map_name);
  1037. if (!err)
  1038. key_scan_pos += strlen(map_opt);
  1039. return err;
  1040. }
  1041. int bpf__config_obj(struct bpf_object *obj,
  1042. struct parse_events_term *term,
  1043. struct perf_evlist *evlist,
  1044. int *error_pos)
  1045. {
  1046. int key_scan_pos = 0;
  1047. int err;
  1048. if (!obj || !term || !term->config)
  1049. return -EINVAL;
  1050. if (strstarts(term->config, "map:")) {
  1051. key_scan_pos = sizeof("map:") - 1;
  1052. err = bpf__obj_config_map(obj, term, evlist, &key_scan_pos);
  1053. goto out;
  1054. }
  1055. err = -BPF_LOADER_ERRNO__OBJCONF_OPT;
  1056. out:
  1057. if (error_pos)
  1058. *error_pos = key_scan_pos;
  1059. return err;
  1060. }
  1061. typedef int (*map_config_func_t)(const char *name, int map_fd,
  1062. const struct bpf_map_def *pdef,
  1063. struct bpf_map_op *op,
  1064. void *pkey, void *arg);
  1065. static int
  1066. foreach_key_array_all(map_config_func_t func,
  1067. void *arg, const char *name,
  1068. int map_fd, const struct bpf_map_def *pdef,
  1069. struct bpf_map_op *op)
  1070. {
  1071. unsigned int i;
  1072. int err;
  1073. for (i = 0; i < pdef->max_entries; i++) {
  1074. err = func(name, map_fd, pdef, op, &i, arg);
  1075. if (err) {
  1076. pr_debug("ERROR: failed to insert value to %s[%u]\n",
  1077. name, i);
  1078. return err;
  1079. }
  1080. }
  1081. return 0;
  1082. }
  1083. static int
  1084. foreach_key_array_ranges(map_config_func_t func, void *arg,
  1085. const char *name, int map_fd,
  1086. const struct bpf_map_def *pdef,
  1087. struct bpf_map_op *op)
  1088. {
  1089. unsigned int i, j;
  1090. int err;
  1091. for (i = 0; i < op->k.array.nr_ranges; i++) {
  1092. unsigned int start = op->k.array.ranges[i].start;
  1093. size_t length = op->k.array.ranges[i].length;
  1094. for (j = 0; j < length; j++) {
  1095. unsigned int idx = start + j;
  1096. err = func(name, map_fd, pdef, op, &idx, arg);
  1097. if (err) {
  1098. pr_debug("ERROR: failed to insert value to %s[%u]\n",
  1099. name, idx);
  1100. return err;
  1101. }
  1102. }
  1103. }
  1104. return 0;
  1105. }
  1106. static int
  1107. bpf_map_config_foreach_key(struct bpf_map *map,
  1108. map_config_func_t func,
  1109. void *arg)
  1110. {
  1111. int err, map_fd;
  1112. struct bpf_map_op *op;
  1113. const struct bpf_map_def *def;
  1114. const char *name = bpf_map__name(map);
  1115. struct bpf_map_priv *priv = bpf_map__priv(map);
  1116. if (IS_ERR(priv)) {
  1117. pr_debug("ERROR: failed to get private from map %s\n", name);
  1118. return -BPF_LOADER_ERRNO__INTERNAL;
  1119. }
  1120. if (!priv || list_empty(&priv->ops_list)) {
  1121. pr_debug("INFO: nothing to config for map %s\n", name);
  1122. return 0;
  1123. }
  1124. def = bpf_map__def(map);
  1125. if (IS_ERR(def)) {
  1126. pr_debug("ERROR: failed to get definition from map %s\n", name);
  1127. return -BPF_LOADER_ERRNO__INTERNAL;
  1128. }
  1129. map_fd = bpf_map__fd(map);
  1130. if (map_fd < 0) {
  1131. pr_debug("ERROR: failed to get fd from map %s\n", name);
  1132. return map_fd;
  1133. }
  1134. list_for_each_entry(op, &priv->ops_list, list) {
  1135. switch (def->type) {
  1136. case BPF_MAP_TYPE_ARRAY:
  1137. case BPF_MAP_TYPE_PERF_EVENT_ARRAY:
  1138. switch (op->key_type) {
  1139. case BPF_MAP_KEY_ALL:
  1140. err = foreach_key_array_all(func, arg, name,
  1141. map_fd, def, op);
  1142. break;
  1143. case BPF_MAP_KEY_RANGES:
  1144. err = foreach_key_array_ranges(func, arg, name,
  1145. map_fd, def,
  1146. op);
  1147. break;
  1148. default:
  1149. pr_debug("ERROR: keytype for map '%s' invalid\n",
  1150. name);
  1151. return -BPF_LOADER_ERRNO__INTERNAL;
  1152. }
  1153. if (err)
  1154. return err;
  1155. break;
  1156. default:
  1157. pr_debug("ERROR: type of '%s' incorrect\n", name);
  1158. return -BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE;
  1159. }
  1160. }
  1161. return 0;
  1162. }
  1163. static int
  1164. apply_config_value_for_key(int map_fd, void *pkey,
  1165. size_t val_size, u64 val)
  1166. {
  1167. int err = 0;
  1168. switch (val_size) {
  1169. case 1: {
  1170. u8 _val = (u8)(val);
  1171. err = bpf_map_update_elem(map_fd, pkey, &_val, BPF_ANY);
  1172. break;
  1173. }
  1174. case 2: {
  1175. u16 _val = (u16)(val);
  1176. err = bpf_map_update_elem(map_fd, pkey, &_val, BPF_ANY);
  1177. break;
  1178. }
  1179. case 4: {
  1180. u32 _val = (u32)(val);
  1181. err = bpf_map_update_elem(map_fd, pkey, &_val, BPF_ANY);
  1182. break;
  1183. }
  1184. case 8: {
  1185. err = bpf_map_update_elem(map_fd, pkey, &val, BPF_ANY);
  1186. break;
  1187. }
  1188. default:
  1189. pr_debug("ERROR: invalid value size\n");
  1190. return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUESIZE;
  1191. }
  1192. if (err && errno)
  1193. err = -errno;
  1194. return err;
  1195. }
  1196. static int
  1197. apply_config_evsel_for_key(const char *name, int map_fd, void *pkey,
  1198. struct perf_evsel *evsel)
  1199. {
  1200. struct xyarray *xy = evsel->fd;
  1201. struct perf_event_attr *attr;
  1202. unsigned int key, events;
  1203. bool check_pass = false;
  1204. int *evt_fd;
  1205. int err;
  1206. if (!xy) {
  1207. pr_debug("ERROR: evsel not ready for map %s\n", name);
  1208. return -BPF_LOADER_ERRNO__INTERNAL;
  1209. }
  1210. if (xy->row_size / xy->entry_size != 1) {
  1211. pr_debug("ERROR: Dimension of target event is incorrect for map %s\n",
  1212. name);
  1213. return -BPF_LOADER_ERRNO__OBJCONF_MAP_EVTDIM;
  1214. }
  1215. attr = &evsel->attr;
  1216. if (attr->inherit) {
  1217. pr_debug("ERROR: Can't put inherit event into map %s\n", name);
  1218. return -BPF_LOADER_ERRNO__OBJCONF_MAP_EVTINH;
  1219. }
  1220. if (perf_evsel__is_bpf_output(evsel))
  1221. check_pass = true;
  1222. if (attr->type == PERF_TYPE_RAW)
  1223. check_pass = true;
  1224. if (attr->type == PERF_TYPE_HARDWARE)
  1225. check_pass = true;
  1226. if (!check_pass) {
  1227. pr_debug("ERROR: Event type is wrong for map %s\n", name);
  1228. return -BPF_LOADER_ERRNO__OBJCONF_MAP_EVTTYPE;
  1229. }
  1230. events = xy->entries / (xy->row_size / xy->entry_size);
  1231. key = *((unsigned int *)pkey);
  1232. if (key >= events) {
  1233. pr_debug("ERROR: there is no event %d for map %s\n",
  1234. key, name);
  1235. return -BPF_LOADER_ERRNO__OBJCONF_MAP_MAPSIZE;
  1236. }
  1237. evt_fd = xyarray__entry(xy, key, 0);
  1238. err = bpf_map_update_elem(map_fd, pkey, evt_fd, BPF_ANY);
  1239. if (err && errno)
  1240. err = -errno;
  1241. return err;
  1242. }
  1243. static int
  1244. apply_obj_config_map_for_key(const char *name, int map_fd,
  1245. const struct bpf_map_def *pdef,
  1246. struct bpf_map_op *op,
  1247. void *pkey, void *arg __maybe_unused)
  1248. {
  1249. int err;
  1250. switch (op->op_type) {
  1251. case BPF_MAP_OP_SET_VALUE:
  1252. err = apply_config_value_for_key(map_fd, pkey,
  1253. pdef->value_size,
  1254. op->v.value);
  1255. break;
  1256. case BPF_MAP_OP_SET_EVSEL:
  1257. err = apply_config_evsel_for_key(name, map_fd, pkey,
  1258. op->v.evsel);
  1259. break;
  1260. default:
  1261. pr_debug("ERROR: unknown value type for '%s'\n", name);
  1262. err = -BPF_LOADER_ERRNO__INTERNAL;
  1263. }
  1264. return err;
  1265. }
  1266. static int
  1267. apply_obj_config_map(struct bpf_map *map)
  1268. {
  1269. return bpf_map_config_foreach_key(map,
  1270. apply_obj_config_map_for_key,
  1271. NULL);
  1272. }
  1273. static int
  1274. apply_obj_config_object(struct bpf_object *obj)
  1275. {
  1276. struct bpf_map *map;
  1277. int err;
  1278. bpf_map__for_each(map, obj) {
  1279. err = apply_obj_config_map(map);
  1280. if (err)
  1281. return err;
  1282. }
  1283. return 0;
  1284. }
  1285. int bpf__apply_obj_config(void)
  1286. {
  1287. struct bpf_object *obj, *tmp;
  1288. int err;
  1289. bpf_object__for_each_safe(obj, tmp) {
  1290. err = apply_obj_config_object(obj);
  1291. if (err)
  1292. return err;
  1293. }
  1294. return 0;
  1295. }
  1296. #define bpf__for_each_map(pos, obj, objtmp) \
  1297. bpf_object__for_each_safe(obj, objtmp) \
  1298. bpf_map__for_each(pos, obj)
  1299. #define bpf__for_each_stdout_map(pos, obj, objtmp) \
  1300. bpf__for_each_map(pos, obj, objtmp) \
  1301. if (bpf_map__name(pos) && \
  1302. (strcmp("__bpf_stdout__", \
  1303. bpf_map__name(pos)) == 0))
  1304. int bpf__setup_stdout(struct perf_evlist *evlist)
  1305. {
  1306. struct bpf_map_priv *tmpl_priv = NULL;
  1307. struct bpf_object *obj, *tmp;
  1308. struct perf_evsel *evsel = NULL;
  1309. struct bpf_map *map;
  1310. int err;
  1311. bool need_init = false;
  1312. bpf__for_each_stdout_map(map, obj, tmp) {
  1313. struct bpf_map_priv *priv = bpf_map__priv(map);
  1314. if (IS_ERR(priv))
  1315. return -BPF_LOADER_ERRNO__INTERNAL;
  1316. /*
  1317. * No need to check map type: type should have been
  1318. * verified by kernel.
  1319. */
  1320. if (!need_init && !priv)
  1321. need_init = !priv;
  1322. if (!tmpl_priv && priv)
  1323. tmpl_priv = priv;
  1324. }
  1325. if (!need_init)
  1326. return 0;
  1327. if (!tmpl_priv) {
  1328. err = parse_events(evlist, "bpf-output/no-inherit=1,name=__bpf_stdout__/",
  1329. NULL);
  1330. if (err) {
  1331. pr_debug("ERROR: failed to create bpf-output event\n");
  1332. return -err;
  1333. }
  1334. evsel = perf_evlist__last(evlist);
  1335. }
  1336. bpf__for_each_stdout_map(map, obj, tmp) {
  1337. struct bpf_map_priv *priv = bpf_map__priv(map);
  1338. if (IS_ERR(priv))
  1339. return -BPF_LOADER_ERRNO__INTERNAL;
  1340. if (priv)
  1341. continue;
  1342. if (tmpl_priv) {
  1343. priv = bpf_map_priv__clone(tmpl_priv);
  1344. if (!priv)
  1345. return -ENOMEM;
  1346. err = bpf_map__set_priv(map, priv, bpf_map_priv__clear);
  1347. if (err) {
  1348. bpf_map_priv__clear(map, priv);
  1349. return err;
  1350. }
  1351. } else if (evsel) {
  1352. struct bpf_map_op *op;
  1353. op = bpf_map__add_newop(map, NULL);
  1354. if (IS_ERR(op))
  1355. return PTR_ERR(op);
  1356. op->op_type = BPF_MAP_OP_SET_EVSEL;
  1357. op->v.evsel = evsel;
  1358. }
  1359. }
  1360. return 0;
  1361. }
  1362. #define ERRNO_OFFSET(e) ((e) - __BPF_LOADER_ERRNO__START)
  1363. #define ERRCODE_OFFSET(c) ERRNO_OFFSET(BPF_LOADER_ERRNO__##c)
  1364. #define NR_ERRNO (__BPF_LOADER_ERRNO__END - __BPF_LOADER_ERRNO__START)
  1365. static const char *bpf_loader_strerror_table[NR_ERRNO] = {
  1366. [ERRCODE_OFFSET(CONFIG)] = "Invalid config string",
  1367. [ERRCODE_OFFSET(GROUP)] = "Invalid group name",
  1368. [ERRCODE_OFFSET(EVENTNAME)] = "No event name found in config string",
  1369. [ERRCODE_OFFSET(INTERNAL)] = "BPF loader internal error",
  1370. [ERRCODE_OFFSET(COMPILE)] = "Error when compiling BPF scriptlet",
  1371. [ERRCODE_OFFSET(PROGCONF_TERM)] = "Invalid program config term in config string",
  1372. [ERRCODE_OFFSET(PROLOGUE)] = "Failed to generate prologue",
  1373. [ERRCODE_OFFSET(PROLOGUE2BIG)] = "Prologue too big for program",
  1374. [ERRCODE_OFFSET(PROLOGUEOOB)] = "Offset out of bound for prologue",
  1375. [ERRCODE_OFFSET(OBJCONF_OPT)] = "Invalid object config option",
  1376. [ERRCODE_OFFSET(OBJCONF_CONF)] = "Config value not set (missing '=')",
  1377. [ERRCODE_OFFSET(OBJCONF_MAP_OPT)] = "Invalid object map config option",
  1378. [ERRCODE_OFFSET(OBJCONF_MAP_NOTEXIST)] = "Target map doesn't exist",
  1379. [ERRCODE_OFFSET(OBJCONF_MAP_VALUE)] = "Incorrect value type for map",
  1380. [ERRCODE_OFFSET(OBJCONF_MAP_TYPE)] = "Incorrect map type",
  1381. [ERRCODE_OFFSET(OBJCONF_MAP_KEYSIZE)] = "Incorrect map key size",
  1382. [ERRCODE_OFFSET(OBJCONF_MAP_VALUESIZE)] = "Incorrect map value size",
  1383. [ERRCODE_OFFSET(OBJCONF_MAP_NOEVT)] = "Event not found for map setting",
  1384. [ERRCODE_OFFSET(OBJCONF_MAP_MAPSIZE)] = "Invalid map size for event setting",
  1385. [ERRCODE_OFFSET(OBJCONF_MAP_EVTDIM)] = "Event dimension too large",
  1386. [ERRCODE_OFFSET(OBJCONF_MAP_EVTINH)] = "Doesn't support inherit event",
  1387. [ERRCODE_OFFSET(OBJCONF_MAP_EVTTYPE)] = "Wrong event type for map",
  1388. [ERRCODE_OFFSET(OBJCONF_MAP_IDX2BIG)] = "Index too large",
  1389. };
  1390. static int
  1391. bpf_loader_strerror(int err, char *buf, size_t size)
  1392. {
  1393. char sbuf[STRERR_BUFSIZE];
  1394. const char *msg;
  1395. if (!buf || !size)
  1396. return -1;
  1397. err = err > 0 ? err : -err;
  1398. if (err >= __LIBBPF_ERRNO__START)
  1399. return libbpf_strerror(err, buf, size);
  1400. if (err >= __BPF_LOADER_ERRNO__START && err < __BPF_LOADER_ERRNO__END) {
  1401. msg = bpf_loader_strerror_table[ERRNO_OFFSET(err)];
  1402. snprintf(buf, size, "%s", msg);
  1403. buf[size - 1] = '\0';
  1404. return 0;
  1405. }
  1406. if (err >= __BPF_LOADER_ERRNO__END)
  1407. snprintf(buf, size, "Unknown bpf loader error %d", err);
  1408. else
  1409. snprintf(buf, size, "%s",
  1410. str_error_r(err, sbuf, sizeof(sbuf)));
  1411. buf[size - 1] = '\0';
  1412. return -1;
  1413. }
  1414. #define bpf__strerror_head(err, buf, size) \
  1415. char sbuf[STRERR_BUFSIZE], *emsg;\
  1416. if (!size)\
  1417. return 0;\
  1418. if (err < 0)\
  1419. err = -err;\
  1420. bpf_loader_strerror(err, sbuf, sizeof(sbuf));\
  1421. emsg = sbuf;\
  1422. switch (err) {\
  1423. default:\
  1424. scnprintf(buf, size, "%s", emsg);\
  1425. break;
  1426. #define bpf__strerror_entry(val, fmt...)\
  1427. case val: {\
  1428. scnprintf(buf, size, fmt);\
  1429. break;\
  1430. }
  1431. #define bpf__strerror_end(buf, size)\
  1432. }\
  1433. buf[size - 1] = '\0';
  1434. int bpf__strerror_prepare_load(const char *filename, bool source,
  1435. int err, char *buf, size_t size)
  1436. {
  1437. size_t n;
  1438. int ret;
  1439. n = snprintf(buf, size, "Failed to load %s%s: ",
  1440. filename, source ? " from source" : "");
  1441. if (n >= size) {
  1442. buf[size - 1] = '\0';
  1443. return 0;
  1444. }
  1445. buf += n;
  1446. size -= n;
  1447. ret = bpf_loader_strerror(err, buf, size);
  1448. buf[size - 1] = '\0';
  1449. return ret;
  1450. }
  1451. int bpf__strerror_probe(struct bpf_object *obj __maybe_unused,
  1452. int err, char *buf, size_t size)
  1453. {
  1454. bpf__strerror_head(err, buf, size);
  1455. case BPF_LOADER_ERRNO__PROGCONF_TERM: {
  1456. scnprintf(buf, size, "%s (add -v to see detail)", emsg);
  1457. break;
  1458. }
  1459. bpf__strerror_entry(EEXIST, "Probe point exist. Try 'perf probe -d \"*\"' and set 'force=yes'");
  1460. bpf__strerror_entry(EACCES, "You need to be root");
  1461. bpf__strerror_entry(EPERM, "You need to be root, and /proc/sys/kernel/kptr_restrict should be 0");
  1462. bpf__strerror_entry(ENOENT, "You need to check probing points in BPF file");
  1463. bpf__strerror_end(buf, size);
  1464. return 0;
  1465. }
  1466. int bpf__strerror_load(struct bpf_object *obj,
  1467. int err, char *buf, size_t size)
  1468. {
  1469. bpf__strerror_head(err, buf, size);
  1470. case LIBBPF_ERRNO__KVER: {
  1471. unsigned int obj_kver = bpf_object__kversion(obj);
  1472. unsigned int real_kver;
  1473. if (fetch_kernel_version(&real_kver, NULL, 0)) {
  1474. scnprintf(buf, size, "Unable to fetch kernel version");
  1475. break;
  1476. }
  1477. if (obj_kver != real_kver) {
  1478. scnprintf(buf, size,
  1479. "'version' ("KVER_FMT") doesn't match running kernel ("KVER_FMT")",
  1480. KVER_PARAM(obj_kver),
  1481. KVER_PARAM(real_kver));
  1482. break;
  1483. }
  1484. scnprintf(buf, size, "Failed to load program for unknown reason");
  1485. break;
  1486. }
  1487. bpf__strerror_end(buf, size);
  1488. return 0;
  1489. }
  1490. int bpf__strerror_config_obj(struct bpf_object *obj __maybe_unused,
  1491. struct parse_events_term *term __maybe_unused,
  1492. struct perf_evlist *evlist __maybe_unused,
  1493. int *error_pos __maybe_unused, int err,
  1494. char *buf, size_t size)
  1495. {
  1496. bpf__strerror_head(err, buf, size);
  1497. bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE,
  1498. "Can't use this config term with this map type");
  1499. bpf__strerror_end(buf, size);
  1500. return 0;
  1501. }
  1502. int bpf__strerror_apply_obj_config(int err, char *buf, size_t size)
  1503. {
  1504. bpf__strerror_head(err, buf, size);
  1505. bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_EVTDIM,
  1506. "Cannot set event to BPF map in multi-thread tracing");
  1507. bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_EVTINH,
  1508. "%s (Hint: use -i to turn off inherit)", emsg);
  1509. bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_EVTTYPE,
  1510. "Can only put raw, hardware and BPF output event into a BPF map");
  1511. bpf__strerror_end(buf, size);
  1512. return 0;
  1513. }
  1514. int bpf__strerror_setup_stdout(struct perf_evlist *evlist __maybe_unused,
  1515. int err, char *buf, size_t size)
  1516. {
  1517. bpf__strerror_head(err, buf, size);
  1518. bpf__strerror_end(buf, size);
  1519. return 0;
  1520. }