nft.c revision d007e1a59e4beaddab430992302d43b122ffc801
1384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso/* 2384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * (C) 2012 by Pablo Neira Ayuso <pablo@netfilter.org> 3384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * 4384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * This program is free software; you can redistribute it and/or modify 5384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * it under the terms of the GNU General Public License as published 6384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * by the Free Software Foundation; either version 2 of the License, or 7384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * (at your option) any later version. 8384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * 9384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso * This code has been sponsored by Sophos Astaro <http://www.sophos.com> 10384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso */ 11384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 12384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <unistd.h> 13384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <fcntl.h> 14384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <sys/types.h> 15384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <sys/socket.h> 16384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <stdbool.h> 17384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <errno.h> 18384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <netdb.h> /* getprotobynumber */ 19384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <time.h> 208b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso#include <stdarg.h> 210a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo#include <inttypes.h> 22384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 23384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <xtables.h> 24384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <libiptc/libxtc.h> 25384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <libiptc/xtcshared.h> 26384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 27384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <stdlib.h> 28384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <string.h> 29384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 30384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <linux/netfilter/x_tables.h> 310391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka#include <linux/netfilter_ipv4/ip_tables.h> 320391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka#include <linux/netfilter_ipv6/ip6_tables.h> 330391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka#include <netinet/ip6.h> 34384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 35384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <linux/netlink.h> 36384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <linux/netfilter/nfnetlink.h> 37384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <linux/netfilter/nf_tables.h> 38384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <linux/netfilter/nf_tables_compat.h> 39384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 40384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <libmnl/libmnl.h> 41a4e1098169a67716a81316c36ce22ddcb33df1c0Tomasz Bursztyka#include <libnftnl/table.h> 42a4e1098169a67716a81316c36ce22ddcb33df1c0Tomasz Bursztyka#include <libnftnl/chain.h> 43a4e1098169a67716a81316c36ce22ddcb33df1c0Tomasz Bursztyka#include <libnftnl/rule.h> 44a4e1098169a67716a81316c36ce22ddcb33df1c0Tomasz Bursztyka#include <libnftnl/expr.h> 45384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 46384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <netinet/in.h> /* inet_ntoa */ 47384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include <arpa/inet.h> 48384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 49384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include "nft.h" 50384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#include "xshared.h" /* proto_to_name */ 51077785df023ad8947d44d19769bc6d91e3917633Tomasz Bursztyka#include "nft-shared.h" 528b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso#include "xtables-config-parser.h" 53384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 54384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void *nft_fn; 55384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 5684909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint mnl_talk(struct nft_handle *h, struct nlmsghdr *nlh, 5784909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo int (*cb)(const struct nlmsghdr *nlh, void *data), 5884909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo void *data) 59384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 60384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 61384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 62384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 63384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (mnl_socket_sendto(h->nl, nlh, nlh->nlmsg_len) < 0) { 64384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_send"); 65384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return -1; 66384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 67384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 68384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_socket_recvfrom(h->nl, buf, sizeof(buf)); 69384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (ret > 0) { 70384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_cb_run(buf, ret, h->seq, h->portid, cb, data); 71384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret <= 0) 72384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso break; 73384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 74384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_socket_recvfrom(h->nl, buf, sizeof(buf)); 75384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 76384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret == -1) { 77384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return -1; 78384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 79384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 80384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 81384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 82384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 83d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic LIST_HEAD(batch_page_list); 84d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int batch_num_pages; 85d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 86d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostruct batch_page { 87d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct list_head head; 88d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct mnl_nlmsg_batch *batch; 89d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso}; 90d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 91d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso/* selected batch page is 256 Kbytes long to load ruleset of 92d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso * half a million rules without hitting -EMSGSIZE due to large 93d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso * iovec. 94d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso */ 95d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso#define BATCH_PAGE_SIZE getpagesize() * 32 96d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 97d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic struct mnl_nlmsg_batch *mnl_nft_batch_alloc(void) 98d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 99d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso static char *buf; 100d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 101d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso /* libmnl needs higher buffer to handle batch overflows */ 102d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso buf = malloc(BATCH_PAGE_SIZE + getpagesize()); 103d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (buf == NULL) 104d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return NULL; 105d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 106d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return mnl_nlmsg_batch_start(buf, BATCH_PAGE_SIZE); 107d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 108d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 109d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic struct mnl_nlmsg_batch * 110d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusomnl_nft_batch_page_add(struct mnl_nlmsg_batch *batch) 111d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 112d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct batch_page *batch_page; 113d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 114d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso batch_page = malloc(sizeof(struct batch_page)); 115d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (batch_page == NULL) 116d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return NULL; 117d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 118d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso batch_page->batch = batch; 119d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso list_add_tail(&batch_page->head, &batch_page_list); 120d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso batch_num_pages++; 121d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 122d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return mnl_nft_batch_alloc(); 123d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 124d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 125d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int nlbuffsiz; 126d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 127d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic void mnl_nft_set_sndbuffer(const struct mnl_socket *nl) 128d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 129d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso int newbuffsiz; 130d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 131d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (batch_num_pages * BATCH_PAGE_SIZE <= nlbuffsiz) 132d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return; 133d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 134d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso newbuffsiz = batch_num_pages * BATCH_PAGE_SIZE; 135d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 136d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso /* Rise sender buffer length to avoid hitting -EMSGSIZE */ 137d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (setsockopt(mnl_socket_get_fd(nl), SOL_SOCKET, SO_SNDBUFFORCE, 138d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso &newbuffsiz, sizeof(socklen_t)) < 0) 139d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return; 140d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 141d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nlbuffsiz = newbuffsiz; 142d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 143d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 144d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic ssize_t mnl_nft_socket_sendmsg(const struct mnl_socket *nl) 145d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 146d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso static const struct sockaddr_nl snl = { 147d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .nl_family = AF_NETLINK 148d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso }; 149d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct iovec iov[batch_num_pages]; 150d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct msghdr msg = { 151d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .msg_name = (struct sockaddr *) &snl, 152d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .msg_namelen = sizeof(snl), 153d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .msg_iov = iov, 154d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .msg_iovlen = batch_num_pages, 155d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso }; 156d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct batch_page *batch_page, *next; 157d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso int i = 0; 158d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 159d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nft_set_sndbuffer(nl); 160d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 161d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso list_for_each_entry_safe(batch_page, next, &batch_page_list, head) { 162d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso iov[i].iov_base = mnl_nlmsg_batch_head(batch_page->batch); 163d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso iov[i].iov_len = mnl_nlmsg_batch_size(batch_page->batch); 164d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso i++; 165d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso#ifdef NL_DEBUG 166d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nlmsg_fprintf(stdout, 167d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nlmsg_batch_head(batch_page->batch), 168d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nlmsg_batch_size(batch_page->batch), 169d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso sizeof(struct nfgenmsg)); 170d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso#endif 171d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso list_del(&batch_page->head); 172d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso free(batch_page->batch); 173d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso free(batch_page); 174d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso batch_num_pages--; 175d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 176d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 177d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return sendmsg(mnl_socket_get_fd(nl), &msg, 0); 178d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 179d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 180d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int cb_err(const struct nlmsghdr *nlh, void *data) 181d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 182d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso /* We can provide better error reporting than iptables-restore */ 183d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso errno = EINVAL; 184d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return MNL_CB_ERROR; 185d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 186d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 187d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic mnl_cb_t cb_ctl_array[NLMSG_MIN_TYPE] = { 188d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso [NLMSG_ERROR] = cb_err, 189d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso}; 190d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 191d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int mnl_nft_batch_talk(struct nft_handle *h) 192d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 193d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso int ret, fd = mnl_socket_get_fd(h->nl); 194d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso char rcv_buf[MNL_SOCKET_BUFFER_SIZE]; 195d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso fd_set readfds; 196d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct timeval tv = { 197d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .tv_sec = 0, 198d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso .tv_usec = 0 199d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso }; 200d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso int err = 0; 201d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 202d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = mnl_nft_socket_sendmsg(h->nl); 203d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret == -1) { 204d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso perror("mnl_socket_sendmsg"); 205d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return -1; 206d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 207d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 208d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso FD_ZERO(&readfds); 209d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso FD_SET(fd, &readfds); 210d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 211d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso /* receive and digest all the acknowledgments from the kernel. */ 212d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = select(fd+1, &readfds, NULL, NULL, &tv); 213d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret == -1) { 214d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso perror("select"); 215d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return -1; 216d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 217d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso while (ret > 0 && FD_ISSET(fd, &readfds)) { 218d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = mnl_socket_recvfrom(h->nl, rcv_buf, sizeof(rcv_buf)); 219d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret == -1) { 220d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso perror("mnl_socket_recvfrom"); 221d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return -1; 222d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 223d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 224d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = mnl_cb_run2(rcv_buf, ret, 0, h->portid, 225d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NULL, NULL, cb_ctl_array, 226d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso MNL_ARRAY_SIZE(cb_ctl_array)); 227d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso /* Continue on error, make sure we get all acknoledgments */ 228d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret == -1) 229d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso err = errno; 230d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 231d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = select(fd+1, &readfds, NULL, NULL, &tv); 232d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret == -1) { 233d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso perror("select"); 234d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return -1; 235d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 236d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso FD_ZERO(&readfds); 237d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso FD_SET(fd, &readfds); 238d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 239d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return err ? -1 : 0; 240d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 241d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 242d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic void mnl_nft_batch_put(struct mnl_nlmsg_batch *batch, int type, 243d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso uint32_t seq) 244d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 245d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct nlmsghdr *nlh; 246d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct nfgenmsg *nfg; 247d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 248d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nlh = mnl_nlmsg_put_header(mnl_nlmsg_batch_current(batch)); 249d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nlh->nlmsg_type = type; 250d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nlh->nlmsg_flags = NLM_F_REQUEST; 251d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nlh->nlmsg_seq = seq; 252d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 253d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nfg = mnl_nlmsg_put_extra_header(nlh, sizeof(*nfg)); 254d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nfg->nfgen_family = AF_INET; 255d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nfg->version = NFNETLINK_V0; 256d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nfg->res_id = NFNL_SUBSYS_NFTABLES; 257d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 258d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (!mnl_nlmsg_batch_next(batch)) 259d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nft_batch_page_add(batch); 260d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 261d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 262d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic void mnl_nft_batch_begin(struct mnl_nlmsg_batch *batch, uint32_t seq) 263d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 264d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nft_batch_put(batch, NFNL_MSG_BATCH_BEGIN, seq); 265d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 266d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 267d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic void mnl_nft_batch_end(struct mnl_nlmsg_batch *batch, uint32_t seq) 268d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 269d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nft_batch_put(batch, NFNL_MSG_BATCH_END, seq); 270d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 271d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 272afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longostruct builtin_table xtables_ipv4[TABLES_MAX] = { 273384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso [RAW] = { 274384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso .name = "raw", 275e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .chains = { 276e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 277e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "PREROUTING", 278890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 2795b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -300, /* NF_IP_PRI_RAW */ 280e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_PRE_ROUTING, 281e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 282e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 283e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "OUTPUT", 284890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 2855b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -300, /* NF_IP_PRI_RAW */ 286e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_OUT, 287e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 288e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 289384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }, 290384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso [MANGLE] = { 291384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso .name = "mangle", 292e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .chains = { 293e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 294e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "PREROUTING", 295890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 2965b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -150, /* NF_IP_PRI_MANGLE */ 297e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_PRE_ROUTING, 298e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 299e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 300e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "INPUT", 301890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3025b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -150, /* NF_IP_PRI_MANGLE */ 303e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_IN, 304e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 305e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 306e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "FORWARD", 307890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3085b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -150, /* NF_IP_PRI_MANGLE */ 309e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_FORWARD, 310e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 311e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 312e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "OUTPUT", 313890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "route", 3145b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -150, /* NF_IP_PRI_MANGLE */ 315e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_OUT, 316e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 317e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 318e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "POSTROUTING", 319890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3205b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = -150, /* NF_IP_PRI_MANGLE */ 321e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_POST_ROUTING, 322e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 323e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 324384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }, 325384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso [FILTER] = { 326384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso .name = "filter", 327e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .chains = { 328e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 329e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "INPUT", 330890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3315b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = 0, /* NF_IP_PRI_FILTER */ 332e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_IN, 333e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 334e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 335e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "FORWARD", 336890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3375b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = 0, /* NF_IP_PRI_FILTER */ 338e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_FORWARD, 339e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 340e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 341e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "OUTPUT", 342890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3435b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = 0, /* NF_IP_PRI_FILTER */ 344e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_OUT, 345e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 346e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 347384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }, 348384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso [SECURITY] = { 349384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso .name = "security", 350e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .chains = { 351e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 352e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "INPUT", 353890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3545b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = 150, /* NF_IP_PRI_SECURITY */ 355e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_IN, 356e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 357e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 358e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "FORWARD", 359890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3605b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = 150, /* NF_IP_PRI_SECURITY */ 361e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_FORWARD, 362e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 363e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso { 364e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .name = "OUTPUT", 365890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "filter", 3665b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso .prio = 150, /* NF_IP_PRI_SECURITY */ 367e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso .hook = NF_INET_LOCAL_OUT, 368e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso }, 369384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }, 370384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }, 371890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso [NAT] = { 372890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .name = "nat", 373890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .chains = { 374890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso { 37586eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso .name = "PREROUTING", 376890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "nat", 377890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .prio = -100, /* NF_IP_PRI_NAT_DST */ 37886eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso .hook = NF_INET_PRE_ROUTING, 379890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso }, 380890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso { 381890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .name = "INPUT", 382890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "nat", 383890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .prio = 100, /* NF_IP_PRI_NAT_SRC */ 384890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .hook = NF_INET_LOCAL_IN, 385890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso }, 386890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso { 387890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .name = "POSTROUTING", 388890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .type = "nat", 389890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .prio = 100, /* NF_IP_PRI_NAT_SRC */ 390890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso .hook = NF_INET_POST_ROUTING, 391890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso }, 39286eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso { 39386eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso .name = "OUTPUT", 39486eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso .type = "nat", 39586eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso .prio = -100, /* NF_IP_PRI_NAT_DST */ 39686eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso .hook = NF_INET_LOCAL_OUT, 39786eed10c9f2c42e0f50eb4e527a48ee9e63146f4Pablo Neira Ayuso }, 398890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso }, 399890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso }, 400384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso}; 401384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 40284909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo#include <linux/netfilter_arp.h> 40384909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo 40484909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longostruct builtin_table xtables_arp[TABLES_MAX] = { 40584909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo [FILTER] = { 40684909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .name = "filter", 40784909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .chains = { 40884909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo { 40984909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .name = "INPUT", 41084909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .type = "filter", 41184909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .prio = NF_IP_PRI_FILTER, 41284909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .hook = NF_ARP_IN, 41384909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo }, 41484909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo { 41584909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .name = "FORWARD", 41684909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .type = "filter", 41784909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .prio = NF_IP_PRI_FILTER, 41884909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .hook = NF_ARP_FORWARD, 41984909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo }, 42084909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo { 42184909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .name = "OUTPUT", 42284909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .type = "filter", 42384909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .prio = NF_IP_PRI_FILTER, 42484909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo .hook = NF_ARP_OUT, 42584909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo }, 42684909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo }, 42784909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo }, 42884909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo}; 42984909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longo 43084909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint 4315705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayusonft_table_builtin_add(struct nft_handle *h, struct builtin_table *_t, 4325705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso bool dormant) 4335b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso{ 4345b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 4355b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso struct nlmsghdr *nlh; 4365b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso struct nft_table *t; 4375b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso int ret; 4385b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso 4395b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso t = nft_table_alloc(); 4405b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso if (t == NULL) 4415b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso return -1; 4425b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso 4435b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso nft_table_attr_set(t, NFT_TABLE_ATTR_NAME, (char *)_t->name); 4445705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso if (dormant) { 4455705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso nft_table_attr_set_u32(t, NFT_TABLE_ATTR_FLAGS, 4465705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso NFT_TABLE_F_DORMANT); 4475705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso } 4485b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso 4490391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_table_nlmsg_build_hdr(buf, NFT_MSG_NEWTABLE, h->family, 4505b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso NLM_F_ACK|NLM_F_EXCL, h->seq); 4515b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso nft_table_nlmsg_build_payload(nlh, t); 4525b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso nft_table_free(t); 4535b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso 4545b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso ret = mnl_talk(h, nlh, NULL, NULL); 4555b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso if (ret < 0) { 4565b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso if (errno != EEXIST) 4575b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso perror("mnl-talk:nft_table_init_one"); 4585b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso } 4595b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso return ret; 4605b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso} 4615b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso 46284909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longostruct nft_chain * 463c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayusonft_chain_builtin_alloc(struct builtin_table *table, 464c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct builtin_chain *chain, int policy) 465384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 466384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 467384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 468384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_alloc(); 469384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (c == NULL) 470c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso return NULL; 471384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 472384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set(c, NFT_CHAIN_ATTR_TABLE, (char *)table->name); 473384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set(c, NFT_CHAIN_ATTR_NAME, (char *)chain->name); 474384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set_u32(c, NFT_CHAIN_ATTR_HOOKNUM, chain->hook); 4755b414b85c33912aec912d260502c8b0c0df794a1Pablo Neira Ayuso nft_chain_attr_set_u32(c, NFT_CHAIN_ATTR_PRIO, chain->prio); 476384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set_u32(c, NFT_CHAIN_ATTR_POLICY, policy); 477890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso nft_chain_attr_set(c, NFT_CHAIN_ATTR_TYPE, (char *)chain->type); 478384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 479c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso return c; 480c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso} 481c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 48284909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longovoid 483c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayusonft_chain_builtin_add(struct nft_handle *h, struct builtin_table *table, 484c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct builtin_chain *chain, int policy) 485c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso{ 486c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 487c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct nlmsghdr *nlh; 488c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct nft_chain *c; 489c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 490c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso c = nft_chain_builtin_alloc(table, chain, policy); 491c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso if (c == NULL) 492c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso return; 493c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 494890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso /* NLM_F_CREATE requests module autoloading */ 4950391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_NEWCHAIN, h->family, 496890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso NLM_F_ACK|NLM_F_EXCL|NLM_F_CREATE, 497890fd9ef76ad0c11695fb0d09a88169e6e46584fPablo Neira Ayuso h->seq); 498384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_nlmsg_build_payload(nlh, c); 499384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_free(c); 500384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 501384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (mnl_talk(h, nlh, NULL, NULL) < 0) { 502384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (errno != EEXIST) 503c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso perror("mnl_talk:nft_chain_builtin_add"); 504c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso } 505c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso} 506c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 507c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso/* find if built-in table already exists */ 50884909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longostruct builtin_table * 50984909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longonft_table_builtin_find(struct nft_handle *h, const char *table) 510c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso{ 511c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso int i; 512c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso bool found = false; 513c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 514c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso for (i=0; i<TABLES_MAX; i++) { 51567da6075a4e7ced0e8cc452d73ce8ab06cbf8cd9Tomasz Bursztyka if (h->tables[i].name == NULL) 51667da6075a4e7ced0e8cc452d73ce8ab06cbf8cd9Tomasz Bursztyka break; 51767da6075a4e7ced0e8cc452d73ce8ab06cbf8cd9Tomasz Bursztyka 518afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longo if (strcmp(h->tables[i].name, table) != 0) 519c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso continue; 520c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 521c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso found = true; 522c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso break; 523384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 524c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 525afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longo return found ? &h->tables[i] : NULL; 526384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 527384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 528c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso/* find if built-in chain already exists */ 52984909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longostruct builtin_chain * 530e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayusonft_chain_builtin_find(struct builtin_table *t, const char *chain) 531384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 532e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso int i; 533c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso bool found = false; 534384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 535e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso for (i=0; i<NF_IP_NUMHOOKS && t->chains[i].name != NULL; i++) { 536e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso if (strcmp(t->chains[i].name, chain) != 0) 537e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso continue; 538384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 539e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso found = true; 540e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso break; 541e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso } 542e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso return found ? &t->chains[i] : NULL; 543e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso} 544e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso 545e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayusostatic void 546e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso__nft_chain_builtin_init(struct nft_handle *h, 547e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso struct builtin_table *table, const char *chain, 548e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso int policy) 549e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso{ 550e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso int i, default_policy; 551e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso 552e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso /* Initialize all built-in chains. Exception, for e one received as 553e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso * parameter, set the default policy as requested. 554e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso */ 555e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso for (i=0; i<NF_IP_NUMHOOKS && table->chains[i].name != NULL; i++) { 556e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso if (chain && strcmp(table->chains[i].name, chain) == 0) 557e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso default_policy = policy; 558e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso else 559e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso default_policy = NF_ACCEPT; 560e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso 561e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso nft_chain_builtin_add(h, table, &table->chains[i], 562e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso default_policy); 563384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 564c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso} 565c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 56684909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint 567c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayusonft_chain_builtin_init(struct nft_handle *h, const char *table, 568c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso const char *chain, int policy) 569c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso{ 570c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso int ret = 0; 571c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct builtin_table *t; 572c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 573afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longo t = nft_table_builtin_find(h, table); 574c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso if (t == NULL) { 575c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso ret = -1; 576c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso goto out; 577c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso } 5785705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso if (nft_table_builtin_add(h, t, false) < 0) { 579e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso /* Built-in table already initialized, skip. */ 580e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso if (errno == EEXIST) 581e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso goto out; 582c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso } 583e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso __nft_chain_builtin_init(h, t, chain, policy); 584c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayusoout: 585c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso return ret; 586384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 587384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 5887244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztykastatic bool nft_chain_builtin(struct nft_chain *c) 5897244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka{ 5907244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka /* Check if this chain has hook number, in that case is built-in. 5917244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka * Should we better export the flags to user-space via nf_tables? 5927244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka */ 5937244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka return nft_chain_attr_get(c, NFT_CHAIN_ATTR_HOOKNUM) != NULL; 5947244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka} 5957244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka 596afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longoint nft_init(struct nft_handle *h, struct builtin_table *t) 597384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 598384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso h->nl = mnl_socket_open(NETLINK_NETFILTER); 599384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (h->nl == NULL) { 600384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_open"); 601384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return -1; 602384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 603384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 604384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (mnl_socket_bind(h->nl, 0, MNL_SOCKET_AUTOPID) < 0) { 605384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_bind"); 606384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return -1; 607384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 608384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso h->portid = mnl_socket_get_portid(h->nl); 609afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longo h->tables = t; 610384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 611d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso INIT_LIST_HEAD(&h->rule_list); 612d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 613d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso h->batch = mnl_nft_batch_alloc(); 614d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 615384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 616384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 617384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 618384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusovoid nft_fini(struct nft_handle *h) 619384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 620384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_socket_close(h->nl); 621b0194cea194b510c675ca05415da15cff57afe47Ana Rey free(mnl_nlmsg_batch_head(h->batch)); 622b0194cea194b510c675ca05415da15cff57afe47Ana Rey mnl_nlmsg_batch_stop(h->batch); 623384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 624384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 625384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_table_add(struct nft_handle *h, const struct nft_table *t) 626384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 627384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 628384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 629384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 6300391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_table_nlmsg_build_hdr(buf, NFT_MSG_NEWTABLE, h->family, 631384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_ACK|NLM_F_EXCL, h->seq); 632384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_table_nlmsg_build_payload(nlh, t); 633384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 634384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return mnl_talk(h, nlh, NULL, NULL); 635384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 636384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 637384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_chain_add(struct nft_handle *h, const struct nft_chain *c) 638384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 639384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 640384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 641384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 6420391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_NEWCHAIN, h->family, 643384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_ACK|NLM_F_EXCL, h->seq); 644384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_nlmsg_build_payload(nlh, c); 645384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 646384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return mnl_talk(h, nlh, NULL, NULL); 647384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 648384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 6495705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayusoint nft_table_set_dormant(struct nft_handle *h, const char *table) 6505705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso{ 6515705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso int ret = 0, i; 6525705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso struct builtin_table *t; 6535705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 654afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longo t = nft_table_builtin_find(h, table); 6555705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso if (t == NULL) { 6565705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso ret = -1; 6575705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso goto out; 6585705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso } 6595705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso /* Add this table as dormant */ 6605705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso if (nft_table_builtin_add(h, t, true) < 0) { 6615705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso /* Built-in table already initialized, skip. */ 6625705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso if (errno == EEXIST) 6635705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso goto out; 6645705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso } 6655705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso for (i=0; t->chains[i].name != NULL && i<NF_INET_NUMHOOKS; i++) 6665705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso __nft_chain_builtin_init(h, t, t->chains[i].name, NF_ACCEPT); 6675705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayusoout: 6685705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso return ret; 6695705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso} 6705705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 6715705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayusoint nft_table_wake_dormant(struct nft_handle *h, const char *table) 6725705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso{ 6735705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 6745705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso struct nlmsghdr *nlh; 6755705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso struct nft_table *t; 6765705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 6775705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso t = nft_table_alloc(); 6785705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso if (t == NULL) 6795705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso return -1; 6805705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 6815705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso nft_table_attr_set(t, NFT_TABLE_ATTR_NAME, (char *)table); 6825705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso nft_table_attr_set_u32(t, NFT_TABLE_ATTR_FLAGS, 0); 6835705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 6840391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_table_nlmsg_build_hdr(buf, NFT_MSG_NEWTABLE, h->family, 6855705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso NLM_F_ACK, h->seq); 6865705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso nft_table_nlmsg_build_payload(nlh, t); 6875705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso nft_table_free(t); 6885705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 6895705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso return mnl_talk(h, nlh, NULL, NULL); 6905705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso} 6915705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso 692384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void nft_chain_print_debug(struct nft_chain *c, struct nlmsghdr *nlh) 693384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 694384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#ifdef NLDEBUG 695384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char tmp[1024]; 696384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 697384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_snprintf(tmp, sizeof(tmp), c, 0, 0); 698384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("DEBUG: chain: %s", tmp); 699384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_nlmsg_fprintf(stdout, nlh, nlh->nlmsg_len, sizeof(struct nfgenmsg)); 700384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#endif 701384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 702384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 703384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int 704384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso__nft_chain_set(struct nft_handle *h, const char *table, 705384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain, int policy, 706384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const struct xt_counters *counters) 707384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 708384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 709384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 710384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 711c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct builtin_table *_t; 712c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso struct builtin_chain *_c; 713e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso int ret; 714c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 715afae1f841bc2c4b39a38fa97d271f3877d00bf3aGiuseppe Longo _t = nft_table_builtin_find(h, table); 716c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso /* if this built-in table does not exists, create it */ 717c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso if (_t != NULL) 7185705ea1f4e3c9cd3d5d9cbcf84b9733ce1f07e57Pablo Neira Ayuso nft_table_builtin_add(h, _t, false); 719c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso 720e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso _c = nft_chain_builtin_find(_t, chain); 721c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso if (_c != NULL) { 722c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso /* This is a built-in chain */ 723c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso c = nft_chain_builtin_alloc(_t, _c, policy); 724c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso if (c == NULL) 725c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso return -1; 726c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso } else { 72720c156f9f4c43857a622f015a3022517601c3600Tomasz Bursztyka errno = ENOENT; 72820c156f9f4c43857a622f015a3022517601c3600Tomasz Bursztyka return -1; 729c74f398c58e7b77ca1c82a15d0b2e9eed4fb82adPablo Neira Ayuso } 730384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 731384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (counters) { 732384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set_u64(c, NFT_CHAIN_ATTR_BYTES, 733384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso counters->bcnt); 734384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set_u64(c, NFT_CHAIN_ATTR_PACKETS, 735384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso counters->pcnt); 736384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 737384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 7380391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_NEWCHAIN, h->family, 739384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_ACK, h->seq); 740384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_nlmsg_build_payload(nlh, c); 741384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 742384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_print_debug(c, nlh); 743384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 744384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_free(c); 745384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 746384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_talk(h, nlh, NULL, NULL); 747384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) 748384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_talk:__nft_chain_policy"); 749384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 750384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret; 751384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 752384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 753384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_chain_set(struct nft_handle *h, const char *table, 754384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain, const char *policy, 755384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const struct xt_counters *counters) 756384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 757384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret = -1; 758384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 759384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_fn = nft_chain_set; 760384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 761384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(policy, "DROP") == 0) 762384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = __nft_chain_set(h, table, chain, NF_DROP, counters); 763384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso else if (strcmp(policy, "ACCEPT") == 0) 764384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = __nft_chain_set(h, table, chain, NF_ACCEPT, counters); 765384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 766384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* the core expects 1 for success and 0 for error */ 767384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret == 0 ? 1 : 0; 768384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 769384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 770f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztykastatic int __add_match(struct nft_rule_expr *e, struct xt_entry_match *m) 771384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 772384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso void *info; 773384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 774384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set(e, NFT_EXPR_MT_NAME, m->u.user.name, strlen(m->u.user.name)); 775384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u32(e, NFT_EXPR_MT_REV, m->u.user.revision); 776384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 777384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso info = calloc(1, m->u.match_size); 778384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (info == NULL) 779f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return -ENOMEM; 780384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 7815eed9118f2620ac07edd553599e2415f00d6f8f3Pablo Neira Ayuso memcpy(info, m->data, m->u.match_size - sizeof(*m)); 782384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set(e, NFT_EXPR_MT_INFO, info, m->u.match_size - sizeof(*m)); 783f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 784f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return 0; 785384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 786384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 787c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayusoint add_match(struct nft_rule *r, struct xt_entry_match *m) 788384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 789384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_expr *expr; 790f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka int ret; 791384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 792384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso expr = nft_rule_expr_alloc("match"); 793384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (expr == NULL) 794f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return -ENOMEM; 795384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 796f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka ret = __add_match(expr, m); 797384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_add_expr(r, expr); 798f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 799f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return ret; 800384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 801384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 802f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztykastatic int __add_target(struct nft_rule_expr *e, struct xt_entry_target *t) 803384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 8045eed9118f2620ac07edd553599e2415f00d6f8f3Pablo Neira Ayuso void *info; 805384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 806384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set(e, NFT_EXPR_TG_NAME, t->u.user.name, 807384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso strlen(t->u.user.name)); 808384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u32(e, NFT_EXPR_TG_REV, t->u.user.revision); 809384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 8105eed9118f2620ac07edd553599e2415f00d6f8f3Pablo Neira Ayuso info = calloc(1, t->u.target_size); 8115eed9118f2620ac07edd553599e2415f00d6f8f3Pablo Neira Ayuso if (info == NULL) 8125eed9118f2620ac07edd553599e2415f00d6f8f3Pablo Neira Ayuso return -ENOMEM; 813384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 8145eed9118f2620ac07edd553599e2415f00d6f8f3Pablo Neira Ayuso memcpy(info, t->data, t->u.target_size - sizeof(*t)); 815384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set(e, NFT_EXPR_TG_INFO, info, t->u.target_size - sizeof(*t)); 816f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 817f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return 0; 818384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 819384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 82084909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint add_target(struct nft_rule *r, struct xt_entry_target *t) 821384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 822384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_expr *expr; 823f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka int ret; 824384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 825384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso expr = nft_rule_expr_alloc("target"); 826384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (expr == NULL) 827f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return -ENOMEM; 828384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 829f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka ret = __add_target(expr, t); 830384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_add_expr(r, expr); 831f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 832f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return ret; 833384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 834384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 83584909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint add_jumpto(struct nft_rule *r, const char *name, int verdict) 836384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 837384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_expr *expr; 838384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 839384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso expr = nft_rule_expr_alloc("immediate"); 840384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (expr == NULL) 841f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return -ENOMEM; 842384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 843384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u32(expr, NFT_EXPR_IMM_DREG, NFT_REG_VERDICT); 844384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u32(expr, NFT_EXPR_IMM_VERDICT, verdict); 845384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_str(expr, NFT_EXPR_IMM_CHAIN, (char *)name); 846384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_add_expr(r, expr); 847f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 848f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return 0; 849384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 850384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 85184909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint add_verdict(struct nft_rule *r, int verdict) 852384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 853384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_expr *expr; 854384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 855384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso expr = nft_rule_expr_alloc("immediate"); 856384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (expr == NULL) 857f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return -ENOMEM; 858384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 859384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u32(expr, NFT_EXPR_IMM_DREG, NFT_REG_VERDICT); 860384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u32(expr, NFT_EXPR_IMM_VERDICT, verdict); 861384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_add_expr(r, expr); 862f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 863f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return 0; 864384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 865384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 866c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayusoint add_action(struct nft_rule *r, struct iptables_command_state *cs, 867d007e1a59e4beaddab430992302d43b122ffc801Pablo Neira Ayuso bool goto_set) 868c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso{ 869c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso int ret = 0; 870c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso 871c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso /* If no target at all, add nothing (default to continue) */ 872c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso if (cs->target != NULL) { 873c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso /* Standard target? */ 874c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso if (strcmp(cs->jumpto, XTC_LABEL_ACCEPT) == 0) 875c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso ret = add_verdict(r, NF_ACCEPT); 876c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso else if (strcmp(cs->jumpto, XTC_LABEL_DROP) == 0) 877c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso ret = add_verdict(r, NF_DROP); 878c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso else if (strcmp(cs->jumpto, XTC_LABEL_RETURN) == 0) 879c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso ret = add_verdict(r, NFT_RETURN); 880c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso else 881c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso ret = add_target(r, cs->target->t); 882c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso } else if (strlen(cs->jumpto) > 0) { 883c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso /* Not standard, then it's a go / jump to chain */ 884d007e1a59e4beaddab430992302d43b122ffc801Pablo Neira Ayuso if (goto_set) 885c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso ret = add_jumpto(r, cs->jumpto, NFT_GOTO); 886c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso else 887c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso ret = add_jumpto(r, cs->jumpto, NFT_JUMP); 888c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso } 889c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso return ret; 890c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso} 891c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso 892384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void nft_rule_print_debug(struct nft_rule *r, struct nlmsghdr *nlh) 893384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 894384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#ifdef NLDEBUG 895384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char tmp[1024]; 896384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 897384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_snprintf(tmp, sizeof(tmp), r, 0, 0); 898384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("DEBUG: rule: %s", tmp); 899384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_nlmsg_fprintf(stdout, nlh, nlh->nlmsg_len, sizeof(struct nfgenmsg)); 900384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso#endif 901384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 902384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 90384909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longoint add_counters(struct nft_rule *r, uint64_t packets, uint64_t bytes) 904384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 905384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_expr *expr; 906384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 907384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso expr = nft_rule_expr_alloc("counter"); 908384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (expr == NULL) 909f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return -ENOMEM; 910384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 911384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u64(expr, NFT_EXPR_CTR_BYTES, packets); 912384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_expr_set_u64(expr, NFT_EXPR_CTR_PACKETS, bytes); 913384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 914384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_add_expr(r, expr); 915f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka 916f6b30db46e5e3d32c76b186361853b5a5ecaf99fTomasz Bursztyka return 0; 917384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 918384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 9194ef77b6d1b52e1fe52a7fd48d38d9233f0961640Pablo Neira Ayusovoid add_compat(struct nft_rule *r, uint32_t proto, bool inv) 9202a87a024e1f77407e332086a4fa664e048280195Pablo Neira Ayuso{ 9212a87a024e1f77407e332086a4fa664e048280195Pablo Neira Ayuso nft_rule_attr_set_u32(r, NFT_RULE_ATTR_COMPAT_PROTO, proto); 9222a87a024e1f77407e332086a4fa664e048280195Pablo Neira Ayuso nft_rule_attr_set_u32(r, NFT_RULE_ATTR_COMPAT_FLAGS, 9232a87a024e1f77407e332086a4fa664e048280195Pablo Neira Ayuso inv ? NFT_RULE_COMPAT_F_INV : 0); 9242a87a024e1f77407e332086a4fa664e048280195Pablo Neira Ayuso} 9252a87a024e1f77407e332086a4fa664e048280195Pablo Neira Ayuso 926cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusostatic struct nft_rule * 927cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusonft_rule_new(struct nft_handle *h, const char *chain, const char *table, 928c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso void *data) 929384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 930384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule *r; 931384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 932384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_alloc(); 933cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso if (r == NULL) 934cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso return NULL; 935384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 936d370c4ad803c37eedfbee5963fac6f7e9968939cTomasz Bursztyka nft_rule_attr_set_u32(r, NFT_RULE_ATTR_FAMILY, h->family); 937384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_set(r, NFT_RULE_ATTR_TABLE, (char *)table); 938384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_set(r, NFT_RULE_ATTR_CHAIN, (char *)chain); 939384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 940c6836c19592dbe1a8be9b0ad76c0ae09abcb82e7Pablo Neira Ayuso if (h->ops->add(r, data) < 0) 941cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso goto err; 942cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 943cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso return r; 944cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusoerr: 945cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_rule_free(r); 946cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso return NULL; 947cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso} 948cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 949d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusoenum rule_update_type { 950d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_APPEND, 951d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_INSERT, 952d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_REPLACE, 953d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_DELETE, 954d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_FLUSH, 955d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_COMMIT, 956d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso NFT_DO_ABORT, 957d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso}; 958d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 959d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostruct rule_update { 960d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct list_head head; 961d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso enum rule_update_type type; 962d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct nft_rule *rule; 963d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso}; 964d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 965d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int rule_update_add(struct nft_handle *h, enum rule_update_type type, 966d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct nft_rule *r) 967d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso{ 968d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct rule_update *rupd; 969d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 970d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso rupd = calloc(1, sizeof(struct rule_update)); 971d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (rupd == NULL) 972d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return -1; 973d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 974d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso rupd->rule = r; 975d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso rupd->type = type; 976d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso list_add_tail(&rupd->head, &h->rule_list); 977d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso h->rule_list_num++; 978d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 979d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 0; 980d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso} 981d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 982cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusoint 983cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusonft_rule_append(struct nft_handle *h, const char *chain, const char *table, 984b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso void *data, uint64_t handle, bool verbose) 985cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso{ 986cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso struct nft_rule *r; 987d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso int type; 988cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 989cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso /* If built-in chains don't exist for this table, create them */ 990cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso if (nft_xtables_config_load(h, XTABLES_CONFIG_DEFAULT, 0) < 0) 991cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_chain_builtin_init(h, table, chain, NF_ACCEPT); 992cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 993cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_fn = nft_rule_append; 994cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 995b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso r = nft_rule_new(h, chain, table, data); 996d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (r == NULL) 997d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 0; 998384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 9990a366d8696582e979d55f6832a797d1217f4b908Tomasz Bursztyka if (handle > 0) { 10000a366d8696582e979d55f6832a797d1217f4b908Tomasz Bursztyka nft_rule_attr_set(r, NFT_RULE_ATTR_HANDLE, &handle); 1001d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type = NFT_DO_REPLACE; 1002cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso } else 1003d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type = NFT_DO_APPEND; 1004384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1005d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (rule_update_add(h, type, r) < 0) 1006d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_free(r); 1007384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1008d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 1; 1009384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1010384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 10111ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayusovoid 10128877968858a8dd6b7ae096988d57a7511c81733dGiuseppe Longonft_rule_print_save(const void *data, 1013cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka struct nft_rule *r, enum nft_rule_print type, 1014cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka unsigned int format) 1015384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 10161ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso const char *chain = nft_rule_attr_get_str(r, NFT_RULE_ATTR_CHAIN); 1017cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka int family = nft_rule_attr_get_u8(r, NFT_RULE_ATTR_FAMILY); 1018cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka struct nft_family_ops *ops; 1019384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1020384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* print chain name */ 10211ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso switch(type) { 10221ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso case NFT_RULE_APPEND: 10231ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso printf("-A %s ", chain); 10241ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso break; 10251ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso case NFT_RULE_DEL: 10261ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso printf("-D %s ", chain); 10271ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso break; 10281ff21a68502d67e056100da7e0da074467bc08edPablo Neira Ayuso } 1029384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1030cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka ops = nft_family_ops_lookup(family); 1031cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka 10328877968858a8dd6b7ae096988d57a7511c81733dGiuseppe Longo if (ops->save_firewall) 10338877968858a8dd6b7ae096988d57a7511c81733dGiuseppe Longo ops->save_firewall(data, format); 1034cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka 1035384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1036384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1037384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int nft_chain_list_cb(const struct nlmsghdr *nlh, void *data) 1038384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1039384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 1040384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list *list = data; 1041384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1042384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_alloc(); 1043384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (c == NULL) { 1044384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("OOM"); 1045384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1046384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1047384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1048384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nft_chain_nlmsg_parse(nlh, c) < 0) { 1049384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("nft_rule_nlmsg_parse"); 1050384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto out; 1051384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1052384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1053f77e1aca9aa8851b469f79c2db80ddb6f49253b2Pablo Neira Ayuso nft_chain_list_add_tail(c, list); 1054384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1055384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return MNL_CB_OK; 1056384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoout: 1057384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_free(c); 1058384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1059384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return MNL_CB_OK; 1060384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1061384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1062384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic struct nft_chain_list *nft_chain_list_get(struct nft_handle *h) 1063384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1064384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 1065384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 1066384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1067384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list *list; 1068384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1069384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_chain_list_alloc(); 1070384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (list == NULL) { 1071b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso errno = ENOMEM; 1072b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso return NULL; 1073384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1074384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 10750391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_GETCHAIN, h->family, 1076384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_DUMP, h->seq); 1077384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1078384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_talk(h, nlh, nft_chain_list_cb, list); 1079384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) 1080384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_talk:nft_chain_list_get"); 1081384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1082384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return list; 1083384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1084384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1085384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostruct nft_chain_list *nft_chain_dump(struct nft_handle *h) 1086384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1087384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return nft_chain_list_get(h); 1088384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1089384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1090384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic const char *policy_name[NF_ACCEPT+1] = { 1091384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso [NF_DROP] = "DROP", 1092384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso [NF_ACCEPT] = "ACCEPT", 1093384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso}; 1094384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1095384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void nft_chain_print_save(struct nft_chain *c, bool basechain) 1096384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1097384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain = nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 1098384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint64_t pkts = nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_PACKETS); 1099384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint64_t bytes = nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_BYTES); 1100384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1101384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* print chain name */ 1102384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (basechain) { 1103384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint32_t pol = NF_ACCEPT; 1104384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1105384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* no default chain policy? don't crash, display accept */ 1106384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nft_chain_attr_get(c, NFT_CHAIN_ATTR_POLICY)) 1107384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso pol = nft_chain_attr_get_u32(c, NFT_CHAIN_ATTR_POLICY); 1108384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 11090a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo printf(":%s %s [%"PRIu64":%"PRIu64"]\n", chain, policy_name[pol], 1110384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso pkts, bytes); 1111384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } else 11120a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo printf(":%s - [%"PRIu64":%"PRIu64"]\n", chain, pkts, bytes); 1113384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1114384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1115384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_chain_save(struct nft_handle *h, struct nft_chain_list *list, 1116384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *table) 1117384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1118384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 1119384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 1120384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1121384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 1122b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1123384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1124384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1125384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 1126384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (c != NULL) { 1127384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_table = 1128384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 1129384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso bool basechain = false; 1130384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1131384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, chain_table) != 0) 1132384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1133384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 11347244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka basechain = nft_chain_builtin(c); 1135384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_print_save(c, basechain); 1136384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 1137384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 1138384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1139384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1140d01b2c28c8101f0d24e1db3f146fd845c2a634e8Giuseppe Longo nft_chain_list_iter_destroy(iter); 1141384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_list_free(list); 1142384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1143384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 1; 1144384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1145384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1146384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int nft_rule_list_cb(const struct nlmsghdr *nlh, void *data) 1147384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1148384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule *r; 1149384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_list *list = data; 1150384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1151384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_alloc(); 1152384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (r == NULL) { 1153384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("OOM"); 1154384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1155384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1156384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1157384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nft_rule_nlmsg_parse(nlh, r) < 0) { 1158384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("nft_rule_nlmsg_parse"); 1159384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto out; 1160384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1161384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1162f77e1aca9aa8851b469f79c2db80ddb6f49253b2Pablo Neira Ayuso nft_rule_list_add_tail(r, list); 1163384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1164384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return MNL_CB_OK; 1165384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoout: 1166384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_free(r); 1167137cc981906f356c971da6de13e777a419382ff4Giuseppe Longo nft_rule_list_free(list); 1168384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1169384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return MNL_CB_OK; 1170384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1171384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1172384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic struct nft_rule_list *nft_rule_list_get(struct nft_handle *h) 1173384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1174384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 1175384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 1176384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_list *list; 1177384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1178384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1179384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_rule_list_alloc(); 1180b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 1181384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1182384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 11830391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_rule_nlmsg_build_hdr(buf, NFT_MSG_GETRULE, h->family, 1184384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_DUMP, h->seq); 1185384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1186384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_talk(h, nlh, nft_rule_list_cb, list); 1187384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) { 1188384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_talk:nft_rule_save"); 1189384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_list_free(list); 1190384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return NULL; 1191384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1192384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1193384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return list; 1194384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1195384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1196384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_save(struct nft_handle *h, const char *table, bool counters) 1197384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1198384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_list *list; 1199384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_list_iter *iter; 1200384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule *r; 1201384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1202384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_rule_list_get(h); 1203b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 1204384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1205384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1206384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_rule_list_iter_create(list); 1207b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1208384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1209384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1210384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_list_iter_next(iter); 1211384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (r != NULL) { 1212384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *rule_table = 1213384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_get_str(r, NFT_RULE_ATTR_TABLE); 1214cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka struct iptables_command_state cs = {}; 1215384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1216384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, rule_table) != 0) 1217384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1218384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1219cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka nft_rule_to_iptables_command_state(r, &cs); 1220cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka 1221cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka nft_rule_print_save(&cs, r, NFT_RULE_APPEND, 1222cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka counters ? 0 : FMT_NOCOUNTS); 1223384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1224384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 1225384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_list_iter_next(iter); 1226384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1227384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1228d01b2c28c8101f0d24e1db3f146fd845c2a634e8Giuseppe Longo nft_rule_list_iter_destroy(iter); 1229384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_list_free(list); 1230384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1231384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* the core expects 1 for success and 0 for error */ 1232384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 1; 1233384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1234384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1235384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void 1236384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso__nft_rule_flush(struct nft_handle *h, const char *table, const char *chain) 1237384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1238384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule *r; 1239384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1240384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_alloc(); 1241384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (r == NULL) 1242384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return; 1243384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1244384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_set(r, NFT_RULE_ATTR_TABLE, (char *)table); 1245384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_set(r, NFT_RULE_ATTR_CHAIN, (char *)chain); 1246384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1247d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (rule_update_add(h, NFT_DO_FLUSH, r) < 0) 1248d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_free(r); 1249384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1250384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1251384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_flush(struct nft_handle *h, const char *chain, const char *table) 1252384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1253384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1254384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list *list; 1255384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 1256384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 1257384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1258384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_fn = nft_rule_flush; 1259384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1260384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_chain_list_get(h); 1261384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (list == NULL) { 1262384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = 0; 1263384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1264384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1265384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1266384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 1267b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1268b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso goto err; 1269384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1270384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 1271384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (c != NULL) { 1272384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *table_name = 1273384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 1274384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_name = 1275384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 1276384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1277384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, table_name) != 0) 1278384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1279384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1280384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (chain != NULL && strcmp(chain, chain_name) != 0) 1281384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1282384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1283384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso __nft_rule_flush(h, table_name, chain_name); 1284384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 128536ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo if (chain != NULL) 128636ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo break; 1287384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 1288384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 1289384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1290384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1291d01b2c28c8101f0d24e1db3f146fd845c2a634e8Giuseppe Longo nft_chain_list_iter_destroy(iter); 1292384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1293384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_list_free(list); 1294384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1295384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* the core expects 1 for success and 0 for error */ 1296384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret == 0 ? 1 : 0; 1297384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1298384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1299384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_chain_user_add(struct nft_handle *h, const char *chain, const char *table) 1300384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1301384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 1302384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 1303384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 1304384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1305384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1306e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso /* If built-in chains don't exist for this table, create them */ 13078b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (nft_xtables_config_load(h, XTABLES_CONFIG_DEFAULT, 0) < 0) 13088b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso nft_chain_builtin_init(h, table, NULL, NF_ACCEPT); 1309e17b5f15c2beda86565a8f4e55fdf12ceb20dd59Pablo Neira Ayuso 1310384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_alloc(); 1311b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (c == NULL) 1312b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso return 0; 1313384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1314384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set(c, NFT_CHAIN_ATTR_TABLE, (char *)table); 1315384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_set(c, NFT_CHAIN_ATTR_NAME, (char *)chain); 1316384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 13170391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_NEWCHAIN, h->family, 1318384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_ACK|NLM_F_EXCL, h->seq); 1319384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_nlmsg_build_payload(nlh, c); 1320384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_free(c); 1321384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1322384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_talk(h, nlh, NULL, NULL); 1323384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) { 1324384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (errno != EEXIST) 1325384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_talk:nft_chain_add"); 1326384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1327384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1328384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* the core expects 1 for success and 0 for error */ 1329384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret == 0 ? 1 : 0; 1330384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1331384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1332384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int __nft_chain_del(struct nft_handle *h, struct nft_chain *c) 1333384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1334384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 1335384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 1336384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1337384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 13380391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_DELCHAIN, h->family, 1339384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_ACK, h->seq); 1340384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_nlmsg_build_payload(nlh, c); 1341384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1342384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_talk(h, nlh, NULL, NULL); 1343384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) { 1344384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (errno != EEXIST && errno != ENOENT) 1345384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_talk:__nft_chain_del"); 1346384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1347384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1348384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret; 1349384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1350384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1351384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_chain_user_del(struct nft_handle *h, const char *chain, const char *table) 1352384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1353384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list *list; 1354384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 1355384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 1356384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret = 0; 1357384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int deleted_ctr = 0; 1358384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1359384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_chain_list_get(h); 1360384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (list == NULL) 1361384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1362384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1363384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 1364b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1365b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso goto err; 1366384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1367384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 1368384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (c != NULL) { 1369384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *table_name = 1370384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 1371384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_name = 1372384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 1373384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1374384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* don't delete built-in chain */ 1375384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nft_chain_builtin(c)) 1376384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1377384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1378384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, table_name) != 0) 1379384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1380384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1381384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (chain != NULL && strcmp(chain, chain_name) != 0) 1382384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1383384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1384384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = __nft_chain_del(h, c); 1385384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) 1386384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso break; 1387384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1388384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso deleted_ctr++; 138936ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo 139036ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo if (chain != NULL) 139136ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo break; 1392384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 1393384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 1394384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1395384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1396d01b2c28c8101f0d24e1db3f146fd845c2a634e8Giuseppe Longo nft_chain_list_iter_destroy(iter); 1397384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1398384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_list_free(list); 1399384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1400384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* chain not found */ 1401384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0 && deleted_ctr == 0) 1402384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso errno = ENOENT; 1403384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1404384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* the core expects 1 for success and 0 for error */ 1405384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret == 0 ? 1 : 0; 1406384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1407384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 14080aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayusostruct nft_chain * 1409e127d223d01aaa0886c7f279110ac36651b9a057Tomasz Bursztykanft_chain_list_find(struct nft_chain_list *list, 14100aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso const char *table, const char *chain) 14119c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso{ 14129c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 14139c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso struct nft_chain *c; 14149c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14159c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 1416b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 14179c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso return NULL; 14189c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14199c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 14209c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso while (c != NULL) { 14219c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso const char *table_name = 14229c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 14239c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso const char *chain_name = 14249c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 14259c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14269c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso if (strcmp(table, table_name) != 0) 14279c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso goto next; 14289c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14299c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso if (strcmp(chain, chain_name) != 0) 14309c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso goto next; 14319c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14320aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso nft_chain_list_iter_destroy(iter); 14339c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso return c; 14349c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayusonext: 14359c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 14369c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso } 14370aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso nft_chain_list_iter_destroy(iter); 14389c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso return NULL; 14399c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso} 14409c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14410aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayusostatic struct nft_chain * 14420aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayusonft_chain_find(struct nft_handle *h, const char *table, const char *chain) 14430aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso{ 14440aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso struct nft_chain_list *list; 14450aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 14460aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso list = nft_chain_list_get(h); 1447b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 14480aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso return NULL; 14490aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 1450e127d223d01aaa0886c7f279110ac36651b9a057Tomasz Bursztyka return nft_chain_list_find(list, table, chain); 14510aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso} 14520aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 1453384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_chain_user_rename(struct nft_handle *h,const char *chain, 1454384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *table, const char *newname) 1455384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 14564493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka char buf[MNL_SOCKET_BUFFER_SIZE]; 14574493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka struct nlmsghdr *nlh; 14584493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka struct nft_chain *c; 14599c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso uint64_t handle; 1460384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1461384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 14624493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka /* If built-in chains don't exist for this table, create them */ 14638b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (nft_xtables_config_load(h, XTABLES_CONFIG_DEFAULT, 0) < 0) 14648b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso nft_chain_builtin_init(h, table, NULL, NF_ACCEPT); 14654493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka 14669c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso /* Find the old chain to be renamed */ 14679c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso c = nft_chain_find(h, table, chain); 14689c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso if (c == NULL) { 14699c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso errno = ENOENT; 14709c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso return -1; 14719c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso } 14729c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso handle = nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_HANDLE); 14739c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso 14749c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso /* Now prepare the new name for the chain */ 14754493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka c = nft_chain_alloc(); 1476b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (c == NULL) 14774493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka return -1; 14784493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka 14794493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka nft_chain_attr_set(c, NFT_CHAIN_ATTR_TABLE, (char *)table); 14809c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso nft_chain_attr_set(c, NFT_CHAIN_ATTR_NAME, (char *)newname); 14819c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso nft_chain_attr_set_u64(c, NFT_CHAIN_ATTR_HANDLE, handle); 1482384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 14830391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_NEWCHAIN, h->family, 14849c541721d318598db45986ee2fd61491fadb53d0Pablo Neira Ayuso NLM_F_ACK, h->seq); 14854493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka nft_chain_nlmsg_build_payload(nlh, c); 14864493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka nft_chain_free(c); 14874493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka 14884493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka ret = mnl_talk(h, nlh, NULL, NULL); 14894493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka if (ret < 0) { 14904493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka if (errno != EEXIST) 14914493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka perror("mnl_talk:nft_chain_rename"); 14924493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka } 14934493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka 14944493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka /* the core expects 1 for success and 0 for error */ 14954493582ccb60a443fc7efeca78edafbefc689aa3Tomasz Bursztyka return ret == 0 ? 1 : 0; 1496384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1497384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1498384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int nft_table_list_cb(const struct nlmsghdr *nlh, void *data) 1499384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1500384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table *t; 1501384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table_list *list = data; 1502384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1503384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso t = nft_table_alloc(); 1504384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (t == NULL) { 1505384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("OOM"); 1506384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1507384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1508384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1509384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nft_table_nlmsg_parse(nlh, t) < 0) { 1510384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("nft_rule_nlmsg_parse"); 1511384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto out; 1512384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1513384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1514f77e1aca9aa8851b469f79c2db80ddb6f49253b2Pablo Neira Ayuso nft_table_list_add_tail(t, list); 1515384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1516384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return MNL_CB_OK; 1517384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoout: 1518384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_table_free(t); 1519384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1520384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return MNL_CB_OK; 1521384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1522384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1523384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic struct nft_table_list *nft_table_list_get(struct nft_handle *h) 1524384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1525384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 1526384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 1527384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1528384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table_list *list; 1529384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1530384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_table_list_alloc(); 1531b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 1532384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1533384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 15340391677c1a0b28c14d01febd9628a543e8e5fd62Tomasz Bursztyka nlh = nft_rule_nlmsg_build_hdr(buf, NFT_MSG_GETTABLE, h->family, 1535384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso NLM_F_DUMP, h->seq); 1536384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1537384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_talk(h, nlh, nft_table_list_cb, list); 1538384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret < 0) 1539384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_talk:nft_table_list_get"); 1540384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1541384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return list; 1542384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1543384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1544384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusobool nft_table_find(struct nft_handle *h, const char *tablename) 1545384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1546384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table_list *list; 1547384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table_list_iter *iter; 1548384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table *t; 1549384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso bool ret = false; 1550384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1551384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_table_list_get(h); 1552384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (list == NULL) 1553384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1554384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1555384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_table_list_iter_create(list); 1556b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1557384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1558384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1559384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso t = nft_table_list_iter_next(iter); 1560384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (t != NULL) { 1561384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *this_tablename = 1562384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_table_attr_get(t, NFT_TABLE_ATTR_NAME); 1563384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1564384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(tablename, this_tablename) == 0) 1565384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return true; 1566384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1567384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso t = nft_table_list_iter_next(iter); 1568384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1569384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1570384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_table_list_free(list); 1571384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1572384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1573384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret; 1574384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1575384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1576384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_for_each_table(struct nft_handle *h, 1577384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int (*func)(struct nft_handle *h, const char *tablename, bool counters), 1578384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso bool counters) 1579384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1580384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret = 1; 1581384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table_list *list; 1582384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table_list_iter *iter; 1583384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_table *t; 1584384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1585384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_table_list_get(h); 1586384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (list == NULL) { 1587384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = 0; 1588384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 1589384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1590384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1591384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_table_list_iter_create(list); 1592b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1593384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1594384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1595384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso t = nft_table_list_iter_next(iter); 1596384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (t != NULL) { 1597384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *tablename = 1598384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_table_attr_get(t, NFT_TABLE_ATTR_NAME); 1599384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1600384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso func(h, tablename, counters); 1601384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1602384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso t = nft_table_list_iter_next(iter); 1603384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1604384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1605384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_table_list_free(list); 1606384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1607384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 1608384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* the core expects 1 for success and 0 for error */ 1609384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret == 0 ? 1 : 0; 1610384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1611384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 16120aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayusoint nft_table_purge_chains(struct nft_handle *h, const char *this_table, 16130aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso struct nft_chain_list *chain_list) 16140aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso{ 16150aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso struct nft_chain_list_iter *iter; 16160aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso struct nft_chain *chain_obj; 16170aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 16180aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso iter = nft_chain_list_iter_create(chain_list); 1619b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 16200aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso return 0; 16210aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 16220aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso chain_obj = nft_chain_list_iter_next(iter); 16230aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso while (chain_obj != NULL) { 16240aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso const char *table = 16250aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso nft_chain_attr_get_str(chain_obj, NFT_CHAIN_ATTR_TABLE); 16260aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 16270aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso if (strcmp(this_table, table) != 0) 16280aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso goto next; 16290aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 16300aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso if (nft_chain_builtin(chain_obj)) 16310aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso goto next; 16320aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 16330aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso if ( __nft_chain_del(h, chain_obj) < 0) { 16340aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso if (errno != EBUSY) 16350aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso return -1; 16360aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso } 16370aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayusonext: 16380aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso chain_obj = nft_chain_list_iter_next(iter); 16390aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso } 16400aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso nft_chain_list_iter_destroy(iter); 16410aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 16420aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso return 0; 16430aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso} 16440aad20f3979e3b6becd40e4ed5bba8d09d90706ePablo Neira Ayuso 1645d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int __nft_rule_del(struct nft_handle *h, struct nft_rule_list *list, 1646d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct nft_rule *r) 1647384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1648384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret; 1649384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1650d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_list_del(r); 1651384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1652d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = rule_update_add(h, NFT_DO_DELETE, r); 1653d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret < 0) { 1654d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_free(r); 1655d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return -1; 1656d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 1657d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 1; 1658384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1659384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 166084909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longostruct nft_rule_list *nft_rule_list_create(struct nft_handle *h) 1661384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1662b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso return nft_rule_list_get(h); 16633aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso} 16643aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 166584909d171585d77fe769f03e2b1b96eab0aa0213Giuseppe Longovoid nft_rule_list_destroy(struct nft_rule_list *list) 16663aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso{ 16673aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso nft_rule_list_free(list); 16683aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso} 16693aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 16703aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayusostatic struct nft_rule * 1671e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayusonft_rule_find(struct nft_handle *h, struct nft_rule_list *list, 1672e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayuso const char *chain, const char *table, void *data, int rulenum) 16733aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso{ 16743aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule *r; 16753aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule_list_iter *iter; 16763aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso int rule_ctr = 0; 16773aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso bool found = false; 16783aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1679384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_rule_list_iter_create(list); 1680b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1681384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1682384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1683384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_list_iter_next(iter); 1684384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (r != NULL) { 1685384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *rule_table = 1686384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_get_str(r, NFT_RULE_ATTR_TABLE); 1687384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *rule_chain = 1688384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_get_str(r, NFT_RULE_ATTR_CHAIN); 1689384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1690384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, rule_table) != 0 || 1691384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso strcmp(chain, rule_chain) != 0) { 1692384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso DEBUGP("different chain / table\n"); 1693384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1694384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1695384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1696384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (rulenum >= 0) { 1697384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* Delete by rule number case */ 16984acee778f5712c4cc574e328183a3252ad81a802Tomasz Bursztyka if (rule_ctr != rulenum) 1699384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 17004acee778f5712c4cc574e328183a3252ad81a802Tomasz Bursztyka found = true; 17014acee778f5712c4cc574e328183a3252ad81a802Tomasz Bursztyka break; 1702384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } else { 1703e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayuso found = h->ops->rule_find(h->ops, r, data); 1704e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayuso if (found) 1705e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayuso break; 1706384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1707384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 17084acee778f5712c4cc574e328183a3252ad81a802Tomasz Bursztyka rule_ctr++; 1709384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_list_iter_next(iter); 1710384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1711384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1712384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_list_iter_destroy(iter); 1713384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 17143aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return found ? r : NULL; 1715384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1716384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1717384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_check(struct nft_handle *h, const char *chain, 1718b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso const char *table, void *data, bool verbose) 1719384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 17203aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule_list *list; 17213aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso int ret; 17223aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1723384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_fn = nft_rule_check; 1724384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 17253aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso list = nft_rule_list_create(h); 1726b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 17273aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return 0; 17283aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1729b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso ret = nft_rule_find(h, list, chain, table, data, -1) ? 1 : 0; 17303aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso if (ret == 0) 17313aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso errno = ENOENT; 17323aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 17333aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso nft_rule_list_destroy(list); 17343aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 17353aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return ret; 1736384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1737384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1738384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_delete(struct nft_handle *h, const char *chain, 1739b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso const char *table, void *data, bool verbose) 1740384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 17413aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso int ret = 0; 17423aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule *r; 17433aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule_list *list; 17443aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1745384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_fn = nft_rule_delete; 1746384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 17473aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso list = nft_rule_list_create(h); 1748b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 17493aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return 0; 17503aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1751b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso r = nft_rule_find(h, list, chain, table, data, -1); 17523aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso if (r != NULL) { 1753d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret =__nft_rule_del(h, list, r); 1754d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret < 0) 1755d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso errno = ENOMEM; 17563aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso } else 17573aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso errno = ENOENT; 17583aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 17593aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso nft_rule_list_destroy(list); 17603aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 17613aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return ret; 1762384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1763384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1764cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusostatic int 1765cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusonft_rule_add(struct nft_handle *h, const char *chain, 1766cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso const char *table, struct iptables_command_state *cs, 1767cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso uint64_t handle, bool verbose) 1768cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso{ 1769cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso struct nft_rule *r; 1770cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1771cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso r = nft_rule_new(h, chain, table, cs); 1772d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (r == NULL) 1773d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 0; 1774d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 177566a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso if (handle > 0) 177666a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso nft_rule_attr_set_u64(r, NFT_RULE_ATTR_POSITION, handle); 1777cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1778d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (rule_update_add(h, NFT_DO_INSERT, r) < 0) { 1779d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_free(r); 1780d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 0; 1781cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso } 1782cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1783d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 1; 1784cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso} 1785cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1786cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusoint nft_rule_insert(struct nft_handle *h, const char *chain, 1787b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso const char *table, void *data, int rulenum, bool verbose) 1788cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso{ 1789cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso struct nft_rule_list *list; 1790cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso struct nft_rule *r; 179166a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso uint64_t handle = 0; 1792cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1793cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso /* If built-in chains don't exist for this table, create them */ 1794cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso if (nft_xtables_config_load(h, XTABLES_CONFIG_DEFAULT, 0) < 0) 1795cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_chain_builtin_init(h, table, chain, NF_ACCEPT); 1796cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1797cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_fn = nft_rule_insert; 1798cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 179966a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso if (rulenum > 0) { 180066a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso list = nft_rule_list_create(h); 180166a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso if (list == NULL) 180266a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso goto err; 1803cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1804b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso r = nft_rule_find(h, list, chain, table, data, rulenum); 180566a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso if (r == NULL) { 180666a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso errno = ENOENT; 180766a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso goto err; 180866a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso } 1809cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 181066a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso handle = nft_rule_attr_get_u64(r, NFT_RULE_ATTR_HANDLE); 181166a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso DEBUGP("adding after rule handle %"PRIu64"\n", handle); 1812cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 181366a5399b6a4383ea4081d99ae852eebc1d65f265Pablo Neira Ayuso nft_rule_list_destroy(list); 1814cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso } 1815cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1816b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso return nft_rule_add(h, chain, table, data, handle, verbose); 1817cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayusoerr: 1818cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_rule_list_destroy(list); 1819cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso return 0; 1820cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso} 1821cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso 1822384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_delete_num(struct nft_handle *h, const char *chain, 18233aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso const char *table, int rulenum, bool verbose) 1824384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 18253aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso int ret = 0; 18263aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule *r; 18273aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule_list *list; 18283aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1829384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_fn = nft_rule_delete_num; 1830384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 18313aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso list = nft_rule_list_create(h); 1832b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 18333aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return 0; 18343aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1835e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayuso r = nft_rule_find(h, list, chain, table, NULL, rulenum); 18363aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso if (r != NULL) { 18373aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso ret = 1; 18383aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 18393aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso DEBUGP("deleting rule by number %d\n", rulenum); 1840d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = __nft_rule_del(h, list, r); 1841d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret < 0) 1842d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso errno = ENOMEM; 18433aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso } else 18443aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso errno = ENOENT; 18453aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 18463aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso nft_rule_list_destroy(list); 18473aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 18483aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return ret; 1849384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1850384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1851384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_replace(struct nft_handle *h, const char *chain, 1852b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso const char *table, void *data, int rulenum, bool verbose) 1853384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 18543aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso int ret = 0; 18553aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule *r; 18563aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso struct nft_rule_list *list; 1857384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1858384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_fn = nft_rule_replace; 1859384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 18603aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso list = nft_rule_list_create(h); 1861b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 18623aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return 0; 18633aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 1864b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso r = nft_rule_find(h, list, chain, table, data, rulenum); 18653aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso if (r != NULL) { 18661298a1014bc14c45de50cc242779dfa382c456c9Pablo Neira Ayuso DEBUGP("replacing rule with handle=%llu\n", 18671298a1014bc14c45de50cc242779dfa382c456c9Pablo Neira Ayuso (unsigned long long) 18681298a1014bc14c45de50cc242779dfa382c456c9Pablo Neira Ayuso nft_rule_attr_get_u64(r, NFT_RULE_ATTR_HANDLE)); 1869384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1870b756cf08d6eff885d808504c674bd7eb5ebabfbbPablo Neira Ayuso ret = nft_rule_append(h, chain, table, data, 1871cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso nft_rule_attr_get_u64(r, NFT_RULE_ATTR_HANDLE), 1872cf95f347e52ca8badc6a7149045d9c09f4fa666dPablo Neira Ayuso verbose); 18733aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso } else 18743aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso errno = ENOENT; 18753aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 18763aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso nft_rule_list_destroy(list); 18773aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso 18783aea037f359795edeb69426e2dde63c59540ba5fPablo Neira Ayuso return ret; 1879384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1880384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1881384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void 1882384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoprint_header(unsigned int format, const char *chain, const char *pol, 1883384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const struct xt_counters *counters, bool basechain, uint32_t refs) 1884384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1885384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("Chain %s", chain); 1886384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (basechain) { 1887384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(" (policy %s", pol); 1888384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (!(format & FMT_NOCOUNTS)) { 1889384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso fputc(' ', stdout); 189036cba824e1689c6255d4e33b7fa82541a774609bPablo Neira Ayuso xtables_print_num(counters->pcnt, (format|FMT_NOTABLE)); 1891384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso fputs("packets, ", stdout); 189236cba824e1689c6255d4e33b7fa82541a774609bPablo Neira Ayuso xtables_print_num(counters->bcnt, (format|FMT_NOTABLE)); 1893384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso fputs("bytes", stdout); 1894384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1895384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(")\n"); 1896384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } else { 1897384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(" (%u references)\n", refs); 1898384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1899384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1900384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (format & FMT_LINENUMBERS) 1901384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%-4s ", "%s "), "num"); 1902384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (!(format & FMT_NOCOUNTS)) { 1903384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (format & FMT_KILOMEGAGIGA) { 1904384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%5s ","%s "), "pkts"); 1905384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%5s ","%s "), "bytes"); 1906384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } else { 1907384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%8s ","%s "), "pkts"); 1908384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%10s ","%s "), "bytes"); 1909384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1910384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1911384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (!(format & FMT_NOTARGET)) 1912384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%-9s ","%s "), "target"); 1913384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso fputs(" prot ", stdout); 1914384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (format & FMT_OPTIONS) 1915384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso fputs("opt", stdout); 1916384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (format & FMT_VIA) { 1917384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT(" %-6s ","%s "), "in"); 1918384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT("%-6s ","%s "), "out"); 1919384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1920384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT(" %-19s ","%s "), "source"); 1921384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf(FMT(" %-19s "," %s "), "destination"); 1922384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("\n"); 1923384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1924384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1925384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int 1926b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka__nft_rule_list(struct nft_handle *h, const char *chain, const char *table, 1927384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int rulenum, unsigned int format, 19288aa384a7d54734fd830840a2593cd4f07749976fTomasz Bursztyka void (*cb)(struct nft_rule *r, unsigned int num, 1929384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso unsigned int format)) 1930384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1931384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_list *list; 1932384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule_list_iter *iter; 1933384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_rule *r; 1934384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int rule_ctr = 0, ret = 0; 1935384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1936384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_rule_list_get(h); 1937b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (list == NULL) 1938384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 1939384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1940384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_rule_list_iter_create(list); 1941b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 1942b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso goto err; 1943384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1944384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_list_iter_next(iter); 1945384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (r != NULL) { 1946384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *rule_table = 1947384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_get_str(r, NFT_RULE_ATTR_TABLE); 1948384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *rule_chain = 1949384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_attr_get_str(r, NFT_RULE_ATTR_CHAIN); 1950384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1951384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso rule_ctr++; 1952384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1953384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, rule_table) != 0 || 1954384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso strcmp(chain, rule_chain) != 0) 1955384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 1956384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 19570b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo if (rulenum > 0 && rule_ctr != rulenum) { 1958384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* List by rule number case */ 19590b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo goto next; 19600b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo } 1961384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 19628aa384a7d54734fd830840a2593cd4f07749976fTomasz Bursztyka cb(r, rule_ctr, format); 19630b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo if (rulenum > 0 && rule_ctr == rulenum) { 19640b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo ret = 1; 19650b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo break; 1966384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 19670b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo 1968384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 1969384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso r = nft_rule_list_iter_next(iter); 1970384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 1971384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1972384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_list_iter_destroy(iter); 1973b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayusoerr: 1974384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_rule_list_free(list); 1975384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1976384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret == 0) 1977384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso errno = ENOENT; 1978384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1979384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret; 1980384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 1981384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1982384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_list(struct nft_handle *h, const char *chain, const char *table, 1983384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int rulenum, unsigned int format) 1984384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 1985e23e66f9d1a25c75df684850b7cd99053708c4d0Tomasz Bursztyka const struct nft_family_ops *ops; 1986384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list *list; 1987384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 1988384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 19890a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo bool found = false; 1990384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 1991aa1601423175c90c37c3e6a3d6975d3e2eb74d1eTomasz Bursztyka /* If built-in chains don't exist for this table, create them */ 1992aa1601423175c90c37c3e6a3d6975d3e2eb74d1eTomasz Bursztyka if (nft_xtables_config_load(h, XTABLES_CONFIG_DEFAULT, 0) < 0) 1993aa1601423175c90c37c3e6a3d6975d3e2eb74d1eTomasz Bursztyka nft_chain_builtin_init(h, table, NULL, NF_ACCEPT); 1994aa1601423175c90c37c3e6a3d6975d3e2eb74d1eTomasz Bursztyka 1995e23e66f9d1a25c75df684850b7cd99053708c4d0Tomasz Bursztyka ops = nft_family_ops_lookup(h->family); 1996e23e66f9d1a25c75df684850b7cd99053708c4d0Tomasz Bursztyka 1997b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka if (chain && rulenum) { 1998b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka __nft_rule_list(h, chain, table, 1999e23e66f9d1a25c75df684850b7cd99053708c4d0Tomasz Bursztyka rulenum, format, ops->print_firewall); 2000b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka return 1; 2001b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka } 2002b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka 2003384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_chain_dump(h); 2004384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2005384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 2006b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 2007b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso goto err; 2008384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2009384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 2010384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (c != NULL) { 2011384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_table = 2012384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 2013384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_name = 2014384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 2015384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint32_t policy = 2016384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_u32(c, NFT_CHAIN_ATTR_POLICY); 2017384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint32_t refs = 2018384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_u32(c, NFT_CHAIN_ATTR_USE); 2019384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct xt_counters ctrs = { 2020384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso .pcnt = nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_PACKETS), 2021384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso .bcnt = nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_BYTES), 2022384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }; 2023384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso bool basechain = false; 2024384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2025384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nft_chain_attr_get(c, NFT_CHAIN_ATTR_HOOKNUM)) 2026384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso basechain = true; 2027384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2028384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, chain_table) != 0) 2029384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 2030384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (chain && strcmp(chain, chain_name) != 0) 2031384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 2032384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 20330a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo if (found) 20340a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo printf("\n"); 20350a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo 20367a1026f59c101a67233c65dd5ef9b0ae15945ca5Tomasz Bursztyka print_header(format, chain_name, policy_name[policy], 20370b3bafcedff19b69ff5a51855da28e8e83c05b71Giuseppe Longo &ctrs, basechain, refs); 20387a1026f59c101a67233c65dd5ef9b0ae15945ca5Tomasz Bursztyka 2039b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka __nft_rule_list(h, chain_name, table, 2040e23e66f9d1a25c75df684850b7cd99053708c4d0Tomasz Bursztyka rulenum, format, ops->print_firewall); 20410a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo 2042da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso /* we printed the chain we wanted, stop processing. */ 2043da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso if (chain) 2044da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso break; 2045da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso 20460a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo found = true; 20470a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo 2048384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 2049384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 2050384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2051384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 205234c59adfae98515468ec50c644c30115fee0b97eGiuseppe Longo nft_chain_list_iter_destroy(iter); 2053b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayusoerr: 2054384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_list_free(list); 2055384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2056384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 1; 2057384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 2058384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2059384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic void 20608aa384a7d54734fd830840a2593cd4f07749976fTomasz Bursztykalist_save(struct nft_rule *r, unsigned int num, unsigned int format) 2061384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 2062cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka struct iptables_command_state cs = {}; 2063cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka 2064cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka nft_rule_to_iptables_command_state(r, &cs); 2065cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka 2066cdc78b1d6bd7b48ec05d78fc6e6cd98473f40357Tomasz Bursztyka nft_rule_print_save(&cs, r, NFT_RULE_APPEND, !(format & FMT_NOCOUNTS)); 2067384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 2068384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2069384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusostatic int 20708dd2627afc462a2591c2f621743cae1a6b98d771Pablo Neira Ayusonft_rule_list_chain_save(struct nft_handle *h, const char *chain, 20718dd2627afc462a2591c2f621743cae1a6b98d771Pablo Neira Ayuso const char *table, struct nft_chain_list *list, 20728dd2627afc462a2591c2f621743cae1a6b98d771Pablo Neira Ayuso int counters) 2073384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 2074384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 2075384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 2076384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2077384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 2078b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 2079384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 2080384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2081384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 2082384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (c != NULL) { 2083384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_table = 2084384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 2085384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_name = 2086384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 2087384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint32_t policy = 2088384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_u32(c, NFT_CHAIN_ATTR_POLICY); 2089384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 20908dd2627afc462a2591c2f621743cae1a6b98d771Pablo Neira Ayuso if (strcmp(table, chain_table) != 0 || 20918dd2627afc462a2591c2f621743cae1a6b98d771Pablo Neira Ayuso (chain && strcmp(chain, chain_name) != 0)) 2092384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 2093384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2094384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* this is a base chain */ 20957244bef43f350ab31ef54db8a81905f6c68acac0Tomasz Bursztyka if (nft_chain_builtin(c)) { 2096384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("-P %s %s", chain_name, policy_name[policy]); 2097384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2098384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (counters) { 20990a5f6c302f2f596f6e6aa0241a7772cf441b026fGiuseppe Longo printf(" -c %"PRIu64" %"PRIu64"\n", 2100384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_PACKETS), 2101384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_u64(c, NFT_CHAIN_ATTR_BYTES)); 2102384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } else 2103384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("\n"); 2104384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } else { 2105384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso printf("-N %s\n", chain_name); 2106384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2107384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 2108384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 2109384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2110384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2111e9a0ef8f1e27f5ef13a27f6cc984e8f2e05afd72Tomasz Bursztyka nft_chain_list_iter_destroy(iter); 2112e9a0ef8f1e27f5ef13a27f6cc984e8f2e05afd72Tomasz Bursztyka 2113384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 1; 2114384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 2115384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2116384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_rule_list_save(struct nft_handle *h, const char *chain, 2117384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *table, int rulenum, int counters) 2118384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 2119384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list *list; 2120384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain_list_iter *iter; 2121384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nft_chain *c; 212210f92fce0a2ea1805c8b269543b8f1738d22bf3dPablo Neira Ayuso int ret = 1; 2123384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2124384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso list = nft_chain_dump(h); 2125384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2126384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* Dump policies and custom chains first */ 212710f92fce0a2ea1805c8b269543b8f1738d22bf3dPablo Neira Ayuso if (!rulenum) 21288dd2627afc462a2591c2f621743cae1a6b98d771Pablo Neira Ayuso nft_rule_list_chain_save(h, chain, table, list, counters); 2129384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2130384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* Now dump out rules in this table */ 2131384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso iter = nft_chain_list_iter_create(list); 2132b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 2133b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso goto err; 2134384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2135384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 2136384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso while (c != NULL) { 2137384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_table = 2138384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_TABLE); 2139384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *chain_name = 2140384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_attr_get_str(c, NFT_CHAIN_ATTR_NAME); 2141384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2142384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (strcmp(table, chain_table) != 0) 2143384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 2144384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (chain && strcmp(chain, chain_name) != 0) 2145384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto next; 2146384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2147b08836a33bac28860400e1e34c8244249f651a6fTomasz Bursztyka ret = __nft_rule_list(h, chain_name, table, rulenum, 214810f92fce0a2ea1805c8b269543b8f1738d22bf3dPablo Neira Ayuso counters ? 0 : FMT_NOCOUNTS, list_save); 2149da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso 2150da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso /* we printed the chain we wanted, stop processing. */ 2151da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso if (chain) 2152da07c930b9e2aaf8df24022a175b1774aa0bdd8bPablo Neira Ayuso break; 2153384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusonext: 2154384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso c = nft_chain_list_iter_next(iter); 2155384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2156e9a0ef8f1e27f5ef13a27f6cc984e8f2e05afd72Tomasz Bursztyka 2157e9a0ef8f1e27f5ef13a27f6cc984e8f2e05afd72Tomasz Bursztyka nft_chain_list_iter_destroy(iter); 2158b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayusoerr: 2159384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nft_chain_list_free(list); 2160384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 216110f92fce0a2ea1805c8b269543b8f1738d22bf3dPablo Neira Ayuso return ret; 2162384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 2163384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2164a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztykaint nft_rule_zero_counters(struct nft_handle *h, const char *chain, 2165a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka const char *table, int rulenum) 2166a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka{ 2167a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka struct iptables_command_state cs = {}; 2168a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka struct nft_rule_list *list; 2169a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka struct nft_rule *r; 2170a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka int ret = 0; 2171a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2172a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka nft_fn = nft_rule_delete; 2173a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2174a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka list = nft_rule_list_create(h); 2175a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka if (list == NULL) 2176a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka return 0; 2177a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2178e2a2c72277b49ac611809b3978365ab3010e1597Pablo Neira Ayuso r = nft_rule_find(h, list, chain, table, NULL, rulenum); 2179a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka if (r == NULL) { 2180a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka errno = ENOENT; 2181a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka ret = 1; 2182a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka goto error; 2183a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka } 2184a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2185a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka nft_rule_to_iptables_command_state(r, &cs); 2186a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2187a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka cs.counters.pcnt = cs.counters.bcnt = 0; 2188a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2189a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka ret = nft_rule_append(h, chain, table, &cs, 2190a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka nft_rule_attr_get_u64(r, NFT_RULE_ATTR_HANDLE), 2191a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka false); 2192a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2193a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztykaerror: 2194a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka nft_rule_list_destroy(list); 2195a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2196a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka return ret; 2197a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka} 2198a69cc575295eedb44f0fa33cd5fcf1cc0114133aTomasz Bursztyka 2199d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayusostatic int nft_action(struct nft_handle *h, int action) 22009e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso{ 2201d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso int flags = NLM_F_CREATE, type; 2202d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso struct rule_update *n, *tmp; 22039e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso struct nlmsghdr *nlh; 2204d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso uint32_t seq = 1; 22059e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso int ret; 22069e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso 2207d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nft_batch_begin(h->batch, seq++); 22089e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso 2209d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso list_for_each_entry_safe(n, tmp, &h->rule_list, head) { 2210d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso switch (n->type) { 2211d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_APPEND: 2212d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type = NFT_MSG_NEWRULE; 2213d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso flags |= NLM_F_APPEND; 2214d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso break; 2215d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_INSERT: 2216d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type = NFT_MSG_NEWRULE; 2217d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso break; 2218d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_REPLACE: 2219d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type = NFT_MSG_NEWRULE; 2220d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso flags |= NLM_F_REPLACE; 2221d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso break; 2222d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_DELETE: 2223d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_FLUSH: 2224d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type = NFT_MSG_DELRULE; 2225d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso break; 2226d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso default: 2227d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return 0; 2228d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 22299e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso 2230d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nlh = nft_rule_nlmsg_build_hdr(mnl_nlmsg_batch_current(h->batch), 2231d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso type, h->family, flags, seq++); 2232d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_nlmsg_build_payload(nlh, n->rule); 2233d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_print_debug(n->rule, nlh); 2234d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2235d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso h->rule_list_num--; 2236d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso list_del(&n->head); 2237d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso nft_rule_free(n->rule); 2238d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso free(n); 2239d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2240d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (!mnl_nlmsg_batch_next(h->batch)) 2241d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso h->batch = mnl_nft_batch_page_add(h->batch); 2242d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso } 2243d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2244d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso switch (action) { 2245d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_COMMIT: 2246d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nft_batch_end(h->batch, seq++); 2247d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso break; 2248d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso case NFT_DO_ABORT: 2249d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso break; 22509e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso } 2251d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2252d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (!mnl_nlmsg_batch_is_empty(h->batch)) 2253d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso h->batch = mnl_nft_batch_page_add(h->batch); 2254d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2255d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso ret = mnl_nft_batch_talk(h); 2256d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso if (ret < 0) 2257d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso perror("mnl_nft_batch_talk:"); 2258d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2259d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso mnl_nlmsg_batch_reset(h->batch); 2260d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso 2261f041efe3c26e3059df1ac8f1775f77423d4be5f6Pablo Neira Ayuso return ret == 0 ? 1 : 0; 22629e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso} 22639e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso 22649e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayusoint nft_commit(struct nft_handle *h) 22659e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso{ 2266d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return nft_action(h, NFT_DO_COMMIT); 22679e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso} 22689e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso 22699e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayusoint nft_abort(struct nft_handle *h) 22709e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso{ 2271d6a127cd5710f8c60e95bfd0378ca352c07140a9Pablo Neira Ayuso return nft_action(h, NFT_DO_ABORT); 22729e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso} 22739e62dc8637f210cdeaed784396fecab9b6e5f043Pablo Neira Ayuso 2274384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoint nft_compatible_revision(const char *name, uint8_t rev, int opt) 2275384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 2276384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct mnl_socket *nl; 2277384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso char buf[MNL_SOCKET_BUFFER_SIZE]; 2278384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nlmsghdr *nlh; 2279384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso uint32_t portid, seq, type; 2280384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int ret = 0; 2281384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 228285512f09680a798ebe92e96ad62eeae863fbc791Pablo Neira Ayuso if (opt == IPT_SO_GET_REVISION_MATCH || 228385512f09680a798ebe92e96ad62eeae863fbc791Pablo Neira Ayuso opt == IP6T_SO_GET_REVISION_MATCH) 2284384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso type = 0; 2285384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso else 2286384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso type = 1; 2287384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2288384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nlh = mnl_nlmsg_put_header(buf); 2289384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nlh->nlmsg_type = (NFNL_SUBSYS_NFT_COMPAT << 8) | NFNL_MSG_COMPAT_GET; 2290384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nlh->nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK; 2291384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nlh->nlmsg_seq = seq = time(NULL); 2292384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2293384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso struct nfgenmsg *nfg = mnl_nlmsg_put_extra_header(nlh, sizeof(*nfg)); 2294384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nfg->nfgen_family = AF_INET; 2295384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nfg->version = NFNETLINK_V0; 2296384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nfg->res_id = 0; 2297384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2298384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_attr_put_strz(nlh, NFTA_COMPAT_NAME, name); 2299384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_attr_put_u32(nlh, NFTA_COMPAT_REV, htonl(rev)); 2300384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_attr_put_u32(nlh, NFTA_COMPAT_TYPE, htonl(type)); 2301384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2302384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso DEBUGP("requesting `%s' rev=%d type=%d via nft_compat\n", 2303384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso name, rev, type); 2304384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2305384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso nl = mnl_socket_open(NETLINK_NETFILTER); 2306384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (nl == NULL) { 2307384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_open"); 2308384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return 0; 2309384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2310384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2311384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (mnl_socket_bind(nl, 0, MNL_SOCKET_AUTOPID) < 0) { 2312384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_bind"); 2313384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 2314384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2315384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso portid = mnl_socket_get_portid(nl); 2316384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2317384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (mnl_socket_sendto(nl, nlh, nlh->nlmsg_len) < 0) { 2318384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_send"); 2319384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 2320384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2321384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2322384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_socket_recvfrom(nl, buf, sizeof(buf)); 2323384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret == -1) { 2324384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_socket_recvfrom"); 2325384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 2326384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2327384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2328384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso ret = mnl_cb_run(buf, ret, seq, portid, NULL, NULL); 2329384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if (ret == -1) { 2330384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso perror("mnl_cb_run"); 2331384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso goto err; 2332384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2333384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2334384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoerr: 2335384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso mnl_socket_close(nl); 2336384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2337384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return ret < 0 ? 0 : 1; 2338384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 2339384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2340384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso/* Translates errno numbers into more human-readable form than strerror. */ 2341384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayusoconst char *nft_strerror(int err) 2342384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso{ 2343384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso unsigned int i; 2344384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso static struct table_struct { 2345384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso void *fn; 2346384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso int err; 2347384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso const char *message; 2348384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } table[] = 2349384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { 2350384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_chain_user_del, ENOTEMPTY, "Chain is not empty" }, 2351384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_chain_user_del, EINVAL, "Can't delete built-in chain" }, 2352384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_chain_user_del, EMLINK, 2353384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso "Can't delete chain with references left" }, 2354384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_chain_user_add, EEXIST, "Chain already exists" }, 2355384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_rule_add, E2BIG, "Index of insertion too big" }, 2356384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_rule_replace, E2BIG, "Index of replacement too big" }, 2357384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_rule_delete_num, E2BIG, "Index of deletion too big" }, 2358384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso/* { TC_READ_COUNTER, E2BIG, "Index of counter too big" }, 2359384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { TC_ZERO_COUNTER, E2BIG, "Index of counter too big" }, */ 2360384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_rule_add, ELOOP, "Loop found in table" }, 2361384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_rule_add, EINVAL, "Target problem" }, 2362384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso /* ENOENT for DELETE probably means no matching rule */ 2363384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_rule_delete, ENOENT, 2364384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso "Bad rule (does a matching rule exist in that chain?)" }, 2365384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_chain_set, ENOENT, "Bad built-in chain name" }, 2366384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { nft_chain_set, EINVAL, "Bad policy name" }, 2367384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { NULL, EPERM, "Permission denied (you must be root)" }, 2368384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { NULL, 0, "Incompatible with this kernel" }, 2369384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { NULL, ENOPROTOOPT, "iptables who? (do you need to insmod?)" }, 2370384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { NULL, ENOSYS, "Will be implemented real soon. I promise ;)" }, 2371384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { NULL, ENOMEM, "Memory allocation problem" }, 2372384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso { NULL, ENOENT, "No chain/target/match by that name" }, 2373384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso }; 2374384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2375384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso for (i = 0; i < sizeof(table)/sizeof(struct table_struct); i++) { 2376384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso if ((!table[i].fn || table[i].fn == nft_fn) 2377384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso && table[i].err == err) 2378384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return table[i].message; 2379384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso } 2380384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso 2381384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso return strerror(err); 2382384958620abab397062b67fb2763e813b63f74f0Pablo Neira Ayuso} 23838b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 23848b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayusostatic void xtables_config_perror(uint32_t flags, const char *fmt, ...) 23858b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso{ 23868b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso va_list args; 23878b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 23888b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso va_start(args, fmt); 23898b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 23908b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (flags & NFT_LOAD_VERBOSE) 23918b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso vfprintf(stderr, fmt, args); 23928b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 23938b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso va_end(args); 23948b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso} 23958b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 23968b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayusoint nft_xtables_config_load(struct nft_handle *h, const char *filename, 23978b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso uint32_t flags) 23988b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso{ 23998b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso struct nft_table_list *table_list = nft_table_list_alloc(); 24008b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso struct nft_chain_list *chain_list = nft_chain_list_alloc(); 24016a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey struct nft_table_list_iter *titer = NULL; 24026a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey struct nft_chain_list_iter *citer = NULL; 24038b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso struct nft_table *table; 24048b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso struct nft_chain *chain; 2405cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo uint32_t table_family, chain_family; 2406cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo bool found = false; 24078b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 24088b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (xtables_config_parse(filename, table_list, chain_list) < 0) { 24098b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (errno == ENOENT) { 24108b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24118b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "configuration file `%s' does not exists\n", 24128b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso filename); 24138b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } else { 24148b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24158b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "Fatal error parsing config file: %s\n", 24168b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso strerror(errno)); 24178b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24186a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey goto err; 24198b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24208b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 24218b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso /* Stage 1) create tables */ 24228b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso titer = nft_table_list_iter_create(table_list); 24238b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso while ((table = nft_table_list_iter_next(titer)) != NULL) { 2424cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo table_family = nft_table_attr_get_u32(table, 2425cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo NFT_TABLE_ATTR_FAMILY); 2426cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo if (h->family != table_family) 2427cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo continue; 2428cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo 2429cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo found = true; 2430cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo 24318b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (nft_table_add(h, table) < 0) { 24328b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (errno == EEXIST) { 24338b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24348b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "table `%s' already exists, skipping\n", 24358b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_table_attr_get(table, NFT_TABLE_ATTR_NAME)); 24368b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } else { 24378b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24388b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "table `%s' cannot be create, reason `%s'. Exitting\n", 24398b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_table_attr_get(table, NFT_TABLE_ATTR_NAME), 24408b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso strerror(errno)); 24416a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey goto err; 24428b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24438b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso continue; 24448b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24458b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, "table `%s' has been created\n", 24468b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_table_attr_get(table, NFT_TABLE_ATTR_NAME)); 24478b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24482c5850147937cd3da104adf654cc7b2d1f0c0a0bPablo Neira Ayuso nft_table_list_iter_destroy(titer); 24492c5850147937cd3da104adf654cc7b2d1f0c0a0bPablo Neira Ayuso nft_table_list_free(table_list); 24508b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 2451cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo if (!found) 24526a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey goto err; 2453cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo 24548b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso /* Stage 2) create chains */ 24558b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso citer = nft_chain_list_iter_create(chain_list); 24568b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso while ((chain = nft_chain_list_iter_next(citer)) != NULL) { 2457cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo chain_family = nft_chain_attr_get_u32(chain, 2458cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo NFT_CHAIN_ATTR_TABLE); 2459cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo if (h->family != chain_family) 2460cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo continue; 2461cea310bd1d078618e87b83410f8c6f75b34de450Giuseppe Longo 24628b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (nft_chain_add(h, chain) < 0) { 24638b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso if (errno == EEXIST) { 24648b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24658b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "chain `%s' already exists in table `%s', skipping\n", 24668b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_chain_attr_get(chain, NFT_CHAIN_ATTR_NAME), 24678b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_chain_attr_get(chain, NFT_CHAIN_ATTR_TABLE)); 24688b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } else { 24698b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24708b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "chain `%s' cannot be create, reason `%s'. Exitting\n", 24718b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_chain_attr_get(chain, NFT_CHAIN_ATTR_NAME), 24728b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso strerror(errno)); 24736a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey goto err; 24748b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24758b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso continue; 24768b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24778b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso 24788b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso xtables_config_perror(flags, 24798b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso "chain `%s' in table `%s' has been created\n", 24808b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_chain_attr_get(chain, NFT_CHAIN_ATTR_NAME), 24818b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso (char *)nft_chain_attr_get(chain, NFT_CHAIN_ATTR_TABLE)); 24828b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso } 24832c5850147937cd3da104adf654cc7b2d1f0c0a0bPablo Neira Ayuso nft_chain_list_iter_destroy(citer); 24842c5850147937cd3da104adf654cc7b2d1f0c0a0bPablo Neira Ayuso nft_chain_list_free(chain_list); 24852c5850147937cd3da104adf654cc7b2d1f0c0a0bPablo Neira Ayuso 24868b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso return 0; 24876a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey 24886a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Reyerr: 24896a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey nft_table_list_free(table_list); 24906a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey nft_chain_list_free(chain_list); 24916a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey 24926a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey if (titer != NULL) 24936a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey nft_table_list_iter_destroy(titer); 24946a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey if (citer != NULL) 2495504119fe14bffde5800a631da89b80ed6043cecbPablo Neira Ayuso nft_chain_list_iter_destroy(citer); 24966a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey 24976a4033b70dfdcc2df66f4ea51c901786a2b6131cAna Rey return -1; 24988b9ea2e3f8d685a6b940691cabf5e82c96254747Pablo Neira Ayuso} 2499b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2500b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longoint nft_chain_zero_counters(struct nft_handle *h, const char *chain, 2501b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo const char *table) 2502b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo{ 2503b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo struct nft_chain_list *list; 2504b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo struct nft_chain_list_iter *iter; 2505b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo struct nft_chain *c; 2506b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo struct nlmsghdr *nlh; 2507b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo char buf[MNL_SOCKET_BUFFER_SIZE]; 2508b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo int ret = 0; 2509b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2510b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo list = nft_chain_list_get(h); 2511b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo if (list == NULL) 2512b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo goto err; 2513b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2514b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo iter = nft_chain_list_iter_create(list); 2515b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso if (iter == NULL) 2516b6d90619891e9a2f804af6af9860da8f95878abfPablo Neira Ayuso goto err; 2517b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2518b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo c = nft_chain_list_iter_next(iter); 2519b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo while (c != NULL) { 2520b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo const char *chain_name = 2521b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_attr_get(c, NFT_CHAIN_ATTR_NAME); 2522b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo const char *chain_table = 2523b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_attr_get(c, NFT_CHAIN_ATTR_TABLE); 2524b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2525b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo if (strcmp(table, chain_table) != 0) 2526b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo goto next; 2527b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2528b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo if (chain != NULL && strcmp(chain, chain_name) != 0) 2529b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo goto next; 2530b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2531b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_attr_set_u64(c, NFT_CHAIN_ATTR_PACKETS, 0); 2532b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_attr_set_u64(c, NFT_CHAIN_ATTR_BYTES, 0); 2533b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2534b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_attr_unset(c, NFT_CHAIN_ATTR_HANDLE); 2535b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2536b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nlh = nft_chain_nlmsg_build_hdr(buf, NFT_MSG_NEWCHAIN, 2537b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo h->family, NLM_F_ACK, h->seq); 2538b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2539b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_nlmsg_build_payload(nlh, c); 2540b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2541b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo ret = mnl_talk(h, nlh, NULL, NULL); 2542b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo if (ret < 0) 2543b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo perror("mnl_talk:nft_chain_zero_counters"); 2544b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 254536ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo if (chain != NULL) 254636ca9bdb288f7ba528307b7695ab94f7fa8e9a2dGiuseppe Longo break; 2547b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longonext: 2548b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo c = nft_chain_list_iter_next(iter); 2549b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo } 2550b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2551b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_list_iter_destroy(iter); 2552b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2553b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longoerr: 2554b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo nft_chain_list_free(list); 2555b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo 2556b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo /* the core expects 1 for success and 0 for error */ 2557b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo return ret == 0 ? 1 : 0; 2558b48126ca92cc44e88aa024e6da7ff245914d6a53Giuseppe Longo} 2559