o Use conntrack netlink attributes: Major change

o Kill action setting: Mask based dumping

1 files changed, 70 insertions, 24 deletions

diff --git a/extensions/libct_proto_udp.c b/extensions/libct_proto_udp.c
index 0088cc5..aa733f0 100644
--- a/extensions/libct_proto_udp.c
+++ b/extensions/libct_proto_udp.c
@@ -11,15 +11,17 @@
 #include <getopt.h>
 #include <stdlib.h>
 #include <netinet/in.h> /* For htons */
-#include <linux/netfilter_ipv4/ip_conntrack_tuple.h>
-#include <linux/netfilter_ipv4/ip_conntrack.h>
+#include <linux/netfilter_ipv4/ip_conntrack_netlink.h>
 #include "libct_proto.h"
+#include "libctnetlink.h"
 
 static struct option opts[] = {
 	{"orig-port-src", 1, 0, '1'},
 	{"orig-port-dst", 1, 0, '2'},
 	{"reply-port-src", 1, 0, '3'},
 	{"reply-port-dst", 1, 0, '4'},
+	{"mask-port-src", 1, 0, '5'},
+	{"mask-port-dst", 1, 0, '6'},
 	{0, 0, 0, 0}
 };
 
@@ -35,6 +37,12 @@ enum udp_param_flags {
 
 	REPL_DPORT_BIT = 3,
 	REPL_DPORT = (1 << REPL_DPORT_BIT),
+
+	MASK_SPORT_BIT = 4,
+	MASK_SPORT = (1 << MASK_SPORT_BIT),
+
+	MASK_DPORT_BIT = 5,
+	MASK_DPORT = (1 << MASK_DPORT_BIT),
 };
 
 void help()
@@ -43,67 +51,105 @@ void help()
 	fprintf(stdout, "--orig-port-dst        original destination port\n");
 	fprintf(stdout, "--reply-port-src       reply source port\n");
 	fprintf(stdout, "--reply-port-dst       reply destination port\n");
+	fprintf(stdout, "--mask-port-src	mask source port\n");
+	fprintf(stdout, "--mask-port-dst	mask destination port\n");
 }
 
-int parse(char c, char *argv[], 
-	   struct ip_conntrack_tuple *orig,
-	   struct ip_conntrack_tuple *reply,
-	   union ip_conntrack_proto *proto,
-	   unsigned int *flags)
+int parse_options(char c, char *argv[], 
+		  struct ctnl_tuple *orig,
+		  struct ctnl_tuple *reply,
+		  struct ctnl_tuple *mask,
+		  union ctnl_protoinfo *proto,
+		  unsigned int *flags)
 {
 	switch(c) {
 		case '1':
 			if (optarg) {
-				orig->src.u.udp.port = htons(atoi(optarg));
+				orig->l4src.udp.port = htons(atoi(optarg));
 				*flags |= ORIG_SPORT;
 			}
 			break;
 		case '2':
 			if (optarg) {
-				orig->dst.u.udp.port = htons(atoi(optarg));
+				orig->l4dst.udp.port = htons(atoi(optarg));
 				*flags |= ORIG_DPORT;
 			}
 			break;
 		case '3':
 			if (optarg) {
-				reply->src.u.udp.port = htons(atoi(optarg));
+				reply->l4src.udp.port = htons(atoi(optarg));
 				*flags |= REPL_SPORT;
 			}
 			break;
 		case '4':
 			if (optarg) {
-				reply->dst.u.udp.port = htons(atoi(optarg));
+				reply->l4dst.udp.port = htons(atoi(optarg));
 				*flags |= REPL_DPORT;
 			}
 			break;
+		case '5':
+			if (optarg) {
+				mask->l4src.udp.port = htons(atoi(optarg));
+				*flags |= MASK_SPORT;
+			}
+			break;
+		case '6':
+			if (optarg) {
+				mask->l4src.udp.port = htons(atoi(optarg));
+				*flags |= MASK_DPORT;
+			}
+			break;
 	}
 	return 1;
 }
 
-int final_check(unsigned int flags)
+int final_check(unsigned int flags,
+		struct ctnl_tuple *orig,
+		struct ctnl_tuple *reply)
 {
-	if ((flags & ORIG_SPORT) && (flags & ORIG_DPORT))
+	if ((flags & (ORIG_SPORT|ORIG_DPORT)) 
+	    && !(flags & (REPL_SPORT|REPL_DPORT))) {
+		reply->l4src.udp.port = orig->l4dst.udp.port;
+		reply->l4dst.udp.port = orig->l4src.udp.port;
+		return 1;
+	} else if (!(flags & (ORIG_SPORT|ORIG_DPORT))
+	            && (flags & (REPL_SPORT|REPL_DPORT))) {
+		orig->l4src.udp.port = reply->l4dst.udp.port;
+		orig->l4dst.udp.port = reply->l4src.udp.port;
 		return 1;
-	else if ((flags & REPL_SPORT) && (flags & REPL_DPORT))
+	}
+	if ((flags & (ORIG_SPORT|ORIG_DPORT)) 
+	    && ((flags & (REPL_SPORT|REPL_DPORT))))
 		return 1;
 
 	return 0;
 }
 
-void print_tuple(struct ip_conntrack_tuple *t)
+void parse_proto(struct nfattr *cda[], struct ctnl_tuple *tuple)
+{
+	if (cda[CTA_PROTO_UDP_SRC-1])
+		tuple->l4src.udp.port =
+			*(u_int16_t *)NFA_DATA(cda[CTA_PROTO_UDP_SRC-1]);
+	if (cda[CTA_PROTO_UDP_DST-1])
+		tuple->l4dst.udp.port =
+			*(u_int16_t *)NFA_DATA(cda[CTA_PROTO_UDP_DST-1]);
+}
+
+void print_proto(struct ctnl_tuple *tuple)
 {
-	fprintf(stdout, "sport=%d dport=%d ", ntohs(t->src.u.udp.port), 
-				             ntohs(t->dst.u.udp.port));
+	fprintf(stdout, "sport=%u dport=%u ", htons(tuple->l4src.udp.port),
+					      htons(tuple->l4dst.udp.port));
 }
 
 static struct ctproto_handler udp = {
-	.name 		= "udp",
-	.protonum	= 17,
-	.parse		= parse,
-	.print_tuple	= print_tuple,
-	.final_check	= final_check,
-	.help		= help,
-	.opts		= opts
+	.name 			= "udp",
+	.protonum		= 17,
+	.parse_opts		= parse_options,
+	.parse_proto		= parse_proto,
+	.print_proto		= print_proto,
+	.final_check		= final_check,
+	.help			= help,
+	.opts			= opts
 };
 
 void __attribute__ ((constructor)) init(void);