网络第5天 - 技术栈

使用IO多路复用（select）实现TCP服务器与客户端的通信

服务器

cpp 复制代码

#include <myhead.h>
#define IP "192.168.124.74"
#define PORT 6666

int cli_link(int sfd,struct sockaddr_in* acin,fd_set* fdset,int* maxfd)
{
	int newfd=0;
	struct sockaddr_in cin;
	socklen_t len=sizeof(cin);

	printf("触发连接事件\n");
	newfd=accept(sfd,(struct sockaddr*)&cin,&len);
	if(newfd<0){
		ERR_MSG("accept");
		return -1;
	}
	printf("newfd:%d %s:%d 连接成功\n",newfd\
			,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));

	acin[newfd-3]=cin; //把客户端地址端口存起来

	FD_SET(newfd,fdset);
	*maxfd= *maxfd>newfd? *maxfd:newfd;

	return 0;
}

int cli_connect(struct sockaddr_in* cin,int i,fd_set* fdset,int* maxfd)
{
	char buf[100]="";
	ssize_t res=0;

	bzero(buf,sizeof(buf));            
	//接收客户端数据                   
	res=recv(i,&buf,sizeof(buf),0);    
	if(res<0){                         
		ERR_MSG("recv");               
		return -1;                        
	}else if(res==0){                  
		printf("[%s:%d]:%s 客户端下线\n",inet_ntoa(cin[i-3].sin_addr),ntohs(cin[i-3].sin_port),buf);   
		close(i);
		FD_CLR(i,fdset);//删除
		while(FD_ISSET(*maxfd,fdset)==0 && (*maxfd)-- >0);//更新maxfd
		return 0;                         
	}                                  
	printf("[%s:%d]:%s\n",inet_ntoa(cin[i-3].sin_addr),\
			ntohs(cin[i-3].sin_port),buf);  

	//向客户端发送数据                 
	//要发送的数据                     
	strcat(buf,"^-^");                 
	res=send(i,buf,sizeof(buf),0); 
	if(res<0){                         
		ERR_MSG("send");               
		return -1;                     
	}                                  
	printf("send success\n");          
	return 0;
}

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("sfd create success\n");

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

	//绑定服务器的IP和端口号
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind IP&PORT success\n");

	//将文件转为被动监听模式
	if(listen(sfd,128)<0){
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	//创建读集合
	fd_set fdset,temp;

	//清空并填充读集合
	FD_ZERO(&fdset);
	FD_SET(0,&fdset);
	FD_SET(sfd,&fdset);

	int res=0;
	char buf[100]="";

	int maxfd=sfd;
	struct sockaddr_in cin[1024-3];

	while(1){
		temp=fdset;

		//调用select
		res=select(maxfd+1,&temp,NULL,NULL,NULL);
		if(res<0){
			ERR_MSG("select");
			return -1;
		}else if(res==0){
			printf("超时\n");
			break;
		}

		for(int i=0;i<=maxfd;i++){
			if(FD_ISSET(i,&temp)==0){
				continue;
			}
			//判断读集合内的数据，根据就绪文件描述符执行相关代码
			if(i==0){
				printf("触发输入事件\n");
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]='\0';
				printf("buf=%s\n",buf);
			}else if(i==sfd){
				//客户端连接服务器
				cli_link(sfd,cin,&fdset,&maxfd);
			}else{
				//客户端与服务器通信
				cli_connect(cin,i,&fdset,&maxfd);    
			}
		}	
	}

	//关闭所有文件描述符
	close(sfd);
	return 0;
}

客户端

cpp 复制代码

#include <myhead.h>

#define PORT 6666
#define IP "192.168.124.74"

int main(int argc, const char *argv[])
{
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\n");

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

	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("connect");
		return -1;
	}
	printf("connect to %s:%d success\n",\
			inet_ntoa(sin.sin_addr),ntohs(sin.sin_port));

	fd_set readfd,tempfd; //创建读体

	FD_ZERO(&readfd); //清空并填充读体
	FD_SET(0,&readfd);
	FD_SET(sfd,&readfd);

	int select_res=0;
	int maxfd=sfd;
	char buf[100]="";

	while(1){
		tempfd=readfd;

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

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

			if(0==i){
				printf("输入数据发送给服务器>>>\n");
				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 to %s:%d:%s success\n",\
						inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf);

			}else if(sfd==i){
				printf("从服务器收到数据\n");
				bzero(buf,sizeof(buf));
				if(recv(sfd,buf,sizeof(buf),0)<0){
					ERR_MSG("recv");
					return -1;
				}
				printf("[%s:%d]:%s\n",\
						inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf);
			}
		}
	}
	close(sfd);
	return 0;
}

现象：