github
/
aria2
mirror of https://github.com/aria2/aria2
1
0
Fork 0
Code Issues Releases Wiki Activity

Setup BT UDP SOCKS5 proxy on demand

pull/1857/head
myl7 2021-12-05 20:57:12 +08:00
parent 2806da3298
commit f2a66a68b3
No known key found for this signature in database
GPG Key ID: 04F1013B67177C88
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/DHTSetup.cc
View File

@ -114,7 +114,10 @@ DHTSetup::setup(DownloadEngine* e, int family)
}
uint16_t port;
auto connection = make_unique<DHTConnectionSocksProxyImpl>(family);
const std::string& proxyUri = e->getOption()->get(PREF_BT_UDP_SOCKS_PROXY);
auto connection = proxyUri.empty()
? make_unique<DHTConnectionImpl>(family)
: make_unique<DHTConnectionSocksProxyImpl>(family);
{
port = e->getBtRegistry()->getUdpPort();
const std::string& addr = e->getOption()->get(
@ -140,18 +143,18 @@ DHTSetup::setup(DownloadEngine* e, int family)
}
A2_LOG_DEBUG(fmt("Initialized local node ID=%s",
util::toHex(localNode->getID(), DHT_ID_LENGTH).c_str()));
{
if (!proxyUri.empty()) {
auto c = static_cast<DHTConnectionSocksProxyImpl*>(connection.get());
const std::string& user =
e->getOption()->get(PREF_BT_UDP_SOCKS_PROXY_USER);
const std::string& passwd =
e->getOption()->get(PREF_BT_UDP_SOCKS_PROXY_PASSWD);
uri::UriStruct us;
uri::parse(us, e->getOption()->get(PREF_BT_UDP_SOCKS_PROXY));
uri::parse(us, proxyUri);
const std::string& host = us.host;
uint16_t port = us.port;
if (!connection->startProxy(host, port, user, passwd,
localNode->getIPAddress(),
localNode->getPort())) {
if (!c->startProxy(host, port, user, passwd, localNode->getIPAddress(),
localNode->getPort())) {
throw DL_ABORT_EX("Error occurred while connecting to SOCKS5 relay "
"server for UDP proxy for DHT and UDP trackers");
}