~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

TOMOYO Linux Cross Reference
Linux/net/netfilter/nft_nat.c

Version: ~ [ linux-5.1-rc5 ] ~ [ linux-5.0.7 ] ~ [ linux-4.20.17 ] ~ [ linux-4.19.34 ] ~ [ linux-4.18.20 ] ~ [ linux-4.17.19 ] ~ [ linux-4.16.18 ] ~ [ linux-4.15.18 ] ~ [ linux-4.14.111 ] ~ [ linux-4.13.16 ] ~ [ linux-4.12.14 ] ~ [ linux-4.11.12 ] ~ [ linux-4.10.17 ] ~ [ linux-4.9.168 ] ~ [ linux-4.8.17 ] ~ [ linux-4.7.10 ] ~ [ linux-4.6.7 ] ~ [ linux-4.5.7 ] ~ [ linux-4.4.178 ] ~ [ linux-4.3.6 ] ~ [ linux-4.2.8 ] ~ [ linux-4.1.52 ] ~ [ linux-4.0.9 ] ~ [ linux-3.19.8 ] ~ [ linux-3.18.138 ] ~ [ linux-3.17.8 ] ~ [ linux-3.16.65 ] ~ [ linux-3.15.10 ] ~ [ linux-3.14.79 ] ~ [ linux-3.13.11 ] ~ [ linux-3.12.74 ] ~ [ linux-3.11.10 ] ~ [ linux-3.10.108 ] ~ [ linux-3.9.11 ] ~ [ linux-3.8.13 ] ~ [ linux-3.7.10 ] ~ [ linux-3.6.11 ] ~ [ linux-3.5.7 ] ~ [ linux-3.4.113 ] ~ [ linux-3.3.8 ] ~ [ linux-3.2.102 ] ~ [ linux-3.1.10 ] ~ [ linux-3.0.101 ] ~ [ linux-2.6.39.4 ] ~ [ linux-2.6.38.8 ] ~ [ linux-2.6.37.6 ] ~ [ linux-2.6.36.4 ] ~ [ linux-2.6.35.14 ] ~ [ linux-2.6.34.15 ] ~ [ linux-2.6.33.20 ] ~ [ linux-2.6.32.71 ] ~ [ linux-2.6.0 ] ~ [ linux-2.4.37.11 ] ~ [ unix-v6-master ] ~ [ ccs-tools-1.8.5 ] ~ [ policy-sample ] ~
Architecture: ~ [ i386 ] ~ [ alpha ] ~ [ m68k ] ~ [ mips ] ~ [ ppc ] ~ [ sparc ] ~ [ sparc64 ] ~

  1 /*
  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
  5  *
  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.
  9  *
 10  */
 11 
 12 #include <linux/module.h>
 13 #include <linux/init.h>
 14 #include <linux/skbuff.h>
 15 #include <linux/ip.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>
 25 #include <net/ip.h>
 26 
 27 struct nft_nat {
 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;
 33         u8                      family;
 34         u16                     flags;
 35 };
 36 
 37 static void nft_nat_eval(const struct nft_expr *expr,
 38                          struct nft_regs *regs,
 39                          const struct nft_pktinfo *pkt)
 40 {
 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;
 45 
 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];
 53 
 54                 } else {
 55                         memcpy(range.min_addr.ip6,
 56                                &regs->data[priv->sreg_addr_min],
 57                                sizeof(range.min_addr.ip6));
 58                         memcpy(range.max_addr.ip6,
 59                                &regs->data[priv->sreg_addr_max],
 60                                sizeof(range.max_addr.ip6));
 61                 }
 62                 range.flags |= NF_NAT_RANGE_MAP_IPS;
 63         }
 64 
 65         if (priv->sreg_proto_min) {
 66                 range.min_proto.all = (__force __be16)nft_reg_load16(
 67                         &regs->data[priv->sreg_proto_min]);
 68                 range.max_proto.all = (__force __be16)nft_reg_load16(
 69                         &regs->data[priv->sreg_proto_max]);
 70                 range.flags |= NF_NAT_RANGE_PROTO_SPECIFIED;
 71         }
 72 
 73         range.flags |= priv->flags;
 74 
 75         regs->verdict.code = nf_nat_setup_info(ct, &range, priv->type);
 76 }
 77 
 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 },
 86 };
 87 
 88 static int nft_nat_validate(const struct nft_ctx *ctx,
 89                             const struct nft_expr *expr,
 90                             const struct nft_data **data)
 91 {
 92         struct nft_nat *priv = nft_expr_priv(expr);
 93         int err;
 94 
 95         err = nft_chain_validate_dependency(ctx->chain, NFT_CHAIN_T_NAT);
 96         if (err < 0)
 97                 return err;
 98 
 99         switch (priv->type) {
100         case NFT_NAT_SNAT:
101                 err = nft_chain_validate_hooks(ctx->chain,
102                                                (1 << NF_INET_POST_ROUTING) |
103                                                (1 << NF_INET_LOCAL_IN));
104                 break;
105         case NFT_NAT_DNAT:
106                 err = nft_chain_validate_hooks(ctx->chain,
107                                                (1 << NF_INET_PRE_ROUTING) |
108                                                (1 << NF_INET_LOCAL_OUT));
109                 break;
110         }
111 
112         return err;
113 }
114 
115 static int nft_nat_init(const struct nft_ctx *ctx, const struct nft_expr *expr,
116                         const struct nlattr * const tb[])
117 {
118         struct nft_nat *priv = nft_expr_priv(expr);
119         unsigned int alen, plen;
120         u32 family;
121         int err;
122 
123         if (tb[NFTA_NAT_TYPE] == NULL ||
124             (tb[NFTA_NAT_REG_ADDR_MIN] == NULL &&
125              tb[NFTA_NAT_REG_PROTO_MIN] == NULL))
126                 return -EINVAL;
127 
128         switch (ntohl(nla_get_be32(tb[NFTA_NAT_TYPE]))) {
129         case NFT_NAT_SNAT:
130                 priv->type = NF_NAT_MANIP_SRC;
131                 break;
132         case NFT_NAT_DNAT:
133                 priv->type = NF_NAT_MANIP_DST;
134                 break;
135         default:
136                 return -EINVAL;
137         }
138 
139         if (tb[NFTA_NAT_FAMILY] == NULL)
140                 return -EINVAL;
141 
142         family = ntohl(nla_get_be32(tb[NFTA_NAT_FAMILY]));
143         if (family != ctx->family)
144                 return -EOPNOTSUPP;
145 
146         switch (family) {
147         case NFPROTO_IPV4:
148                 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip);
149                 break;
150         case NFPROTO_IPV6:
151                 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip6);
152                 break;
153         default:
154                 return -EAFNOSUPPORT;
155         }
156         priv->family = family;
157 
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);
162                 if (err < 0)
163                         return err;
164 
165                 if (tb[NFTA_NAT_REG_ADDR_MAX]) {
166                         priv->sreg_addr_max =
167                                 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MAX]);
168 
169                         err = nft_validate_register_load(priv->sreg_addr_max,
170                                                          alen);
171                         if (err < 0)
172                                 return err;
173                 } else {
174                         priv->sreg_addr_max = priv->sreg_addr_min;
175                 }
176         }
177 
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]);
182 
183                 err = nft_validate_register_load(priv->sreg_proto_min, plen);
184                 if (err < 0)
185                         return err;
186 
187                 if (tb[NFTA_NAT_REG_PROTO_MAX]) {
188                         priv->sreg_proto_max =
189                                 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MAX]);
190 
191                         err = nft_validate_register_load(priv->sreg_proto_max,
192                                                          plen);
193                         if (err < 0)
194                                 return err;
195                 } else {
196                         priv->sreg_proto_max = priv->sreg_proto_min;
197                 }
198         }
199 
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)
203                         return -EINVAL;
204         }
205 
206         return nf_ct_netns_get(ctx->net, family);
207 }
208 
209 static int nft_nat_dump(struct sk_buff *skb, const struct nft_expr *expr)
210 {
211         const struct nft_nat *priv = nft_expr_priv(expr);
212 
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;
217                 break;
218         case NF_NAT_MANIP_DST:
219                 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_DNAT)))
220                         goto nla_put_failure;
221                 break;
222         }
223 
224         if (nla_put_be32(skb, NFTA_NAT_FAMILY, htonl(priv->family)))
225                 goto nla_put_failure;
226 
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;
233         }
234 
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;
241         }
242 
243         if (priv->flags != 0) {
244                 if (nla_put_be32(skb, NFTA_NAT_FLAGS, htonl(priv->flags)))
245                         goto nla_put_failure;
246         }
247 
248         return 0;
249 
250 nla_put_failure:
251         return -1;
252 }
253 
254 static void
255 nft_nat_destroy(const struct nft_ctx *ctx, const struct nft_expr *expr)
256 {
257         const struct nft_nat *priv = nft_expr_priv(expr);
258 
259         nf_ct_netns_put(ctx->net, priv->family);
260 }
261 
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,
271 };
272 
273 static struct nft_expr_type nft_nat_type __read_mostly = {
274         .name           = "nat",
275         .ops            = &nft_nat_ops,
276         .policy         = nft_nat_policy,
277         .maxattr        = NFTA_NAT_MAX,
278         .owner          = THIS_MODULE,
279 };
280 
281 static int __init nft_nat_module_init(void)
282 {
283         return nft_register_expr(&nft_nat_type);
284 }
285 
286 static void __exit nft_nat_module_exit(void)
287 {
288         nft_unregister_expr(&nft_nat_type);
289 }
290 
291 module_init(nft_nat_module_init);
292 module_exit(nft_nat_module_exit);
293 
294 MODULE_LICENSE("GPL");
295 MODULE_AUTHOR("Tomasz Bursztyka <tomasz.bursztyka@linux.intel.com>");
296 MODULE_ALIAS_NFT_EXPR("nat");
297 

~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

kernel.org | git.kernel.org | LWN.net | Project Home | Wiki (Japanese) | Wiki (English) | SVN repository | Mail admin

Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.

osdn.jp