path: root/src/common/flwd_network_connect.c

#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;
}