11月22日作业

select客户端

#include <myhd.h>
#define SERIP "192.168.114.118"
#define SERPORT 8888
#define CLIIP "192.168.114.118"
#define CLIPORT 6666
int main(int argc, const char *argv[])
{
	//1、创建客户端用于通信的套接字
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd==-1)
	{
		perror("socket error");
		return -1;
	}
	printf("cfd=%d\n",cfd);
	//2、绑定
	//2.1、填充地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;  //使用ipv4通信
	cin.sin_port=htons(CLIPORT);  //服务器端口号
	cin.sin_addr.s_addr=inet_addr(CLIIP);
	//2.2、绑定工作
	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");
	//3、连接服务器
	//3.1、填充服务器地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;  //使用ipv4通信
	sin.sin_port=htons(SERPORT);  //服务器端口号
	sin.sin_addr.s_addr=inet_addr(SERIP);  //服务器ip地址
	//3.2、连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("connect error");
		return -1;
	}
	printf("connect success\n");
	//4、收发数据
	char buf[128]="";
	char rbuf[128]="";

	int res=0;   //接收select的返回值

	定义用于检测文件描述符的集合
	fd_set readfds;
	清空容器中的内容
	FD_ZERO(&readfds);
	将要检测的文件描述符放入集合中
	FD_SET(cfd,&readfds);
	FD_SET(0,&readfds);
	while(1)
	{
		res=select(cfd+1,&readfds,NULL,NULL,NULL);
		if(res==-1)
		{
			perror("select error");
			return -1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return -1;
		}

		bzero(buf,sizeof(buf));
		bzero(rbuf,sizeof(rbuf));
		//printf("请输入：");
		fgets(buf,sizeof(buf),stdin);  //从标准输入中读取数据
		buf[strlen(buf)-1]='\0';
		//将数据发送给服务器
		send(cfd,buf,sizeof(buf),0);
		//输入quit退出
		if(strcmp(buf,"quit")==0)
		{
			break;
		}
		//接收服务器发送来的消息
		int res=recv(cfd,rbuf,sizeof(rbuf),0);
		if(res==0)
		{
			printf("服务器已经关闭\n");
			break;
		}
		printf("rbuf=%s\n",rbuf);
	}
	//5、关闭客户端套接字文件
	close(cfd);
	return 0;
}

poll服务器

#include <myhd.h>
#define PORT 8888     //端口号
#define IP "192.168.114.118"  //IP地址
int main(int argc, const char *argv[])
{
	//1、创建用于接受链接的套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd==-1)
	{
		perror("socket error");
		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 error");
		return -1;
	}
	printf("设置端口快速重用成功\n");

	//2、绑定IP地址和端口号
	//2.1、填充要绑定的地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	//2.2、绑定
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("bing error");
		return -1;
	}
	printf("bing success _%d_%s_%s_\n",__LINE__,__FILE__,__func__);
	//3、将套接字设置成被动监听状态
	if(listen(sfd,128)==-1)
	{
		perror("listen error");
		return -1;
	}
	printf("listen success _%d_%s_%s_\n",__LINE__,__FILE__,__func__);
	//4、阻塞等待客户端链接请求，如果有新的客户端连接，则创建一个新的用于通信的套接字
	//4.1、定义客户端地址信息结构体
	struct sockaddr_in cin;   //客户端地址信息结构体
	cin.sin_family=AF_INET;
	socklen_t socklen=sizeof(cin);  //客户端地址信息大小

	//5、收发数据使用newfd完成通信
	char buf[128]="";
	定义一个集合管理文件描述符数组
	struct pollfd fds[1024];
	//将0号文件描述符放入
	fds[0].fd=0;
	fds[0].events=POLLIN;  //表明要进行读事件
	//将sfd放入集合
	fds[3].fd=sfd;
	fds[3].events=POLLIN; 

	int res=0;  //接收poll返回的结果
	int newfd=-1;  //存放用于最新连接客户端的套接字
	int i=4;

	while(1)
	{
		res=poll(fds,1024,-1);  //1和2不放
		if(res<0)
		{
			perror("poll error");
			return -1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return -1;
		}
		for(int j=0;j<=(i<newfd?newfd:i);j++)
		{
			printf("for循环开始\n");
			if(j==sfd)
			{
				//4.2、阻塞接收客户端的连接请求，并且获取客户端的地址信息
				newfd=accept(sfd,(struct sockaddr*)&cin,&socklen);
				if(newfd==-1)
				{
					perror("accept error");
					return -1;
				}
				printf("accept success _%d_%s_%s_\n",__LINE__,__FILE__,__func__);
				i=newfd;
				fds[i].fd=newfd;
				fds[i].events=POLLOUT;  //写事件
			}
			else if(j==0)
			{
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]='\0';
				printf("触发终端输入事件：%s\n",buf);
			}
			else if(j==1||j==2)
			{
				continue;
			}
			else
			{
				//清空字符串
				bzero(buf,sizeof(buf));
				//read(newfd,buf,sizeof(buf));  //从套接字中读取客户端发来的消息
				int ret=recv(i,buf,sizeof(buf),0);
				//判断收到的结果
				if(ret==0)
				{
					printf("客户端已经下线\n");
					break;
				}
				else if(ret<0)
				{
					perror("recv error");
					return -1;
				}
				printf("[%s：%d]：%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),buf);
				//将读取的信息加上一些字符发送回去
				strcat(buf,"*_*");
				//write(newfd,buf,sizeof(buf));
				send(i,buf,sizeof(buf),0);
			}
		}
	}
	//6、关闭所有套接字
	close(sfd);
	//close(newfd);
	return 0;
}