add network test code.

git-svn-id: https://rt-thread.googlecode.com/svn/trunk@1518 bbd45198-f89e-11dd-88c7-29a3b14d5316

examples/test/net_test.c

+/*
+ * Net Test Utilities for RT-Thread
+ */
+#include <rtthread.h>
+#include <finsh.h>
+#include <lwip/api.h>
+#include <lwip/sockets.h>
+/* 
+ * UDP echo server
+ */
+#define UDP_ECHO_PORT   7
+rt_thread_t udpecho_tid = RT_NULL;
+void udpecho_entry(void *parameter)
+{
+	struct netconn *conn;
+	struct netbuf *buf;
+	struct ip_addr *addr;
+	unsigned short port;
+
+	conn = netconn_new(NETCONN_UDP);
+	netconn_bind(conn, IP_ADDR_ANY, 7);
+
+	while(1)
+	{
+        /* received data to buffer */
+		buf = netconn_recv(conn);
+
+		addr = netbuf_fromaddr(buf);
+		port = netbuf_fromport(buf);
+
+        /* send the data to buffer */
+		netconn_connect(conn, addr, port);
+
+		/* reset address, and send to client */
+		buf->addr = RT_NULL;
+		netconn_send(conn, buf);
+
+        /* release buffer */
+		netbuf_delete(buf);
+	}
+}
+/*
+ * UDP socket echo server
+ */
+#define UDP_SOCKET_ECHO_PORT	700
+#define UDP_SOCKET_BUFFER_SIZE	4096
+rt_thread_t udpecho_socket_tid = RT_NULL;
+void udpecho_socket_entry(void *parameter)
+{
+	int sock;
+	int bytes_read;
+	char *recv_data;
+	rt_uint32_t addr_len;
+	struct sockaddr_in server_addr, client_addr;
+
+	/* allocate the data buffer */
+	recv_data = rt_malloc(UDP_SOCKET_BUFFER_SIZE);
+	if (recv_data == RT_NULL)
+	{
+		/* no memory yet */
+		rt_kprintf("no memory\n");
+		goto _exit;
+	}
+	/* create a UDP socket */
+	if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
+	{
+		rt_kprintf("create socket error\n");
+		goto _exit;
+	}
+
+	/* initialize server address */
+	server_addr.sin_family = AF_INET;
+	server_addr.sin_port = htons(UDP_SOCKET_ECHO_PORT);
+	server_addr.sin_addr.s_addr = INADDR_ANY;
+	rt_memset(&(server_addr.sin_zero),0, sizeof(server_addr.sin_zero));
+
+	/* bind socket to server address */
+	if (bind(sock,(struct sockaddr *)&server_addr,
+			sizeof(struct sockaddr)) == -1)
+	{
+		/* bind failed */
+		rt_kprintf("bind error\n");
+		goto _exit;
+	}
+
+	addr_len = sizeof(struct sockaddr);
+	while (1)
+	{
+		/* try to receive from UDP socket */
+		bytes_read = recvfrom(sock, recv_data, UDP_SOCKET_BUFFER_SIZE, 0,
+							 (struct sockaddr *)&client_addr, &addr_len);
+
+		/* send back */
+		sendto(sock, recv_data, bytes_read, 0,
+			(struct sockaddr *)&client_addr, addr_len);
+	}
+
+_exit:
+	rt_free(recv_data);
+	return;
+}
+
+/*
+ * TCP echo server
+ */
+#define TCP_ECHO_PORT   7
+rt_thread_t tcpecho_tid = RT_NULL;
+void tcpecho_entry(void *parameter)
+{
+	struct netconn *conn, *newconn;
+	err_t err;
+
+	/* Create a new connection identifier. */
+	conn = netconn_new(NETCONN_TCP);
+
+	/* Bind connection to well known port number 7. */
+	netconn_bind(conn, NULL, TCP_ECHO_PORT);
+
+	/* Tell connection to go into listening mode. */
+	netconn_listen(conn);
+
+	while(1)
+	{
+		/* Grab new connection. */
+		newconn = netconn_accept(conn);
+		/* Process the new connection. */
+		if(newconn != NULL)
+		{
+			struct netbuf *buf;
+			void *data;
+			u16_t len;
+
+			while((buf = netconn_recv(newconn)) != NULL)
+			{
+				do
+				{
+					netbuf_data(buf, &data, &len);
+					err = netconn_write(newconn, data, len, NETCONN_COPY);
+					if(err != ERR_OK){}
+				}
+				while(netbuf_next(buf) >= 0);
+				netbuf_delete(buf);
+			}
+			/* Close connection and discard connection identifier. */
+			netconn_delete(newconn);
+		}
+	}
+}
+
+/*
+ * TCP socket echo server
+ */
+#define TCP_SOCKET_ECHO_PORT	700
+#define TCP_SOCKET_BUFFER_SIZE	4096
+rt_thread_t tcpecho_socket_tid = RT_NULL;
+void tcpecho_socket_entry(void *parameter)
+{
+   char *recv_data;
+   rt_uint32_t sin_size;
+   int sock = -1, connected, bytes_received;
+   struct sockaddr_in server_addr, client_addr;
+
+	recv_data = rt_malloc(TCP_SOCKET_BUFFER_SIZE);
+	if (recv_data == RT_NULL)
+	{
+		rt_kprintf("no memory\n");
+		goto _exit;
+	}
+
+	/* create a TCP socket */
+	if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1)
+	{
+		rt_kprintf("create socket error\n");
+		goto _exit;
+	}
+
+	/* initialize server address */
+	server_addr.sin_family = AF_INET;
+	server_addr.sin_port = htons(TCP_SOCKET_ECHO_PORT);
+	server_addr.sin_addr.s_addr = INADDR_ANY;
+	rt_memset(&(server_addr.sin_zero),8, sizeof(server_addr.sin_zero));
+
+	/* bind to server address */
+	if (bind(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
+	{
+		rt_kprintf("bind address failed\n");
+		goto _exit;
+   }
+
+	/* listen */
+	if (listen(sock, 5) == -1)
+	{
+		rt_kprintf("listen error\n");
+		goto _exit;
+	}
+
+	sin_size = sizeof(struct sockaddr_in);
+	while(1)
+	{
+		/* accept client connected */
+		connected = accept(sock, (struct sockaddr *)&client_addr, &sin_size);
+		if (connected > 0)
+		{
+			int timeout;
+			
+			/* set timeout option */
+			timeout = 5000; /* 5second */
+			setsockopt(connected, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(timeout));
+
+			/* handle this client */
+			while (1)
+			{
+				/* receive data from this connection */
+				bytes_received = recv(connected,recv_data, TCP_SOCKET_BUFFER_SIZE, 0);
+				if (bytes_received <= 0)
+				{
+					rt_kprintf("close client connection, errno: %d, socket error: %d\n", 
+						rt_get_errno(),
+						lwip_get_error());
+					/* connection closed. */
+					lwip_close(connected);
+					break;
+				}
+
+				/* send data to client */
+				send(connected, recv_data, bytes_received, 0);
+			}
+		}
+   }
+
+_exit:
+	/* close socket */
+	if (sock != -1) lwip_close(sock);
+	rt_free(recv_data);
+
+	return ;
+}
+
+/*
+ * NetIO TCP server
+ */
+
+/* network test utilities entry */
+void net_test(void)
+{
+	/* start UDP echo server */
+	if (udpecho_tid == RT_NULL)
+	{
+		udpecho_tid = rt_thread_create("uecho",
+									udpecho_entry, RT_NULL,
+									512, RT_THREAD_PRIORITY_MAX/2, 5);
+		if (udpecho_tid != RT_NULL)
+			rt_thread_startup(udpecho_tid);
+	}
+	if (udpecho_socket_tid == RT_NULL)
+	{
+		udpecho_socket_tid = rt_thread_create("uecho_s",
+									udpecho_socket_entry, RT_NULL,
+									512, RT_THREAD_PRIORITY_MAX/2 + 1, 5);
+		if (udpecho_socket_tid != RT_NULL)
+			rt_thread_startup(udpecho_socket_tid);
+	}
+
+	if (tcpecho_tid == RT_NULL)
+	{
+		tcpecho_tid = rt_thread_create("techo",
+									tcpecho_entry, RT_NULL,
+									512, RT_THREAD_PRIORITY_MAX/2 + 2, 5);
+		if (tcpecho_tid != RT_NULL)
+			rt_thread_startup(tcpecho_tid);
+	}
+	if (tcpecho_socket_tid == RT_NULL)
+	{
+		tcpecho_socket_tid = rt_thread_create("techo_s",
+									tcpecho_socket_entry, RT_NULL,
+									512, RT_THREAD_PRIORITY_MAX/2 + 3, 5);
+		if (tcpecho_socket_tid != RT_NULL)
+			rt_thread_startup(tcpecho_socket_tid);
+	}
+}
+FINSH_FUNCTION_EXPORT(net_test, network test);
+