i2pd/SOCKS.cpp

226 lines
5.5 KiB
C++
Raw Normal View History

2014-07-14 20:40:06 +04:00
#include "SOCKS.h"
#include "Identity.h"
#include "NetDb.h"
#include "Destination.h"
2014-10-16 04:52:17 +04:00
#include "ClientContext.h"
#include "I2PEndian.h"
2014-07-14 20:40:06 +04:00
#include <cstring>
namespace i2p
{
namespace proxy
{
2014-09-14 19:38:34 +04:00
const uint8_t socks_leaseset_timeout = 10;
const uint8_t socks_timeout = 60;
2014-07-14 20:40:06 +04:00
void SOCKS4AHandler::AsyncSockRead()
{
LogPrint("--- socks4a async sock read");
if(m_sock) {
if (m_state == INITIAL) {
m_sock->async_receive(boost::asio::buffer(m_sock_buff, socks_buffer_size),
std::bind(&SOCKS4AHandler::HandleSockRecv, this,
std::placeholders::_1, std::placeholders::_2));
} else {
LogPrint("--- socks4a state?? ", m_state);
2014-07-14 20:40:06 +04:00
}
} else {
LogPrint("--- socks4a no socket for read");
}
}
void SOCKS4AHandler::Terminate() {
CloseStream();
CloseSock();
delete this; // HACK: ew
2014-07-14 20:40:06 +04:00
}
void SOCKS4AHandler::SocksFailed()
{
LogPrint("--- socks4a failed");
//TODO: send the right response
boost::asio::async_write(*m_sock, boost::asio::buffer("\x00\x5b 12345"),
std::bind(&SOCKS4AHandler::SentSocksFailed, this,
std::placeholders::_1));
2014-07-14 20:40:06 +04:00
}
2014-07-14 20:40:06 +04:00
void SOCKS4AHandler::CloseSock()
{
if (m_sock) {
LogPrint("--- socks4a close sock");
m_sock->close();
delete m_sock;
m_sock = nullptr;
}
}
void SOCKS4AHandler::CloseStream()
{
if (m_stream) {
LogPrint("--- socks4a close stream");
m_stream.reset ();
}
}
2014-07-14 20:40:06 +04:00
2014-09-14 19:38:34 +04:00
const size_t socks_hostname_size = 1024;
const size_t socks_ident_size = 1024;
const size_t destb32_len = 52;
2014-07-14 20:40:06 +04:00
void SOCKS4AHandler::HandleSockRecv(const boost::system::error_code & ecode, std::size_t len)
{
LogPrint("--- socks4a sock recv: ", len);
//TODO: we may not have received all the data :(
2014-07-14 20:40:06 +04:00
if(ecode) {
LogPrint(" --- sock recv got error: ", ecode);
Terminate();
return;
}
char hostbuff[socks_hostname_size];
char identbuff[socks_ident_size];
std::memset(hostbuff, 0, sizeof(hostbuff));
std::memset(identbuff, 0, sizeof(hostbuff));
std::string dest;
2014-07-14 20:40:06 +04:00
uint16_t port = 0;
uint32_t address = 0;
uint16_t idx1 = 0;
uint16_t idx2 = 0;
2014-07-14 20:40:06 +04:00
LogPrint(eLogDebug,"--- socks4a state initial ", len);
2014-07-14 20:40:06 +04:00
// check valid request
if( m_sock_buff[0] != 4 || m_sock_buff[1] != 1 || m_sock_buff[len-1] ) {
LogPrint(eLogError,"--- socks4a rejected invalid");
SocksFailed();
return;
}
2014-07-14 20:40:06 +04:00
// get port
port = bufbe16toh(m_sock_buff + 2);
// get address
address = bufbe32toh(m_sock_buff + 4);
2014-07-14 20:40:06 +04:00
// check valid request
if( address == 0 || address > 255 ) {
LogPrint(eLogError,"--- socks4a rejected because it's actually socks4");
SocksFailed();
return;
}
2014-07-14 20:40:06 +04:00
// read ident
do {
LogPrint(eLogDebug,"--- socks4a ", (int) m_sock_buff[9+idx1]);
identbuff[idx1] = m_sock_buff[8+idx1];
} while( identbuff[idx1++] && idx1 < socks_ident_size );
2014-07-14 20:40:06 +04:00
LogPrint(eLogInfo,"--- socks4a ident ", identbuff);
// read hostname
do {
hostbuff[idx2] = m_sock_buff[8+idx1+idx2];
} while( hostbuff[idx2++] && idx2 < socks_hostname_size );
LogPrint(eLogInfo,"--- socks4a requested ", hostbuff, ":" , port);
dest = std::string(hostbuff);
if(dest.find(".i2p") == std::string::npos) {
LogPrint("--- socks4a invalid hostname: ", dest);
SocksFailed();
return;
2014-07-14 20:40:06 +04:00
}
m_parent->GetLocalDestination ()->CreateStream (
std::bind (&SOCKS4AHandler::HandleStreamRequestComplete,
this, std::placeholders::_1), dest);
2014-07-14 20:40:06 +04:00
}
void SOCKS4AHandler::ConnectionSuccess()
{
LogPrint(eLogInfo,"--- socks4a connection success");
//TODO: send the right response
boost::asio::async_write(*m_sock, boost::asio::buffer("\x00\x5a 12345"),
std::bind(&SOCKS4AHandler::SentConnectionSuccess, this,
std::placeholders::_1));
}
void SOCKS4AHandler::SentSocksFailed(const boost::system::error_code & ecode)
2014-07-14 20:40:06 +04:00
{
if (!ecode) {
Terminate();
}
else
{
LogPrint (eLogError,"--- socks4a Closing socket after sending failure because: ", ecode.message ());
2014-07-14 20:40:06 +04:00
Terminate();
}
}
void SOCKS4AHandler::SentConnectionSuccess(const boost::system::error_code & ecode)
2014-07-14 20:40:06 +04:00
{
if (!ecode) {
LogPrint (eLogInfo,"--- socks4a New I2PTunnel connection");
auto connection = std::make_shared<i2p::client::I2PTunnelConnection>((i2p::client::I2PTunnel *)m_parent, m_sock, m_stream);
m_parent->AddConnection (connection);
connection->I2PConnect ();
}
else
{
LogPrint (eLogError,"--- socks4a Closing socket after sending success because: ", ecode.message ());
Terminate();
}
2014-07-14 20:40:06 +04:00
}
void SOCKS4AHandler::HandleStreamRequestComplete (std::shared_ptr<i2p::stream::Stream> stream)
2014-07-14 20:40:06 +04:00
{
if (stream)
{
m_stream = stream;
2014-07-14 20:40:06 +04:00
ConnectionSuccess();
}
else
{
LogPrint (eLogError,"--- socks4a Issue when creating the stream, check the previous warnings for more info.");
2014-07-14 20:40:06 +04:00
SocksFailed();
}
}
void SOCKS4AServer::Start ()
2014-07-14 20:40:06 +04:00
{
m_Acceptor.listen ();
Accept ();
2014-07-14 20:40:06 +04:00
}
void SOCKS4AServer::Stop ()
2014-07-14 20:40:06 +04:00
{
m_Acceptor.close();
m_Timer.cancel ();
ClearConnections ();
2014-07-14 20:40:06 +04:00
}
void SOCKS4AServer::Accept ()
2014-07-14 20:40:06 +04:00
{
auto newSocket = new boost::asio::ip::tcp::socket (GetService ());
m_Acceptor.async_accept (*newSocket, std::bind (&SOCKS4AServer::HandleAccept, this,
std::placeholders::_1, newSocket));
}
void SOCKS4AServer::HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket)
2014-07-14 20:40:06 +04:00
{
if (!ecode)
{
LogPrint(eLogDebug,"--- socks4a accepted");
new SOCKS4AHandler(this, socket);
2014-07-14 20:40:06 +04:00
Accept();
}
else
{
LogPrint (eLogError,"--- socks4a Closing socket on accept because: ", ecode.message ());
delete socket;
}
2014-07-14 20:40:06 +04:00
}
2014-07-14 20:40:06 +04:00
}
}