Implement space saving algorithm

Author: gogapp

Makefile

@@ -44,7 +44,7 @@ SRCS-y += sol/main.c
 # Libraries.
 SRCS-y += lib/mailbox.c lib/net.c lib/flow.c lib/ipip.c \
 	lib/luajit-ffi-cdata.c lib/launch.c lib/lpm.c lib/acl.c lib/varip.c \
-	lib/l2.c
+	lib/l2.c lib/space_saving.c
 
 LDLIBS += $(LDIR) -Bstatic -lluajit-5.1 -Bdynamic -lm -lmnl
 CFLAGS += $(WERROR_FLAGS) -I${GATEKEEPER}/include -I/usr/local/include/luajit-2.0/

include/space_saving.h

@@ -0,0 +1,119 @@
+/*
+ * Gatekeeper - DoS protection system.
+ * Copyright (C) 2016 Digirati LTDA.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef _SPACE_SAVING_H_
+#define _SPACE_SAVING_H_
+
+#define OVRFLOW 1000000
+
+#include <netinet/in.h>
+
+#include <rte_ether.h>
+#include <rte_hash.h>
+#include <rte_jhash.h>
+
+#include "list.h"
+//#include "gatekeeper_gk.h"
+//#include "gatekeeper_net.h"
+//#include "gatekeeper_flow.h"
+
+#define DEFAULT_HASH_FUNC rte_jhash
+
+extern struct ip_key *hh_table;
+extern struct list_head bkt_head_ip4;
+extern struct list_head bkt_head_ip6;
+extern int streamlen;
+extern int mx;
+
+struct ip_key {
+	uint16_t proto;
+	union {
+		struct v4{
+			struct in_addr src;
+			struct in_addr dst;
+		} v4;
+
+		struct v6{
+			struct in6_addr src;
+			struct in6_addr dst;
+		} v6; 
+	} k;	
+};
+
+/* Data structure for Counter bucket. */
+struct counter_bucket
+{
+	uint16_t proto;
+	int bkt_id;
+
+	union {
+		/* Bucket for IPV4 address. */
+		struct rte_hash *bkt_ip4;
+
+		/* Bucket for IPV6 address. */
+		struct rte_hash *bkt_ip6;
+	} bkt;
+
+	struct list_head list;
+};
+
+/* Data structure of IP data. */
+struct ip_data
+{
+	int err;
+	int bkt_id;
+	struct ip_key key;
+	struct counter_bucket ct_bucket;
+};
+
+/* 
+ * Create a counter table of size = 1.0/epsion.
+ * @epsilon is the error parameter for space saving algorithm.
+ */
+struct rte_hash *
+create_counter_table(unsigned int socket_id, uint16_t proto, int counter_id,
+	int ht_size);
+
+/* Destroy a counter table. */
+void destroy_counter_table(uint16_t proto, int counter_id);
+
+/* 
+ * Create a counter bucket.
+ * Size of each bucket is set to 100 by default.
+ * TODO: Find a way to vary the size of a bucket to ensure 
+ *		 optimum memory usage.
+ */
+struct rte_hash *
+create_bucket(unsigned int socket_id, uint16_t proto, int bkt_id);
+
+/* Increment Counter Algorithm. */
+static int
+increment_counter(unsigned int socket_id, uint16_t proto, struct ip_data **element);
+
+/* Space Saving algorithm. */
+//static
+int space_saving(unsigned int socket_id, uint16_t proto, struct ip_key *key,
+	struct rte_hash *ct_table);
+
+int SSiterate(struct rte_hash *ct_table, int proto, int threshold);
+
+int SSEstLow(struct rte_hash *ct_table, struct ip_key *key);
+
+int SSEstUpp(struct rte_hash *ct_table, struct ip_key *key);
+ 
+#endif /* _SPACE_SAVING_H */