网络编程 day5

select

#include <my.h>
#define SER_IP "192.168.125.178"
#define SER_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;
	}
	printf("socket success sfd=%d\n",sfd);
	//绑定IP地址和段口号
	//填充地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family =AF_INET;  //地址族
	sin.sin_port = htons(SER_PORT);  //端口号
	sin.sin_addr.s_addr =inet_addr(SER_IP); //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");
	//4 阻塞等待客户端连接请求
	//定义地址信息结构体变量用于接受客户端的地址信息
	struct sockaddr_in cin;
	socklen_t socklen =sizeof(cin);  //接受地址信息的长度
	//1创建一个文件描述符检测集合
	fd_set readfds,tempfds;
	//2清除容器中的内容
	FD_ZERO(&readfds);
	//3 将0号和sfd文件描述符放入集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	while(1)
	{
		tempfds=readfds;
		//4阻塞检测文件描述符集合是否有事情产生
		int res=select(sfd+1,&tempfds,NULL,NULL,NULL);
		if(res==0)
		{
			printf("time out\n");
			return -1;
		}else if(res==-1)
		{
			perror("select error");
			return -1;
		}
		//程序执行到此，说明检测的集合中有事件产生了，要判断那个文件描述符还在
		if(FD_ISSET(sfd,&tempfds))
		{
			int newfd =accept(sfd,(struct sockaddr*)&cin,&socklen);
			if(newfd==-1)
			{
				perror("accept error");
				return -1;
			}
			printf("[%s:%d]:已连接,newfd =%d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		}
		//判断0号文件描述符是否在集合中
		if(FD_ISSET(0,&tempfds))
		{
			//另一个阻塞事件
			char buf[128]="";
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;
			printf("触发了键盘输入事件：%s\n",buf);
		}
		/*

		//5数据通信
		char rbuf[128]="";    //接受消息
		while(1)
		{
		bzero(rbuf,sizeof(rbuf));// 清空容器
		//从套接字文件中读取消息
		int ret =read(newfd,rbuf,sizeof(rbuf));
		if(ret==0)
		{
		printf("客户端已下线\n");
		break;
		}
		printf("[%s:%d]:%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),rbuf);
		//将字符串加个笑脸回去
		strcat(rbuf,"*-*");
		write(newfd,rbuf,strlen(rbuf));
		printf("发送成功\n");
		}
		close(newfd);
		*/
	}
		close(sfd);
		return 0;
	}

poll

#include <my.h>
#define SER_IP "192.168.125.178"
#define SER_PORT 8888
#define CLI_IP "192.168.125.67"
#define CLI_PORT 9999
int main(int argc,const char*argv[])
{
	//创建套接字文件描述符
	int cfd =socket(AF_INET,SOCK_STREAM,0);
	if(cfd==-1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success cfd =%d\n",cfd);
	//设置端口号快速重用
	int reuse =1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
	{
		perror("setsockopt error");
		return -1;
	}
	printf("端口号快速重用成功\n");
	//2绑定IP和段口号
	//填充地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family =AF_INET;
	cin.sin_port =htons(CLI_PORT);
	cin.sin_addr.s_addr =inet_addr(CLI_IP);
	//绑定工作
	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");
	//连接服务器
	//3.1填充要连接的服务器地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;//协议族
	sin.sin_port =htons(SER_PORT);//服务器端口号
	sin.sin_addr.s_addr=inet_addr(SER_IP);//服务器IP
	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("connet error");
		return -1;
	}
	printf("connect success\n");
	//创建一个用于检测文件描述符的容器
	struct pollfd pfds[2];
	pfds[0].fd=0;  //表示第一个元素检测0号文件描述符
	pfds[0].events =POLLIN;    //要检测0号文件描述符的读事件
	pfds[1].fd=cfd;           //表示第二个元素检测cfd文件的描述符
	pfds[1].events=POLLIN;    //要检测cfd文件描述符的读事件

	//4数据收发
	char wbuf[128]="";
	while(1)
	{
		int res =poll(pfds,2,-1);
		if(res==0)
		{
			printf("time out\n");
			return -1;
		}else if(res==-1)
		{
			perror("poll error");
			return -1;
		}
		if(pfds[0].revents==POLLIN)
		{
			//从终端获取数据
			fgets(wbuf,sizeof(wbuf),stdin);
			wbuf[strlen(wbuf)-1]='\0';
			if(strcmp(wbuf,"quit")==0)
			{
				break;
			}
			//将数据发送给服务器
			send(cfd,wbuf,strlen(wbuf),0);
		}
		//判断是否为服务器发来消息
		if(pfds[1].revents==POLLIN)
		{
			//接收服务器发来的消息
			bzero(wbuf,sizeof(wbuf));
			recv(cfd,wbuf,sizeof(wbuf),0);
			printf(" 收到的服务器消息为：%s\n",wbuf);
		}
	}
	close(cfd);
	return 0;
}