linux下connect阻塞處理

主要參考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;
}

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末吨铸,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子祖秒,更是在濱河造成了極大的恐慌诞吱,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,544評論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件竭缝,死亡現(xiàn)場離奇詭異房维,居然都是意外死亡,警方通過查閱死者的電腦和手機抬纸,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,430評論 3 392
  • 文/潘曉璐 我一進店門咙俩,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事阿趁∧せ祝” “怎么了?”我有些...
    開封第一講書人閱讀 162,764評論 0 353
  • 文/不壞的土叔 我叫張陵脖阵,是天一觀的道長皂股。 經(jīng)常有香客問我,道長命黔,這世上最難降的妖魔是什么呜呐? 我笑而不...
    開封第一講書人閱讀 58,193評論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮悍募,結(jié)果婚禮上蘑辑,老公的妹妹穿的比我還像新娘。我一直安慰自己坠宴,他們只是感情好洋魂,可當(dāng)我...
    茶點故事閱讀 67,216評論 6 388
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著啄踊,像睡著了一般忧设。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上颠通,一...
    開封第一講書人閱讀 51,182評論 1 299
  • 那天址晕,我揣著相機與錄音,去河邊找鬼顿锰。 笑死谨垃,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的硼控。 我是一名探鬼主播刘陶,決...
    沈念sama閱讀 40,063評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼牢撼!你這毒婦竟也來了匙隔?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,917評論 0 274
  • 序言:老撾萬榮一對情侶失蹤熏版,失蹤者是張志新(化名)和其女友劉穎纷责,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體撼短,經(jīng)...
    沈念sama閱讀 45,329評論 1 310
  • 正文 獨居荒郊野嶺守林人離奇死亡再膳,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,543評論 2 332
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了曲横。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片喂柒。...
    茶點故事閱讀 39,722評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出灾杰,到底是詐尸還是另有隱情蚊丐,我是刑警寧澤,帶...
    沈念sama閱讀 35,425評論 5 343
  • 正文 年R本政府宣布吭露,位于F島的核電站吠撮,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏讲竿。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,019評論 3 326
  • 文/蒙蒙 一弄屡、第九天 我趴在偏房一處隱蔽的房頂上張望题禀。 院中可真熱鬧,春花似錦膀捷、人聲如沸迈嘹。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,671評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽秀仲。三九已至,卻和暖如春壶笼,著一層夾襖步出監(jiān)牢的瞬間神僵,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,825評論 1 269
  • 我被黑心中介騙來泰國打工覆劈, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留保礼,地道東北人。 一個月前我還...
    沈念sama閱讀 47,729評論 2 368
  • 正文 我出身青樓责语,卻偏偏與公主長得像炮障,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子坤候,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,614評論 2 353