14 #include <arpa/inet.h>
16 #include <linux/netfilter/nf_tables.h>
19 #include <libmnl/libmnl.h>
20 #include <libnftnl/expr.h>
21 #include <libnftnl/rule.h>
24 enum nft_registers sreg_qnum;
26 uint16_t queues_total;
30 static int nftnl_expr_queue_set(
struct nftnl_expr *e, uint16_t type,
31 const void *data, uint32_t data_len)
36 case NFTNL_EXPR_QUEUE_NUM:
37 queue->queuenum = *((uint16_t *)data);
39 case NFTNL_EXPR_QUEUE_TOTAL:
40 queue->queues_total = *((uint16_t *)data);
42 case NFTNL_EXPR_QUEUE_FLAGS:
43 queue->flags = *((uint16_t *)data);
45 case NFTNL_EXPR_QUEUE_SREG_QNUM:
46 queue->sreg_qnum = *((uint32_t *)data);
55 nftnl_expr_queue_get(
const struct nftnl_expr *e, uint16_t type,
61 case NFTNL_EXPR_QUEUE_NUM:
62 *data_len =
sizeof(queue->queuenum);
63 return &queue->queuenum;
64 case NFTNL_EXPR_QUEUE_TOTAL:
65 *data_len =
sizeof(queue->queues_total);
66 return &queue->queues_total;
67 case NFTNL_EXPR_QUEUE_FLAGS:
68 *data_len =
sizeof(queue->flags);
70 case NFTNL_EXPR_QUEUE_SREG_QNUM:
71 *data_len =
sizeof(queue->sreg_qnum);
72 return &queue->sreg_qnum;
77 static int nftnl_expr_queue_cb(
const struct nlattr *attr,
void *data)
79 const struct nlattr **tb = data;
80 int type = mnl_attr_get_type(attr);
82 if (mnl_attr_type_valid(attr, NFTA_QUEUE_MAX) < 0)
87 case NFTA_QUEUE_TOTAL:
88 case NFTA_QUEUE_FLAGS:
89 if (mnl_attr_validate(attr, MNL_TYPE_U16) < 0)
92 case NFTA_QUEUE_SREG_QNUM:
93 if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0)
103 nftnl_expr_queue_build(
struct nlmsghdr *nlh,
const struct nftnl_expr *e)
107 if (e->flags & (1 << NFTNL_EXPR_QUEUE_NUM))
108 mnl_attr_put_u16(nlh, NFTA_QUEUE_NUM, htons(queue->queuenum));
109 if (e->flags & (1 << NFTNL_EXPR_QUEUE_TOTAL))
110 mnl_attr_put_u16(nlh, NFTA_QUEUE_TOTAL, htons(queue->queues_total));
111 if (e->flags & (1 << NFTNL_EXPR_QUEUE_FLAGS))
112 mnl_attr_put_u16(nlh, NFTA_QUEUE_FLAGS, htons(queue->flags));
113 if (e->flags & (1 << NFTNL_EXPR_QUEUE_SREG_QNUM))
114 mnl_attr_put_u32(nlh, NFTA_QUEUE_SREG_QNUM, htonl(queue->sreg_qnum));
118 nftnl_expr_queue_parse(
struct nftnl_expr *e,
struct nlattr *attr)
121 struct nlattr *tb[NFTA_QUEUE_MAX+1] = {};
123 if (mnl_attr_parse_nested(attr, nftnl_expr_queue_cb, tb) < 0)
126 if (tb[NFTA_QUEUE_NUM]) {
127 queue->queuenum = ntohs(mnl_attr_get_u16(tb[NFTA_QUEUE_NUM]));
128 e->flags |= (1 << NFTNL_EXPR_QUEUE_NUM);
130 if (tb[NFTA_QUEUE_TOTAL]) {
131 queue->queues_total = ntohs(mnl_attr_get_u16(tb[NFTA_QUEUE_TOTAL]));
132 e->flags |= (1 << NFTNL_EXPR_QUEUE_TOTAL);
134 if (tb[NFTA_QUEUE_FLAGS]) {
135 queue->flags = ntohs(mnl_attr_get_u16(tb[NFTA_QUEUE_FLAGS]));
136 e->flags |= (1 << NFTNL_EXPR_QUEUE_FLAGS);
138 if (tb[NFTA_QUEUE_SREG_QNUM]) {
139 queue->sreg_qnum = ntohl(mnl_attr_get_u32(tb[NFTA_QUEUE_SREG_QNUM]));
140 e->flags |= (1 << NFTNL_EXPR_QUEUE_SREG_QNUM);
147 nftnl_expr_queue_json_parse(
struct nftnl_expr *e, json_t *root,
148 struct nftnl_parse_err *err)
155 if (nftnl_jansson_parse_val(root,
"num", NFTNL_TYPE_U16, &type, err) == 0)
156 nftnl_expr_set_u16(e, NFTNL_EXPR_QUEUE_NUM, type);
158 if (nftnl_jansson_parse_val(root,
"total", NFTNL_TYPE_U16, &code, err) == 0)
159 nftnl_expr_set_u16(e, NFTNL_EXPR_QUEUE_TOTAL, code);
161 if (nftnl_jansson_parse_val(root,
"flags", NFTNL_TYPE_U16, &code, err) == 0)
162 nftnl_expr_set_u16(e, NFTNL_EXPR_QUEUE_FLAGS, code);
164 if (nftnl_jansson_parse_val(root,
"sreg_qnum", NFTNL_TYPE_U32, &sreg_qnum,
166 nftnl_expr_set_u32(e, NFTNL_EXPR_QUEUE_SREG_QNUM, sreg_qnum);
175 static int nftnl_expr_queue_snprintf_default(
char *buf,
size_t len,
176 const struct nftnl_expr *e)
179 int ret, remain = len, offset = 0;
180 uint16_t total_queues;
182 if (e->flags & (1 << NFTNL_EXPR_QUEUE_NUM)) {
183 total_queues = queue->queuenum + queue->queues_total - 1;
185 ret = snprintf(buf + offset, len,
"num %u", queue->queuenum);
186 SNPRINTF_BUFFER_SIZE(ret, remain, offset);
188 if (queue->queues_total && total_queues != queue->queuenum) {
189 ret = snprintf(buf + offset, len,
"-%u", total_queues);
190 SNPRINTF_BUFFER_SIZE(ret, remain, offset);
193 ret = snprintf(buf + offset, len,
" ");
194 SNPRINTF_BUFFER_SIZE(ret, remain, offset);
197 if (e->flags & (1 << NFTNL_EXPR_QUEUE_SREG_QNUM)) {
198 ret = snprintf(buf + offset, len,
"sreg_qnum %u ",
200 SNPRINTF_BUFFER_SIZE(ret, remain, offset);
203 if (e->flags & (1 << NFTNL_EXPR_QUEUE_FLAGS)) {
204 if (queue->flags & (NFT_QUEUE_FLAG_BYPASS)) {
205 ret = snprintf(buf + offset, len,
"bypass ");
206 SNPRINTF_BUFFER_SIZE(ret, remain, offset);
208 if (queue->flags & (NFT_QUEUE_FLAG_CPU_FANOUT)) {
209 ret = snprintf(buf + offset, len,
"fanout ");
210 SNPRINTF_BUFFER_SIZE(ret, remain, offset);
216 static int nftnl_expr_queue_export(
char *buf,
size_t size,
217 const struct nftnl_expr *e,
int type)
220 NFTNL_BUF_INIT(b, buf, size);
222 if (e->flags & (1 << NFTNL_EXPR_QUEUE_NUM))
223 nftnl_buf_u32(&b, type, queue->queuenum, NUM);
224 if (e->flags & (1 << NFTNL_EXPR_QUEUE_TOTAL))
225 nftnl_buf_u32(&b, type, queue->queues_total, TOTAL);
226 if (e->flags & (1 << NFTNL_EXPR_QUEUE_FLAGS))
227 nftnl_buf_u32(&b, type, queue->flags, FLAGS);
228 if (e->flags & (1 << NFTNL_EXPR_QUEUE_SREG_QNUM))
229 nftnl_buf_u32(&b, type, queue->sreg_qnum, SREG_QNUM);
231 return nftnl_buf_done(&b);
235 nftnl_expr_queue_snprintf(
char *buf,
size_t len, uint32_t type,
236 uint32_t flags,
const struct nftnl_expr *e)
239 case NFTNL_OUTPUT_DEFAULT:
240 return nftnl_expr_queue_snprintf_default(buf, len, e);
241 case NFTNL_OUTPUT_XML:
242 case NFTNL_OUTPUT_JSON:
243 return nftnl_expr_queue_export(buf, len, e, type);
250 static bool nftnl_expr_queue_cmp(
const struct nftnl_expr *e1,
251 const struct nftnl_expr *e2)
257 if (e1->flags & (1 << NFTNL_EXPR_QUEUE_NUM))
258 eq &= (q1->queuenum == q2->queuenum);
259 if (e1->flags & (1 << NFTNL_EXPR_QUEUE_TOTAL))
260 eq &= (q1->queues_total == q2->queues_total);
261 if (e1->flags & (1 << NFTNL_EXPR_QUEUE_FLAGS))
262 eq &= (q1->flags == q2->flags);
263 if (e1->flags & (1 << NFTNL_EXPR_QUEUE_SREG_QNUM))
264 eq &= (q1->sreg_qnum == q2->sreg_qnum);
269 struct expr_ops expr_ops_queue = {
272 .max_attr = NFTA_QUEUE_MAX,
273 .cmp = nftnl_expr_queue_cmp,
274 .set = nftnl_expr_queue_set,
275 .get = nftnl_expr_queue_get,
276 .parse = nftnl_expr_queue_parse,
277 .build = nftnl_expr_queue_build,
278 .snprintf = nftnl_expr_queue_snprintf,
279 .json_parse = nftnl_expr_queue_json_parse,