【Linux网络编程部分----多进程高并发poll模型】

YINLEREN##

于 2023-02-26 20:43:18 发布

阅读量515

点赞数

文章标签：网络 tcp/ip c语言 linux

本文链接：https://blog.csdn.net/qq_53380078/article/details/129230173

版权

前言

本文采用 Visual Studio 2022 运用远程登陆，来对虚拟机内的 Linux 进行操作

如果不知道怎么使用的话可参考： linux远程开发——使用vs2019远程连接linux_vs2019远程调试linux_似末的博客-CSDN博客https://blog.csdn.net/wmcy123/article/details/123415371?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522167741273216782427438492%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=167741273216782427438492&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduend~default-3-123415371-null-null.142^v73^insert_down1,201^v4^add_ask,239^v2^insert_chatgpt&utm_term=vs%E9%93%BE%E6%8E%A5linux&spm=1018.2226.3001.4187

背景

在TCP协议下，使用套接字（socket）多进程（fork）之间进行通信，并且客户端能够从服务器设定下的地址上获取文件。

分析

要分别从服务器和客户端两个方面入手，采取网络编程所固定的格式来进行编写

服务器：要不断的监听是否有客户端来进行通信，并且采用多进程的方式使得客户端之间互不影响

客户端：要可随时加入服务器，并且能够自主下载服务端下面的指定文件

编写步骤

服务器：

        第一步：创建套接字
        第二步：监听套接字
        第三步：等待客户端连接
        第四步：发送数据给客户端

第五步：关闭套接字

客户端：

        第一步：创建套接字
        第二步：链接服务器
        第三步：读写服务器
        第四步：关闭套接字

服务器端代码

一、我们先做前期准备，先创建套接字（socket），并先确定其中的内容

int tcp_sock = socket(AF_INET, SOCK_STREAM, 0);      //创建套接字
	struct sockaddr_in Ipv4 {};                      //定义一个Ipv4的结构体信息
	Ipv4.sin_family = AF_INET;                       //协议为IPV4
	Ipv4.sin_addr.s_addr = 0;                        //Ip地址
	Ipv4.sin_port = htons(PORT);                     //端口号  此处PORT为宏定义

二、创建完后，在对其进行绑定，并且开始监听

if (bind(tcp_sock, (struct sockaddr*)&Ipv4, sizeof(Ipv4)) < 0) //绑定
	{
		perror("bind");
		return 1;
	}
	listen(tcp_sock, 10);    //监听将主动套接字变为被动套接字  监听10路

三、创建 pollfd 类型的结构来确保我们使用 poll 函数

struct pollfd fds[SIZE]{};    // 将服务器加入到轮询表中 
fds[index].fd = tcp_sock;     // 有数据可读事件 */

四、准备工作已经做完了，就开始服务器的操作

while (true)
	{
		if (poll(fds, SIZE, 3000) == 0)
		{
			//printf("time out \n");        //检测
			continue;
		}
		for (int i = 0; i < index + 1; i++)
		{
			if (tcp_sock == fds[i].fd)        //主服务器  
			{
				if (fds[i].revents & POLLIN)
				{
					int cid = accept(tcp_sock, NULL, NULL);
					fds[index].fd = cid;      //将服务器加入到轮询表中
					fds[index].events = POLLIN; //有数据可读事件
					index++;
					printf("someone coming\n");
				}
			}
			else //与客户端交互
			{
				if (fork() == 0)  //多进程  子进程进行操作。避免出现孤儿进程
				{
					if (fds[i].revents & POLLIN)
					{
						memset(buf, 0, SIZE);   //将buf的内容全部置为0
						buf[SIZE - 1] = 1;
						len = read(fds[i].fd, buf, SIZE);  //读取数据
						if (len > 0)
						{
							if (strcmp(buf, "file") == 0)
							{
								printf("sent file to %d\n", i);
								dir(fds[i].fd);  // 目录文件操作
								read(fds[i].fd, buf, SIZE);
								file(fds[i].fd, buf); 
								memset(buf, 0, SIZE);
							}
							if (buf[SIZE - 1] != 0)
							{
								printf("%d --- :%s\n", i, buf);  
							}
							write(fds[i].fd, buf, len);
						}
					}
					exit(-1);
				}	
			}
			waitpid(-1, NULL, WNOHANG);
		}
	}