nft_objref.c 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255
  1. // SPDX-License-Identifier: GPL-2.0-only
  2. /*
  3. * Copyright (c) 2012-2016 Pablo Neira Ayuso <pablo@netfilter.org>
  4. */
  5. #include <linux/init.h>
  6. #include <linux/module.h>
  7. #include <linux/skbuff.h>
  8. #include <linux/netlink.h>
  9. #include <linux/netfilter.h>
  10. #include <linux/netfilter/nf_tables.h>
  11. #include <net/netfilter/nf_tables.h>
  12. #define nft_objref_priv(expr) *((struct nft_object **)nft_expr_priv(expr))
  13. static void nft_objref_eval(const struct nft_expr *expr,
  14. struct nft_regs *regs,
  15. const struct nft_pktinfo *pkt)
  16. {
  17. struct nft_object *obj = nft_objref_priv(expr);
  18. obj->ops->eval(obj, regs, pkt);
  19. }
  20. static int nft_objref_init(const struct nft_ctx *ctx,
  21. const struct nft_expr *expr,
  22. const struct nlattr * const tb[])
  23. {
  24. struct nft_object *obj = nft_objref_priv(expr);
  25. u8 genmask = nft_genmask_next(ctx->net);
  26. u32 objtype;
  27. if (!tb[NFTA_OBJREF_IMM_NAME] ||
  28. !tb[NFTA_OBJREF_IMM_TYPE])
  29. return -EINVAL;
  30. objtype = ntohl(nla_get_be32(tb[NFTA_OBJREF_IMM_TYPE]));
  31. obj = nft_obj_lookup(ctx->net, ctx->table,
  32. tb[NFTA_OBJREF_IMM_NAME], objtype,
  33. genmask);
  34. if (IS_ERR(obj))
  35. return -ENOENT;
  36. nft_objref_priv(expr) = obj;
  37. obj->use++;
  38. return 0;
  39. }
  40. static int nft_objref_dump(struct sk_buff *skb, const struct nft_expr *expr)
  41. {
  42. const struct nft_object *obj = nft_objref_priv(expr);
  43. if (nla_put_string(skb, NFTA_OBJREF_IMM_NAME, obj->key.name) ||
  44. nla_put_be32(skb, NFTA_OBJREF_IMM_TYPE,
  45. htonl(obj->ops->type->type)))
  46. goto nla_put_failure;
  47. return 0;
  48. nla_put_failure:
  49. return -1;
  50. }
  51. static void nft_objref_deactivate(const struct nft_ctx *ctx,
  52. const struct nft_expr *expr,
  53. enum nft_trans_phase phase)
  54. {
  55. struct nft_object *obj = nft_objref_priv(expr);
  56. if (phase == NFT_TRANS_COMMIT)
  57. return;
  58. obj->use--;
  59. }
  60. static void nft_objref_activate(const struct nft_ctx *ctx,
  61. const struct nft_expr *expr)
  62. {
  63. struct nft_object *obj = nft_objref_priv(expr);
  64. obj->use++;
  65. }
  66. static struct nft_expr_type nft_objref_type;
  67. static const struct nft_expr_ops nft_objref_ops = {
  68. .type = &nft_objref_type,
  69. .size = NFT_EXPR_SIZE(sizeof(struct nft_object *)),
  70. .eval = nft_objref_eval,
  71. .init = nft_objref_init,
  72. .activate = nft_objref_activate,
  73. .deactivate = nft_objref_deactivate,
  74. .dump = nft_objref_dump,
  75. };
  76. struct nft_objref_map {
  77. struct nft_set *set;
  78. enum nft_registers sreg:8;
  79. struct nft_set_binding binding;
  80. };
  81. static void nft_objref_map_eval(const struct nft_expr *expr,
  82. struct nft_regs *regs,
  83. const struct nft_pktinfo *pkt)
  84. {
  85. struct nft_objref_map *priv = nft_expr_priv(expr);
  86. const struct nft_set *set = priv->set;
  87. const struct nft_set_ext *ext;
  88. struct nft_object *obj;
  89. bool found;
  90. found = set->ops->lookup(nft_net(pkt), set, &regs->data[priv->sreg],
  91. &ext);
  92. if (!found) {
  93. regs->verdict.code = NFT_BREAK;
  94. return;
  95. }
  96. obj = *nft_set_ext_obj(ext);
  97. obj->ops->eval(obj, regs, pkt);
  98. }
  99. static int nft_objref_map_init(const struct nft_ctx *ctx,
  100. const struct nft_expr *expr,
  101. const struct nlattr * const tb[])
  102. {
  103. struct nft_objref_map *priv = nft_expr_priv(expr);
  104. u8 genmask = nft_genmask_next(ctx->net);
  105. struct nft_set *set;
  106. int err;
  107. set = nft_set_lookup_global(ctx->net, ctx->table,
  108. tb[NFTA_OBJREF_SET_NAME],
  109. tb[NFTA_OBJREF_SET_ID], genmask);
  110. if (IS_ERR(set))
  111. return PTR_ERR(set);
  112. if (!(set->flags & NFT_SET_OBJECT))
  113. return -EINVAL;
  114. priv->sreg = nft_parse_register(tb[NFTA_OBJREF_SET_SREG]);
  115. err = nft_validate_register_load(priv->sreg, set->klen);
  116. if (err < 0)
  117. return err;
  118. priv->binding.flags = set->flags & NFT_SET_OBJECT;
  119. err = nf_tables_bind_set(ctx, set, &priv->binding);
  120. if (err < 0)
  121. return err;
  122. priv->set = set;
  123. return 0;
  124. }
  125. static int nft_objref_map_dump(struct sk_buff *skb, const struct nft_expr *expr)
  126. {
  127. const struct nft_objref_map *priv = nft_expr_priv(expr);
  128. if (nft_dump_register(skb, NFTA_OBJREF_SET_SREG, priv->sreg) ||
  129. nla_put_string(skb, NFTA_OBJREF_SET_NAME, priv->set->name))
  130. goto nla_put_failure;
  131. return 0;
  132. nla_put_failure:
  133. return -1;
  134. }
  135. static void nft_objref_map_deactivate(const struct nft_ctx *ctx,
  136. const struct nft_expr *expr,
  137. enum nft_trans_phase phase)
  138. {
  139. struct nft_objref_map *priv = nft_expr_priv(expr);
  140. nf_tables_deactivate_set(ctx, priv->set, &priv->binding, phase);
  141. }
  142. static void nft_objref_map_activate(const struct nft_ctx *ctx,
  143. const struct nft_expr *expr)
  144. {
  145. struct nft_objref_map *priv = nft_expr_priv(expr);
  146. priv->set->use++;
  147. }
  148. static void nft_objref_map_destroy(const struct nft_ctx *ctx,
  149. const struct nft_expr *expr)
  150. {
  151. struct nft_objref_map *priv = nft_expr_priv(expr);
  152. nf_tables_destroy_set(ctx, priv->set);
  153. }
  154. static struct nft_expr_type nft_objref_type;
  155. static const struct nft_expr_ops nft_objref_map_ops = {
  156. .type = &nft_objref_type,
  157. .size = NFT_EXPR_SIZE(sizeof(struct nft_objref_map)),
  158. .eval = nft_objref_map_eval,
  159. .init = nft_objref_map_init,
  160. .activate = nft_objref_map_activate,
  161. .deactivate = nft_objref_map_deactivate,
  162. .destroy = nft_objref_map_destroy,
  163. .dump = nft_objref_map_dump,
  164. };
  165. static const struct nft_expr_ops *
  166. nft_objref_select_ops(const struct nft_ctx *ctx,
  167. const struct nlattr * const tb[])
  168. {
  169. if (tb[NFTA_OBJREF_SET_SREG] &&
  170. (tb[NFTA_OBJREF_SET_NAME] ||
  171. tb[NFTA_OBJREF_SET_ID]))
  172. return &nft_objref_map_ops;
  173. else if (tb[NFTA_OBJREF_IMM_NAME] &&
  174. tb[NFTA_OBJREF_IMM_TYPE])
  175. return &nft_objref_ops;
  176. return ERR_PTR(-EOPNOTSUPP);
  177. }
  178. static const struct nla_policy nft_objref_policy[NFTA_OBJREF_MAX + 1] = {
  179. [NFTA_OBJREF_IMM_NAME] = { .type = NLA_STRING,
  180. .len = NFT_OBJ_MAXNAMELEN - 1 },
  181. [NFTA_OBJREF_IMM_TYPE] = { .type = NLA_U32 },
  182. [NFTA_OBJREF_SET_SREG] = { .type = NLA_U32 },
  183. [NFTA_OBJREF_SET_NAME] = { .type = NLA_STRING,
  184. .len = NFT_SET_MAXNAMELEN - 1 },
  185. [NFTA_OBJREF_SET_ID] = { .type = NLA_U32 },
  186. };
  187. static struct nft_expr_type nft_objref_type __read_mostly = {
  188. .name = "objref",
  189. .select_ops = nft_objref_select_ops,
  190. .policy = nft_objref_policy,
  191. .maxattr = NFTA_OBJREF_MAX,
  192. .owner = THIS_MODULE,
  193. };
  194. static int __init nft_objref_module_init(void)
  195. {
  196. return nft_register_expr(&nft_objref_type);
  197. }
  198. static void __exit nft_objref_module_exit(void)
  199. {
  200. nft_unregister_expr(&nft_objref_type);
  201. }
  202. module_init(nft_objref_module_init);
  203. module_exit(nft_objref_module_exit);
  204. MODULE_LICENSE("GPL");
  205. MODULE_AUTHOR("Pablo Neira Ayuso <pablo@netfilter.org>");
  206. MODULE_ALIAS_NFT_EXPR("objref");
  207. MODULE_DESCRIPTION("nftables stateful object reference module");