commit 0102be9d7ac3476729303f7eb27e68adafadf7e8
author Alexander Afanasyev <alexander.afanasyev@ucla.edu> Thu Jan 09 14:31:09 2014 -0800
committer Alexander Afanasyev <alexander.afanasyev@ucla.edu> Thu Jan 09 15:14:31 2014 -0800
tree f7da7b858fe8278c428211c17ac14c7ba53fb8c5
parent f75a0aad5dba72d9bfcbb2207194ec5bec711c0a

transport/unix: Implementing connect timeout

This timeout is not strictly necessary for unix socket, but will be
useful later for other types of transports.

Change-Id: I2c3b155e0e75da4a127fffba2fecb964f2d8eff1

src/transport/unix-transport.cpp

diff --git a/src/transport/unix-transport.cpp b/src/transport/unix-transport.cpp
index 2704df1..5ee6a61 100644
--- a/src/transport/unix-transport.cpp
+++ b/src/transport/unix-transport.cpp
@@ -33,6 +33,7 @@
     , socket_(*transport_.ioService_)
     , partialDataSize_(0)
     , connectionInProgress_(false)
+    , connectTimer_(*transport_.ioService_)
   {
   }
 
@@ -40,6 +41,7 @@
   connectHandler(const boost::system::error_code& error)
   {
     connectionInProgress_ = false;
+    connectTimer_.cancel();
 
     if (!error)
       {
@@ -62,6 +64,18 @@
         throw Transport::Error(error, "error while connecting to the forwarder");
       }
   }
+
+  void
+  connectTimeoutHandler(const boost::system::error_code& error)
+  {
+    if (error) // e.g., cancelled timer
+      return;
+
+    connectionInProgress_ = false;
+    transport_.isConnected_ = false;
+    socket_.close();
+    throw Transport::Error(error, "error while connecting to the forwarder");
+  }
   
   void
   connect()
@@ -71,12 +85,18 @@
       socket_.open();
       socket_.async_connect(protocol::endpoint(transport_.unixSocket_),
                             func_lib::bind(&Impl::connectHandler, this, _1));
+
+      // Wait at most 4 seconds to connect
+      /// @todo Decide whether this number should be configurable
+      connectTimer_.expires_from_now(boost::posix_time::seconds(4));
+      connectTimer_.async_wait(func_lib::bind(&Impl::connectTimeoutHandler, this, _1));
     }
   }
 
   void 
   close()
   {
+    connectTimer_.cancel();
     socket_.close();
     transport_.isConnected_ = false;
   }
@@ -209,6 +229,8 @@
 
   std::list< Block > sendQueue_;
   bool connectionInProgress_;
+
+  boost::asio::deadline_timer connectTimer_;
 };
 
 UnixTransport::UnixTransport(const std::string &unixSocket/* = "/tmp/.ndnd.sock"*/)