linux下socket connect 阻塞方式阻塞時間控制

原創

2020-02-22 07:19

同事今天問我，如何在linux下的c代碼裏面控制connect的阻塞時間。應用的背景是：linux下的c程序有兩個目標IP需要connect，如果用阻塞方式，當其中一個IP不能連接的情況下，程序將阻塞在connect函數上。
本來以爲用setsockopt修改個什麼參數就可以搞定，結果baidu了半天也沒有結果。倒是在網上搜到很多這樣的解決方案：將connect方式設置爲非阻塞方式，這樣程序一旦執行就會馬上返回，但問題是，到底有沒有連接上呢，你需要等待一段時間，然後使用函數判斷連接是否正常。試了下，貌似由於系統的區別，這個方法在我的linux環境下不適用，主要是後面判斷是否連接上的函數不行，不管連接上或沒連接上都返回一個值。
看來網上的方法也不能盡信，最後在公司前人的一段代碼裏面找到了解決方法，這裏分享下：

sigset(SIGALRM, u_alarm_handler);
alarm(2);
code = connect(socket_fd, (struct sockaddr*)&socket_st, sizeof(struct sockaddr_in));
alarm(0);
sigrelse(SIGALRM);

     首先定義一箇中斷信號處理函數u_alarm_handler，用於超時後的報警處理，然後定義一個2秒的定時器，執行connect，當系統connect成功，則系統正常執行下去；如果connect不成功阻塞在這裏，則超過定義的2秒後，系統會產生一個信號，觸發執行u_alarm_handler函數，當執行完u_alarm_handler後，程序將繼續從connect的下面一行執行下去。
     其中，處理函數可以如下定義，也可以加入更多的錯誤處理。