TCP 客戶端 服務端詳細程式碼
轉自:http://www.myexception.cn/program/1912019.html
TCP網路程式設計中connect()、listen()和accept()三者之間的關係
基於 TCP 的網路程式設計開發分為伺服器端和客戶端兩部分,常見的核心步驟和流程如下:
connect()函式
對於客戶端的 connect() 函式,該函式的功能為客戶端主動連線伺服器,建立連線是通過三次握手,而這個連線的過程是由核心完成,不是這個函式完成的,這個函式的作用僅僅是通知 Linux 核心,讓 Linux 核心自動完成 TCP 三次握手連線(三次握手詳情,請看《淺談 TCP 三次握手》),最後把連線的結果返回給這個函式的返回值(成功連線為0, 失敗為-1)。
通常的情況,客戶端的 connect() 函式預設會一直阻塞,直到三次握手成功或超時失敗才返回(正常的情況,這個過程很快完成)。
listen()函式
對於伺服器,它是被動連線的。舉一個生活中的例子,通常的情況下,移動的客服(相當於伺服器)是等待著客戶(相當於客戶端)電話的到來。而這個過程,需要呼叫listen()函式。
#include<sys/socket.h>
int listen(int sockfd, int backlog);
listen() 函式的主要作用就是將套接字( sockfd )變成被動的連線監聽套接字(被動等待客戶端的連線),至於引數 backlog 的作用是設定核心中連線佇列的長度(這個長度有什麼用,後面做詳細的解釋)。這裡需要注意的是,listen()函式不會阻塞,它主要做的事情為,將該套接字和套接字對應的連線佇列長度告訴 Linux 核心,然後,listen()函式就結束。
這樣的話,當有一個客戶端主動連線(connect()),Linux 核心就自動完成TCP 3次握手,將建立好的連結自動儲存到佇列中,如此重複。
所以,只要 TCP 伺服器呼叫了 listen(),客戶端就可以通過 connect() 和伺服器建立連線,而這個連線的過程是由核心完成。
下面為測試的伺服器和客戶端程式碼,執行程式時,要先執行伺服器,再執行客戶端:
伺服器:
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> int main(int argc, char *argv[]) { unsigned short port = 8000; int sockfd; sockfd = socket(AF_INET, SOCK_STREAM, 0);// 建立通訊端點:套接字 if(sockfd < 0) { perror("socket"); exit(-1); } struct sockaddr_in my_addr; bzero(&my_addr, sizeof(my_addr)); my_addr.sin_family = AF_INET; my_addr.sin_port = htons(port); my_addr.sin_addr.s_addr = htonl(INADDR_ANY); int err_log = bind(sockfd, (struct sockaddr*)&my_addr, sizeof(my_addr)); if( err_log != 0) { perror("binding"); close(sockfd); exit(-1); } err_log = listen(sockfd, 10); if(err_log != 0) { perror("listen"); close(sockfd); exit(-1); } printf("listen client @port=%d...\n",port); sleep(10); // 延時10s system("netstat -an | grep 8000"); // 檢視連線狀態 return 0; }
客戶端:
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
int main(int argc, char *argv[])
{
unsigned short port = 8000; // 伺服器的埠號
char *server_ip = "10.221.20.12"; // 伺服器ip地址
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);// 建立通訊端點:套接字
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
struct sockaddr_in server_addr;
bzero(&server_addr,sizeof(server_addr)); // 初始化伺服器地址
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(port);
inet_pton(AF_INET, server_ip, &server_addr.sin_addr);
int err_log = connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)); // 主動連線伺服器
if(err_log != 0)
{
perror("connect");
close(sockfd);
exit(-1);
}
system("netstat -an | grep 8000"); // 檢視連線狀態
while(1);
return 0;
}
執行程式時,要先執行伺服器,再執行客戶端,執行結果如下:
三次握手的連線佇列
這裡詳細的介紹一下 listen() 函式的第二個引數( backlog)的作用:告訴核心連線佇列的長度。
為了更好的理解
backlog 引數,我們必須認識到核心為任何一個給定的監聽套介面維護兩個佇列:
1、未完成連線佇列(incomplete connection queue),每個這樣的 SYN 分節對應其中一項:已由某個客戶發出併到達伺服器,而伺服器正在等待完成相應的 TCP三次握手過程。這些套介面處於 SYN_RCVD 狀態。
2、已完成連線佇列(completed connection queue),每個已完成 TCP 三次握手過程的客戶對應其中一項。這些套介面處於 ESTABLISHED 狀態。
當來自客戶的 SYN 到達時,TCP 在未完成連線佇列中建立一個新項,然後響應以三次握手的第二個分節:伺服器的 SYN 響應,其中稍帶對客戶 SYN 的 ACK(即SYN+ACK),這一項一直保留在未完成連線佇列中,直到三次握手的第三個分節(客戶對伺服器 SYN 的 ACK )到達或者該項超時為止(曾經源自Berkeley的實現為這些未完成連線的項設定的超時值為75秒)。
如果三次握手正常完成,該項就從未完成連線佇列移到已完成連線佇列的隊尾。
backlog 引數歷史上被定義為上面兩個佇列的大小之和,大多數實現預設值為 5,當伺服器把這個完成連線佇列的某個連線取走後,這個佇列的位置又空出一個,這樣來回實現動態平衡,但在高併發 web 伺服器中此值顯然不夠。
accept()函式
accept()函式功能是,從處於
established 狀態的連線佇列頭部取出一個已經完成的連線,如果這個佇列沒有已經完成的連線,accept()函式就會阻塞,直到取出佇列中已完成的使用者連線為止。
如果,伺服器不能及時呼叫 accept() 取走佇列中已完成的連線,佇列滿掉後會怎樣呢?UNP(《unix網路程式設計》)告訴我們,伺服器的連線佇列滿掉後,伺服器不會對再對建立新連線的syn進行應答,所以客戶端的 connect 就會返回 ETIMEDOUT。但實際上Linux的並不是這樣的!
下面為測試程式碼,伺服器 listen() 函式只指定佇列長度為 2,客戶端有 6 個不同的套接字主動連線伺服器,同時,保證客戶端的 6 個 connect()函式都先呼叫完畢,伺服器的 accpet() 才開始呼叫。
伺服器:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
int main(int argc, char *argv[])
{
unsigned short port = 8000;
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
struct sockaddr_in my_addr;
bzero(&my_addr, sizeof(my_addr));
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(port);
my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
int err_log = bind(sockfd, (struct sockaddr*)&my_addr, sizeof(my_addr));
if( err_log != 0)
{
perror("binding");
close(sockfd);
exit(-1);
}
err_log = listen(sockfd, 2); // 等待佇列為2
if(err_log != 0)
{
perror("listen");
close(sockfd);
exit(-1);
}
printf("after listen\n");
sleep(20); //延時 20秒
printf("listen client @port=%d...\n",port);
int i = 0;
while(1)
{
struct sockaddr_in client_addr;
char cli_ip[INET_ADDRSTRLEN] = "";
socklen_t cliaddr_len = sizeof(client_addr);
int connfd;
connfd = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);
if(connfd < 0)
{
perror("accept");
continue;
}
inet_ntop(AF_INET, &client_addr.sin_addr, cli_ip, INET_ADDRSTRLEN);
printf("-----------%d------\n", ++i);
printf("client ip=%s,port=%d\n", cli_ip,ntohs(client_addr.sin_port));
char recv_buf[512] = {0};
while( recv(connfd, recv_buf, sizeof(recv_buf), 0) > 0 )
{
printf("recv data ==%s\n",recv_buf);
break;
}
close(connfd); //關閉已連線套接字
//printf("client closed!\n");
}
close(sockfd); //關閉監聽套接字
return 0;
}
客戶端:
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
void test_connect()
{
unsigned short port = 8000; // 伺服器的埠號
char *server_ip = "10.221.20.12"; // 伺服器ip地址
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);// 建立通訊端點:套接字
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
struct sockaddr_in server_addr;
bzero(&server_addr,sizeof(server_addr)); // 初始化伺服器地址
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(port);
inet_pton(AF_INET, server_ip, &server_addr.sin_addr);
int err_log = connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)); // 主動連線伺服器
if(err_log != 0)
{
perror("connect");
close(sockfd);
exit(-1);
}
printf("err_log ========= %d\n", err_log);
char send_buf[100]="this is for test";
send(sockfd, send_buf, strlen(send_buf), 0); // 向伺服器傳送資訊
//close(sockfd);
}
int main(int argc, char *argv[])
{
pid_t pid;
pid = fork();
if(0 == pid){
test_connect(); // 1
pid_t pid = fork();
if(0 == pid){
test_connect(); // 2
}else if(pid > 0){
test_connect(); // 3
}
}else if(pid > 0){
test_connect(); // 4
pid_t pid = fork();
if(0 == pid){
test_connect(); // 5
}else if(pid > 0){
test_connect(); // 6
}
}
while(1);
return 0;
}
同樣是先執行伺服器,在執行客戶端,伺服器 accept()函式前延時了 20 秒, 保證了客戶端的 connect() 全部呼叫完畢後再呼叫 accept(),執行結果如下:
按照 UNP 的說法,連線佇列滿後(這裡設定長度為 2,發了 6 個連線),以後再呼叫 connect() 應該統統超時失敗,但實際上測試結果是:有的 connect()立刻成功返回了,有的經過明顯延遲後成功返回了。對於伺服器
accpet() 函式也是這樣的結果:有的立馬成功返回,有的延遲後成功返回。
對於上面伺服器的程式碼,我們把lisen()的第二個引數改為大於 6 的數(如 10),重新執行程式,發現,客戶端 connect() 立馬返回成功, 伺服器 accpet() 函式也立馬返回成功。
TCP 的連線佇列滿後,Linux 不會如書中所說的全部拒絕連線,有些會延時連線,有些會連線不上(當佇列的長度指定為 0 ),寫程式時伺服器的 listen() 的第二個引數最好還是根據需要填寫,寫太大不好(具體可以看cat /proc/sys/net/core/somaxconn,預設最大值限制是 128),浪費資源,寫太小也不好,延時建立連線(也有可能連線不上)。