day5 IO模型分类

最新推荐文章于 2024-09-14 20:37:51 发布

y_xuan1999

最新推荐文章于 2024-09-14 20:37:51 发布

阅读量32

点赞数

分类专栏：网络编程文章标签：网络服务器 linux

本文链接：https://blog.csdn.net/m0_74205826/article/details/130117076

版权

网络编程专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这篇文章展示了如何使用IO多路复用（select函数）来实现一个TCP服务器，该服务器监听指定端口，接受客户端连接，并进行数据传输。客户端通过套接字连接到服务器，发送和接收数据。代码中包含了错误处理和文件描述符管理，确保了通信的稳定性和可靠性。

摘要由CSDN通过智能技术生成

IO多路复用

TCP服务器

#include <stdio.h>	
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <signal.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>
#include <sys/time.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d",__LINE__);\
	perror(msg);\
	return -1;\
}while(0)

#define IP "192.168.8.176"
#define PORT 11123

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

	//允许端口被快速重用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
	}

	//填充地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family        = AF_INET;
	sin.sin_port          = htons(PORT);
	sin.sin_addr.s_addr   = inet_addr(IP);

	//绑定地址信息结构体
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
	}

	//将套接字设置为被动监听状态
	if(listen(sfd,128) < 0)
	{
		ERR_MSG("listen");
	}


	//设置一个读集合
	fd_set readfds,tmpfds;

	//由于readfds中需要防止检测的文件描述符，所以不能让他是随机值
	//所以需要将readfds清空
	FD_ZERO(&readfds);
	FD_ZERO(&tmpfds);

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

	//另存客户端地址信息，0,1,2，sfd不可能有对应的客户端
	struct sockaddr_in saveCin[1024-4];

	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	char buf[128] = "";
	int s_res = 0;
	int newfd = -1;
	ssize_t res = 0;
	int maxfd = sfd;

	while(1)
	{

		tmpfds = readfds;
		s_res = select(maxfd+1,&tmpfds,NULL,NULL,NULL);
		if(s_res < 0)
		{
			ERR_MSG("select");
		}
		else if(0 == s_res)
		{
			printf("time out ...\n");
			return -1;
		}

		//判断集合中剩下那个文件描述符，走对应程序
		for(int i = 0; i <= maxfd; i++)
		{
			if(!FD_ISSET(i,&tmpfds))
			{
				continue;
			}

			if(0 == i)
			{
				printf("键盘输入事件：");
				int sndfd = -1;
				res = scanf("%d%s",&sndfd,buf);
				while(getchar() != 10);
				if(res != 2)
				{
					fprintf(stderr,"请输入正确的格式：int string\n");
					continue;
				}

				//能运行到当前位置，则代表输入的格式整数
				if(sndfd <= sfd || sndfd >= 1024 || !FD_ISSET(sndfd,&readfds))
				{
					fprintf(stderr,"sndfd = %d 文件描述符错误\n",sndfd);
					continue;
				}

				if(send(sndfd,buf,sizeof(buf),0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
			}
			else if(sfd == i)
			{
				newfd = accept(sfd,(struct sockaddr *)&cin,&addrlen);
				if(newfd < 0)
				{
					ERR_MSG("accept");
				}
				printf("[%s:%d] newfd=%d 连接成功__%d__\n", \
						inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);	

				saveCin[newfd-4] = cin;
				//将newfd添加到读集合中去
				FD_SET(newfd,&readfds);

				//更新maxfd
				maxfd = maxfd > newfd ? maxfd : newfd;
				
			}
			else
			{
				bzero(buf,sizeof(buf));
				//接收
				res = recv(i,buf,sizeof(buf),0);
				if(res < 0)
				{
					ERR_MSG("recv");
				}
				else if(0 == res)
				{
					printf("[%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
					//从目前最大文件描述符往小的判断直到判断到在集合中
					//该文件描述符就是最大的
					int j = 0;
					for(j = maxfd; j >= 0; j--)
					{
						if(FD_ISSET(j,&readfds))
						{
							maxfd = j;
							break;
						}
					}
					if(j < 0)
						maxfd = -1;
					continue;
				}
					printf("[%s : %d] newfd = %d : %s\n",\
							inet_ntoa(saveCin[i-4].sin_addr),ntohs(saveCin[i-4].sin_port),i,buf);	
			}

		}
	}

	close(newfd);
	close(sfd);
	return 0;
}

客户端

#include <stdio.h>	
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/select.h>
#include <sys/time.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d\n",__LINE__);\
	perror(msg);\
}while(0)

#define IP "192.168.8.176"
#define PORT 11123

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

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

	//绑定客户端的地址信息结构体，非必须绑定的
	//如果不绑定，则由操作系统自动选择一个端口号，以及本机可用IP绑定到套接字上

	//填充地址信息结构体
	//真实的地址信息结构体根据地址族指定 AF_INET：man 7 ip 
	struct sockaddr_in sin;
	sin.sin_family       = AF_INET;      //必须填AF_INET
	sin.sin_port         = htons(PORT);  //端口号 1024_49151
	sin.sin_addr.s_addr  = inet_addr(IP);//本机IP ifconfig

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

    //设置一个读集合
    fd_set readfds, tmpfds;     //vi -t fd_set，集合中只有一个成员是长整形数组

    //由于readfds中需要防止要检测的文件描述符，所以不能让他是随机值
    //所以需要将readfds清空
    FD_ZERO(&readfds);
    FD_ZERO(&tmpfds);

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

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

	while(1)
	{
		tmpfds = readfds;
		s_res = select(cfd + 1, &tmpfds, NULL, NULL, NULL);
		if (s_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if (0 == s_res)
		{
			printf("time out...\n");
			return -1;
		}

		if(FD_ISSET(0,&tmpfds))
		{
			bzero(buf,sizeof(buf));
			printf("请输入>>> ");
			res = scanf("%s",buf);
			if(send(cfd,buf,sizeof(buf),0) < 0)
			{
				ERR_MSG("send");
				return -1;
			}
		}
		
		if(FD_ISSET(cfd,&tmpfds))
		{
			bzero(buf,sizeof(buf));
			res = recv(cfd,buf,sizeof(buf),0);
			if(res < 0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0 == res)
			{
				printf("客户端退出\n");
				return -1;
			}
			printf("A说：");
			printf("%s\n",buf);
		}
	}

	close(cfd);
	return 0;
}