mirror of https://github.com/oxen-io/lokinet
* implement path transfer message
* update dht docs * update other docs * start working on dht for hidden services * fix up unit tests for dht * update makefile and other build filespull/5/head
Jeff Becker
6 years ago
parent
7d9bffdf9c
commit
18b50f4a74
@ -0,0 +1,31 @@
|
|||||||
|
#ifndef LLARP_MESSAGES_DHT_HPP
|
||||||
|
#define LLARP_MESSAGES_DHT_HPP
|
||||||
|
#include <llarp/dht.hpp>
|
||||||
|
#include <llarp/routing/message.hpp>
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
namespace llarp
|
||||||
|
{
|
||||||
|
namespace routing
|
||||||
|
{
|
||||||
|
struct DHTMessage : public IMessage
|
||||||
|
{
|
||||||
|
std::vector< llarp::dht::IMessage* > M;
|
||||||
|
uint64_t V = 0;
|
||||||
|
|
||||||
|
~DHTMessage();
|
||||||
|
|
||||||
|
bool
|
||||||
|
DecodeKey(llarp_buffer_t key, llarp_buffer_t* val);
|
||||||
|
|
||||||
|
bool
|
||||||
|
BEncode(llarp_buffer_t* buf) const;
|
||||||
|
|
||||||
|
bool
|
||||||
|
HandleMessage(IMessageHandler* h, llarp_router* r) const;
|
||||||
|
};
|
||||||
|
} // namespace routing
|
||||||
|
} // namespace llarp
|
||||||
|
|
||||||
|
#endif
|
||||||
@ -0,0 +1,61 @@
|
|||||||
|
#include <llarp/messages/dht.hpp>
|
||||||
|
#include "../router.hpp"
|
||||||
|
|
||||||
|
namespace llarp
|
||||||
|
{
|
||||||
|
namespace routing
|
||||||
|
{
|
||||||
|
DHTMessage::~DHTMessage()
|
||||||
|
{
|
||||||
|
for(auto& msg : M)
|
||||||
|
delete msg;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
DHTMessage::DecodeKey(llarp_buffer_t key, llarp_buffer_t* val)
|
||||||
|
{
|
||||||
|
llarp::dht::Key_t from;
|
||||||
|
from.Zero();
|
||||||
|
if(llarp_buffer_eq(key, "M"))
|
||||||
|
{
|
||||||
|
return llarp::dht::DecodeMesssageList(from, val, M);
|
||||||
|
}
|
||||||
|
else if(llarp_buffer_eq(key, "V"))
|
||||||
|
{
|
||||||
|
return bencode_read_integer(val, &V);
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
DHTMessage::BEncode(llarp_buffer_t* buf) const
|
||||||
|
{
|
||||||
|
if(!bencode_start_dict(buf))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
if(!BEncodeWriteDictMsgType(buf, "A", "M"))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictBEncodeList("M", M, buf))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictInt(buf, "V", LLARP_PROTO_VERSION))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
return bencode_end(buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
DHTMessage::HandleMessage(IMessageHandler* h, llarp_router* r) const
|
||||||
|
{
|
||||||
|
// set source as us
|
||||||
|
llarp::dht::Key_t us = r->pubkey();
|
||||||
|
for(auto& msg : M)
|
||||||
|
{
|
||||||
|
msg->From = us;
|
||||||
|
if(!r->dht->impl.RelayRequestForPath(from, msg))
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace routing
|
||||||
|
} // namespace llarp
|
||||||
@ -0,0 +1,65 @@
|
|||||||
|
#include <llarp/messages/path_transfer.hpp>
|
||||||
|
#include "../router.hpp"
|
||||||
|
|
||||||
|
namespace llarp
|
||||||
|
{
|
||||||
|
namespace routing
|
||||||
|
{
|
||||||
|
PathTransferMessage::PathTransferMessage() : IMessage()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
PathTransferMessage::~PathTransferMessage()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
PathTransferMessage::DecodeKey(llarp_buffer_t key, llarp_buffer_t* val)
|
||||||
|
{
|
||||||
|
bool read = false;
|
||||||
|
if(!BEncodeMaybeReadDictEntry("P", P, read, key, val))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeMaybeReadDictEntry("T", T, read, key, val))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeMaybeReadDictInt("V", V, read, key, val))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeMaybeReadDictEntry("Y", Y, read, key, val))
|
||||||
|
return false;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
PathTransferMessage::BEncode(llarp_buffer_t* buf) const
|
||||||
|
{
|
||||||
|
if(!bencode_start_dict(buf))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictMsgType(buf, "A", "T"))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictEntry("P", P, buf))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictEntry("T", T, buf))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictInt(buf, "V", LLARP_PROTO_VERSION))
|
||||||
|
return false;
|
||||||
|
if(!BEncodeWriteDictEntry("Y", Y, buf))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
return bencode_end(buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
PathTransferMessage::HandleMessage(IMessageHandler* h,
|
||||||
|
llarp_router* r) const
|
||||||
|
{
|
||||||
|
auto path = r->paths.GetByUpstream(r->pubkey(), P);
|
||||||
|
if(path)
|
||||||
|
{
|
||||||
|
return path->HandleDownstream(T.Buffer(), Y, r);
|
||||||
|
}
|
||||||
|
llarp::Warn("No such local path for path transfer src=", from,
|
||||||
|
" dst=", P);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
} // namespace routing
|
||||||
|
|
||||||
|
} // namespace llarp
|
||||||
Loading…
Reference in New Issue