create new project.

1 files changed, 363 insertions, 0 deletions

diff --git a/src/common/flwd_network_connect.c b/src/common/flwd_network_connect.c
new file mode 100644
index 0000000..e875e87
--- /dev/null
+++ b/src/common/flwd_network_connect.c
@@ -0,0 +1,363 @@
+#include "flowood.h"
+#include "flowood_fun.h"
+#include "flwd_net.h"
+#include "MESA_htable.h"
+#include "MESA_list_queue.h"
+#include "MESA_handle_logger.h"
+#include "MESA_list_count.h"
+#include <assert.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <pthread.h>
+#include <string.h>
+#include <assert.h>
+#include <arpa/inet.h>
+#include <netinet/tcp.h>
+#include <netinet/udp.h>
+
+
+typedef struct{
+	char region_str[FLWD_REGION_STR_LEN_MAX];
+	unsigned int vlan_id;
+	unsigned int conn_forward_gateway_array[FLWD_NETWORK_NxM_MAX]; /* ����������ת������IP, ������ */
+	int conn_forward_gateway_array_num;
+}flwd_network_gdev_data_t;
+
+/*  
+	NOTE:
+		GDEV��FWD֮����NxM��ϵ, ͨ�������ñ�, ��֮access����, ����һ��ѡ���˵Ļ�ԾIP, Ӧ�ý����ĸ�FWD����ͨ���ĸ�GDEV����ȥ.
+
+		active_IP:  access---->forwad---->gdev---->INTERNET.
+
+
+	config:
+		region	vlan		gdev_redirect_ip	forward_manage_ip	forward_feedback_ip
+
+
+	region: �������, ��ʡ����;
+	vlan	 : ��ֹһ���ֵ��ڵ�GDEV��froward���ܲ�ͨ, ������������񡢿�VLAN�ȵ�, ���������ʶ;
+	gdev_redirect_ip: GDEV�Ļ�����IP
+	forward_manage_ip: ת�����ع�����IP
+	forward_feedback_ip: ת�����ص������ݻش���IP
+
+	������ȫ��һ��, ��access�����зֶ˿ڵ�����, Ҫȫ��ͬ��, ���������������޸�!!!!
+*/
+
+
+
+
+
+static int flwd_network_gdev_key_cmp(const uchar * key1, uint size1, const uchar * key2, uint size2)
+{
+	if(size1 != size2){
+		return -1;
+	}
+
+	return memcmp(key1, key2, size1);
+}
+
+static uint flwd_network_gdev_key2index(const MESA_htable_handle table, const uchar * key, uint size)
+{
+	unsigned int hash = 131;
+	unsigned seed = 13131;
+	unsigned int i;
+
+	for(i = 0; i < size; i++){
+		hash = hash * seed + *key++;
+	}
+	
+	return hash;
+}
+
+static void flwd_network_gdev_data_free(void *data)
+{
+	assert(0); /* ��������ʱ��Զ����free, �Ͳ����ߵ�����, ����!!! */
+}
+
+/*
+	NOTE:
+		��htable��gdev_ip��Ϊkey,
+		flwd_network_gdev_data_t�ṹΪdata, 
+		��������ȥ���ĸ�gdev, ����ͨ����Щforward.
+*/
+static MESA_htable_handle flwd_create_network_gdev_table(void)
+{
+	int opt_int;
+	MESA_htable_handle htable;
+
+	htable = MESA_htable_born();
+	assert(htable != NULL);
+
+	opt_int = 0; /* ��ʼ�����ٸĶ�, ����ģʽ */
+	MESA_htable_set_opt(htable, MHO_THREAD_SAFE, &opt_int, sizeof(int));
+
+	opt_int = 1024;
+	MESA_htable_set_opt(htable, MHO_HASH_SLOT_SIZE, &opt_int, sizeof(int));
+
+	opt_int = 1000;
+	MESA_htable_set_opt(htable, MHO_HASH_MAX_ELEMENT_NUM, &opt_int, sizeof(int));
+
+	opt_int = 0;
+	MESA_htable_set_opt(htable, MHO_EXPIRE_TIME, &opt_int, sizeof(int)); /* ����̭ */
+
+	opt_int = HASH_ELIMINATE_ALGO_FIFO; /* ���߳�����ģʽ����FIFO */
+	MESA_htable_set_opt(htable, MHO_ELIMIMINATE_TYPE, &opt_int, sizeof(int));
+
+	MESA_htable_set_opt(htable, MHO_CBFUN_KEY_COMPARE, (void *)&flwd_network_gdev_key_cmp, sizeof(void *));
+
+	MESA_htable_set_opt(htable, MHO_CBFUN_KEY_TO_INDEX, (void *)&flwd_network_gdev_key2index, sizeof(void *));
+
+	MESA_htable_set_opt(htable, MHO_CBFUN_DATA_FREE, (void *)&flwd_network_gdev_data_free, sizeof(void *));
+	
+	opt_int = 0;
+	MESA_htable_set_opt(htable, MHO_AUTO_UPDATE_TIME, &opt_int, sizeof(int));
+
+	opt_int = 0;
+	MESA_htable_set_opt(htable, MHO_SCREEN_PRINT_CTRL, &opt_int, sizeof(int));
+
+	opt_int = 100;
+	MESA_htable_set_opt(htable, MHO_HASH_LIST_COLLIDE_THRESHOLD, &opt_int, sizeof(int));	
+
+	char *err_log = (char *)"./flwd_hash_collide.log";
+	MESA_htable_set_opt(htable, MHO_HASH_LOG_FILE, (void *)err_log, strlen(err_log));	
+	
+	int ret = MESA_htable_mature(htable);
+	assert(ret >= 0);
+
+	return htable;
+}
+
+/*  
+	pattern:
+		#region	vlan_id		gdev_redirect_ip	
+*/
+static int flwd_network_gdev_cfg_parse(void)
+{
+	FILE *fp;
+	char __raw_line_buf[1024];
+	char line_buf[1024];
+	const char *delim = "\t ";
+	char *save_ptr;
+	flwd_network_gdev_data_t *tmp_data;
+	char *section, *region_str, *gdev_ip_str;
+	unsigned int hkey;
+	int ret;
+
+	fp = fopen("./conf/ip_reuse/network_gdev.cfg", "r");
+	if(NULL == fp){
+		return -1;
+	}
+
+	while(fgets(__raw_line_buf, 1024, fp)){
+		if('#' ==__raw_line_buf[0]){
+			continue;
+		}
+		memcpy(line_buf, __raw_line_buf, 1024);
+		flwd_del_last_rn(line_buf, 1024);
+		tmp_data = (flwd_network_gdev_data_t *)calloc(1, sizeof(flwd_network_gdev_data_t));
+
+		/* region */
+		section = strtok_r(line_buf, delim, &save_ptr);
+		if(NULL == section){
+			return -1;
+		}
+		strncpy(tmp_data->region_str, section, FLWD_REGION_STR_LEN_MAX-1);
+		region_str = section;
+
+		/* vlan_id */
+		section = strtok_r(NULL, delim, &save_ptr);
+		if(NULL == section){
+			return -1;
+		}
+		tmp_data->vlan_id = (unsigned int)atoi(section);
+
+		/* gdev_redirect_ip */
+		section = strtok_r(NULL, delim, &save_ptr);
+		if(NULL == section){
+			return -1;
+		}
+		if(inet_pton(AF_INET, section, &hkey) <= 0){
+			return -1;
+		}
+		gdev_ip_str = section;
+
+		ret = MESA_htable_add(flwd_global_val.flwd_network_conn_table, (const uchar *)&hkey, sizeof(int), tmp_data);
+		if(ret >= 0){
+			flwd_log(10, "parse network_gdev.cfg, %s\t%u\t%s\n", region_str, tmp_data->vlan_id, gdev_ip_str);
+		}else{
+			flwd_log(30, "parse network_gdev.cfg error, %s\n", __raw_line_buf);
+		}
+
+		while(strtok_r(NULL, delim, &save_ptr));
+	}
+
+	fclose(fp);
+
+	return 0;
+}
+
+
+static  int flwd_network_gdev_fwd_relate_cb(const uchar * key, uint size, void * data, void * user)
+{
+	flwd_network_gdev_data_t *hdata = (flwd_network_gdev_data_t *)data;
+	flwd_network_gdev_data_t *tmp_fwd_data = (flwd_network_gdev_data_t *)user;
+
+	if(strlen(hdata->region_str) != strlen(tmp_fwd_data->region_str)){
+		return ITERATE_CB_RET_CONTINUE_FLAG; /* MESA_htable_iterate_bytime�޷����ػص�������ֵ, �����ø�trick, ��conn_forward_gateway_array_num=2�����ҵ���, ������1 */
+	}
+	
+	if(strncmp(hdata->region_str, tmp_fwd_data->region_str,strlen(hdata->region_str)) != 0){
+		return ITERATE_CB_RET_CONTINUE_FLAG;
+	}
+
+	if(hdata->vlan_id != tmp_fwd_data->vlan_id){
+		return ITERATE_CB_RET_CONTINUE_FLAG;
+	}
+
+	tmp_fwd_data->conn_forward_gateway_array_num = 2; /* bingo! */
+
+	if(hdata->conn_forward_gateway_array_num >= FLWD_NETWORK_NxM_MAX){
+		assert(0);
+	}
+
+	hdata->conn_forward_gateway_array[hdata->conn_forward_gateway_array_num] = tmp_fwd_data->conn_forward_gateway_array[0];
+	hdata->conn_forward_gateway_array_num++;
+
+	/* NOTE: �м�!!!  �����ҵ�����һ�������е�GDEV, �˴�Ҳ�����ж�callback, ��Ϊһ�������ڻ�����������������GDEV, Ҫȫ����һ�� */
+	return ITERATE_CB_RET_CONTINUE_FLAG;
+}
+
+static void flwd_network_gdev_fwd_relate(flwd_network_gdev_data_t *tmp_data)
+{
+	/* �������� */
+	MESA_htable_iterate_bytime(flwd_global_val.flwd_network_conn_table, ITERATE_TYPE_OLDEST_FIRST, 
+							 flwd_network_gdev_fwd_relate_cb, tmp_data);
+
+	return ;
+}
+
+
+/*
+	#region	vlan_id		forward_ip	
+*/
+static int flwd_network_forward_cfg_parse(void)
+{
+	FILE *fp;
+	char line_buf[1024];
+	const char *delim = "\t ";
+	char *save_ptr;
+	flwd_network_gdev_data_t tmp_data;
+	char *section;
+	char *forward_ip;
+
+	fp = fopen("./conf/ip_reuse/network_forward.cfg", "r");
+	if(NULL == fp){
+		return -1;
+	}
+
+	while(fgets(line_buf, 1024, fp)){
+		if('#' ==line_buf[0]){
+			continue;
+		}
+		flwd_del_last_rn(line_buf, 1024);
+		memset(&tmp_data, 0, sizeof(flwd_network_gdev_data_t));
+
+		section = strtok_r(line_buf, delim, &save_ptr);
+		if(NULL == section){
+			return -1;
+		}
+		strncpy(tmp_data.region_str, section, FLWD_REGION_STR_LEN_MAX-1);
+
+		section = strtok_r(NULL, delim, &save_ptr);
+		if(NULL == section){
+			return -1;
+		}
+		tmp_data.vlan_id = (unsigned int)atoi(section);
+
+		section = strtok_r(NULL, delim, &save_ptr);
+		if(NULL == section){
+			return -1;
+		}
+		if(inet_pton(AF_INET, section, &tmp_data.conn_forward_gateway_array[0]) <= 0){ /* ����һ������0, ��ʱ�洢 forward_ip */
+			return -1;
+		}
+		forward_ip = section;
+		tmp_data.conn_forward_gateway_array_num = 1;
+
+		flwd_network_gdev_fwd_relate(&tmp_data);
+		/* MESA_htable_iterate_bytime�޷����ػص�������ֵ, �����ø�trick, ��conn_forward_gateway_array_num=2�����ҵ���, ������1 */
+		if(2 != tmp_data.conn_forward_gateway_array_num){
+			flwd_log(30, "can't found related gdev ip in network_gdev.cfg, network_forward.cfg->forward ip:%s\n", forward_ip);
+			return -1;
+		}else{
+			flwd_log(10, "parse network_gdev.cfg->%s\n", forward_ip);
+		}
+
+		while(strtok_r(NULL, delim, &save_ptr));
+	}
+
+	fclose(fp);
+
+	return 0;
+}
+
+/* TODO, 
+
+   ///����ѡ��Ļ�ԾIP���ڵ�GDEV_IP��ID, �������, ȷ��Ӧ�ô��ĸ�FWD����
+
+   ��һ���ֵ���˵, �������ظ�����һ��ת������, ����ͨ�����е�INLINE-DEVICE, 
+
+   �����IP��ַ������������.
+  
+*/
+unsigned int flwd_search_fwd_ip_by_gdev_ip(unsigned int gdev_ip_net_order)
+{
+	flwd_network_gdev_data_t *hdata;
+	char ip_str[16];
+	int rnd_num;
+
+	hdata = (flwd_network_gdev_data_t *)MESA_htable_search(flwd_global_val.flwd_network_conn_table, (unsigned char *)&gdev_ip_net_order, sizeof(int));
+	if(NULL == hdata){
+		inet_ntop(AF_INET, &gdev_ip_net_order, ip_str, 16);
+		flwd_log(30, "not found gdev ip %s in network_gdev.cfg\n", ip_str);
+		return -1;
+	}
+
+	/* NxM���˹�ϵ, ��������, �����ѡһ��ת������ */
+	rnd_num = rand() % hdata->conn_forward_gateway_array_num;
+
+	return hdata->conn_forward_gateway_array[rnd_num];
+	/*
+		TODO:
+		дһ��gdev_ip��FWD���ݿ�IP�ľ�̬������,
+
+		���߿���һ��rip·��Э��, �Զ��㲥, �Զ�����.
+
+		����ʱ���ֶ�д��, ʵ�ʾ���10.0.6.201��em1����IP��ַ: 172.16.1.201.
+	*/
+
+	//inet_pton(AF_INET, "10.0.6.201", &gdev_ip_net_order);
+	//inet_pton(AF_INET, "172.16.1.201", &gdev_ip_net_order);
+	//inet_pton(AF_INET, "47.74.128.220", &gdev_ip_net_order);
+	//return gdev_ip_net_order;
+}
+
+int flwd_network_conn_init(void)
+{
+	flwd_global_val.flwd_network_conn_table = flwd_create_network_gdev_table();
+
+	/* NOTE, �����ȵ���flwd_network_gdev_cfg_parse, ���ӵ�htable��, �����ٵ���flwd_network_forward_cfg_parse */
+	if(flwd_network_gdev_cfg_parse() < 0){
+		flwd_log(30, "flwd_network_gdev_cfg_parse error!\n");
+		return -1;
+	}
+
+	/* NOTE, �����ȵ���flwd_network_gdev_cfg_parse, ���ӵ�htable��, �����ٵ���flwd_network_forward_cfg_parse */
+	if(flwd_network_forward_cfg_parse() < 0){
+		flwd_log(30, "flwd_network_forward_cfg_parse error!\n");
+		return -1;
+	}
+	return 0;
+}
+
+