From 2fad0c845c6dee51d8756a4f558783d8a2344784 Mon Sep 17 00:00:00 2001 From: Ana Rey Date: Tue, 19 Nov 2013 13:01:56 +0100 Subject: table/chain: add u8 setter and getter for family values These are needed to set the family value for chain and table. Signed-off-by: Ana Rey Signed-off-by: Pablo Neira Ayuso --- src/chain.c | 13 +++++++++++++ src/libnftables.map | 4 ++++ src/table.c | 13 +++++++++++++ 3 files changed, 30 insertions(+) (limited to 'src') diff --git a/src/chain.c b/src/chain.c index a58999d..0145d32 100644 --- a/src/chain.c +++ b/src/chain.c @@ -206,6 +206,12 @@ void nft_chain_attr_set_u64(struct nft_chain *c, uint16_t attr, uint64_t data) } EXPORT_SYMBOL(nft_chain_attr_set_u64); +void nft_chain_attr_set_u8(struct nft_chain *c, uint16_t attr, uint8_t data) +{ + nft_chain_attr_set(c, attr, &data); +} +EXPORT_SYMBOL(nft_chain_attr_set_u8); + void nft_chain_attr_set_str(struct nft_chain *c, uint16_t attr, const char *str) { nft_chain_attr_set(c, attr, str); @@ -272,6 +278,13 @@ uint64_t nft_chain_attr_get_u64(struct nft_chain *c, uint16_t attr) } EXPORT_SYMBOL(nft_chain_attr_get_u64); +uint8_t nft_chain_attr_get_u8(struct nft_chain *c, uint16_t attr) +{ + const uint8_t *val = nft_chain_attr_get(c, attr); + return val ? *val : 0; +} +EXPORT_SYMBOL(nft_chain_attr_get_u8); + struct nlmsghdr * nft_chain_nlmsg_build_hdr(char *buf, uint16_t cmd, uint16_t family, uint16_t type, uint32_t seq) diff --git a/src/libnftables.map b/src/libnftables.map index aa47714..3bdfeda 100644 --- a/src/libnftables.map +++ b/src/libnftables.map @@ -6,8 +6,10 @@ global: nft_table_attr_unset; nft_table_attr_set; nft_table_attr_get; + nft_table_attr_set_u8; nft_table_attr_set_u32; nft_table_attr_set_str; + nft_table_attr_get_u8; nft_table_attr_get_u32; nft_table_attr_get_str; nft_table_parse; @@ -32,11 +34,13 @@ global: nft_chain_attr_is_set; nft_chain_attr_unset; nft_chain_attr_set; + nft_chain_attr_set_u8; nft_chain_attr_set_u32; nft_chain_attr_set_s32; nft_chain_attr_set_u64; nft_chain_attr_set_str; nft_chain_attr_get; + nft_chain_attr_get_u8; nft_chain_attr_get_u32; nft_chain_attr_get_s32; nft_chain_attr_get_u64; diff --git a/src/table.c b/src/table.c index fe37f90..cf24cae 100644 --- a/src/table.c +++ b/src/table.c @@ -103,6 +103,12 @@ void nft_table_attr_set_u32(struct nft_table *t, uint16_t attr, uint32_t val) } EXPORT_SYMBOL(nft_table_attr_set_u32); +void nft_table_attr_set_u8(struct nft_table *t, uint16_t attr, uint8_t val) +{ + nft_table_attr_set(t, attr, &val); +} +EXPORT_SYMBOL(nft_table_attr_set_u8); + void nft_table_attr_set_str(struct nft_table *t, uint16_t attr, const char *str) { nft_table_attr_set(t, attr, str); @@ -133,6 +139,13 @@ uint32_t nft_table_attr_get_u32(struct nft_table *t, uint16_t attr) } EXPORT_SYMBOL(nft_table_attr_get_u32); +uint8_t nft_table_attr_get_u8(struct nft_table *t, uint16_t attr) +{ + const void *ret = nft_table_attr_get(t, attr); + return ret == NULL ? 0 : *((uint8_t *)ret); +} +EXPORT_SYMBOL(nft_table_attr_get_u8); + const char *nft_table_attr_get_str(struct nft_table *t, uint16_t attr) { return nft_table_attr_get(t, attr); -- cgit v1.2.3