更為常用的網路檔案模型:多路複用,再也不用建立一個執行緒去監聽!!

2020-10-21 13:00:24

更為常用的網路檔案模型:多路複用,再也不用建立一個執行緒去監聽!!

select 函數設定三個檔案集合(分別是讀、寫、執行),然後將需要被檢測的檔案描述符加入到檔案集合中,延時一段時間,如果這段時間內有檔案描述符發生動作,那麼這個檔案描述符將會在時間結束後保留在集合中,沒有發生動作的,會在時間結束時從集合中刪除。

在這裡插入圖片描述

  • 標頭檔案:
/* According to POSIX.1-2001, POSIX.1-2008 */
 #include <sys/select.h>
/* According to earlier standards */
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
  • 函數原型:
  int select(int nfds,    //表示檔案集合最大檔案描述符+1(0~1023)下標加1
  			fd_set *readfds,  //讀檔案集合
  			fd_set *writefds,  //寫檔案集合
            fd_set *exceptfds,  //執行檔案集合
            struct timeval *timeout);  //設定超時時間
   void FD_CLR(int fd, fd_set *set);    //從set集合中刪除fd描述符
   int  FD_ISSET(int fd, fd_set *set); //判斷fd描述符是否在集合內
   void FD_SET(int fd, fd_set *set); //新增fd描述符到集合內
   void FD_ZERO(fd_set *set); //清空集合set
  • 引數分析:
select (…)內容
nfds表示檔案集合最大檔案描述符+1(0~1023)下標加1
readfds讀檔案集合
writefds寫檔案集合
exceptfds執行檔案集合
timeout)設定超時時間
  • 返回值:
    0 表示超過時間沒有檔案描述符發生動作
    大於0 表示有檔案描述符發生動作
    -1 表示出錯

記錄:send多少個位元組,recv就多少個,如果recv少了,則還會觸發recv直至收到send發來的全部位元組

DEMO:

在這裡插入圖片描述

/********************************************************************
  > File Name: main.c
  > Author: xiening
  > Mail:  1606598696@qq.com  
  > Created Time: 2020年10月20日 星期二 16時48分02秒
 *******************************************************************/

#include "list.h"

#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/types.h>        
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <string.h>

#define TCPPORT 5000

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

	//建立socket
	int tcpsock = socket(AF_INET, SOCK_STREAM, 0);
	if(tcpsock == -1)
	{
		fprintf(stderr, "socket failed: %s\n", strerror(errno));
	}

	int optval = 1;
	int optlen = sizeof(int);
	int ret = setsockopt(tcpsock, SOL_SOCKET, SO_REUSEADDR, &optval, optlen);
	if(ret == -1)
	{
		fprintf(stderr, "setsockopt failed :%s\n", strerror(errno));
		return -1;
	}


	//網路結構體:
	struct sockaddr_in src;
	src.sin_family = AF_INET;
	src.sin_addr.s_addr = inet_addr("0.0.0.0");
	src.sin_port = htons(TCPPORT);

	//繫結地址:
	ret = bind(tcpsock, (struct sockaddr*)&src, sizeof(struct sockaddr));
	if(ret == -1)
	{
		fprintf(stderr, "bind failed: %s\n", strerror(errno));
		return -1;
	}

	//監聽通訊端:
	ret = listen(tcpsock, 3);
	if(ret == -1)
	{
		fprintf(stderr, "bind failed: %s\n", strerror(errno));
		return -1;
	}

	//建立連結串列:初始化連結串列
	pfdlist head = init_list();

	//讀集合:
	fd_set rset;



	while(1)
	{
		//FD_ZERO(&rset); //清空集合set
		
		//清空集合set
		FD_ZERO(&rset);
		
		//將tcpsock新增到集合中:
		FD_SET(tcpsock, &rset);

		//時間設定:3秒
		struct timeval t;
		t.tv_sec = 3;
		t.tv_usec = 0;

		//描述符最大值:
		int FDMAX = tcpsock;
		//將連結串列中的描述符加入集合中,並找出最大描述符
		pfdlist tmp = head;
		while(tmp->next != NULL)
		{
			tmp = tmp->next;
			FD_SET(tmp->acceptfd, &rset); //新增fd描述符到集合內
			FDMAX = FDMAX>(tmp->acceptfd)?FDMAX:(tmp->acceptfd);
		}

		//多路複用
		int ret = select(FDMAX+1, &rset, NULL, NULL, &t);
		//在3秒內沒有發生動作
		if(ret == 0)
		{
			printf("time out\n");
			continue;
		}
		else if(ret == -1)
		{
			fprintf(stderr, "select failed: %s\n", strerror(errno));
			return -1;
		}
		//3秒內有動作:
		else if(ret > 0)
		{
			//tcpsock發生動作:
			if(FD_ISSET(tcpsock, &rset))
			{
				struct sockaddr_in destaddr;
				int len = sizeof(struct sockaddr_in);
				int acceptfd = accept(tcpsock, (struct sockaddr*)&destaddr, &len);
				if(acceptfd == -1)
				{
					fprintf(stderr, "accept failed: %s\n", strerror(errno));
					return -1;
				}
				//加入連結串列:
				else 
				{
					printf("有新的連線加入\n");
					insert_list(head, acceptfd);
				//	continue;
				}
			}


			//遍歷連結串列:
			tmp = head;
			while(tmp->next != NULL)
			{
				tmp = tmp->next;
				if(FD_ISSET(tmp->acceptfd, &rset)) //判斷通訊描述符是否在集合內
				{
					char buf[32];
					bzero(buf, 32);
					int ret = recv(tmp->acceptfd, buf, 32, 0);
					if(ret != -1)
						printf("recv:%s", buf);
			
					//通訊結束,刪除節點
					if(strcmp(buf, "88\n") == 0)
					{
						printf("對方斷開了通訊, 刪除該通訊通訊端\n");
						close(tmp->acceptfd);
						delete_node(head, tmp->acceptfd);
					}
				}
			}

		}

	
	}


	

    return 0 ;
}


/********************************************************************
  > File Name: list.c
  > Author: xiening
  > Mail:  1606598696@qq.com  
  > Created Time: 2020年10月20日 星期二 16時31分58秒
 *******************************************************************/

#include "list.h"


//初始化連結串列:
pfdlist init_list()
{
	pfdlist node = malloc(sizeof(fdlist));
	node->next = NULL;
	return node;
}


//尾插法:
void insert_list(pfdlist head, int newfd)
{
	pfdlist tmp = head;

	//遍歷連結串列:到尾部
	while(tmp->next != NULL)
	{
		tmp = tmp->next;
		//排除重複的
		if(tmp->acceptfd == newfd)
		{
			printf("newfd already exist\n");
			return ;
		}
	}


	pfdlist node = malloc(sizeof(fdlist));
	node->acceptfd = newfd;
	node->next = NULL;
	tmp->next = node;

}


//遍歷列印連結串列:
void display_list(pfdlist head)
{
	pfdlist tmp = head;
	while(tmp->next != NULL)
	{
		tmp = tmp->next;
		printf("%d-", tmp->acceptfd);
	}
	printf("over\n");
}

//刪除某個節點:
void delete_node(pfdlist head, int fd)
{
	pfdlist tmp = head;

	//遍歷連結串列:
	while(tmp->next != NULL)
	{
		//儲存上一個節點
		pfdlist tmp1;
		tmp1 = tmp;
		tmp = tmp->next;
		
		//匹配節點
		if(tmp->acceptfd == fd)
		{
			tmp1->next = tmp->next;
			free(tmp);
			return ;
		}
	}
	if(tmp->next == NULL)
	{
		printf("delete_node failed fd not exist\n");
	}
	
}

//刪除整個連結串列:
void delete_all(pfdlist head)
{
	pfdlist tmp = head->next;

	//遍歷連結串列
	while(tmp != NULL)
	{
		//記錄下一個節點
		pfdlist tmp1 = tmp->next;

		//刪除節點
		free(tmp);
		tmp = tmp1;

	}
}




/********************************************************************
  > File Name: net.h
  > Author: xiening
  > Mail:  1606598696@qq.com  
  > Created Time: 2020年10月20日 星期二 16時29分33秒
 *******************************************************************/

#ifndef LIST_H
#define LIST_H 

#include <stdio.h>
#include <netinet/in.h>
#include <stdlib.h>


typedef struct FDLIST
{
	int acceptfd;
	//struct sockaddr_in destaddr;
	struct FDLIST *next;

}fdlist, *pfdlist;

//刪除整個連結串列:
void delete_all(pfdlist head);

//刪除某個節點:
void delete_node(pfdlist head, int fd);

//遍歷列印連結串列:
void display_list(pfdlist head);

//尾插法:
void insert_list(pfdlist head, int newfd);

//初始化連結串列:
pfdlist init_list();



#endif