summaryrefslogtreecommitdiffstats
path: root/extensions
diff options
context:
space:
mode:
Diffstat (limited to 'extensions')
-rw-r--r--extensions/Makefile.am3
-rw-r--r--extensions/libct_proto_tcp.c25
-rw-r--r--extensions/libct_proto_udp.c24
3 files changed, 51 insertions, 1 deletions
diff --git a/extensions/Makefile.am b/extensions/Makefile.am
index ae78346..ab29a6d 100644
--- a/extensions/Makefile.am
+++ b/extensions/Makefile.am
@@ -8,7 +8,8 @@ INCLUDES=-I../include -I/lib/modules/$(shell (uname -r))/build/include
CFLAGS=-fPIC -Wall
LIBS=
-lib_LTLIBRARIES = libct_proto_tcp.la libct_proto_udp.la
+lib_LTLIBRARIES = libct_proto_tcp.la libct_proto_udp.la libct_proto_icmp.la
libct_proto_tcp_la_SOURCES = libct_proto_tcp.c
libct_proto_udp_la_SOURCES = libct_proto_udp.c
+libct_proto_icmp_la_SOURCES = libct_proto_icmp.c
diff --git a/extensions/libct_proto_tcp.c b/extensions/libct_proto_tcp.c
index 58005b0..a2243dc 100644
--- a/extensions/libct_proto_tcp.c
+++ b/extensions/libct_proto_tcp.c
@@ -54,6 +54,15 @@ static const char *states[] = {
"LISTEN"
};
+void help()
+{
+ fprintf(stdout, "--orig-port-src original source port\n");
+ 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, "--state TCP state, fe. ESTABLISHED\n");
+}
+
int parse(char c, char *argv[],
struct ip_conntrack_tuple *orig,
struct ip_conntrack_tuple *reply,
@@ -104,6 +113,20 @@ int parse(char c, char *argv[],
return 1;
}
+int final_check(unsigned int flags)
+{
+ if (!(flags & ORIG_SPORT))
+ return 0;
+ else if (!(flags & ORIG_DPORT))
+ return 0;
+ else if (!(flags & REPL_SPORT))
+ return 0;
+ else if (!(flags & REPL_DPORT))
+ return 0;
+
+ return 1;
+}
+
void print_tuple(struct ip_conntrack_tuple *t)
{
fprintf(stdout, "sport=%d dport=%d ", ntohs(t->src.u.tcp.port),
@@ -121,6 +144,8 @@ static struct ctproto_handler tcp = {
.parse = parse,
.print_tuple = print_tuple,
.print_proto = print_proto,
+ .final_check = final_check,
+ .help = help,
.opts = opts
};
diff --git a/extensions/libct_proto_udp.c b/extensions/libct_proto_udp.c
index 5675a05..8e20bd5 100644
--- a/extensions/libct_proto_udp.c
+++ b/extensions/libct_proto_udp.c
@@ -37,6 +37,14 @@ enum udp_param_flags {
REPL_DPORT = (1 << REPL_DPORT_BIT),
};
+void help()
+{
+ fprintf(stdout, "--orig-port-src original source port\n");
+ 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");
+}
+
int parse(char c, char *argv[],
struct ip_conntrack_tuple *orig,
struct ip_conntrack_tuple *reply,
@@ -72,6 +80,20 @@ int parse(char c, char *argv[],
return 1;
}
+int final_check(unsigned int flags)
+{
+ if (!(flags & ORIG_SPORT))
+ return 0;
+ else if (!(flags & ORIG_DPORT))
+ return 0;
+ else if (!(flags & REPL_SPORT))
+ return 0;
+ else if (!(flags & REPL_DPORT))
+ return 0;
+
+ return 1;
+}
+
void print_tuple(struct ip_conntrack_tuple *t)
{
fprintf(stdout, "sport=%d dport=%d ", ntohs(t->src.u.udp.port),
@@ -83,6 +105,8 @@ static struct ctproto_handler udp = {
.protonum = 17,
.parse = parse,
.print_tuple = print_tuple,
+ .final_check = final_check,
+ .help = help,
.opts = opts
};