华清远见22071班9.20作业

1.tcp的select服务器代码

#include<sys/socket.h>
#include<sys/types.h>
#include<stdio.h>
#include<unistd.h>
#include<string.h>
#include<arpa/inet.h>
#include<sys/time.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)
int updata(int maxfd,fd_set readfd);

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;
	}
	//填充ip和端口信息到结构体中
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 						//地址族必须是AF_INET
	sin.sin_port   		= htons(8888); 					//端口号的网络字节序（1024~49151）
	sin.sin_addr.s_addr = inet_addr("192.168.31.109");  //ubuntu 的本机地址,用桥接
	//绑定服务器的ip地址和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	//将套接字设置为被动监听状态
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	fd_set readfd,tempfd;
	FD_ZERO(&readfd);
	FD_ZERO(&tempfd);

	FD_SET(0,&readfd);
	FD_SET(sfd,&readfd);

	int maxfd=sfd;
	int st_res=0;

	char buf[128]="";
	ssize_t res;
	int newfd=0;
	

	//获取新的文件描述符,该描述符用于通信
	struct sockaddr_in a;
	int size=sizeof(a);
	struct sockaddr_in str[1024-4];

	while(1)
	{
		tempfd=readfd;
		st_res=select(maxfd+1,&tempfd,NULL,NULL,NULL);
		if(st_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		if(st_res==0)
		{
			printf("超时\n");
			break;
		}
		int i=0;
		for(i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tempfd)==0)
			{
				continue;
			}
			if(0==i)
			{
				int sendfd; 
				bzero(buf, sizeof(buf));
				res = scanf("%d %s", &sendfd, buf);
				while(getchar()!=10);
				if(res != 2)
				{
					fprintf(stderr, "请输入正确格式： fd string\n");
					continue;
				}
				if(sendfd <=3 || FD_ISSET(sendfd, &readfd)==0)
				{
					fprintf(stderr, "输入的文件描述符：%d 有误\n", sendfd);
					continue;
				}
				if(send(sendfd, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("发送成功\n");
			}
			else if(sfd==i)
			{
				printf("客户端链接\n");
				newfd=accept(sfd,(struct sockaddr*)&a,&size);
				if(newfd<0)
				{
					ERR_MSG("accept");
					return -1;
				}
				printf("[%d]newfd=%d\n",ntohs(a.sin_port),newfd);
				FD_SET(newfd,&readfd);
				maxfd=maxfd>newfd?maxfd:newfd;
				str[newfd-4]=a;
			}
			else
			{
				bzero(buf,sizeof(buf));
				//接收
				res=recv(i,buf,sizeof(buf),0);
				if(res<0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(res==0)
				{
					printf("%d客户端关闭\n",i);
					close(i);
					FD_CLR(i,&readfd);
					maxfd=updata(maxfd,readfd);
				}
				else
				{
				printf("[%s:%d]%d客户端:%s\n",inet_ntoa(str[i-4].sin_addr),str[i-4].sin_port,i,buf);
				}
			}
		}
	}

	close(sfd);
	return 0;
}


int updata(int maxfd,fd_set readfd)
{
	for(int j=maxfd;j>=0;j--)
	{
		if(FD_ISSET(j,&readfd))
		{
			return j;
		}
	}
	return 0;
}

2.tcp的select 客户端代码

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)
int main(int argc, const char *argv[])
{
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}

	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(8888);
	sin.sin_addr.s_addr=inet_addr("192.168.31.109");

	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	fd_set readfd,tempfd;
	FD_ZERO(&readfd);
	FD_ZERO(&tempfd);
	FD_SET(0,&readfd);
	FD_SET(sfd,&readfd);

	char buf[128]="";
	ssize_t res;
	int st_res=0;
	while(1)
	{
		tempfd=readfd;
		st_res=select(sfd+1,&tempfd,NULL,NULL,NULL);
		if(st_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		if(st_res==0)
		{
			printf("超时\n");
			break;
		}

		if(FD_ISSET(0,&tempfd))
		{
			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("发送成功\n");
		}	
		if(FD_ISSET(sfd,&tempfd))	
		{
			bzero(buf,sizeof(buf));
			res=recv(sfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{
				printf("对端关闭\n");
				break;
			}
			printf(":%s\n",buf);
		}
	}
	close(sfd);
	return 0;
}

 3.tcp的poll客户端代码

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>
#include<poll.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)
int main(int argc, const char *argv[])
{
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}

	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(8888);
	sin.sin_addr.s_addr=inet_addr("192.168.31.109");

	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	struct pollfd fds[2];
	fds[0].fd=0;
	fds[0].events=POLLIN;
	fds[1].fd=sfd;
	fds[1].events=POLLIN;
	char buf[128]="";
	ssize_t res;
	int pl_res=0;
	while(1)
	{
		pl_res=poll(fds,2,-1);
		if(pl_res<0)
		{
			ERR_MSG("poll");
			return -1;
		}
		else if(0==pl_res)
		{
			printf("time out \n");
			break;
		}

		if((fds[0].revents&POLLIN)!=0)
		{
			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("发送成功\n");
		}
		if((fds[1].revents&POLLIN)!=0)
		{
			bzero(buf,sizeof(buf));
			res=recv(sfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{
				printf("对端关闭\n");
				break;
			}
			printf(":%s\n",buf);
		}
	}
	close(sfd);
	return 0;
}