2 * Copyright (c) 2008-2009 Patrick McHardy <kaber@trash.net>
3 * Copyright (c) 2012 Pablo Neira Ayuso <pablo@netfilter.org>
4 * Copyright (c) 2012 Intel Corporation
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms and conditions of the GNU General Public License,
8 * version 2, as published by the Free Software Foundation.
12 #include <linux/module.h>
13 #include <linux/init.h>
14 #include <linux/skbuff.h>
16 #include <linux/string.h>
17 #include <linux/netlink.h>
18 #include <linux/netfilter.h>
19 #include <linux/netfilter_ipv4.h>
20 #include <linux/netfilter/nfnetlink.h>
21 #include <linux/netfilter/nf_tables.h>
22 #include <net/netfilter/nf_conntrack.h>
23 #include <net/netfilter/nf_nat.h>
24 #include <net/netfilter/nf_tables.h>
28 enum nft_registers sreg_addr_min:8;
29 enum nft_registers sreg_addr_max:8;
30 enum nft_registers sreg_proto_min:8;
31 enum nft_registers sreg_proto_max:8;
32 enum nf_nat_manip_type type:8;
37 static void nft_nat_eval(const struct nft_expr *expr,
38 struct nft_regs *regs,
39 const struct nft_pktinfo *pkt)
41 const struct nft_nat *priv = nft_expr_priv(expr);
42 enum ip_conntrack_info ctinfo;
43 struct nf_conn *ct = nf_ct_get(pkt->skb, &ctinfo);
44 struct nf_nat_range2 range;
46 memset(&range, 0, sizeof(range));
47 if (priv->sreg_addr_min) {
48 if (priv->family == AF_INET) {
49 range.min_addr.ip = (__force __be32)
50 regs->data[priv->sreg_addr_min];
51 range.max_addr.ip = (__force __be32)
52 regs->data[priv->sreg_addr_max];
55 memcpy(range.min_addr.ip6,
56 ®s->data[priv->sreg_addr_min],
57 sizeof(range.min_addr.ip6));
58 memcpy(range.max_addr.ip6,
59 ®s->data[priv->sreg_addr_max],
60 sizeof(range.max_addr.ip6));
62 range.flags |= NF_NAT_RANGE_MAP_IPS;
65 if (priv->sreg_proto_min) {
66 range.min_proto.all = (__force __be16)nft_reg_load16(
67 ®s->data[priv->sreg_proto_min]);
68 range.max_proto.all = (__force __be16)nft_reg_load16(
69 ®s->data[priv->sreg_proto_max]);
70 range.flags |= NF_NAT_RANGE_PROTO_SPECIFIED;
73 range.flags |= priv->flags;
75 regs->verdict.code = nf_nat_setup_info(ct, &range, priv->type);
78 static const struct nla_policy nft_nat_policy[NFTA_NAT_MAX + 1] = {
79 [NFTA_NAT_TYPE] = { .type = NLA_U32 },
80 [NFTA_NAT_FAMILY] = { .type = NLA_U32 },
81 [NFTA_NAT_REG_ADDR_MIN] = { .type = NLA_U32 },
82 [NFTA_NAT_REG_ADDR_MAX] = { .type = NLA_U32 },
83 [NFTA_NAT_REG_PROTO_MIN] = { .type = NLA_U32 },
84 [NFTA_NAT_REG_PROTO_MAX] = { .type = NLA_U32 },
85 [NFTA_NAT_FLAGS] = { .type = NLA_U32 },
88 static int nft_nat_validate(const struct nft_ctx *ctx,
89 const struct nft_expr *expr,
90 const struct nft_data **data)
92 struct nft_nat *priv = nft_expr_priv(expr);
95 err = nft_chain_validate_dependency(ctx->chain, NFT_CHAIN_T_NAT);
101 err = nft_chain_validate_hooks(ctx->chain,
102 (1 << NF_INET_POST_ROUTING) |
103 (1 << NF_INET_LOCAL_IN));
106 err = nft_chain_validate_hooks(ctx->chain,
107 (1 << NF_INET_PRE_ROUTING) |
108 (1 << NF_INET_LOCAL_OUT));
115 static int nft_nat_init(const struct nft_ctx *ctx, const struct nft_expr *expr,
116 const struct nlattr * const tb[])
118 struct nft_nat *priv = nft_expr_priv(expr);
119 unsigned int alen, plen;
123 if (tb[NFTA_NAT_TYPE] == NULL ||
124 (tb[NFTA_NAT_REG_ADDR_MIN] == NULL &&
125 tb[NFTA_NAT_REG_PROTO_MIN] == NULL))
128 switch (ntohl(nla_get_be32(tb[NFTA_NAT_TYPE]))) {
130 priv->type = NF_NAT_MANIP_SRC;
133 priv->type = NF_NAT_MANIP_DST;
139 if (tb[NFTA_NAT_FAMILY] == NULL)
142 family = ntohl(nla_get_be32(tb[NFTA_NAT_FAMILY]));
143 if (ctx->family != NFPROTO_INET && ctx->family != family)
148 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip);
151 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip6);
154 return -EAFNOSUPPORT;
156 priv->family = family;
158 if (tb[NFTA_NAT_REG_ADDR_MIN]) {
159 priv->sreg_addr_min =
160 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MIN]);
161 err = nft_validate_register_load(priv->sreg_addr_min, alen);
165 if (tb[NFTA_NAT_REG_ADDR_MAX]) {
166 priv->sreg_addr_max =
167 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MAX]);
169 err = nft_validate_register_load(priv->sreg_addr_max,
174 priv->sreg_addr_max = priv->sreg_addr_min;
178 plen = FIELD_SIZEOF(struct nf_nat_range, min_addr.all);
179 if (tb[NFTA_NAT_REG_PROTO_MIN]) {
180 priv->sreg_proto_min =
181 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MIN]);
183 err = nft_validate_register_load(priv->sreg_proto_min, plen);
187 if (tb[NFTA_NAT_REG_PROTO_MAX]) {
188 priv->sreg_proto_max =
189 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MAX]);
191 err = nft_validate_register_load(priv->sreg_proto_max,
196 priv->sreg_proto_max = priv->sreg_proto_min;
200 if (tb[NFTA_NAT_FLAGS]) {
201 priv->flags = ntohl(nla_get_be32(tb[NFTA_NAT_FLAGS]));
202 if (priv->flags & ~NF_NAT_RANGE_MASK)
206 return nf_ct_netns_get(ctx->net, family);
209 static int nft_nat_dump(struct sk_buff *skb, const struct nft_expr *expr)
211 const struct nft_nat *priv = nft_expr_priv(expr);
213 switch (priv->type) {
214 case NF_NAT_MANIP_SRC:
215 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_SNAT)))
216 goto nla_put_failure;
218 case NF_NAT_MANIP_DST:
219 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_DNAT)))
220 goto nla_put_failure;
224 if (nla_put_be32(skb, NFTA_NAT_FAMILY, htonl(priv->family)))
225 goto nla_put_failure;
227 if (priv->sreg_addr_min) {
228 if (nft_dump_register(skb, NFTA_NAT_REG_ADDR_MIN,
229 priv->sreg_addr_min) ||
230 nft_dump_register(skb, NFTA_NAT_REG_ADDR_MAX,
231 priv->sreg_addr_max))
232 goto nla_put_failure;
235 if (priv->sreg_proto_min) {
236 if (nft_dump_register(skb, NFTA_NAT_REG_PROTO_MIN,
237 priv->sreg_proto_min) ||
238 nft_dump_register(skb, NFTA_NAT_REG_PROTO_MAX,
239 priv->sreg_proto_max))
240 goto nla_put_failure;
243 if (priv->flags != 0) {
244 if (nla_put_be32(skb, NFTA_NAT_FLAGS, htonl(priv->flags)))
245 goto nla_put_failure;
255 nft_nat_destroy(const struct nft_ctx *ctx, const struct nft_expr *expr)
257 const struct nft_nat *priv = nft_expr_priv(expr);
259 nf_ct_netns_put(ctx->net, priv->family);
262 static struct nft_expr_type nft_nat_type;
263 static const struct nft_expr_ops nft_nat_ops = {
264 .type = &nft_nat_type,
265 .size = NFT_EXPR_SIZE(sizeof(struct nft_nat)),
266 .eval = nft_nat_eval,
267 .init = nft_nat_init,
268 .destroy = nft_nat_destroy,
269 .dump = nft_nat_dump,
270 .validate = nft_nat_validate,
273 static struct nft_expr_type nft_nat_type __read_mostly = {
276 .policy = nft_nat_policy,
277 .maxattr = NFTA_NAT_MAX,
278 .owner = THIS_MODULE,
281 #ifdef CONFIG_NF_TABLES_INET
282 static void nft_nat_inet_eval(const struct nft_expr *expr,
283 struct nft_regs *regs,
284 const struct nft_pktinfo *pkt)
286 const struct nft_nat *priv = nft_expr_priv(expr);
288 if (priv->family == nft_pf(pkt))
289 nft_nat_eval(expr, regs, pkt);
292 static const struct nft_expr_ops nft_nat_inet_ops = {
293 .type = &nft_nat_type,
294 .size = NFT_EXPR_SIZE(sizeof(struct nft_nat)),
295 .eval = nft_nat_inet_eval,
296 .init = nft_nat_init,
297 .destroy = nft_nat_destroy,
298 .dump = nft_nat_dump,
299 .validate = nft_nat_validate,
302 static struct nft_expr_type nft_inet_nat_type __read_mostly = {
304 .family = NFPROTO_INET,
305 .ops = &nft_nat_inet_ops,
306 .policy = nft_nat_policy,
307 .maxattr = NFTA_NAT_MAX,
308 .owner = THIS_MODULE,
311 static int nft_nat_inet_module_init(void)
313 return nft_register_expr(&nft_inet_nat_type);
316 static void nft_nat_inet_module_exit(void)
318 nft_unregister_expr(&nft_inet_nat_type);
321 static int nft_nat_inet_module_init(void) { return 0; }
322 static void nft_nat_inet_module_exit(void) { }
325 static int __init nft_nat_module_init(void)
327 int ret = nft_nat_inet_module_init();
332 ret = nft_register_expr(&nft_nat_type);
334 nft_nat_inet_module_exit();
339 static void __exit nft_nat_module_exit(void)
341 nft_nat_inet_module_exit();
342 nft_unregister_expr(&nft_nat_type);
345 module_init(nft_nat_module_init);
346 module_exit(nft_nat_module_exit);
348 MODULE_LICENSE("GPL");
349 MODULE_AUTHOR("Tomasz Bursztyka <tomasz.bursztyka@linux.intel.com>");
350 MODULE_ALIAS_NFT_EXPR("nat");