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 mt->rev = *((uint32_t *)data);
51 case 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]);
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);
167 static int nftnl_expr_match_json_parse(
struct nftnl_expr *e, json_t *root,
168 struct nftnl_parse_err *err)
173 name = nftnl_jansson_parse_str(root,
"name", err);
175 nftnl_expr_set_str(e, NFTNL_EXPR_MT_NAME, name);
185 static int nftnl_expr_match_xml_parse(
struct nftnl_expr *e, mxml_node_t *tree,
186 struct nftnl_parse_err *err)
191 name = nftnl_mxml_str_parse(tree,
"name", MXML_DESCEND_FIRST,
192 NFTNL_XML_MAND, err);
194 nftnl_expr_set_str(e, NFTNL_EXPR_MT_NAME, name);
205 static int nftnl_expr_match_export(
char *buf,
size_t size,
206 const struct nftnl_expr *e,
int type)
209 NFTNL_BUF_INIT(b, buf, size);
211 if (e->flags & (1 << NFTNL_EXPR_MT_NAME))
212 nftnl_buf_str(&b, type, mt->name, NAME);
214 return nftnl_buf_done(&b);
218 nftnl_expr_match_snprintf(
char *buf,
size_t len, uint32_t type,
219 uint32_t flags,
const struct nftnl_expr *e)
224 case NFTNL_OUTPUT_DEFAULT:
225 return snprintf(buf, len,
"name %s rev %u ",
226 match->name, match->rev);
227 case NFTNL_OUTPUT_XML:
228 case NFTNL_OUTPUT_JSON:
229 return nftnl_expr_match_export(buf, len, e, type);
236 static void nftnl_expr_match_free(
const struct nftnl_expr *e)
243 struct expr_ops expr_ops_match = {
246 .max_attr = NFTA_MATCH_MAX,
247 .free = nftnl_expr_match_free,
248 .set = nftnl_expr_match_set,
249 .get = nftnl_expr_match_get,
250 .parse = nftnl_expr_match_parse,
251 .build = nftnl_expr_match_build,
252 .snprintf = nftnl_expr_match_snprintf,
253 .xml_parse = nftnl_expr_match_xml_parse,
254 .json_parse = nftnl_expr_match_json_parse,