CSAPP:Tiny Web服務器

Web 服務器，一個多麼莊嚴而神聖的名詞！沒有接觸之前，它神祕而不可侵犯；學習研究之後，它又是如此地平易近人。今天，讓我們一起走近web服務器。

此時，想像一下，當你自己動手實現了一個服務器時，你會是怎麼的興奮；尤其是將一個真正的瀏覽器指向我們自己實現的服務器時，看着它顯示自己本機上的文本及圖片時，那將是怎樣的激動時刻呀！

下面就來看看它的實現全過程：

Tiny的準備

#ifndef __CSAPP_H__ #define __CSAPP_H__ #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <string.h> #include <ctype.h> #include <setjmp.h> #include <signal.h> #include <sys/time.h> #include <sys/types.h> #include <sys/wait.h> #include <sys/stat.h> #include <fcntl.h> #include <sys/mman.h> #include <errno.h> #include <math.h> #include <semaphore.h> #include <sys/socket.h> #include <netdb.h> #include <netinet/in.h> #include <arpa/inet.h> /* 默認的文件訪問權限爲 DEF_MODE & ~DEF_UMASK */ #define DEF_MODE S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH #define DEF_UMASK S_IWGRP|S_IWOTH typedef struct sockaddr SA; #define RIO_BUFSIZE 8192 typedef struct { int rio_fd; /* 內部緩存區的描述符 */ int rio_cnt; /* 內部緩存區剩下還未讀的字節數 */ char *rio_bufptr; /* 指向內部緩存區中下一個未讀字節 */ char rio_buf[RIO_BUFSIZE]; /* 內部緩存區 */ } rio_t; extern char **environ; #define MAXLINE 8192 /* 每行最大字符數 */ #define MAXBUF 8192 /* I/O緩存區的最大容量 */ #define LISTENQ 1024 /* 監聽的第二個參數 */ /* helper functions */ ssize_t rio_writen(int fd,void *usrbuf,size_t n); void rio_readinitb(rio_t *rp,int fd); ssize_t rio_readlineb(rio_t *rp,void *usrbuf,size_t maxlen); int open_clientfd(char *hostname, int portno); int open_listenfd(int portno); #endif void doit(int fd); void read_requesthdrs(rio_t *rp); int parse_uri(char *uri,char *filename,char *cgiargs); void serve_static(int fd,char *filename,int filesize); void get_filetype(char *filename,char *filetype); void serve_dynamic(int fd,char *filename,char *cgiargs); void clienterror(int fd,char *cause,char *errum,char *shorting,char *longmsg);

Tiny的main程序

int main(int argc,char **argv) { int listenfd,connfd,port,clientlen; struct sockaddr_in clientaddr; if(argc != 2){ fprintf(stderr,"usage:%s <port>/n",argv[0]); exit(0); } port = atoi(argv[1]); listenfd = open_listenfd(port); while(1){ clientlen = sizeof(clientaddr); connfd = accept(listenfd,(SA *)&clientaddr,&clientlen); doit(connfd); close(connfd); } }

Tiny是一個迭代服務器，監聽在命令行中確定的端口上的連接請求。在通過open_listenedfd函數打開一個監聽套接字以後，Tiny執行典型的無限服務循環，反覆地接受一個連接(accept)請求，執行事務(doit)，最後關閉連接描述符(close)。

Tiny的doit函數

void doit(int fd) { int is_static; struct stat sbuf; char buf[MAXLINE],method[MAXLINE],uri[MAXLINE],version[MAXLINE]; char filename[MAXLINE],cgiargs[MAXLINE]; rio_t rio; rio_readinitb(&rio,fd); rio_readlineb(&rio,buf,MAXLINE); sscanf(buf,"%s %s %s",method,uri,version); if(strcasecmp(method,"GET")){ clienterror(fd,method,"501","Not Implemented","Tiny does not implement this method"); return; } read_requesthdrs(&rio); is_static = parse_uri(uri,filename,cgiargs); if(stat(filename,&sbuf) < 0){ clienterror(fd,filename,"404","Not found","Tiny couldn't read the file"); return; } if(is_static){ if(!(S_ISREG(sbuf.st_mode)) || !(S_IRUSR & sbuf.st_mode)){ clienterror(fd,filename,"403","Forbidden","Tiny couldn't read the file"); return; } serve_static(fd,filename,sbuf.st_size); } else{ if(!(S_ISREG(sbuf.st_mode)) || !(S_IRUSR & sbuf.st_mode)){ clienterror(fd,filename,"403","Forbidden","Tiny couldn't read the file"); return; } serve_dynamic(fd,filename,cgiargs); } }

rio_readinitb(&rio,fd) ：將程序的內部緩存區與描述符相關聯。

rio_readlineb(&rio,buf,MAXLINE) :從內部緩存區讀出一個文本行至buf中，以null字符來結束這個文本行。當然，每行最大的字符數量不能超過MAXLINE。

sscanf(buf,"%s %s %s",method,uri,version) :作爲例子，一般此時buf中存放的是“GET / HTTP/1.1”,所以可知method爲“GET”，uri爲“/”，version爲“HTTP/1.1”。其中sscanf的功能：把buf中的字符串以空格爲分隔符分別傳送到method、uri及version中。

strcasecmp(method,"GET") :忽略大小寫比較method與“GET”的大小，相等的話返回0。

read_requesthdrs(&rio) :讀並忽略請求報頭。

parse_uri(uri,filename,cgiargs) :解析uri，得文件名存入filename中，參數存入cgiargs中。

stat(filename,&sbuf) :將文件filename中的各個元數據填寫進sbuf中，如果找不到文件返回0。

S_ISREG(sbuf,st_mode) :此文件爲普通文件。

S_IRUSR & sbuf.st_mode :有讀取權限。

serve_static(fd,filename,sbuf.st_size) :提供靜態服務。

serve_dynamic(fd,filename,cgiargs) :提供動態服務。

從doit函數中可知，我們的Tiny Web服務器只支持“GET”方法，其他方法請求的話則會發送一條錯誤消息，主程序返回，並等待下一個請求。否則，我們讀並忽略請求報頭。（其實，我們在請求服務時，直接不用寫請求報頭即可，寫上只是爲了符合HTTP協議標準）。

然後，我們將uri解析爲一個文件名和一個可能爲空的CGI參數，並且設置一個標誌位，表明請求的是靜態內容還是動態內容。通過stat函數判斷文件是否存在。

最後，如果請求的是靜態內容，我們需要檢驗它是否是一個普通文件，並且可讀。條件通過，則我們服務器向客服端發送靜態內容；相似的，如果請求的是動態內容，我就覈實該文件是否是可執行文件，如果是則執行該文件，並提供動態功能。

Tiny的clienterror函數

void clienterror(int fd,char *cause,char *errnum,char *shortmsg,char *longmsg) { char buf[MAXLINE],body[MAXLINE]; sprintf(body,"<html><title>Tiny Error</title>"); sprintf(body,"%s<body bgcolor=""ffffff"">/r/n",body); sprintf(body,"%s%s:%s/r/n",body,longmsg,shortmsg); sprintf(body,"%s<p>%s:%s/r/n",body,longmsg,cause); sprintf(body,"%s<hr><i>The Tiny Web server</i>/r/n",body); sprintf(buf,"HTTP/1.0 %s %s/r/n",errnum,shortmsg); rio_writen(fd,buf,strlen(buf)); sprintf(buf,"Content-type:text/html/r/n"); rio_writen(fd,buf,strlen(buf)); sprintf(buf,"Content-length:%d/r/n/r/n",strlen(body)); rio_writen(fd,buf,strlen(buf)); rio_writen(fd,body,strlen(body)); }

向客戶端返回錯誤信息。

sprintf(buf,"------------"):將字符串“------------”輸送到buf中。

rio_writen(fd,buf,strlen(buf)):將buf中的字符串寫入fd描述符中。

Tiny的read_requesthdrs函數

void read_requesthdrs(rio_t *rp) { char buf[MAXLINE]; rio_readlineb(rp,buf,MAXLINE); while(strcmp(buf,"/r/n")) rio_readlineb(rp,buf,MAXLINE); return; }

Tiny不需要請求報頭中的任何信息，所以我們這個函數就是來跳過這些請求報頭的。具體做法就是讀這些請求報頭，直到空行，然後返回。OK！

Tiny的parse_uri函數

int parse_uri(char *uri,char *filename,char *cgiargs) { char *ptr; if(!strstr(uri,"cgi-bin")){ strcpy(cgiargs,""); strcpy(filename,"."); strcat(filename,uri); if(uri[strlen(uri)-1] == '/') strcat(filename,"home.html"); return 1; }else{ ptr = index(uri,'?'); if(ptr){ strcpy(cgiargs,ptr+1); *ptr = '/0'; } else strcpy(cgiargs,""); strcpy(filename,"."); strcat(filename,uri); return 0; } }

根據uri中是否含有cgi-bin來判斷請求的是靜態內容還是動態內容。如果沒有cgi-bin，則說明請求的是靜態內容。那麼，我們需把cgiargs置NULL，然後獲得文件名，如果我們請求的uri最後爲 “/”，則自動添加上home.html。比如說，我們請求的是“/”,則返回的文件名爲“./home.html”,而我們請求“/logo.gif”,則返回的文件名爲“./logo.gif”。如果uri中含有cgi-bin，則說明請求的是動態內容。那麼，我們需要把參數拷貝到cgiargs中，把要執行的文件路徑寫入filename。舉例來說，uri爲/cgi-bin/adder?3&5,則cigargs中   存放的是3&5，filename中存放的是“./cgi-bin/adder”,OK!

index(uri,'?') : 找出uri字符串中第一個出現參數‘？’的地址，並將此地址返回。

 

Tiny的serve_static函數void serve_static(int fd,char *filename,int filesize) { int srcfd; char *srcp,filetype[MAXLINE],buf[MAXBUF]; get_filetype(filename,filetype); sprintf(buf,"HTTP/1.0 200 OK/r/n"); sprintf(buf,"%sServer:Tiny Web Server/r/n",buf); sprintf(buf,"%sContent-length:%d/r/n",buf,filesize); sprintf(buf,"%sContent-type:%s/r/n/r/n",buf,filetype); rio_writen(fd,buf,strlen(buf)); srcfd = open(filename,O_RDONLY,0); srcp = mmap(0,filesize,PROT_READ,MAP_PRIVATE,srcfd,0); close(srcfd); rio_writen(fd,srcp,filesize); munmap(srcp,filesize); } void get_filetype(char *filename,char *filetype) { if(strstr(filename,".html")) strcpy(filetype,"text/html"); else if(strstr(filename,".gif")) strcpy(filetype,"image/gif"); else if(strstr(filename,",jpeg")) strcpy(filetype,"image.jpeg"); else strcpy(filetype,"text/plain"); }

打開文件名爲filename的文件，把它映射到一個虛擬存儲器空間，將文件的前filesize字節映射到從地址srcp開始的虛擬存儲區域。關閉文件描述符srcfd，把虛擬存儲區的數據寫入fd描述符，最後釋放虛擬存儲器區域。

Tiny的serve_dynamic函數void serve_dynamic(int fd,char *filename,char *cgiargs) { char buf[MAXLINE],*emptylist[] = { NULL }; sprintf(buf,"HTTP/1.0 200 OK/r/n"); rio_writen(fd,buf,strlen(buf)); sprintf(buf,"Server:Tiny Web Server/r/n"); rio_writen(fd,buf,strlen(buf)); if(fork() == 0){ setenv("QUERY_STRING",cgiargs,1); dup2(fd,STDOUT_FILENO); execve(filename,emptylist,environ); } wait(NULL); }

Tiny通過派生一個子進程並在子進程的上下文中運行一個cgi程序（可執行文件），來提供各種類型的動態內容。

setenv("QUERY_STRING",cgiargs,1) :設置QUERY_STRING環境變量。

dup2 （fd，STDOUT_FILENO) ：重定向它的標準輸出到已連接描述符。此時，任何寫到標準輸出的東西都直接寫到客戶端。

execve(filename,emptylist,environ) :加載運行cgi程序。

====================from csapp.c========================ssize_t rio_writen(int fd, void *usrbuf, size_t n) { size_t nleft = n; ssize_t nwritten; char *bufp = usrbuf; while (nleft > 0) { if ((nwritten = write(fd, bufp, nleft)) <= 0) { if (errno == EINTR) nwritten = 0; else return -1; } nleft -= nwritten; bufp += nwritten; } return n; } static ssize_t rio_read(rio_t *rp, char *usrbuf, size_t n) { int cnt; while (rp->rio_cnt <= 0) { /* 如果緩存區空，則重新填充 */ rp->rio_cnt = read(rp->rio_fd, rp->rio_buf, sizeof(rp->rio_buf)); if (rp->rio_cnt < 0) { if (errno != EINTR) return -1; } else if (rp->rio_cnt == 0) /* EOF */ return 0; else rp->rio_bufptr = rp->rio_buf; /* 重新設置緩存區指針 */ } /* 從內部緩存區拷貝 min(n, rp->rio_cnt) 個字節到usrbuf*/ cnt = n; if (rp->rio_cnt < n) cnt = rp->rio_cnt; memcpy(usrbuf, rp->rio_bufptr, cnt); rp->rio_bufptr += cnt; rp->rio_cnt -= cnt; return cnt; } void rio_readinitb(rio_t *rp, int fd) { rp->rio_fd = fd; rp->rio_cnt = 0; rp->rio_bufptr = rp->rio_buf; } ssize_t rio_readlineb(rio_t *rp, void *usrbuf, size_t maxlen) { int n, rc; char c, *bufp = usrbuf; for (n = 1; n < maxlen; n++) { if ((rc = rio_read(rp, &c, 1)) == 1) { *bufp++ = c; if (c == '/n') break; }else if (rc == 0) { if (n == 1) return 0; /* EOF, no data read */ else break; /* EOF, some data was read */ } else return -1; /* error */ } *bufp = 0; return n; } int open_clientfd(char *hostname, int port) { int clientfd; struct hostent *hp; struct sockaddr_in serveraddr; if ((clientfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) return -1; if ((hp = gethostbyname(hostname)) == NULL) return -2; bzero((char *) &serveraddr, sizeof(serveraddr)); serveraddr.sin_family = AF_INET; bcopy((char *)hp->h_addr_list[0], (char *)&serveraddr.sin_addr.s_addr, hp->h_length); serveraddr.sin_port = htons(port); if (connect(clientfd, (SA *) &serveraddr, sizeof(serveraddr)) < 0) return -1; return clientfd; } int open_listenfd(int port) { int listenfd, optval=1; struct sockaddr_in serveraddr; /* 創建一個套接字描述符 */ if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) return -1; /* Eliminates "Address already in use" error from bind. */ if (setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, (const void *)&optval , sizeof(int)) < 0) return -1; /* Listenfd will be an endpoint for all requests to port on any IP address for this host */ bzero((char *) &serveraddr, sizeof(serveraddr)); serveraddr.sin_family = AF_INET; serveraddr.sin_addr.s_addr = htonl(INADDR_ANY); serveraddr.sin_port = htons((unsigned short)port); if (bind(listenfd, (SA *)&serveraddr, sizeof(serveraddr)) < 0) return -1; /* Make it a listening socket ready to accept connection requests */ if (listen(listenfd, LISTENQ) < 0) return -1; return listenfd; }