Socket demo
一、select
1.原理
select是Linux系统提供的一种服务,无需使用循环不断询问或因等待IO而导致进程被阻塞
select就是将我们想要监听的sockfd交给select,而由系统来管理select,如果sockfd有事件产生,则select则通知程序。
select使用一个结构体来管理sockfd,其内有一个bitmap,向select添加sockfd也就是将bitmap对应值进行更改为1。同理,当有事件发生时,系统返回的bitmap中如果值为1则代表此sockfd有事件 发生。由于select返回的是一个bitmap而不是发生事件的集合,因此还需要遍历整个bitmap找出有事件发生的sockfd,导致其效率很低。
select只有水平触发,监听的最大值为1024,也可更改
2.常用函数
void FD_CLR(int fd, fd_set *set);//清除某一个被监视的文件描述符。
int FD_ISSET(int fd, fd_set *set);//测试一个文件描述符是否是集合中的一员
void FD_SET(int fd, fd_set *set);//添加一个文件描述符,将set中的某一位设置成1;
void FD_ZERO(fd_set *set);//清空集合中的文件描述符,将每一位都设置为0;
int select(int maxfd, fd_set* readset, fd_set* writeset, fd_set* exceptset,
const struct timeval* timeout);返回发生事件的sockfd的个数
3.代码
#include "TcpServer.h"
using namespace MySocket;
const int CLIENTSIZE = 100;
int main(int argc, char *argv[])
{
if (argc != 2)
return -1;
TcpServer server;
server.InitServer(atoi(argv[1]));
int maxfd = server.listenfd;
int client[CLIENTSIZE];
for (int i = 0; i < CLIENTSIZE; ++i)
{
client[i] = -1;
}
fd_set allset;//
FD_ZERO(&allset);//将allset都清空
FD_SET(server.listenfd, &allset);//将listenfd设置为监听
while (1)
{
fd_set tmpset = allset;//这里复制的原因是:返回发生事件的bitmap将原来的bitmap覆盖了
int ret = select(maxfd + 1, &tmpset, NULL, NULL, NULL);//返回发生事件的sockfd的个数
if (ret < 0)
{
printf("select failed..\n");
break;
}
else if (ret == 0)
{
printf("select timeout..\n");
continue;
}
for (int i = 0; i <= maxfd; ++i)
{
// printf("%d %d\n", i, maxfd);
if (!FD_ISSET(i, &tmpset))
continue;
if (i == server.listenfd)
{
printf("waiting for conneting...\n");
struct sockaddr_in cliaddr;
socklen_t clilen = sizeof(cliaddr);
if (server.Accept() == false)
{
printf("connect failed!\n");
return -1;
}
int confd = server.clientfd;
if (confd > CLIENTSIZE)
{
printf("too many connection...\n");
return -1;
}
printf("sock=%d connet success.\n\n", confd);
client[confd] = confd;
FD_SET(confd, &allset);
if (confd > maxfd)
{
maxfd = confd;
}
}
else
{
server.clientfd = i;
char buf[1000];
printf("dealing client sockfd..\n");
if (server.Read(buf, 0))
{
printf("%s\n", buf);
}
else
{
printf("socket=%d disconnect\n", i);
server.CloseClient();
FD_CLR(i, &allset);
}
}
}
}
return 0;
}
二、poll
1.原理
相比select,poll变化不大,原理也相同,poll使用数组来取代bitmap,因而也就没有大小限制。同样,poll也需要遍历整个数组来找出有事件发生的sockfd,其也只有水平触发。
2.常用函数
poll结构体
struct pollfd{
int fd; //文件描述符
short events; //等待的事件
short revents; //实际发生的事件
};
int poll(struct pollfd *fds, nfds_t nfds, int timeout);监视并等待多个文件描述符的属性变化
3.代码
#include "TcpServer.h"
#include "poll.h"
using namespace MySocket;
const int CLIENTSIZE = 100;
const int FDSIZE=1024;
int main(int argc, char *argv[])
{
if (argc != 2)
return -1;
TcpServer server;
server.InitServer(atoi(argv[1]));
int client[CLIENTSIZE];
for (int i = 0; i < CLIENTSIZE; ++i)
{
client[i] = -1;
}
//poll设置
struct pollfd fds[FDSIZE];
for(int i=0;i<FDSIZE;++i){
fds[i].fd=-1;
}
fds[server.listenfd].fd=server.listenfd;
fds[server.listenfd].events=POLLIN;
int maxfd = server.listenfd;
while (1)
{
int ret=poll(fds,maxfd+1,-1);
if (ret < 0)
{
printf("poll failed..\n");
break;
}
else if (ret == 0)
{
printf("poll timeout..\n");
continue;
}
for (int i = 0; i <= maxfd; ++i)
{
// printf("%d %d\n", i, maxfd);
if(fds[i].fd<0)continue;
if((fds[i].revents&POLLIN)==0) continue;
if (i == server.listenfd)
{
printf("waiting for conneting...\n");
struct sockaddr_in cliaddr;
socklen_t clilen = sizeof(cliaddr);
if (server.Accept() == false)
{
printf("connect failed!\n");
return -1;
}
int confd = server.clientfd;
if (confd > CLIENTSIZE)
{
printf("too many connection...\n");
return -1;
}
printf("sock=%d connet success.\n\n", confd);
client[confd] = confd;
fds[confd].fd=confd;
fds[confd].events=POLLIN;
fds[confd].revents=0;
if (confd > maxfd)
{
maxfd = confd;
}
}
else
{
server.clientfd = i;
char buf[1000];
printf("deal client sockfd..\n");
if (server.Read(buf, 0))
{
printf("%s\n", buf);
}
else
{
printf("socket=%d disconnect\n", i);
server.CloseClient();
fds[i].fd=-1;
}
}
}
}
return 0;
}
三、epoll
1.原理
select和poll效率都比较低,其原因就是每次都需要遍历才能找到发生事件的sockfd。epoll的一个巨大改进之处在于将发生事件的sockfd使用一个数据进行保存,提高了效率。epoll具有水平触发和边缘触发两种触发模式。
2.常用函数
struct epoll_event结构如下:
//保存触发事件的某个文件描述符相关的数据(与具体使用方式有关)
typedef union epoll_data {
void *ptr;
int fd;
__uint32_t u32;
__uint64_t u64;
} epoll_data_t;
//感兴趣的事件和被触发的事件
struct epoll_event {
__uint32_t events; /* Epoll events */
epoll_data_t data; /* User data variable */
};
int epoll_create(int size);//创建一个epoll的句柄。自从linux2.6.8之后,size参数是被忽略的。
int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);//epoll的事件注册函数,它不同于select()是在监听事件时告诉内核要监听什么类型的事件,而是在这里先注册要监听的事件类型。第一个参数是epoll_create()的返回值。第二个参数表示动作,用三个宏来表示:EPOLL_CTL_ADD:注册新的fd到epfd中;EPOLL_CTL_MOD:修改已经注册的fd的监听事件;EPOLL_CTL_DEL:从epfd中删除一个fd;第三个参数是需要监听的fd。第四个参数是告诉内核需要监听什么事
int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);//收集在epoll监控的事件中已经发送的事件。参数events是分配好的epoll_event结构体数组,epoll将会把发生的事件赋值到events数组中
3. 代码
#include "TcpServer.h"
#include "sys/epoll.h"
using namespace MySocket;
const int CLIENTSIZE = 100;
const int EVENTSIZE = 100;
int main(int argc, char *argv[])
{
if (argc != 2)
return -1;
TcpServer server;
server.InitServer(atoi(argv[1]));
int client[CLIENTSIZE];
for (int i = 0; i < CLIENTSIZE; ++i)
{
client[i] = -1;
}
int epollfd = epoll_create(1); // linux2.6.8版本后此值被忽略,只要大于0即可
struct epoll_event ev;
ev.data.fd = server.listenfd;
ev.events = EPOLLIN;
epoll_ctl(epollfd, EPOLL_CTL_ADD, server.listenfd, &ev);
int maxfd = server.listenfd;
while (1)
{
epoll_event events[EVENTSIZE];
int ret = epoll_wait(epollfd, events, EVENTSIZE, -1);//-1:忽略超时
if (ret < 0)
{
printf("epoll failed..\n");
break;
}
else if (ret == 0)
{
printf("epoll timeout..\n");
continue;
}
for (int i = 0; i < ret; ++i)
{
// printf("%d %d\n", i, maxfd);
if ((events[i].data.fd == server.listenfd) && (events[i].events & EPOLLIN))
{
printf("waiting for conneting...\n");
struct sockaddr_in cliaddr;
socklen_t clilen = sizeof(cliaddr);
if (server.Accept() == false)
{
printf("connect failed!\n");
return -1;
}
int confd = server.clientfd;
if (confd > CLIENTSIZE)
{
printf("too many connection...\n");
return -1;
}
printf("sock=%d connet success.\n\n", confd);
client[confd] = confd;
memcpy(&ev, 0, sizeof(ev));
ev.data.fd = confd;
ev.events = EPOLLIN;
epoll_ctl(epollfd, EPOLL_CTL_ADD, confd, &ev);
if (confd > maxfd)
{
maxfd = confd;
}
}
else
{
server.clientfd = i;
char buf[1000];
printf("dealing client sockfd..\n");
if (server.Read(buf, 0))
{
printf("%s\n", buf);
}
else
{
printf("socket=%d disconnect\n", i);
server.CloseClient();
memcpy(&ev, 0, sizeof(ev));
ev.data.fd = events[i].data.fd;
ev.events = EPOLLIN;
epoll_ctl(epollfd,EPOLL_CTL_DEL,events[i].data.fd,&ev);
close(events[i].data.fd);
}
}
}
}
return 0;
}