mirror of https://github.com/PurpleI2P/i2pd.git
orignal
3 months ago
5 changed files with 170 additions and 118 deletions
@ -0,0 +1,154 @@
@@ -0,0 +1,154 @@
|
||||
/*
|
||||
* Copyright (c) 2024, The PurpleI2P Project |
||||
* |
||||
* This file is part of Purple i2pd project and licensed under BSD3 |
||||
* |
||||
* See full license text in LICENSE file at top of project tree |
||||
* |
||||
*/ |
||||
|
||||
#ifndef SOCKS5_H__ |
||||
#define SOCKS5_H__ |
||||
|
||||
#include <string> |
||||
#include <memory> |
||||
#include <boost/asio.hpp> |
||||
#include "I2PEndian.h" |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace transport |
||||
{ |
||||
// SOCKS5 constants
|
||||
const uint8_t SOCKS5_VER = 0x05; |
||||
const uint8_t SOCKS5_CMD_CONNECT = 0x01; |
||||
const uint8_t SOCKS5_CMD_UDP_ASSOCIATE = 0x03; |
||||
const uint8_t SOCKS5_ATYP_IPV4 = 0x01; |
||||
const uint8_t SOCKS5_ATYP_IPV6 = 0x04; |
||||
const uint8_t SOCKS5_ATYP_NAME = 0x03; |
||||
const size_t SOCKS5_UDP_IPV4_REQUEST_HEADER_SIZE = 10; |
||||
const size_t SOCKS5_UDP_IPV6_REQUEST_HEADER_SIZE = 22; |
||||
|
||||
// SOCKS5 handshake
|
||||
template<typename Socket, typename Handler> |
||||
void Socks5Connect (Socket& s, Handler handler, std::shared_ptr<std::vector<uint8_t> > buff, uint16_t port) |
||||
{ |
||||
if (buff && buff->size () >= 6) |
||||
{ |
||||
(*buff)[0] = SOCKS5_VER; |
||||
(*buff)[1] = SOCKS5_CMD_CONNECT; |
||||
(*buff)[2] = 0x00; |
||||
htobe16buf(buff->data () + buff->size () - 2, port); |
||||
boost::asio::async_write(s, boost::asio::buffer(*buff), boost::asio::transfer_all(), |
||||
[buff, &s, handler](const boost::system::error_code& ec, std::size_t transferred) |
||||
{ |
||||
(void) transferred; |
||||
if (!ec) |
||||
{ |
||||
auto readbuff = std::make_shared<std::vector<int8_t> >(262); // max possible
|
||||
boost::asio::async_read(s, boost::asio::buffer(readbuff->data (), 7), boost::asio::transfer_all(), // read min reply size
|
||||
[readbuff, &s, handler](const boost::system::error_code& ec, std::size_t transferred) |
||||
{ |
||||
if (!ec) |
||||
{ |
||||
if (!(*readbuff)[1]) // succeeded
|
||||
{ |
||||
boost::system::error_code ec; |
||||
size_t moreBytes = s.available(ec); |
||||
if (!ec && moreBytes) // read remaining portion of reply
|
||||
boost::asio::read (s, boost::asio::buffer(readbuff->data (), moreBytes), boost::asio::transfer_all (), ec); |
||||
if (!ec) |
||||
handler (boost::system::error_code ()); |
||||
else |
||||
handler (boost::asio::error::make_error_code (boost::asio::error::connection_aborted)); |
||||
} |
||||
else |
||||
handler (boost::asio::error::make_error_code (boost::asio::error::connection_refused)); |
||||
} |
||||
else |
||||
handler (ec); |
||||
}); |
||||
} |
||||
else |
||||
handler (ec); |
||||
}); |
||||
} |
||||
else |
||||
handler (boost::asio::error::make_error_code (boost::asio::error::no_buffer_space)); |
||||
} |
||||
|
||||
template<typename Socket, typename Handler> |
||||
void Socks5Connect (Socket& s, const boost::asio::ip::tcp::endpoint& ep, Handler handler) |
||||
{ |
||||
std::shared_ptr<std::vector<uint8_t> > buff; |
||||
if(ep.address ().is_v4 ()) |
||||
{ |
||||
buff = std::make_shared<std::vector<uint8_t> >(10); |
||||
(*buff)[3] = SOCKS5_ATYP_IPV4; |
||||
auto addrbytes = ep.address ().to_v4().to_bytes(); |
||||
memcpy(buff->data () + 4, addrbytes.data(), 4); |
||||
} |
||||
else if (ep.address ().is_v6 ()) |
||||
{ |
||||
buff = std::make_shared<std::vector<uint8_t> >(22); |
||||
(*buff)[3] = SOCKS5_ATYP_IPV6; |
||||
auto addrbytes = ep.address ().to_v6().to_bytes(); |
||||
memcpy(buff->data () + 4, addrbytes.data(), 16); |
||||
} |
||||
if (buff) |
||||
Socks5Connect (s, handler, buff, ep.port ()); |
||||
else |
||||
handler (boost::asio::error::make_error_code (boost::asio::error::fault)); |
||||
} |
||||
|
||||
template<typename Socket, typename Handler> |
||||
void Socks5Connect (Socket& s, const std::pair<std::string, uint16_t>& ep, Handler handler) |
||||
{ |
||||
auto& addr = ep.first; |
||||
if (addr.length () <= 255) |
||||
{ |
||||
auto buff = std::make_shared<std::vector<uint8_t> >(addr.length () + 7); |
||||
(*buff)[3] = SOCKS5_ATYP_NAME; |
||||
(*buff)[4] = addr.length (); |
||||
memcpy (buff->data () + 5, addr.c_str (), addr.length ()); |
||||
Socks5Connect (s, handler, buff, ep.second); |
||||
} |
||||
else |
||||
handler (boost::asio::error::make_error_code (boost::asio::error::name_too_long)); |
||||
} |
||||
|
||||
|
||||
template<typename Socket, typename Endpoint, typename Handler> |
||||
void Socks5Handshake (Socket& s, Endpoint ep, Handler handler) |
||||
{ |
||||
static const uint8_t methodSelection[3] = { SOCKS5_VER, 0x01, 0x00 }; // 1 method, no auth
|
||||
boost::asio::async_write(s, boost::asio::buffer(methodSelection, 3), boost::asio::transfer_all(), |
||||
[&s, ep, handler] (const boost::system::error_code& ec, std::size_t transferred) |
||||
{ |
||||
(void) transferred; |
||||
if (!ec) |
||||
{ |
||||
auto readbuff = std::make_shared<std::vector<uint8_t> >(2); |
||||
boost::asio::async_read(s, boost::asio::buffer(*readbuff), boost::asio::transfer_all(), |
||||
[&s, ep, handler, readbuff] (const boost::system::error_code& ec, std::size_t transferred) |
||||
{ |
||||
if (!ec) |
||||
{ |
||||
if (transferred == 2 && (*readbuff)[1] == 0x00) // no auth
|
||||
Socks5Connect (s, ep, handler); |
||||
else |
||||
handler (boost::asio::error::make_error_code (boost::asio::error::invalid_argument)); |
||||
} |
||||
else |
||||
handler (ec); |
||||
}); |
||||
} |
||||
else |
||||
handler (ec); |
||||
}); |
||||
} |
||||
|
||||
} |
||||
} |
||||
|
||||
#endif |
||||
Loading…
Reference in new issue