11.23

select实现客户端代码

#include<myhead.h>
#define SERIP "192.168.114.125"
#define SERPORT 8888
#define CLIIP "192.168.114,125"
#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);    //服务器IP地址
	//2.2绑定工作
	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
	{
		perror("bind error");
		return -1;
	}
	//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");
 
	定义一个用来检测文件描述符的集合
	fd_set readfds;
 
	清空容器中的内容
	FD_ZERO(&readfds);
	将要检测的文件描述符放入集合中
	FD_SET(cfd,&readfds);
	FD_SET(0,&readfds);
 
	int res=0;//接收select的返回值
	//4.收发数据
	char buf[128]="";
	char rbuf[128]="";
 
	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;
		}else
		{
			bzero(buf,sizeof(buf));
			bzero(rbuf,sizeof(rbuf));
 
				fgets(buf,sizeof(buf),stdin);    //从标准输入中读取数据
				buf[strlen(buf)-1]='\0';
				//将数据发送给服务器
				send(cfd,buf,sizeof(buf),0);
				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<myhead.h>
#define PORT 8888
#define IP "192.168.114.125"
int main(int argc, const char *argv[])
{
	//1.创建用于接收连接的套接字
	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;
	}	//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("bind error");
		return -1;
	}
	//3.将套接字设置为被动监听
	if( listen(sfd,128)==-1)
	{
		perror("listen error");
		return -1;
	}
	//4.阻塞等待接收连接
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;
	socklen_t socklen=sizeof(cin);
	int newfd;
	//文件描述符集合数组
	struct pollfd fds[1024];
	fds[0].fd=0;
	fds[0].events=POLLIN;
	fds[1].fd=sfd;
	fds[1].events=POLLIN;
	struct sockaddr_in save[1024];
	int res=0;
	char buf[128]="";
	int maxfd=sfd;
	while(1)
	{ 
		res=poll(fds,maxfd+1,-1);
			if(res == -1)
			{
				perror("poll error");
				return -1;
			}
			else if(res == 0)
			{
				printf("timeout\n");
				return -1;
			}
 
		if(fds[1].revents==POLLIN)
		{	if((newfd=accept(sfd,(struct sockaddr *)&cin,&socklen))==-1)
			{
				perror("accept error");
				return -1;
			}
			printf("accept success newfd=%d\n",newfd);
			fds[newfd].fd=newfd;	
			fds[newfd].events=POLLIN;
			save[newfd]=cin;
			if(newfd>maxfd)
			{	
                maxfd=newfd;
            }
			
		}
		if(fds[0].revents==POLLIN)
		{
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]='\0';
			printf("触发终端输入:%s\n",buf);
			for(int i=4;i<=maxfd;i++)
			{send(fds[i].fd,buf,sizeof(buf),0);}
 
 
		}
		for(int i=4;i<maxfd+1;i++)
		{   if(fds[i].revents==POLLIN)
			{
				char buf[128]="";
				bzero(buf,sizeof(buf));
				int res= recv(i,buf,sizeof(buf),0);
				if(res == 0)
				{
					printf("客户端下线\n");
					close(fds[i].fd);
					fds[i].fd=-1;
					for(int j=maxfd;j>3;j--)
					{  			
						if(fds[j].fd!=-1)
						{
						maxfd=fds[j].fd;
						break;
						}
					}
					continue;
				}
 
				else if(res<0)
				{
					perror("recv error");
					return -1;
				}
				                printf("%s:%d]:%s\n",inet_ntoa(save[i].sin_addr),ntohs(save[i].sin_port),buf);	
			}
		}
 
 
 
	}
 
//6.关闭套接字
close(sfd);
	return 0;