Blame view
net/netfilter/nft_payload.c
11.4 KB
96518518c netfilter: add nf... |
1 |
/* |
ef1f7df91 netfilter: nf_tab... |
2 |
* Copyright (c) 2008-2009 Patrick McHardy <kaber@trash.net> |
181409698 netfilter: nft_pa... |
3 |
* Copyright (c) 2016 Pablo Neira Ayuso <pablo@netfilter.org> |
96518518c netfilter: add nf... |
4 5 6 7 8 9 10 11 12 |
* * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License version 2 as * published by the Free Software Foundation. * * Development of this code funded by Astaro AG (http://www.astaro.com/) */ #include <linux/kernel.h> |
8cfd23e67 netfilter: nft_pa... |
13 |
#include <linux/if_vlan.h> |
96518518c netfilter: add nf... |
14 15 16 17 18 19 20 |
#include <linux/init.h> #include <linux/module.h> #include <linux/netlink.h> #include <linux/netfilter.h> #include <linux/netfilter/nf_tables.h> #include <net/netfilter/nf_tables_core.h> #include <net/netfilter/nf_tables.h> |
181409698 netfilter: nft_pa... |
21 22 23 24 |
/* For layer 4 checksum field offset. */ #include <linux/tcp.h> #include <linux/udp.h> #include <linux/icmpv6.h> |
96518518c netfilter: add nf... |
25 |
|
8cfd23e67 netfilter: nft_pa... |
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
/* add vlan header into the user buffer for if tag was removed by offloads */ static bool nft_payload_copy_vlan(u32 *d, const struct sk_buff *skb, u8 offset, u8 len) { int mac_off = skb_mac_header(skb) - skb->data; u8 vlan_len, *vlanh, *dst_u8 = (u8 *) d; struct vlan_ethhdr veth; vlanh = (u8 *) &veth; if (offset < ETH_HLEN) { u8 ethlen = min_t(u8, len, ETH_HLEN - offset); if (skb_copy_bits(skb, mac_off, &veth, ETH_HLEN)) return false; veth.h_vlan_proto = skb->vlan_proto; memcpy(dst_u8, vlanh + offset, ethlen); len -= ethlen; if (len == 0) return true; dst_u8 += ethlen; offset = ETH_HLEN; } else if (offset >= VLAN_ETH_HLEN) { offset -= VLAN_HLEN; goto skip; } veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb)); veth.h_vlan_encapsulated_proto = skb->protocol; vlanh += offset; vlan_len = min_t(u8, len, VLAN_ETH_HLEN - offset); memcpy(dst_u8, vlanh, vlan_len); len -= vlan_len; if (!len) return true; dst_u8 += vlan_len; skip: return skb_copy_bits(skb, offset + mac_off, dst_u8, len) == 0; } |
96518518c netfilter: add nf... |
72 |
static void nft_payload_eval(const struct nft_expr *expr, |
a55e22e92 netfilter: nf_tab... |
73 |
struct nft_regs *regs, |
96518518c netfilter: add nf... |
74 75 76 77 |
const struct nft_pktinfo *pkt) { const struct nft_payload *priv = nft_expr_priv(expr); const struct sk_buff *skb = pkt->skb; |
49499c3e6 netfilter: nf_tab... |
78 |
u32 *dest = ®s->data[priv->dreg]; |
96518518c netfilter: add nf... |
79 |
int offset; |
8cfd23e67 netfilter: nft_pa... |
80 |
dest[priv->len / NFT_REG32_SIZE] = 0; |
96518518c netfilter: add nf... |
81 82 83 84 |
switch (priv->base) { case NFT_PAYLOAD_LL_HEADER: if (!skb_mac_header_was_set(skb)) goto err; |
8cfd23e67 netfilter: nft_pa... |
85 86 87 88 89 90 91 |
if (skb_vlan_tag_present(skb)) { if (!nft_payload_copy_vlan(dest, skb, priv->offset, priv->len)) goto err; return; } |
96518518c netfilter: add nf... |
92 93 94 95 96 97 |
offset = skb_mac_header(skb) - skb->data; break; case NFT_PAYLOAD_NETWORK_HEADER: offset = skb_network_offset(skb); break; case NFT_PAYLOAD_TRANSPORT_HEADER: |
a20877b5e netfilter: nf_tab... |
98 99 |
if (!pkt->tprot_set) goto err; |
c54032e05 netfilter: nf_tab... |
100 |
offset = pkt->xt.thoff; |
96518518c netfilter: add nf... |
101 102 103 104 105 |
break; default: BUG(); } offset += priv->offset; |
fad136ea0 netfilter: nf_tab... |
106 |
if (skb_copy_bits(skb, offset, dest, priv->len) < 0) |
96518518c netfilter: add nf... |
107 108 109 |
goto err; return; err: |
a55e22e92 netfilter: nf_tab... |
110 |
regs->verdict.code = NFT_BREAK; |
96518518c netfilter: add nf... |
111 112 113 |
} static const struct nla_policy nft_payload_policy[NFTA_PAYLOAD_MAX + 1] = { |
7ec3f7b47 netfilter: nft_pa... |
114 115 116 117 118 119 120 |
[NFTA_PAYLOAD_SREG] = { .type = NLA_U32 }, [NFTA_PAYLOAD_DREG] = { .type = NLA_U32 }, [NFTA_PAYLOAD_BASE] = { .type = NLA_U32 }, [NFTA_PAYLOAD_OFFSET] = { .type = NLA_U32 }, [NFTA_PAYLOAD_LEN] = { .type = NLA_U32 }, [NFTA_PAYLOAD_CSUM_TYPE] = { .type = NLA_U32 }, [NFTA_PAYLOAD_CSUM_OFFSET] = { .type = NLA_U32 }, |
96518518c netfilter: add nf... |
121 122 123 124 125 126 127 |
}; static int nft_payload_init(const struct nft_ctx *ctx, const struct nft_expr *expr, const struct nlattr * const tb[]) { struct nft_payload *priv = nft_expr_priv(expr); |
96518518c netfilter: add nf... |
128 |
|
c29b72e02 netfilter: nft_pa... |
129 |
priv->base = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_BASE])); |
96518518c netfilter: add nf... |
130 131 |
priv->offset = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_OFFSET])); priv->len = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_LEN])); |
b1c96ed37 netfilter: nf_tab... |
132 |
priv->dreg = nft_parse_register(tb[NFTA_PAYLOAD_DREG]); |
96518518c netfilter: add nf... |
133 |
|
1ec10212f netfilter: nf_tab... |
134 135 |
return nft_validate_register_store(ctx, priv->dreg, NULL, NFT_DATA_VALUE, priv->len); |
96518518c netfilter: add nf... |
136 137 138 139 140 |
} static int nft_payload_dump(struct sk_buff *skb, const struct nft_expr *expr) { const struct nft_payload *priv = nft_expr_priv(expr); |
b1c96ed37 netfilter: nf_tab... |
141 |
if (nft_dump_register(skb, NFTA_PAYLOAD_DREG, priv->dreg) || |
96518518c netfilter: add nf... |
142 143 144 145 146 147 148 149 150 |
nla_put_be32(skb, NFTA_PAYLOAD_BASE, htonl(priv->base)) || nla_put_be32(skb, NFTA_PAYLOAD_OFFSET, htonl(priv->offset)) || nla_put_be32(skb, NFTA_PAYLOAD_LEN, htonl(priv->len))) goto nla_put_failure; return 0; nla_put_failure: return -1; } |
ef1f7df91 netfilter: nf_tab... |
151 152 |
static const struct nft_expr_ops nft_payload_ops = { .type = &nft_payload_type, |
96518518c netfilter: add nf... |
153 |
.size = NFT_EXPR_SIZE(sizeof(struct nft_payload)), |
96518518c netfilter: add nf... |
154 155 156 |
.eval = nft_payload_eval, .init = nft_payload_init, .dump = nft_payload_dump, |
ef1f7df91 netfilter: nf_tab... |
157 |
}; |
c29b72e02 netfilter: nft_pa... |
158 159 160 161 162 163 164 |
const struct nft_expr_ops nft_payload_fast_ops = { .type = &nft_payload_type, .size = NFT_EXPR_SIZE(sizeof(struct nft_payload)), .eval = nft_payload_eval, .init = nft_payload_init, .dump = nft_payload_dump, }; |
181409698 netfilter: nft_pa... |
165 166 167 168 169 170 171 172 173 174 175 176 177 178 |
static inline void nft_csum_replace(__sum16 *sum, __wsum fsum, __wsum tsum) { *sum = csum_fold(csum_add(csum_sub(~csum_unfold(*sum), fsum), tsum)); if (*sum == 0) *sum = CSUM_MANGLED_0; } static bool nft_payload_udp_checksum(struct sk_buff *skb, unsigned int thoff) { struct udphdr *uh, _uh; uh = skb_header_pointer(skb, thoff, sizeof(_uh), &_uh); if (!uh) return false; |
5fd02ebe6 netfilter: fix a ... |
179 |
return (__force bool)uh->check; |
181409698 netfilter: nft_pa... |
180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 |
} static int nft_payload_l4csum_offset(const struct nft_pktinfo *pkt, struct sk_buff *skb, unsigned int *l4csum_offset) { switch (pkt->tprot) { case IPPROTO_TCP: *l4csum_offset = offsetof(struct tcphdr, check); break; case IPPROTO_UDP: if (!nft_payload_udp_checksum(skb, pkt->xt.thoff)) return -1; /* Fall through. */ case IPPROTO_UDPLITE: *l4csum_offset = offsetof(struct udphdr, check); break; case IPPROTO_ICMPV6: *l4csum_offset = offsetof(struct icmp6hdr, icmp6_cksum); break; default: return -1; } *l4csum_offset += pkt->xt.thoff; return 0; } static int nft_payload_l4csum_update(const struct nft_pktinfo *pkt, struct sk_buff *skb, __wsum fsum, __wsum tsum) { int l4csum_offset; __sum16 sum; /* If we cannot determine layer 4 checksum offset or this packet doesn't * require layer 4 checksum recalculation, skip this packet. */ if (nft_payload_l4csum_offset(pkt, skb, &l4csum_offset) < 0) return 0; if (skb_copy_bits(skb, l4csum_offset, &sum, sizeof(sum)) < 0) return -1; /* Checksum mangling for an arbitrary amount of bytes, based on * inet_proto_csum_replace*() functions. */ if (skb->ip_summed != CHECKSUM_PARTIAL) { nft_csum_replace(&sum, fsum, tsum); if (skb->ip_summed == CHECKSUM_COMPLETE) { skb->csum = ~csum_add(csum_sub(~(skb->csum), fsum), tsum); } } else { sum = ~csum_fold(csum_add(csum_sub(csum_unfold(sum), fsum), tsum)); } if (!skb_make_writable(skb, l4csum_offset + sizeof(sum)) || skb_store_bits(skb, l4csum_offset, &sum, sizeof(sum)) < 0) return -1; return 0; } |
053d20f57 netfilter: nft_pa... |
244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 |
static int nft_payload_csum_inet(struct sk_buff *skb, const u32 *src, __wsum fsum, __wsum tsum, int csum_offset) { __sum16 sum; if (skb_copy_bits(skb, csum_offset, &sum, sizeof(sum)) < 0) return -1; nft_csum_replace(&sum, fsum, tsum); if (!skb_make_writable(skb, csum_offset + sizeof(sum)) || skb_store_bits(skb, csum_offset, &sum, sizeof(sum)) < 0) return -1; return 0; } |
7ec3f7b47 netfilter: nft_pa... |
259 260 261 262 263 264 265 266 267 |
static void nft_payload_set_eval(const struct nft_expr *expr, struct nft_regs *regs, const struct nft_pktinfo *pkt) { const struct nft_payload_set *priv = nft_expr_priv(expr); struct sk_buff *skb = pkt->skb; const u32 *src = ®s->data[priv->sreg]; int offset, csum_offset; __wsum fsum, tsum; |
7ec3f7b47 netfilter: nft_pa... |
268 269 270 271 272 273 274 275 276 277 278 |
switch (priv->base) { case NFT_PAYLOAD_LL_HEADER: if (!skb_mac_header_was_set(skb)) goto err; offset = skb_mac_header(skb) - skb->data; break; case NFT_PAYLOAD_NETWORK_HEADER: offset = skb_network_offset(skb); break; case NFT_PAYLOAD_TRANSPORT_HEADER: |
a20877b5e netfilter: nf_tab... |
279 280 |
if (!pkt->tprot_set) goto err; |
7ec3f7b47 netfilter: nft_pa... |
281 282 283 284 285 286 287 288 |
offset = pkt->xt.thoff; break; default: BUG(); } csum_offset = offset + priv->csum_offset; offset += priv->offset; |
053d20f57 netfilter: nft_pa... |
289 |
if ((priv->csum_type == NFT_PAYLOAD_CSUM_INET || priv->csum_flags) && |
7ec3f7b47 netfilter: nft_pa... |
290 291 |
(priv->base != NFT_PAYLOAD_TRANSPORT_HEADER || skb->ip_summed != CHECKSUM_PARTIAL)) { |
7ec3f7b47 netfilter: nft_pa... |
292 293 |
fsum = skb_checksum(skb, offset, priv->len, 0); tsum = csum_partial(src, priv->len, 0); |
7ec3f7b47 netfilter: nft_pa... |
294 |
|
053d20f57 netfilter: nft_pa... |
295 296 |
if (priv->csum_type == NFT_PAYLOAD_CSUM_INET && nft_payload_csum_inet(skb, src, fsum, tsum, csum_offset)) |
7ec3f7b47 netfilter: nft_pa... |
297 |
goto err; |
181409698 netfilter: nft_pa... |
298 299 300 301 |
if (priv->csum_flags && nft_payload_l4csum_update(pkt, skb, fsum, tsum) < 0) goto err; |
7ec3f7b47 netfilter: nft_pa... |
302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 |
} if (!skb_make_writable(skb, max(offset + priv->len, 0)) || skb_store_bits(skb, offset, src, priv->len) < 0) goto err; return; err: regs->verdict.code = NFT_BREAK; } static int nft_payload_set_init(const struct nft_ctx *ctx, const struct nft_expr *expr, const struct nlattr * const tb[]) { struct nft_payload_set *priv = nft_expr_priv(expr); priv->base = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_BASE])); priv->offset = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_OFFSET])); priv->len = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_LEN])); priv->sreg = nft_parse_register(tb[NFTA_PAYLOAD_SREG]); if (tb[NFTA_PAYLOAD_CSUM_TYPE]) priv->csum_type = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_CSUM_TYPE])); if (tb[NFTA_PAYLOAD_CSUM_OFFSET]) priv->csum_offset = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_CSUM_OFFSET])); |
181409698 netfilter: nft_pa... |
330 331 332 333 334 335 336 337 338 |
if (tb[NFTA_PAYLOAD_CSUM_FLAGS]) { u32 flags; flags = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_CSUM_FLAGS])); if (flags & ~NFT_PAYLOAD_L4CSUM_PSEUDOHDR) return -EINVAL; priv->csum_flags = flags; } |
7ec3f7b47 netfilter: nft_pa... |
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 |
switch (priv->csum_type) { case NFT_PAYLOAD_CSUM_NONE: case NFT_PAYLOAD_CSUM_INET: break; default: return -EOPNOTSUPP; } return nft_validate_register_load(priv->sreg, priv->len); } static int nft_payload_set_dump(struct sk_buff *skb, const struct nft_expr *expr) { const struct nft_payload_set *priv = nft_expr_priv(expr); if (nft_dump_register(skb, NFTA_PAYLOAD_SREG, priv->sreg) || nla_put_be32(skb, NFTA_PAYLOAD_BASE, htonl(priv->base)) || nla_put_be32(skb, NFTA_PAYLOAD_OFFSET, htonl(priv->offset)) || nla_put_be32(skb, NFTA_PAYLOAD_LEN, htonl(priv->len)) || nla_put_be32(skb, NFTA_PAYLOAD_CSUM_TYPE, htonl(priv->csum_type)) || nla_put_be32(skb, NFTA_PAYLOAD_CSUM_OFFSET, |
181409698 netfilter: nft_pa... |
361 362 |
htonl(priv->csum_offset)) || nla_put_be32(skb, NFTA_PAYLOAD_CSUM_FLAGS, htonl(priv->csum_flags))) |
7ec3f7b47 netfilter: nft_pa... |
363 364 365 366 367 368 369 370 371 372 373 374 375 376 |
goto nla_put_failure; return 0; nla_put_failure: return -1; } static const struct nft_expr_ops nft_payload_set_ops = { .type = &nft_payload_type, .size = NFT_EXPR_SIZE(sizeof(struct nft_payload_set)), .eval = nft_payload_set_eval, .init = nft_payload_set_init, .dump = nft_payload_set_dump, }; |
0ca743a55 netfilter: nf_tab... |
377 378 379 |
static const struct nft_expr_ops * nft_payload_select_ops(const struct nft_ctx *ctx, const struct nlattr * const tb[]) |
c29b72e02 netfilter: nft_pa... |
380 381 382 |
{ enum nft_payload_bases base; unsigned int offset, len; |
7ec3f7b47 netfilter: nft_pa... |
383 |
if (tb[NFTA_PAYLOAD_BASE] == NULL || |
c29b72e02 netfilter: nft_pa... |
384 385 386 387 388 389 390 391 392 393 394 395 396 |
tb[NFTA_PAYLOAD_OFFSET] == NULL || tb[NFTA_PAYLOAD_LEN] == NULL) return ERR_PTR(-EINVAL); base = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_BASE])); switch (base) { case NFT_PAYLOAD_LL_HEADER: case NFT_PAYLOAD_NETWORK_HEADER: case NFT_PAYLOAD_TRANSPORT_HEADER: break; default: return ERR_PTR(-EOPNOTSUPP); } |
7ec3f7b47 netfilter: nft_pa... |
397 398 399 400 401 402 403 404 |
if (tb[NFTA_PAYLOAD_SREG] != NULL) { if (tb[NFTA_PAYLOAD_DREG] != NULL) return ERR_PTR(-EINVAL); return &nft_payload_set_ops; } if (tb[NFTA_PAYLOAD_DREG] == NULL) return ERR_PTR(-EINVAL); |
c29b72e02 netfilter: nft_pa... |
405 |
offset = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_OFFSET])); |
45d9bcda2 netfilter: nf_tab... |
406 |
len = ntohl(nla_get_be32(tb[NFTA_PAYLOAD_LEN])); |
c29b72e02 netfilter: nft_pa... |
407 |
|
f627ed91d netfilter: nf_tab... |
408 409 |
if (len <= 4 && is_power_of_2(len) && IS_ALIGNED(offset, len) && base != NFT_PAYLOAD_LL_HEADER) |
c29b72e02 netfilter: nft_pa... |
410 411 412 413 |
return &nft_payload_fast_ops; else return &nft_payload_ops; } |
4e24877e6 netfilter: nf_tab... |
414 |
struct nft_expr_type nft_payload_type __read_mostly = { |
ef1f7df91 netfilter: nf_tab... |
415 |
.name = "payload", |
c29b72e02 netfilter: nft_pa... |
416 |
.select_ops = nft_payload_select_ops, |
96518518c netfilter: add nf... |
417 418 |
.policy = nft_payload_policy, .maxattr = NFTA_PAYLOAD_MAX, |
ef1f7df91 netfilter: nf_tab... |
419 |
.owner = THIS_MODULE, |
96518518c netfilter: add nf... |
420 |
}; |