From 0a4e6a9f76f86bf822ae6738a6e1fc5c8e5b4fa4 Mon Sep 17 00:00:00 2001
From: Daniel Borkmann <borkmann@iogearbox.net>
Date: Thu, 29 Nov 2012 15:58:12 +0100
Subject: [PATCH] flowtop: add state descriptions for dccp, sctp

Signed-off-by: Daniel Borkmann <borkmann@iogearbox.net>
---
 src/flowtop.c | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/src/flowtop.c b/src/flowtop.c
index 32756567..d22d3b5b 100644
--- a/src/flowtop.c
+++ b/src/flowtop.c
@@ -27,6 +27,8 @@
 #include <ctype.h>
 #include <libnetfilter_conntrack/libnetfilter_conntrack.h>
 #include <libnetfilter_conntrack/libnetfilter_conntrack_tcp.h>
+#include <libnetfilter_conntrack/libnetfilter_conntrack_dccp.h>
+#include <libnetfilter_conntrack/libnetfilter_conntrack_sctp.h>
 #include <GeoIP.h>
 #include <GeoIPCity.h>
 #include <netinet/in.h>
@@ -165,6 +167,54 @@ static const uint8_t tcp_states[] = {
 	TCP_CONNTRACK_NONE,
 };
 
+static const char *const dccp_state2str[DCCP_CONNTRACK_MAX] = {
+	[DCCP_CONNTRACK_NONE]		= "NOSTATE",
+	[DCCP_CONNTRACK_REQUEST]	= "REQUEST",
+	[DCCP_CONNTRACK_RESPOND]	= "RESPOND",
+	[DCCP_CONNTRACK_PARTOPEN]	= "PARTOPEN",
+	[DCCP_CONNTRACK_OPEN]		= "OPEN",
+	[DCCP_CONNTRACK_CLOSEREQ]	= "CLOSEREQ",
+	[DCCP_CONNTRACK_CLOSING]	= "CLOSING",
+	[DCCP_CONNTRACK_TIMEWAIT]	= "TIMEWAIT",
+	[DCCP_CONNTRACK_IGNORE]		= "IGNORE",
+	[DCCP_CONNTRACK_INVALID]	= "INVALID",
+};
+
+static const uint8_t dccp_states[] = {
+	DCCP_CONNTRACK_NONE,
+	DCCP_CONNTRACK_REQUEST,
+	DCCP_CONNTRACK_RESPOND,
+	DCCP_CONNTRACK_PARTOPEN,
+	DCCP_CONNTRACK_OPEN,
+	DCCP_CONNTRACK_CLOSEREQ,
+	DCCP_CONNTRACK_CLOSING,
+	DCCP_CONNTRACK_TIMEWAIT,
+	DCCP_CONNTRACK_IGNORE,
+	DCCP_CONNTRACK_INVALID,
+};
+
+static const char *const sctp_state2str[SCTP_CONNTRACK_MAX] = {
+	[SCTP_CONNTRACK_NONE]		= "NOSTATE",
+	[SCTP_CONNTRACK_CLOSED]		= "CLOSED",
+	[SCTP_CONNTRACK_COOKIE_WAIT]	= "COOKIE_WAIT",
+	[SCTP_CONNTRACK_COOKIE_ECHOED]	= "COOKIE_ECHOED",
+	[SCTP_CONNTRACK_ESTABLISHED]	= "ESTABLISHED",
+	[SCTP_CONNTRACK_SHUTDOWN_SENT]	= "SHUTDOWN_SENT",
+	[SCTP_CONNTRACK_SHUTDOWN_RECD]	= "SHUTDOWN_RECD",
+	[SCTP_CONNTRACK_SHUTDOWN_ACK_SENT] = "SHUTDOWN_ACK_SENT",
+};
+
+static const uint8_t sctp_states[] = {
+	SCTP_CONNTRACK_NONE,
+	SCTP_CONNTRACK_CLOSED,
+	SCTP_CONNTRACK_COOKIE_WAIT,
+	SCTP_CONNTRACK_COOKIE_ECHOED,
+	SCTP_CONNTRACK_ESTABLISHED,
+	SCTP_CONNTRACK_SHUTDOWN_SENT,
+	SCTP_CONNTRACK_SHUTDOWN_RECD,
+	SCTP_CONNTRACK_SHUTDOWN_ACK_SENT,
+};
+
 static const struct nfct_filter_ipv4 filter_ipv4 = {
 	.addr = __constant_htonl(INADDR_LOOPBACK),
 	.mask = 0xffffffff,
-- 
2.11.4.GIT