华清远见嵌入式学习——网络编程——作业4

作业要求：①使用IO多路复用中的select函数实现TCP并发服务器客户端

                  ②使用IO多路复用中的poll函数实现TCP并发服务器的服务器端

一、

代码

#include <myhead.h>

#define SERPORT 8888              //服务器端口号
#define SERIP "192.168.114.113"       //服务器IP地址

int main(int argc, const char *argv[])
{
	//创建用于通信的套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd == -1)
	{
		perror("socket error");
		return -1;
	}

	//连接服务器
	///填充服务器地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(SERPORT);
	sin.sin_addr.s_addr = inet_addr(SERIP);

	///连接服务器
	if(connect(cfd,(struct sockaddr *)&sin,sizeof(sin)) == -1)
	{
		perror("connect error");
		return -1;
	}
	
	//创建用于检测文件描述符的集合
	fd_set readfds,tempfds;
	
	//清空集合
	FD_ZERO(&readfds);
	
	//将要检测的文件描述符放入集合中
	FD_SET(cfd,&readfds);
	FD_SET(0,&readfds);

	int res = 0;    //接收select的返回值
	int maxfd = cfd;  //集合中值最大的文件描述符

	//向服务器进行数据的收发
	char buf[128] = "";
	int ret = 0;    //接收recv的返回值
	while(1)
	{
		tempfds = readfds;
	
		res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(res == -1)
		{
			perror("select error");
			return -1;
		}else if(res == 0)
		{
			printf("time out\n");
			return -1;			
		}
		
		//遍历集合中所有的文件描述符
		for(int i = 0;i <= maxfd;i++)
		{
			//判断当前文件描述符是否在集合中
			if(!FD_ISSET(i,&readfds))
			{
				continue;
			}
	
			
			//判断0号文件描述符是否还在集合中
			if(0 == i)
			{
				//从标准输入中读取数据
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1] == 0;

				//将数据发送到服务器
				if(send(cfd,buf,sizeof(buf),0) == -1)
				{
					perror("send error");
					return -1;
				}

			}else if(cfd == i)     //判断cfd是否还在集合中
			{
				//接收来自服务器的消息
				ret = recv(cfd,buf,sizeof(buf),0);
				if(ret == -1)
				{
					perror("recv error");
					return -1;
				}else if(ret == 0)
				{
					printf("服务器已关闭\n");
					return -1;
				}
				
				printf("服务器消息：%s\n",buf);
			}
		}
	}
	
	//关闭文件描述符
	close(cfd);
	
	return 0; 
}

效果图

二、

代码

#include <myhead.h>

#define IP "192.168.114.118"
#define PORT 8888

int main(int argc, const char *argv[])
{
	//创建用于连接的套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd == -1)
	{
		perror("socket error");
		return -1;
	}

	//设置端口号快速重用
	int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
	{
		perror("setsockopt error");
		return -1;
	}

	//绑定服务器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)) == -1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");

	//将连接用套接字设置为被动监听状态
	if(listen(sfd,128) == -1)
	{
		perror("listen error");
		return -1;
	}
	printf("listen success\n");

	//定义一个集合管理sfd和打开的通信用文件描述符
	struct pollfd fds[1024];
	int maxfd = 0;


	//手动放入sfd
	fds[0].fd = sfd;
	fds[0].events = POLLIN;     //表明为读事件

	//将fds中其余元素初始化为-1
	for(int i = 4;i <= 1024;i++)
	{
		fds[i].fd = -1;
	}

	//填充客户端地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family = AF_INET;
	socklen_t socklen = sizeof(cin);


	char cbuf[128] = "";  //给客户端用的容器
	int nfd;
	int res = 0;  //接收poll返回的结果
	while(1)
	{
		res = poll(fds,maxfd+1,-1);
		if(res == -1)
		{
			perror("select");
			return -1;
		}
		else if(res == 0)
		{
			continue;;
		}
		else if(res > 0)                //说明检测到了有文件描述符对应的缓冲区的数据发生了改变
		{
			if(fds[0].revents ==  POLLIN)    //表明有新的客户连接进来了
			{
				int nfd = accept(sfd,(struct sockaddr*)&cin,&socklen);  //阻塞在此处，直到有客户端连接上来
				if(nfd == -1)   //增加这些错误的判断非常重要，可以帮助找到出现问题的地方
				{
					perror("accept");
					return -1;
				}

				//将新的文件描述符加入到集合中
				for(int i = 1;i < 1024;i++)
				{
					if( fds[i].fd == -1)
					{
						fds[i].fd = nfd;
						fds[i].events = POLLIN;
						break;
					}
				}

				//更新最大的文件描述符
				if(nfd > maxfd)
				{
					maxfd = nfd;
				}
			}

			for(int i = 1;i <= maxfd;i++)     //轮询客户端对应的文件描述符
			{
				if(fds[i].revents == POLLIN)  //说明此文件描述符对应的客户端发送来了数据
				{
					int ret = read(fds[i].fd,cbuf,sizeof(cbuf));
					if(ret == -1)
					{
						perror("read");
						exit(-1);
					}
					else if(ret == 0)
					{
						printf("client closed\n");
						close(fds[i].fd);   //关闭对应的文件描述符
						fds[i].fd = -1;   //在fds中清空对应的文件描述符
					}
					else if(ret > 0)
					{
						printf("read buf = %s\n",cbuf);
						write(fds[i].fd,cbuf,strlen(cbuf)+1);
					}


				}
			}
		}
	}
	//关闭所有套接字
	close(sfd);

	return 0;

}

效果图