主要參考https://www.cnblogs.com/Anker/p/6413642.html
一码倦,問題:
我在做一個tcp通信的項目,負(fù)責(zé)客戶端锭碳,使用非阻塞connect遇到connect阻塞的問題袁稽,使用阻塞connect則遇到 EINPROGRESS的錯誤,需要找到辦法解決擒抛。特此記錄以作學(xué)習(xí)所用推汽。
二,非阻塞和阻塞connect
對于阻塞式套接字歧沪,調(diào)用connect函數(shù)將激發(fā)TCP的三次握手過程歹撒,而且僅在連接建立成功或者出錯時才返回,阻塞時長幾十秒到幾分鐘不等诊胞;
對于非阻塞式套接字暖夭,如果調(diào)用connect函數(shù)會之間返回-1(表示出錯),且錯誤為EINPROGRESS撵孤,表示連接建立迈着,建立啟動但是尚未完成;
如果返回0邪码,則表示連接已經(jīng)建立裕菠,這通常是在服務(wù)器和客戶在同一臺主機上時發(fā)生。
三闭专,解決方法:
select是一種IO多路復(fù)用機制奴潘,它允許進程指示內(nèi)核等待多個事件的任何一個發(fā)生,并且在有一個或者多個事件發(fā)生或者經(jīng)歷一段指定的時間后才喚醒它影钉。connect本身并不具有設(shè)置超時功能画髓,如果想對套接字的IO操作設(shè)置超時,可使用select函數(shù)平委。
對于select和非阻塞connect雀扶,注意兩點:[1] 當(dāng)連接成功建立時,描述符變成可寫肆汹; [2] 當(dāng)連接建立遇到錯誤時愚墓,描述符變?yōu)榧纯勺x,也可寫昂勉,遇到這種情況浪册,可調(diào)用getsockopt函數(shù)。
四岗照,實現(xiàn)步驟:
(1) 創(chuàng)建socket村象,并利用fcntl將其設(shè)置為非阻塞
(2) 調(diào)用connect函數(shù),如果返回0攒至,則連接建立厚者;如果返回-1,檢查errno 迫吐,如果值為 EINPROGRESS库菲,則連接正在建立;
(3) 為了控制連接建立時間志膀,將該socket描述符加入到select的可讀可寫集合中熙宇,采用select函數(shù)設(shè)定超時;
(4) 如果規(guī)定時間內(nèi)成功建立溉浙,則描述符變?yōu)榭蓪懱讨梗环駝t,采用getsockopt函數(shù)捕獲錯誤信息戳稽;
(5) 恢復(fù)套接字的文件狀態(tài)并返回馆蠕。
測試代碼如下:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <fcntl.h>
#include <errno.h>
int main(int argc, char **argv)
{
if (argc < 3) {
printf("please input ip and port, for example ./main 120.12.34.56 80.\n");
return -1;
}
char *ipaddr = argv[1];
unsigned int port = atoi(argv[2]);
int fd = 0;
struct sockaddr_in addr;
fd_set fdr, fdw;
struct timeval timeout;
int err = 0;
int errlen = sizeof(err);
fd = socket(AF_INET,SOCK_STREAM,0);
if (fd < 0) {
fprintf(stderr, "create socket failed,error:%s.\n", strerror(errno));
return -1;
}
bzero(&addr, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
inet_pton(AF_INET, ipaddr, &addr.sin_addr);
/*設(shè)置套接字為非阻塞*/
int flags = fcntl(fd, F_GETFL, 0);
if (flags < 0) {
fprintf(stderr, "Get flags error:%s\n", strerror(errno));
close(fd);
return -1;
}
flags |= O_NONBLOCK;
if (fcntl(fd, F_SETFL, flags) < 0) {
fprintf(stderr, "Set flags error:%s\n", strerror(errno));
close(fd);
return -1;
}
/*阻塞情況下linux系統(tǒng)默認(rèn)超時時間為75s*/
int rc = connect(fd, (struct sockaddr*)&addr, sizeof(addr));
if (rc != 0) {
if (errno == EINPROGRESS) {
printf("Doing connection.\n");
/*正在處理連接*/
FD_ZERO(&fdr);
FD_ZERO(&fdw);
FD_SET(fd, &fdr);
FD_SET(fd, &fdw);
timeout.tv_sec = 10;
timeout.tv_usec = 0;
rc = select(fd + 1, &fdr, &fdw, NULL, &timeout);
printf("rc is: %d\n", rc);
/*select調(diào)用失敗*/
if (rc < 0) {
fprintf(stderr, "connect error:%s\n", strerror(errno));
close(fd);
return -1;
}
/*連接超時*/
if (rc == 0) {
fprintf(stderr, "Connect timeout.\n");
close(fd);
return -1;
}
/*[1] 當(dāng)連接成功建立時,描述符變成可寫,rc=1*/
if (rc == 1 && FD_ISSET(fd, &fdw)) {
printf("Connect success\n");
close(fd);
return 0;
}
/*[2] 當(dāng)連接建立遇到錯誤時惊奇,描述符變?yōu)榧纯勺x互躬,也可寫,rc=2 遇到這種情況赊时,可調(diào)用getsockopt函數(shù)*/
if (rc == 2) {
if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errlen) == -1) {
fprintf(stderr, "getsockopt(SO_ERROR): %s", strerror(errno));
close(fd);
return -1;
}
if (err) {
errno = err;
fprintf(stderr, "connect error:%s\n", strerror(errno));
close(fd);
return -1;
}
}
}
fprintf(stderr, "connect failed, error:%s.\n", strerror(errno));
return -1;
}
return 0;
}