在C ++中使用connect()时设置超时

时间:2009-10-20 05:48:31

标签: c++ sockets timeout

我在linux上使用c ++中的套接字连接到使用connect()的服务器,但它似乎没有超时值。什么是最简单/最好的方式给它几秒钟的超时。我将发布我所拥有的代码,包括对connect()的阻塞调用:

using namespace std;
int main( int argc, char* argv[] )
{
  // Interpret the command line arguments
  string uri_string = "";

  if ( argc != 2 )
  {
    cerr << "Usage: " << argv[0];
    cerr << " <URI>" << endl;
    return 1;
  }
  else
  {
    uri_string = argv[1];
  }

  // Create URI object and have it parse the uri_string
  URI *uri = URI::Parse(uri_string);

  if ( uri == NULL )
  {
    cerr << "Error: Cannot parse URI." << endl;
    return 2;
  }

  // Check the port number specified, if none use port 80
  unsigned port = 80;
  if ( uri->Is_port_defined() )
  {
    port = uri->Get_port();
  }

  // Create TCP socket and connect to server
  int tcp_sock = socket( AF_INET, SOCK_STREAM, 0 );
  if ( tcp_sock < 0 )
  {
    cerr << "Unable to create TCP socket." << endl;
    return 3;
  }

  sockaddr_in server;
  socklen_t slen = sizeof(server);

  server.sin_family = AF_INET;
  server.sin_port = htons( port );
  cout << "Resolving " << uri->Get_host() << "... ";
  hostent *hostp = gethostbyname( uri->Get_host().c_str() );
  memcpy( &server.sin_addr, hostp->h_addr, hostp->h_length );

  struct in_addr **pptr;
  pptr = (struct in_addr **)hostp->h_addr_list;

  string ip_addr = "";
  while( *pptr != NULL ) {
    ip_addr += inet_ntoa(**(pptr++));
  }
  cout << ip_addr << endl;

  cout << "Connecting to " << uri->Get_host() << "|" << ip_addr << "|:";
  cout << port << "... ";
  if ( connect( tcp_sock, (sockaddr*)&server, slen ) < 0 )
  {
    cerr << "Unable to connect to server via TCP." << endl;
    close( tcp_sock );
    return 4;
  }
  cout << "connected." << endl;

  // rest of code
}

4 个答案:

答案 0 :(得分:5)

使套接字成为非阻塞套接字,然后使用带有超时值的select()或poll()来检查可写性。如果select()以超时返回,则表示没有及时连接,您可以关闭套接字并处理连接失败。如果它完成后返回,一切都很好,你可以继续。

答案 1 :(得分:1)

查看this newsgroup post,了解如何在Lunux中为套接字连接设置超时的详细信息和示例代码。基本思想是使用非阻塞套接字,然后等待连接。

答案 2 :(得分:1)

为套接字设置非阻塞模式,并使用select检查是否建立了连接。 here是样本

答案 3 :(得分:1)

在没有select()的情况下进行非阻塞的另一种方法是检查EWOULDBLOCK的GetLastError()。

// set socket as nonblocking
unsigned long value = 1;
if (ioctlsocket(yourSocket, FIONBIO, &value) == SOCKET_ERROR)
    return SOCKET_ERROR;

// (call connect here)

// check if connect would block
if (GetLastError() == EWOULDBLOCK)
    return TIMEOUT;