您好,登錄后才能下訂單哦!
(上一篇地址)前面使用socket完成一個服務器對應多個客戶端的小實驗的時候,針對TCP連接,我們必須得創建新的進程來與新的客戶端通信。那么,就意味著,1000個客戶端就有有1000個server進程,這顯然是不實際的。如果,我們可以提前把要監聽的文件描述符放到一個集合里,一旦其中一個發生事件(不管是連上,還是通信),就去處理。這樣,會方便很多。所以,今天學習一下IO復用。
最流行的I/O模型是阻塞I/O模型,缺省時,所有的套接口都是阻塞的。
如果一個或多個I/O條件滿足(例如:輸入已準備好被讀,或者描述字可以承接更多輸出的時候)我們就能夠被通知到,這樣的能力被稱為I/O復用,是由函數select和poll支持的。
/* According to POSIX.1-2001 */
#include <sys/select.h>
/* According to earlier standards */
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
int select(int nfds, fd_set *readfds, fd_set *writefds,
fd_set *exceptfds, struct timeval *timeout);
void FD_CLR(int fd, fd_set *set);//從集合中刪除一個描述字
int FD_ISSET(int fd, fd_set *set);//描述字是否在該集合中
void FD_SET(int fd, fd_set *set);//添加一個描述字到集合中
void FD_ZERO(fd_set *set);//清空描述字集合
時間的結構體如下:
struct timeval(
long tv_sec; //秒
long tv_usec;//微秒
);
timeout參數有三種可能
根本不等待:檢查描述字后立即返回,這稱為輪詢。定時器的值必須為0
select使用描述字集,它一般是一個整數數組,每個數中的每一位對應一個描述字。
使用select完成之前socket的測試,流程如下:
客戶端代碼不變。
#include < sys/types.h>
#include < sys/socket.h>
#include < netinet/in.h> //sockaddr_in
#include < stdio.h>
#include < string.h>
//TCP
int main()
{
int fd;
int ret;
int addrLen;
char acbuf[20] = "";
struct sockaddr_in serAddr = {0};
struct sockaddr_in myAddr = {0};
//1.socket();
fd = socket(PF_INET,SOCK_STREAM,0);
if(fd == -1)
{
perror("socket");
return -1;
}
//2.連接connect() 服務器的地址
serAddr.sin_family = AF_INET;
serAddr.sin_port = htons(1234);
serAddr.sin_addr.s_addr = inet_addr("192.168.159.5");
ret = connect(fd,(struct sockaddr *)&serAddr,sizeof(struct sockaddr_in));
if(ret == -1)
{
perror("connect");
return -1;
}
//獲取自己的地址
addrLen = sizeof(struct sockaddr_in);
ret = getsockname(fd,(struct sockaddr *)&myAddr,&addrLen);
if(ret == -1)
{
perror("getsockname");
return -1;
}
printf("client---ip: %s , port: %d\n",\
inet_ntoa(myAddr.sin_addr),ntohs(myAddr.sin_port));
//3.通信
while(1)
{
printf("send: ");
fflush(stdout);
scanf("%s",acbuf);
if(strcmp(acbuf,"exit") == 0)
{
break;
}
write(fd,acbuf,strlen(acbuf));
}
//4.close()
close(fd);
return 0;
}
服務器端:
select.c
#include < sys/types.h>
#include < sys/socket.h>
#include < netinet/in.h> //sockaddr_in
#include < stdio.h>
#include < string.h>
#include < signal.h>
#include < sys/select.h>
#include < unistd.h>
#include < sys/time.h>
//TCP
int main()
{
int fd;
int clientfd;
int ret;
pid_t pid;
int i;
int maxfd; //當前最大套接字
int nEvent;
fd_set set = {0}; //監聽集合
fd_set oldset = {0}; //存放所有要監聽的文件描述符
struct timeval time = {0};
int reuse = 0;
char acbuf[20] = "";
char client_addr[100] = "";
struct sockaddr_in addr = {0}; //自己的地址
struct sockaddr_in clientAddr = {0}; //連上的客戶端的地址
int addrLen = sizeof(struct sockaddr_in);
signal(SIGCHLD,SIG_IGN);
//1.socket()
fd = socket(PF_INET,SOCK_STREAM,0);
if(fd == -1)
{
perror("socket");
return -1;
}
//會出現沒有活動的套接字仍然存在,會禁止綁定端口,出現錯誤:address already in use .
//由TCP套接字TIME_WAIT引起,bind 返回 EADDRINUSE,該狀態會保留2-4分鐘
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
{
perror("setsockopet error\n");
return -1;
}
//2.bind()
addr.sin_family = AF_INET;
addr.sin_port = htons(1234);
addr.sin_addr.s_addr = inet_addr("192.168.159.5");
ret = bind(fd,(struct sockaddr *)&addr,addrLen);
if(ret == -1)
{
perror("bind");
return -1;
}
//3.listen()
ret = listen(fd,10);
if(ret == -1)
{
perror("listen");
return -1;
}
//創建監聽集合
FD_ZERO(&oldset);
FD_SET(fd,&oldset);
//maxfdp1:當前等待的最大套接字。比如:當前fd的值=3,則最大的套接字就是3
//所以每當有客戶端連接進來,就比較一下文件描述符
maxfd = fd;
//select
//select之前,set放的是所有要監聽的文件描述符;{3,4,5}
//select之后,set只剩下有發生事件的文件描述符。{3}
while(1)
{
set = oldset;
printf("before accept.\n");
time.tv_sec = 5;
nEvent = select(maxfd + 1,&set,NULL,NULL,&time); //返回文件描述符的個數(即事件的個數)
printf("after accept.%d\n",nEvent);
if(nEvent == -1)
{
perror("select");
return -1;
}
else if(nEvent == 0) //超時
{
printf("time out");
return 1;
}
else
{
//有事件發生
//判斷是否是客戶端產生的事件
for(i = 0 ; i <= maxfd ; i++)
{
if(FD_ISSET(i,&set))
{
if(i == fd)
{
clientfd = accept(fd,(struct sockaddr *)&clientAddr,&addrLen);
FD_SET(clientfd,&oldset);
printf("client ip:%s ,port:%u\n",inet_ntoa(clientAddr.sin_addr),ntohs(clientAddr.sin_port));
if(clientfd > maxfd)
{
maxfd = clientfd;
}
}
else
{
memset(acbuf,0,20);
if(read(i,acbuf,20) == 0) //客戶端退出
{
close(i);
//還要從集合里刪除
FD_CLR(i,&oldset);
}
else
printf("receive: %s\n",acbuf);
}
}
}
}
}
return 0;
}
epoll用到的函數有以下幾個:
#include <sys/epoll.h>
int epoll_create(int size);//創建epoll
int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);//操作函數
int epoll_wait(int epfd, struct epoll_event *events,
int maxevents, int timeout);
事件集合的結構體:
(這里 ,還要注意,epoll的超時參數是int,單位是us)
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h> //sockaddr_in
#include <stdio.h>
#include <string.h>
#include <signal.h>
#include <sys/epoll.h>
//epoll
//epoll_wait() epoll_creat() epoll_ctl()
//TCP
int main()
{
int fd;
int clientfd;
int ret;
pid_t pid;
int i;
int epfd;
int nEvent;
struct epoll_event event = {0};
struct epoll_event rtl_events[20] = {0}; //事件結果集
int reuse = 0;
char acbuf[20] = "";
char client_addr[100] = "";
struct sockaddr_in addr = {0}; //自己的地址
struct sockaddr_in clientAddr = {0}; //連上的客戶端的地址
int addrLen = sizeof(struct sockaddr_in);
signal(SIGCHLD,SIG_IGN);
//1.socket()
fd = socket(PF_INET,SOCK_STREAM,0);
if(fd == -1)
{
perror("socket");
return -1;
}
//會出現沒有活動的套接字仍然存在,會禁止綁定端口,出現錯誤:address already in use .
//由TCP套接字TIME_WAIT引起,bind 返回 EADDRINUSE,該狀態會保留2-4分鐘
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
{
perror("setsockopet error\n");
return -1;
}
//2.bind()
addr.sin_family = AF_INET;
addr.sin_port = htons(1234);
addr.sin_addr.s_addr = inet_addr("192.168.159.5");
ret = bind(fd,(struct sockaddr *)&addr,addrLen);
if(ret == -1)
{
perror("bind");
return -1;
}
//3.listen()
ret = listen(fd,10);
if(ret == -1)
{
perror("listen");
return -1;
}
epfd = epoll_create(1000); //同時監聽的文件描述符
event.data.fd = fd;
event.events = EPOLLIN; //讀
epoll_ctl(epfd,EPOLL_CTL_ADD,fd, &event);
while(1)
{
// nEvent = epoll_wait(epfd,rtl_events,20,-1); //-1:阻塞 0:非阻塞
nEvent = epoll_wait(epfd,rtl_events,20,5000);
if(nEvent == -1)
{
perror("epoll_wait");
return -1;
}
else if(nEvent == 0)
{
printf("time out.");
}
else
{
//有事件發生,立即處理
for(i = 0; i < nEvent;i++)
{
//如果是 服務器fd
if( rtl_events[i].data.fd == fd )
{
clientfd = accept(fd,(struct sockaddr *)&clientAddr,&addrLen);
//添加
event.data.fd = clientfd;
event.events = EPOLLIN; //讀
epoll_ctl(epfd,EPOLL_CTL_ADD,clientfd,&event);
printf("client ip:%s ,port:%u\n",inet_ntoa(clientAddr.sin_addr),ntohs(clientAddr.sin_port));
}
else
{
//否則 客戶端fd
memset(acbuf,0,20);
ret = read(rtl_events[i].data.fd,acbuf,20);
printf("%d\n",ret);
if( ret == 0) //客戶端退出
{
close(rtl_events[i].data.fd);
//從集合里刪除
epoll_ctl(epfd,EPOLL_CTL_DEL,rtl_events[i].data.fd,NULL);
}
else
printf("receive: %s\n",acbuf);
}
}
}
}
return 0;
}
運行結果如前,正常收發。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。