您好,登錄后才能下訂單哦!
在 TCP/IP 協議中,"IP地址 + TCP或UDP端口號" 可以唯一標識網絡通訊中的一個進程,"IP地址+端口號" 就稱為 socket。本文以一個簡單的 TCP 協議為例,介紹如何創建基于 TCP 協議的網絡程序。
TCP 協議通訊流程
下圖描述了 TCP 協議的通訊流程(此圖來自互聯網):
下圖則描述 TCP 建立連接的過程(此圖來自互聯網):
服務器調用 socket()、bind()、listen() 函數完成初始化后,調用 accept() 阻塞等待,處于監聽端口的狀態,客戶端調用 socket() 初始化后,調用 connect() 發出 SYN 段并阻塞等待服務器應答,服務器應答一個SYN-ACK 段,客戶端收到后從 connect() 返回,同時應答一個 ACK 段,服務器收到后從 accept() 返回。
TCP 連接建立后數據傳輸的過程:
建立連接后,TCP 協議提供全雙工的通信服務,但是一般的客戶端/服務器程序的流程是由客戶端主動發起請求,服務器被動處理請求,一問一答的方式。因此,服務器從 accept() 返回后立刻調用 read(),讀 socket 就像讀管道一樣,如果沒有數據到達就阻塞等待,這時客戶端調用 write() 發送請求給服務器,服務器收到后從 read() 返回,對客戶端的請求進行處理,在此期間客戶端調用 read() 阻塞等待服務器的應答,服務器調用 write() 將處理結果發回給客戶端,再次調用 read() 阻塞等待下一條請求,客戶端收到后從 read() 返回,發送下一條請求,如此循環下去。
下圖描述了關閉 TCP 連接的過程:
如果客戶端沒有更多的請求了,就調用 close() 關閉連接,就像寫端關閉的管道一樣,服務器的 read() 返回 0,這樣服務器就知道客戶端關閉了連接,也調用 close() 關閉連接。注意,任何一方調用 close() 后,連接的兩個傳輸方向都關閉,不能再發送數據了。如果一方調用 shutdown() 則連接處于半關閉狀態,仍可接收對方發來的數據。
在學習 socket 編程時要注意應用程序和 TCP 協議層是如何交互的:
下面通過一個簡單的 TCP 網絡程序來理解相關概念。程序分為服務器端和客戶端兩部分,它們之間通過 socket 進行通信。
服務器端程序
下面是一個非常簡單的服務器端程序,它從客戶端讀字符,然后將每個字符轉換為大寫并回送給客戶端:
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <netinet/in.h> #define MAXLINE 80 #define SERV_PORT 8000 int main(void) { struct sockaddr_in servaddr, cliaddr; socklen_t cliaddr_len; int listenfd, connfd; char buf[MAXLINE]; char str[INET_ADDRSTRLEN]; int i, n; // socket() 打開一個網絡通訊端口,如果成功的話, // 就像 open() 一樣返回一個文件描述符, // 應用程序可以像讀寫文件一樣用 read/write 在網絡上收發數據。 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 和 servaddr 綁定在一起, // 使 listenfd 這個用于網絡通訊的文件描述符監聽 servaddr 所描述的地址和端口號。 bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)); // listen() 聲明 listenfd 處于監聽狀態, // 并且最多允許有 20 個客戶端處于連接待狀態,如果接收到更多的連接請求就忽略。 listen(listenfd, 20); printf("Accepting connections ...\n"); while (1) { cliaddr_len = sizeof(cliaddr); // 典型的服務器程序可以同時服務于多個客戶端, // 當有客戶端發起連接時,服務器調用的 accept() 返回并接受這個連接, // 如果有大量的客戶端發起連接而服務器來不及處理,尚未 accept 的客戶端就處于連接等待狀態。 connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddr_len); n = read(connfd, buf, MAXLINE); printf("received from %s at PORT %d\n", inet_ntop(AF_INET, &cliaddr.sin_addr, str, sizeof(str)), ntohs(cliaddr.sin_port)); for (i = 0; i < n; i++) { buf[i] = toupper(buf[i]); } write(connfd, buf, n); close(connfd); } }
把上面的代碼保存到文件 server.c 文件中,并執行下面的命令編譯:
$ gcc server.c -o server
然后運行編譯出來的 server 程序:
$ ./server
此時我們可以通過 ss 命令來查看主機上的端口監聽情況:
如上圖所示,server 程序已經開始監聽主機的 8000 端口了。
下面讓我們介紹一下這段程序中用到的 socket 相關的 API。
int socket(int family, int type, int protocol);
socket() 打開一個網絡通訊端口,如果成功的話,就像 open() 一樣返回一個文件描述符,應用程序可以像讀寫文件一樣用 read/write 在網絡上收發數據。對于IPv4,family 參數指定為 AF_INET。對于 TCP 協議,type 參數指定為 SOCK_STREAM,表示面向流的傳輸協議。如果是 UDP 協議,則 type 參數指定為 SOCK_DGRAM,表示面向數據報的傳輸協議。protocol 指定為 0 即可。
int bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen);
服務器需要調用 bind 函數綁定一個固定的網絡地址和端口號。bind() 的作用是將參數 sockfd 和 myaddr 綁定在一起,使 sockfd 這個用于網絡通訊的文件描述符監聽 myaddr 所描述的地址和端口號。struct sockaddr *是一個通用指針類型,myaddr 參數實際上可以接受多種協議的 sockaddr 結構體,而它們的長度各不相同,所以需要第三個參數 addrlen 指定結構體的長度。
程序中對 myaddr 參數的初始化為:
bzero(&servaddr, sizeof(servaddr)); servaddr.sin_family = AF_INET; servaddr.sin_addr.s_addr = htonl(INADDR_ANY); servaddr.sin_port = htons(SERV_PORT);
首先將整個結構體清零,然后設置地址類型為 AF_INET,網絡地址為 INADDR_ANY,這個宏表示本地的任意 IP 地址,因為服務器可能有多個網卡,每個網卡也可能綁定多個 IP 地址,這樣設置可以在所有的 IP 地址上監聽,直到與某個客戶端建立了連接時才確定下來到底用哪個 IP 地址,端口號為 SERV_PORT,我們定義為 8000。
int listen(int sockfd, int backlog);
listen() 聲明 sockfd 處于監聽狀態,并且最多允許有 backlog 個客戶端處于連接待狀態,如果接收到更多的連接請求就忽略。
int accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen);
三方握手完成后,服務器調用 accept() 接受連接,如果服務器調用 accept() 時還沒有客戶端的連接請求,就阻塞等待直到有客戶端連接上來。cliaddr 是一個傳出參數,accept() 返回時傳出客戶端的地址和端口號。addrlen 參數是一個傳入傳出參數(value-result argument),傳入的是調用者提供的緩沖區 cliaddr 的長度以避免緩沖區溢出問題,傳出的是客戶端地址結構體的實際長度(有可能沒有占滿調用者提供的緩沖區)。如果給 cliaddr 參數傳 NULL,表示不關心客戶端的地址。
服務器程序的主要結構如下:
while (1) { cliaddr_len = sizeof(cliaddr); connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddr_len); n = read(connfd, buf, MAXLINE); ...... close(connfd); }
整個是一個 while 死循環,每次循環處理一個客戶端連接。由于 cliaddr_len 是傳入傳出參數,每次調用 accept( ) 之前應該重新賦初值。accept() 的參數 listenfd 是先前的監聽文件描述符,而 accept() 的返回值是另外一個文件描述符 connfd,之后與客戶端之間就通過這個 connfd 通訊,最后關閉 connfd 斷開連接,而不關閉 listenfd,再次回到循環開頭 listenfd 仍然用作 accept 的參數。
客戶端程序
下面是客戶端程序,它從命令行參數中獲得一個字符串發給服務器,然后接收服務器返回的字符串并打印:
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <netinet/in.h> #include <arpa/inet.h> #define MAXLINE 80 #define SERV_PORT 8000 int main(int argc, char *argv[]) { struct sockaddr_in servaddr; char buf[MAXLINE]; int sockfd, n; char *str; if (argc != 2) { fputs("usage: ./client message\n", stderr); exit(1); } str = argv[1]; sockfd = socket(AF_INET, SOCK_STREAM, 0); bzero(&servaddr, sizeof(servaddr)); servaddr.sin_family = AF_INET; inet_pton(AF_INET, "127.0.0.1", &servaddr.sin_addr); servaddr.sin_port = htons(SERV_PORT); // 由于客戶端不需要固定的端口號,因此不必調用 bind(),客戶端的端口號由內核自動分配。 // 注意,客戶端不是不允許調用 bind(),只是沒有必要調用 bind() 固定一個端口號, // 服務器也不是必須調用 bind(),但如果服務器不調用 bind(),內核會自動給服務器分配監聽端口, // 每次啟動服務器時端口號都不一樣,客戶端要連接服務器就會遇到麻煩。 connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)); write(sockfd, str, strlen(str)); n = read(sockfd, buf, MAXLINE); printf("Response from server:\n"); write(STDOUT_FILENO, buf, n); printf("\n"); close(sockfd); return 0; }
把上面的代碼保存到文件 client.c 文件中,并執行下面的命令編譯:
$ gcc client.c -o client
然后運行編譯出來的 client 程序:
$ ./client hello
此時服務器端會收到請求并返回轉換為大寫的字符串,并輸出相應的信息:
而客戶端在發送請求后會收到轉換過的字符串:
在客戶端的代碼中有兩點需要注意:
1. 由于客戶端不需要固定的端口號,因此不必調用 bind(),客戶端的端口號由內核自動分配。
2. 客戶端需要調用 connect() 連接服務器,connect 和 bind 的參數形式一致,區別在于 bind 的參數是自己的地址,而 connect 的參數是對方的地址。
至此我們已經使用 socket 技術完成了一個最簡單的客戶端服務器程序,雖然離實際應用還非常遙遠,但就學習而言已經足夠了。
提升服務器端的響應能力
雖然我們的服務器程序可以響應客戶端的請求,但是這樣的效率太低了。一般情況下服務器程序需要能夠同時處理多個客戶端的請求。可以通過 fork 系統調用創建子進程來處理每個請求,下面是大體的實現思路:
listenfd = socket(...); bind(listenfd, ...); listen(listenfd, ...); while (1) { connfd = accept(listenfd, ...); n = fork(); if (n == -1) { perror("call to fork"); exit(1); } else if (n == 0) { // 在子進程中處理客戶端的請求。 close(listenfd); while (1) { read(connfd, ...); ... write(connfd, ...); } close(connfd); exit(0); } else { close(connfd); } }
此時父進程的任務就是不斷的創建子進程,而由子進程去響應客戶端的具體請求。通過這種方式,可以極大的提升服務器端的響應能力。
總結
本文通過一個簡單的建基于 TCP 協議的網絡程序介紹了 linux socket 編程中的基本概念。通過它我們可以了解到 socket 程序工作的基本原理,以及一些解決性能問題的思路。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。