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

同事今天問我,如何在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()
{
}
這個方法相對網上的作法而言代碼量小,更精巧,看來老代碼也是有不少值得學習的地方。    

linux

相關文章
相關標籤/搜索