Loading...
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_nat_core.h>
25#include <net/netfilter/nf_tables.h>
26#include <net/netfilter/nf_nat_l3proto.h>
27#include <net/ip.h>
28
29struct nft_nat {
30 enum nft_registers sreg_addr_min:8;
31 enum nft_registers sreg_addr_max:8;
32 enum nft_registers sreg_proto_min:8;
33 enum nft_registers sreg_proto_max:8;
34 enum nf_nat_manip_type type:8;
35 u8 family;
36 u16 flags;
37};
38
39static void nft_nat_eval(const struct nft_expr *expr,
40 struct nft_regs *regs,
41 const struct nft_pktinfo *pkt)
42{
43 const struct nft_nat *priv = nft_expr_priv(expr);
44 enum ip_conntrack_info ctinfo;
45 struct nf_conn *ct = nf_ct_get(pkt->skb, &ctinfo);
46 struct nf_nat_range range;
47
48 memset(&range, 0, sizeof(range));
49 if (priv->sreg_addr_min) {
50 if (priv->family == AF_INET) {
51 range.min_addr.ip = (__force __be32)
52 regs->data[priv->sreg_addr_min];
53 range.max_addr.ip = (__force __be32)
54 regs->data[priv->sreg_addr_max];
55
56 } else {
57 memcpy(range.min_addr.ip6,
58 ®s->data[priv->sreg_addr_min],
59 sizeof(range.min_addr.ip6));
60 memcpy(range.max_addr.ip6,
61 ®s->data[priv->sreg_addr_max],
62 sizeof(range.max_addr.ip6));
63 }
64 range.flags |= NF_NAT_RANGE_MAP_IPS;
65 }
66
67 if (priv->sreg_proto_min) {
68 range.min_proto.all =
69 *(__be16 *)®s->data[priv->sreg_proto_min];
70 range.max_proto.all =
71 *(__be16 *)®s->data[priv->sreg_proto_max];
72 range.flags |= NF_NAT_RANGE_PROTO_SPECIFIED;
73 }
74
75 range.flags |= priv->flags;
76
77 regs->verdict.code = nf_nat_setup_info(ct, &range, priv->type);
78}
79
80static const struct nla_policy nft_nat_policy[NFTA_NAT_MAX + 1] = {
81 [NFTA_NAT_TYPE] = { .type = NLA_U32 },
82 [NFTA_NAT_FAMILY] = { .type = NLA_U32 },
83 [NFTA_NAT_REG_ADDR_MIN] = { .type = NLA_U32 },
84 [NFTA_NAT_REG_ADDR_MAX] = { .type = NLA_U32 },
85 [NFTA_NAT_REG_PROTO_MIN] = { .type = NLA_U32 },
86 [NFTA_NAT_REG_PROTO_MAX] = { .type = NLA_U32 },
87 [NFTA_NAT_FLAGS] = { .type = NLA_U32 },
88};
89
90static int nft_nat_validate(const struct nft_ctx *ctx,
91 const struct nft_expr *expr,
92 const struct nft_data **data)
93{
94 struct nft_nat *priv = nft_expr_priv(expr);
95 int err;
96
97 err = nft_chain_validate_dependency(ctx->chain, NFT_CHAIN_T_NAT);
98 if (err < 0)
99 return err;
100
101 switch (priv->type) {
102 case NFT_NAT_SNAT:
103 err = nft_chain_validate_hooks(ctx->chain,
104 (1 << NF_INET_POST_ROUTING) |
105 (1 << NF_INET_LOCAL_IN));
106 break;
107 case NFT_NAT_DNAT:
108 err = nft_chain_validate_hooks(ctx->chain,
109 (1 << NF_INET_PRE_ROUTING) |
110 (1 << NF_INET_LOCAL_OUT));
111 break;
112 }
113
114 return err;
115}
116
117static int nft_nat_init(const struct nft_ctx *ctx, const struct nft_expr *expr,
118 const struct nlattr * const tb[])
119{
120 struct nft_nat *priv = nft_expr_priv(expr);
121 unsigned int alen, plen;
122 u32 family;
123 int err;
124
125 if (tb[NFTA_NAT_TYPE] == NULL ||
126 (tb[NFTA_NAT_REG_ADDR_MIN] == NULL &&
127 tb[NFTA_NAT_REG_PROTO_MIN] == NULL))
128 return -EINVAL;
129
130 switch (ntohl(nla_get_be32(tb[NFTA_NAT_TYPE]))) {
131 case NFT_NAT_SNAT:
132 priv->type = NF_NAT_MANIP_SRC;
133 break;
134 case NFT_NAT_DNAT:
135 priv->type = NF_NAT_MANIP_DST;
136 break;
137 default:
138 return -EINVAL;
139 }
140
141 err = nft_nat_validate(ctx, expr, NULL);
142 if (err < 0)
143 return err;
144
145 if (tb[NFTA_NAT_FAMILY] == NULL)
146 return -EINVAL;
147
148 family = ntohl(nla_get_be32(tb[NFTA_NAT_FAMILY]));
149 if (family != ctx->afi->family)
150 return -EOPNOTSUPP;
151
152 switch (family) {
153 case NFPROTO_IPV4:
154 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip);
155 break;
156 case NFPROTO_IPV6:
157 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip6);
158 break;
159 default:
160 return -EAFNOSUPPORT;
161 }
162 priv->family = family;
163
164 if (tb[NFTA_NAT_REG_ADDR_MIN]) {
165 priv->sreg_addr_min =
166 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MIN]);
167 err = nft_validate_register_load(priv->sreg_addr_min, alen);
168 if (err < 0)
169 return err;
170
171 if (tb[NFTA_NAT_REG_ADDR_MAX]) {
172 priv->sreg_addr_max =
173 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MAX]);
174
175 err = nft_validate_register_load(priv->sreg_addr_max,
176 alen);
177 if (err < 0)
178 return err;
179 } else {
180 priv->sreg_addr_max = priv->sreg_addr_min;
181 }
182 }
183
184 plen = FIELD_SIZEOF(struct nf_nat_range, min_addr.all);
185 if (tb[NFTA_NAT_REG_PROTO_MIN]) {
186 priv->sreg_proto_min =
187 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MIN]);
188
189 err = nft_validate_register_load(priv->sreg_proto_min, plen);
190 if (err < 0)
191 return err;
192
193 if (tb[NFTA_NAT_REG_PROTO_MAX]) {
194 priv->sreg_proto_max =
195 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MAX]);
196
197 err = nft_validate_register_load(priv->sreg_proto_max,
198 plen);
199 if (err < 0)
200 return err;
201 } else {
202 priv->sreg_proto_max = priv->sreg_proto_min;
203 }
204 }
205
206 if (tb[NFTA_NAT_FLAGS]) {
207 priv->flags = ntohl(nla_get_be32(tb[NFTA_NAT_FLAGS]));
208 if (priv->flags & ~NF_NAT_RANGE_MASK)
209 return -EINVAL;
210 }
211
212 return nf_ct_netns_get(ctx->net, family);
213}
214
215static int nft_nat_dump(struct sk_buff *skb, const struct nft_expr *expr)
216{
217 const struct nft_nat *priv = nft_expr_priv(expr);
218
219 switch (priv->type) {
220 case NF_NAT_MANIP_SRC:
221 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_SNAT)))
222 goto nla_put_failure;
223 break;
224 case NF_NAT_MANIP_DST:
225 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_DNAT)))
226 goto nla_put_failure;
227 break;
228 }
229
230 if (nla_put_be32(skb, NFTA_NAT_FAMILY, htonl(priv->family)))
231 goto nla_put_failure;
232
233 if (priv->sreg_addr_min) {
234 if (nft_dump_register(skb, NFTA_NAT_REG_ADDR_MIN,
235 priv->sreg_addr_min) ||
236 nft_dump_register(skb, NFTA_NAT_REG_ADDR_MAX,
237 priv->sreg_addr_max))
238 goto nla_put_failure;
239 }
240
241 if (priv->sreg_proto_min) {
242 if (nft_dump_register(skb, NFTA_NAT_REG_PROTO_MIN,
243 priv->sreg_proto_min) ||
244 nft_dump_register(skb, NFTA_NAT_REG_PROTO_MAX,
245 priv->sreg_proto_max))
246 goto nla_put_failure;
247 }
248
249 if (priv->flags != 0) {
250 if (nla_put_be32(skb, NFTA_NAT_FLAGS, htonl(priv->flags)))
251 goto nla_put_failure;
252 }
253
254 return 0;
255
256nla_put_failure:
257 return -1;
258}
259
260static void
261nft_nat_destroy(const struct nft_ctx *ctx, const struct nft_expr *expr)
262{
263 const struct nft_nat *priv = nft_expr_priv(expr);
264
265 nf_ct_netns_put(ctx->net, priv->family);
266}
267
268static struct nft_expr_type nft_nat_type;
269static const struct nft_expr_ops nft_nat_ops = {
270 .type = &nft_nat_type,
271 .size = NFT_EXPR_SIZE(sizeof(struct nft_nat)),
272 .eval = nft_nat_eval,
273 .init = nft_nat_init,
274 .destroy = nft_nat_destroy,
275 .dump = nft_nat_dump,
276 .validate = nft_nat_validate,
277};
278
279static struct nft_expr_type nft_nat_type __read_mostly = {
280 .name = "nat",
281 .ops = &nft_nat_ops,
282 .policy = nft_nat_policy,
283 .maxattr = NFTA_NAT_MAX,
284 .owner = THIS_MODULE,
285};
286
287static int __init nft_nat_module_init(void)
288{
289 return nft_register_expr(&nft_nat_type);
290}
291
292static void __exit nft_nat_module_exit(void)
293{
294 nft_unregister_expr(&nft_nat_type);
295}
296
297module_init(nft_nat_module_init);
298module_exit(nft_nat_module_exit);
299
300MODULE_LICENSE("GPL");
301MODULE_AUTHOR("Tomasz Bursztyka <tomasz.bursztyka@linux.intel.com>");
302MODULE_ALIAS_NFT_EXPR("nat");
1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * Copyright (c) 2008-2009 Patrick McHardy <kaber@trash.net>
4 * Copyright (c) 2012 Pablo Neira Ayuso <pablo@netfilter.org>
5 * Copyright (c) 2012 Intel Corporation
6 */
7
8#include <linux/module.h>
9#include <linux/init.h>
10#include <linux/skbuff.h>
11#include <linux/ip.h>
12#include <linux/string.h>
13#include <linux/netlink.h>
14#include <linux/netfilter.h>
15#include <linux/netfilter_ipv4.h>
16#include <linux/netfilter/nfnetlink.h>
17#include <linux/netfilter/nf_tables.h>
18#include <net/netfilter/nf_conntrack.h>
19#include <net/netfilter/nf_nat.h>
20#include <net/netfilter/nf_tables.h>
21#include <net/ip.h>
22
23struct nft_nat {
24 enum nft_registers sreg_addr_min:8;
25 enum nft_registers sreg_addr_max:8;
26 enum nft_registers sreg_proto_min:8;
27 enum nft_registers sreg_proto_max:8;
28 enum nf_nat_manip_type type:8;
29 u8 family;
30 u16 flags;
31};
32
33static void nft_nat_eval(const struct nft_expr *expr,
34 struct nft_regs *regs,
35 const struct nft_pktinfo *pkt)
36{
37 const struct nft_nat *priv = nft_expr_priv(expr);
38 enum ip_conntrack_info ctinfo;
39 struct nf_conn *ct = nf_ct_get(pkt->skb, &ctinfo);
40 struct nf_nat_range2 range;
41
42 memset(&range, 0, sizeof(range));
43 if (priv->sreg_addr_min) {
44 if (priv->family == AF_INET) {
45 range.min_addr.ip = (__force __be32)
46 regs->data[priv->sreg_addr_min];
47 range.max_addr.ip = (__force __be32)
48 regs->data[priv->sreg_addr_max];
49
50 } else {
51 memcpy(range.min_addr.ip6,
52 ®s->data[priv->sreg_addr_min],
53 sizeof(range.min_addr.ip6));
54 memcpy(range.max_addr.ip6,
55 ®s->data[priv->sreg_addr_max],
56 sizeof(range.max_addr.ip6));
57 }
58 range.flags |= NF_NAT_RANGE_MAP_IPS;
59 }
60
61 if (priv->sreg_proto_min) {
62 range.min_proto.all = (__force __be16)nft_reg_load16(
63 ®s->data[priv->sreg_proto_min]);
64 range.max_proto.all = (__force __be16)nft_reg_load16(
65 ®s->data[priv->sreg_proto_max]);
66 range.flags |= NF_NAT_RANGE_PROTO_SPECIFIED;
67 }
68
69 range.flags |= priv->flags;
70
71 regs->verdict.code = nf_nat_setup_info(ct, &range, priv->type);
72}
73
74static const struct nla_policy nft_nat_policy[NFTA_NAT_MAX + 1] = {
75 [NFTA_NAT_TYPE] = { .type = NLA_U32 },
76 [NFTA_NAT_FAMILY] = { .type = NLA_U32 },
77 [NFTA_NAT_REG_ADDR_MIN] = { .type = NLA_U32 },
78 [NFTA_NAT_REG_ADDR_MAX] = { .type = NLA_U32 },
79 [NFTA_NAT_REG_PROTO_MIN] = { .type = NLA_U32 },
80 [NFTA_NAT_REG_PROTO_MAX] = { .type = NLA_U32 },
81 [NFTA_NAT_FLAGS] = { .type = NLA_U32 },
82};
83
84static int nft_nat_validate(const struct nft_ctx *ctx,
85 const struct nft_expr *expr,
86 const struct nft_data **data)
87{
88 struct nft_nat *priv = nft_expr_priv(expr);
89 int err;
90
91 err = nft_chain_validate_dependency(ctx->chain, NFT_CHAIN_T_NAT);
92 if (err < 0)
93 return err;
94
95 switch (priv->type) {
96 case NFT_NAT_SNAT:
97 err = nft_chain_validate_hooks(ctx->chain,
98 (1 << NF_INET_POST_ROUTING) |
99 (1 << NF_INET_LOCAL_IN));
100 break;
101 case NFT_NAT_DNAT:
102 err = nft_chain_validate_hooks(ctx->chain,
103 (1 << NF_INET_PRE_ROUTING) |
104 (1 << NF_INET_LOCAL_OUT));
105 break;
106 }
107
108 return err;
109}
110
111static int nft_nat_init(const struct nft_ctx *ctx, const struct nft_expr *expr,
112 const struct nlattr * const tb[])
113{
114 struct nft_nat *priv = nft_expr_priv(expr);
115 unsigned int alen, plen;
116 u32 family;
117 int err;
118
119 if (tb[NFTA_NAT_TYPE] == NULL ||
120 (tb[NFTA_NAT_REG_ADDR_MIN] == NULL &&
121 tb[NFTA_NAT_REG_PROTO_MIN] == NULL))
122 return -EINVAL;
123
124 switch (ntohl(nla_get_be32(tb[NFTA_NAT_TYPE]))) {
125 case NFT_NAT_SNAT:
126 priv->type = NF_NAT_MANIP_SRC;
127 break;
128 case NFT_NAT_DNAT:
129 priv->type = NF_NAT_MANIP_DST;
130 break;
131 default:
132 return -EINVAL;
133 }
134
135 if (tb[NFTA_NAT_FAMILY] == NULL)
136 return -EINVAL;
137
138 family = ntohl(nla_get_be32(tb[NFTA_NAT_FAMILY]));
139 if (ctx->family != NFPROTO_INET && ctx->family != family)
140 return -EOPNOTSUPP;
141
142 switch (family) {
143 case NFPROTO_IPV4:
144 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip);
145 break;
146 case NFPROTO_IPV6:
147 alen = FIELD_SIZEOF(struct nf_nat_range, min_addr.ip6);
148 break;
149 default:
150 return -EAFNOSUPPORT;
151 }
152 priv->family = family;
153
154 if (tb[NFTA_NAT_REG_ADDR_MIN]) {
155 priv->sreg_addr_min =
156 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MIN]);
157 err = nft_validate_register_load(priv->sreg_addr_min, alen);
158 if (err < 0)
159 return err;
160
161 if (tb[NFTA_NAT_REG_ADDR_MAX]) {
162 priv->sreg_addr_max =
163 nft_parse_register(tb[NFTA_NAT_REG_ADDR_MAX]);
164
165 err = nft_validate_register_load(priv->sreg_addr_max,
166 alen);
167 if (err < 0)
168 return err;
169 } else {
170 priv->sreg_addr_max = priv->sreg_addr_min;
171 }
172 }
173
174 plen = FIELD_SIZEOF(struct nf_nat_range, min_addr.all);
175 if (tb[NFTA_NAT_REG_PROTO_MIN]) {
176 priv->sreg_proto_min =
177 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MIN]);
178
179 err = nft_validate_register_load(priv->sreg_proto_min, plen);
180 if (err < 0)
181 return err;
182
183 if (tb[NFTA_NAT_REG_PROTO_MAX]) {
184 priv->sreg_proto_max =
185 nft_parse_register(tb[NFTA_NAT_REG_PROTO_MAX]);
186
187 err = nft_validate_register_load(priv->sreg_proto_max,
188 plen);
189 if (err < 0)
190 return err;
191 } else {
192 priv->sreg_proto_max = priv->sreg_proto_min;
193 }
194 }
195
196 if (tb[NFTA_NAT_FLAGS]) {
197 priv->flags = ntohl(nla_get_be32(tb[NFTA_NAT_FLAGS]));
198 if (priv->flags & ~NF_NAT_RANGE_MASK)
199 return -EINVAL;
200 }
201
202 return nf_ct_netns_get(ctx->net, family);
203}
204
205static int nft_nat_dump(struct sk_buff *skb, const struct nft_expr *expr)
206{
207 const struct nft_nat *priv = nft_expr_priv(expr);
208
209 switch (priv->type) {
210 case NF_NAT_MANIP_SRC:
211 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_SNAT)))
212 goto nla_put_failure;
213 break;
214 case NF_NAT_MANIP_DST:
215 if (nla_put_be32(skb, NFTA_NAT_TYPE, htonl(NFT_NAT_DNAT)))
216 goto nla_put_failure;
217 break;
218 }
219
220 if (nla_put_be32(skb, NFTA_NAT_FAMILY, htonl(priv->family)))
221 goto nla_put_failure;
222
223 if (priv->sreg_addr_min) {
224 if (nft_dump_register(skb, NFTA_NAT_REG_ADDR_MIN,
225 priv->sreg_addr_min) ||
226 nft_dump_register(skb, NFTA_NAT_REG_ADDR_MAX,
227 priv->sreg_addr_max))
228 goto nla_put_failure;
229 }
230
231 if (priv->sreg_proto_min) {
232 if (nft_dump_register(skb, NFTA_NAT_REG_PROTO_MIN,
233 priv->sreg_proto_min) ||
234 nft_dump_register(skb, NFTA_NAT_REG_PROTO_MAX,
235 priv->sreg_proto_max))
236 goto nla_put_failure;
237 }
238
239 if (priv->flags != 0) {
240 if (nla_put_be32(skb, NFTA_NAT_FLAGS, htonl(priv->flags)))
241 goto nla_put_failure;
242 }
243
244 return 0;
245
246nla_put_failure:
247 return -1;
248}
249
250static void
251nft_nat_destroy(const struct nft_ctx *ctx, const struct nft_expr *expr)
252{
253 const struct nft_nat *priv = nft_expr_priv(expr);
254
255 nf_ct_netns_put(ctx->net, priv->family);
256}
257
258static struct nft_expr_type nft_nat_type;
259static const struct nft_expr_ops nft_nat_ops = {
260 .type = &nft_nat_type,
261 .size = NFT_EXPR_SIZE(sizeof(struct nft_nat)),
262 .eval = nft_nat_eval,
263 .init = nft_nat_init,
264 .destroy = nft_nat_destroy,
265 .dump = nft_nat_dump,
266 .validate = nft_nat_validate,
267};
268
269static struct nft_expr_type nft_nat_type __read_mostly = {
270 .name = "nat",
271 .ops = &nft_nat_ops,
272 .policy = nft_nat_policy,
273 .maxattr = NFTA_NAT_MAX,
274 .owner = THIS_MODULE,
275};
276
277#ifdef CONFIG_NF_TABLES_INET
278static void nft_nat_inet_eval(const struct nft_expr *expr,
279 struct nft_regs *regs,
280 const struct nft_pktinfo *pkt)
281{
282 const struct nft_nat *priv = nft_expr_priv(expr);
283
284 if (priv->family == nft_pf(pkt))
285 nft_nat_eval(expr, regs, pkt);
286}
287
288static const struct nft_expr_ops nft_nat_inet_ops = {
289 .type = &nft_nat_type,
290 .size = NFT_EXPR_SIZE(sizeof(struct nft_nat)),
291 .eval = nft_nat_inet_eval,
292 .init = nft_nat_init,
293 .destroy = nft_nat_destroy,
294 .dump = nft_nat_dump,
295 .validate = nft_nat_validate,
296};
297
298static struct nft_expr_type nft_inet_nat_type __read_mostly = {
299 .name = "nat",
300 .family = NFPROTO_INET,
301 .ops = &nft_nat_inet_ops,
302 .policy = nft_nat_policy,
303 .maxattr = NFTA_NAT_MAX,
304 .owner = THIS_MODULE,
305};
306
307static int nft_nat_inet_module_init(void)
308{
309 return nft_register_expr(&nft_inet_nat_type);
310}
311
312static void nft_nat_inet_module_exit(void)
313{
314 nft_unregister_expr(&nft_inet_nat_type);
315}
316#else
317static int nft_nat_inet_module_init(void) { return 0; }
318static void nft_nat_inet_module_exit(void) { }
319#endif
320
321static int __init nft_nat_module_init(void)
322{
323 int ret = nft_nat_inet_module_init();
324
325 if (ret)
326 return ret;
327
328 ret = nft_register_expr(&nft_nat_type);
329 if (ret)
330 nft_nat_inet_module_exit();
331
332 return ret;
333}
334
335static void __exit nft_nat_module_exit(void)
336{
337 nft_nat_inet_module_exit();
338 nft_unregister_expr(&nft_nat_type);
339}
340
341module_init(nft_nat_module_init);
342module_exit(nft_nat_module_exit);
343
344MODULE_LICENSE("GPL");
345MODULE_AUTHOR("Tomasz Bursztyka <tomasz.bursztyka@linux.intel.com>");
346MODULE_ALIAS_NFT_EXPR("nat");