IO多路tcp客户端

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<string.h>
#include<unistd.h>
#include<sys/select.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr,"_%d_",__LINE__);\
    perror(msg);\
    }while(0)

#define PORT 8888 //1024~49151
#define IP "192.168.31.127"

int update_maxfd(int maxfd,fd_set readfds)
{
	int i=maxfd;
	for(;i>0;i++)
	{
		if(FD_ISSET(i,&readfds))
			return i;
	}
	return 0;
}
int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("create socket success\n");



	// 填充地址信息结构体，真实的地址信息结构体与协议族相关
	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");



	//创建一个读集合
	fd_set readfds;
	fd_set tempfds;
	//将集合清空
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	//将0号文件描述符添加到集合
	FD_SET(0,&readfds);
	//将sfd文件描述符添加
	FD_SET(sfd,&readfds);

	//最大文件描述符
	int maxfd=sfd;
	int select_res=0;
	char buf[128]="";
	ssize_t res=0;
	int i;
	//定义一个数组，存储每次连接成功的客户端的信息
	struct sockaddr_in save_addr[1024-4];
	while(1)
	{
		tempfds=readfds;
		select_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(select_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(select_res==0)
		{
			fprintf(stderr,"select time out\n");
			break;
		}
		//能运行到当前位置说明集合中文件描述符准备就绪
		//且集合只会剩下产生的文件描述符
		//例如，0号产生事件，则只会剩下0号

		for(i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tempfds)==0)
			{
				continue;
			}

			if(i==0)
			{
				//触发键盘输入事件
				printf("触发键盘输入事件:");
				fflush(stdout);

				bzero(buf, sizeof(buf));

				//从终端获取文件描述符以及要发送的内容
				int sendfd;
				int res = scanf("%d %s", &sendfd, buf);
				while(getchar() != 10);
				if(res != 2)
				{
					fprintf(stderr, "请输入正确格式：文件描述符 要发送的内容\n");
					continue;
				}

				//判断获取到的文件描述符是否在readfds集合中
				if(FD_ISSET(sendfd, &readfds) == 0)
				{
					fprintf(stderr, "sendfd=%d 文件描述符不在集合中\n", sendfd);
					continue;
				}


				if(send(sendfd, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send message success\n");
			}
			else if(i==sfd)
			{
				printf("触发客户端连接事件：");
				fflush(stdout);


				//循环接收
				bzero(buf,sizeof(buf));
				size_t res=recv(i,buf,sizeof(buf),0);
				if(res<0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(res ==0)
				{
					printf("[%s:%d] newfd=%d读取失败\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),i);
					//关闭文件描述符
					close(i);
					//将文件描述符从集合中剔除
					FD_CLR(i,&readfds);
					//更新maxfd
					maxfd=update_maxfd(maxfd,readfds);
				}
				else
				{
					printf("[%s:%d] newfd=%d %s\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),i,buf);
				}

			}
		}
	}
	close(sfd);


	return 0;
}