8.15作业

select的TCP服务器

#include
#include
#include
#include
#define PORT 7777
#define IP "192.168.112.89"
int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	//设置允许端口被快速复用
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;     //必须填AF_INET
	sin.sin_port=htons(PORT);   //端口号:1024~49151
	sin.sin_addr.s_addr=inet_addr(IP);	//本机IP ifconfig
	//绑定服务器的IP和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	//将套接字设置为被动监听状态
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	struct sockaddr_in cin;//存储客户端的地址信息
	socklen_t addrlen=sizeof(cin);
	//从已完成链接的队列中获取一个客户端信息，生成一个新的文件描述符
	//该文件描述符才是与客户端通信的文件描述符
	fd_set readfds,rreadfds;
	FD_ZERO(&readfds);
	FD_ZERO(&rreadfds);
	FD_SET(0,&rreadfds);
	FD_SET(sfd,&rreadfds);
	char buf[128]="";
	int s_res=-1;
	int maxfd=sfd;
	int newfd=-1;
	struct sockaddr_in saveCin[1024-3];
	ssize_t res=0;
	while(1)
	{
		readfds=rreadfds;
		s_res=select(maxfd+1,&readfds,NULL,NULL,NULL);
		if(s_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res)
		{
			printf("time out\n");
			break;
		}
		for(int i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&readfds)==0)
				continue;
			if(0==i)
			{
				printf("输入事件触发\n");
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				printf("%s\n",buf);
			}
			else if(sfd==i)
			{
				printf("客户端连接事件触发\n");
				newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
				if(newfd<0)
				{
					ERR_MSG("accept");
					return -1;
				}
				printf("[%s : %d]newfd=%d 客户端连接成功\n",\
						inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
				FD_SET(newfd,&rreadfds);
				maxfd=maxfd>newfd?maxfd:newfd;
				saveCin[newfd-3]=cin;
			}
			else if(FD_ISSET(i,&readfds))
			{
				printf("触发交互事件\n");
				bzero(buf,sizeof(buf));
				//接收数据
				res=recv(i,buf,sizeof(buf),0);
				if(res<0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(0==res)
				{
					printf("[%s : %d]newfd=%d 客户端下线\n",\
							inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i);
					close(i);
					FD_CLR(i,&rreadfds);
					while(FD_ISSET(maxfd,&rreadfds)==0&&maxfd-->=0);
					continue;
				}
				printf("[%s : %d]newfd=%d : %s\n",\
						inet_ntoa(saveCin[i-3].sin_addr),ntohs(saveCin[i-        3].sin_port),i,buf);
				if(send(i,buf,sizeof(buf),0)<0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");

			}
		}
	}

	close(sfd);
	return 0;
}

select的TCP客户端

#include
#include
#include
#include
#define PORT 7777
#define IP "192.168.112.89"
int main(int argc, const char *argv[])
{
	//创建套接字
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;     //必须填AF_INET
	sin.sin_port=htons(PORT);   //端口号:1024~49151
	sin.sin_addr.s_addr=inet_addr(IP);	//本机IP ifconfig
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");
	fd_set readfds,rreadfds;
	FD_ZERO(&readfds);
	FD_ZERO(&rreadfds);
	FD_SET(0,&rreadfds);
	FD_SET(cfd,&rreadfds);
	int maxfd=cfd;
	int s_res=-1;
	char buf[128]="";
	size_t res=0;
	while(1)
	{
		readfds=rreadfds;
		s_res=select(cfd+1,&readfds,NULL,NULL,NULL);
		if(s_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res)
		{
			printf("time out\n");
			break;
		}
		if(FD_ISSET(0,&readfds))
		{
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;
			if(send(cfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
		}
		else if(FD_ISSET(cfd,&readfds))
		{
			res=recv(cfd,buf,sizeof(buf),0);
			if(res<0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(0==res)
				{
					printf("[%s : %d]newfd=%d 服务器掉线\n",IP,PORT,cfd);
					break;
					close(cfd);
					FD_CLR(cfd,&rreadfds);
					while(FD_ISSET(maxfd,&rreadfds)==0&&maxfd-->=0);
					continue;
				}
				printf("[%s : %d]newfd=%d : %s\n",IP,PORT,cfd,buf);
			}
		}
	close(cfd);
	return 0;
}

搭建poll客户端

#include
#include
#include
#include
#define PORT 7777
#define IP "192.168.112.89"
int main(int argc, const char *argv[])
{
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("cfd=%d\n",cfd);
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success");
	struct pollfd fds[2];
	fds[0].fd=0;
	fds[0].events=POLLIN;
	fds[1].fd=cfd;
	fds[1].events=POLLIN;

	char buf[128]="";
	ssize_t res=0;
	int p_res=-1;
	while(1)
	{
		p_res=poll(fds,2,-1);
		if(p_res<0)
		{
			ERR_MSG("poll");
			return -1;
		}
		else if(0==p_res)
		{
			printf("time out\n");
			break;
		}
		if((fds[0].revents&POLLIN)!=0)
		{
			printf("键盘触发事件\n");
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;
			if(send(cfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		if(fds[1].revents&POLLIN)
		{
			printf("触发服务器交互事件\n");
			bzero(buf,sizeof(buf));
			res=recv(cfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{
				printf("[%s : %d]cfd=%d 服务器掉线\n",IP,PORT,cfd);
				break;
			}
			printf("[%s : %d]cfd=%d %s\n",IP,PORT,cfd,buf);
		}
	}
	return 0;
}