# 2023/3/16 select与poll函数练习

于 2023-03-16 22:13:36 发布
阅读量35 收藏
点赞数
文章标签： 网络 c语言 开发语言 ubuntu tcp/ip
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/L1051878/article/details/129601765
版权

2023/3/16 select与poll函数练习

select
客户端

#include "head.h"

#define SER_PORT 2222
#define SER_IP "192.168.31.59"

#define CLI_PORT 5555
#define CLI_IP "192.168.31.59"

int main(int argc, const char *argv[])
{
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	printf("socket create success __%d__\n",__LINE__);

	//客户端地址信息结构体
	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)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}

	printf("bind success __%d__\n",__LINE__);

	//服务器端地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family         = AF_INET;
	sin.sin_port           = htons(SER_PORT);
	sin.sin_addr.s_addr    = inet_addr(SER_IP);
	socklen_t addrlen      = sizeof(sin);

	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,addrlen) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}

	printf("connect success __%d__\n",__LINE__);
	
	//设置集合
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	//将需要的文件描述符填充到集合中
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);


	int s_res = 0;
	ssize_t res = -1;
	char buf[128] = "";

	while(1)
	{
		tempfds = readfds;

		//让内核去监测集合中的文件描述符是否准备就绪
		s_res = select(cfd+1,&tempfds,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,&tempfds))
		{
			//发送数据
			memset(buf,0,sizeof(buf));
			printf("请输入：");
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf) - 1] = 0;

			if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				return-1;
			}else
			{
				printf("send success\n");
			}
			if(strcmp(buf,"quit") == 0)
			{
				printf("客户端退出\n");
				break;
			}

			printf("send success __%d__\n",__LINE__);
		}
		if(FD_ISSET(cfd,&tempfds))
		{
			//接收数据
			memset(buf,0,sizeof(buf));
			res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
			if(res < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}else if(0 == res)
			{
				printf("服务器端关闭\n");
				break;
			}
			printf("[%s : %d] buf = %s __%d__\n",\
					inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf,__LINE__);
		}
	}
	//关闭所有文件描述符
	close(cfd);


	return 0;
}

服务端

#include "head.h"

#define SER_PORT 2222   //1024~49151
#define SER_IP "192.168.31.59" //IP地址，本机IP

#define CLI_PORT 5555
#define CLI_IP "192.168.31.59"

int deal_cli_msg(int newfd,struct sockaddr_in cin);

int main(int argc, const char *argv[])
{

	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd = %d __%d__\n",sfd,__LINE__);

	//设置允许端口被快速服用
	int resue = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&resue,sizeof(resue)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("setsockopt success __%d__\n",__LINE__);

	//填充服务器的地址信息结构体
	//真实的地址信息结构体根据地址族执行，AF_INET：man 7 ip
	struct sockaddr_in sin;
	sin.sin_family     = AF_INET;//必须填AF_INET
	sin.sin_port       = htons(SER_PORT);//端口号的网络字节序，1024~49151
	sin.sin_addr.s_addr= inet_addr(SER_IP);//IP地址,ifconfig

	//绑定--必须绑定
	if(bind (sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}

	printf("bind success __%d__\n",__LINE__);


	//存储连接成功的客户端的地址信息
	struct sockaddr_in cin;
	cin.sin_family     = AF_INET;//必须填AF_INET
	cin.sin_port       = htons(CLI_PORT);//端口号的网络字节序，1024~49151
	cin.sin_addr.s_addr= inet_addr(CLI_IP);//IP地址,ifconfig
	socklen_t addtlen = sizeof(cin);


	//将套接字设置为被动监听状态
	if(listen(sfd,128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);

	fd_set readfds,tempfds;//设置集合
	FD_ZERO(&readfds);//清空集合
	FD_ZERO(&tempfds);//清空集合

	//将需要的文件描述符填充到集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);

	int s_res = 0;
	int newfd = -1;
	int maxfd = sfd;
	ssize_t res = -1;
	char buf[128] = "";
	struct sockaddr_in saveCin[1024-4];

	while(1)
	{
		tempfds = readfds;

		//让内核去监测集合中的文件描述符是否准备就绪
		s_res = select(maxfd+1,&tempfds,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,&tempfds) == 0)
			{
				continue;
			}

			if(0 == i)
			{
				int sndfd;
				res = scanf("%d %s",&sndfd,buf);
				while(getchar() != 10);
				if(res != 2)
				{
					fprintf(stderr,"输入格式错误\n");
					continue;
				}
				if(sndfd >= 1024 || sndfd <= sfd || !FD_ISSET(sndfd,&readfds))
				{
					fprintf(stderr,"sndfd = %d 没有对应客户端\n",sndfd);
					continue;
				}
				if(send(sndfd,buf,sizeof(buf),0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
			}
			else if(sfd == i)
			{

				newfd = accept(sfd,(struct sockaddr*)&cin,&addtlen);
				if(newfd < 0)
				{
					ERR_MSG("accept");
					return -1;
				}

				printf("[%s  :  %d]  newfd = %d 客户端连接成功\n",\
						inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

				saveCin[newfd-4] = cin;
				//将新生成的文件描述符newfd，添加到集合中
				FD_SET(newfd,&readfds);

				//更新maxfd
				maxfd = maxfd>newfd?maxfd:newfd;
			}else
			{
				printf("客户发送信息\n");
				//接收
				memset(buf,0,sizeof(buf));
				res = recv(i,buf,sizeof(buf),0);
				if(res < 0)
				{
					ERR_MSG("recv");
				}else if(0 == res)
				{
					fprintf(stderr,"[%s : %d] newfd = %d 客户端断开连接\n",\
							inet_ntoa(saveCin[i-4].sin_addr),ntohs(saveCin[i-4].sin_port),i);
					//关闭文件描述符
					close(i);
					//从集合中移除
					FD_CLR(i,&readfds);
					//更新maxfd
					for(int j = maxfd;i>sfd;j--)
					{
						if(FD_ISSET(j,&readfds))
						{
							maxfd = j;
							break;
						}
					}

				}else
				{

					fprintf(stderr,"[%s : %d] newfd = %d buf = %s\n",\
							inet_ntoa(saveCin[i-4].sin_addr),ntohs(saveCin[i-4].sin_port),i,buf);
					//发送,将数据拼接一个*_*发送回去
					strcat(buf,"*_*\n");
					if(send(i,buf,sizeof(buf),0) < 0)
					{
						ERR_MSG("send");
						return -1;
					}
					printf("send success\n");
				}
			}
		}
	}

	//关闭所有套接字文件描述符
	close(sfd);
	return 0;
}

int deal_cli_msg(int newfd,struct sockaddr_in cin)
{
	char buf[128] = "";
	ssize_t res = -1;
	while(1)
	{
		//接收
		memset(buf,0,sizeof(buf));
		res = recv(newfd,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),newfd);
			break;
		}
		if(strcmp(buf,"quit") == 0)
		{
			printf("退出服务器\n");
			break;
		}
		printf("[%s : %d] newfd = %d buf = %s\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);


		//发送,将数据拼接一个*_*发送回去
		strcat(buf,"*_*\n");
		if(send(newfd,buf,sizeof(buf),0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");
	}
	return 0;
}

poll
客户端

#include "head.h"

#define SER_PORT 2222
#define SER_IP "192.168.31.59"

#define CLI_PORT 5555
#define CLI_IP "192.168.31.59"

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	printf("socket create success cfd = %d __%d__\n",cfd,__LINE__);

	//填充客户器的地址信息结构体
	//真实的地址信息结构体根据地址族执行，AF_INET：man 7 ip
	struct sockaddr_in cin;
	cin.sin_family     = AF_INET;//必须填AF_INET
	cin.sin_port       = htons(CLI_PORT);//客户端的端口号的网络字节序，1024~49151
	cin.sin_addr.s_addr= inet_addr(CLI_IP);//IP地址,ifconfig

	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);

	//填充服务器自身的地址信息结构体
	//给connect函数使用，指定要连接到哪个服务器
	struct sockaddr_in sin;
	sin.sin_family        = AF_INET;//必须填AF_INET
	sin.sin_port          = htons(SER_PORT);//服务器端口号的网络字节序
	sin.sin_addr.s_addr   = inet_addr(SER_IP);//服务器端IP地址的网络字节序，ifconfig
	socklen_t addrlen = sizeof(sin);


	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,addrlen) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success __%d__\n",__LINE__);

	//创建集合
	struct pollfd fds[2];

	//将0号添加到集合中
	fds[0].fd = 0;
	fds[0].events = POLLIN;//读事件

	//将cfd添加到集合中
	fds[1].fd = cfd;
	fds[1].events = POLLIN;//读事件

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

	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;
		}

		//有文件描述符产生事件
		//判断所有文件描述符实际产生的事件（revents）中是否有POLLIN
		//有可能同时产生多个事件，其中包含POLLIN
		//因此需要通过按位与的发方式提取出revents中是否含有POLLIN
		if(fds[0].revents & POLLIN)
		{

			//发送
			memset(buf,0,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf) -1] = 0;
			if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			if(strcmp(buf,"quit") == 0)
			{
				printf("客户端退出\n");
				break;
			}
			printf("send success __%d__\n",__LINE__);

		}

		if(fds[1].revents & POLLIN) // 服务器交互事件
		{

			//接收数据
			memset(buf,0,sizeof(buf));
			res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
			//res = recv(sfd,buf,sizeof(buf),0);
			if(res < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}else if(0 == res)
			{
				fprintf(stderr,"服务器关闭\n");
				break;
			}
			printf("[%s : %d]buf = %s\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf);

		}
	}

	//关闭所有文件
	close(cfd);
	return 0;
}
天涯~过客
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值