更为常用的网络文件模型:多路复用,再也不用创建一个线程去监听!!
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