同事今天問我,如何在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的下面一行執行下去。
其中,處理函數可以如下定義,也可以加入更多的錯誤處理。
- void u_alarm_handler()
- {
- }
這個方法相對網上的做法而言代碼量小,更精巧,看來老代碼也是有很多值得學習的地方。
http://blog.chinaunix.net/space.php?uid=16792259&do=blog&id=3064785