ip6_flowlabel.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847
  1. /*
  2. * ip6_flowlabel.c IPv6 flowlabel manager.
  3. *
  4. * This program is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU General Public License
  6. * as published by the Free Software Foundation; either version
  7. * 2 of the License, or (at your option) any later version.
  8. *
  9. * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
  10. */
  11. #include <linux/capability.h>
  12. #include <linux/errno.h>
  13. #include <linux/types.h>
  14. #include <linux/socket.h>
  15. #include <linux/net.h>
  16. #include <linux/netdevice.h>
  17. #include <linux/if_arp.h>
  18. #include <linux/in6.h>
  19. #include <linux/route.h>
  20. #include <linux/proc_fs.h>
  21. #include <linux/seq_file.h>
  22. #include <linux/slab.h>
  23. #include <linux/export.h>
  24. #include <linux/pid_namespace.h>
  25. #include <net/net_namespace.h>
  26. #include <net/sock.h>
  27. #include <net/ipv6.h>
  28. #include <net/ndisc.h>
  29. #include <net/protocol.h>
  30. #include <net/ip6_route.h>
  31. #include <net/addrconf.h>
  32. #include <net/rawv6.h>
  33. #include <net/icmp.h>
  34. #include <net/transp_v6.h>
  35. #include <asm/uaccess.h>
  36. #define FL_MIN_LINGER 6 /* Minimal linger. It is set to 6sec specified
  37. in old IPv6 RFC. Well, it was reasonable value.
  38. */
  39. #define FL_MAX_LINGER 60 /* Maximal linger timeout */
  40. /* FL hash table */
  41. #define FL_MAX_PER_SOCK 32
  42. #define FL_MAX_SIZE 4096
  43. #define FL_HASH_MASK 255
  44. #define FL_HASH(l) (ntohl(l)&FL_HASH_MASK)
  45. static atomic_t fl_size = ATOMIC_INIT(0);
  46. static struct ip6_flowlabel __rcu *fl_ht[FL_HASH_MASK+1];
  47. static void ip6_fl_gc(unsigned long dummy);
  48. static DEFINE_TIMER(ip6_fl_gc_timer, ip6_fl_gc, 0, 0);
  49. /* FL hash table lock: it protects only of GC */
  50. static DEFINE_SPINLOCK(ip6_fl_lock);
  51. /* Big socket sock */
  52. static DEFINE_SPINLOCK(ip6_sk_fl_lock);
  53. #define for_each_fl_rcu(hash, fl) \
  54. for (fl = rcu_dereference_bh(fl_ht[(hash)]); \
  55. fl != NULL; \
  56. fl = rcu_dereference_bh(fl->next))
  57. #define for_each_fl_continue_rcu(fl) \
  58. for (fl = rcu_dereference_bh(fl->next); \
  59. fl != NULL; \
  60. fl = rcu_dereference_bh(fl->next))
  61. #define for_each_sk_fl_rcu(np, sfl) \
  62. for (sfl = rcu_dereference_bh(np->ipv6_fl_list); \
  63. sfl != NULL; \
  64. sfl = rcu_dereference_bh(sfl->next))
  65. static inline struct ip6_flowlabel *__fl_lookup(struct net *net, __be32 label)
  66. {
  67. struct ip6_flowlabel *fl;
  68. for_each_fl_rcu(FL_HASH(label), fl) {
  69. if (fl->label == label && net_eq(fl->fl_net, net))
  70. return fl;
  71. }
  72. return NULL;
  73. }
  74. static struct ip6_flowlabel *fl_lookup(struct net *net, __be32 label)
  75. {
  76. struct ip6_flowlabel *fl;
  77. rcu_read_lock_bh();
  78. fl = __fl_lookup(net, label);
  79. if (fl && !atomic_inc_not_zero(&fl->users))
  80. fl = NULL;
  81. rcu_read_unlock_bh();
  82. return fl;
  83. }
  84. static void fl_free(struct ip6_flowlabel *fl)
  85. {
  86. if (fl) {
  87. if (fl->share == IPV6_FL_S_PROCESS)
  88. put_pid(fl->owner.pid);
  89. release_net(fl->fl_net);
  90. kfree(fl->opt);
  91. kfree_rcu(fl, rcu);
  92. }
  93. }
  94. static void fl_release(struct ip6_flowlabel *fl)
  95. {
  96. spin_lock_bh(&ip6_fl_lock);
  97. fl->lastuse = jiffies;
  98. if (atomic_dec_and_test(&fl->users)) {
  99. unsigned long ttd = fl->lastuse + fl->linger;
  100. if (time_after(ttd, fl->expires))
  101. fl->expires = ttd;
  102. ttd = fl->expires;
  103. if (fl->opt && fl->share == IPV6_FL_S_EXCL) {
  104. struct ipv6_txoptions *opt = fl->opt;
  105. fl->opt = NULL;
  106. kfree(opt);
  107. }
  108. if (!timer_pending(&ip6_fl_gc_timer) ||
  109. time_after(ip6_fl_gc_timer.expires, ttd))
  110. mod_timer(&ip6_fl_gc_timer, ttd);
  111. }
  112. spin_unlock_bh(&ip6_fl_lock);
  113. }
  114. static void ip6_fl_gc(unsigned long dummy)
  115. {
  116. int i;
  117. unsigned long now = jiffies;
  118. unsigned long sched = 0;
  119. spin_lock(&ip6_fl_lock);
  120. for (i=0; i<=FL_HASH_MASK; i++) {
  121. struct ip6_flowlabel *fl;
  122. struct ip6_flowlabel __rcu **flp;
  123. flp = &fl_ht[i];
  124. while ((fl = rcu_dereference_protected(*flp,
  125. lockdep_is_held(&ip6_fl_lock))) != NULL) {
  126. if (atomic_read(&fl->users) == 0) {
  127. unsigned long ttd = fl->lastuse + fl->linger;
  128. if (time_after(ttd, fl->expires))
  129. fl->expires = ttd;
  130. ttd = fl->expires;
  131. if (time_after_eq(now, ttd)) {
  132. *flp = fl->next;
  133. fl_free(fl);
  134. atomic_dec(&fl_size);
  135. continue;
  136. }
  137. if (!sched || time_before(ttd, sched))
  138. sched = ttd;
  139. }
  140. flp = &fl->next;
  141. }
  142. }
  143. if (!sched && atomic_read(&fl_size))
  144. sched = now + FL_MAX_LINGER;
  145. if (sched) {
  146. mod_timer(&ip6_fl_gc_timer, sched);
  147. }
  148. spin_unlock(&ip6_fl_lock);
  149. }
  150. static void __net_exit ip6_fl_purge(struct net *net)
  151. {
  152. int i;
  153. spin_lock(&ip6_fl_lock);
  154. for (i = 0; i <= FL_HASH_MASK; i++) {
  155. struct ip6_flowlabel *fl;
  156. struct ip6_flowlabel __rcu **flp;
  157. flp = &fl_ht[i];
  158. while ((fl = rcu_dereference_protected(*flp,
  159. lockdep_is_held(&ip6_fl_lock))) != NULL) {
  160. if (net_eq(fl->fl_net, net) &&
  161. atomic_read(&fl->users) == 0) {
  162. *flp = fl->next;
  163. fl_free(fl);
  164. atomic_dec(&fl_size);
  165. continue;
  166. }
  167. flp = &fl->next;
  168. }
  169. }
  170. spin_unlock(&ip6_fl_lock);
  171. }
  172. static struct ip6_flowlabel *fl_intern(struct net *net,
  173. struct ip6_flowlabel *fl, __be32 label)
  174. {
  175. struct ip6_flowlabel *lfl;
  176. fl->label = label & IPV6_FLOWLABEL_MASK;
  177. spin_lock_bh(&ip6_fl_lock);
  178. if (label == 0) {
  179. for (;;) {
  180. fl->label = htonl(net_random())&IPV6_FLOWLABEL_MASK;
  181. if (fl->label) {
  182. lfl = __fl_lookup(net, fl->label);
  183. if (lfl == NULL)
  184. break;
  185. }
  186. }
  187. } else {
  188. /*
  189. * we dropper the ip6_fl_lock, so this entry could reappear
  190. * and we need to recheck with it.
  191. *
  192. * OTOH no need to search the active socket first, like it is
  193. * done in ipv6_flowlabel_opt - sock is locked, so new entry
  194. * with the same label can only appear on another sock
  195. */
  196. lfl = __fl_lookup(net, fl->label);
  197. if (lfl != NULL) {
  198. atomic_inc(&lfl->users);
  199. spin_unlock_bh(&ip6_fl_lock);
  200. return lfl;
  201. }
  202. }
  203. fl->lastuse = jiffies;
  204. fl->next = fl_ht[FL_HASH(fl->label)];
  205. rcu_assign_pointer(fl_ht[FL_HASH(fl->label)], fl);
  206. atomic_inc(&fl_size);
  207. spin_unlock_bh(&ip6_fl_lock);
  208. return NULL;
  209. }
  210. /* Socket flowlabel lists */
  211. struct ip6_flowlabel * fl6_sock_lookup(struct sock *sk, __be32 label)
  212. {
  213. struct ipv6_fl_socklist *sfl;
  214. struct ipv6_pinfo *np = inet6_sk(sk);
  215. label &= IPV6_FLOWLABEL_MASK;
  216. rcu_read_lock_bh();
  217. for_each_sk_fl_rcu(np, sfl) {
  218. struct ip6_flowlabel *fl = sfl->fl;
  219. if (fl->label == label) {
  220. fl->lastuse = jiffies;
  221. atomic_inc(&fl->users);
  222. rcu_read_unlock_bh();
  223. return fl;
  224. }
  225. }
  226. rcu_read_unlock_bh();
  227. return NULL;
  228. }
  229. EXPORT_SYMBOL_GPL(fl6_sock_lookup);
  230. void fl6_free_socklist(struct sock *sk)
  231. {
  232. struct ipv6_pinfo *np = inet6_sk(sk);
  233. struct ipv6_fl_socklist *sfl;
  234. if (!rcu_access_pointer(np->ipv6_fl_list))
  235. return;
  236. spin_lock_bh(&ip6_sk_fl_lock);
  237. while ((sfl = rcu_dereference_protected(np->ipv6_fl_list,
  238. lockdep_is_held(&ip6_sk_fl_lock))) != NULL) {
  239. np->ipv6_fl_list = sfl->next;
  240. spin_unlock_bh(&ip6_sk_fl_lock);
  241. fl_release(sfl->fl);
  242. kfree_rcu(sfl, rcu);
  243. spin_lock_bh(&ip6_sk_fl_lock);
  244. }
  245. spin_unlock_bh(&ip6_sk_fl_lock);
  246. }
  247. /* Service routines */
  248. /*
  249. It is the only difficult place. flowlabel enforces equal headers
  250. before and including routing header, however user may supply options
  251. following rthdr.
  252. */
  253. struct ipv6_txoptions *fl6_merge_options(struct ipv6_txoptions * opt_space,
  254. struct ip6_flowlabel * fl,
  255. struct ipv6_txoptions * fopt)
  256. {
  257. struct ipv6_txoptions * fl_opt = fl->opt;
  258. if (fopt == NULL || fopt->opt_flen == 0)
  259. return fl_opt;
  260. if (fl_opt != NULL) {
  261. opt_space->hopopt = fl_opt->hopopt;
  262. opt_space->dst0opt = fl_opt->dst0opt;
  263. opt_space->srcrt = fl_opt->srcrt;
  264. opt_space->opt_nflen = fl_opt->opt_nflen;
  265. } else {
  266. if (fopt->opt_nflen == 0)
  267. return fopt;
  268. opt_space->hopopt = NULL;
  269. opt_space->dst0opt = NULL;
  270. opt_space->srcrt = NULL;
  271. opt_space->opt_nflen = 0;
  272. }
  273. opt_space->dst1opt = fopt->dst1opt;
  274. opt_space->opt_flen = fopt->opt_flen;
  275. return opt_space;
  276. }
  277. EXPORT_SYMBOL_GPL(fl6_merge_options);
  278. static unsigned long check_linger(unsigned long ttl)
  279. {
  280. if (ttl < FL_MIN_LINGER)
  281. return FL_MIN_LINGER*HZ;
  282. if (ttl > FL_MAX_LINGER && !capable(CAP_NET_ADMIN))
  283. return 0;
  284. return ttl*HZ;
  285. }
  286. static int fl6_renew(struct ip6_flowlabel *fl, unsigned long linger, unsigned long expires)
  287. {
  288. linger = check_linger(linger);
  289. if (!linger)
  290. return -EPERM;
  291. expires = check_linger(expires);
  292. if (!expires)
  293. return -EPERM;
  294. fl->lastuse = jiffies;
  295. if (time_before(fl->linger, linger))
  296. fl->linger = linger;
  297. if (time_before(expires, fl->linger))
  298. expires = fl->linger;
  299. if (time_before(fl->expires, fl->lastuse + expires))
  300. fl->expires = fl->lastuse + expires;
  301. return 0;
  302. }
  303. static struct ip6_flowlabel *
  304. fl_create(struct net *net, struct sock *sk, struct in6_flowlabel_req *freq,
  305. char __user *optval, int optlen, int *err_p)
  306. {
  307. struct ip6_flowlabel *fl = NULL;
  308. int olen;
  309. int addr_type;
  310. int err;
  311. olen = optlen - CMSG_ALIGN(sizeof(*freq));
  312. err = -EINVAL;
  313. if (olen > 64 * 1024)
  314. goto done;
  315. err = -ENOMEM;
  316. fl = kzalloc(sizeof(*fl), GFP_KERNEL);
  317. if (fl == NULL)
  318. goto done;
  319. if (olen > 0) {
  320. struct msghdr msg;
  321. struct flowi6 flowi6;
  322. int junk;
  323. err = -ENOMEM;
  324. fl->opt = kmalloc(sizeof(*fl->opt) + olen, GFP_KERNEL);
  325. if (fl->opt == NULL)
  326. goto done;
  327. memset(fl->opt, 0, sizeof(*fl->opt));
  328. fl->opt->tot_len = sizeof(*fl->opt) + olen;
  329. err = -EFAULT;
  330. if (copy_from_user(fl->opt+1, optval+CMSG_ALIGN(sizeof(*freq)), olen))
  331. goto done;
  332. msg.msg_controllen = olen;
  333. msg.msg_control = (void*)(fl->opt+1);
  334. memset(&flowi6, 0, sizeof(flowi6));
  335. err = ip6_datagram_send_ctl(net, sk, &msg, &flowi6, fl->opt,
  336. &junk, &junk, &junk);
  337. if (err)
  338. goto done;
  339. err = -EINVAL;
  340. if (fl->opt->opt_flen)
  341. goto done;
  342. if (fl->opt->opt_nflen == 0) {
  343. kfree(fl->opt);
  344. fl->opt = NULL;
  345. }
  346. }
  347. fl->fl_net = hold_net(net);
  348. fl->expires = jiffies;
  349. err = fl6_renew(fl, freq->flr_linger, freq->flr_expires);
  350. if (err)
  351. goto done;
  352. fl->share = freq->flr_share;
  353. addr_type = ipv6_addr_type(&freq->flr_dst);
  354. if ((addr_type & IPV6_ADDR_MAPPED) ||
  355. addr_type == IPV6_ADDR_ANY) {
  356. err = -EINVAL;
  357. goto done;
  358. }
  359. fl->dst = freq->flr_dst;
  360. atomic_set(&fl->users, 1);
  361. switch (fl->share) {
  362. case IPV6_FL_S_EXCL:
  363. case IPV6_FL_S_ANY:
  364. break;
  365. case IPV6_FL_S_PROCESS:
  366. fl->owner.pid = get_task_pid(current, PIDTYPE_PID);
  367. break;
  368. case IPV6_FL_S_USER:
  369. fl->owner.uid = current_euid();
  370. break;
  371. default:
  372. err = -EINVAL;
  373. goto done;
  374. }
  375. return fl;
  376. done:
  377. fl_free(fl);
  378. *err_p = err;
  379. return NULL;
  380. }
  381. static int mem_check(struct sock *sk)
  382. {
  383. struct ipv6_pinfo *np = inet6_sk(sk);
  384. struct ipv6_fl_socklist *sfl;
  385. int room = FL_MAX_SIZE - atomic_read(&fl_size);
  386. int count = 0;
  387. if (room > FL_MAX_SIZE - FL_MAX_PER_SOCK)
  388. return 0;
  389. for_each_sk_fl_rcu(np, sfl)
  390. count++;
  391. if (room <= 0 ||
  392. ((count >= FL_MAX_PER_SOCK ||
  393. (count > 0 && room < FL_MAX_SIZE/2) || room < FL_MAX_SIZE/4) &&
  394. !capable(CAP_NET_ADMIN)))
  395. return -ENOBUFS;
  396. return 0;
  397. }
  398. static inline void fl_link(struct ipv6_pinfo *np, struct ipv6_fl_socklist *sfl,
  399. struct ip6_flowlabel *fl)
  400. {
  401. spin_lock_bh(&ip6_sk_fl_lock);
  402. sfl->fl = fl;
  403. sfl->next = np->ipv6_fl_list;
  404. rcu_assign_pointer(np->ipv6_fl_list, sfl);
  405. spin_unlock_bh(&ip6_sk_fl_lock);
  406. }
  407. int ipv6_flowlabel_opt_get(struct sock *sk, struct in6_flowlabel_req *freq)
  408. {
  409. struct ipv6_pinfo *np = inet6_sk(sk);
  410. struct ipv6_fl_socklist *sfl;
  411. rcu_read_lock_bh();
  412. for_each_sk_fl_rcu(np, sfl) {
  413. if (sfl->fl->label == (np->flow_label & IPV6_FLOWLABEL_MASK)) {
  414. spin_lock_bh(&ip6_fl_lock);
  415. freq->flr_label = sfl->fl->label;
  416. freq->flr_dst = sfl->fl->dst;
  417. freq->flr_share = sfl->fl->share;
  418. freq->flr_expires = (sfl->fl->expires - jiffies) / HZ;
  419. freq->flr_linger = sfl->fl->linger / HZ;
  420. spin_unlock_bh(&ip6_fl_lock);
  421. rcu_read_unlock_bh();
  422. return 0;
  423. }
  424. }
  425. rcu_read_unlock_bh();
  426. return -ENOENT;
  427. }
  428. int ipv6_flowlabel_opt(struct sock *sk, char __user *optval, int optlen)
  429. {
  430. int uninitialized_var(err);
  431. struct net *net = sock_net(sk);
  432. struct ipv6_pinfo *np = inet6_sk(sk);
  433. struct in6_flowlabel_req freq;
  434. struct ipv6_fl_socklist *sfl1=NULL;
  435. struct ipv6_fl_socklist *sfl;
  436. struct ipv6_fl_socklist __rcu **sflp;
  437. struct ip6_flowlabel *fl, *fl1 = NULL;
  438. if (optlen < sizeof(freq))
  439. return -EINVAL;
  440. if (copy_from_user(&freq, optval, sizeof(freq)))
  441. return -EFAULT;
  442. switch (freq.flr_action) {
  443. case IPV6_FL_A_PUT:
  444. spin_lock_bh(&ip6_sk_fl_lock);
  445. for (sflp = &np->ipv6_fl_list;
  446. (sfl = rcu_dereference(*sflp))!=NULL;
  447. sflp = &sfl->next) {
  448. if (sfl->fl->label == freq.flr_label) {
  449. if (freq.flr_label == (np->flow_label&IPV6_FLOWLABEL_MASK))
  450. np->flow_label &= ~IPV6_FLOWLABEL_MASK;
  451. *sflp = rcu_dereference(sfl->next);
  452. spin_unlock_bh(&ip6_sk_fl_lock);
  453. fl_release(sfl->fl);
  454. kfree_rcu(sfl, rcu);
  455. return 0;
  456. }
  457. }
  458. spin_unlock_bh(&ip6_sk_fl_lock);
  459. return -ESRCH;
  460. case IPV6_FL_A_RENEW:
  461. rcu_read_lock_bh();
  462. for_each_sk_fl_rcu(np, sfl) {
  463. if (sfl->fl->label == freq.flr_label) {
  464. err = fl6_renew(sfl->fl, freq.flr_linger, freq.flr_expires);
  465. rcu_read_unlock_bh();
  466. return err;
  467. }
  468. }
  469. rcu_read_unlock_bh();
  470. if (freq.flr_share == IPV6_FL_S_NONE &&
  471. ns_capable(net->user_ns, CAP_NET_ADMIN)) {
  472. fl = fl_lookup(net, freq.flr_label);
  473. if (fl) {
  474. err = fl6_renew(fl, freq.flr_linger, freq.flr_expires);
  475. fl_release(fl);
  476. return err;
  477. }
  478. }
  479. return -ESRCH;
  480. case IPV6_FL_A_GET:
  481. if (freq.flr_label & ~IPV6_FLOWLABEL_MASK)
  482. return -EINVAL;
  483. fl = fl_create(net, sk, &freq, optval, optlen, &err);
  484. if (fl == NULL)
  485. return err;
  486. sfl1 = kmalloc(sizeof(*sfl1), GFP_KERNEL);
  487. if (freq.flr_label) {
  488. err = -EEXIST;
  489. rcu_read_lock_bh();
  490. for_each_sk_fl_rcu(np, sfl) {
  491. if (sfl->fl->label == freq.flr_label) {
  492. if (freq.flr_flags&IPV6_FL_F_EXCL) {
  493. rcu_read_unlock_bh();
  494. goto done;
  495. }
  496. fl1 = sfl->fl;
  497. atomic_inc(&fl1->users);
  498. break;
  499. }
  500. }
  501. rcu_read_unlock_bh();
  502. if (fl1 == NULL)
  503. fl1 = fl_lookup(net, freq.flr_label);
  504. if (fl1) {
  505. recheck:
  506. err = -EEXIST;
  507. if (freq.flr_flags&IPV6_FL_F_EXCL)
  508. goto release;
  509. err = -EPERM;
  510. if (fl1->share == IPV6_FL_S_EXCL ||
  511. fl1->share != fl->share ||
  512. ((fl1->share == IPV6_FL_S_PROCESS) &&
  513. (fl1->owner.pid == fl->owner.pid)) ||
  514. ((fl1->share == IPV6_FL_S_USER) &&
  515. uid_eq(fl1->owner.uid, fl->owner.uid)))
  516. goto release;
  517. err = -ENOMEM;
  518. if (sfl1 == NULL)
  519. goto release;
  520. if (fl->linger > fl1->linger)
  521. fl1->linger = fl->linger;
  522. if ((long)(fl->expires - fl1->expires) > 0)
  523. fl1->expires = fl->expires;
  524. fl_link(np, sfl1, fl1);
  525. fl_free(fl);
  526. return 0;
  527. release:
  528. fl_release(fl1);
  529. goto done;
  530. }
  531. }
  532. err = -ENOENT;
  533. if (!(freq.flr_flags&IPV6_FL_F_CREATE))
  534. goto done;
  535. err = -ENOMEM;
  536. if (sfl1 == NULL || (err = mem_check(sk)) != 0)
  537. goto done;
  538. fl1 = fl_intern(net, fl, freq.flr_label);
  539. if (fl1 != NULL)
  540. goto recheck;
  541. if (!freq.flr_label) {
  542. if (copy_to_user(&((struct in6_flowlabel_req __user *) optval)->flr_label,
  543. &fl->label, sizeof(fl->label))) {
  544. /* Intentionally ignore fault. */
  545. }
  546. }
  547. fl_link(np, sfl1, fl);
  548. return 0;
  549. default:
  550. return -EINVAL;
  551. }
  552. done:
  553. fl_free(fl);
  554. kfree(sfl1);
  555. return err;
  556. }
  557. #ifdef CONFIG_PROC_FS
  558. struct ip6fl_iter_state {
  559. struct seq_net_private p;
  560. struct pid_namespace *pid_ns;
  561. int bucket;
  562. };
  563. #define ip6fl_seq_private(seq) ((struct ip6fl_iter_state *)(seq)->private)
  564. static struct ip6_flowlabel *ip6fl_get_first(struct seq_file *seq)
  565. {
  566. struct ip6_flowlabel *fl = NULL;
  567. struct ip6fl_iter_state *state = ip6fl_seq_private(seq);
  568. struct net *net = seq_file_net(seq);
  569. for (state->bucket = 0; state->bucket <= FL_HASH_MASK; ++state->bucket) {
  570. for_each_fl_rcu(state->bucket, fl) {
  571. if (net_eq(fl->fl_net, net))
  572. goto out;
  573. }
  574. }
  575. fl = NULL;
  576. out:
  577. return fl;
  578. }
  579. static struct ip6_flowlabel *ip6fl_get_next(struct seq_file *seq, struct ip6_flowlabel *fl)
  580. {
  581. struct ip6fl_iter_state *state = ip6fl_seq_private(seq);
  582. struct net *net = seq_file_net(seq);
  583. for_each_fl_continue_rcu(fl) {
  584. if (net_eq(fl->fl_net, net))
  585. goto out;
  586. }
  587. try_again:
  588. if (++state->bucket <= FL_HASH_MASK) {
  589. for_each_fl_rcu(state->bucket, fl) {
  590. if (net_eq(fl->fl_net, net))
  591. goto out;
  592. }
  593. goto try_again;
  594. }
  595. fl = NULL;
  596. out:
  597. return fl;
  598. }
  599. static struct ip6_flowlabel *ip6fl_get_idx(struct seq_file *seq, loff_t pos)
  600. {
  601. struct ip6_flowlabel *fl = ip6fl_get_first(seq);
  602. if (fl)
  603. while (pos && (fl = ip6fl_get_next(seq, fl)) != NULL)
  604. --pos;
  605. return pos ? NULL : fl;
  606. }
  607. static void *ip6fl_seq_start(struct seq_file *seq, loff_t *pos)
  608. __acquires(RCU)
  609. {
  610. rcu_read_lock_bh();
  611. return *pos ? ip6fl_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
  612. }
  613. static void *ip6fl_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  614. {
  615. struct ip6_flowlabel *fl;
  616. if (v == SEQ_START_TOKEN)
  617. fl = ip6fl_get_first(seq);
  618. else
  619. fl = ip6fl_get_next(seq, v);
  620. ++*pos;
  621. return fl;
  622. }
  623. static void ip6fl_seq_stop(struct seq_file *seq, void *v)
  624. __releases(RCU)
  625. {
  626. rcu_read_unlock_bh();
  627. }
  628. static int ip6fl_seq_show(struct seq_file *seq, void *v)
  629. {
  630. struct ip6fl_iter_state *state = ip6fl_seq_private(seq);
  631. if (v == SEQ_START_TOKEN)
  632. seq_printf(seq, "%-5s %-1s %-6s %-6s %-6s %-8s %-32s %s\n",
  633. "Label", "S", "Owner", "Users", "Linger", "Expires", "Dst", "Opt");
  634. else {
  635. struct ip6_flowlabel *fl = v;
  636. seq_printf(seq,
  637. "%05X %-1d %-6d %-6d %-6ld %-8ld %pi6 %-4d\n",
  638. (unsigned int)ntohl(fl->label),
  639. fl->share,
  640. ((fl->share == IPV6_FL_S_PROCESS) ?
  641. pid_nr_ns(fl->owner.pid, state->pid_ns) :
  642. ((fl->share == IPV6_FL_S_USER) ?
  643. from_kuid_munged(seq_user_ns(seq), fl->owner.uid) :
  644. 0)),
  645. atomic_read(&fl->users),
  646. fl->linger/HZ,
  647. (long)(fl->expires - jiffies)/HZ,
  648. &fl->dst,
  649. fl->opt ? fl->opt->opt_nflen : 0);
  650. }
  651. return 0;
  652. }
  653. static const struct seq_operations ip6fl_seq_ops = {
  654. .start = ip6fl_seq_start,
  655. .next = ip6fl_seq_next,
  656. .stop = ip6fl_seq_stop,
  657. .show = ip6fl_seq_show,
  658. };
  659. static int ip6fl_seq_open(struct inode *inode, struct file *file)
  660. {
  661. struct seq_file *seq;
  662. struct ip6fl_iter_state *state;
  663. int err;
  664. err = seq_open_net(inode, file, &ip6fl_seq_ops,
  665. sizeof(struct ip6fl_iter_state));
  666. if (!err) {
  667. seq = file->private_data;
  668. state = ip6fl_seq_private(seq);
  669. rcu_read_lock();
  670. state->pid_ns = get_pid_ns(task_active_pid_ns(current));
  671. rcu_read_unlock();
  672. }
  673. return err;
  674. }
  675. static int ip6fl_seq_release(struct inode *inode, struct file *file)
  676. {
  677. struct seq_file *seq = file->private_data;
  678. struct ip6fl_iter_state *state = ip6fl_seq_private(seq);
  679. put_pid_ns(state->pid_ns);
  680. return seq_release_net(inode, file);
  681. }
  682. static const struct file_operations ip6fl_seq_fops = {
  683. .owner = THIS_MODULE,
  684. .open = ip6fl_seq_open,
  685. .read = seq_read,
  686. .llseek = seq_lseek,
  687. .release = ip6fl_seq_release,
  688. };
  689. static int __net_init ip6_flowlabel_proc_init(struct net *net)
  690. {
  691. if (!proc_create("ip6_flowlabel", S_IRUGO, net->proc_net,
  692. &ip6fl_seq_fops))
  693. return -ENOMEM;
  694. return 0;
  695. }
  696. static void __net_exit ip6_flowlabel_proc_fini(struct net *net)
  697. {
  698. remove_proc_entry("ip6_flowlabel", net->proc_net);
  699. }
  700. #else
  701. static inline int ip6_flowlabel_proc_init(struct net *net)
  702. {
  703. return 0;
  704. }
  705. static inline void ip6_flowlabel_proc_fini(struct net *net)
  706. {
  707. }
  708. #endif
  709. static void __net_exit ip6_flowlabel_net_exit(struct net *net)
  710. {
  711. ip6_fl_purge(net);
  712. ip6_flowlabel_proc_fini(net);
  713. }
  714. static struct pernet_operations ip6_flowlabel_net_ops = {
  715. .init = ip6_flowlabel_proc_init,
  716. .exit = ip6_flowlabel_net_exit,
  717. };
  718. int ip6_flowlabel_init(void)
  719. {
  720. return register_pernet_subsys(&ip6_flowlabel_net_ops);
  721. }
  722. void ip6_flowlabel_cleanup(void)
  723. {
  724. del_timer(&ip6_fl_gc_timer);
  725. unregister_pernet_subsys(&ip6_flowlabel_net_ops);
  726. }