Niobe开发板TCP联网演示

struct sockaddr_in {
	sa_family_t sin_family;
	in_port_t sin_port;
	struct in_addr sin_addr;
	uint8_t sin_zero[8];
};

	//服务端地址信息
	struct sockaddr_in server_sock;
	bzero(&server_sock, sizeof(server_sock));
	server_sock.sin_family = AF_INET;
	//server_sock.sin_addr.s_addr = htonl(INADDR_ANY);
	inet_pton(AF_INET, "192.168.1.121", &server_sock.sin_addr);
	server_sock.sin_port = htons(_PROT_);

typedef struct {
	unsigned long fds_bits[FD_SETSIZE / 8 / sizeof(long)];
} fd_set;

	fd_set fds;
	FD_ZERO(&fds);
    FD_SET(sock_fd,&fds);//将sock_fd添加至fds

    if(FD_ISSET(sock_fd,&fds))//判断sock_fd是否在fds中
        ;//

int WifiConnect(const char *ssid, const char *psk)

int socket(int domain, int type, int protocol)

int bind(int fd, const struct sockaddr *addr, socklen_t len)

int listen(int fd, int backlog)

int accept(int socket, struct sockaddr *address, socklen_t *address_len)

ssize_t recv(int fd, void *buf, size_t len, int flags)

ssize_t send(int fd, const void *buf, size_t len, int flags)

int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout)

static void TCPServerTaskWithSelect(void)
{
	//服务端地址信息
	struct sockaddr_in server_sock;
	//客户端地址信息
	struct sockaddr_in client_sock;
	int sin_size = sizeof(struct sockaddr_in);
	struct sockaddr_in *cli_addr;
	//连接Wifi
	if(WifiConnect(WIFI_SSID,WIFI_PASSWORD)!=0)
	{
		perror("Wifi connect error!");
		exit(1);
	}
	in_addr_t localIpaddr = GetLocalIpaddr();
	printf("ip = %x\n",localIpaddr);

	//创建socket
	printf("socket begin\n");
	int sock_fd=-1,new_fd=-1;
	if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
	{
		perror("socket is error\r\n");
		exit(1);
	}
	printf("sock_fd=%d\n",sock_fd);

	bzero(&server_sock, sizeof(server_sock));
	server_sock.sin_family = AF_INET;
	//server_sock.sin_addr.s_addr = htonl(INADDR_ANY);
	inet_pton(AF_INET, "192.168.1.114", &server_sock.sin_addr);
	server_sock.sin_port = htons(_PROT_);
	char str[IP_LEN];
	inet_ntop(AF_INET, &server_sock.sin_addr, str, sizeof(str));
	printf("tcp server IP_addr: %s at PORT %d\n",str,ntohs(server_sock.sin_port));
	
	//调用bind函数绑定socket和地址
	if (bind(sock_fd, (struct sockaddr *)&server_sock, sizeof(struct sockaddr)) == -1)
	{
		perror("bind is error\r\n");
		exit(1);
	}
	//调用listen函数监听(指定port监听)
	if (listen(sock_fd, TCP_BACKLOG) == -1)
	{
		perror("listen is error\r\n");
		exit(1);
	}
	//init fd_arr
	printf("start select\r\n");
	int i=0;
	for(;i<_MAX_SIZE_;++i)
	{
		fd_arr[i]=-1;
	}
	add_fd_arr(sock_fd);
	fd_set fds;
	FD_ZERO(&fds);

	while(true)
	{
		//更新max_fd和文件描述符集
		max_fd=sock_fd;
		for(int ii=0; ii<_MAX_SIZE_; ii++)
		{
			if(fd_arr[ii]!=-1)
			{
				FD_SET(fd_arr[ii],&fds);
				if(fd_arr[ii]>max_fd)
				{
					max_fd=fd_arr[ii];
				}
			}
		}
		//printf("update max_fd = %d\n",max_fd);
		printf("select wait...\n");
        int ret = select(max_fd + 1, &fds, NULL, NULL, NULL);
		if(ret<=0){
			//超时等
			printf("[DISCOVERY]ret:%d\n", ret);
			continue;
		}
		if(FD_ISSET(sock_fd,&fds)) //判断是否发生accept
		{
			new_fd = accept(sock_fd, (struct sockaddr *)&client_sock, (socklen_t *)&sin_size);
			//printf("accept new fd = %d\n",new_fd);
			if(-1!=new_fd)
			{
				//打印客户端的ip和port
				char str[IP_LEN];
				inet_ntop(AF_INET, &client_sock.sin_addr, str, sizeof(str));
				printf("connected from %s at PORT %d\n",str,ntohs(client_sock.sin_port));				
				if(1==add_fd_arr(new_fd))
				{
					perror("fd_arr is full,close new_fd\n");
					close(new_fd);//此处比较粗陋，应该关闭久未发生信息交付的fd
				}								
			}
			continue;
		}
		//客户端发来信息
		for(i=0;i<_MAX_SIZE_;++i)
		{
			if(fd_arr[i]!=-1&&FD_ISSET(fd_arr[i],&fds))
			{
				char buf[1024];
				memset(buf,'\0',sizeof(buf));
				ssize_t size=recv(fd_arr[i],buf,sizeof(buf)-1,0);
				//"exit"特殊字符串，表示客户端申请断开链接
				if(size==0 || size==-1 || memcmp(buf,"exit",size)==0)
				{
					printf("remote client fd is %d, close,size is %d,msg is %s\n",fd_arr[i],size,buf);					
					FD_CLR(fd_arr[i],&fds);
					close(fd_arr[i]);
					fd_arr[i]=-1;
				}else
				{
					printf("recv: fd:%d,msg:%s",fd_arr[i],buf);
					if ((ret = send(fd_arr[i], tcpSendBuf, strlen(tcpSendBuf) + 1, 0)) == -1)
					{
						perror("send error \r\n");
					}					
				}
				osDelay(10);
				continue;
			}
		}
	}	
	close(sock_fd);
	FD_CLR(sock_fd,&fds);
	sock_fd=-1;
	return ;
}

// 默认WiFi名和密码
#define WIFI_SSID "aaa"
#define WIFI_PASSWORD "talkweb1996"

// 默认服务端的ip地址和端口号
#define _PROT_ 8800
#define _SERVER_IP_ "192.168.1.112"

        # "TW303_Network_mqttclient:network_mqttclient_example",
        # "TW402_APP_oled_u8g2:app_oled_u8g2_example",
         "TW304_Network_tcpserver:network_tcpserver_demo",
        # "TW304_Network_tcpclient:network_tcpclient_demo",
        # "TW305_Network_udpclient:network_udpclient_demo",

callback function for wifi connect

WaitConnectResult:wait success[1]s
WiFi connect succeed!
begain to dhcp<-- DHCP state:Inprogress -->

<-- DHCP state:OK -->
server :
	server_id : 192.168.1.1
	mask : 255.255.255.0, 1
	gw : 192.168.1.1
	T0 : 7200
	T1 : 3600
	T2 : 6300
clients <1> :
	mac_idx mac             addr            state   lease   tries   rto     
	0       849dc22100d4    192.168.1.112   10      0       1       4     

tcp server IP_addr: 192.168.1.112 at PORT 8800
select wait...

connected from 192.168.1.114 at PORT 61941
select wait...
recv: fd:1,msg:0000
select wait...

recv: fd:1,msg:1111
select wait...

recv: fd:1,msg:2222
select wait...

remote client fd is 1, close,size is 4,msg is exit

select wait...