Mirrors
/
wownero
mirror of https://git.wownero.com/wownero/wownero.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #7890 

2a8a511 Quicker resource cleanup on p2p socks timeout (Lee *!* Clagett)
This commit is contained in:
luigi1111 2021-09-23 22:06:35 -05:00
parent 75ee729a3a 2a8a51129f
commit 237acd120e
No known key found for this signature in database
GPG Key ID: F4ACA0183641E010
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/net/socks.h
View File

@ -201,6 +201,13 @@ namespace socks
std::shared_ptr<client> self_; std::shared_ptr<client> self_;
void operator()(boost::system::error_code error = boost::system::error_code{}); void operator()(boost::system::error_code error = boost::system::error_code{});
}; };
//! Calls `async_close` on `self` at destruction. NOP if `nullptr`.
struct close_on_exit
{
std::shared_ptr<client> self;
~close_on_exit() { async_close{std::move(self)}(); }
};
}; };
template<typename Handler> template<typename Handler>

5
src/p2p/net_node.cpp
View File

@ -342,6 +342,7 @@ namespace nodetool
} }
}; };
net::socks::client::close_on_exit close_client{};
boost::unique_future<client_result> socks_result{}; boost::unique_future<client_result> socks_result{};
{ {
boost::promise<client_result> socks_promise{}; boost::promise<client_result> socks_promise{};
@ -350,6 +351,7 @@ namespace nodetool
auto client = net::socks::make_connect_client( auto client = net::socks::make_connect_client(
boost::asio::ip::tcp::socket{service}, net::socks::version::v4a, notify{std::move(socks_promise)} boost::asio::ip::tcp::socket{service}, net::socks::version::v4a, notify{std::move(socks_promise)}
); );
close_client.self = client;
if (!start_socks(std::move(client), proxy, remote)) if (!start_socks(std::move(client), proxy, remote))
return boost::none; return boost::none;
} }
@ -371,7 +373,10 @@ namespace nodetool
{ {
auto result = socks_result.get(); auto result = socks_result.get();
if (!result.first) if (!result.first)
{
close_client.self.reset();
return {std::move(result.second)}; return {std::move(result.second)};
}
MERROR("Failed to make socks connection to " << remote.str() << " (via " << proxy << "): " << result.first.message()); MERROR("Failed to make socks connection to " << remote.str() << " (via " << proxy << "): " << result.first.message());
} }