嵌入式Linux网络编程之：网络高级编程
2．select()
使用fcntl()函数虽然可以实现非阻塞I/O或信号驱动I/O，但在实际使用时往往会对资源是否准备完毕进行循环测试，这样就大大增加了不必要的CPU资源的占用。在这里可以使用select()函数来解决这个问题，同时，使用select()函数还可以设置等待的时间，可以说功能更加强大。下面是使用select()函数的服务器端源代码。客户端程序基本上与10.2.3小节中的例子相同，仅加入一行sleep()函数，使得客户端进程等待几秒钟才结束。
/*net_select.c*/
#includesys/types.h>
#includesys/socket.h>
#includestdio.h>
#includestdlib.h>
#includestring.h>
#includesys/time.h>
#includesys/ioctl.h>
#includeunistd.h>
#includenetinet/in.h>
#definePORT4321
#defineMAX_QUE_CONN_NM5
#defineMAX_SOCK_FDFD_SETSIZE
#defineBUFFER_SIZE1024
intmain()
{
structsockaddr_inserver_sockaddr,client_sockaddr;
intsin_size,count;
fd_setinset,tmp_inset;
intsockfd,client_fd,fd;
charbuf[BUFFER_SIZE];
if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
{
perror(socket);
exit(1);
}
server_sockaddr.sin_family=AF_INET;
server_sockaddr.sin_port=htons(PORT);
server_sockaddr.sin_addr.s_addr=INADDR_ANY;
bzero((server_sockaddr.sin_zero),8);
inti=1;/*允许重复使用本地地址与套接字进行绑定*/
setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,i,sizeof(i));
if(bind(sockfd,(structsockaddr*)server_sockaddr,
sizeof(structsockaddr))==-1)
{
perror(bind);
exit(1);
}
if(listen(sockfd,MAX_QUE_CONN_NM)==-1)
{
perror(listen);
exit(1);
}
printf(listening....n);
/*将调用socket()函数的描述符作为文件描述符*/
FD_ZERO(inset);
FD_SET(sockfd,inset);
while(1)
{
tmp_inset=inset;
sin_size=sizeof(structsockaddr_in);
memset(buf,0,sizeof(buf));
/*调用select()函数*/
if(!(select(MAX_SOCK_FD,tmp_inset,NULL,NULL,NULL)>0))
{
perror(select);
}
for(fd=0;fdMAX_SOCK_FD;fd++)
{
if(FD_ISSET(fd,tmp_inset)>0)
{
if(fd==sockfd)
{/*服务端接收客户端的连接请求*/
if((client_fd=accept(sockfd,
(structsockaddr*)client_sockaddr,sin_size))==-1)
{
perror(accept);
exit(1);
}
FD_SET(client_fd,inset);
printf(Newconnectionfrom%d(socket)n,client_fd);
}
else/*处理从客户端发来的消息*/
{
if((count=recv(client_fd,buf,BUFFER_SIZE,0))>0)
{
printf(Receivedamessagefrom%d:%sn,
client_fd,buf);
}
else
{
close(fd);
FD_CLR(fd,inset);
printf(Client%d(socket)hasleftn,fd);
}
}
}/*endofifFD_ISSET*/
}/*endofforfd*/
}/*endifwhilewhile*/
close(sockfd);
exit(0);
}
加入微信
获取电子行业最新资讯
搜索微信公众号：EEPW
或用微信扫描左侧二维码
