diff options
author | Ken-ichirou MATSUZAWA <chamaken@gmail.com> | 2016-11-15 16:38:16 +0900 |
---|---|---|
committer | Pablo Neira Ayuso <pablo@netfilter.org> | 2016-11-24 12:52:37 +0100 |
commit | ba196a97e810746e5660fe3f57c87c0ed0f2b324 (patch) | |
tree | 1a74551e89521304ccea2443daf718a8786e5bee /utils | |
parent | ac35f78f7a13d3464009f4d8d1576d308e3d17c8 (diff) |
utils: nf-log: attaching a conntrack information
This patch enables nf-log in utils directory to show conntrack
information if libnetfilter_conntrack exists.
Signed-off-by: Ken-ichirou MATSUZAWA <chamas@h4.dion.ne.jp>
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
Diffstat (limited to 'utils')
-rw-r--r-- | utils/Makefile.am | 4 | ||||
-rw-r--r-- | utils/nf-log.c | 103 |
2 files changed, 106 insertions, 1 deletions
diff --git a/utils/Makefile.am b/utils/Makefile.am index dfe5f34..a848b10 100644 --- a/utils/Makefile.am +++ b/utils/Makefile.am @@ -9,6 +9,10 @@ nfulnl_test_LDFLAGS = -dynamic nf_log_SOURCES = nf-log.c nf_log_LDADD = ../src/libnetfilter_log.la nf_log_LDFLAGS = -dynamic -lmnl +if BUILD_NFCT +nf_log_LDFLAGS += $(LIBNETFILTER_CONNTRACK_LIBS) +nf_log_CFLAGS = -DBUILD_NFCT +endif if BUILD_IPULOG check_PROGRAMS += ulog_test diff --git a/utils/nf-log.c b/utils/nf-log.c index 5f2a192..ad8369c 100644 --- a/utils/nf-log.c +++ b/utils/nf-log.c @@ -3,15 +3,108 @@ #include <stdlib.h> #include <arpa/inet.h> -#include <linux/netfilter/nfnetlink_log.h> +#include <libnetfilter_log/linux_nfnetlink_log.h> #include <libmnl/libmnl.h> #include <libnetfilter_log/libnetfilter_log.h> +#ifdef BUILD_NFCT +#include <linux/netfilter/nf_conntrack_common.h> +#include <linux/netfilter/nf_conntrack_tuple_common.h> +#include <libnetfilter_conntrack/libnetfilter_conntrack.h> +#endif + +#ifdef BUILD_NFCT +static int print_ctinfo(const struct nlattr *const attr) +{ + uint32_t ctinfo; + + if (attr == NULL) + return MNL_CB_OK; + + ctinfo = ntohl(mnl_attr_get_u32(attr)); + printf(" ip_conntrack_info:"); + + switch (CTINFO2DIR(ctinfo)) { + case IP_CT_DIR_ORIGINAL: + printf(" ORIGINAL /"); + break; + case IP_CT_DIR_REPLY: + printf(" REPLY /"); + break; + default: + printf(" unknown dir: %d\n", CTINFO2DIR(ctinfo)); + return MNL_CB_ERROR; + } + + switch (ctinfo) { + case IP_CT_ESTABLISHED: + case IP_CT_ESTABLISHED_REPLY: + printf(" ESTABLISHED\n"); + break; + case IP_CT_RELATED: + case IP_CT_RELATED_REPLY: + printf(" RELATED\n"); + break; + case IP_CT_NEW: + case IP_CT_NEW_REPLY: + printf(" NEW\n"); + break; + default: + printf(" unknown ctinfo: %d\n", ctinfo); + return MNL_CB_ERROR; + } + + return MNL_CB_OK; +} + +static int print_nfct(uint8_t family, + const struct nlattr *const info_attr, + const struct nlattr *const ct_attr) +{ + char buf[4096]; + struct nf_conntrack *ct = NULL; + + if (info_attr != NULL) + print_ctinfo(info_attr); + + if (ct_attr == NULL) + return MNL_CB_OK; + + ct = nfct_new(); + if (ct == NULL) { + perror("nfct_new"); + return MNL_CB_ERROR; + } + + if (nfct_payload_parse(mnl_attr_get_payload(ct_attr), + mnl_attr_get_payload_len(ct_attr), + family, ct) < 0) { + perror("nfct_payload_parse"); + nfct_destroy(ct); + return MNL_CB_ERROR; + } + + nfct_snprintf(buf, sizeof(buf), ct, 0, NFCT_O_DEFAULT, 0); + printf(" %s\n", buf); + nfct_destroy(ct); + + return MNL_CB_OK; +} +#else +static int print_nfct(uint8_t family, + const struct nlattr *const info_attr, + const struct nlattr *const ct_attr) +{ + return MNL_CB_OK; +} +#endif + static int log_cb(const struct nlmsghdr *nlh, void *data) { struct nlattr *attrs[NFULA_MAX + 1] = { NULL }; struct nfulnl_msg_packet_hdr *ph = NULL; + struct nfgenmsg *nfg; const char *prefix = NULL; uint32_t mark = 0; char buf[4096]; @@ -21,6 +114,8 @@ static int log_cb(const struct nlmsghdr *nlh, void *data) if (ret != MNL_CB_OK) return ret; + nfg = mnl_nlmsg_get_payload(nlh); + if (attrs[NFULA_PACKET_HDR]) ph = mnl_attr_get_payload(attrs[NFULA_PACKET_HDR]); if (attrs[NFULA_PREFIX]) @@ -38,6 +133,8 @@ static int log_cb(const struct nlmsghdr *nlh, void *data) return MNL_CB_ERROR; printf("%s (ret=%d)\n", buf, ret); + print_nfct(nfg->nfgen_family, attrs[NFULA_CT_INFO], attrs[NFULA_CT]); + return MNL_CB_OK; } @@ -108,6 +205,10 @@ int main(int argc, char *argv[]) exit(EXIT_FAILURE); } +#ifdef BUILD_NFCT + mnl_attr_put_u16(nlh, NFULA_CFG_FLAGS, htons(NFULNL_CFG_F_CONNTRACK)); +#endif + if (mnl_socket_sendto(nl, nlh, nlh->nlmsg_len) < 0) { perror("mnl_socket_sendto"); exit(EXIT_FAILURE); |