12#include <linux/netfilter/nf_tables.h>
15#include <libmnl/libmnl.h>
16#include <libnftnl/expr.h>
17#include <libnftnl/rule.h>
31static int nftnl_expr_objref_set(
struct nftnl_expr *e, uint16_t type,
32 const void *data, uint32_t data_len)
37 case NFTNL_EXPR_OBJREF_IMM_TYPE:
38 memcpy(&objref->imm.type, data, data_len);
40 case NFTNL_EXPR_OBJREF_IMM_NAME:
41 objref->imm.name = strdup(data);
42 if (!objref->imm.name)
45 case NFTNL_EXPR_OBJREF_SET_SREG:
46 memcpy(&objref->set.sreg, data, data_len);
48 case NFTNL_EXPR_OBJREF_SET_NAME:
49 objref->set.name = strdup(data);
50 if (!objref->set.name)
53 case NFTNL_EXPR_OBJREF_SET_ID:
54 memcpy(&objref->set.id, data, data_len);
60static const void *nftnl_expr_objref_get(
const struct nftnl_expr *e,
61 uint16_t type, uint32_t *data_len)
66 case NFTNL_EXPR_OBJREF_IMM_TYPE:
67 *data_len =
sizeof(objref->imm.type);
68 return &objref->imm.type;
69 case NFTNL_EXPR_OBJREF_IMM_NAME:
70 *data_len = strlen(objref->imm.name) + 1;
71 return objref->imm.name;
72 case NFTNL_EXPR_OBJREF_SET_SREG:
73 *data_len =
sizeof(objref->set.sreg);
74 return &objref->set.sreg;
75 case NFTNL_EXPR_OBJREF_SET_NAME:
76 *data_len = strlen(objref->set.name) + 1;
77 return objref->set.name;
78 case NFTNL_EXPR_OBJREF_SET_ID:
79 *data_len =
sizeof(objref->set.id);
80 return &objref->set.id;
85static int nftnl_expr_objref_cb(
const struct nlattr *attr,
void *data)
87 int type = mnl_attr_get_type(attr);
88 const struct nlattr **tb = data;
90 if (mnl_attr_type_valid(attr, NFTA_OBJREF_MAX) < 0)
94 case NFTA_OBJREF_IMM_TYPE:
95 if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0)
98 case NFTA_OBJREF_IMM_NAME:
99 case NFTA_OBJREF_SET_NAME:
100 if (mnl_attr_validate(attr, MNL_TYPE_STRING) < 0)
103 case NFTA_OBJREF_SET_SREG:
104 case NFTA_OBJREF_SET_ID:
105 if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0)
114static void nftnl_expr_objref_build(
struct nlmsghdr *nlh,
115 const struct nftnl_expr *e)
119 if (e->flags & (1 << NFTNL_EXPR_OBJREF_IMM_TYPE))
120 mnl_attr_put_u32(nlh, NFTA_OBJREF_IMM_TYPE,
121 htonl(objref->imm.type));
122 if (e->flags & (1 << NFTNL_EXPR_OBJREF_IMM_NAME))
123 mnl_attr_put_str(nlh, NFTA_OBJREF_IMM_NAME, objref->imm.name);
124 if (e->flags & (1 << NFTNL_EXPR_OBJREF_SET_SREG))
125 mnl_attr_put_u32(nlh, NFTA_OBJREF_SET_SREG,
126 htonl(objref->set.sreg));
127 if (e->flags & (1 << NFTNL_EXPR_OBJREF_SET_NAME))
128 mnl_attr_put_str(nlh, NFTA_OBJREF_SET_NAME, objref->set.name);
129 if (e->flags & (1 << NFTNL_EXPR_OBJREF_SET_ID))
130 mnl_attr_put_u32(nlh, NFTA_OBJREF_SET_ID,
131 htonl(objref->set.id));
134static int nftnl_expr_objref_parse(
struct nftnl_expr *e,
struct nlattr *attr)
137 struct nlattr *tb[NFTA_OBJREF_MAX + 1] = {};
139 if (mnl_attr_parse_nested(attr, nftnl_expr_objref_cb, tb) < 0)
142 if (tb[NFTA_OBJREF_IMM_TYPE]) {
144 ntohl(mnl_attr_get_u32(tb[NFTA_OBJREF_IMM_TYPE]));
145 e->flags |= (1 << NFTNL_EXPR_OBJREF_IMM_TYPE);
147 if (tb[NFTA_OBJREF_IMM_NAME]) {
149 strdup(mnl_attr_get_str(tb[NFTA_OBJREF_IMM_NAME]));
150 e->flags |= (1 << NFTNL_EXPR_OBJREF_IMM_NAME);
152 if (tb[NFTA_OBJREF_SET_SREG]) {
154 ntohl(mnl_attr_get_u32(tb[NFTA_OBJREF_SET_SREG]));
155 e->flags |= (1 << NFTNL_EXPR_OBJREF_SET_SREG);
157 if (tb[NFTA_OBJREF_SET_NAME]) {
159 strdup(mnl_attr_get_str(tb[NFTA_OBJREF_SET_NAME]));
160 e->flags |= (1 << NFTNL_EXPR_OBJREF_SET_NAME);
162 if (tb[NFTA_OBJREF_SET_ID]) {
164 ntohl(mnl_attr_get_u32(tb[NFTA_OBJREF_SET_ID]));
165 e->flags |= (1 << NFTNL_EXPR_OBJREF_SET_ID);
171static int nftnl_expr_objref_snprintf(
char *buf,
size_t len,
173 const struct nftnl_expr *e)
177 if (e->flags & (1 << NFTNL_EXPR_OBJREF_SET_SREG))
178 return snprintf(buf, len,
"sreg %u set %s ",
179 objref->set.sreg, objref->set.name);
181 return snprintf(buf, len,
"type %u name %s ",
182 objref->imm.type, objref->imm.name);
185static void nftnl_expr_objref_free(
const struct nftnl_expr *e)
189 xfree(objref->imm.name);
190 xfree(objref->set.name);
193static struct attr_policy objref_attr_policy[__NFTNL_EXPR_OBJREF_MAX] = {
194 [NFTNL_EXPR_OBJREF_IMM_TYPE] = { .maxlen =
sizeof(uint32_t) },
195 [NFTNL_EXPR_OBJREF_IMM_NAME] = { .maxlen = NFT_NAME_MAXLEN },
196 [NFTNL_EXPR_OBJREF_SET_SREG] = { .maxlen =
sizeof(uint32_t) },
197 [NFTNL_EXPR_OBJREF_SET_NAME] = { .maxlen = NFT_NAME_MAXLEN },
198 [NFTNL_EXPR_OBJREF_SET_ID] = { .maxlen =
sizeof(uint32_t) },
201struct expr_ops expr_ops_objref = {
204 .nftnl_max_attr = __NFTNL_EXPR_OBJREF_MAX - 1,
205 .attr_policy = objref_attr_policy,
206 .free = nftnl_expr_objref_free,
207 .set = nftnl_expr_objref_set,
208 .get = nftnl_expr_objref_get,
209 .parse = nftnl_expr_objref_parse,
210 .build = nftnl_expr_objref_build,
211 .output = nftnl_expr_objref_snprintf,