/** *************************************************************************
 *  @file          main_l2switch.c
 *  @brief	  基于FAST架构的软件二层交换示例程序
 * 
 *  详细说明
 * 
 *  @date	   2017/04/08 10:39:17 星期六
 *  @author		XDL(Copyright  2017  XuDongLai)
 *  @email		<XuDongLai0923@163.com>
 *  @version	0.2.0
 ****************************************************************************/
/*
 * main_l2switch.c
 *
 * Copyright (C) 2017 - XuDongLai
 *
 * 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 2 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/>.
 */
#include <fast.h>
#include <unistd.h>


#define NM08_PORT_CNT   4	   /*设备端口数量*/
#define NM08_NEIGH_MAX  128	   /*每个端口上最大存储邻居MAC个数*/
#define MAC_LEN 6				/*MAC地址长度*/
#define MAC_LIFE_TIME 10	   /*MAC地址生命时长为300秒，可调整*/

int debug = 0;
#define xprintf(argc...) if(debug)printf(argc);
/*端口计数*/
struct nm08_port_stats
{
	u64 recv_pkts;
	u64 recv_bytes;
	u64 send_pkts;
	u64 send_bytes;
};/*32B*/

/*MAC地址，有效位标记和MAC地址学习或更新时间*/

struct nm08_port_mac
{
	u8  port;	 /*地址所在端口号信息*/
	u8  valid;	 /*地址有效位标记*/
	u8  addr[MAC_LEN];  /*存储MAC地址*/	
	struct timeval tv;  /*存储此MAC的开始有效时间*/
};/*24B*/

/*08的邻居信息表*/
struct nm08_neigh_table
{
	struct nm08_port_stats port[NM08_PORT_CNT];
	struct nm08_port_mac mac[NM08_NEIGH_MAX];
};

struct nm08_neigh_table *nm08_table = NULL;

/**
* @brief 
*
* @param addr1
* @param addr2
*
* @return 
*/
int ether_addr_equal(u8 *addr1,u8 *addr2)
{
	u16 *a = (u16 *)addr1;
	u16 *b = (u16 *)addr2;
	
	return ((a[0] ^ b[0]) | (a[1] ^ b[1]) | (a[2] ^ b[2])) != 0; 
}


/**
* @brief 
*
* @param inport
* @param index
*/
/*更新MAC地址对应的有效时间*/
void update_mac_time(u8 inport,u8 index)
{
	struct timeval tv_now;
	gettimeofday(&tv_now,NULL);
	tv_now.tv_sec += MAC_LIFE_TIME;	   /*MAC地址生命周期设置，每次更新，此MAC地址有效时长为300秒，用户可调整*/
	nm08_table->mac[index].tv = tv_now;/*更新当前时间*/
	if(nm08_table->mac[index].port != inport)
	{
		/*机器换了端口*/
	}
	nm08_table->mac[index].port = inport;/*更新MAC所在端口号，有可能是从另外端口拔出，插到了新端口上，如此可以快速更新*/
	xprintf("update_mac_time->port:%d,index:%d\n",inport,index);
}

/**
* @brief 
*
* @param inport
* @param src_mac
*/
/*地址学习过程，将报文的源MAC学习到对应端口MAC表中*/
void learn_smac(u8 inport,u8 *src_mac)
{
	/*更新之前查找空白存储MAC位置*/
	int i = 0,j = -1;

	xprintf("learn_smac->\n");
		
	for(i= 0;i<NM08_NEIGH_MAX;i++)
	{
		if (nm08_table->mac[i].valid == 1)/*MAC地址有效*/
		{
			if(!ether_addr_equal(src_mac,&nm08_table->mac[i].addr[0]))
			{
				/*当前存储空间地址与输入学习的地址相同，则更新此MAC地址的有效时间*/
				xprintf("learn_smac->Update TIME!\n");
				update_mac_time(inport,i);
				return;
			}
		}
		else if(j == -1)/*第一次找到有效可存储MAC地址的索引*/
		{
			j = i;/*记录下当前可用的有效位置索引*/
		}
	}
	/*此端口已经存储的MAC地址表中找不到需要学习的MAC地址，则取出最后一个空白地址*/
	if(j == -1)/*有效地址没有空白可用*/
	{
		xprintf("learn_smac->Can't learning more!\n");
		return;
	}
	/*有空白位置可以存储此新学习的MAC*/
	memcpy(&nm08_table->mac[j].addr[0],src_mac,MAC_LEN);
	update_mac_time(inport,j);/*更新此MAC地址的有效时间*/
	nm08_table->mac[j].valid = 1;/*将MAC地址信息置为有效状态*/
	xprintf("learn_smac->add new MAC,port:%d,index:%d\n",inport,j);
}

/**
* @brief 
*
* @param inport
* @param dst_mac
*
* @return 
*/
int find_dmac(u8 inport,u8 *dst_mac)
{
	int ret = -1,i = 0;
	
	
	for(;i< NM08_NEIGH_MAX;i++)/*i != inport 表示不在此MAC的输入端口表中查找*/
	{
		/*判断地址是否有效，MAC地址是否相等*/
		if (nm08_table->mac[i].valid == 1 &&
		    nm08_table->mac[i].port != inport &&
		    !ether_addr_equal(dst_mac,&nm08_table->mac[i].addr[0]))
		{
			ret = nm08_table->mac[i].port;/*如果有效且相等，则返回对应的端口号（即输出端口号）*/
			break;
		}

	}
	xprintf("find_dmac->ret = %d\n",ret);
	return ret;/*返回-1表示没有学习到，则需要泛洪*/
}

/**
* @brief 
*
* @param pkt
* @param pkt_len
*/
void pkt_send_normal(struct fast_packet *pkt,int pkt_len)
{
	xprintf("pkt_send_normal->%p,outport:%d,len:%d\n",pkt,pkt->um.outport,pkt_len);
	pkt->um.pktsrc = 1;/*报文来源为CPU输入*/
	pkt->um.pktdst = 0;/*报文目的为硬件输出*/
	pkt->um.dstmid = 5;/*直接从硬件GOE模块输出，不走解析、查表等模块*/
	fast_ua_send(pkt,pkt_len);
}

/**
* @brief 
*
* @param pkt
* @param pkt_len
*/
void pkt_send_flood(struct fast_packet *pkt,int pkt_len)
{
	int i = 0,inport = pkt->um.inport;/*保存输入端口*/
	xprintf("------pkt_send_flood------\n");
	for(;i< NM08_PORT_CNT;i++)/*除输入端口外，其他都发送一份*/
	{
		if(i != inport)
		{
			pkt->um.outport = i;
			pkt_send_normal(pkt,pkt_len);
		}
	}
}

/**
* @brief 
*/
void nm08_show_mac_info(void)
{
	int i = 0,j = 0,max_cnt = 0;
	char buf[400];
	int port_mac_cnt[NM08_PORT_CNT] = {0};
	struct nm08_port_mac mac[NM08_PORT_CNT][NM08_NEIGH_MAX] = {{0}};

	for(;i<NM08_NEIGH_MAX;i++)
	{
		if (nm08_table->mac[i].valid == 1)
		{
			mac[nm08_table->mac[i].port][port_mac_cnt[nm08_table->mac[i].port]] = nm08_table->mac[i];
			port_mac_cnt[nm08_table->mac[i].port]++;
		}
	}
	/*记录一个端口最多的MAC地址数量，确定我们最大需要输出几行*/
	max_cnt = port_mac_cnt[0];
	for(i=1;i<NM08_PORT_CNT;i++)
	{
		if(port_mac_cnt[i] > max_cnt)
			max_cnt = port_mac_cnt[i];
	}
	max_cnt++;/*最后多打印一行空行*/
	xprintf("\nID               PORT0              PORT1              PORT2              PORT3\n");

	for(j = 0;j<max_cnt;j++)
	{			
		sprintf(buf,"%2d ",j);
		for(i=0;i< NM08_PORT_CNT;i++)
		{
			
			if (mac[i][j].valid == 1)
			{
				sprintf(buf,"%s  %02X:%02X:%02X:%02X:%02X:%02X",buf,
				        mac[i][j].addr[0],
				        mac[i][j].addr[1],
				        mac[i][j].addr[2],
				        mac[i][j].addr[3],
				        mac[i][j].addr[4],
				        mac[i][j].addr[5]);
			}
			else
			{
				sprintf(buf,"%s                  .",buf);
			}
		}
		xprintf("%s\n",buf);
	}
}

/**
* @brief 
*
* @param argv
*
* @return 
*/
/*MAC地址老化处理线程*/
void *nm08_mac_aging(void *argv)
{
	struct timeval tv_now;
	int i = 0,k = 0,aging_cnt = 0;

	while(1)
	{
		gettimeofday(&tv_now,NULL);

		for(i = 0;i<NM08_NEIGH_MAX;i++)
		{
			/*判断地址是否有效*/
			if (nm08_table->mac[i].valid == 1)
			{
				/*此处仅用秒来做比较，如果需要更高精度可将微秒加入再比较*/
				if(nm08_table->mac[i].tv.tv_sec < tv_now.tv_sec)
				{
					/*此MAC地址的有效时间已到*/
					nm08_table->mac[i].valid = 0;
					aging_cnt++;
				}
			}
		}
		
		xprintf("aging[%d]->invalid mac:%d\n",k++,aging_cnt);
		aging_cnt = 0;/*统计每次老化MAC地址个数，输出后归零*/
		sleep(5);/*老化时间误差，每10秒才判断一次。如果提高精度可缩短时间*/
		nm08_show_mac_info();
	}
}

/**
* @brief 
*/
void nm08_start_aging(void)
{	
	pthread_t tid;

	/*创建地址老化处理线程*/
	if(pthread_create(&tid, NULL, nm08_mac_aging, NULL))
	{
		xprintf("Create nm08_mac_aging thread error!\n");
		exit(0);
	}
	else
	{
		xprintf("Create nm08_mac_aging thread OK!\n");			
	}
}

/**
* @brief 
*
* @param pkt
* @param pkt_len
*
* @return 
*/
int callback(struct fast_packet *pkt,int pkt_len)
{
	int outport = -1;

	xprintf("inport:%d,dstmid:%d,len:%d,dmac:%02X:%02X:%02X:%02X:%02X:%02X,smac:%02X:%02X:%02X:%02X:%02X:%02X\n",pkt->um.inport,pkt->um.dstmid,pkt_len,
	       pkt->data[0],pkt->data[1],pkt->data[2],pkt->data[3],pkt->data[4],pkt->data[5],
	       pkt->data[6],pkt->data[7],pkt->data[8],pkt->data[9],pkt->data[10],pkt->data[11]);

	//print_pkt(pkt,pkt_len);
	//pkt->um.outport = 1;
	//pkt_send_normal(pkt,pkt_len);/*正常发送报文*/
	//return 0;
	/*MAC地址学习过程*/
	learn_smac(pkt->um.inport,&pkt->data[MAC_LEN]);/*用源MAC位置开始学习*/

	/*查表过程*/
	outport = find_dmac(pkt->um.inport,pkt->data);/*用目的MAC地址开始查表*/

	/*发送报文*/
	if(outport == -1)/*报文需要泛洪操作*/
	{
		pkt_send_flood(pkt,pkt_len);/*泛洪发送，保留输入端口不变调用*/
	}
	else/*正常转发*/
	{
		pkt->um.outport = outport;/*修改报文输出端口号*/
		pkt_send_normal(pkt,pkt_len);/*正常发送报文*/
	}
	return 0;
}

/**
* @brief 
*/
void ua_init(u8 mid)
{
	int ret = 0;
	/*向系统注册，自己进程处理报文模块ID为1的所有报文*/
	if((ret=fast_ua_init(mid,callback)))//UA模块实例化(输入参数1:接收模块ID号,输入参数2:接收报文的回调处理函数)
	{
		perror("fast_ua_init!\n");
		exit (ret);//如果初始化失败,则需要打印失败信息,并将程序结束退出!
	}
}

/**
* @brief 
*
* @param argc
* @param argv[]
*
* @return 
*/
int main(int argc,char* argv[])
{
	u8 mid = 129;
	debug = 1;
	if(argc == 2)
	{
		debug = atoi(argv[1]);
		
	}
	else if(argc == 3)
	{
		debug = atoi(argv[1]);
		mid = atoi(argv[2]);
	}

	if(mid <129)
	{
		printf("Usage:\n\t%s debug mid\n",argv[0]);
		printf("\tdeubg[0,1],UA mid rang [129,255]!\n");
		exit(0);
	}
	/*申请地址表存储空间*/
	nm08_table = (struct nm08_neigh_table *)malloc(sizeof(struct nm08_neigh_table));
	/*空间清零*/
	memset(nm08_table,0,sizeof(struct nm08_neigh_table));

	/*UA模块初始化	*/
	ua_init(mid);
	
	/*配置硬件规则，将硬件所有报文送到模块ID为1的进程处理*/
	fast_reg_wr(FAST_ACTION_REG_ADDR|FAST_DEFAULT_RULE_ADDR,ACTION_SET_MID<<28|mid);

	/*启动地址学习表老化线程*/
	nm08_start_aging();
	
	/*启动线程接收分派给UA进程的报文*/
	fast_ua_recv();
	
	/*主进程进入循环休眠中,数据处理主要在回调函数*/
	while(1){sleep(9999);}
	return 0;
}