i2pd/libi2pd_client/SAM.h

247 lines
9.7 KiB
C
Raw Normal View History

/*
* Copyright (c) 2013-2020, 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
*/
2014-09-24 20:01:26 +04:00
#ifndef SAM_H__
#define SAM_H__
2014-09-24 22:59:03 +04:00
#include <inttypes.h>
#include <string>
2014-09-25 00:39:31 +04:00
#include <map>
#include <list>
2014-09-24 20:01:26 +04:00
#include <thread>
2014-10-06 05:59:05 +04:00
#include <mutex>
2014-11-23 00:35:58 +03:00
#include <memory>
2014-09-24 20:01:26 +04:00
#include <boost/asio.hpp>
#include "util.h"
#include "Identity.h"
#include "LeaseSet.h"
2014-09-24 22:59:03 +04:00
#include "Streaming.h"
#include "Destination.h"
2014-09-24 20:01:26 +04:00
namespace i2p
{
namespace client
2014-09-24 20:01:26 +04:00
{
2015-03-27 16:44:27 +03:00
const size_t SAM_SOCKET_BUFFER_SIZE = 8192;
const int SAM_SOCKET_CONNECTION_MAX_IDLE = 3600; // in seconds
2017-01-31 19:16:55 +03:00
const int SAM_SESSION_READINESS_CHECK_INTERVAL = 20; // in seconds
2014-09-24 22:59:03 +04:00
const char SAM_HANDSHAKE[] = "HELLO VERSION";
2014-12-16 23:54:02 +03:00
const char SAM_HANDSHAKE_REPLY[] = "HELLO REPLY RESULT=OK VERSION=%s\n";
2018-02-25 16:47:39 +03:00
const char SAM_HANDSHAKE_NOVERSION[] = "HELLO REPLY RESULT=NOVERSION\n";
2017-01-31 19:16:55 +03:00
const char SAM_HANDSHAKE_I2P_ERROR[] = "HELLO REPLY RESULT=I2P_ERROR\n";
2014-09-25 21:22:25 +04:00
const char SAM_SESSION_CREATE[] = "SESSION CREATE";
2014-10-03 05:40:15 +04:00
const char SAM_SESSION_CREATE_REPLY_OK[] = "SESSION STATUS RESULT=OK DESTINATION=%s\n";
const char SAM_SESSION_CREATE_DUPLICATED_ID[] = "SESSION STATUS RESULT=DUPLICATED_ID\n";
2017-01-31 19:16:55 +03:00
const char SAM_SESSION_CREATE_DUPLICATED_DEST[] = "SESSION STATUS RESULT=DUPLICATED_DEST\n";
2018-02-16 14:28:22 +03:00
const char SAM_SESSION_CREATE_INVALID_ID[] = "SESSION STATUS RESULT=INVALID_ID\n";
const char SAM_SESSION_STATUS_INVALID_KEY[] = "SESSION STATUS RESULT=INVALID_KEY\n";
2017-01-31 19:16:55 +03:00
const char SAM_SESSION_STATUS_I2P_ERROR[] = "SESSION STATUS RESULT=I2P_ERROR MESSAGE=%s\n";
2014-09-25 21:58:09 +04:00
const char SAM_STREAM_CONNECT[] = "STREAM CONNECT";
const char SAM_STREAM_STATUS_OK[] = "STREAM STATUS RESULT=OK\n";
const char SAM_STREAM_STATUS_INVALID_ID[] = "STREAM STATUS RESULT=INVALID_ID\n";
2019-03-29 16:29:28 +03:00
const char SAM_STREAM_STATUS_INVALID_KEY[] = "STREAM STATUS RESULT=INVALID_KEY\n";
const char SAM_STREAM_STATUS_CANT_REACH_PEER[] = "STREAM STATUS RESULT=CANT_REACH_PEER\n";
const char SAM_STREAM_STATUS_I2P_ERROR[] = "STREAM STATUS RESULT=I2P_ERROR\n";
2017-01-31 19:16:55 +03:00
const char SAM_STREAM_ACCEPT[] = "STREAM ACCEPT";
const char SAM_STREAM_FORWARD[] = "STREAM FORWARD";
2015-03-27 04:23:59 +03:00
const char SAM_DATAGRAM_SEND[] = "DATAGRAM SEND";
2019-07-10 20:30:31 +03:00
const char SAM_RAW_SEND[] = "RAW SEND";
2014-09-30 19:08:38 +04:00
const char SAM_DEST_GENERATE[] = "DEST GENERATE";
2017-01-31 19:16:55 +03:00
const char SAM_DEST_REPLY[] = "DEST REPLY PUB=%s PRIV=%s\n";
2014-09-30 19:08:38 +04:00
const char SAM_DEST_REPLY_I2P_ERROR[] = "DEST REPLY RESULT=I2P_ERROR\n";
2014-10-03 00:55:01 +04:00
const char SAM_NAMING_LOOKUP[] = "NAMING LOOKUP";
2020-12-19 04:48:08 +03:00
const char SAM_NAMING_REPLY[] = "NAMING REPLY RESULT=OK NAME=%s VALUE=%s\n";
2016-02-01 06:37:38 +03:00
const char SAM_DATAGRAM_RECEIVED[] = "DATAGRAM RECEIVED DESTINATION=%s SIZE=%lu\n";
const char SAM_RAW_RECEIVED[] = "RAW RECEIVED SIZE=%lu\n";
2014-10-03 23:08:41 +04:00
const char SAM_NAMING_REPLY_INVALID_KEY[] = "NAMING REPLY RESULT=INVALID_KEY NAME=%s\n";
2018-09-02 22:39:23 +03:00
const char SAM_NAMING_REPLY_KEY_NOT_FOUND[] = "NAMING REPLY RESULT=KEY_NOT_FOUND NAME=%s\n";
2017-01-31 19:16:55 +03:00
const char SAM_PARAM_MIN[] = "MIN";
const char SAM_PARAM_MAX[] = "MAX";
const char SAM_PARAM_STYLE[] = "STYLE";
const char SAM_PARAM_ID[] = "ID";
2014-09-29 22:18:06 +04:00
const char SAM_PARAM_SILENT[] = "SILENT";
2017-01-31 19:16:55 +03:00
const char SAM_PARAM_DESTINATION[] = "DESTINATION";
2014-12-17 00:23:42 +03:00
const char SAM_PARAM_NAME[] = "NAME";
2017-01-31 19:16:55 +03:00
const char SAM_PARAM_SIGNATURE_TYPE[] = "SIGNATURE_TYPE";
2017-11-14 23:05:07 +03:00
const char SAM_PARAM_CRYPTO_TYPE[] = "CRYPTO_TYPE";
2015-03-27 04:23:59 +03:00
const char SAM_PARAM_SIZE[] = "SIZE";
const char SAM_PARAM_HOST[] = "HOST";
const char SAM_PARAM_PORT[] = "PORT";
2017-01-31 19:16:55 +03:00
const char SAM_VALUE_TRANSIENT[] = "TRANSIENT";
const char SAM_VALUE_STREAM[] = "STREAM";
const char SAM_VALUE_DATAGRAM[] = "DATAGRAM";
2017-01-31 19:16:55 +03:00
const char SAM_VALUE_RAW[] = "RAW";
const char SAM_VALUE_TRUE[] = "true";
2017-01-30 04:38:18 +03:00
const char SAM_VALUE_FALSE[] = "false";
2014-09-25 21:22:25 +04:00
enum SAMSocketType
{
eSAMSocketTypeUnknown,
eSAMSocketTypeSession,
2014-09-26 23:40:57 +04:00
eSAMSocketTypeStream,
2014-11-23 05:56:59 +03:00
eSAMSocketTypeAcceptor,
eSAMSocketTypeForward,
2014-11-23 05:56:59 +03:00
eSAMSocketTypeTerminated
2014-09-25 21:22:25 +04:00
};
2014-09-24 22:59:03 +04:00
class SAMBridge;
struct SAMSession;
2018-04-25 18:27:56 +03:00
class SAMSocket: public std::enable_shared_from_this<SAMSocket>
2014-09-24 22:59:03 +04:00
{
public:
2018-01-15 16:19:57 +03:00
typedef boost::asio::ip::tcp::socket Socket_t;
2018-04-24 16:45:16 +03:00
SAMSocket (SAMBridge& owner);
~SAMSocket ();
2014-12-04 19:24:00 +03:00
2018-04-24 16:45:16 +03:00
Socket_t& GetSocket () { return m_Socket; };
2014-09-24 22:59:03 +04:00
void ReceiveHandshake ();
2014-12-04 19:24:00 +03:00
void SetSocketType (SAMSocketType socketType) { m_SocketType = socketType; };
SAMSocketType GetSocketType () const { return m_SocketType; };
2014-09-24 22:59:03 +04:00
void Terminate (const char* reason);
2017-06-02 18:43:33 +03:00
2018-04-25 18:25:49 +03:00
bool IsSession(const std::string & id) const;
private:
2018-04-25 18:25:49 +03:00
void TerminateClose() { Terminate(nullptr); }
2018-04-25 18:25:49 +03:00
void HandleHandshakeReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-09-24 22:59:03 +04:00
void HandleHandshakeReplySent (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-09-25 21:22:25 +04:00
void HandleMessage (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2017-01-31 19:16:55 +03:00
void SendMessageReply (const char * msg, size_t len, bool close);
2014-09-25 21:22:25 +04:00
void HandleMessageReplySent (const boost::system::error_code& ecode, std::size_t bytes_transferred, bool close);
2014-09-24 22:59:03 +04:00
void Receive ();
void HandleReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2017-01-31 19:16:55 +03:00
void I2PReceive ();
2014-09-26 23:40:57 +04:00
void HandleI2PReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-11-23 19:33:58 +03:00
void HandleI2PAccept (std::shared_ptr<i2p::stream::Stream> stream);
void HandleI2PForward (std::shared_ptr<i2p::stream::Stream> stream, boost::asio::ip::tcp::endpoint ep);
void HandleWriteI2PData (const boost::system::error_code& ecode, size_t sz);
void HandleI2PDatagramReceive (const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
void HandleI2PRawDatagramReceive (uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
2014-09-24 22:59:03 +04:00
2014-09-25 21:22:25 +04:00
void ProcessSessionCreate (char * buf, size_t len);
2017-03-29 20:59:48 +03:00
void ProcessStreamConnect (char * buf, size_t len, size_t rem);
2014-09-26 23:40:57 +04:00
void ProcessStreamAccept (char * buf, size_t len);
void ProcessStreamForward (char * buf, size_t len);
void ProcessDestGenerate (char * buf, size_t len);
2014-10-03 00:55:01 +04:00
void ProcessNamingLookup (char * buf, size_t len);
2017-01-31 19:16:55 +03:00
void SendI2PError(const std::string & msg);
size_t ProcessDatagramSend (char * buf, size_t len, const char * data); // from SAM 1.0
2015-03-27 04:23:59 +03:00
void ExtractParams (char * buf, std::map<std::string, std::string>& params);
2014-09-25 21:22:25 +04:00
void Connect (std::shared_ptr<const i2p::data::LeaseSet> remote, std::shared_ptr<SAMSession> session = nullptr);
void HandleConnectLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet);
2020-12-19 04:48:08 +03:00
void SendNamingLookupReply (const std::string& name, std::shared_ptr<const i2p::data::IdentityEx> identity);
2019-03-29 18:59:59 +03:00
void HandleNamingLookupLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet, std::string name);
void HandleSessionReadinessCheckTimer (const boost::system::error_code& ecode);
void SendSessionCreateReplyOk ();
void WriteI2PData(size_t sz);
void WriteI2PDataImmediate(uint8_t * ptr, size_t sz);
2018-01-15 16:19:57 +03:00
void HandleWriteI2PDataImmediate(const boost::system::error_code & ec, uint8_t * buff);
2018-04-25 18:25:49 +03:00
void HandleStreamSend(const boost::system::error_code & ec);
2014-09-24 22:59:03 +04:00
private:
SAMBridge& m_Owner;
2018-04-24 16:45:16 +03:00
Socket_t m_Socket;
boost::asio::deadline_timer m_Timer;
2014-09-24 22:59:03 +04:00
char m_Buffer[SAM_SOCKET_BUFFER_SIZE + 1];
2015-03-27 21:02:27 +03:00
size_t m_BufferOffset;
2014-09-24 22:59:03 +04:00
uint8_t m_StreamBuffer[SAM_SOCKET_BUFFER_SIZE];
2014-09-25 21:22:25 +04:00
SAMSocketType m_SocketType;
std::string m_ID; // nickname
2014-09-29 22:18:06 +04:00
bool m_IsSilent;
2018-01-06 06:48:51 +03:00
bool m_IsAccepting; // for eSAMSocketTypeAcceptor only
2014-11-23 19:33:58 +03:00
std::shared_ptr<i2p::stream::Stream> m_Stream;
2017-01-31 19:16:55 +03:00
};
2014-09-24 22:59:03 +04:00
enum SAMSessionType
{
eSAMSessionTypeUnknown,
eSAMSessionTypeStream,
eSAMSessionTypeDatagram,
eSAMSessionTypeRaw
};
2014-09-25 00:39:31 +04:00
struct SAMSession
{
2018-04-25 18:25:49 +03:00
SAMBridge & m_Bridge;
2015-02-24 23:40:50 +03:00
std::shared_ptr<ClientDestination> localDestination;
2017-01-31 19:16:55 +03:00
std::shared_ptr<boost::asio::ip::udp::endpoint> UDPEndpoint;
2018-04-25 18:25:49 +03:00
std::string Name;
SAMSessionType Type;
2018-04-24 16:45:16 +03:00
SAMSession (SAMBridge & parent, const std::string & name, SAMSessionType type, std::shared_ptr<ClientDestination> dest);
2016-12-23 03:38:17 +03:00
~SAMSession ();
void CloseStreams ();
2014-09-25 00:39:31 +04:00
};
class SAMBridge: private i2p::util::RunnableService
2014-09-24 20:01:26 +04:00
{
public:
2020-02-04 23:31:04 +03:00
SAMBridge (const std::string& address, int port, bool singleThread);
2014-09-24 20:01:26 +04:00
~SAMBridge ();
void Start ();
void Stop ();
2017-01-31 19:16:55 +03:00
boost::asio::io_service& GetService () { return GetIOService (); };
std::shared_ptr<SAMSession> CreateSession (const std::string& id, SAMSessionType type, const std::string& destination, // empty string means transient
2014-12-17 00:23:42 +03:00
const std::map<std::string, std::string> * params);
2014-09-25 00:39:31 +04:00
void CloseSession (const std::string& id);
2016-04-03 05:16:49 +03:00
std::shared_ptr<SAMSession> FindSession (const std::string& id) const;
2014-09-24 20:01:26 +04:00
2018-04-25 18:25:49 +03:00
std::list<std::shared_ptr<SAMSocket> > ListSockets(const std::string & id) const;
2018-04-24 16:45:16 +03:00
2017-01-31 19:16:55 +03:00
/** send raw data to remote endpoint from our UDP Socket */
void SendTo(const uint8_t * buf, size_t len, std::shared_ptr<boost::asio::ip::udp::endpoint> remote);
2017-01-30 04:38:18 +03:00
void AddSocket(std::shared_ptr<SAMSocket> socket);
2018-04-25 18:25:49 +03:00
void RemoveSocket(const std::shared_ptr<SAMSocket> & socket);
bool ResolveSignatureType (const std::string& name, i2p::data::SigningKeyType& type) const;
2019-01-23 18:52:17 +03:00
2014-09-24 20:01:26 +04:00
private:
void Accept ();
2018-04-25 18:25:49 +03:00
void HandleAccept(const boost::system::error_code& ecode, std::shared_ptr<SAMSocket> socket);
2014-09-24 20:01:26 +04:00
void ReceiveDatagram ();
void HandleReceivedDatagram (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-09-24 20:01:26 +04:00
private:
2020-02-04 23:31:04 +03:00
bool m_IsSingleThread;
2014-09-24 20:01:26 +04:00
boost::asio::ip::tcp::acceptor m_Acceptor;
boost::asio::ip::udp::endpoint m_DatagramEndpoint, m_SenderEndpoint;
boost::asio::ip::udp::socket m_DatagramSocket;
mutable std::mutex m_SessionsMutex;
2016-04-03 05:16:49 +03:00
std::map<std::string, std::shared_ptr<SAMSession> > m_Sessions;
2018-04-25 18:25:49 +03:00
mutable std::mutex m_OpenSocketsMutex;
std::list<std::shared_ptr<SAMSocket> > m_OpenSockets;
2014-10-24 05:14:17 +04:00
uint8_t m_DatagramReceiveBuffer[i2p::datagram::MAX_DATAGRAM_SIZE+1];
2019-01-23 18:52:17 +03:00
std::map<std::string, i2p::data::SigningKeyType> m_SignatureTypes;
public:
// for HTTP
const decltype(m_Sessions)& GetSessions () const { return m_Sessions; };
2017-01-31 19:16:55 +03:00
};
2014-09-24 20:01:26 +04:00
}
}
#endif