vik*_*.rk 4 c unix sockets tcp client-server
我正在阅读经典书籍Unix Network Programming,当我偶然发现这个程序时(第 6.8 节,第 179-180 页)
#include "unp.h"
int
main(int argc, char **argv)
{
int i, maxi, maxfd, listenfd, connfd, sockfd;
int nready, client[FD_SETSIZE];
ssize_t n;
fd_set rset, allset;
char buf[MAXLINE];
socklen_t clilen;
struct sockaddr_in cliaddr, servaddr;
listenfd = Socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(SERV_PORT);
Bind(listenfd, (SA *) &servaddr, sizeof(servaddr));
Listen(listenfd, LISTENQ);
maxfd = listenfd; /* initialize */
maxi = -1; /* index into client[] array */
for (i = 0; i < FD_SETSIZE; i++)
client[i] = -1; /* -1 indicates available entry */
FD_ZERO(&allset);
FD_SET(listenfd, &allset);
for ( ; ; ) {
rset = allset; /* structure assignment */
nready = Select(maxfd+1, &rset, NULL, NULL, NULL);
if (FD_ISSET(listenfd, &rset)) { /* new client connection */
clilen = sizeof(cliaddr);
connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);
for (i = 0; i < FD_SETSIZE; i++)
if (client[i] < 0) {
client[i] = connfd; /* save descriptor */
break;
}
if (i == FD_SETSIZE)
err_quit("too many clients");
FD_SET(connfd, &allset); /* add new descriptor to set */
if (connfd > maxfd)
maxfd = connfd; /* for select */
if (i > maxi)
maxi = i; /* max index in client[] array */
if (--nready <= 0)
continue; /* no more readable descriptors */
}
for (i = 0; i <= maxi; i++) { /* check all clients for data */
if ( (sockfd = client[i]) < 0)
continue;
if (FD_ISSET(sockfd, &rset)) {
if ( (n = Read(sockfd, buf, MAXLINE)) == 0) {
/*4connection closed by client */
Close(sockfd);
FD_CLR(sockfd, &allset);
client[i] = -1;
} else
Writen(sockfd, buf, n);
if (--nready <= 0)
break; /* no more readable descriptors */
}
}
}
}
Run Code Online (Sandbox Code Playgroud)
作者提到这个程序对于 DOS 攻击是不安全的。引自书中,
“不幸的是,我们刚刚展示的服务器存在问题。考虑一下如果恶意客户端连接到服务器,发送一个字节的数据(除了换行符),然后进入睡眠状态会发生什么。服务器将调用read(系统调用),它将从客户端读取单个字节的数据,然后在下一次调用read 中阻塞,等待来自该客户端的更多数据。然后服务器被这个客户端阻塞,并且不会为任何服务其他客户端,直到恶意客户端发送换行符或终止”
我不确定我是否正确理解这一点。为什么这个恶意客户端会第二次调用read系统调用,因为它只发送了 1 个字节的数据,第一次调用select 时会通知它。对select的后续调用将永远不会设置此恶意文件描述符,因为没有活动。我在这里错过了什么吗?
我的猜测是代码中有一个错字,而不是Read,应该是书中其他地方提到的Readline方法的某个版本。
注意:代码包含Read和Select(大写 R 和 S),它们只是read和select系统调用的错误处理包装器