日期:2014-05-16  浏览次数:20686 次

linux下socket connect 阻塞方式 阻塞时间控制(转)

? ? ?同事今天问我,如何在linux下的c代码里面控制connect的阻塞时间。应用的背景是:linux下的c程序有两个目标IP需要connect,如果用阻塞方式,当其中一个IP不能连接的情况下,程序将阻塞在connect函数上。
???? 本来以为用setsockopt修改个什么参数就可以搞定,结果baidu了半天也没有结果。倒是在网上搜到很多这样的解决方案:将connect方式设置为非阻塞方式,这样程序一旦执行就会马上返回,但问题是,到底有没有连接上呢,你需要等待一段时间,然后使用函数判断连接是否正常。试了下,貌似由于系统的区别,这个方法在我的linux环境下不适用,主要是后面判断是否连接上的函数不行,不管连接上或没连接上都返回一个值。
????? 看来网上的方法也不能尽信,最后在公司前人的一段代码里面找到了解决方法,这里分享下:

  1. sigset(SIGALRM,?u_alarm_handler);
  2. alarm(2);
  3. code?=?connect(socket_fd,?(struct sockaddr*)&socket_st,?sizeof(struct sockaddr_in));
  4. alarm(0);
  5. sigrelse(SIGALRM);
? ?? 首先定义一个中断信号处理函数u_alarm_handler,用于超时后的报警处理,然后定义一个2秒的定时器,执行connect,当系统connect成功,则系统正常执行下去;如果connect不成功阻塞在这里,则超过定义的2秒后,系统会产生一个信号,触发执行u_alarm_handler函数, 当执行完u_alarm_handler后,程序将继续从connect的下面一行执行下去。
???? 其中,处理函数可以如下定义,也可以加入更多的错误处理。
?????
  1. void u_alarm_handler()
  2. {
  3. }
?? ? 这个方法相对网上的做法而言代码量小,更精巧,看来老代码也是有很多值得学习的地方。