i2pd/LeaseSet.h

114 lines
3.5 KiB
C
Raw Normal View History

2013-11-25 03:10:27 +04:00
#ifndef LEASE_SET_H__
#define LEASE_SET_H__
#include <inttypes.h>
2013-11-29 16:52:09 +04:00
#include <string.h>
2014-01-02 03:19:03 +04:00
#include <vector>
2016-02-09 18:46:27 +03:00
#include <memory>
2013-12-20 07:05:45 +04:00
#include "Identity.h"
2013-11-25 03:10:27 +04:00
namespace i2p
{
2014-07-29 21:44:54 +04:00
namespace tunnel
{
2016-05-25 21:17:34 +03:00
class InboundTunnel;
2014-07-29 21:44:54 +04:00
}
2013-11-25 03:10:27 +04:00
namespace data
2013-11-29 16:52:09 +04:00
{
2016-02-29 05:43:18 +03:00
const int LEASE_ENDDATE_THRESHOLD = 51000; // in milliseconds
2013-11-25 03:10:27 +04:00
struct Lease
{
2015-03-23 19:55:42 +03:00
IdentHash tunnelGateway;
2013-11-25 03:10:27 +04:00
uint32_t tunnelID;
2016-02-09 23:27:23 +03:00
uint64_t endDate; // 0 means invalid
bool isUpdated; // trasient
};
struct LeaseCmp
{
bool operator() (std::shared_ptr<const Lease> l1, std::shared_ptr<const Lease> l2) const
{
if (l1->tunnelID != l2->tunnelID)
return l1->tunnelID < l2->tunnelID;
else
return l1->tunnelGateway < l2->tunnelGateway;
};
2013-11-25 03:10:27 +04:00
};
2016-05-25 21:17:34 +03:00
const size_t MAX_LS_BUFFER_SIZE = 3072;
const size_t LEASE_SIZE = 44; // 32 + 4 + 8
2016-02-02 19:55:38 +03:00
const uint8_t MAX_NUM_LEASES = 16;
2013-11-25 03:10:27 +04:00
class LeaseSet: public RoutingDestination
{
public:
2016-02-08 03:45:06 +03:00
LeaseSet (const uint8_t * buf, size_t len, bool storeLeases = true);
2015-04-08 16:39:02 +03:00
~LeaseSet () { delete[] m_Buffer; };
void Update (const uint8_t * buf, size_t len);
2016-02-17 06:57:38 +03:00
bool IsNewer (const uint8_t * buf, size_t len) const;
2016-02-09 23:27:23 +03:00
void PopulateLeases (); // from buffer
2015-11-03 17:15:49 +03:00
std::shared_ptr<const IdentityEx> GetIdentity () const { return m_Identity; };
2014-10-03 23:08:41 +04:00
2014-07-29 22:31:55 +04:00
const uint8_t * GetBuffer () const { return m_Buffer; };
size_t GetBufferLen () const { return m_BufferLen; };
2015-04-08 17:34:16 +03:00
bool IsValid () const { return m_IsValid; };
2016-02-11 06:51:08 +03:00
const std::vector<std::shared_ptr<const Lease> > GetNonExpiredLeases (bool withThreshold = true) const;
2014-01-15 05:57:33 +04:00
bool HasExpiredLeases () const;
2016-02-08 03:45:06 +03:00
bool IsExpired () const;
2016-02-10 06:42:01 +03:00
bool IsEmpty () const { return m_Leases.empty (); };
2016-02-08 03:45:06 +03:00
uint64_t GetExpirationTime () const { return m_ExpirationTime; };
bool operator== (const LeaseSet& other) const
{ return m_BufferLen == other.m_BufferLen && !memcmp (m_Buffer, other.m_Buffer, m_BufferLen); };
2016-02-09 18:46:27 +03:00
// implements RoutingDestination
const IdentHash& GetIdentHash () const { return m_Identity->GetIdentHash (); };
2013-11-25 03:10:27 +04:00
const uint8_t * GetEncryptionPublicKey () const { return m_EncryptionKey; };
bool IsDestination () const { return true; };
2014-07-22 04:14:11 +04:00
private:
2015-11-03 17:15:49 +03:00
void ReadFromBuffer (bool readIdentity = true);
2016-02-17 06:57:38 +03:00
uint64_t ExtractTimestamp (const uint8_t * buf, size_t len) const; // min expiration time
2013-11-25 03:10:27 +04:00
private:
2016-02-08 03:45:06 +03:00
bool m_IsValid, m_StoreLeases; // we don't need to store leases for floodfill
2016-02-09 23:27:23 +03:00
std::set<std::shared_ptr<Lease>, LeaseCmp> m_Leases;
2016-02-08 03:45:06 +03:00
uint64_t m_ExpirationTime; // in milliseconds
2015-11-03 17:15:49 +03:00
std::shared_ptr<const IdentityEx> m_Identity;
2013-11-25 03:10:27 +04:00
uint8_t m_EncryptionKey[256];
2015-04-08 16:39:02 +03:00
uint8_t * m_Buffer;
2014-07-29 21:44:54 +04:00
size_t m_BufferLen;
2013-11-25 03:10:27 +04:00
};
2016-05-25 21:17:34 +03:00
class LocalLeaseSet
{
public:
LocalLeaseSet (std::shared_ptr<const IdentityEx> identity, const uint8_t * encryptionPublicKey, std::vector<std::shared_ptr<i2p::tunnel::InboundTunnel> > tunnels);
~LocalLeaseSet () { delete[] m_Buffer; };
const uint8_t * GetBuffer () const { return m_Buffer; };
2016-05-25 22:10:28 +03:00
uint8_t * GetSignature () { return m_Buffer + m_BufferLen - GetSignatureLen (); };
2016-05-25 21:17:34 +03:00
size_t GetBufferLen () const { return m_BufferLen; };
size_t GetSignatureLen () const { return m_Identity->GetSignatureLen (); };
const IdentHash& GetIdentHash () const { return m_Identity->GetIdentHash (); };
2016-05-25 22:10:28 +03:00
bool IsExpired () const;
bool operator== (const LeaseSet& other) const
{ return m_BufferLen == other.GetBufferLen () && !memcmp (other.GetBuffer (), other.GetBuffer (), m_BufferLen); };
2016-05-25 21:17:34 +03:00
private:
2016-05-25 22:10:28 +03:00
uint64_t m_ExpirationTime; // in milliseconds
2016-05-25 21:17:34 +03:00
std::shared_ptr<const IdentityEx> m_Identity;
uint8_t * m_Buffer;
size_t m_BufferLen;
};
2013-11-25 03:10:27 +04:00
}
}
#endif