18 #include <netinet/in.h>
19 #include <sys/types.h>
24 #include <linux/netfilter.h>
25 #include <linux/netfilter/nf_tables.h>
26 #include <linux/netfilter/nfnetlink.h>
28 #include <libmnl/libmnl.h>
29 #include <libnftnl/set.h>
31 static struct nftnl_set *set_parse_file(
const char *file, uint16_t format)
35 struct nftnl_parse_err *err;
38 s = nftnl_set_alloc();
44 fd = open(file, O_RDONLY);
50 if (read(fd, data,
sizeof(data)) < 0) {
57 err = nftnl_parse_err_alloc();
63 if (nftnl_set_parse(s, format, data, err) < 0) {
64 nftnl_parse_perror(
"Unable to parse file", err);
65 nftnl_parse_err_free(err);
69 nftnl_parse_err_free(err);
71 nftnl_set_set_u32(s, NFTNL_SET_ID, 1);
76 int main(
int argc,
char *argv[])
78 struct mnl_socket *nl;
79 char buf[MNL_SOCKET_BUFFER_SIZE];
81 uint32_t portid, seq, set_seq;
84 uint16_t family, format, outformat;
85 struct mnl_nlmsg_batch *batch;
88 printf(
"Usage: %s {xml|json} <file>\n", argv[0]);
92 if (strcmp(argv[1],
"xml") == 0) {
93 format = NFTNL_PARSE_XML;
94 outformat = NFTNL_OUTPUT_XML;
95 }
else if (strcmp(argv[1],
"json") == 0) {
96 format = NFTNL_PARSE_JSON;
97 outformat = NFTNL_OUTPUT_JSON;
99 printf(
"Unknown format: xml, json\n");
103 s = set_parse_file(argv[2], format);
107 nftnl_set_fprintf(stdout, s, outformat, 0);
108 fprintf(stdout,
"\n");
111 batching = nftnl_batch_is_supported();
113 perror(
"cannot talk to nfnetlink");
117 batch = mnl_nlmsg_batch_start(buf,
sizeof(buf));
120 nftnl_batch_begin(mnl_nlmsg_batch_current(batch), seq++);
121 mnl_nlmsg_batch_next(batch);
124 family = nftnl_set_get_u32(s, NFTNL_SET_FAMILY);
127 nlh = nftnl_set_nlmsg_build_hdr(mnl_nlmsg_batch_current(batch),
128 NFT_MSG_NEWSET, family,
129 NLM_F_CREATE|NLM_F_ACK, seq++);
130 nftnl_set_nlmsg_build_payload(nlh, s);
132 mnl_nlmsg_batch_next(batch);
135 nftnl_batch_end(mnl_nlmsg_batch_current(batch), seq++);
136 mnl_nlmsg_batch_next(batch);
139 nl = mnl_socket_open(NETLINK_NETFILTER);
141 perror(
"mnl_socket_open");
145 if (mnl_socket_bind(nl, 0, MNL_SOCKET_AUTOPID) < 0) {
146 perror(
"mnl_socket_bind");
149 portid = mnl_socket_get_portid(nl);
151 if (mnl_socket_sendto(nl, mnl_nlmsg_batch_head(batch),
152 mnl_nlmsg_batch_size(batch)) < 0) {
153 perror(
"mnl_socket_send");
157 mnl_nlmsg_batch_stop(batch);
159 ret = mnl_socket_recvfrom(nl, buf,
sizeof(buf));
161 ret = mnl_cb_run(buf, ret, set_seq, portid, NULL, NULL);
164 ret = mnl_socket_recvfrom(nl, buf,
sizeof(buf));
171 mnl_socket_close(nl);