|
|
|
@ -15,9 +15,9 @@ namespace i2p
@@ -15,9 +15,9 @@ namespace i2p
|
|
|
|
|
{ |
|
|
|
|
namespace client |
|
|
|
|
{ |
|
|
|
|
SAMSocket::SAMSocket (SAMBridge& owner): |
|
|
|
|
SAMSocket::SAMSocket (SAMBridge& owner): |
|
|
|
|
m_Owner (owner), m_Socket (m_Owner.GetService ()), m_Timer (m_Owner.GetService ()), |
|
|
|
|
m_BufferOffset (0), m_SocketType (eSAMSocketTypeUnknown), m_IsSilent (false), |
|
|
|
|
m_BufferOffset (0), m_SocketType (eSAMSocketTypeUnknown), m_IsSilent (false), |
|
|
|
|
m_Stream (nullptr), m_Session (nullptr) |
|
|
|
|
{ |
|
|
|
|
} |
|
|
|
@ -25,21 +25,21 @@ namespace client
@@ -25,21 +25,21 @@ namespace client
|
|
|
|
|
SAMSocket::~SAMSocket () |
|
|
|
|
{ |
|
|
|
|
Terminate (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::CloseStream () |
|
|
|
|
{ |
|
|
|
|
if (m_Stream) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
m_Stream->Close (); |
|
|
|
|
m_Stream.reset (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::Terminate () |
|
|
|
|
{ |
|
|
|
|
CloseStream (); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
switch (m_SocketType) |
|
|
|
|
{ |
|
|
|
|
case eSAMSocketTypeSession: |
|
|
|
@ -47,14 +47,14 @@ namespace client
@@ -47,14 +47,14 @@ namespace client
|
|
|
|
|
break; |
|
|
|
|
case eSAMSocketTypeStream: |
|
|
|
|
{ |
|
|
|
|
if (m_Session) |
|
|
|
|
if (m_Session) |
|
|
|
|
m_Session->DelSocket (shared_from_this ()); |
|
|
|
|
break; |
|
|
|
|
} |
|
|
|
|
case eSAMSocketTypeAcceptor: |
|
|
|
|
{ |
|
|
|
|
if (m_Session) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
m_Session->DelSocket (shared_from_this ()); |
|
|
|
|
if (m_Session->localDestination) |
|
|
|
|
m_Session->localDestination->StopAcceptingStreams (); |
|
|
|
@ -71,8 +71,8 @@ namespace client
@@ -71,8 +71,8 @@ namespace client
|
|
|
|
|
|
|
|
|
|
void SAMSocket::ReceiveHandshake () |
|
|
|
|
{ |
|
|
|
|
m_Socket.async_read_some (boost::asio::buffer(m_Buffer, SAM_SOCKET_BUFFER_SIZE), |
|
|
|
|
std::bind(&SAMSocket::HandleHandshakeReceived, shared_from_this (), |
|
|
|
|
m_Socket.async_read_some (boost::asio::buffer(m_Buffer, SAM_SOCKET_BUFFER_SIZE), |
|
|
|
|
std::bind(&SAMSocket::HandleHandshakeReceived, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -85,7 +85,7 @@ namespace client
@@ -85,7 +85,7 @@ namespace client
|
|
|
|
|
Terminate (); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
m_Buffer[bytes_transferred] = 0; |
|
|
|
|
char * eol = (char *)memchr (m_Buffer, '\n', bytes_transferred); |
|
|
|
|
if (eol) |
|
|
|
@ -94,8 +94,8 @@ namespace client
@@ -94,8 +94,8 @@ namespace client
|
|
|
|
|
char * separator = strchr (m_Buffer, ' '); |
|
|
|
|
if (separator) |
|
|
|
|
{ |
|
|
|
|
separator = strchr (separator + 1, ' '); |
|
|
|
|
if (separator) |
|
|
|
|
separator = strchr (separator + 1, ' '); |
|
|
|
|
if (separator) |
|
|
|
|
*separator = 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -117,13 +117,13 @@ namespace client
@@ -117,13 +117,13 @@ namespace client
|
|
|
|
|
{ |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t l = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_HANDSHAKE_REPLY, version.c_str ()); |
|
|
|
|
#else |
|
|
|
|
#else |
|
|
|
|
size_t l = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_HANDSHAKE_REPLY, version.c_str ()); |
|
|
|
|
#endif |
|
|
|
|
boost::asio::async_write (m_Socket, boost::asio::buffer (m_Buffer, l), boost::asio::transfer_all (), |
|
|
|
|
std::bind(&SAMSocket::HandleHandshakeReplySent, shared_from_this (), |
|
|
|
|
std::bind(&SAMSocket::HandleHandshakeReplySent, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
SendMessageReply (SAM_HANDSHAKE_I2P_ERROR, strlen (SAM_HANDSHAKE_I2P_ERROR), true); |
|
|
|
|
} |
|
|
|
@ -145,25 +145,25 @@ namespace client
@@ -145,25 +145,25 @@ namespace client
|
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
m_Socket.async_read_some (boost::asio::buffer(m_Buffer, SAM_SOCKET_BUFFER_SIZE), |
|
|
|
|
std::bind(&SAMSocket::HandleMessage, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
} |
|
|
|
|
m_Socket.async_read_some (boost::asio::buffer(m_Buffer, SAM_SOCKET_BUFFER_SIZE), |
|
|
|
|
std::bind(&SAMSocket::HandleMessage, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::SendMessageReply (const char * msg, size_t len, bool close) |
|
|
|
|
{ |
|
|
|
|
if (!m_IsSilent) |
|
|
|
|
if (!m_IsSilent) |
|
|
|
|
boost::asio::async_write (m_Socket, boost::asio::buffer (msg, len), boost::asio::transfer_all (), |
|
|
|
|
std::bind(&SAMSocket::HandleMessageReplySent, shared_from_this (), |
|
|
|
|
std::bind(&SAMSocket::HandleMessageReplySent, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2, close)); |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
if (close) |
|
|
|
|
Terminate (); |
|
|
|
|
else |
|
|
|
|
Receive (); |
|
|
|
|
} |
|
|
|
|
Receive (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::HandleMessageReplySent (const boost::system::error_code& ecode, std::size_t bytes_transferred, bool close) |
|
|
|
@ -179,8 +179,8 @@ namespace client
@@ -179,8 +179,8 @@ namespace client
|
|
|
|
|
if (close) |
|
|
|
|
Terminate (); |
|
|
|
|
else |
|
|
|
|
Receive (); |
|
|
|
|
} |
|
|
|
|
Receive (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::HandleMessage (const boost::system::error_code& ecode, std::size_t bytes_transferred) |
|
|
|
@ -205,8 +205,8 @@ namespace client
@@ -205,8 +205,8 @@ namespace client
|
|
|
|
|
char * separator = strchr (m_Buffer, ' '); |
|
|
|
|
if (separator) |
|
|
|
|
{ |
|
|
|
|
separator = strchr (separator + 1, ' '); |
|
|
|
|
if (separator) |
|
|
|
|
separator = strchr (separator + 1, ' '); |
|
|
|
|
if (separator) |
|
|
|
|
*separator = 0; |
|
|
|
|
else |
|
|
|
|
separator = eol; |
|
|
|
@ -236,12 +236,12 @@ namespace client
@@ -236,12 +236,12 @@ namespace client
|
|
|
|
|
*separator = ' '; |
|
|
|
|
*eol = '\n'; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
// since it's SAM v1 reply is not expected
|
|
|
|
|
Receive (); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogError, "SAM: unexpected message ", m_Buffer); |
|
|
|
|
Terminate (); |
|
|
|
|
} |
|
|
|
@ -254,7 +254,7 @@ namespace client
@@ -254,7 +254,7 @@ namespace client
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogWarning, "SAM: incomplete message ", bytes_transferred); |
|
|
|
|
m_BufferOffset = bytes_transferred; |
|
|
|
|
// try to receive remaining message
|
|
|
|
@ -268,10 +268,10 @@ namespace client
@@ -268,10 +268,10 @@ namespace client
|
|
|
|
|
LogPrint (eLogDebug, "SAM: session create: ", buf); |
|
|
|
|
std::map<std::string, std::string> params; |
|
|
|
|
ExtractParams (buf, params); |
|
|
|
|
std::string& style = params[SAM_PARAM_STYLE]; |
|
|
|
|
std::string& style = params[SAM_PARAM_STYLE]; |
|
|
|
|
std::string& id = params[SAM_PARAM_ID]; |
|
|
|
|
std::string& destination = params[SAM_PARAM_DESTINATION]; |
|
|
|
|
m_ID = id; |
|
|
|
|
m_ID = id; |
|
|
|
|
if (m_Owner.FindSession (id)) |
|
|
|
|
{ |
|
|
|
|
// session exists
|
|
|
|
@ -302,7 +302,7 @@ namespace client
@@ -302,7 +302,7 @@ namespace client
|
|
|
|
|
forward = std::make_shared<boost::asio::ip::udp::endpoint>(addr, port); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// create destination
|
|
|
|
|
// create destination
|
|
|
|
|
m_Session = m_Owner.CreateSession (id, destination == SAM_VALUE_TRANSIENT ? "" : destination, ¶ms); |
|
|
|
|
if (m_Session) |
|
|
|
|
{ |
|
|
|
@ -311,7 +311,7 @@ namespace client
@@ -311,7 +311,7 @@ namespace client
|
|
|
|
|
{ |
|
|
|
|
m_Session->UDPEndpoint = forward; |
|
|
|
|
auto dest = m_Session->localDestination->CreateDatagramDestination (); |
|
|
|
|
dest->SetReceiver (std::bind (&SAMSocket::HandleI2PDatagramReceive, shared_from_this (), |
|
|
|
|
dest->SetReceiver (std::bind (&SAMSocket::HandleI2PDatagramReceive, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -321,7 +321,7 @@ namespace client
@@ -321,7 +321,7 @@ namespace client
|
|
|
|
|
{ |
|
|
|
|
m_Timer.expires_from_now (boost::posix_time::seconds(SAM_SESSION_READINESS_CHECK_INTERVAL)); |
|
|
|
|
m_Timer.async_wait (std::bind (&SAMSocket::HandleSessionReadinessCheckTimer, |
|
|
|
|
shared_from_this (), std::placeholders::_1)); |
|
|
|
|
shared_from_this (), std::placeholders::_1)); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
@ -339,7 +339,7 @@ namespace client
@@ -339,7 +339,7 @@ namespace client
|
|
|
|
|
m_Timer.expires_from_now (boost::posix_time::seconds(SAM_SESSION_READINESS_CHECK_INTERVAL)); |
|
|
|
|
m_Timer.async_wait (std::bind (&SAMSocket::HandleSessionReadinessCheckTimer, |
|
|
|
|
shared_from_this (), std::placeholders::_1)); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -352,7 +352,7 @@ namespace client
@@ -352,7 +352,7 @@ namespace client
|
|
|
|
|
priv[l1] = 0; |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t l2 = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_SESSION_CREATE_REPLY_OK, priv); |
|
|
|
|
#else |
|
|
|
|
#else |
|
|
|
|
size_t l2 = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_SESSION_CREATE_REPLY_OK, priv); |
|
|
|
|
#endif |
|
|
|
|
SendMessageReply (m_Buffer, l2, false); |
|
|
|
@ -366,7 +366,7 @@ namespace client
@@ -366,7 +366,7 @@ namespace client
|
|
|
|
|
std::string& id = params[SAM_PARAM_ID]; |
|
|
|
|
std::string& destination = params[SAM_PARAM_DESTINATION]; |
|
|
|
|
std::string& silent = params[SAM_PARAM_SILENT]; |
|
|
|
|
if (silent == SAM_VALUE_TRUE) m_IsSilent = true; |
|
|
|
|
if (silent == SAM_VALUE_TRUE) m_IsSilent = true; |
|
|
|
|
m_ID = id; |
|
|
|
|
m_Session = m_Owner.FindSession (id); |
|
|
|
|
if (m_Session) |
|
|
|
@ -390,7 +390,7 @@ namespace client
@@ -390,7 +390,7 @@ namespace client
|
|
|
|
|
SendMessageReply(SAM_SESSION_STATUS_INVALID_KEY, strlen(SAM_SESSION_STATUS_INVALID_KEY), true); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
SendMessageReply (SAM_STREAM_STATUS_INVALID_ID, strlen(SAM_STREAM_STATUS_INVALID_ID), true); |
|
|
|
|
SendMessageReply (SAM_STREAM_STATUS_INVALID_ID, strlen(SAM_STREAM_STATUS_INVALID_ID), true); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::Connect (std::shared_ptr<const i2p::data::LeaseSet> remote) |
|
|
|
@ -399,7 +399,7 @@ namespace client
@@ -399,7 +399,7 @@ namespace client
|
|
|
|
|
m_Session->AddSocket (shared_from_this ()); |
|
|
|
|
m_Stream = m_Session->localDestination->CreateStream (remote); |
|
|
|
|
m_Stream->Send ((uint8_t *)m_Buffer, 0); // connect
|
|
|
|
|
I2PReceive (); |
|
|
|
|
I2PReceive (); |
|
|
|
|
SendMessageReply (SAM_STREAM_STATUS_OK, strlen(SAM_STREAM_STATUS_OK), false); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -421,17 +421,17 @@ namespace client
@@ -421,17 +421,17 @@ namespace client
|
|
|
|
|
ExtractParams (buf, params); |
|
|
|
|
std::string& id = params[SAM_PARAM_ID]; |
|
|
|
|
std::string& silent = params[SAM_PARAM_SILENT]; |
|
|
|
|
if (silent == SAM_VALUE_TRUE) m_IsSilent = true; |
|
|
|
|
if (silent == SAM_VALUE_TRUE) m_IsSilent = true; |
|
|
|
|
m_ID = id; |
|
|
|
|
m_Session = m_Owner.FindSession (id); |
|
|
|
|
if (m_Session) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
m_SocketType = eSAMSocketTypeAcceptor; |
|
|
|
|
m_Session->AddSocket (shared_from_this ()); |
|
|
|
|
if (!m_Session->localDestination->IsAcceptingStreams ()) |
|
|
|
|
m_Session->localDestination->AcceptOnce (std::bind (&SAMSocket::HandleI2PAccept, shared_from_this (), std::placeholders::_1)); |
|
|
|
|
SendMessageReply (SAM_STREAM_STATUS_OK, strlen(SAM_STREAM_STATUS_OK), false); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
SendMessageReply (SAM_STREAM_STATUS_INVALID_ID, strlen(SAM_STREAM_STATUS_INVALID_ID), true); |
|
|
|
|
} |
|
|
|
@ -443,9 +443,9 @@ namespace client
@@ -443,9 +443,9 @@ namespace client
|
|
|
|
|
ExtractParams (buf, params); |
|
|
|
|
size_t size = std::stoi(params[SAM_PARAM_SIZE]), offset = data - buf; |
|
|
|
|
if (offset + size <= len) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
if (m_Session) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
auto d = m_Session->localDestination->GetDatagramDestination (); |
|
|
|
|
if (d) |
|
|
|
|
{ |
|
|
|
@ -458,24 +458,24 @@ namespace client
@@ -458,24 +458,24 @@ namespace client
|
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
LogPrint (eLogError, "SAM: session is not created from DATAGRAM SEND"); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogWarning, "SAM: sent datagram size ", size, " exceeds buffer ", len - offset); |
|
|
|
|
return 0; // try to receive more
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return offset + size; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::ProcessDestGenerate () |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogDebug, "SAM: dest generate"); |
|
|
|
|
auto keys = i2p::data::PrivateKeys::CreateRandomKeys (); |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t len = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_DEST_REPLY, |
|
|
|
|
size_t len = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_DEST_REPLY, |
|
|
|
|
keys.GetPublic ()->ToBase64 ().c_str (), keys.ToBase64 ().c_str ()); |
|
|
|
|
#else |
|
|
|
|
size_t len = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_DEST_REPLY, |
|
|
|
|
#else |
|
|
|
|
size_t len = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_DEST_REPLY, |
|
|
|
|
keys.GetPublic ()->ToBase64 ().c_str (), keys.ToBase64 ().c_str ()); |
|
|
|
|
#endif |
|
|
|
|
SendMessageReply (m_Buffer, len, false); |
|
|
|
@ -504,12 +504,12 @@ namespace client
@@ -504,12 +504,12 @@ namespace client
|
|
|
|
|
std::bind (&SAMSocket::HandleNamingLookupLeaseSetRequestComplete, |
|
|
|
|
shared_from_this (), std::placeholders::_1, ident)); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogError, "SAM: naming failed, unknown address ", name); |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t len = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY_INVALID_KEY, name.c_str()); |
|
|
|
|
#else |
|
|
|
|
#else |
|
|
|
|
size_t len = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY_INVALID_KEY, name.c_str()); |
|
|
|
|
#endif |
|
|
|
|
SendMessageReply (m_Buffer, len, false); |
|
|
|
@ -521,7 +521,7 @@ namespace client
@@ -521,7 +521,7 @@ namespace client
|
|
|
|
|
LogPrint (eLogError, "SAM: i2p error ", msg); |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t len = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_SESSION_STATUS_I2P_ERROR, msg.c_str()); |
|
|
|
|
#else |
|
|
|
|
#else |
|
|
|
|
size_t len = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_SESSION_STATUS_I2P_ERROR, msg.c_str()); |
|
|
|
|
#endif |
|
|
|
|
SendMessageReply (m_Buffer, len, true); |
|
|
|
@ -530,30 +530,30 @@ namespace client
@@ -530,30 +530,30 @@ namespace client
|
|
|
|
|
void SAMSocket::HandleNamingLookupLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet, i2p::data::IdentHash ident) |
|
|
|
|
{ |
|
|
|
|
if (leaseSet) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
context.GetAddressBook ().InsertAddress (leaseSet->GetIdentity ()); |
|
|
|
|
SendNamingLookupReply (leaseSet->GetIdentity ()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogError, "SAM: naming lookup failed. LeaseSet for ", ident.ToBase32 (), " not found"); |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t len = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY_INVALID_KEY, |
|
|
|
|
size_t len = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY_INVALID_KEY, |
|
|
|
|
context.GetAddressBook ().ToAddress (ident).c_str()); |
|
|
|
|
#else |
|
|
|
|
#else |
|
|
|
|
size_t len = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY_INVALID_KEY, |
|
|
|
|
context.GetAddressBook ().ToAddress (ident).c_str()); |
|
|
|
|
#endif |
|
|
|
|
SendMessageReply (m_Buffer, len, false); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::SendNamingLookupReply (std::shared_ptr<const i2p::data::IdentityEx> identity) |
|
|
|
|
{ |
|
|
|
|
auto base64 = identity->ToBase64 (); |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t l = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY, base64.c_str ()); |
|
|
|
|
#else |
|
|
|
|
size_t l = sprintf_s (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY, base64.c_str ()); |
|
|
|
|
#else |
|
|
|
|
size_t l = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_NAMING_REPLY, base64.c_str ()); |
|
|
|
|
#endif |
|
|
|
|
SendMessageReply (m_Buffer, l, false); |
|
|
|
@ -561,7 +561,7 @@ namespace client
@@ -561,7 +561,7 @@ namespace client
|
|
|
|
|
|
|
|
|
|
void SAMSocket::ExtractParams (char * buf, std::map<std::string, std::string>& params) |
|
|
|
|
{ |
|
|
|
|
char * separator; |
|
|
|
|
char * separator; |
|
|
|
|
do |
|
|
|
|
{ |
|
|
|
|
separator = strchr (buf, ' '); |
|
|
|
@ -572,11 +572,11 @@ namespace client
@@ -572,11 +572,11 @@ namespace client
|
|
|
|
|
*value = 0; |
|
|
|
|
value++; |
|
|
|
|
params[buf] = value; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
buf = separator + 1; |
|
|
|
|
} |
|
|
|
|
while (separator); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::Receive () |
|
|
|
|
{ |
|
|
|
@ -586,7 +586,7 @@ namespace client
@@ -586,7 +586,7 @@ namespace client
|
|
|
|
|
Terminate (); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
m_Socket.async_read_some (boost::asio::buffer(m_Buffer + m_BufferOffset, SAM_SOCKET_BUFFER_SIZE - m_BufferOffset), |
|
|
|
|
m_Socket.async_read_some (boost::asio::buffer(m_Buffer + m_BufferOffset, SAM_SOCKET_BUFFER_SIZE - m_BufferOffset), |
|
|
|
|
std::bind((m_SocketType == eSAMSocketTypeStream) ? &SAMSocket::HandleReceived : &SAMSocket::HandleMessage, |
|
|
|
|
shared_from_this (), std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
} |
|
|
|
@ -602,17 +602,17 @@ namespace client
@@ -602,17 +602,17 @@ namespace client
|
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
if (m_Stream) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
auto s = shared_from_this (); |
|
|
|
|
m_Stream->AsyncSend ((uint8_t *)m_Buffer, bytes_transferred, |
|
|
|
|
[s](const boost::system::error_code& ecode) |
|
|
|
|
{ |
|
|
|
|
if (!ecode) |
|
|
|
|
s->Receive (); |
|
|
|
|
else |
|
|
|
|
else |
|
|
|
|
s->m_Owner.GetService ().post ([s] { s->Terminate (); }); |
|
|
|
|
}); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -622,7 +622,7 @@ namespace client
@@ -622,7 +622,7 @@ namespace client
|
|
|
|
|
{ |
|
|
|
|
if (m_Stream->GetStatus () == i2p::stream::eStreamStatusNew || |
|
|
|
|
m_Stream->GetStatus () == i2p::stream::eStreamStatusOpen) // regular
|
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
m_Stream->AsyncReceive (boost::asio::buffer (m_StreamBuffer, SAM_SOCKET_BUFFER_SIZE), |
|
|
|
|
std::bind (&SAMSocket::HandleI2PReceive, shared_from_this (), |
|
|
|
|
std::placeholders::_1, std::placeholders::_2), |
|
|
|
@ -638,10 +638,10 @@ namespace client
@@ -638,10 +638,10 @@ namespace client
|
|
|
|
|
std::bind (&SAMSocket::HandleWriteI2PData, shared_from_this (), std::placeholders::_1)); |
|
|
|
|
} |
|
|
|
|
else // no more data
|
|
|
|
|
Terminate (); |
|
|
|
|
} |
|
|
|
|
Terminate (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::HandleI2PReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred) |
|
|
|
|
{ |
|
|
|
@ -653,17 +653,17 @@ namespace client
@@ -653,17 +653,17 @@ namespace client
|
|
|
|
|
if (bytes_transferred > 0) |
|
|
|
|
boost::asio::async_write (m_Socket, boost::asio::buffer (m_StreamBuffer, bytes_transferred), |
|
|
|
|
std::bind (&SAMSocket::HandleWriteI2PData, shared_from_this (), std::placeholders::_1)); // postpone termination
|
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
auto s = shared_from_this (); |
|
|
|
|
m_Owner.GetService ().post ([s] { s->Terminate (); }); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
auto s = shared_from_this (); |
|
|
|
|
m_Owner.GetService ().post ([s] { s->Terminate (); }); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
@ -694,7 +694,7 @@ namespace client
@@ -694,7 +694,7 @@ namespace client
|
|
|
|
|
context.GetAddressBook ().InsertAddress (stream->GetRemoteIdentity ()); |
|
|
|
|
auto session = m_Owner.FindSession (m_ID); |
|
|
|
|
if (session) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
// find more pending acceptors
|
|
|
|
|
for (auto it: session->ListSockets ()) |
|
|
|
|
if (it->m_SocketType == eSAMSocketTypeAcceptor) |
|
|
|
@ -710,19 +710,19 @@ namespace client
@@ -710,19 +710,19 @@ namespace client
|
|
|
|
|
const size_t ident_len = ident_ptr->GetFullLen(); |
|
|
|
|
uint8_t* ident = new uint8_t[ident_len]; |
|
|
|
|
|
|
|
|
|
// send remote peer address as base64
|
|
|
|
|
// send remote peer address as base64
|
|
|
|
|
const size_t l = ident_ptr->ToBuffer (ident, ident_len); |
|
|
|
|
const size_t l1 = i2p::data::ByteStreamToBase64 (ident, l, (char *)m_StreamBuffer, SAM_SOCKET_BUFFER_SIZE); |
|
|
|
|
delete[] ident; |
|
|
|
|
m_StreamBuffer[l1] = '\n'; |
|
|
|
|
HandleI2PReceive (boost::system::error_code (), l1 +1); // we send identity like it has been received from stream
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
I2PReceive (); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
LogPrint (eLogWarning, "SAM: I2P acceptor has been reset"); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMSocket::HandleI2PDatagramReceive (const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len) |
|
|
|
|
{ |
|
|
|
@ -750,12 +750,12 @@ namespace client
@@ -750,12 +750,12 @@ namespace client
|
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
#ifdef _MSC_VER |
|
|
|
|
size_t l = sprintf_s ((char *)m_StreamBuffer, SAM_SOCKET_BUFFER_SIZE, SAM_DATAGRAM_RECEIVED, base64.c_str (), (long unsigned int)len); |
|
|
|
|
#else |
|
|
|
|
size_t l = snprintf ((char *)m_StreamBuffer, SAM_SOCKET_BUFFER_SIZE, SAM_DATAGRAM_RECEIVED, base64.c_str (), (long unsigned int)len); |
|
|
|
|
size_t l = sprintf_s ((char *)m_StreamBuffer, SAM_SOCKET_BUFFER_SIZE, SAM_DATAGRAM_RECEIVED, base64.c_str (), (long unsigned int)len); |
|
|
|
|
#else |
|
|
|
|
size_t l = snprintf ((char *)m_StreamBuffer, SAM_SOCKET_BUFFER_SIZE, SAM_DATAGRAM_RECEIVED, base64.c_str (), (long unsigned int)len); |
|
|
|
|
#endif |
|
|
|
|
if (len < SAM_SOCKET_BUFFER_SIZE - l) |
|
|
|
|
{ |
|
|
|
|
if (len < SAM_SOCKET_BUFFER_SIZE - l) |
|
|
|
|
{ |
|
|
|
|
memcpy (m_StreamBuffer + l, buf, len); |
|
|
|
|
boost::asio::async_write (m_Socket, boost::asio::buffer (m_StreamBuffer, len + l), |
|
|
|
|
std::bind (&SAMSocket::HandleWriteI2PData, shared_from_this (), std::placeholders::_1)); |
|
|
|
@ -770,7 +770,7 @@ namespace client
@@ -770,7 +770,7 @@ namespace client
|
|
|
|
|
UDPEndpoint(nullptr) |
|
|
|
|
{ |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
SAMSession::~SAMSession () |
|
|
|
|
{ |
|
|
|
|
CloseStreams(); |
|
|
|
@ -800,7 +800,7 @@ namespace client
@@ -800,7 +800,7 @@ namespace client
|
|
|
|
|
{ |
|
|
|
|
if (m_IsRunning) |
|
|
|
|
Stop (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMBridge::Start () |
|
|
|
|
{ |
|
|
|
@ -819,26 +819,26 @@ namespace client
@@ -819,26 +819,26 @@ namespace client
|
|
|
|
|
m_Sessions.clear (); |
|
|
|
|
m_Service.stop (); |
|
|
|
|
if (m_Thread) |
|
|
|
|
{ |
|
|
|
|
m_Thread->join (); |
|
|
|
|
{ |
|
|
|
|
m_Thread->join (); |
|
|
|
|
delete m_Thread; |
|
|
|
|
m_Thread = nullptr; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMBridge::Run () |
|
|
|
|
{ |
|
|
|
|
void SAMBridge::Run () |
|
|
|
|
{ |
|
|
|
|
while (m_IsRunning) |
|
|
|
|
{ |
|
|
|
|
try |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
m_Service.run (); |
|
|
|
|
} |
|
|
|
|
catch (std::exception& ex) |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogError, "SAM: runtime exception: ", ex.what ()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMBridge::Accept () |
|
|
|
@ -855,7 +855,7 @@ namespace client
@@ -855,7 +855,7 @@ namespace client
|
|
|
|
|
boost::system::error_code ec; |
|
|
|
|
auto ep = socket->GetSocket ().remote_endpoint (ec); |
|
|
|
|
if (!ec) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
LogPrint (eLogDebug, "SAM: new connection from ", ep); |
|
|
|
|
socket->ReceiveHandshake (); |
|
|
|
|
} |
|
|
|
@ -869,10 +869,10 @@ namespace client
@@ -869,10 +869,10 @@ namespace client
|
|
|
|
|
Accept (); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
std::shared_ptr<SAMSession> SAMBridge::CreateSession (const std::string& id, const std::string& destination, |
|
|
|
|
std::shared_ptr<SAMSession> SAMBridge::CreateSession (const std::string& id, const std::string& destination, |
|
|
|
|
const std::map<std::string, std::string> * params) |
|
|
|
|
{ |
|
|
|
|
std::shared_ptr<ClientDestination> localDestination = nullptr; |
|
|
|
|
std::shared_ptr<ClientDestination> localDestination = nullptr; |
|
|
|
|
if (destination != "") |
|
|
|
|
{ |
|
|
|
|
i2p::data::PrivateKeys keys; |
|
|
|
@ -887,10 +887,10 @@ namespace client
@@ -887,10 +887,10 @@ namespace client
|
|
|
|
|
{ |
|
|
|
|
auto it = params->find (SAM_PARAM_SIGNATURE_TYPE); |
|
|
|
|
if (it != params->end ()) |
|
|
|
|
// TODO: extract string values
|
|
|
|
|
// TODO: extract string values
|
|
|
|
|
signatureType = std::stoi(it->second); |
|
|
|
|
} |
|
|
|
|
localDestination = i2p::client::context.CreateNewLocalDestination (true, signatureType, params); |
|
|
|
|
localDestination = i2p::client::context.CreateNewLocalDestination (true, signatureType, params); |
|
|
|
|
} |
|
|
|
|
if (localDestination) |
|
|
|
|
{ |
|
|
|
@ -911,13 +911,13 @@ namespace client
@@ -911,13 +911,13 @@ namespace client
|
|
|
|
|
std::unique_lock<std::mutex> l(m_SessionsMutex); |
|
|
|
|
auto it = m_Sessions.find (id); |
|
|
|
|
if (it != m_Sessions.end ()) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
session = it->second; |
|
|
|
|
m_Sessions.erase (it); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if (session) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
session->localDestination->StopAcceptingStreams (); |
|
|
|
|
session->CloseStreams (); |
|
|
|
|
} |
|
|
|
@ -943,9 +943,9 @@ namespace client
@@ -943,9 +943,9 @@ namespace client
|
|
|
|
|
void SAMBridge::ReceiveDatagram () |
|
|
|
|
{ |
|
|
|
|
m_DatagramSocket.async_receive_from ( |
|
|
|
|
boost::asio::buffer (m_DatagramReceiveBuffer, i2p::datagram::MAX_DATAGRAM_SIZE), |
|
|
|
|
boost::asio::buffer (m_DatagramReceiveBuffer, i2p::datagram::MAX_DATAGRAM_SIZE), |
|
|
|
|
m_SenderEndpoint, |
|
|
|
|
std::bind (&SAMBridge::HandleReceivedDatagram, this, std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
std::bind (&SAMBridge::HandleReceivedDatagram, this, std::placeholders::_1, std::placeholders::_2)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SAMBridge::HandleReceivedDatagram (const boost::system::error_code& ecode, std::size_t bytes_transferred) |
|
|
|
@ -955,7 +955,7 @@ namespace client
@@ -955,7 +955,7 @@ namespace client
|
|
|
|
|
m_DatagramReceiveBuffer[bytes_transferred] = 0; |
|
|
|
|
char * eol = strchr ((char *)m_DatagramReceiveBuffer, '\n'); |
|
|
|
|
*eol = 0; eol++; |
|
|
|
|
size_t payloadLen = bytes_transferred - ((uint8_t *)eol - m_DatagramReceiveBuffer); |
|
|
|
|
size_t payloadLen = bytes_transferred - ((uint8_t *)eol - m_DatagramReceiveBuffer); |
|
|
|
|
LogPrint (eLogDebug, "SAM: datagram received ", m_DatagramReceiveBuffer," size=", payloadLen); |
|
|
|
|
char * sessionID = strchr ((char *)m_DatagramReceiveBuffer, ' '); |
|
|
|
|
if (sessionID) |
|
|
|
@ -967,12 +967,12 @@ namespace client
@@ -967,12 +967,12 @@ namespace client
|
|
|
|
|
*destination = 0; destination++; |
|
|
|
|
auto session = FindSession (sessionID); |
|
|
|
|
if (session) |
|
|
|
|
{ |
|
|
|
|
{ |
|
|
|
|
i2p::data::IdentityEx dest; |
|
|
|
|
dest.FromBase64 (destination); |
|
|
|
|
session->localDestination->GetDatagramDestination ()-> |
|
|
|
|
SendDatagramTo ((uint8_t *)eol, payloadLen, dest.GetIdentHash ()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
LogPrint (eLogError, "SAM: Session ", sessionID, " not found"); |
|
|
|
|
} |
|
|
|
|