2023/8/31 -- 网络编程

I/O多路复用

select的TCP服务器模型：

#include <myhead.h>

#define PORT 8888
#define IP "192.168.114.153"

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

int main(int argc, const char *argv[])
{
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(-1 == sfd){
		ERR_MSG("socket");
		return -1;
	}
	int reuse = 1;
	if(-1 == setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))){
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速复用成功\n");
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	if(-1 == bind(sfd,(struct sockaddr*)&sin,sizeof(sin))){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	if(-1 == listen(sfd,128)){
		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;
	int newfd = -1;
	char buf[128] = "";
	ssize_t res = 0;

	struct sockaddr_in savecin[1024];
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	while(1){
		tempfds = readfds;
		if(select(maxfd+1,&tempfds,NULL,NULL,NULL) == -1){
			ERR_MSG("select");
			return -1;
		}
		for(int i = 0;i <= maxfd;i++){
			if(FD_ISSET(i,&tempfds) == 0)
				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(i == sfd){
				printf("触发客户端连接事件\n");
				newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
				if(-1 == newfd){
					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(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 buf = %s\n",inet_ntoa(savecin[i].sin_addr),ntohs(savecin[i].sin_port),i,buf);
				strcat(buf," *_* ");
				if(-1 == send(i,buf,sizeof(buf),0)){
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
			}
		}
	}
	close(sfd);
	close(newfd);
	return 0;
}

select的TCP客户端模型：

#define PORT 8888
#define IP "192.168.114.153"

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

int main(int argc, const char *argv[])
{
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(-1 == cfd){
		ERR_MSG("socket");
		return -1;
	}

	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	if(-1 == connect(cfd,(struct sockaddr*)&sin,sizeof(sin))){
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);

	int maxfd = cfd;
	char buf[128] = "";
	ssize_t res = 0;
	ssize_t ret = 0;
	while(1){
		tempfds = readfds;
		if(-1 == select(maxfd+1,&readfds,NULL,NULL,NULL)){
			ERR_MSG("select");
			return -1;
		}
		if(FD_ISSET(0,&tempfds)){
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1] = 0;
			ret = send(cfd,buf,sizeof(buf),0);
			if(-1 == ret){
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
			FD_SET(cfd,&tempfds);
		}
		if(FD_ISSET(cfd,&tempfds)){
			bzero(buf,sizeof(buf));
			res = recv(cfd,buf,sizeof(buf),0);
			if(-1 == res){
				ERR_MSG("recv");
				return -1;
			}
			printf("cfd = %d buf = %s\n",cfd,buf);
			FD_SET(0,&tempfds);
		}
	}
	close(cfd);
	return 0;
}

效果图：