11#include <linux/netfilter.h>
12#include <linux/netfilter/nf_tables.h>
15#include <libmnl/libmnl.h>
16#include <libnftnl/object.h>
18static struct nftnl_obj *obj_add_parse(
int argc,
char *argv[])
20 size_t timeout_array_size;
26 if (strcmp(argv[1],
"ip") == 0)
27 family = NFPROTO_IPV4;
28 else if (strcmp(argv[1],
"ip6") == 0)
29 family = NFPROTO_IPV6;
30 else if (strcmp(argv[1],
"inet") == 0)
31 family = NFPROTO_INET;
32 else if (strcmp(argv[1],
"bridge") == 0)
33 family = NFPROTO_BRIDGE;
34 else if (strcmp(argv[1],
"arp") == 0)
37 fprintf(stderr,
"Unknown family: ip, ip6, inet, bridge, arp\n");
41 if (strcmp(argv[4],
"udp") == 0)
42 l4proto = IPPROTO_UDP;
43 else if (strcmp(argv[4],
"tcp") == 0)
44 l4proto = IPPROTO_TCP;
46 fprintf(stderr,
"Unknown layer 4 protocol\n");
50 t = nftnl_obj_alloc();
56 timeout_array_size =
sizeof(uint32_t) * (NFTNL_CTTIMEOUT_TCP_MAX);
57 timeout = calloc(1, timeout_array_size);
58 if (timeout == NULL) {
63 timeout[NFTNL_CTTIMEOUT_TCP_ESTABLISHED] = 111;
64 timeout[NFTNL_CTTIMEOUT_TCP_CLOSE] = 16;
65 timeout[NFTNL_CTTIMEOUT_TCP_CLOSE_WAIT] = 14;
66 nftnl_obj_set_u32(t, NFTNL_OBJ_FAMILY, family);
67 nftnl_obj_set_u32(t, NFTNL_OBJ_TYPE, NFT_OBJECT_CT_TIMEOUT);
68 nftnl_obj_set_str(t, NFTNL_OBJ_TABLE, argv[2]);
69 nftnl_obj_set_str(t, NFTNL_OBJ_NAME, argv[3]);
70 nftnl_obj_set_u8(t, NFTNL_OBJ_CT_TIMEOUT_L4PROTO, l4proto);
71 nftnl_obj_set_u16(t, NFTNL_OBJ_CT_TIMEOUT_L3PROTO, NFPROTO_IPV4);
72 nftnl_obj_set_data(t, NFTNL_OBJ_CT_TIMEOUT_ARRAY,
73 timeout, timeout_array_size);
78int main(
int argc,
char *argv[])
80 struct mnl_socket *nl;
81 char buf[MNL_SOCKET_BUFFER_SIZE];
83 uint32_t portid, seq, obj_seq, family;
85 struct mnl_nlmsg_batch *batch;
89 fprintf(stderr,
"%s <family> <table> <name> <protocol> \n", argv[0]);
93 t = obj_add_parse(argc, argv);
99 batch = mnl_nlmsg_batch_start(buf,
sizeof(buf));
101 nftnl_batch_begin(mnl_nlmsg_batch_current(batch), seq++);
102 mnl_nlmsg_batch_next(batch);
105 family = nftnl_obj_get_u32(t, NFTNL_OBJ_FAMILY);
106 nlh = nftnl_nlmsg_build_hdr(mnl_nlmsg_batch_current(batch),
107 NFT_MSG_NEWOBJ, family, NLM_F_ACK | NLM_F_CREATE, seq++);
108 nftnl_obj_nlmsg_build_payload(nlh, t);
110 mnl_nlmsg_batch_next(batch);
112 nftnl_batch_end(mnl_nlmsg_batch_current(batch), seq++);
113 mnl_nlmsg_batch_next(batch);
115 nl = mnl_socket_open(NETLINK_NETFILTER);
117 perror(
"mnl_socket_open");
121 if (mnl_socket_bind(nl, 0, MNL_SOCKET_AUTOPID) < 0) {
122 perror(
"mnl_socket_bind");
125 portid = mnl_socket_get_portid(nl);
127 if (mnl_socket_sendto(nl, mnl_nlmsg_batch_head(batch),
128 mnl_nlmsg_batch_size(batch)) < 0) {
129 perror(
"mnl_socket_send");
133 mnl_nlmsg_batch_stop(batch);
135 ret = mnl_socket_recvfrom(nl, buf,
sizeof(buf));
137 ret = mnl_cb_run(buf, ret, obj_seq, portid, NULL, NULL);
140 ret = mnl_socket_recvfrom(nl, buf,
sizeof(buf));
146 mnl_socket_close(nl);