【Linux网络编程】基于TCP流 I/O多路转接(poll) 的高性能http服务器
生活随笔
收集整理的這篇文章主要介紹了
【Linux网络编程】基于TCP流 I/O多路转接(poll) 的高性能http服务器
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
服務器比較簡陋,為了學習poll的使用,只向客戶端回寫一條html語句。啟動服務器后,瀏覽器發起請求,服務端向瀏覽器寫回html,響應字符串,然后可以看到,瀏覽器解析并顯示 Hello Poll!.
?
啟動服務端:
用瀏覽器訪問:
?
瀏覽器解析出字符串:
?
?
完整代碼:
1 #include <stdio.h> 2 #include <stdlib.h> 3 #include <unistd.h> 4 #include <netinet/in.h> 5 #include <arpa/inet.h> 6 #include <unistd.h> 7 #include <string.h> 8 #include <poll.h> 9 #include <sys/socket.h> 10 #include <sys/types.h> 11 #include <sys/stat.h> 12 13 14 #define POLLFD_SIZE 1024 /* struct pollfd 結構體數組最大上限 */ 15 16 17 18 /* 關心描述符集事件數組*/ 19 struct pollfd array_pollfd[POLLFD_SIZE]; 20 21 /* 結構體成員詳情 22 struct pollfd 23 { 24 int fd; // 關心的描述符 25 short events; // 關心的事件 26 short revents; // 發生的事件 27 }; 28 */ 29 30 /* 獲取一個監聽連接的sockfd */ 31 int run_getsockfd(const char*ip, int port); 32 33 /* 執行poll檢測 */ 34 void run_poll(int listen_sockfd); 35 36 /* 響應客戶端的連接,并添加新的描述符到關心事件中 */ 37 void run_accept(int listen_sock); 38 39 /* 當與客戶端連接的描述符有事件就緒時,做出響應 */ 40 void run_action( int index); 41 42 int main(int argc, char **argv) 43 { 44 if(argc != 3) 45 { 46 printf("usage: [server_ip] [server_port]"); 47 return 1; 48 } 49 50 int listen_sockfd = run_getsockfd(argv[1], atoi(argv[2])); 51 52 run_poll(listen_sockfd); 53 54 return 0; 55 } 56 57 58 /* 調用poll 并檢測返回事件 */ 59 void run_poll(int listen_sockfd) 60 { 61 /* 將負責監聽連接的sockfd注冊事件 */ 62 array_pollfd[0].fd = listen_sockfd; 63 array_pollfd[0].events = POLLIN; 64 65 /* 初始化數組中的描述符 */ 66 int idx_init = 1; 67 for(; idx_init < POLLFD_SIZE; ++idx_init) 68 { 69 array_pollfd[idx_init].fd = -1; 70 } 71 int timeout = 1000; /* 設定一秒后超時 */ 72 73 74 while(1) 75 { 76 int ret_poll = poll(array_pollfd, POLLFD_SIZE, timeout); 77 78 if(ret_poll == 0) /* 超時 */ 79 printf("timeout\n"); 80 else if(ret_poll < 0) /* 執行出錯*/ 81 perror("poll()"); 82 else 83 {/* 有關心的事件就緒 */ 84 85 /* 遍歷數組,輪詢檢測poll的結果 */ 86 int idx_check = 0; 87 for(idx_check = 0; idx_check < POLLFD_SIZE; ++idx_check) 88 { 89 if(idx_check == 0 && array_pollfd[0].revents & POLLIN) 90 {/* listen_sockfd 讀事件就緒 */ 91 run_accept(listen_sockfd); 92 } 93 else if(idx_check != 0) 94 {/* 與客戶端連接的sockfd 有事件就緒 */ 95 run_action(idx_check); 96 } 97 } 98 } 99 } // end while 1 100 } 101 102 103 /* 當與客戶端連接的描述符有事件就緒時,做出響應 */ 104 void run_action( int index) 105 { 106 if(array_pollfd[index].revents & POLLIN) 107 {/* 客戶端讀事件發生 */ 108 char buf[1024]; /* 存儲從客戶端讀來的消息 */ 109 memset(buf, 0, sizeof(buf)); 110 ssize_t s = read(array_pollfd[index].fd, buf, sizeof(buf)-1); 111 if(s > 0) 112 { 113 buf[s-1] = '\0'; 114 printf("client say$ %s \n", buf); 115 array_pollfd[index].events = POLLOUT; 116 } 117 else if( s <= 0) 118 { 119 printf("client quit!\n"); 120 close(array_pollfd[index].fd); 121 array_pollfd[index].fd = -1; 122 } 123 124 } 125 else if (array_pollfd[index].revents & POLLOUT) 126 {/* 客戶端寫事件發生 */ 127 /* 使用瀏覽器測試,寫回到客戶端,瀏覽器會解析字符串,顯示 Hellp Epoll! */ 128 const char* msg = "HTTP/1.1 200 OK\r\n\r\n<html><br/><h1>Hello poll!</h1></html>"; 129 write(array_pollfd[index].fd, msg, strlen(msg)); 130 close(array_pollfd[index].fd); 131 array_pollfd[index].fd = -1; 132 } 133 } 134 135 136 /* 響應客戶端的連接,并添加新的描述符到關心事件中 */ 137 void run_accept(int listen_sock) 138 { 139 struct sockaddr_in cliaddr; 140 socklen_t clilen = sizeof(cliaddr); 141 142 int new_sock = accept(listen_sock, (struct sockaddr*)&cliaddr, &clilen); 143 if( new_sock < 0) 144 { 145 perror("accept"); 146 return ; 147 } 148 149 printf("與客戶端連接成功: ip %s port %d \n", inet_ntoa(cliaddr.sin_addr), ntohs(cliaddr.sin_port)); 150 /* 將新socket描述符添加到數組中 */ 151 int idx_find = 1; 152 for(; idx_find < POLLFD_SIZE; ++idx_find) 153 { 154 if(array_pollfd[idx_find].fd < 0) 155 { 156 array_pollfd[idx_find].fd = new_sock; 157 array_pollfd[idx_find].events = POLLIN ; 158 break; 159 } 160 } 161 if(idx_find == POLLFD_SIZE) 162 { 163 perror("連接超出最大限度,add array_pollfd[]"); 164 return; 165 } 166 167 } 168 169 /* 獲取一個監聽socket */ 170 int run_getsockfd(const char* ip, int port) 171 { 172 int sock = socket(AF_INET, SOCK_STREAM, 0); 173 if( sock < 0){ 174 perror("socket()"); 175 exit(1); 176 } 177 178 int opt = 1; 179 setsockopt(sock , SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)); 180 181 struct sockaddr_in server; 182 bzero(&server, sizeof(server)); 183 server.sin_addr.s_addr = inet_addr(ip); 184 server.sin_port = htons(port); 185 server.sin_family = AF_INET; 186 187 if(bind(sock, (struct sockaddr *)&server, sizeof(server) ) < 0){ 188 perror("bind()"); 189 exit(2); 190 } 191 192 if(listen(sock, 5) < 0){ 193 perror("listen()"); 194 exit(3); 195 } 196 197 return sock; 198 }?
轉載于:https://www.cnblogs.com/jiangzhaowei/p/8831108.html
總結
以上是生活随笔為你收集整理的【Linux网络编程】基于TCP流 I/O多路转接(poll) 的高性能http服务器的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 机器学习—SVM
- 下一篇: public、protected、def