|
|
|
|
@ -121,7 +121,7 @@ namespace client
|
|
|
|
|
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
|
|
|
|
|
@ -252,6 +252,7 @@ namespace client
|
|
|
|
|
Terminate ();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
LogPrint (eLogWarning, "SAM: incomplete message ", bytes_transferred);
|
|
|
|
|
@ -278,6 +279,29 @@ namespace client
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::shared_ptr<boost::asio::ip::udp::endpoint> forward = nullptr;
|
|
|
|
|
if (style == SAM_VALUE_DATAGRAM && params.find(SAM_VALUE_HOST) != params.end() && params.find(SAM_VALUE_PORT) != params.end())
|
|
|
|
|
{
|
|
|
|
|
// udp forward selected
|
|
|
|
|
boost::system::error_code e;
|
|
|
|
|
// TODO: support hostnames in udp forward
|
|
|
|
|
auto addr = boost::asio::ip::address::from_string(params[SAM_VALUE_HOST], e);
|
|
|
|
|
if (e)
|
|
|
|
|
{
|
|
|
|
|
// not an ip address
|
|
|
|
|
SendI2PError("Invalid IP Address in HOST");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto port = std::stoi(params[SAM_VALUE_PORT]);
|
|
|
|
|
if (port == -1)
|
|
|
|
|
{
|
|
|
|
|
SendI2PError("Invalid port");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
forward = std::make_shared<boost::asio::ip::udp::endpoint>(addr, port);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// create destination
|
|
|
|
|
m_Session = m_Owner.CreateSession (id, destination == SAM_VALUE_TRANSIENT ? "" : destination, ¶ms);
|
|
|
|
|
if (m_Session)
|
|
|
|
|
@ -285,6 +309,7 @@ namespace client
|
|
|
|
|
m_SocketType = eSAMSocketTypeSession;
|
|
|
|
|
if (style == SAM_VALUE_DATAGRAM)
|
|
|
|
|
{
|
|
|
|
|
m_Session->UDPEndpoint = forward;
|
|
|
|
|
auto dest = m_Session->localDestination->CreateDatagramDestination ();
|
|
|
|
|
dest->SetReceiver (std::bind (&SAMSocket::HandleI2PDatagramReceive, shared_from_this (),
|
|
|
|
|
std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5));
|
|
|
|
|
@ -491,7 +516,18 @@ namespace client
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void SAMSocket::HandleNamingLookupLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet, i2p::data::IdentHash ident)
|
|
|
|
|
void SAMSocket::SendI2PError(const std::string & msg)
|
|
|
|
|
{
|
|
|
|
|
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
|
|
|
|
|
size_t len = snprintf (m_Buffer, SAM_SOCKET_BUFFER_SIZE, SAM_SESSION_STATUS_I2P_ERROR, msg.c_str());
|
|
|
|
|
#endif
|
|
|
|
|
SendMessageReply (m_Buffer, len, true);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void SAMSocket::HandleNamingLookupLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet, i2p::data::IdentHash ident)
|
|
|
|
|
{
|
|
|
|
|
if (leaseSet)
|
|
|
|
|
{
|
|
|
|
|
@ -692,23 +728,45 @@ namespace client
|
|
|
|
|
{
|
|
|
|
|
LogPrint (eLogDebug, "SAM: datagram received ", len);
|
|
|
|
|
auto base64 = from.ToBase64 ();
|
|
|
|
|
auto ep = m_Session->UDPEndpoint;
|
|
|
|
|
if (ep)
|
|
|
|
|
{
|
|
|
|
|
// udp forward enabled
|
|
|
|
|
size_t bsz = base64.size();
|
|
|
|
|
size_t sz = bsz + 1 + len;
|
|
|
|
|
// build datagram body
|
|
|
|
|
uint8_t * data = new uint8_t[sz];
|
|
|
|
|
// Destination
|
|
|
|
|
memcpy(data, base64.c_str(), bsz);
|
|
|
|
|
// linefeed
|
|
|
|
|
data[bsz] = '\n';
|
|
|
|
|
// Payload
|
|
|
|
|
memcpy(data+bsz+1, buf, len);
|
|
|
|
|
// send to remote endpoint
|
|
|
|
|
m_Owner.SendTo(data, sz, ep);
|
|
|
|
|
delete [] data;
|
|
|
|
|
}
|
|
|
|
|
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);
|
|
|
|
|
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 = 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)
|
|
|
|
|
{
|
|
|
|
|
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));
|
|
|
|
|
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));
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
LogPrint (eLogWarning, "SAM: received datagram size ", len," exceeds buffer");
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
LogPrint (eLogWarning, "SAM: received datagram size ", len," exceeds buffer");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
SAMSession::SAMSession (std::shared_ptr<ClientDestination> dest):
|
|
|
|
|
localDestination (dest)
|
|
|
|
|
localDestination (dest),
|
|
|
|
|
UDPEndpoint(nullptr)
|
|
|
|
|
{
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -873,6 +931,14 @@ namespace client
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void SAMBridge::SendTo(const uint8_t * buf, size_t len, std::shared_ptr<boost::asio::ip::udp::endpoint> remote)
|
|
|
|
|
{
|
|
|
|
|
if(remote)
|
|
|
|
|
{
|
|
|
|
|
m_DatagramSocket.send_to(boost::asio::buffer(buf, len), *remote);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void SAMBridge::ReceiveDatagram ()
|
|
|
|
|
{
|
|
|
|
|
m_DatagramSocket.async_receive_from (
|
|
|
|
|
|
orignal
8 years ago
committed by
GitHub
