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) | 設定超時時間 |
/********************************************************************
> 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