国庆day2---select实现服务器并发

select.c：

#include <myhead.h>

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

#define IP "192.168.1.3"
#define PORT 8888

int main(int argc, const char *argv[])
{
	//创建报式套接字socket
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(-1==sfd){
		ERR_MSG("socket error");
		return -1;
	}
	printf("socket success\n");

	//允许端口快速复用setsockopt
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))){
		ERR_MSG("setsockopt error");
		return -1;
	}

	//定义一个地址结构体
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//将套接字和IP、端口绑定bind
	if(-1==bind(sfd,(struct sockaddr*)&sin,sizeof(sin))){
		ERR_MSG("bind error");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置成被动监听状态
	if(-1==listen(sfd,128)){
		ERR_MSG("listen error");
		return -1;
	}

	int newfd=-1;

	fd_set readfds,tempfds;

	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);

	int maxfd=sfd;
	struct sockaddr_in saveCin[1024];

	int s_res=0;
	ssize_t res=0;

	char buf[128]="";

	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);

	while(1){
		tempfds=readfds;

		s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(-1==s_res){
			ERR_MSG("select error");
			return -1;
		}
		else if(0==s_res){
			printf("time out...\n");
			return -1;
		}

		for(int i=0;i<=maxfd;i++){
			if(!FD_ISSET(i,&tempfds)){
				continue;
			}
			if(0==i){
				printf("触发键盘输入事件>>>\n");
				bzero(buf,sizeof(buf));
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				printf(":%s\n",buf);
			}
			else if(sfd==i){
				printf("触发客户端连接事件>>>\n");
				if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen)) == -1){
					ERR_MSG("accept");
					return -1;
				}
				printf("[%s:%d] newfd=%d 连接成功\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

				saveCin[newfd]=cin;

				FD_SET(newfd,&readfds);
				maxfd = maxfd>newfd ? maxfd : newfd;
			}
			else{
				printf("触发客户端交互事件>>>\n");
				bzero(buf,sizeof(buf));

				res = recv(i,buf,sizeof(buf),0);
				if(-1 == res){
					ERR_MSG("recv");
					return -1;
				}
				else if(0 == res){
					printf("[%s:%d] newfd=%d 下线了\n",inet_ntoa(saveCin[i].sin_addr),ntohs(saveCin[i].sin_port),i);

					close(i);

					FD_CLR(i,&readfds);

					while(FD_ISSET(maxfd,&readfds) == 0 && maxfd-->=0);
					continue;
				}
				printf("[%s:%d] newfd=%d : %s\n",inet_ntoa(saveCin[i].sin_addr),ntohs(saveCin[i].sin_port),i,buf);

				strcat(buf,"*-*");
				if(send(i,buf,sizeof(buf),0) == -1){
					ERR_MSG("send");
					return -1;
				}
				printf("发送成功\n");
			}
		}
	}
	close(sfd);
	
	return 0;
}

效果图：