From ba196a97e810746e5660fe3f57c87c0ed0f2b324 Mon Sep 17 00:00:00 2001 From: Ken-ichirou MATSUZAWA Date: Tue, 15 Nov 2016 16:38:16 +0900 Subject: 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 Signed-off-by: Pablo Neira Ayuso --- configure.ac | 5 ++- utils/Makefile.am | 4 +++ utils/nf-log.c | 103 +++++++++++++++++++++++++++++++++++++++++++++++++++++- 3 files changed, 110 insertions(+), 2 deletions(-) diff --git a/configure.ac b/configure.ac index ead9399..c914e00 100644 --- a/configure.ac +++ b/configure.ac @@ -19,7 +19,7 @@ AM_PROG_LIBTOOL AC_PROG_INSTALL AC_PROG_LN_S -case "$host" in +case "$host" in *-*-linux* | *-*-uclinux*) ;; *) AC_MSG_ERROR([Linux only, dude!]);; esac @@ -31,6 +31,9 @@ AM_CONDITIONAL([BUILD_IPULOG], [test "x$with_ipulog" != xno]) dnl Dependencies PKG_CHECK_MODULES([LIBNFNETLINK], [libnfnetlink >= 0.0.41]) PKG_CHECK_MODULES([LIBMNL], [libmnl >= 1.0.3]) +PKG_CHECK_MODULES([LIBNETFILTER_CONNTRACK], [libnetfilter_conntrack >= 1.0.2], + [HAVE_LNFCT=1], [HAVE_LNFCT=0]) +AM_CONDITIONAL([BUILD_NFCT], [test "$HAVE_LNFCT" -eq 1]) dnl Output the makefile AC_CONFIG_FILES([Makefile src/Makefile include/Makefile 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 #include -#include +#include #include #include +#ifdef BUILD_NFCT +#include +#include +#include +#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); -- cgit v1.2.3