#include <rtthread.h>
#include <lwip/sockets.h>

#define SERV_PORT 10001
#define BUF_SIZE  1024
#define BACKLOG	  5

static const char send_data[] = "This is TCP Server from RT-Thread.";

void tcpserver6(void)
{
	int sockfd, clientfd;
	struct sockaddr_in6 server_addr6, client_addr6;
	int bytes_received;
	char *recv_data;
	rt_uint32_t sin_size;
	rt_bool_t stop = RT_FALSE;

	recv_data = rt_malloc(BUF_SIZE);
	if(recv_data == RT_NULL)
	{
		rt_kprintf("No memory\n");
		return ;
	}
	
	if((sockfd = socket(PF_INET6, SOCK_STREAM, 0)) == -1)
	{
		rt_kprintf("Socket error\n");
		rt_free(recv_data);
		return ;
	}
	
	server_addr6.sin6_family = AF_INET6;
	memcpy(server_addr6.sin6_addr.s6_addr, IP6_ADDR_ANY, 16);
	server_addr6.sin6_port = htons(SERV_PORT);

	if(bind(sockfd, (struct sockaddr *)&server_addr6, sizeof(struct sockaddr)) == -1)
	{
		rt_kprintf("Bind error\n");
		rt_free(recv_data);
		return ;
	}
	if(listen(sockfd, BACKLOG) == -1)
	{
		rt_kprintf("Listen error\n");
		rt_free(recv_data);
		return ;
	}

	rt_sprintf(recv_data, "%4d", SERV_PORT);
	rt_kprintf("\nTCPServer Waiting for client on port %s...\n", recv_data);
	
	while(stop != RT_TRUE)
	{ 
		sin_size = sizeof(struct sockaddr_in6);
		clientfd = accept(sockfd, (struct sockaddr *)&client_addr6, &sin_size);
		rt_kprintf("I got a connection from (IP:%s, PORT:%d\n)", inet6_ntoa(client_addr6.sin6_addr), ntohs(client_addr6.sin6_port));
		while(1)
		{
			send(clientfd, send_data, strlen(send_data), 0);

			bytes_received = recv(clientfd, recv_data, BUF_SIZE, 0);
			if(bytes_received <= 0)
			{
				closesocket(clientfd);
				break;
			}
			recv_data[bytes_received] = '\0';
			if(strcmp(recv_data, "q") == 0 || strcmp(recv_data, "Q") == 0)
			{
				closesocket(clientfd);
				break;
			}
			else if(strcmp(recv_data, "exit") == 0)
			{
				closesocket(clientfd);
				stop = RT_TRUE;
				break;
			}
			else
			{
				rt_kprintf("RECEIVED DATA = %s\n", recv_data);
			}
		}
	}
	
	closesocket(sockfd);
	rt_free(recv_data);
	
	return ;
}
#ifdef RT_USING_FINSH
#include <finsh.h>
FINSH_FUNCTION_EXPORT(tcpserver6, start tcp server via ipv6 );
#endif