act_simple.c 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211
  1. /*
  2. * net/sched/simp.c Simple example of an action
  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: Jamal Hadi Salim (2005)
  10. *
  11. */
  12. #include <linux/module.h>
  13. #include <linux/init.h>
  14. #include <linux/kernel.h>
  15. #include <linux/netdevice.h>
  16. #include <linux/skbuff.h>
  17. #include <linux/rtnetlink.h>
  18. #include <net/pkt_sched.h>
  19. #define TCA_ACT_SIMP 22
  20. #include <linux/tc_act/tc_defact.h>
  21. #include <net/tc_act/tc_defact.h>
  22. #define SIMP_TAB_MASK 7
  23. static struct tcf_common *tcf_simp_ht[SIMP_TAB_MASK + 1];
  24. static u32 simp_idx_gen;
  25. static DEFINE_RWLOCK(simp_lock);
  26. static struct tcf_hashinfo simp_hash_info = {
  27. .htab = tcf_simp_ht,
  28. .hmask = SIMP_TAB_MASK,
  29. .lock = &simp_lock,
  30. };
  31. static int tcf_simp(struct sk_buff *skb, struct tc_action *a, struct tcf_result *res)
  32. {
  33. struct tcf_defact *d = a->priv;
  34. spin_lock(&d->tcf_lock);
  35. d->tcf_tm.lastuse = jiffies;
  36. d->tcf_bstats.bytes += skb->len;
  37. d->tcf_bstats.packets++;
  38. /* print policy string followed by _ then packet count
  39. * Example if this was the 3rd packet and the string was "hello"
  40. * then it would look like "hello_3" (without quotes)
  41. **/
  42. printk("simple: %s_%d\n",
  43. (char *)d->tcfd_defdata, d->tcf_bstats.packets);
  44. spin_unlock(&d->tcf_lock);
  45. return d->tcf_action;
  46. }
  47. static int tcf_simp_release(struct tcf_defact *d, int bind)
  48. {
  49. int ret = 0;
  50. if (d) {
  51. if (bind)
  52. d->tcf_bindcnt--;
  53. d->tcf_refcnt--;
  54. if (d->tcf_bindcnt <= 0 && d->tcf_refcnt <= 0) {
  55. kfree(d->tcfd_defdata);
  56. tcf_hash_destroy(&d->common, &simp_hash_info);
  57. ret = 1;
  58. }
  59. }
  60. return ret;
  61. }
  62. static int alloc_defdata(struct tcf_defact *d, u32 datalen, void *defdata)
  63. {
  64. d->tcfd_defdata = kmemdup(defdata, datalen, GFP_KERNEL);
  65. if (unlikely(!d->tcfd_defdata))
  66. return -ENOMEM;
  67. d->tcfd_datalen = datalen;
  68. return 0;
  69. }
  70. static int realloc_defdata(struct tcf_defact *d, u32 datalen, void *defdata)
  71. {
  72. kfree(d->tcfd_defdata);
  73. return alloc_defdata(d, datalen, defdata);
  74. }
  75. static int tcf_simp_init(struct rtattr *rta, struct rtattr *est,
  76. struct tc_action *a, int ovr, int bind)
  77. {
  78. struct rtattr *tb[TCA_DEF_MAX];
  79. struct tc_defact *parm;
  80. struct tcf_defact *d;
  81. struct tcf_common *pc;
  82. void *defdata;
  83. u32 datalen = 0;
  84. int ret = 0;
  85. if (rta == NULL || rtattr_parse_nested(tb, TCA_DEF_MAX, rta) < 0)
  86. return -EINVAL;
  87. if (tb[TCA_DEF_PARMS - 1] == NULL ||
  88. RTA_PAYLOAD(tb[TCA_DEF_PARMS - 1]) < sizeof(*parm))
  89. return -EINVAL;
  90. parm = RTA_DATA(tb[TCA_DEF_PARMS - 1]);
  91. defdata = RTA_DATA(tb[TCA_DEF_DATA - 1]);
  92. if (defdata == NULL)
  93. return -EINVAL;
  94. datalen = RTA_PAYLOAD(tb[TCA_DEF_DATA - 1]);
  95. if (datalen <= 0)
  96. return -EINVAL;
  97. pc = tcf_hash_check(parm->index, a, bind, &simp_hash_info);
  98. if (!pc) {
  99. pc = tcf_hash_create(parm->index, est, a, sizeof(*d), bind,
  100. &simp_idx_gen, &simp_hash_info);
  101. if (unlikely(!pc))
  102. return -ENOMEM;
  103. d = to_defact(pc);
  104. ret = alloc_defdata(d, datalen, defdata);
  105. if (ret < 0) {
  106. kfree(pc);
  107. return ret;
  108. }
  109. ret = ACT_P_CREATED;
  110. } else {
  111. d = to_defact(pc);
  112. if (!ovr) {
  113. tcf_simp_release(d, bind);
  114. return -EEXIST;
  115. }
  116. realloc_defdata(d, datalen, defdata);
  117. }
  118. spin_lock_bh(&d->tcf_lock);
  119. d->tcf_action = parm->action;
  120. spin_unlock_bh(&d->tcf_lock);
  121. if (ret == ACT_P_CREATED)
  122. tcf_hash_insert(pc, &simp_hash_info);
  123. return ret;
  124. }
  125. static inline int tcf_simp_cleanup(struct tc_action *a, int bind)
  126. {
  127. struct tcf_defact *d = a->priv;
  128. if (d)
  129. return tcf_simp_release(d, bind);
  130. return 0;
  131. }
  132. static inline int tcf_simp_dump(struct sk_buff *skb, struct tc_action *a,
  133. int bind, int ref)
  134. {
  135. unsigned char *b = skb->tail;
  136. struct tcf_defact *d = a->priv;
  137. struct tc_defact opt;
  138. struct tcf_t t;
  139. opt.index = d->tcf_index;
  140. opt.refcnt = d->tcf_refcnt - ref;
  141. opt.bindcnt = d->tcf_bindcnt - bind;
  142. opt.action = d->tcf_action;
  143. RTA_PUT(skb, TCA_DEF_PARMS, sizeof(opt), &opt);
  144. RTA_PUT(skb, TCA_DEF_DATA, d->tcfd_datalen, d->tcfd_defdata);
  145. t.install = jiffies_to_clock_t(jiffies - d->tcf_tm.install);
  146. t.lastuse = jiffies_to_clock_t(jiffies - d->tcf_tm.lastuse);
  147. t.expires = jiffies_to_clock_t(d->tcf_tm.expires);
  148. RTA_PUT(skb, TCA_DEF_TM, sizeof(t), &t);
  149. return skb->len;
  150. rtattr_failure:
  151. skb_trim(skb, b - skb->data);
  152. return -1;
  153. }
  154. static struct tc_action_ops act_simp_ops = {
  155. .kind = "simple",
  156. .hinfo = &simp_hash_info,
  157. .type = TCA_ACT_SIMP,
  158. .capab = TCA_CAP_NONE,
  159. .owner = THIS_MODULE,
  160. .act = tcf_simp,
  161. .dump = tcf_simp_dump,
  162. .cleanup = tcf_simp_cleanup,
  163. .init = tcf_simp_init,
  164. .walk = tcf_generic_walker,
  165. };
  166. MODULE_AUTHOR("Jamal Hadi Salim(2005)");
  167. MODULE_DESCRIPTION("Simple example action");
  168. MODULE_LICENSE("GPL");
  169. static int __init simp_init_module(void)
  170. {
  171. int ret = tcf_register_action(&act_simp_ops);
  172. if (!ret)
  173. printk("Simple TC action Loaded\n");
  174. return ret;
  175. }
  176. static void __exit simp_cleanup_module(void)
  177. {
  178. tcf_unregister_action(&act_simp_ops);
  179. }
  180. module_init(simp_init_module);
  181. module_exit(simp_cleanup_module);