select函数
函数原型:
int select(int nfds, fd_set *readfds, fd_set *writefds,
fd_set *exceptfds, struct timeval *timeout);
参1: 所监听的所有文件描述符中,最大的文件描述符+1,fd4+1
参2/3/4: 所监听的文件描述符“可读”事件。readfds();
所监听的文件描述符“可写”事件。writefds();
所监听的文件描述符“异常”事件。exceptfds();
参5:定时阻塞监控时间,3种情况
1.NULL,永远等下去
2.设置timeval,等待固定时间
3.设置timeval里时间均为0,检查描述字后立即返回,轮询。
返回值:成功:所监听的所有的监听集合中,满足条件的总数。
失败:
例:read返回fd1、fd2,write返回fd2、fd4,except返回fd3、fd4,则返回值等于6。
void FD_CLR(int fd, fd_set *set);将set清空。
int FD_ISSET(int fd, fd_set *set);将fd从set中清除出去。
void FD_SET(int fd, fd_set *set);判断fd是否在集合中。
void FD_ZERO(fd_set *set);将fd设置到set集合中去。
代码实现
出错处理函数头文件wrap.h
#ifndef __WRAP_H_
#define __WRAP_H_
void perr_exit(const char *s);
int Accept(int fd,struct sockaddr *sa,socklen_t *salenptr);
int Bind(int fd,const struct sockaddr *sa,socklen_t salen);
int Connect(int fd,const struct sockaddr *sa,socklen_t salen);
int Listen(int fd,int backlog);
int Socket(int family,int type,int protocol);
ssize_t Read(int fd,void *ptr,size_t nbytes);
int Close(int fd);
ssize_t Readn(int fd,void *vptr,size_t n);
ssize_t Writen(int fd,const void *vptr,size_t n);
ssize_t my_read(int fd,char *ptr);
ssize_t Readline(int fd,void *vptr,size_t maxlen);
#endif
出错处理函数wrap.c
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <sys/socket.h>
void perr_exit(const char *s)
{
perror(s);
exit(-1);
}
int Accept(int fd,struct sockaddr *sa,socklen_t *salenptr)
{
int n;
again:
if((n=accept(fd,sa,salenptr))<0){
if((errno==ECONNABORTED)||(errno==EINTR))
goto again;
else
perr_exit("accept error");
}
return n;
}
int Bind(int fd,const struct sockaddr *sa,socklen_t salen)
{
int n;
if((n=bind(fd,sa,salen))<0)
perr_exit("Bind error");
return n;
}
int Connect(int fd,const struct sockaddr *sa,socklen_t salen)
{
int n;
printf("----------------in wrap before connect() \n");
n=connect(fd,sa,salen);
printf("----------------in wrap connect() return %d\n",n);
if(n<0){
perr_exit("connect error");
}
return 0;
}
int Listen(int fd,int backlog)
{
int n;
if((n=listen(fd,backlog))<0)
perr_exit("listen error");
return n;
}
int Socket(int family,int type,int protocol)
{
int n;
if((n=socket(family,type,protocol))<0)
perr_exit("socket error");
return n;
}
ssize_t Read(int fd,void *ptr,size_t nbytes)
{
ssize_t n;
again:
if((n=read(fd,ptr,nbytes))==-1){
if(errno==EINTR)
goto again;
else
return -1;
}
return n;
}
ssize_t Write(int fd,const void *ptr,size_t nbytes)
{
ssize_t n;
again:
if((n=write(fd,ptr,nbytes))==-1){
if(errno==EINTR)
goto again;
else
return -1;
}
return n;
}
int Close(int fd)
{
int n;
if((n=close(fd))==-1)
perr_exit("close error");
return n;
}
/*参三:应该读取的字节数*/
ssize_t Readn(int fd,void *vptr,size_t n)
{
size_t nleft;//usigned int 剩余未读取的字节数
ssize_t nread;//int 实际读到的字节数
char *ptr;
ptr=vptr;
nleft=n;//n 未读取字节数
while(nleft>0){
if((nread=read(fd,ptr,nleft))<0){
if(errno==EINTR)
nread=0;
else
return -1;
}else if(nread==0)
break;
nleft-=nread;
ptr+=nread;
}
return n-nleft;
}
ssize_t Writen(int fd,const void *vptr,size_t n)
{
size_t nleft;
ssize_t nwritten;
const char *ptr;
ptr=vptr;
nleft=n;
while(nleft>0){
if((nwritten=write(fd,ptr,nleft))<=0){
if(nwritten<0&&errno==EINTR)
nwritten=0;
else
return -1;
}
nleft-=nwritten;
ptr+=nwritten;
}
return n;
}
static ssize_t my_read(int fd,char *ptr)
{
static int read_cnt;
static char *read_ptr;
static char read_buf[100];
if(read_cnt<=0){
again:
if((read_cnt=read(fd,read_buf,sizeof(read_buf)))<0){
if(errno==EINTR)
goto again;
return -1;
}else if(read_cnt==0)
return 0;
read_ptr=read_buf;
}
read_cnt--;
*ptr=*read_ptr++;
return 1;
}
/*readline --- fgets*/
//传出参数 vptr
ssize_t Readline(int fd,void *vptr,size_t maxlen)
{
ssize_t n,rc;
char c,*ptr;
ptr=vptr;
for(n=1;n<maxlen;n++){
if((rc=my_read(fd,&c))==1){
*ptr++=c;
if(c=='\n')
break;
}else if(rc==0){
*ptr=0;
return n-1;
}else
return -1;
}
*ptr=0;
return n;
}
服务端函数server.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <ctype.h>
#include "wrap.h"
#define SERV_PORT 6666
int main(int argc,char *argv[])
{
int i,j,n,maxi;
int nready,client[FD_SETSIZE]; /*自定义数组client,防止遍历1024个文件描述符 FD_SETSIZE默认为1024*/
int maxfd,listenfd,connfd,sockfd;
char buf[BUFSIZ],str[INET_ADDRSTRLEN]; /* #define INET_ADDRSTRLEN 16 */
struct sockaddr_in clie_addr,serv_addr;
socklen_t clie_addr_len;
fd_set rset,allset;
listenfd=Socket(AF_INET,SOCK_STREAM,0);
bzero(&serv_addr,sizeof(serv_addr));
serv_addr.sin_family=AF_INET;
serv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
serv_addr.sin_port=htons(SERV_PORT);
Bind(listenfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr));
Listen(listenfd,20);
maxfd=listenfd;
/*起初listenfd即为最大文件描述符*/
maxi=-1;
for(i=0;i<FD_SETSIZE;i++) /*将来用作client[]的下标,初始值指向0个元素之前下标位置*/
client[i]=-1; /*用-1初始化client[]*/
FD_ZERO(&allset);
FD_SET(listenfd,&allset); /*构造select监控文件描述符集*/
while(1){
rset=allset;
nready=select(maxfd+1,&rset,NULL,NULL,NULL);
if(nready<0)
perr_exit("select error");
if(FD_ISSET(listenfd,&rset)){ /*说明有新的客户端链接请求*/
clie_addr_len=sizeof(clie_addr);
connfd=Accept(listenfd,(struct sockaddr *)&clie_addr,&clie_addr_len); /*Accept不会阻塞*/
printf("received from %s at PORT %d\n",
inet_ntop(AF_INET,&clie_addr.sin_addr,str,sizeof(str)),
ntohs(clie_addr.sin_port));
for(i=0;i<FD_SETSIZE;i++)
if(client[i]<0){
client[i]=connfd; /*找client[]中没有使用的位置*/
break; /*保存accept返回的文件描述符到client[]里*/
}
if(i==FD_SETSIZE){
fputs("too many clients\n",stderr); /*达到select能监控的文件个数上限1024*/
exit(1);
}
FD_SET(connfd,&allset); /*向监控文件描述符集合allset添加新的文件描述符connfd*/
if(connfd>maxfd)
maxfd=connfd; /*select第一个参数需要*/
if(i>maxi)
maxi=i; /*保证maxi存的总是client[]最后一个元素下标*/
if(--nready==0)
continue;
}
for(i=0;i<=maxi;i++){
if((sockfd=client[i])<0)
continue;
if(FD_ISSET(sockfd,&rset)){
if((n=Read(sockfd,buf,sizeof(buf)))==0){ /*当client关闭链接时,服务器端也关闭对应链接*/
Close(sockfd); /*解除select对此文件描述符的监控*/
FD_CLR(sockfd,&allset);
client[i]=-1;
}else if(n>0)
for(j=0;j<n;j++){
buf[j]=toupper(buf[j]);
sleep(10);
Writen(sockfd,buf,n);
}
if(--nready==0)
break; /*跳出for,但还在while中*/
}
}
}
Close(listenfd);
return 0;
}
在终端输入:
make
./server
再启动一个终端连接服务端
nc 127.0.0.1 6666