17 #include <arpa/inet.h> 19 #include <libmnl/libmnl.h> 21 #include <linux/netfilter/nf_tables.h> 22 #include <linux/netfilter/nf_tables_compat.h> 24 #include <libnftnl/expr.h> 25 #include <libnftnl/rule.h> 28 #define XT_EXTENSION_MAXNAMELEN 29 31 char name[XT_EXTENSION_MAXNAMELEN];
38 nftnl_expr_match_set(
struct nftnl_expr *e, uint16_t type,
39 const void *data, uint32_t data_len)
44 case NFTNL_EXPR_MT_NAME:
45 snprintf(mt->name,
sizeof(mt->name),
"%.*s", data_len,
48 case NFTNL_EXPR_MT_REV:
49 memcpy(&mt->rev, data,
sizeof(mt->rev));
51 case NFTNL_EXPR_MT_INFO:
52 if (e->flags & (1 << NFTNL_EXPR_MT_INFO))
56 mt->data_len = data_len;
65 nftnl_expr_match_get(
const struct nftnl_expr *e, uint16_t type,
71 case NFTNL_EXPR_MT_NAME:
72 *data_len =
sizeof(mt->name);
74 case NFTNL_EXPR_MT_REV:
75 *data_len =
sizeof(mt->rev);
77 case NFTNL_EXPR_MT_INFO:
78 *data_len = mt->data_len;
84 static int nftnl_expr_match_cb(
const struct nlattr *attr,
void *data)
86 const struct nlattr **tb = data;
87 int type = mnl_attr_get_type(attr);
89 if (mnl_attr_type_valid(attr, NFTA_MATCH_MAX) < 0)
94 if (mnl_attr_validate(attr, MNL_TYPE_NUL_STRING) < 0)
98 if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0)
101 case NFTA_MATCH_INFO:
102 if (mnl_attr_validate(attr, MNL_TYPE_BINARY) < 0)
112 nftnl_expr_match_build(
struct nlmsghdr *nlh,
const struct nftnl_expr *e)
116 if (e->flags & (1 << NFTNL_EXPR_MT_NAME))
117 mnl_attr_put_strz(nlh, NFTA_MATCH_NAME, mt->name);
118 if (e->flags & (1 << NFTNL_EXPR_MT_REV))
119 mnl_attr_put_u32(nlh, NFTA_MATCH_REV, htonl(mt->rev));
120 if (e->flags & (1 << NFTNL_EXPR_MT_INFO))
121 mnl_attr_put(nlh, NFTA_MATCH_INFO, mt->data_len, mt->data);
124 static int nftnl_expr_match_parse(
struct nftnl_expr *e,
struct nlattr *attr)
127 struct nlattr *tb[NFTA_MATCH_MAX+1] = {};
129 if (mnl_attr_parse_nested(attr, nftnl_expr_match_cb, tb) < 0)
132 if (tb[NFTA_MATCH_NAME]) {
133 snprintf(match->name, XT_EXTENSION_MAXNAMELEN,
"%s",
134 mnl_attr_get_str(tb[NFTA_MATCH_NAME]));
136 match->name[XT_EXTENSION_MAXNAMELEN-1] =
'\0';
137 e->flags |= (1 << NFTNL_EXPR_MT_NAME);
140 if (tb[NFTA_MATCH_REV]) {
141 match->rev = ntohl(mnl_attr_get_u32(tb[NFTA_MATCH_REV]));
142 e->flags |= (1 << NFTNL_EXPR_MT_REV);
145 if (tb[NFTA_MATCH_INFO]) {
146 uint32_t len = mnl_attr_get_payload_len(tb[NFTA_MATCH_INFO]);
149 if (e->flags & (1 << NFTNL_EXPR_MT_INFO))
152 match_data = calloc(1, len);
153 if (match_data == NULL)
156 memcpy(match_data, mnl_attr_get_payload(tb[NFTA_MATCH_INFO]), len);
158 match->data = match_data;
159 match->data_len = len;
161 e->flags |= (1 << NFTNL_EXPR_MT_INFO);
168 nftnl_expr_match_snprintf(
char *buf,
size_t len, uint32_t type,
169 uint32_t flags,
const struct nftnl_expr *e)
174 case NFTNL_OUTPUT_DEFAULT:
175 return snprintf(buf, len,
"name %s rev %u ",
176 match->name, match->rev);
177 case NFTNL_OUTPUT_XML:
178 case NFTNL_OUTPUT_JSON:
185 static void nftnl_expr_match_free(
const struct nftnl_expr *e)
192 struct expr_ops expr_ops_match = {
195 .max_attr = NFTA_MATCH_MAX,
196 .free = nftnl_expr_match_free,
197 .set = nftnl_expr_match_set,
198 .get = nftnl_expr_match_get,
199 .parse = nftnl_expr_match_parse,
200 .build = nftnl_expr_match_build,
201 .snprintf = nftnl_expr_match_snprintf,