mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-04-28 11:47:48 +02:00
common Identity for LeaseSet and RouterInfo
This commit is contained in:
parent
6f1d4f4300
commit
d714d7fe6c
9 changed files with 105 additions and 28 deletions
|
@ -3,13 +3,20 @@
|
|||
#include <cryptopp/gzip.h>
|
||||
#include "Log.h"
|
||||
#include "RouterInfo.h"
|
||||
#include "RouterContext.h"
|
||||
#include "Streaming.h"
|
||||
|
||||
namespace i2p
|
||||
{
|
||||
namespace stream
|
||||
{
|
||||
void StreamingDestination::HandleNextPacket (const uint8_t * buf, size_t len)
|
||||
Stream::Stream (const i2p::data::IdentHash& destination):
|
||||
m_SendStreamID (0)
|
||||
{
|
||||
m_RecvStreamID = i2p::context.GetRandomNumberGenerator ().GenerateWord32 ();
|
||||
}
|
||||
|
||||
void Stream::HandleNextPacket (const uint8_t * buf, size_t len)
|
||||
{
|
||||
const uint8_t * end = buf + len;
|
||||
buf += 4; // sendStreamID
|
||||
|
@ -42,15 +49,45 @@ namespace stream
|
|||
if (flags & PACKET_FLAG_FROM_INCLUDED)
|
||||
{
|
||||
LogPrint ("From identity");
|
||||
optionalData += sizeof (i2p::data::RouterIdentity);
|
||||
optionalData += sizeof (i2p::data::Identity);
|
||||
}
|
||||
|
||||
// we have reached payload section
|
||||
std::string str((const char *)buf, end-buf);
|
||||
LogPrint ("Payload: ", str);
|
||||
}
|
||||
|
||||
StreamingDestination m_SharedLocalDestination;
|
||||
|
||||
void StreamingDestination::HandleNextPacket (const uint8_t * buf, size_t len)
|
||||
{
|
||||
uint32_t sendStreamID = *(uint32_t *)(buf);
|
||||
auto it = m_Streams.find (sendStreamID);
|
||||
if (it != m_Streams.end ())
|
||||
it->second->HandleNextPacket (buf, len);
|
||||
else
|
||||
LogPrint ("Unknown stream ", sendStreamID);
|
||||
}
|
||||
|
||||
|
||||
Stream * StreamingDestination::CreateNewStream (const i2p::data::IdentHash& destination)
|
||||
{
|
||||
/*i2p::data::LeaseSet * leaseSet = i2p::data::netdb.FindLeaseSet (destination);
|
||||
if (!leaseSet)
|
||||
{
|
||||
i2p::data::netdb.RequestDestination (destination);
|
||||
sleep (5); // wait for 5 seconds
|
||||
leaseSet = i2p::data::netdb.FindLeaseSet (destination);
|
||||
if (!leaseSet)
|
||||
{
|
||||
LogPrint ("Couldn't find LeaseSet");
|
||||
return nullptr;
|
||||
}
|
||||
} */
|
||||
Stream * s = new Stream (destination);
|
||||
m_Streams[s->GetRecvStreamID ()] = s;
|
||||
return s;
|
||||
}
|
||||
|
||||
void HandleDataMessage (i2p::data::IdentHash * destination, const uint8_t * buf, size_t len)
|
||||
{
|
||||
uint32_t length = be32toh (*(uint32_t *)buf);
|
||||
|
@ -63,12 +100,32 @@ namespace stream
|
|||
decompressor.Put (buf, length);
|
||||
decompressor.MessageEnd();
|
||||
uint8_t uncompressed[2048];
|
||||
int uncomressedSize = decompressor.MaxRetrievable ();
|
||||
decompressor.Get (uncompressed, uncomressedSize);
|
||||
int uncompressedSize = decompressor.MaxRetrievable ();
|
||||
decompressor.Get (uncompressed, uncompressedSize);
|
||||
// then forward to streaming engine
|
||||
// TODO: we have onle one destination, might be more
|
||||
m_SharedLocalDestination.HandleNextPacket (uncompressed, uncompressedSize);
|
||||
}
|
||||
else
|
||||
LogPrint ("Data: protocol ", buf[9], " is not supported");
|
||||
}
|
||||
|
||||
I2NPMessage * CreateDataMessage (Stream * s, uint8_t * payload, size_t len)
|
||||
{
|
||||
I2NPMessage * msg = NewI2NPMessage ();
|
||||
CryptoPP::Gzip compressor;
|
||||
compressor.Put (payload, len);
|
||||
compressor.MessageEnd();
|
||||
int size = compressor.MaxRetrievable ();
|
||||
uint8_t * buf = msg->GetPayload ();
|
||||
*(uint16_t *)buf = htobe32 (size); // length
|
||||
buf += 4;
|
||||
compressor.Get (buf, size);
|
||||
buf[9] = 6; // streaming protocol
|
||||
msg->len += size + 4;
|
||||
FillI2NPMessageHeader (msg, eI2NPData);
|
||||
|
||||
return msg;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue