select的TCP服务器

#include "myhead.h"
#define ERR_MSG(msg) do{\
	printf("__%d__:",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 3696//端口号
#define IP "192.168.114.106"//本机ip
int keybord_events(void)//键盘输入事件
{
	char buf[128]="";
	bzero(buf,sizeof(buf));
	fgets(buf,sizeof(buf),stdin);
	buf[strlen(buf)-1]=0;
	printf(":%s\n",buf);
	return 0;

}
//客户端连接事件
int cliConnect_events(int sfd,struct sockaddr_in saveCin[],fd_set* preadfds,int*pmaxfd)
{
	int newfd=-1;
	struct sockaddr_in cin;//存储客户端的地址信息
	socklen_t addrlen=sizeof(cin);//真实的地址信息结构体的大小
	newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
	if(newfd<0)
	{
		ERR_MSG("newfd");
		return -1;
	}
	printf("[%s:%d]客户端链接成功 newfd=%d\n",\
			inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
	//另存cin到newfd对应的下标位置
	saveCin[newfd]=cin;
	//将newfd添加到集合中
	FD_SET(newfd,preadfds);
	*pmaxfd=*pmaxfd>newfd?*pmaxfd:newfd;//更新maxfd
}
//客户端交互事件
int cliRcvSnd_evebts(int fd,struct sockaddr_in* saveCin,fd_set* preadfds,int*pmaxfd)
{
	char buf[128]="";
	//清空字符串
	bzero(buf,sizeof(buf));
	//接收
	ssize_t res=recv(fd,buf,sizeof(buf),0);
	if(res<0)
	{
		ERR_MSG("recv");
		return -1;
	}
	else if(0==res)
	{

		printf("[%s:%d]客户端下线 newfd=%d\n",\
				inet_ntoa(saveCin[fd].sin_addr),ntohs(saveCin[fd].sin_port),fd);
		close(fd);//关闭文件描述符
		FD_CLR(fd,preadfds);//将文件描述符从集合中剔除
		//由于剔除的文件描述符可能是最大文件描述符更新maxfd
		/*	for(int j=maxfd;j>=0;j--)
			{
			if(FD_ISSET(j,&readfds))
			{
			break;
			}
			maxfd=j;
			}*/
		//从大往小判断，若在集合中则改变maxfd
		while(FD_ISSET(*pmaxfd,preadfds)==0&&(*pmaxfd)-->=0);
		return 0;
	}
	printf("[%s:%d] newfd=%d:%s\n",inet_ntoa(saveCin[fd].sin_addr),\
			ntohs(saveCin[fd].sin_port),fd,buf);
	//发送
	strcat(buf," (030)y ");
	if(send(fd,buf,sizeof(buf),0)<0)
	{
		ERR_MSG("send");
		return -1;
	}
	printf("send success\n");

}
int main(int argc, const char *argv[])
{	
	//创建流式套接字 socket
	int sfd =socket(AF_INET,SOCK_STREAM,0);
	{
		if(sfd <0)
		{
			ERR_MSG("socket");		
			return -1;
		}
	}
	printf("socket create success sfd=%d\n",sfd);
	//允许端口被快速服用成功
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("端口允许复用\n");
	//填充地址信息结构体给bind函数绑定
	//真实的地址信息结构体根据地址族指定 AF_INET：man 7 ip

	struct sockaddr_in sin;
	sin.sin_family =AF_INET;//必须填写AF_INET
	sin.sin_port =htons(PORT);//端口号的网络字节序
	sin.sin_addr.s_addr=inet_addr(IP);//本机IP
	//绑定服务器的地址信息--必须绑定bind
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("socket success\n");
	//将套接字设置为被动监听状态 listen
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	//创建一个集合
	fd_set readfds,tempfds;
	//清空集合
	FD_ZERO(&readfds);
	//将需要监测的文件描述符添加到集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	//储存最大的文件描述符
	int maxfd=sfd;
	//获取一个已经完成的客户端信息，生成一个新的文件描述符 accept;
	int s_res=-1;
	ssize_t res=-1;
	char buf[128]="";
	int newfd=-1;	
	struct sockaddr_in saveCin[1024];//备份链接成功的客户端地址信息，用下标来对应文件扫描
	while(1)
	{
		tempfds=readfds;
		//执行IO多复用函数
		s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(s_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res)
		{
			printf("time out,,\n");
			break;
		}
		printf("__%d__",__LINE__);
		for(int i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tempfds)==0)
				continue;
			if(0==i)
			{
				printf("触发键盘输入事件\n");
				keybord_events();
			}
			else if(sfd==i)
			{
				printf("触发客户端连接事件\n");
				cliConnect_events(sfd,saveCin,&readfds,&maxfd);

			}
			else
			{
			 	printf("触发客户端交互事件\n");
				cliRcvSnd_evebts(i,saveCin,&readfds,&maxfd);
			}
		}
	}


	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;

}

 测试：

socket create success sfd=3
端口允许复用
socket success
listen success
__154__触发客户端连接事件
[192.168.114.198:58293]客户端链接成功 newfd=4
__154__触发客户端连接事件
[192.168.114.198:58294]客户端链接成功 newfd=5
__154__触发客户端交互事件
[192.168.114.198:58293] newfd=4:1
send success
__154__触发客户端交互事件
[192.168.114.198:58293] newfd=4:1
send success
__154__触发客户端交互事件
[192.168.114.198:58294] newfd=5:2
send success
__154__触发客户端交互事件
[192.168.114.198:58294] newfd=5:2
send success
1
__154__触发键盘输入事件
:1
3
__154__触发键盘输入事件
:3
__154__触发客户端交互事件
[192.168.114.198:58293]客户端下线 newfd=4
__154__触发客户端交互事件
[192.168.114.198:58294]客户端下线 newfd=5
__154__触发客户端连接事件
[192.168.114.198:58296]客户端链接成功 newfd=4
__154__触发客户端连接事件
[192.168.114.198:58297]客户端链接成功 newfd=5
__154__触发客户端交互事件
[192.168.114.198:58296]客户端下线 newfd=4
__154__触发客户端交互事件
[192.168.114.198:58297]客户端下线 newfd=5

 思维导图