mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-04-30 12:47:48 +02:00
131 lines
4.2 KiB
C++
131 lines
4.2 KiB
C++
/*
|
|
* Copyright (c) 2013-2022, 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 TRANSPORT_SESSION_H__
|
|
#define TRANSPORT_SESSION_H__
|
|
|
|
#include <inttypes.h>
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <mutex>
|
|
#include "Identity.h"
|
|
#include "Crypto.h"
|
|
#include "RouterInfo.h"
|
|
#include "I2NPProtocol.h"
|
|
#include "Timestamp.h"
|
|
|
|
namespace i2p {
|
|
namespace transport {
|
|
const size_t IPV4_HEADER_SIZE = 20;
|
|
const size_t IPV6_HEADER_SIZE = 40;
|
|
const size_t UDP_HEADER_SIZE = 8;
|
|
|
|
class SignedData {
|
|
public:
|
|
|
|
SignedData() {}
|
|
|
|
SignedData(const SignedData &other) {
|
|
m_Stream << other.m_Stream.rdbuf();
|
|
}
|
|
|
|
void Reset() {
|
|
m_Stream.str("");
|
|
}
|
|
|
|
void Insert(const uint8_t *buf, size_t len) {
|
|
m_Stream.write((char *) buf, len);
|
|
}
|
|
|
|
template<typename T>
|
|
void Insert(T t) {
|
|
m_Stream.write((char *) &t, sizeof(T));
|
|
}
|
|
|
|
bool Verify(std::shared_ptr<const i2p::data::IdentityEx> ident, const uint8_t *signature) const {
|
|
return ident->Verify((const uint8_t *) m_Stream.str().c_str(), m_Stream.str().size(), signature);
|
|
}
|
|
|
|
void Sign(const i2p::data::PrivateKeys &keys, uint8_t *signature) const {
|
|
keys.Sign((const uint8_t *) m_Stream.str().c_str(), m_Stream.str().size(), signature);
|
|
}
|
|
|
|
private:
|
|
|
|
std::stringstream m_Stream;
|
|
};
|
|
|
|
class TransportSession {
|
|
public:
|
|
|
|
TransportSession(std::shared_ptr<const i2p::data::RouterInfo> router, int terminationTimeout) :
|
|
m_NumSentBytes(0), m_NumReceivedBytes(0), m_IsOutgoing(router),
|
|
m_TerminationTimeout(terminationTimeout),
|
|
m_LastActivityTimestamp(i2p::util::GetSecondsSinceEpoch()) {
|
|
if (router)
|
|
m_RemoteIdentity = router->GetRouterIdentity();
|
|
m_CreationTime = m_LastActivityTimestamp;
|
|
}
|
|
|
|
virtual ~TransportSession() {};
|
|
|
|
virtual void Done() = 0;
|
|
|
|
std::string GetIdentHashBase64() const {
|
|
return m_RemoteIdentity ? m_RemoteIdentity->GetIdentHash().ToBase64() : "";
|
|
}
|
|
|
|
std::shared_ptr<const i2p::data::IdentityEx> GetRemoteIdentity() {
|
|
std::lock_guard<std::mutex> l(m_RemoteIdentityMutex);
|
|
return m_RemoteIdentity;
|
|
}
|
|
|
|
void SetRemoteIdentity(std::shared_ptr<const i2p::data::IdentityEx> ident) {
|
|
std::lock_guard<std::mutex> l(m_RemoteIdentityMutex);
|
|
m_RemoteIdentity = ident;
|
|
}
|
|
|
|
size_t GetNumSentBytes() const { return m_NumSentBytes; };
|
|
|
|
size_t GetNumReceivedBytes() const { return m_NumReceivedBytes; };
|
|
|
|
bool IsOutgoing() const { return m_IsOutgoing; };
|
|
|
|
int GetTerminationTimeout() const { return m_TerminationTimeout; };
|
|
|
|
void SetTerminationTimeout(int terminationTimeout) { m_TerminationTimeout = terminationTimeout; };
|
|
|
|
bool IsTerminationTimeoutExpired(uint64_t ts) const {
|
|
return ts >= m_LastActivityTimestamp + GetTerminationTimeout();
|
|
};
|
|
|
|
uint32_t GetCreationTime() const { return m_CreationTime; };
|
|
|
|
void SetCreationTime(uint32_t ts) { m_CreationTime = ts; }; // for introducers
|
|
|
|
virtual uint32_t GetRelayTag() const { return 0; };
|
|
|
|
virtual void SendLocalRouterInfo(bool update = false) { SendI2NPMessages({CreateDatabaseStoreMsg()}); };
|
|
|
|
virtual void SendI2NPMessages(const std::vector<std::shared_ptr<I2NPMessage> > &msgs) = 0;
|
|
|
|
protected:
|
|
|
|
std::shared_ptr<const i2p::data::IdentityEx> m_RemoteIdentity;
|
|
mutable std::mutex m_RemoteIdentityMutex;
|
|
size_t m_NumSentBytes, m_NumReceivedBytes;
|
|
bool m_IsOutgoing;
|
|
int m_TerminationTimeout;
|
|
uint64_t m_LastActivityTimestamp;
|
|
uint32_t m_CreationTime; // seconds since epoch
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif
|