我们这里简单的说下 select 的作用,并给出 select 的客户端实例。我们知道 select 是IO 多路复用的一个最简单支持,poll 和 epoll 是 select 的升级版。我们通常会遇到这样一个问题:当客户端阻塞在 fgets() 等待客户输入的时候,服务器端断开连接。而客户端却不能及时知道,只有在客户输入完毕并发送到服务器的时候才知道连接已经断开,但是此时可能已经过了很长时间了。如果我们想及时知道服务器断开连接怎么办呢?
我们知道不管是 fgets() 等待客户输入还是 read() 从套接口读取数据,都是 IO 操作。我们不能阻塞在某个 IO 操作中的一个,这样其他 IO 操作会无法进行,即使其他 IO 操作上有数据了我们也无法及时读取。select 的原理是这样的:我们将这些 IO 操作所要操作的文件描述符放到一起(比如一个数组中),然后阻塞在 select() 函数上,为什么要阻塞在这里呢?其实这时的 select 实在不停的遍历这个数组,查看其中的文件描述符上是否可读/可写,一旦可读/可写,select 返回,停止阻塞。然后我们对可读/可写的文件描述如做相应的操作即可。
int select(nfds, readfds, writefds, exceptfds, timeout)
nfds 是指定 select() 要遍历的最大文件描述符 + 1,readfds 就是放文件描述的数组,这个数组里面关心的是该数组中文件描述符的读事件,wretefds 也是放文件描述符的数组,这个数组里面关心的是该数组中文件描述符的写事件,exceptfds 也是放文件描述符的数组,这个数组关心的是该数组中文件描述符的出错事件。timeout 是 select 阻塞的时间。如果设置为 空指针,那么将永远阻塞下去直到某个描述符有事件发生(就绪)。否则的话就会在阻塞由 timeout 指定的时间后返回,无论关心的文件描述符是否有事件发生。select 返回有事件发生的文件描述符个数,失败返回 -1,超时返回 0!
int FD_ISSET(int fd, fd_set *set);
服务端 Client.c
中间文件 errwrap.h
中间文件 errwrap.c
客户端编译:gcc Client.c errwrap.c -o Client
我们知道不管是 fgets() 等待客户输入还是 read() 从套接口读取数据,都是 IO 操作。我们不能阻塞在某个 IO 操作中的一个,这样其他 IO 操作会无法进行,即使其他 IO 操作上有数据了我们也无法及时读取。select 的原理是这样的:我们将这些 IO 操作所要操作的文件描述符放到一起(比如一个数组中),然后阻塞在 select() 函数上,为什么要阻塞在这里呢?其实这时的 select 实在不停的遍历这个数组,查看其中的文件描述符上是否可读/可写,一旦可读/可写,select 返回,停止阻塞。然后我们对可读/可写的文件描述如做相应的操作即可。
int select(nfds, readfds, writefds, exceptfds, timeout)
nfds 是指定 select() 要遍历的最大文件描述符 + 1,readfds 就是放文件描述的数组,这个数组里面关心的是该数组中文件描述符的读事件,wretefds 也是放文件描述符的数组,这个数组里面关心的是该数组中文件描述符的写事件,exceptfds 也是放文件描述符的数组,这个数组关心的是该数组中文件描述符的出错事件。timeout 是 select 阻塞的时间。如果设置为 空指针,那么将永远阻塞下去直到某个描述符有事件发生(就绪)。否则的话就会在阻塞由 timeout 指定的时间后返回,无论关心的文件描述符是否有事件发生。select 返回有事件发生的文件描述符个数,失败返回 -1,超时返回 0!
int FD_ISSET(int fd, fd_set *set);
这个函数中,fd是set数组里的一个,查看该fd是否有数据可读或者可写(取决于set是读集合还是写集合),如果有,则返回非0;
服务端,Server.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "errwrap.h"
#define MAXLINE 80
#define SERVERPORT 8000
int main(int argc, char *argv[])
{
int i, maxi, maxfd, listenfd, confd, sockfd;
int nret, client[FD_SETSIZE]; // FD_SETSIZE 默认为1024
ssize_t n;
fd_set rset, allset;
char buf[MAXLINE];
char str[INET_ADDRSTRLEN]; //#define INET_ADDRSTRLEN 16
socklen_t clientaddr_len;
struct sockaddr_in clientaddr, serveraddr;
listenfd = Socket(AF_INET, SOCK_STREAM, 0);//获得一个监听fd
bzero(&serveraddr, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
serveraddr.sin_port = htons(SERVERPORT);
Bind(listenfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
Listen(listenfd, 128);
maxfd = listenfd; //maxfd为现在监听的文件描述符的最大值
maxi = -1; /* client[]的下标*/
for (i = 0; i < FD_SETSIZE; i++)
client[i] = -1; /* 用-1初始化client[] */
FD_ZERO(&allset);
FD_SET(listenfd, &allset); // 设置allset(所有监听文件描述符集合最新状态)
for ( ; ; )
{
rset = allset; //当前文件描述符集合
nret = select(maxfd+1, &rset, NULL, NULL, NULL);//检测“读”
if (nret < 0)
printf("enter 0, nret: %d\n", nret);
if (FD_ISSET(listenfd, &rset)) //如果是新的连接到来
{
clientaddr_len = sizeof(clientaddr);
confd = Accept(listenfd, (struct sockaddr *)&clientaddr, &clientaddr_len);
printf("received from %s at PORT %d\n",
inet_ntop(AF_INET, &clientaddr.sin_addr, str, sizeof(str)),
ntohs(clientaddr.sin_port));
for (i = 0; i < FD_SETSIZE; i++)
{
if (client[i] < 0)
{
client[i] = confd; //将新的文件描述符添加到client里
break;
}
}
// 达到select能监控的文件个数上限1024
if (i == FD_SETSIZE)
{
fputs("too many clients\n", stderr);
exit(1);
}
FD_SET(confd, &allset); //将新的新的文件描述符添加到监控信号集里
if (confd > maxfd)
maxfd = confd; //更新最大的文件描述符
if (i > maxi)
maxi = i; // 更新client最大下标
if (--nret == 0)
continue; /* 如果没有更多的就绪文件描述符继续回到上面select阻塞监听,负责处理未
处理完的就绪文件描述符*/
}
for (i = 0; i <= maxi; i++)
{ // 检测哪个clients 有数据就绪
if ( (sockfd = client[i]) < 0)
continue;
if (FD_ISSET(sockfd, &rset)) //对连接到服务器的每个客户机(文件描述符),检测是否有数据到达
{
if ( (n = Read(sockfd, buf, MAXLINE)) == 0)
{
// 当client关闭链接时,服务器端也关闭对应链接
Close(sockfd);
FD_CLR(sockfd, &allset); // 解除select监控此文件描述符
client[i] = -1;
}
else
{//处理每个有数据到达的fd
int j;
for (j = 0; j < n; j++)
buf[j] = toupper(buf[j]);
Write(sockfd, buf, n);
}
if (--nret == 0)//处理的那个fd不是最后一个
break;
}
}
}
close(listenfd);
return 0;
}
服务端 Client.c
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include "errwrap.h"
#define MAXLINE 80
#define SERV_PORT 8000
int main(int argc, char *argv[])
{
if(argc<2)
{
err_exit("参数过少");
}
struct sockaddr_in servaddr;
char buf[MAXLINE];
int sockfd, n;
sockfd = Socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
servaddr.sin_port = htons(SERV_PORT);
Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
while (fgets(buf, MAXLINE, stdin) != NULL)
{
Write(sockfd, buf, strlen(buf));
n = Read(sockfd, buf, MAXLINE);
if (n == 0)
printf("the other side has been closed.\n");
else
Write(STDOUT_FILENO, buf, n);
}
Close(sockfd);
return 0;
}
中间文件 errwrap.h
#ifndef __ERRWRAP__H
#define __ERRWRAP__H
void perr_exit(const char *s);
void err_exit(const char *s);
int Accept(int fd, struct sockaddr *sa, socklen_t *salenptr);
void Bind(int fd, const struct sockaddr *sa, socklen_t salen);
void Connect(int fd, const struct sockaddr *sa, socklen_t salen);
void Listen(int fd, int backlog);
int Socket(int family, int type, int protocol);
ssize_t Read(int fd, void *ptr, size_t nbytes);
ssize_t Write(int fd, const void *ptr, size_t nbytes);
void Close(int fd);
ssize_t Readn(int fd, void *vptr, size_t n);
ssize_t Writen(int fd, const void *vptr, size_t n);
#endif
中间文件 errwrap.c
#include <stdlib.h>
#include <errno.h>
#include <sys/socket.h>
#include <stdio.h>
//错误处理与退出
void perr_exit(const char *s)
{
perror(s);
exit(1);
}
void err_exit(const char *s)
{
printf("%s\n",s);
exit(1);
}
//发生连接错误或者被信号中断后,将再次进入accept。
int Accept(int fd, struct sockaddr *sa, socklen_t *salenptr)
{
int n;
again:
if ( (n = accept(fd, sa, salenptr)) < 0) {
if ((errno == ECONNABORTED) || (errno == EINTR))
goto again;
else
perr_exit("accept error");
}
return n;
}
void Bind(int fd, const struct sockaddr *sa, socklen_t salen)
{
if (bind(fd, sa, salen) < 0)
perr_exit("bind error");
}
void Connect(int fd, const struct sockaddr *sa, socklen_t salen)
{
if (connect(fd, sa, salen) < 0)
perr_exit("connect error");
}
void Listen(int fd, int backlog)
{
if (listen(fd, backlog) < 0)
perr_exit("listen error");
}
int Socket(int family, int type, int protocol)
{
int n;
if ( (n = socket(family, type, protocol)) < 0)
perr_exit("socket error");
return n;
}
//没有读取到数据,失败原因是因为信号中断,则重新读取
ssize_t Read(int fd, void *ptr, size_t nbytes)
{
ssize_t n;
again:
if ( (n = read(fd, ptr, nbytes)) == -1) {
if (errno == EINTR)
goto again;
else
return -1;
}
return n;
}
ssize_t Write(int fd, const void *ptr, size_t nbytes)
{
ssize_t n;
again:
if ( (n = write(fd, ptr, nbytes)) == -1) {
if (errno == EINTR)
goto again;
else
return -1;
}
return n;
}
void Close(int fd)
{
if (close(fd) == -1)
perr_exit("close error");
}
ssize_t Readn(int fd, void *vptr, size_t n)
{
size_t nleft;
ssize_t nread;
char *ptr;
ptr = vptr;
nleft = n;
while (nleft > 0) {
if ( (nread = read(fd, ptr, nleft)) < 0) {
if (errno == EINTR)
nread = 0;
else
return -1;
} else if (nread == 0)
break;
nleft -= nread;
ptr += nread;
}
return n - nleft;
}
ssize_t Writen(int fd, const void *vptr, size_t n)
{
size_t nleft;
ssize_t nwritten;
const char *ptr;
ptr = vptr;
nleft = n;
while (nleft > 0) {
if ( (nwritten = write(fd, ptr, nleft)) <= 0) {
if (nwritten < 0 && errno == EINTR)
nwritten = 0;
else
return -1;
}
nleft -= nwritten;
ptr += nwritten;
}
return n;
}
客户端编译:gcc Client.c errwrap.c -o Client
服务端编译: gcc Server.c errwrap.c -o Server
先执行服务端: ./server
再执行客户端: ./Client 127.0.0.1
在客户端输入:duanjinhui
返回 DUANJINHUI
服务端会显示:received from 127.0.0.1 at PORT 56641
可以再开几个终端执行Client程序,可以看到服务端的并发现象