9.20io多路复用select创建客户端，服务器

客户端

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <sys/select.h>
#include <sys/time.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr, "__%d__:", __LINE__);\
	perror(msg);\
}while(0)

#define PORT 	8888
#define IP 		"192.168.31.175"
int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	//填充连接服务器IP和端口
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//连接服务器的IP和端口
	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	char buf[128]="";
	ssize_t res=0;
	int st_res=0;
	int newfd;
	int maxfd=sfd;
	//创建读集合，并清空
	fd_set readfds, tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	while(1)
	{
		tempfds=readfds;
		st_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(st_res==0)
		{
			printf("time out...\n");
			return -1;
		}
		else if(st_res<0)
		{
			ERR_MSG("select");
			return -1;
		}

		//
		if(FD_ISSET(0,&tempfds))
		{
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;
			//发送
			if(send(sfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		if(FD_ISSET(sfd,&tempfds))
		{
			//接收
			bzero(buf,sizeof(buf));
			res=recv(sfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{
				fprintf(stderr,"服务器关闭\n");
				break;
			}
			printf("[%s:%d]: %s\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf);
		}
	}

	close(sfd);

	return 0;
}

服务器： 

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>
#include<sys/un.h>
#include <sys/time.h>
#include <sys/types.h>

#define ERR_MSG(msg)do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 8888
#define IP "192.168.31.175"
int updatemaxfd(int maxfd,fd_set readfds);
int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	//填充IP和端口到地址信息结构体

	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//绑定服务器的IP和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	//listen监听
	if(listen(sfd,10)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");


	//创建读集合
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	//把需要的文件描述符放到读集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);

	int maxfd=sfd;
	int st_res=0;
	char buf[128]="";
	ssize_t res=0;
	int newfd=0;
	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);

	int i = 0;

	while(1)
	{
		tempfds = readfds;
		//调用select函数
		st_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);	
		if(st_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==st_res)
		{
			printf("time out...\n");
			break;
		}
		//运行到这里，表示文件描述符集合中航油文件描述符准备就绪

		for(i=0;i<=maxfd;i++)
		{
			//分别判断i对应的文件描述符在不在集合中
			if(FD_ISSET(i,&tempfds)==0)
			{
				continue;
			}
			//能运行到这里说明i在tempfds中
			if(0==i)
			{	

				printf("触发键盘输入事件\n");
				bzero(buf,sizeof(buf));
				fgets(buf,sizeof(buf),stdin);
				printf("buf=%s\n",buf);


			}
			else if(sfd==i)
			{
				printf("触发客户端连接事件\n");
				//获取新的文件描述符
				newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
				if(newfd<0)
				{
					ERR_MSG("accept");
					return -1;
				}

				printf("[%s: %d] newfd=%d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

				//newfd添加到集合中
				FD_SET(newfd,&readfds);
				//更新maxfd
				maxfd=maxfd>newfd?maxfd:newfd;
			}
			else
			{

				printf("触发客户端交互事件\n");
				bzero(buf,sizeof(buf));
				res=recv(i,buf,sizeof(buf),0);
				if(res<0)
				{

					ERR_MSG("recv");
					return -1;
				}
				else if(0==res)
				{
					fprintf(stderr,"newfd=%d客户端关闭\n",i);
					close(i);//关闭不需要的文件描述符
					//将文件描述符从集合中剔除
					FD_CLR(i,&readfds);
					//更新maxfd
					maxfd=updatemaxfd(maxfd,readfds);
				}
				else
				{

					printf("newfd=%d:%s\n",i,buf);
				}
			}

		}

	}
	close(sfd);
	return 0;
}
int updatemaxfd(int maxfd,fd_set readfds)
{
	for(int j=maxfd;j>=0;j--)
	{
		if(FD_ISSET(j,&readfds))
		{
			return j;
		}
	}
	return 0;
}