网络编程day5

思维导图

多路复用select实现服务器端

#include <header.h>
#define SER_PORT 8888
#define SER_IP "192.168.125.156"
int main(int argc, const char *argv[])
{
	//为通信创建一个节点
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd==-1)
	{
		perror("socket");
		return -1;
	}
	printf("socket success sfd=%d\n",sfd);
	//设置地址能够快速重用
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
	{
		perror("setsockopt");
		return 1;
	}
	//绑定ip地址和端口号
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(SER_PORT);
	sin.sin_addr.s_addr=inet_addr(SER_IP);
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");
	//设置套接字为监听状态
	if(listen(sfd,128)==-1)
	{
		perror("listen");
		return -1;
	}
	printf("listen success\n");
	//阻塞等待客户端的连接
	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);
	//准备一个集合，和它的备份
	fd_set readfds,tempfds;
	//清空集合内容
	FD_ZERO(&readfds);
	//将要检测的文件描述符放到集合里
	FD_SET(sfd,&readfds);
	FD_SET(0,&readfds);
	int newfd=0;     //存放客户端套接字变量
	int maxfd=sfd;   //存放容器中最大的文件描述符 
	struct sockaddr_in cin_arr[1024];   //存储客户端地址信息的结构体
	while(1)
	{
		//备份一下
		tempfds=readfds;
		//用select检测集合中是否有事件要发生
		int res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(res==-1)
		{
			perror("select");
			return 1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return 1;
		}
		//当程序执行到这，说明集合中已经有一个或多个事件
		//只需要判断哪些文件描述符还在集合中
		//将所有的文件描述符遍历一遍
		for(int i=0;i<maxfd;i++)
		{
			//判断当前的i文件描述符有没有事件
			if(!FD_ISSET(i,&tempfds))
			{
				continue;
			}
			//如果到这，说明i触发了事件
			//判断i是哪个文件描述符
			if(i=sfd)        //如果是sfd则执行accept函数
			{    
				newfd=accept(i,(struct sockaddr*)&cin,&addrlen);
				if(newfd==-1)
				{
					perror("accept");
					return -1;
				}
				printf("there is a new client accept\n");
				printf("ip:%s\nduankouhao:%d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
				//将newfd放入readfds集合里，以便检测是否有客户端发来消息
				FD_SET(newfd,&readfds);
				//更新maxfd
				maxfd=maxfd>newfd?maxfd:newfd;
				//将套接字地址信息结构体放入数组
				cin_arr[newfd]=cin;
			}
			else if(i==0)          //如果是0（输入）则解除阻塞执行输入
			{
				char buf[128]={0};
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				printf("键盘输入事件完成:%s\n",buf);
				//循环将消息发送给所有客户端
				for(int k=4;k<=maxfd;k++)
				{
					send(k,buf,strlen(buf),0);
				}
			}
			else
			{
				//与客户端通信
				char rbuf[128]={0};
				//清空容器
				bzero(rbuf,sizeof(rbuf));
				//从套接字读取数据
				int res=recv(i,rbuf,sizeof(rbuf),0);
				if(res==0)
				{
					printf("the client is offline\n");
					//关闭套接字
					close(i);
					//将当前套接字文件描述符从集合中移除
					FD_CLR(i,&readfds);
					//更新套接字
					for(int j=maxfd;j>=sfd;j--)
					{
						if(FD_ISSET(j,&readfds))
						{
							maxfd=j;
							break;
						}
					}
					continue;
				}
				printf("[%s:%d]:%s\n",inet_ntoa(cin_arr[i].sin_addr),ntohs(cin_arr[i].sin_port),rbuf);
				//将消息转发给所有客户端
				for(int k=4;k<=maxfd;k++)
				{
					if(k!=i)
					{
						send(k,rbuf,strlen(rbuf),0);
					}
				}
			}
		}
	}
	close(sfd);
	return 0;
}

多路复用poll实现客户端

#include <header.h>
#define SER_PORT 8888
#define SER_IP "192.168.125.156"
#define CLI_PORT 8889
#define CLI_IP "192.168.125.156"
int main(int argc, const char *argv[])
{
	//1.创建用于通讯的套接字文件描述符
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd==-1)
	{
		perror("socket");
		return -1;
	}
	printf("cfd=%d\n",cfd);
	//2.绑定ip地址和端口号
	//填充客户端地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;
	cin.sin_port=htons(CLI_PORT);
	cin.sin_addr.s_addr=inet_addr(CLI_IP);
	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
	{
		perror("bind");
		return -1;
	}
	printf("bind success\n");
	//3.连接服务器
	//3.1准备对端地址结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(SER_PORT);
	sin.sin_addr.s_addr=inet_addr(SER_IP);
	//3.2连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("connect");
		return -1;
	}
	printf("connect success\n");
	//使用poll完成多个阻塞任务的并发执行
	struct pollfd pfd[2];
	pfd[0].fd=0;
	pfd[0].events=POLLIN;
	pfd[1].fd=cfd;
	pfd[1].events=POLLIN;
	//4.数据收发
	char wbuf[128]={0};
	char rbuf[128]={0};
	while(1)
	{
		int res=poll(pfd,2,-1);
		if(res==-1)
		{
			perror("poll");
			return 1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return 1;
		}
		//当程序执行到这，说明容器中必有事件产生
		//如果是0（输入）产生的，就执行输入
		if(pfd[0].revents==POLLIN)
		{
			//从终端上获取发送的数据
			fgets(wbuf,sizeof(wbuf),stdin);
			wbuf[strlen(wbuf)-1]=0;
			send(cfd,wbuf,strlen(wbuf),0);
			printf("send success\n");
		}
		//如果是cfd产生的，就执行接受服务器发来的消息
		if(pfd[1].revents==POLLIN)
		{
			bzero(rbuf,sizeof(rbuf));
			recv(cfd,rbuf,sizeof(rbuf),0);
			printf("the service message is:%s\n",rbuf);
		}
	}
	//5.关闭套接字
	close(cfd);
	return 0;
}