bpf_iter.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843
  1. // SPDX-License-Identifier: GPL-2.0-only
  2. /* Copyright (c) 2020 Facebook */
  3. #include <linux/fs.h>
  4. #include <linux/anon_inodes.h>
  5. #include <linux/filter.h>
  6. #include <linux/bpf.h>
  7. #include <linux/rcupdate_trace.h>
  8. struct bpf_iter_target_info {
  9. struct list_head list;
  10. const struct bpf_iter_reg *reg_info;
  11. u32 btf_id; /* cached value */
  12. };
  13. struct bpf_iter_link {
  14. struct bpf_link link;
  15. struct bpf_iter_aux_info aux;
  16. struct bpf_iter_target_info *tinfo;
  17. };
  18. struct bpf_iter_priv_data {
  19. struct bpf_iter_target_info *tinfo;
  20. const struct bpf_iter_seq_info *seq_info;
  21. struct bpf_prog *prog;
  22. u64 session_id;
  23. u64 seq_num;
  24. bool done_stop;
  25. u8 target_private[] __aligned(8);
  26. };
  27. static struct list_head targets = LIST_HEAD_INIT(targets);
  28. static DEFINE_MUTEX(targets_mutex);
  29. /* protect bpf_iter_link changes */
  30. static DEFINE_MUTEX(link_mutex);
  31. /* incremented on every opened seq_file */
  32. static atomic64_t session_id;
  33. static int prepare_seq_file(struct file *file, struct bpf_iter_link *link,
  34. const struct bpf_iter_seq_info *seq_info);
  35. static void bpf_iter_inc_seq_num(struct seq_file *seq)
  36. {
  37. struct bpf_iter_priv_data *iter_priv;
  38. iter_priv = container_of(seq->private, struct bpf_iter_priv_data,
  39. target_private);
  40. iter_priv->seq_num++;
  41. }
  42. static void bpf_iter_dec_seq_num(struct seq_file *seq)
  43. {
  44. struct bpf_iter_priv_data *iter_priv;
  45. iter_priv = container_of(seq->private, struct bpf_iter_priv_data,
  46. target_private);
  47. iter_priv->seq_num--;
  48. }
  49. static void bpf_iter_done_stop(struct seq_file *seq)
  50. {
  51. struct bpf_iter_priv_data *iter_priv;
  52. iter_priv = container_of(seq->private, struct bpf_iter_priv_data,
  53. target_private);
  54. iter_priv->done_stop = true;
  55. }
  56. static inline bool bpf_iter_target_support_resched(const struct bpf_iter_target_info *tinfo)
  57. {
  58. return tinfo->reg_info->feature & BPF_ITER_RESCHED;
  59. }
  60. static bool bpf_iter_support_resched(struct seq_file *seq)
  61. {
  62. struct bpf_iter_priv_data *iter_priv;
  63. iter_priv = container_of(seq->private, struct bpf_iter_priv_data,
  64. target_private);
  65. return bpf_iter_target_support_resched(iter_priv->tinfo);
  66. }
  67. /* maximum visited objects before bailing out */
  68. #define MAX_ITER_OBJECTS 1000000
  69. /* bpf_seq_read, a customized and simpler version for bpf iterator.
  70. * The following are differences from seq_read():
  71. * . fixed buffer size (PAGE_SIZE)
  72. * . assuming NULL ->llseek()
  73. * . stop() may call bpf program, handling potential overflow there
  74. */
  75. static ssize_t bpf_seq_read(struct file *file, char __user *buf, size_t size,
  76. loff_t *ppos)
  77. {
  78. struct seq_file *seq = file->private_data;
  79. size_t n, offs, copied = 0;
  80. int err = 0, num_objs = 0;
  81. bool can_resched;
  82. void *p;
  83. mutex_lock(&seq->lock);
  84. if (!seq->buf) {
  85. seq->size = PAGE_SIZE << 3;
  86. seq->buf = kvmalloc(seq->size, GFP_KERNEL);
  87. if (!seq->buf) {
  88. err = -ENOMEM;
  89. goto done;
  90. }
  91. }
  92. if (seq->count) {
  93. n = min(seq->count, size);
  94. err = copy_to_user(buf, seq->buf + seq->from, n);
  95. if (err) {
  96. err = -EFAULT;
  97. goto done;
  98. }
  99. seq->count -= n;
  100. seq->from += n;
  101. copied = n;
  102. goto done;
  103. }
  104. seq->from = 0;
  105. p = seq->op->start(seq, &seq->index);
  106. if (!p)
  107. goto stop;
  108. if (IS_ERR(p)) {
  109. err = PTR_ERR(p);
  110. seq->op->stop(seq, p);
  111. seq->count = 0;
  112. goto done;
  113. }
  114. err = seq->op->show(seq, p);
  115. if (err > 0) {
  116. /* object is skipped, decrease seq_num, so next
  117. * valid object can reuse the same seq_num.
  118. */
  119. bpf_iter_dec_seq_num(seq);
  120. seq->count = 0;
  121. } else if (err < 0 || seq_has_overflowed(seq)) {
  122. if (!err)
  123. err = -E2BIG;
  124. seq->op->stop(seq, p);
  125. seq->count = 0;
  126. goto done;
  127. }
  128. can_resched = bpf_iter_support_resched(seq);
  129. while (1) {
  130. loff_t pos = seq->index;
  131. num_objs++;
  132. offs = seq->count;
  133. p = seq->op->next(seq, p, &seq->index);
  134. if (pos == seq->index) {
  135. pr_info_ratelimited("buggy seq_file .next function %ps "
  136. "did not updated position index\n",
  137. seq->op->next);
  138. seq->index++;
  139. }
  140. if (IS_ERR_OR_NULL(p))
  141. break;
  142. /* got a valid next object, increase seq_num */
  143. bpf_iter_inc_seq_num(seq);
  144. if (seq->count >= size)
  145. break;
  146. if (num_objs >= MAX_ITER_OBJECTS) {
  147. if (offs == 0) {
  148. err = -EAGAIN;
  149. seq->op->stop(seq, p);
  150. goto done;
  151. }
  152. break;
  153. }
  154. err = seq->op->show(seq, p);
  155. if (err > 0) {
  156. bpf_iter_dec_seq_num(seq);
  157. seq->count = offs;
  158. } else if (err < 0 || seq_has_overflowed(seq)) {
  159. seq->count = offs;
  160. if (offs == 0) {
  161. if (!err)
  162. err = -E2BIG;
  163. seq->op->stop(seq, p);
  164. goto done;
  165. }
  166. break;
  167. }
  168. if (can_resched)
  169. cond_resched();
  170. }
  171. stop:
  172. offs = seq->count;
  173. if (IS_ERR(p)) {
  174. seq->op->stop(seq, NULL);
  175. err = PTR_ERR(p);
  176. goto done;
  177. }
  178. /* bpf program called if !p */
  179. seq->op->stop(seq, p);
  180. if (!p) {
  181. if (!seq_has_overflowed(seq)) {
  182. bpf_iter_done_stop(seq);
  183. } else {
  184. seq->count = offs;
  185. if (offs == 0) {
  186. err = -E2BIG;
  187. goto done;
  188. }
  189. }
  190. }
  191. n = min(seq->count, size);
  192. err = copy_to_user(buf, seq->buf, n);
  193. if (err) {
  194. err = -EFAULT;
  195. goto done;
  196. }
  197. copied = n;
  198. seq->count -= n;
  199. seq->from = n;
  200. done:
  201. if (!copied)
  202. copied = err;
  203. else
  204. *ppos += copied;
  205. mutex_unlock(&seq->lock);
  206. return copied;
  207. }
  208. static const struct bpf_iter_seq_info *
  209. __get_seq_info(struct bpf_iter_link *link)
  210. {
  211. const struct bpf_iter_seq_info *seq_info;
  212. if (link->aux.map) {
  213. seq_info = link->aux.map->ops->iter_seq_info;
  214. if (seq_info)
  215. return seq_info;
  216. }
  217. return link->tinfo->reg_info->seq_info;
  218. }
  219. static int iter_open(struct inode *inode, struct file *file)
  220. {
  221. struct bpf_iter_link *link = inode->i_private;
  222. return prepare_seq_file(file, link, __get_seq_info(link));
  223. }
  224. static int iter_release(struct inode *inode, struct file *file)
  225. {
  226. struct bpf_iter_priv_data *iter_priv;
  227. struct seq_file *seq;
  228. seq = file->private_data;
  229. if (!seq)
  230. return 0;
  231. iter_priv = container_of(seq->private, struct bpf_iter_priv_data,
  232. target_private);
  233. if (iter_priv->seq_info->fini_seq_private)
  234. iter_priv->seq_info->fini_seq_private(seq->private);
  235. bpf_prog_put(iter_priv->prog);
  236. seq->private = iter_priv;
  237. return seq_release_private(inode, file);
  238. }
  239. const struct file_operations bpf_iter_fops = {
  240. .open = iter_open,
  241. .read = bpf_seq_read,
  242. .release = iter_release,
  243. };
  244. /* The argument reg_info will be cached in bpf_iter_target_info.
  245. * The common practice is to declare target reg_info as
  246. * a const static variable and passed as an argument to
  247. * bpf_iter_reg_target().
  248. */
  249. int bpf_iter_reg_target(const struct bpf_iter_reg *reg_info)
  250. {
  251. struct bpf_iter_target_info *tinfo;
  252. tinfo = kzalloc(sizeof(*tinfo), GFP_KERNEL);
  253. if (!tinfo)
  254. return -ENOMEM;
  255. tinfo->reg_info = reg_info;
  256. INIT_LIST_HEAD(&tinfo->list);
  257. mutex_lock(&targets_mutex);
  258. list_add(&tinfo->list, &targets);
  259. mutex_unlock(&targets_mutex);
  260. return 0;
  261. }
  262. void bpf_iter_unreg_target(const struct bpf_iter_reg *reg_info)
  263. {
  264. struct bpf_iter_target_info *tinfo;
  265. bool found = false;
  266. mutex_lock(&targets_mutex);
  267. list_for_each_entry(tinfo, &targets, list) {
  268. if (reg_info == tinfo->reg_info) {
  269. list_del(&tinfo->list);
  270. kfree(tinfo);
  271. found = true;
  272. break;
  273. }
  274. }
  275. mutex_unlock(&targets_mutex);
  276. WARN_ON(found == false);
  277. }
  278. static void cache_btf_id(struct bpf_iter_target_info *tinfo,
  279. struct bpf_prog *prog)
  280. {
  281. tinfo->btf_id = prog->aux->attach_btf_id;
  282. }
  283. bool bpf_iter_prog_supported(struct bpf_prog *prog)
  284. {
  285. const char *attach_fname = prog->aux->attach_func_name;
  286. struct bpf_iter_target_info *tinfo = NULL, *iter;
  287. u32 prog_btf_id = prog->aux->attach_btf_id;
  288. const char *prefix = BPF_ITER_FUNC_PREFIX;
  289. int prefix_len = strlen(prefix);
  290. if (strncmp(attach_fname, prefix, prefix_len))
  291. return false;
  292. mutex_lock(&targets_mutex);
  293. list_for_each_entry(iter, &targets, list) {
  294. if (iter->btf_id && iter->btf_id == prog_btf_id) {
  295. tinfo = iter;
  296. break;
  297. }
  298. if (!strcmp(attach_fname + prefix_len, iter->reg_info->target)) {
  299. cache_btf_id(iter, prog);
  300. tinfo = iter;
  301. break;
  302. }
  303. }
  304. mutex_unlock(&targets_mutex);
  305. if (tinfo) {
  306. prog->aux->ctx_arg_info_size = tinfo->reg_info->ctx_arg_info_size;
  307. prog->aux->ctx_arg_info = tinfo->reg_info->ctx_arg_info;
  308. }
  309. return tinfo != NULL;
  310. }
  311. const struct bpf_func_proto *
  312. bpf_iter_get_func_proto(enum bpf_func_id func_id, const struct bpf_prog *prog)
  313. {
  314. const struct bpf_iter_target_info *tinfo;
  315. const struct bpf_func_proto *fn = NULL;
  316. mutex_lock(&targets_mutex);
  317. list_for_each_entry(tinfo, &targets, list) {
  318. if (tinfo->btf_id == prog->aux->attach_btf_id) {
  319. const struct bpf_iter_reg *reg_info;
  320. reg_info = tinfo->reg_info;
  321. if (reg_info->get_func_proto)
  322. fn = reg_info->get_func_proto(func_id, prog);
  323. break;
  324. }
  325. }
  326. mutex_unlock(&targets_mutex);
  327. return fn;
  328. }
  329. static void bpf_iter_link_release(struct bpf_link *link)
  330. {
  331. struct bpf_iter_link *iter_link =
  332. container_of(link, struct bpf_iter_link, link);
  333. if (iter_link->tinfo->reg_info->detach_target)
  334. iter_link->tinfo->reg_info->detach_target(&iter_link->aux);
  335. }
  336. static void bpf_iter_link_dealloc(struct bpf_link *link)
  337. {
  338. struct bpf_iter_link *iter_link =
  339. container_of(link, struct bpf_iter_link, link);
  340. kfree(iter_link);
  341. }
  342. static int bpf_iter_link_replace(struct bpf_link *link,
  343. struct bpf_prog *new_prog,
  344. struct bpf_prog *old_prog)
  345. {
  346. int ret = 0;
  347. mutex_lock(&link_mutex);
  348. if (old_prog && link->prog != old_prog) {
  349. ret = -EPERM;
  350. goto out_unlock;
  351. }
  352. if (link->prog->type != new_prog->type ||
  353. link->prog->expected_attach_type != new_prog->expected_attach_type ||
  354. link->prog->aux->attach_btf_id != new_prog->aux->attach_btf_id) {
  355. ret = -EINVAL;
  356. goto out_unlock;
  357. }
  358. old_prog = xchg(&link->prog, new_prog);
  359. bpf_prog_put(old_prog);
  360. out_unlock:
  361. mutex_unlock(&link_mutex);
  362. return ret;
  363. }
  364. static void bpf_iter_link_show_fdinfo(const struct bpf_link *link,
  365. struct seq_file *seq)
  366. {
  367. struct bpf_iter_link *iter_link =
  368. container_of(link, struct bpf_iter_link, link);
  369. bpf_iter_show_fdinfo_t show_fdinfo;
  370. seq_printf(seq,
  371. "target_name:\t%s\n",
  372. iter_link->tinfo->reg_info->target);
  373. show_fdinfo = iter_link->tinfo->reg_info->show_fdinfo;
  374. if (show_fdinfo)
  375. show_fdinfo(&iter_link->aux, seq);
  376. }
  377. static int bpf_iter_link_fill_link_info(const struct bpf_link *link,
  378. struct bpf_link_info *info)
  379. {
  380. struct bpf_iter_link *iter_link =
  381. container_of(link, struct bpf_iter_link, link);
  382. char __user *ubuf = u64_to_user_ptr(info->iter.target_name);
  383. bpf_iter_fill_link_info_t fill_link_info;
  384. u32 ulen = info->iter.target_name_len;
  385. const char *target_name;
  386. u32 target_len;
  387. if (!ulen ^ !ubuf)
  388. return -EINVAL;
  389. target_name = iter_link->tinfo->reg_info->target;
  390. target_len = strlen(target_name);
  391. info->iter.target_name_len = target_len + 1;
  392. if (ubuf) {
  393. if (ulen >= target_len + 1) {
  394. if (copy_to_user(ubuf, target_name, target_len + 1))
  395. return -EFAULT;
  396. } else {
  397. char zero = '\0';
  398. if (copy_to_user(ubuf, target_name, ulen - 1))
  399. return -EFAULT;
  400. if (put_user(zero, ubuf + ulen - 1))
  401. return -EFAULT;
  402. return -ENOSPC;
  403. }
  404. }
  405. fill_link_info = iter_link->tinfo->reg_info->fill_link_info;
  406. if (fill_link_info)
  407. return fill_link_info(&iter_link->aux, info);
  408. return 0;
  409. }
  410. static const struct bpf_link_ops bpf_iter_link_lops = {
  411. .release = bpf_iter_link_release,
  412. .dealloc = bpf_iter_link_dealloc,
  413. .update_prog = bpf_iter_link_replace,
  414. .show_fdinfo = bpf_iter_link_show_fdinfo,
  415. .fill_link_info = bpf_iter_link_fill_link_info,
  416. };
  417. bool bpf_link_is_iter(struct bpf_link *link)
  418. {
  419. return link->ops == &bpf_iter_link_lops;
  420. }
  421. int bpf_iter_link_attach(const union bpf_attr *attr, bpfptr_t uattr,
  422. struct bpf_prog *prog)
  423. {
  424. struct bpf_iter_target_info *tinfo = NULL, *iter;
  425. struct bpf_link_primer link_primer;
  426. union bpf_iter_link_info linfo;
  427. struct bpf_iter_link *link;
  428. u32 prog_btf_id, linfo_len;
  429. bpfptr_t ulinfo;
  430. int err;
  431. if (attr->link_create.target_fd || attr->link_create.flags)
  432. return -EINVAL;
  433. memset(&linfo, 0, sizeof(union bpf_iter_link_info));
  434. ulinfo = make_bpfptr(attr->link_create.iter_info, uattr.is_kernel);
  435. linfo_len = attr->link_create.iter_info_len;
  436. if (bpfptr_is_null(ulinfo) ^ !linfo_len)
  437. return -EINVAL;
  438. if (!bpfptr_is_null(ulinfo)) {
  439. err = bpf_check_uarg_tail_zero(ulinfo, sizeof(linfo),
  440. linfo_len);
  441. if (err)
  442. return err;
  443. linfo_len = min_t(u32, linfo_len, sizeof(linfo));
  444. if (copy_from_bpfptr(&linfo, ulinfo, linfo_len))
  445. return -EFAULT;
  446. }
  447. prog_btf_id = prog->aux->attach_btf_id;
  448. mutex_lock(&targets_mutex);
  449. list_for_each_entry(iter, &targets, list) {
  450. if (iter->btf_id == prog_btf_id) {
  451. tinfo = iter;
  452. break;
  453. }
  454. }
  455. mutex_unlock(&targets_mutex);
  456. if (!tinfo)
  457. return -ENOENT;
  458. /* Only allow sleepable program for resched-able iterator */
  459. if (prog->sleepable && !bpf_iter_target_support_resched(tinfo))
  460. return -EINVAL;
  461. link = kzalloc(sizeof(*link), GFP_USER | __GFP_NOWARN);
  462. if (!link)
  463. return -ENOMEM;
  464. bpf_link_init(&link->link, BPF_LINK_TYPE_ITER, &bpf_iter_link_lops, prog);
  465. link->tinfo = tinfo;
  466. err = bpf_link_prime(&link->link, &link_primer);
  467. if (err) {
  468. kfree(link);
  469. return err;
  470. }
  471. if (tinfo->reg_info->attach_target) {
  472. err = tinfo->reg_info->attach_target(prog, &linfo, &link->aux);
  473. if (err) {
  474. bpf_link_cleanup(&link_primer);
  475. return err;
  476. }
  477. }
  478. return bpf_link_settle(&link_primer);
  479. }
  480. static void init_seq_meta(struct bpf_iter_priv_data *priv_data,
  481. struct bpf_iter_target_info *tinfo,
  482. const struct bpf_iter_seq_info *seq_info,
  483. struct bpf_prog *prog)
  484. {
  485. priv_data->tinfo = tinfo;
  486. priv_data->seq_info = seq_info;
  487. priv_data->prog = prog;
  488. priv_data->session_id = atomic64_inc_return(&session_id);
  489. priv_data->seq_num = 0;
  490. priv_data->done_stop = false;
  491. }
  492. static int prepare_seq_file(struct file *file, struct bpf_iter_link *link,
  493. const struct bpf_iter_seq_info *seq_info)
  494. {
  495. struct bpf_iter_priv_data *priv_data;
  496. struct bpf_iter_target_info *tinfo;
  497. struct bpf_prog *prog;
  498. u32 total_priv_dsize;
  499. struct seq_file *seq;
  500. int err = 0;
  501. mutex_lock(&link_mutex);
  502. prog = link->link.prog;
  503. bpf_prog_inc(prog);
  504. mutex_unlock(&link_mutex);
  505. tinfo = link->tinfo;
  506. total_priv_dsize = offsetof(struct bpf_iter_priv_data, target_private) +
  507. seq_info->seq_priv_size;
  508. priv_data = __seq_open_private(file, seq_info->seq_ops,
  509. total_priv_dsize);
  510. if (!priv_data) {
  511. err = -ENOMEM;
  512. goto release_prog;
  513. }
  514. if (seq_info->init_seq_private) {
  515. err = seq_info->init_seq_private(priv_data->target_private, &link->aux);
  516. if (err)
  517. goto release_seq_file;
  518. }
  519. init_seq_meta(priv_data, tinfo, seq_info, prog);
  520. seq = file->private_data;
  521. seq->private = priv_data->target_private;
  522. return 0;
  523. release_seq_file:
  524. seq_release_private(file->f_inode, file);
  525. file->private_data = NULL;
  526. release_prog:
  527. bpf_prog_put(prog);
  528. return err;
  529. }
  530. int bpf_iter_new_fd(struct bpf_link *link)
  531. {
  532. struct bpf_iter_link *iter_link;
  533. struct file *file;
  534. unsigned int flags;
  535. int err, fd;
  536. if (link->ops != &bpf_iter_link_lops)
  537. return -EINVAL;
  538. flags = O_RDONLY | O_CLOEXEC;
  539. fd = get_unused_fd_flags(flags);
  540. if (fd < 0)
  541. return fd;
  542. file = anon_inode_getfile("bpf_iter", &bpf_iter_fops, NULL, flags);
  543. if (IS_ERR(file)) {
  544. err = PTR_ERR(file);
  545. goto free_fd;
  546. }
  547. iter_link = container_of(link, struct bpf_iter_link, link);
  548. err = prepare_seq_file(file, iter_link, __get_seq_info(iter_link));
  549. if (err)
  550. goto free_file;
  551. fd_install(fd, file);
  552. return fd;
  553. free_file:
  554. fput(file);
  555. free_fd:
  556. put_unused_fd(fd);
  557. return err;
  558. }
  559. struct bpf_prog *bpf_iter_get_info(struct bpf_iter_meta *meta, bool in_stop)
  560. {
  561. struct bpf_iter_priv_data *iter_priv;
  562. struct seq_file *seq;
  563. void *seq_priv;
  564. seq = meta->seq;
  565. if (seq->file->f_op != &bpf_iter_fops)
  566. return NULL;
  567. seq_priv = seq->private;
  568. iter_priv = container_of(seq_priv, struct bpf_iter_priv_data,
  569. target_private);
  570. if (in_stop && iter_priv->done_stop)
  571. return NULL;
  572. meta->session_id = iter_priv->session_id;
  573. meta->seq_num = iter_priv->seq_num;
  574. return iter_priv->prog;
  575. }
  576. int bpf_iter_run_prog(struct bpf_prog *prog, void *ctx)
  577. {
  578. struct bpf_run_ctx run_ctx, *old_run_ctx;
  579. int ret;
  580. if (prog->sleepable) {
  581. rcu_read_lock_trace();
  582. migrate_disable();
  583. might_fault();
  584. old_run_ctx = bpf_set_run_ctx(&run_ctx);
  585. ret = bpf_prog_run(prog, ctx);
  586. bpf_reset_run_ctx(old_run_ctx);
  587. migrate_enable();
  588. rcu_read_unlock_trace();
  589. } else {
  590. rcu_read_lock();
  591. migrate_disable();
  592. old_run_ctx = bpf_set_run_ctx(&run_ctx);
  593. ret = bpf_prog_run(prog, ctx);
  594. bpf_reset_run_ctx(old_run_ctx);
  595. migrate_enable();
  596. rcu_read_unlock();
  597. }
  598. /* bpf program can only return 0 or 1:
  599. * 0 : okay
  600. * 1 : retry the same object
  601. * The bpf_iter_run_prog() return value
  602. * will be seq_ops->show() return value.
  603. */
  604. return ret == 0 ? 0 : -EAGAIN;
  605. }
  606. BPF_CALL_4(bpf_for_each_map_elem, struct bpf_map *, map, void *, callback_fn,
  607. void *, callback_ctx, u64, flags)
  608. {
  609. return map->ops->map_for_each_callback(map, callback_fn, callback_ctx, flags);
  610. }
  611. const struct bpf_func_proto bpf_for_each_map_elem_proto = {
  612. .func = bpf_for_each_map_elem,
  613. .gpl_only = false,
  614. .ret_type = RET_INTEGER,
  615. .arg1_type = ARG_CONST_MAP_PTR,
  616. .arg2_type = ARG_PTR_TO_FUNC,
  617. .arg3_type = ARG_PTR_TO_STACK_OR_NULL,
  618. .arg4_type = ARG_ANYTHING,
  619. };
  620. BPF_CALL_4(bpf_loop, u32, nr_loops, void *, callback_fn, void *, callback_ctx,
  621. u64, flags)
  622. {
  623. bpf_callback_t callback = (bpf_callback_t)callback_fn;
  624. u64 ret;
  625. u32 i;
  626. /* Note: these safety checks are also verified when bpf_loop
  627. * is inlined, be careful to modify this code in sync. See
  628. * function verifier.c:inline_bpf_loop.
  629. */
  630. if (flags)
  631. return -EINVAL;
  632. if (nr_loops > BPF_MAX_LOOPS)
  633. return -E2BIG;
  634. for (i = 0; i < nr_loops; i++) {
  635. ret = callback((u64)i, (u64)(long)callback_ctx, 0, 0, 0);
  636. /* return value: 0 - continue, 1 - stop and return */
  637. if (ret)
  638. return i + 1;
  639. }
  640. return i;
  641. }
  642. const struct bpf_func_proto bpf_loop_proto = {
  643. .func = bpf_loop,
  644. .gpl_only = false,
  645. .ret_type = RET_INTEGER,
  646. .arg1_type = ARG_ANYTHING,
  647. .arg2_type = ARG_PTR_TO_FUNC,
  648. .arg3_type = ARG_PTR_TO_STACK_OR_NULL,
  649. .arg4_type = ARG_ANYTHING,
  650. };
  651. struct bpf_iter_num_kern {
  652. int cur; /* current value, inclusive */
  653. int end; /* final value, exclusive */
  654. } __aligned(8);
  655. __bpf_kfunc_start_defs();
  656. __bpf_kfunc int bpf_iter_num_new(struct bpf_iter_num *it, int start, int end)
  657. {
  658. struct bpf_iter_num_kern *s = (void *)it;
  659. BUILD_BUG_ON(sizeof(struct bpf_iter_num_kern) != sizeof(struct bpf_iter_num));
  660. BUILD_BUG_ON(__alignof__(struct bpf_iter_num_kern) != __alignof__(struct bpf_iter_num));
  661. /* start == end is legit, it's an empty range and we'll just get NULL
  662. * on first (and any subsequent) bpf_iter_num_next() call
  663. */
  664. if (start > end) {
  665. s->cur = s->end = 0;
  666. return -EINVAL;
  667. }
  668. /* avoid overflows, e.g., if start == INT_MIN and end == INT_MAX */
  669. if ((s64)end - (s64)start > BPF_MAX_LOOPS) {
  670. s->cur = s->end = 0;
  671. return -E2BIG;
  672. }
  673. /* user will call bpf_iter_num_next() first,
  674. * which will set s->cur to exactly start value;
  675. * underflow shouldn't matter
  676. */
  677. s->cur = start - 1;
  678. s->end = end;
  679. return 0;
  680. }
  681. __bpf_kfunc int *bpf_iter_num_next(struct bpf_iter_num* it)
  682. {
  683. struct bpf_iter_num_kern *s = (void *)it;
  684. /* check failed initialization or if we are done (same behavior);
  685. * need to be careful about overflow, so convert to s64 for checks,
  686. * e.g., if s->cur == s->end == INT_MAX, we can't just do
  687. * s->cur + 1 >= s->end
  688. */
  689. if ((s64)(s->cur + 1) >= s->end) {
  690. s->cur = s->end = 0;
  691. return NULL;
  692. }
  693. s->cur++;
  694. return &s->cur;
  695. }
  696. __bpf_kfunc void bpf_iter_num_destroy(struct bpf_iter_num *it)
  697. {
  698. struct bpf_iter_num_kern *s = (void *)it;
  699. s->cur = s->end = 0;
  700. }
  701. __bpf_kfunc_end_defs();