|
|
@ -1189,17 +1189,19 @@ namespace transport |
|
|
|
case eSocksProxy: |
|
|
|
case eSocksProxy: |
|
|
|
{ |
|
|
|
{ |
|
|
|
// TODO: support username/password auth etc
|
|
|
|
// TODO: support username/password auth etc
|
|
|
|
uint8_t buff[3] = {0x05, 0x01, 0x00}; |
|
|
|
static const uint8_t buff[3] = {0x05, 0x01, 0x00}; |
|
|
|
boost::asio::async_write(conn->GetSocket(), boost::asio::buffer(buff, 3), boost::asio::transfer_all(), [=] (const boost::system::error_code & ec, std::size_t transferred) { |
|
|
|
boost::asio::async_write(conn->GetSocket(), boost::asio::buffer(buff, 3), boost::asio::transfer_all(), |
|
|
|
|
|
|
|
[] (const boost::system::error_code & ec, std::size_t transferred) |
|
|
|
|
|
|
|
{ |
|
|
|
(void) transferred; |
|
|
|
(void) transferred; |
|
|
|
if(ec) |
|
|
|
if(ec) |
|
|
|
{ |
|
|
|
{ |
|
|
|
LogPrint(eLogWarning, "NTCP2: socks5 write error ", ec.message()); |
|
|
|
LogPrint(eLogWarning, "NTCP2: socks5 write error ", ec.message()); |
|
|
|
} |
|
|
|
} |
|
|
|
}); |
|
|
|
}); |
|
|
|
uint8_t readbuff[2]; |
|
|
|
auto readbuff = std::make_shared<std::array<uint8_t, 2> >(); |
|
|
|
boost::asio::async_read(conn->GetSocket(), boost::asio::buffer(readbuff, 2), |
|
|
|
boost::asio::async_read(conn->GetSocket(), boost::asio::buffer(*readbuff, 2), |
|
|
|
[=](const boost::system::error_code & ec, std::size_t transferred) |
|
|
|
[this, readbuff, timer, conn, host, port, addrtype](const boost::system::error_code & ec, std::size_t transferred) |
|
|
|
{ |
|
|
|
{ |
|
|
|
LogPrint(eLogError, "NTCP2: ", transferred); |
|
|
|
LogPrint(eLogError, "NTCP2: ", transferred); |
|
|
|
if(ec) |
|
|
|
if(ec) |
|
|
@ -1211,19 +1213,19 @@ namespace transport |
|
|
|
} |
|
|
|
} |
|
|
|
else if(transferred == 2) |
|
|
|
else if(transferred == 2) |
|
|
|
{ |
|
|
|
{ |
|
|
|
if(readbuff[1] == 0xba) |
|
|
|
if((*readbuff)[1] == 0xba) |
|
|
|
{ |
|
|
|
{ |
|
|
|
AfterSocksHandshake(conn, timer, host, port, addrtype); |
|
|
|
AfterSocksHandshake(conn, timer, host, port, addrtype); |
|
|
|
return; |
|
|
|
return; |
|
|
|
} |
|
|
|
} |
|
|
|
else if (readbuff[1] == 0xff) |
|
|
|
else if ((*readbuff)[1] == 0xff) |
|
|
|
{ |
|
|
|
{ |
|
|
|
LogPrint(eLogError, "NTCP2: socks5 proxy rejected authentication"); |
|
|
|
LogPrint(eLogError, "NTCP2: socks5 proxy rejected authentication"); |
|
|
|
timer->cancel(); |
|
|
|
timer->cancel(); |
|
|
|
conn->Terminate(); |
|
|
|
conn->Terminate(); |
|
|
|
return; |
|
|
|
return; |
|
|
|
} |
|
|
|
} |
|
|
|
LogPrint(eLogError, "NTCP2:", readbuff[1]); |
|
|
|
LogPrint(eLogError, "NTCP2:", (*readbuff)[1]); |
|
|
|
} |
|
|
|
} |
|
|
|
LogPrint(eLogError, "NTCP2: socks5 server gave invalid response"); |
|
|
|
LogPrint(eLogError, "NTCP2: socks5 server gave invalid response"); |
|
|
|
timer->cancel(); |
|
|
|
timer->cancel(); |
|
|
@ -1245,7 +1247,9 @@ namespace transport |
|
|
|
std::ostream out(&writebuff); |
|
|
|
std::ostream out(&writebuff); |
|
|
|
out << req.to_string(); |
|
|
|
out << req.to_string(); |
|
|
|
|
|
|
|
|
|
|
|
boost::asio::async_write(conn->GetSocket(), writebuff.data(), boost::asio::transfer_all(), [=](const boost::system::error_code & ec, std::size_t transferred) { |
|
|
|
boost::asio::async_write(conn->GetSocket(), writebuff.data(), boost::asio::transfer_all(), |
|
|
|
|
|
|
|
[](const boost::system::error_code & ec, std::size_t transferred) |
|
|
|
|
|
|
|
{ |
|
|
|
(void) transferred; |
|
|
|
(void) transferred; |
|
|
|
if(ec) |
|
|
|
if(ec) |
|
|
|
LogPrint(eLogError, "NTCP2: http proxy write error ", ec.message()); |
|
|
|
LogPrint(eLogError, "NTCP2: http proxy write error ", ec.message()); |
|
|
@ -1253,7 +1257,7 @@ namespace transport |
|
|
|
|
|
|
|
|
|
|
|
boost::asio::streambuf * readbuff = new boost::asio::streambuf; |
|
|
|
boost::asio::streambuf * readbuff = new boost::asio::streambuf; |
|
|
|
boost::asio::async_read_until(conn->GetSocket(), *readbuff, "\r\n\r\n", |
|
|
|
boost::asio::async_read_until(conn->GetSocket(), *readbuff, "\r\n\r\n", |
|
|
|
[=] (const boost::system::error_code & ec, std::size_t transferred) |
|
|
|
[this, readbuff, timer, conn] (const boost::system::error_code & ec, std::size_t transferred) |
|
|
|
{ |
|
|
|
{ |
|
|
|
if(ec) |
|
|
|
if(ec) |
|
|
|
{ |
|
|
|
{ |
|
|
@ -1275,10 +1279,8 @@ namespace transport |
|
|
|
return; |
|
|
|
return; |
|
|
|
} |
|
|
|
} |
|
|
|
else |
|
|
|
else |
|
|
|
{ |
|
|
|
|
|
|
|
LogPrint(eLogError, "NTCP2: http proxy rejected request ", res.code); |
|
|
|
LogPrint(eLogError, "NTCP2: http proxy rejected request ", res.code); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
else |
|
|
|
else |
|
|
|
LogPrint(eLogError, "NTCP2: http proxy gave malformed response"); |
|
|
|
LogPrint(eLogError, "NTCP2: http proxy gave malformed response"); |
|
|
|
timer->cancel(); |
|
|
|
timer->cancel(); |
|
|
@ -1304,7 +1306,8 @@ namespace transport |
|
|
|
auto timeout = NTCP_CONNECT_TIMEOUT * 5; |
|
|
|
auto timeout = NTCP_CONNECT_TIMEOUT * 5; |
|
|
|
conn->SetTerminationTimeout(timeout * 2); |
|
|
|
conn->SetTerminationTimeout(timeout * 2); |
|
|
|
timer->expires_from_now (boost::posix_time::seconds(timeout)); |
|
|
|
timer->expires_from_now (boost::posix_time::seconds(timeout)); |
|
|
|
timer->async_wait ([conn, timeout](const boost::system::error_code& ecode) { |
|
|
|
timer->async_wait ([conn, timeout](const boost::system::error_code& ecode) |
|
|
|
|
|
|
|
{ |
|
|
|
if (ecode != boost::asio::error::operation_aborted) |
|
|
|
if (ecode != boost::asio::error::operation_aborted) |
|
|
|
{ |
|
|
|
{ |
|
|
|
LogPrint (eLogInfo, "NTCP2: Not connected in ", timeout, " seconds"); |
|
|
|
LogPrint (eLogInfo, "NTCP2: Not connected in ", timeout, " seconds"); |
|
|
|