13 #include <netinet/in.h>
15 #include <linux/netfilter.h>
16 #include <linux/netfilter/nf_tables.h>
19 #include <libmnl/libmnl.h>
20 #include <libnftnl/object.h>
22 static struct nftnl_obj *obj_add_parse(
int argc,
char *argv[])
24 size_t timeout_array_size;
30 if (strcmp(argv[1],
"ip") == 0)
31 family = NFPROTO_IPV4;
32 else if (strcmp(argv[1],
"ip6") == 0)
33 family = NFPROTO_IPV6;
34 else if (strcmp(argv[1],
"bridge") == 0)
35 family = NFPROTO_BRIDGE;
36 else if (strcmp(argv[1],
"arp") == 0)
39 fprintf(stderr,
"Unknown family: ip, ip6, bridge, arp\n");
43 if (strcmp(argv[4],
"udp") == 0)
44 l4proto = IPPROTO_UDP;
45 else if (strcmp(argv[4],
"tcp") == 0)
46 l4proto = IPPROTO_TCP;
48 fprintf(stderr,
"Unknown layer 4 protocol\n");
52 t = nftnl_obj_alloc();
58 timeout_array_size =
sizeof(uint32_t) * (NFTNL_CTTIMEOUT_TCP_MAX);
59 timeout = calloc(1, timeout_array_size);
60 if (timeout == NULL) {
65 timeout[NFTNL_CTTIMEOUT_TCP_ESTABLISHED] = 111;
66 timeout[NFTNL_CTTIMEOUT_TCP_CLOSE] = 16;
67 timeout[NFTNL_CTTIMEOUT_TCP_CLOSE_WAIT] = 14;
68 nftnl_obj_set_u32(t, NFTNL_OBJ_FAMILY, family);
69 nftnl_obj_set_u32(t, NFTNL_OBJ_TYPE, NFT_OBJECT_CT_TIMEOUT);
70 nftnl_obj_set_str(t, NFTNL_OBJ_TABLE, argv[2]);
71 nftnl_obj_set_str(t, NFTNL_OBJ_NAME, argv[3]);
72 nftnl_obj_set_u8(t, NFTNL_OBJ_CT_TIMEOUT_L4PROTO, l4proto);
73 nftnl_obj_set_u16(t, NFTNL_OBJ_CT_TIMEOUT_L3PROTO, NFPROTO_IPV4);
74 nftnl_obj_set_data(t, NFTNL_OBJ_CT_TIMEOUT_ARRAY,
75 timeout, timeout_array_size);
80 int main(
int argc,
char *argv[])
82 struct mnl_socket *nl;
83 char buf[MNL_SOCKET_BUFFER_SIZE];
85 uint32_t portid, seq, obj_seq, family;
87 struct mnl_nlmsg_batch *batch;
91 fprintf(stderr,
"%s <family> <table> <name> <protocol> \n", argv[0]);
95 t = obj_add_parse(argc, argv);
101 batch = mnl_nlmsg_batch_start(buf,
sizeof(buf));
103 nftnl_batch_begin(mnl_nlmsg_batch_current(batch), seq++);
104 mnl_nlmsg_batch_next(batch);
107 family = nftnl_obj_get_u32(t, NFTNL_OBJ_FAMILY);
108 nlh = nftnl_nlmsg_build_hdr(mnl_nlmsg_batch_current(batch),
109 NFT_MSG_NEWOBJ, family, NLM_F_ACK | NLM_F_CREATE, seq++);
110 nftnl_obj_nlmsg_build_payload(nlh, t);
112 mnl_nlmsg_batch_next(batch);
114 nftnl_batch_end(mnl_nlmsg_batch_current(batch), seq++);
115 mnl_nlmsg_batch_next(batch);
117 nl = mnl_socket_open(NETLINK_NETFILTER);
119 perror(
"mnl_socket_open");
123 if (mnl_socket_bind(nl, 0, MNL_SOCKET_AUTOPID) < 0) {
124 perror(
"mnl_socket_bind");
127 portid = mnl_socket_get_portid(nl);
129 if (mnl_socket_sendto(nl, mnl_nlmsg_batch_head(batch),
130 mnl_nlmsg_batch_size(batch)) < 0) {
131 perror(
"mnl_socket_send");
135 mnl_nlmsg_batch_stop(batch);
137 ret = mnl_socket_recvfrom(nl, buf,
sizeof(buf));
139 ret = mnl_cb_run(buf, ret, obj_seq, portid, NULL, NULL);
142 ret = mnl_socket_recvfrom(nl, buf,
sizeof(buf));
148 mnl_socket_close(nl);