em_meta.c 23 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013
  1. /*
  2. * net/sched/em_meta.c Metadata ematch
  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: Thomas Graf <tgraf@suug.ch>
  10. *
  11. * ==========================================================================
  12. *
  13. * The metadata ematch compares two meta objects where each object
  14. * represents either a meta value stored in the kernel or a static
  15. * value provided by userspace. The objects are not provided by
  16. * userspace itself but rather a definition providing the information
  17. * to build them. Every object is of a certain type which must be
  18. * equal to the object it is being compared to.
  19. *
  20. * The definition of a objects conists of the type (meta type), a
  21. * identifier (meta id) and additional type specific information.
  22. * The meta id is either TCF_META_TYPE_VALUE for values provided by
  23. * userspace or a index to the meta operations table consisting of
  24. * function pointers to type specific meta data collectors returning
  25. * the value of the requested meta value.
  26. *
  27. * lvalue rvalue
  28. * +-----------+ +-----------+
  29. * | type: INT | | type: INT |
  30. * def | id: DEV | | id: VALUE |
  31. * | data: | | data: 3 |
  32. * +-----------+ +-----------+
  33. * | |
  34. * ---> meta_ops[INT][DEV](...) |
  35. * | |
  36. * ----------- |
  37. * V V
  38. * +-----------+ +-----------+
  39. * | type: INT | | type: INT |
  40. * obj | id: DEV | | id: VALUE |
  41. * | data: 2 |<--data got filled out | data: 3 |
  42. * +-----------+ +-----------+
  43. * | |
  44. * --------------> 2 equals 3 <--------------
  45. *
  46. * This is a simplified schema, the complexity varies depending
  47. * on the meta type. Obviously, the length of the data must also
  48. * be provided for non-numeric types.
  49. *
  50. * Additionally, type dependent modifiers such as shift operators
  51. * or mask may be applied to extend the functionaliy. As of now,
  52. * the variable length type supports shifting the byte string to
  53. * the right, eating up any number of octets and thus supporting
  54. * wildcard interface name comparisons such as "ppp%" matching
  55. * ppp0..9.
  56. *
  57. * NOTE: Certain meta values depend on other subsystems and are
  58. * only available if that subsystem is enabled in the kernel.
  59. */
  60. #include <linux/slab.h>
  61. #include <linux/module.h>
  62. #include <linux/types.h>
  63. #include <linux/kernel.h>
  64. #include <linux/sched.h>
  65. #include <linux/string.h>
  66. #include <linux/skbuff.h>
  67. #include <linux/random.h>
  68. #include <linux/if_vlan.h>
  69. #include <linux/tc_ematch/tc_em_meta.h>
  70. #include <net/dst.h>
  71. #include <net/route.h>
  72. #include <net/pkt_cls.h>
  73. #include <net/sock.h>
  74. struct meta_obj {
  75. unsigned long value;
  76. unsigned int len;
  77. };
  78. struct meta_value {
  79. struct tcf_meta_val hdr;
  80. unsigned long val;
  81. unsigned int len;
  82. };
  83. struct meta_match {
  84. struct meta_value lvalue;
  85. struct meta_value rvalue;
  86. };
  87. static inline int meta_id(struct meta_value *v)
  88. {
  89. return TCF_META_ID(v->hdr.kind);
  90. }
  91. static inline int meta_type(struct meta_value *v)
  92. {
  93. return TCF_META_TYPE(v->hdr.kind);
  94. }
  95. #define META_COLLECTOR(FUNC) static void meta_##FUNC(struct sk_buff *skb, \
  96. struct tcf_pkt_info *info, struct meta_value *v, \
  97. struct meta_obj *dst, int *err)
  98. /**************************************************************************
  99. * System status & misc
  100. **************************************************************************/
  101. META_COLLECTOR(int_random)
  102. {
  103. get_random_bytes(&dst->value, sizeof(dst->value));
  104. }
  105. static inline unsigned long fixed_loadavg(int load)
  106. {
  107. int rnd_load = load + (FIXED_1/200);
  108. int rnd_frac = ((rnd_load & (FIXED_1-1)) * 100) >> FSHIFT;
  109. return ((rnd_load >> FSHIFT) * 100) + rnd_frac;
  110. }
  111. META_COLLECTOR(int_loadavg_0)
  112. {
  113. dst->value = fixed_loadavg(avenrun[0]);
  114. }
  115. META_COLLECTOR(int_loadavg_1)
  116. {
  117. dst->value = fixed_loadavg(avenrun[1]);
  118. }
  119. META_COLLECTOR(int_loadavg_2)
  120. {
  121. dst->value = fixed_loadavg(avenrun[2]);
  122. }
  123. /**************************************************************************
  124. * Device names & indices
  125. **************************************************************************/
  126. static inline int int_dev(struct net_device *dev, struct meta_obj *dst)
  127. {
  128. if (unlikely(dev == NULL))
  129. return -1;
  130. dst->value = dev->ifindex;
  131. return 0;
  132. }
  133. static inline int var_dev(struct net_device *dev, struct meta_obj *dst)
  134. {
  135. if (unlikely(dev == NULL))
  136. return -1;
  137. dst->value = (unsigned long) dev->name;
  138. dst->len = strlen(dev->name);
  139. return 0;
  140. }
  141. META_COLLECTOR(int_dev)
  142. {
  143. *err = int_dev(skb->dev, dst);
  144. }
  145. META_COLLECTOR(var_dev)
  146. {
  147. *err = var_dev(skb->dev, dst);
  148. }
  149. /**************************************************************************
  150. * vlan tag
  151. **************************************************************************/
  152. META_COLLECTOR(int_vlan_tag)
  153. {
  154. unsigned short tag;
  155. if (skb_vlan_tag_present(skb))
  156. dst->value = skb_vlan_tag_get(skb);
  157. else if (!__vlan_get_tag(skb, &tag))
  158. dst->value = tag;
  159. else
  160. *err = -1;
  161. }
  162. /**************************************************************************
  163. * skb attributes
  164. **************************************************************************/
  165. META_COLLECTOR(int_priority)
  166. {
  167. dst->value = skb->priority;
  168. }
  169. META_COLLECTOR(int_protocol)
  170. {
  171. /* Let userspace take care of the byte ordering */
  172. dst->value = tc_skb_protocol(skb);
  173. }
  174. META_COLLECTOR(int_pkttype)
  175. {
  176. dst->value = skb->pkt_type;
  177. }
  178. META_COLLECTOR(int_pktlen)
  179. {
  180. dst->value = skb->len;
  181. }
  182. META_COLLECTOR(int_datalen)
  183. {
  184. dst->value = skb->data_len;
  185. }
  186. META_COLLECTOR(int_maclen)
  187. {
  188. dst->value = skb->mac_len;
  189. }
  190. META_COLLECTOR(int_rxhash)
  191. {
  192. dst->value = skb_get_hash(skb);
  193. }
  194. /**************************************************************************
  195. * Netfilter
  196. **************************************************************************/
  197. META_COLLECTOR(int_mark)
  198. {
  199. dst->value = skb->mark;
  200. }
  201. /**************************************************************************
  202. * Traffic Control
  203. **************************************************************************/
  204. META_COLLECTOR(int_tcindex)
  205. {
  206. dst->value = skb->tc_index;
  207. }
  208. /**************************************************************************
  209. * Routing
  210. **************************************************************************/
  211. META_COLLECTOR(int_rtclassid)
  212. {
  213. if (unlikely(skb_dst(skb) == NULL))
  214. *err = -1;
  215. else
  216. #ifdef CONFIG_IP_ROUTE_CLASSID
  217. dst->value = skb_dst(skb)->tclassid;
  218. #else
  219. dst->value = 0;
  220. #endif
  221. }
  222. META_COLLECTOR(int_rtiif)
  223. {
  224. if (unlikely(skb_rtable(skb) == NULL))
  225. *err = -1;
  226. else
  227. dst->value = inet_iif(skb);
  228. }
  229. /**************************************************************************
  230. * Socket Attributes
  231. **************************************************************************/
  232. #define skip_nonlocal(skb) \
  233. (unlikely(skb->sk == NULL))
  234. META_COLLECTOR(int_sk_family)
  235. {
  236. if (skip_nonlocal(skb)) {
  237. *err = -1;
  238. return;
  239. }
  240. dst->value = skb->sk->sk_family;
  241. }
  242. META_COLLECTOR(int_sk_state)
  243. {
  244. if (skip_nonlocal(skb)) {
  245. *err = -1;
  246. return;
  247. }
  248. dst->value = skb->sk->sk_state;
  249. }
  250. META_COLLECTOR(int_sk_reuse)
  251. {
  252. if (skip_nonlocal(skb)) {
  253. *err = -1;
  254. return;
  255. }
  256. dst->value = skb->sk->sk_reuse;
  257. }
  258. META_COLLECTOR(int_sk_bound_if)
  259. {
  260. if (skip_nonlocal(skb)) {
  261. *err = -1;
  262. return;
  263. }
  264. /* No error if bound_dev_if is 0, legal userspace check */
  265. dst->value = skb->sk->sk_bound_dev_if;
  266. }
  267. META_COLLECTOR(var_sk_bound_if)
  268. {
  269. if (skip_nonlocal(skb)) {
  270. *err = -1;
  271. return;
  272. }
  273. if (skb->sk->sk_bound_dev_if == 0) {
  274. dst->value = (unsigned long) "any";
  275. dst->len = 3;
  276. } else {
  277. struct net_device *dev;
  278. rcu_read_lock();
  279. dev = dev_get_by_index_rcu(sock_net(skb->sk),
  280. skb->sk->sk_bound_dev_if);
  281. *err = var_dev(dev, dst);
  282. rcu_read_unlock();
  283. }
  284. }
  285. META_COLLECTOR(int_sk_refcnt)
  286. {
  287. if (skip_nonlocal(skb)) {
  288. *err = -1;
  289. return;
  290. }
  291. dst->value = atomic_read(&skb->sk->sk_refcnt);
  292. }
  293. META_COLLECTOR(int_sk_rcvbuf)
  294. {
  295. const struct sock *sk = skb_to_full_sk(skb);
  296. if (!sk) {
  297. *err = -1;
  298. return;
  299. }
  300. dst->value = sk->sk_rcvbuf;
  301. }
  302. META_COLLECTOR(int_sk_shutdown)
  303. {
  304. const struct sock *sk = skb_to_full_sk(skb);
  305. if (!sk) {
  306. *err = -1;
  307. return;
  308. }
  309. dst->value = sk->sk_shutdown;
  310. }
  311. META_COLLECTOR(int_sk_proto)
  312. {
  313. const struct sock *sk = skb_to_full_sk(skb);
  314. if (!sk) {
  315. *err = -1;
  316. return;
  317. }
  318. dst->value = sk->sk_protocol;
  319. }
  320. META_COLLECTOR(int_sk_type)
  321. {
  322. const struct sock *sk = skb_to_full_sk(skb);
  323. if (!sk) {
  324. *err = -1;
  325. return;
  326. }
  327. dst->value = sk->sk_type;
  328. }
  329. META_COLLECTOR(int_sk_rmem_alloc)
  330. {
  331. const struct sock *sk = skb_to_full_sk(skb);
  332. if (!sk) {
  333. *err = -1;
  334. return;
  335. }
  336. dst->value = sk_rmem_alloc_get(sk);
  337. }
  338. META_COLLECTOR(int_sk_wmem_alloc)
  339. {
  340. const struct sock *sk = skb_to_full_sk(skb);
  341. if (!sk) {
  342. *err = -1;
  343. return;
  344. }
  345. dst->value = sk_wmem_alloc_get(sk);
  346. }
  347. META_COLLECTOR(int_sk_omem_alloc)
  348. {
  349. const struct sock *sk = skb_to_full_sk(skb);
  350. if (!sk) {
  351. *err = -1;
  352. return;
  353. }
  354. dst->value = atomic_read(&sk->sk_omem_alloc);
  355. }
  356. META_COLLECTOR(int_sk_rcv_qlen)
  357. {
  358. const struct sock *sk = skb_to_full_sk(skb);
  359. if (!sk) {
  360. *err = -1;
  361. return;
  362. }
  363. dst->value = sk->sk_receive_queue.qlen;
  364. }
  365. META_COLLECTOR(int_sk_snd_qlen)
  366. {
  367. const struct sock *sk = skb_to_full_sk(skb);
  368. if (!sk) {
  369. *err = -1;
  370. return;
  371. }
  372. dst->value = sk->sk_write_queue.qlen;
  373. }
  374. META_COLLECTOR(int_sk_wmem_queued)
  375. {
  376. const struct sock *sk = skb_to_full_sk(skb);
  377. if (!sk) {
  378. *err = -1;
  379. return;
  380. }
  381. dst->value = sk->sk_wmem_queued;
  382. }
  383. META_COLLECTOR(int_sk_fwd_alloc)
  384. {
  385. const struct sock *sk = skb_to_full_sk(skb);
  386. if (!sk) {
  387. *err = -1;
  388. return;
  389. }
  390. dst->value = sk->sk_forward_alloc;
  391. }
  392. META_COLLECTOR(int_sk_sndbuf)
  393. {
  394. const struct sock *sk = skb_to_full_sk(skb);
  395. if (!sk) {
  396. *err = -1;
  397. return;
  398. }
  399. dst->value = sk->sk_sndbuf;
  400. }
  401. META_COLLECTOR(int_sk_alloc)
  402. {
  403. const struct sock *sk = skb_to_full_sk(skb);
  404. if (!sk) {
  405. *err = -1;
  406. return;
  407. }
  408. dst->value = (__force int) sk->sk_allocation;
  409. }
  410. META_COLLECTOR(int_sk_hash)
  411. {
  412. if (skip_nonlocal(skb)) {
  413. *err = -1;
  414. return;
  415. }
  416. dst->value = skb->sk->sk_hash;
  417. }
  418. META_COLLECTOR(int_sk_lingertime)
  419. {
  420. const struct sock *sk = skb_to_full_sk(skb);
  421. if (!sk) {
  422. *err = -1;
  423. return;
  424. }
  425. dst->value = sk->sk_lingertime / HZ;
  426. }
  427. META_COLLECTOR(int_sk_err_qlen)
  428. {
  429. const struct sock *sk = skb_to_full_sk(skb);
  430. if (!sk) {
  431. *err = -1;
  432. return;
  433. }
  434. dst->value = sk->sk_error_queue.qlen;
  435. }
  436. META_COLLECTOR(int_sk_ack_bl)
  437. {
  438. const struct sock *sk = skb_to_full_sk(skb);
  439. if (!sk) {
  440. *err = -1;
  441. return;
  442. }
  443. dst->value = sk->sk_ack_backlog;
  444. }
  445. META_COLLECTOR(int_sk_max_ack_bl)
  446. {
  447. const struct sock *sk = skb_to_full_sk(skb);
  448. if (!sk) {
  449. *err = -1;
  450. return;
  451. }
  452. dst->value = sk->sk_max_ack_backlog;
  453. }
  454. META_COLLECTOR(int_sk_prio)
  455. {
  456. const struct sock *sk = skb_to_full_sk(skb);
  457. if (!sk) {
  458. *err = -1;
  459. return;
  460. }
  461. dst->value = sk->sk_priority;
  462. }
  463. META_COLLECTOR(int_sk_rcvlowat)
  464. {
  465. const struct sock *sk = skb_to_full_sk(skb);
  466. if (!sk) {
  467. *err = -1;
  468. return;
  469. }
  470. dst->value = sk->sk_rcvlowat;
  471. }
  472. META_COLLECTOR(int_sk_rcvtimeo)
  473. {
  474. const struct sock *sk = skb_to_full_sk(skb);
  475. if (!sk) {
  476. *err = -1;
  477. return;
  478. }
  479. dst->value = sk->sk_rcvtimeo / HZ;
  480. }
  481. META_COLLECTOR(int_sk_sndtimeo)
  482. {
  483. const struct sock *sk = skb_to_full_sk(skb);
  484. if (!sk) {
  485. *err = -1;
  486. return;
  487. }
  488. dst->value = sk->sk_sndtimeo / HZ;
  489. }
  490. META_COLLECTOR(int_sk_sendmsg_off)
  491. {
  492. const struct sock *sk = skb_to_full_sk(skb);
  493. if (!sk) {
  494. *err = -1;
  495. return;
  496. }
  497. dst->value = sk->sk_frag.offset;
  498. }
  499. META_COLLECTOR(int_sk_write_pend)
  500. {
  501. const struct sock *sk = skb_to_full_sk(skb);
  502. if (!sk) {
  503. *err = -1;
  504. return;
  505. }
  506. dst->value = sk->sk_write_pending;
  507. }
  508. /**************************************************************************
  509. * Meta value collectors assignment table
  510. **************************************************************************/
  511. struct meta_ops {
  512. void (*get)(struct sk_buff *, struct tcf_pkt_info *,
  513. struct meta_value *, struct meta_obj *, int *);
  514. };
  515. #define META_ID(name) TCF_META_ID_##name
  516. #define META_FUNC(name) { .get = meta_##name }
  517. /* Meta value operations table listing all meta value collectors and
  518. * assigns them to a type and meta id. */
  519. static struct meta_ops __meta_ops[TCF_META_TYPE_MAX + 1][TCF_META_ID_MAX + 1] = {
  520. [TCF_META_TYPE_VAR] = {
  521. [META_ID(DEV)] = META_FUNC(var_dev),
  522. [META_ID(SK_BOUND_IF)] = META_FUNC(var_sk_bound_if),
  523. },
  524. [TCF_META_TYPE_INT] = {
  525. [META_ID(RANDOM)] = META_FUNC(int_random),
  526. [META_ID(LOADAVG_0)] = META_FUNC(int_loadavg_0),
  527. [META_ID(LOADAVG_1)] = META_FUNC(int_loadavg_1),
  528. [META_ID(LOADAVG_2)] = META_FUNC(int_loadavg_2),
  529. [META_ID(DEV)] = META_FUNC(int_dev),
  530. [META_ID(PRIORITY)] = META_FUNC(int_priority),
  531. [META_ID(PROTOCOL)] = META_FUNC(int_protocol),
  532. [META_ID(PKTTYPE)] = META_FUNC(int_pkttype),
  533. [META_ID(PKTLEN)] = META_FUNC(int_pktlen),
  534. [META_ID(DATALEN)] = META_FUNC(int_datalen),
  535. [META_ID(MACLEN)] = META_FUNC(int_maclen),
  536. [META_ID(NFMARK)] = META_FUNC(int_mark),
  537. [META_ID(TCINDEX)] = META_FUNC(int_tcindex),
  538. [META_ID(RTCLASSID)] = META_FUNC(int_rtclassid),
  539. [META_ID(RTIIF)] = META_FUNC(int_rtiif),
  540. [META_ID(SK_FAMILY)] = META_FUNC(int_sk_family),
  541. [META_ID(SK_STATE)] = META_FUNC(int_sk_state),
  542. [META_ID(SK_REUSE)] = META_FUNC(int_sk_reuse),
  543. [META_ID(SK_BOUND_IF)] = META_FUNC(int_sk_bound_if),
  544. [META_ID(SK_REFCNT)] = META_FUNC(int_sk_refcnt),
  545. [META_ID(SK_RCVBUF)] = META_FUNC(int_sk_rcvbuf),
  546. [META_ID(SK_SNDBUF)] = META_FUNC(int_sk_sndbuf),
  547. [META_ID(SK_SHUTDOWN)] = META_FUNC(int_sk_shutdown),
  548. [META_ID(SK_PROTO)] = META_FUNC(int_sk_proto),
  549. [META_ID(SK_TYPE)] = META_FUNC(int_sk_type),
  550. [META_ID(SK_RMEM_ALLOC)] = META_FUNC(int_sk_rmem_alloc),
  551. [META_ID(SK_WMEM_ALLOC)] = META_FUNC(int_sk_wmem_alloc),
  552. [META_ID(SK_OMEM_ALLOC)] = META_FUNC(int_sk_omem_alloc),
  553. [META_ID(SK_WMEM_QUEUED)] = META_FUNC(int_sk_wmem_queued),
  554. [META_ID(SK_RCV_QLEN)] = META_FUNC(int_sk_rcv_qlen),
  555. [META_ID(SK_SND_QLEN)] = META_FUNC(int_sk_snd_qlen),
  556. [META_ID(SK_ERR_QLEN)] = META_FUNC(int_sk_err_qlen),
  557. [META_ID(SK_FORWARD_ALLOCS)] = META_FUNC(int_sk_fwd_alloc),
  558. [META_ID(SK_ALLOCS)] = META_FUNC(int_sk_alloc),
  559. [META_ID(SK_HASH)] = META_FUNC(int_sk_hash),
  560. [META_ID(SK_LINGERTIME)] = META_FUNC(int_sk_lingertime),
  561. [META_ID(SK_ACK_BACKLOG)] = META_FUNC(int_sk_ack_bl),
  562. [META_ID(SK_MAX_ACK_BACKLOG)] = META_FUNC(int_sk_max_ack_bl),
  563. [META_ID(SK_PRIO)] = META_FUNC(int_sk_prio),
  564. [META_ID(SK_RCVLOWAT)] = META_FUNC(int_sk_rcvlowat),
  565. [META_ID(SK_RCVTIMEO)] = META_FUNC(int_sk_rcvtimeo),
  566. [META_ID(SK_SNDTIMEO)] = META_FUNC(int_sk_sndtimeo),
  567. [META_ID(SK_SENDMSG_OFF)] = META_FUNC(int_sk_sendmsg_off),
  568. [META_ID(SK_WRITE_PENDING)] = META_FUNC(int_sk_write_pend),
  569. [META_ID(VLAN_TAG)] = META_FUNC(int_vlan_tag),
  570. [META_ID(RXHASH)] = META_FUNC(int_rxhash),
  571. }
  572. };
  573. static inline struct meta_ops *meta_ops(struct meta_value *val)
  574. {
  575. return &__meta_ops[meta_type(val)][meta_id(val)];
  576. }
  577. /**************************************************************************
  578. * Type specific operations for TCF_META_TYPE_VAR
  579. **************************************************************************/
  580. static int meta_var_compare(struct meta_obj *a, struct meta_obj *b)
  581. {
  582. int r = a->len - b->len;
  583. if (r == 0)
  584. r = memcmp((void *) a->value, (void *) b->value, a->len);
  585. return r;
  586. }
  587. static int meta_var_change(struct meta_value *dst, struct nlattr *nla)
  588. {
  589. int len = nla_len(nla);
  590. dst->val = (unsigned long)kmemdup(nla_data(nla), len, GFP_KERNEL);
  591. if (dst->val == 0UL)
  592. return -ENOMEM;
  593. dst->len = len;
  594. return 0;
  595. }
  596. static void meta_var_destroy(struct meta_value *v)
  597. {
  598. kfree((void *) v->val);
  599. }
  600. static void meta_var_apply_extras(struct meta_value *v,
  601. struct meta_obj *dst)
  602. {
  603. int shift = v->hdr.shift;
  604. if (shift && shift < dst->len)
  605. dst->len -= shift;
  606. }
  607. static int meta_var_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
  608. {
  609. if (v->val && v->len &&
  610. nla_put(skb, tlv, v->len, (void *) v->val))
  611. goto nla_put_failure;
  612. return 0;
  613. nla_put_failure:
  614. return -1;
  615. }
  616. /**************************************************************************
  617. * Type specific operations for TCF_META_TYPE_INT
  618. **************************************************************************/
  619. static int meta_int_compare(struct meta_obj *a, struct meta_obj *b)
  620. {
  621. /* Let gcc optimize it, the unlikely is not really based on
  622. * some numbers but jump free code for mismatches seems
  623. * more logical. */
  624. if (unlikely(a->value == b->value))
  625. return 0;
  626. else if (a->value < b->value)
  627. return -1;
  628. else
  629. return 1;
  630. }
  631. static int meta_int_change(struct meta_value *dst, struct nlattr *nla)
  632. {
  633. if (nla_len(nla) >= sizeof(unsigned long)) {
  634. dst->val = *(unsigned long *) nla_data(nla);
  635. dst->len = sizeof(unsigned long);
  636. } else if (nla_len(nla) == sizeof(u32)) {
  637. dst->val = nla_get_u32(nla);
  638. dst->len = sizeof(u32);
  639. } else
  640. return -EINVAL;
  641. return 0;
  642. }
  643. static void meta_int_apply_extras(struct meta_value *v,
  644. struct meta_obj *dst)
  645. {
  646. if (v->hdr.shift)
  647. dst->value >>= v->hdr.shift;
  648. if (v->val)
  649. dst->value &= v->val;
  650. }
  651. static int meta_int_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
  652. {
  653. if (v->len == sizeof(unsigned long)) {
  654. if (nla_put(skb, tlv, sizeof(unsigned long), &v->val))
  655. goto nla_put_failure;
  656. } else if (v->len == sizeof(u32)) {
  657. if (nla_put_u32(skb, tlv, v->val))
  658. goto nla_put_failure;
  659. }
  660. return 0;
  661. nla_put_failure:
  662. return -1;
  663. }
  664. /**************************************************************************
  665. * Type specific operations table
  666. **************************************************************************/
  667. struct meta_type_ops {
  668. void (*destroy)(struct meta_value *);
  669. int (*compare)(struct meta_obj *, struct meta_obj *);
  670. int (*change)(struct meta_value *, struct nlattr *);
  671. void (*apply_extras)(struct meta_value *, struct meta_obj *);
  672. int (*dump)(struct sk_buff *, struct meta_value *, int);
  673. };
  674. static const struct meta_type_ops __meta_type_ops[TCF_META_TYPE_MAX + 1] = {
  675. [TCF_META_TYPE_VAR] = {
  676. .destroy = meta_var_destroy,
  677. .compare = meta_var_compare,
  678. .change = meta_var_change,
  679. .apply_extras = meta_var_apply_extras,
  680. .dump = meta_var_dump
  681. },
  682. [TCF_META_TYPE_INT] = {
  683. .compare = meta_int_compare,
  684. .change = meta_int_change,
  685. .apply_extras = meta_int_apply_extras,
  686. .dump = meta_int_dump
  687. }
  688. };
  689. static inline const struct meta_type_ops *meta_type_ops(struct meta_value *v)
  690. {
  691. return &__meta_type_ops[meta_type(v)];
  692. }
  693. /**************************************************************************
  694. * Core
  695. **************************************************************************/
  696. static int meta_get(struct sk_buff *skb, struct tcf_pkt_info *info,
  697. struct meta_value *v, struct meta_obj *dst)
  698. {
  699. int err = 0;
  700. if (meta_id(v) == TCF_META_ID_VALUE) {
  701. dst->value = v->val;
  702. dst->len = v->len;
  703. return 0;
  704. }
  705. meta_ops(v)->get(skb, info, v, dst, &err);
  706. if (err < 0)
  707. return err;
  708. if (meta_type_ops(v)->apply_extras)
  709. meta_type_ops(v)->apply_extras(v, dst);
  710. return 0;
  711. }
  712. static int em_meta_match(struct sk_buff *skb, struct tcf_ematch *m,
  713. struct tcf_pkt_info *info)
  714. {
  715. int r;
  716. struct meta_match *meta = (struct meta_match *) m->data;
  717. struct meta_obj l_value, r_value;
  718. if (meta_get(skb, info, &meta->lvalue, &l_value) < 0 ||
  719. meta_get(skb, info, &meta->rvalue, &r_value) < 0)
  720. return 0;
  721. r = meta_type_ops(&meta->lvalue)->compare(&l_value, &r_value);
  722. switch (meta->lvalue.hdr.op) {
  723. case TCF_EM_OPND_EQ:
  724. return !r;
  725. case TCF_EM_OPND_LT:
  726. return r < 0;
  727. case TCF_EM_OPND_GT:
  728. return r > 0;
  729. }
  730. return 0;
  731. }
  732. static void meta_delete(struct meta_match *meta)
  733. {
  734. if (meta) {
  735. const struct meta_type_ops *ops = meta_type_ops(&meta->lvalue);
  736. if (ops && ops->destroy) {
  737. ops->destroy(&meta->lvalue);
  738. ops->destroy(&meta->rvalue);
  739. }
  740. }
  741. kfree(meta);
  742. }
  743. static inline int meta_change_data(struct meta_value *dst, struct nlattr *nla)
  744. {
  745. if (nla) {
  746. if (nla_len(nla) == 0)
  747. return -EINVAL;
  748. return meta_type_ops(dst)->change(dst, nla);
  749. }
  750. return 0;
  751. }
  752. static inline int meta_is_supported(struct meta_value *val)
  753. {
  754. return !meta_id(val) || meta_ops(val)->get;
  755. }
  756. static const struct nla_policy meta_policy[TCA_EM_META_MAX + 1] = {
  757. [TCA_EM_META_HDR] = { .len = sizeof(struct tcf_meta_hdr) },
  758. };
  759. static int em_meta_change(struct net *net, void *data, int len,
  760. struct tcf_ematch *m)
  761. {
  762. int err;
  763. struct nlattr *tb[TCA_EM_META_MAX + 1];
  764. struct tcf_meta_hdr *hdr;
  765. struct meta_match *meta = NULL;
  766. err = nla_parse(tb, TCA_EM_META_MAX, data, len, meta_policy);
  767. if (err < 0)
  768. goto errout;
  769. err = -EINVAL;
  770. if (tb[TCA_EM_META_HDR] == NULL)
  771. goto errout;
  772. hdr = nla_data(tb[TCA_EM_META_HDR]);
  773. if (TCF_META_TYPE(hdr->left.kind) != TCF_META_TYPE(hdr->right.kind) ||
  774. TCF_META_TYPE(hdr->left.kind) > TCF_META_TYPE_MAX ||
  775. TCF_META_ID(hdr->left.kind) > TCF_META_ID_MAX ||
  776. TCF_META_ID(hdr->right.kind) > TCF_META_ID_MAX)
  777. goto errout;
  778. meta = kzalloc(sizeof(*meta), GFP_KERNEL);
  779. if (meta == NULL) {
  780. err = -ENOMEM;
  781. goto errout;
  782. }
  783. memcpy(&meta->lvalue.hdr, &hdr->left, sizeof(hdr->left));
  784. memcpy(&meta->rvalue.hdr, &hdr->right, sizeof(hdr->right));
  785. if (!meta_is_supported(&meta->lvalue) ||
  786. !meta_is_supported(&meta->rvalue)) {
  787. err = -EOPNOTSUPP;
  788. goto errout;
  789. }
  790. if (meta_change_data(&meta->lvalue, tb[TCA_EM_META_LVALUE]) < 0 ||
  791. meta_change_data(&meta->rvalue, tb[TCA_EM_META_RVALUE]) < 0)
  792. goto errout;
  793. m->datalen = sizeof(*meta);
  794. m->data = (unsigned long) meta;
  795. err = 0;
  796. errout:
  797. if (err && meta)
  798. meta_delete(meta);
  799. return err;
  800. }
  801. static void em_meta_destroy(struct tcf_ematch *m)
  802. {
  803. if (m)
  804. meta_delete((struct meta_match *) m->data);
  805. }
  806. static int em_meta_dump(struct sk_buff *skb, struct tcf_ematch *em)
  807. {
  808. struct meta_match *meta = (struct meta_match *) em->data;
  809. struct tcf_meta_hdr hdr;
  810. const struct meta_type_ops *ops;
  811. memset(&hdr, 0, sizeof(hdr));
  812. memcpy(&hdr.left, &meta->lvalue.hdr, sizeof(hdr.left));
  813. memcpy(&hdr.right, &meta->rvalue.hdr, sizeof(hdr.right));
  814. if (nla_put(skb, TCA_EM_META_HDR, sizeof(hdr), &hdr))
  815. goto nla_put_failure;
  816. ops = meta_type_ops(&meta->lvalue);
  817. if (ops->dump(skb, &meta->lvalue, TCA_EM_META_LVALUE) < 0 ||
  818. ops->dump(skb, &meta->rvalue, TCA_EM_META_RVALUE) < 0)
  819. goto nla_put_failure;
  820. return 0;
  821. nla_put_failure:
  822. return -1;
  823. }
  824. static struct tcf_ematch_ops em_meta_ops = {
  825. .kind = TCF_EM_META,
  826. .change = em_meta_change,
  827. .match = em_meta_match,
  828. .destroy = em_meta_destroy,
  829. .dump = em_meta_dump,
  830. .owner = THIS_MODULE,
  831. .link = LIST_HEAD_INIT(em_meta_ops.link)
  832. };
  833. static int __init init_em_meta(void)
  834. {
  835. return tcf_em_register(&em_meta_ops);
  836. }
  837. static void __exit exit_em_meta(void)
  838. {
  839. tcf_em_unregister(&em_meta_ops);
  840. }
  841. MODULE_LICENSE("GPL");
  842. module_init(init_em_meta);
  843. module_exit(exit_em_meta);
  844. MODULE_ALIAS_TCF_EMATCH(TCF_EM_META);