* [PATCH libnftnl 1/1] payload: add payload mangling support
2015-11-09 14:59 [PATCH libnftnl 0/1] payload mangling support Patrick McHardy
@ 2015-11-09 14:59 ` Patrick McHardy
0 siblings, 0 replies; 2+ messages in thread
From: Patrick McHardy @ 2015-11-09 14:59 UTC (permalink / raw)
To: pablo; +Cc: netfilter-devel
Signed-off-by: Patrick McHardy <kaber@trash.net>
---
include/libnftnl/expr.h | 4 ++++
include/linux/netfilter/nf_tables.h | 2 ++
src/expr/payload.c | 42 +++++++++++++++++++++++++++++++++----
3 files changed, 44 insertions(+), 4 deletions(-)
diff --git a/include/libnftnl/expr.h b/include/libnftnl/expr.h
index 4a37581..3facee2 100644
--- a/include/libnftnl/expr.h
+++ b/include/libnftnl/expr.h
@@ -43,6 +43,8 @@ enum {
NFTNL_EXPR_PAYLOAD_BASE,
NFTNL_EXPR_PAYLOAD_OFFSET,
NFTNL_EXPR_PAYLOAD_LEN,
+ NFTNL_EXPR_PAYLOAD_SREG,
+ NFTNL_EXPR_PAYLOAD_CSUM_OFFSET,
};
enum {
@@ -218,6 +220,8 @@ enum {
NFT_EXPR_PAYLOAD_BASE,
NFT_EXPR_PAYLOAD_OFFSET,
NFT_EXPR_PAYLOAD_LEN,
+ NFT_EXPR_PAYLOAD_SREG,
+ NFT_EXPR_PAYLOAD_CSUM_OFFSET,
};
enum {
diff --git a/include/linux/netfilter/nf_tables.h b/include/linux/netfilter/nf_tables.h
index 9796d82..7336e98 100644
--- a/include/linux/netfilter/nf_tables.h
+++ b/include/linux/netfilter/nf_tables.h
@@ -611,6 +611,8 @@ enum nft_payload_attributes {
NFTA_PAYLOAD_BASE,
NFTA_PAYLOAD_OFFSET,
NFTA_PAYLOAD_LEN,
+ NFTA_PAYLOAD_SREG,
+ NFTA_PAYLOAD_CSUM_OFFSET,
__NFTA_PAYLOAD_MAX
};
#define NFTA_PAYLOAD_MAX (__NFTA_PAYLOAD_MAX - 1)
diff --git a/src/expr/payload.c b/src/expr/payload.c
index be10663..490ee23 100644
--- a/src/expr/payload.c
+++ b/src/expr/payload.c
@@ -25,10 +25,12 @@
#include <libnftnl/rule.h>
struct nftnl_expr_payload {
+ enum nft_registers sreg;
enum nft_registers dreg;
enum nft_payload_bases base;
uint32_t offset;
uint32_t len;
+ uint32_t csum_offset;
};
static int
@@ -38,6 +40,9 @@ nftnl_expr_payload_set(struct nftnl_expr *e, uint16_t type,
struct nftnl_expr_payload *payload = nftnl_expr_data(e);
switch(type) {
+ case NFTNL_EXPR_PAYLOAD_SREG:
+ payload->sreg = *((uint32_t *)data);
+ break;
case NFTNL_EXPR_PAYLOAD_DREG:
payload->dreg = *((uint32_t *)data);
break;
@@ -50,6 +55,9 @@ nftnl_expr_payload_set(struct nftnl_expr *e, uint16_t type,
case NFTNL_EXPR_PAYLOAD_LEN:
payload->len = *((unsigned int *)data);
break;
+ case NFTNL_EXPR_PAYLOAD_CSUM_OFFSET:
+ payload->csum_offset = *((unsigned int *)data);
+ break;
default:
return -1;
}
@@ -63,6 +71,9 @@ nftnl_expr_payload_get(const struct nftnl_expr *e, uint16_t type,
struct nftnl_expr_payload *payload = nftnl_expr_data(e);
switch(type) {
+ case NFTNL_EXPR_PAYLOAD_SREG:
+ *data_len = sizeof(payload->sreg);
+ return &payload->sreg;
case NFTNL_EXPR_PAYLOAD_DREG:
*data_len = sizeof(payload->dreg);
return &payload->dreg;
@@ -75,6 +86,9 @@ nftnl_expr_payload_get(const struct nftnl_expr *e, uint16_t type,
case NFTNL_EXPR_PAYLOAD_LEN:
*data_len = sizeof(payload->len);
return &payload->len;
+ case NFTNL_EXPR_PAYLOAD_CSUM_OFFSET:
+ *data_len = sizeof(payload->csum_offset);
+ return &payload->csum_offset;
}
return NULL;
}
@@ -88,10 +102,12 @@ static int nftnl_expr_payload_cb(const struct nlattr *attr, void *data)
return MNL_CB_OK;
switch(type) {
+ case NFTA_PAYLOAD_SREG:
case NFTA_PAYLOAD_DREG:
case NFTA_PAYLOAD_BASE:
case NFTA_PAYLOAD_OFFSET:
case NFTA_PAYLOAD_LEN:
+ case NFTA_PAYLOAD_CSUM_OFFSET:
if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0)
abi_breakage();
break;
@@ -106,6 +122,8 @@ nftnl_expr_payload_build(struct nlmsghdr *nlh, struct nftnl_expr *e)
{
struct nftnl_expr_payload *payload = nftnl_expr_data(e);
+ if (e->flags & (1 << NFTNL_EXPR_PAYLOAD_SREG))
+ mnl_attr_put_u32(nlh, NFTA_PAYLOAD_SREG, htonl(payload->sreg));
if (e->flags & (1 << NFTNL_EXPR_PAYLOAD_DREG))
mnl_attr_put_u32(nlh, NFTA_PAYLOAD_DREG, htonl(payload->dreg));
if (e->flags & (1 << NFTNL_EXPR_PAYLOAD_BASE))
@@ -114,6 +132,9 @@ nftnl_expr_payload_build(struct nlmsghdr *nlh, struct nftnl_expr *e)
mnl_attr_put_u32(nlh, NFTA_PAYLOAD_OFFSET, htonl(payload->offset));
if (e->flags & (1 << NFTNL_EXPR_PAYLOAD_LEN))
mnl_attr_put_u32(nlh, NFTA_PAYLOAD_LEN, htonl(payload->len));
+ if (e->flags & (1 << NFT_EXPR_PAYLOAD_CSUM_OFFSET))
+ mnl_attr_put_u32(nlh, NFTA_PAYLOAD_CSUM_OFFSET,
+ htonl(payload->csum_offset));
}
static int
@@ -125,6 +146,10 @@ nftnl_expr_payload_parse(struct nftnl_expr *e, struct nlattr *attr)
if (mnl_attr_parse_nested(attr, nftnl_expr_payload_cb, tb) < 0)
return -1;
+ if (tb[NFTA_PAYLOAD_SREG]) {
+ payload->sreg = ntohl(mnl_attr_get_u32(tb[NFTA_PAYLOAD_SREG]));
+ e->flags |= (1 << NFT_EXPR_PAYLOAD_SREG);
+ }
if (tb[NFTA_PAYLOAD_DREG]) {
payload->dreg = ntohl(mnl_attr_get_u32(tb[NFTA_PAYLOAD_DREG]));
e->flags |= (1 << NFTNL_EXPR_PAYLOAD_DREG);
@@ -141,7 +166,10 @@ nftnl_expr_payload_parse(struct nftnl_expr *e, struct nlattr *attr)
payload->len = ntohl(mnl_attr_get_u32(tb[NFTA_PAYLOAD_LEN]));
e->flags |= (1 << NFTNL_EXPR_PAYLOAD_LEN);
}
-
+ if (tb[NFTA_PAYLOAD_CSUM_OFFSET]) {
+ payload->csum_offset = ntohl(mnl_attr_get_u32(tb[NFTA_PAYLOAD_CSUM_OFFSET]));
+ e->flags |= (1 << NFT_EXPR_PAYLOAD_CSUM_OFFSET);
+ }
return 0;
}
@@ -273,9 +301,15 @@ nftnl_expr_payload_snprintf(char *buf, size_t len, uint32_t type,
switch (type) {
case NFTNL_OUTPUT_DEFAULT:
- return snprintf(buf, len, "load %ub @ %s header + %u => reg %u ",
- payload->len, base2str(payload->base),
- payload->offset, payload->dreg);
+ if (payload->sreg)
+ return snprintf(buf, len, "write reg %u => %ub @ %s header + %u csum_off %u ",
+ payload->sreg,
+ payload->len, base2str(payload->base),
+ payload->offset, payload->csum_offset);
+ else
+ return snprintf(buf, len, "load %ub @ %s header + %u => reg %u ",
+ payload->len, base2str(payload->base),
+ payload->offset, payload->dreg);
case NFTNL_OUTPUT_XML:
case NFTNL_OUTPUT_JSON:
return nftnl_expr_payload_export(buf, len, flags, e, type);
--
2.5.0
^ permalink raw reply related [flat|nested] 2+ messages in thread