diff options
author | Pablo Neira Ayuso <pablo@netfilter.org> | 2019-12-17 18:16:55 +0100 |
---|---|---|
committer | Florian Westphal <fw@strlen.de> | 2019-12-17 23:10:46 +0100 |
commit | 0ceb0151bda7bb947c9313d16eada75c41c51719 (patch) | |
tree | 7153d1516d99ffe023a2299dfc165dc04d2e222b /src/socket.c | |
parent | 01fbc1574b9eec352528b5e6a909d8c6b61633d2 (diff) |
socket: add parse and build userdata interface
Add support for meta userdata area.
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
Signed-off-by: Florian Westphal <fw@strlen.de>
Diffstat (limited to 'src/socket.c')
-rw-r--r-- | src/socket.c | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/src/socket.c b/src/socket.c index e10b3226..d78a163a 100644 --- a/src/socket.c +++ b/src/socket.c @@ -43,6 +43,55 @@ static void socket_expr_clone(struct expr *new, const struct expr *expr) new->socket.key = expr->socket.key; } +#define NFTNL_UDATA_SOCKET_KEY 0 +#define NFTNL_UDATA_SOCKET_MAX 1 + +static int socket_expr_build_udata(struct nftnl_udata_buf *udbuf, + const struct expr *expr) +{ + nftnl_udata_put_u32(udbuf, NFTNL_UDATA_SOCKET_KEY, expr->socket.key); + + return 0; +} + +static int socket_parse_udata(const struct nftnl_udata *attr, void *data) +{ + const struct nftnl_udata **ud = data; + uint8_t type = nftnl_udata_type(attr); + uint8_t len = nftnl_udata_len(attr); + + switch (type) { + case NFTNL_UDATA_SOCKET_KEY: + if (len != sizeof(uint32_t)) + return -1; + break; + default: + return 0; + } + + ud[type] = attr; + return 0; +} + +static struct expr *socket_expr_parse_udata(const struct nftnl_udata *attr) +{ + const struct nftnl_udata *ud[NFTNL_UDATA_SOCKET_MAX + 1] = {}; + uint32_t key; + int err; + + err = nftnl_udata_parse(nftnl_udata_get(attr), nftnl_udata_len(attr), + socket_parse_udata, ud); + if (err < 0) + return NULL; + + if (!ud[NFTNL_UDATA_SOCKET_KEY]) + return NULL; + + key = nftnl_udata_get_u32(ud[NFTNL_UDATA_SOCKET_KEY]); + + return socket_expr_alloc(&internal_location, key); +} + const struct expr_ops socket_expr_ops = { .type = EXPR_SOCKET, .name = "socket", @@ -50,6 +99,8 @@ const struct expr_ops socket_expr_ops = { .json = socket_expr_json, .cmp = socket_expr_cmp, .clone = socket_expr_clone, + .build_udata = socket_expr_build_udata, + .parse_udata = socket_expr_parse_udata, }; struct expr *socket_expr_alloc(const struct location *loc, enum nft_socket_keys key) |