Add timeout for run_next ()

nanocurrency · Mar 7, 2020 · e28dad0 · e28dad0
1 parent b1e9d6a
commit e28dad0
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/nano/node/bootstrap/bootstrap_server.cpp b/nano/node/bootstrap/bootstrap_server.cpp
@@ -692,8 +692,19 @@ void nano::bootstrap_server::run_next (nano::unique_lock<std::mutex> & lock_a)
 		// Realtime
 		auto request (std::move (requests.front ()));
 		requests.pop ();
+		auto timeout_check (requests.empty ());
 		lock_a.unlock ();
 		request->visit (visitor);
+		if (timeout_check)
+		{
+			std::weak_ptr<nano::bootstrap_server> this_w (shared_from_this ());
+			node->alarm.add (std::chrono::steady_clock::now () + (node->config.tcp_io_timeout * 2) + std::chrono::seconds (1), [this_w]() {
+				if (auto this_l = this_w.lock ())
+				{
+					this_l->timeout ();
+				}
+			});
+		}
 	}
 }