You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
lokinet/llarp/link/utp.cpp

881 lines
23 KiB
C++

#include <llarp/link/utp.hpp>
#include "router.hpp"
#include <llarp/messages/link_intro.hpp>
6 years ago
#include <llarp/messages/discard.hpp>
#include <llarp/buffer.hpp>
#include <llarp/endian.h>
#include <utp.h>
6 years ago
#include <cassert>
6 years ago
#include <tuple>
6 years ago
#include <deque>
6 years ago
#ifdef __linux__
#include <linux/errqueue.h>
#include <netinet/ip_icmp.h>
#endif
namespace llarp
{
namespace utp
{
6 years ago
constexpr size_t FragmentHashSize = 32;
6 years ago
constexpr size_t FragmentNonceSize = 24;
constexpr size_t FragmentOverheadSize =
FragmentHashSize + FragmentNonceSize;
6 years ago
constexpr size_t FragmentBodyPayloadSize = 1024;
6 years ago
constexpr size_t FragmentBodyOverhead = sizeof(uint32_t) * 2;
constexpr size_t FragmentBodySize =
FragmentBodyOverhead + FragmentBodyPayloadSize;
6 years ago
constexpr size_t FragmentBufferSize =
6 years ago
FragmentOverheadSize + FragmentBodySize;
6 years ago
typedef llarp::AlignedBuffer< FragmentBufferSize > FragmentBuffer;
6 years ago
typedef llarp::AlignedBuffer< MAX_LINK_MSG_SIZE > MessageBuffer;
struct LinkLayer;
struct BaseSession : public ILinkSession
{
RouterContact remoteRC;
6 years ago
utp_socket* sock;
LinkLayer* parent;
bool gotLIM;
PubKey remoteTransportPubKey;
Addr remoteAddr;
SharedSecret sessionKey;
llarp_time_t lastActive;
6 years ago
const static llarp_time_t sessionTimeout = 30 * 1000;
6 years ago
std::deque< FragmentBuffer > sendq;
6 years ago
size_t sendBufOffset;
FragmentBuffer recvBuf;
6 years ago
size_t recvBufOffset;
6 years ago
MessageBuffer recvMsg;
size_t recvMsgOffset;
bool stalled = false;
6 years ago
6 years ago
void
6 years ago
Alive();
6 years ago
6 years ago
/// base
BaseSession(LinkLayer* p);
6 years ago
/// outbound
BaseSession(LinkLayer* p, utp_socket* s, const RouterContact& rc,
6 years ago
const AddressInfo& addr);
/// inbound
BaseSession(LinkLayer* p, utp_socket* s, const Addr& remote);
6 years ago
enum State
{
eInitial,
eConnecting,
eLinkEstablished, // when utp connection is established
eCryptoHandshake, // crypto handshake initiated
eSessionReady, // session is ready
eClose // utp connection is closed
};
6 years ago
llarp_router*
Router();
State state;
void
6 years ago
OnLinkEstablished(LinkLayer* p)
{
6 years ago
parent = p;
EnterState(eLinkEstablished);
llarp::LogDebug("link established with ", remoteAddr);
}
6 years ago
void
EnterState(State st);
6 years ago
BaseSession();
virtual ~BaseSession();
void
6 years ago
PumpWrite()
{
6 years ago
while(sendq.size() > 0 && !stalled)
{
6 years ago
ssize_t expect = FragmentBufferSize - sendBufOffset;
ssize_t s = write_ll(sendq.front().data() + sendBufOffset, expect);
if(s != expect)
6 years ago
{
6 years ago
llarp::LogDebug("stalled at offset=", sendBufOffset, " sz=", s,
" to ", remoteAddr);
sendBufOffset += s;
6 years ago
stalled = true;
}
6 years ago
else
{
sendBufOffset = 0;
6 years ago
sendq.pop_front();
6 years ago
}
}
}
6 years ago
ssize_t
6 years ago
write_ll(byte_t* buf, size_t sz)
{
6 years ago
if(sock == nullptr)
{
llarp::LogWarn("write_ll failed: no socket");
6 years ago
return 0;
}
6 years ago
ssize_t s = utp_write(sock, buf, sz);
llarp::LogDebug("write_ll ", s, " of ", sz, " bytes to ", remoteAddr);
return s;
}
bool
6 years ago
VerifyThenDecrypt(byte_t* buf);
void
6 years ago
EncryptThenHash(const byte_t* ptr, uint32_t sz, bool isLastFragment);
bool
6 years ago
QueueWriteBuffers(llarp_buffer_t buf)
{
6 years ago
llarp::LogDebug("write ", buf.sz, " bytes to ", remoteAddr);
lastActive = llarp_time_now_ms();
6 years ago
size_t sz = buf.sz;
byte_t* ptr = buf.base;
while(sz)
{
6 years ago
uint32_t s = std::min(FragmentBodyPayloadSize, sz);
6 years ago
EncryptThenHash(ptr, s, ((sz - s) == 0));
ptr += s;
sz -= s;
}
return true;
}
6 years ago
void
Connect()
{
utp_connect(sock, remoteAddr, remoteAddr.SockLen());
EnterState(eConnecting);
}
void
OutboundLinkEstablished(LinkLayer* p)
{
OnLinkEstablished(p);
OutboundHandshake();
6 years ago
}
// send first message
6 years ago
void
OutboundHandshake();
6 years ago
// mix keys
6 years ago
bool
6 years ago
DoKeyExchange(llarp_transport_dh_func dh, const KeyExchangeNonce& n,
const PubKey& other, const SecretKey& secret)
{
ShortHash t_h;
AlignedBuffer< 64 > tmp;
memcpy(tmp.data(), sessionKey, sessionKey.size());
memcpy(tmp.data() + sessionKey.size(), n, n.size());
// t_h = HS(K + L.n)
if(!Router()->crypto.shorthash(t_h, ConstBuffer(tmp)))
{
llarp::LogError("failed to mix key to ", remoteAddr);
return false;
}
// K = TKE(a.p, B_a.e, sk, t_h)
if(!dh(sessionKey, other, secret, t_h))
{
llarp::LogError("key exchange with ", other, " failed");
6 years ago
return false;
}
llarp::LogDebug("keys mixed with session to ", remoteAddr);
6 years ago
return true;
}
void
6 years ago
TickImpl(llarp_time_t now)
{
}
void
6 years ago
Close();
bool
Recv(const void* buf, size_t sz)
{
6 years ago
Alive();
byte_t* ptr = (byte_t*)buf;
llarp::LogDebug("utp read ", sz, " from ", remoteAddr);
6 years ago
size_t s = sz;
6 years ago
// process leftovers
if(recvBufOffset)
{
6 years ago
auto left = FragmentBufferSize - recvBufOffset;
6 years ago
if(s >= left)
6 years ago
{
6 years ago
// yes it fills it
6 years ago
llarp::LogDebug("process leftovers, offset=", recvBufOffset,
" sz=", s, " left=", left);
memcpy(recvBuf.data() + recvBufOffset, ptr, left);
s -= left;
recvBufOffset = 0;
ptr += left;
6 years ago
if(!VerifyThenDecrypt(recvBuf.data()))
6 years ago
return false;
}
}
6 years ago
// process full fragments
6 years ago
while(s >= FragmentBufferSize)
{
6 years ago
recvBufOffset = 0;
6 years ago
llarp::LogDebug("process full sz=", s);
if(!VerifyThenDecrypt(ptr))
return false;
ptr += FragmentBufferSize;
s -= FragmentBufferSize;
}
if(s)
{
// hold onto leftovers
llarp::LogDebug("leftovers sz=", s);
6 years ago
memcpy(recvBuf.data() + recvBufOffset, ptr, s);
recvBufOffset += s;
6 years ago
}
return true;
}
bool
InboundLIM(const LinkIntroMessage* msg);
bool
OutboundLIM(const LinkIntroMessage* msg);
bool
6 years ago
IsTimedOut(llarp_time_t now) const
{
if(now < lastActive)
return false;
6 years ago
auto dlt = now - lastActive;
if(dlt >= sessionTimeout)
{
llarp::LogDebug("session timeout reached for ", remoteAddr);
return true;
}
return false;
}
const PubKey&
6 years ago
RemotePubKey() const
{
return remoteRC.pubkey;
}
const Addr&
6 years ago
RemoteEndpoint() const
{
return remoteAddr;
}
void
MarkEstablished();
6 years ago
}; // namespace utp
struct LinkLayer : public ILinkLayer
{
utp_context* _utp_ctx = nullptr;
6 years ago
llarp_router* router = nullptr;
static uint64
6 years ago
OnRead(utp_callback_arguments* arg);
static uint64
SendTo(utp_callback_arguments* arg)
{
LinkLayer* l =
static_cast< LinkLayer* >(utp_context_get_userdata(arg->context));
6 years ago
llarp::LogDebug("utp_sendto ", Addr(*arg->address), " ", arg->len,
" bytes");
if(sendto(l->m_udp.fd, arg->buf, arg->len, arg->flags, arg->address,
arg->address_len)
== -1)
{
llarp::LogError("sendto failed: ", strerror(errno));
}
return 0;
}
static uint64
OnError(utp_callback_arguments* arg)
{
llarp::LogError(utp_error_code_names[arg->error_code]);
return 0;
}
static uint64
6 years ago
OnStateChange(utp_callback_arguments*);
static uint64
OnAccept(utp_callback_arguments*);
6 years ago
static uint64
OnLog(utp_callback_arguments* arg)
{
llarp::LogDebug(arg->buf);
return 0;
}
6 years ago
LinkLayer(llarp_router* r) : ILinkLayer()
{
6 years ago
router = r;
_utp_ctx = utp_init(2);
utp_context_set_userdata(_utp_ctx, this);
utp_set_callback(_utp_ctx, UTP_SENDTO, &LinkLayer::SendTo);
utp_set_callback(_utp_ctx, UTP_ON_ACCEPT, &LinkLayer::OnAccept);
6 years ago
utp_set_callback(_utp_ctx, UTP_ON_STATE_CHANGE,
&LinkLayer::OnStateChange);
utp_set_callback(_utp_ctx, UTP_ON_READ, &LinkLayer::OnRead);
6 years ago
utp_set_callback(_utp_ctx, UTP_ON_ERROR, &LinkLayer::OnError);
utp_set_callback(_utp_ctx, UTP_LOG, &LinkLayer::OnLog);
6 years ago
utp_context_set_option(_utp_ctx, UTP_LOG_NORMAL, 1);
utp_context_set_option(_utp_ctx, UTP_LOG_MTU, 1);
utp_context_set_option(_utp_ctx, UTP_LOG_DEBUG, 1);
6 years ago
utp_context_set_option(_utp_ctx, UTP_SNDBUF, MAX_LINK_MSG_SIZE * 16);
6 years ago
utp_context_set_option(_utp_ctx, UTP_RCVBUF, MAX_LINK_MSG_SIZE * 64);
}
~LinkLayer()
{
utp_destroy(_utp_ctx);
}
uint16_t
Rank() const
{
return 1;
}
void
RecvFrom(const Addr& from, const void* buf, size_t sz)
{
utp_process_udp(_utp_ctx, (const byte_t*)buf, sz, from, from.SockLen());
}
6 years ago
#ifdef __linux__
void
ProcessICMP()
{
do
{
byte_t vec_buf[4096], ancillary_buf[4096];
struct iovec iov = {vec_buf, sizeof(vec_buf)};
struct sockaddr_in remote;
struct msghdr msg;
ssize_t len;
struct cmsghdr* cmsg;
struct sock_extended_err* e;
struct sockaddr* icmp_addr;
struct sockaddr_in* icmp_sin;
memset(&msg, 0, sizeof(msg));
msg.msg_name = &remote;
msg.msg_namelen = sizeof(remote);
msg.msg_iov = &iov;
msg.msg_iovlen = 1;
msg.msg_flags = 0;
msg.msg_control = ancillary_buf;
msg.msg_controllen = sizeof(ancillary_buf);
len = recvmsg(m_udp.fd, &msg, MSG_ERRQUEUE | MSG_DONTWAIT);
if(len < 0)
{
if(errno == EAGAIN || errno == EWOULDBLOCK)
errno = 0;
else
llarp::LogError("failed to read icmp for utp ", strerror(errno));
return;
}
for(cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg))
{
if(cmsg->cmsg_type != IP_RECVERR)
{
continue;
}
if(cmsg->cmsg_level != SOL_IP)
{
continue;
}
e = (struct sock_extended_err*)CMSG_DATA(cmsg);
if(!e)
continue;
if(e->ee_origin != SO_EE_ORIGIN_ICMP)
{
continue;
}
icmp_addr = (struct sockaddr*)SO_EE_OFFENDER(e);
icmp_sin = (struct sockaddr_in*)icmp_addr;
if(icmp_sin->sin_port != 0)
{
continue;
}
if(e->ee_type == 3 && e->ee_code == 4)
{
utp_process_icmp_fragmentation(_utp_ctx, vec_buf, len,
(struct sockaddr*)&remote,
sizeof(remote), e->ee_info);
}
else
{
utp_process_icmp_error(_utp_ctx, vec_buf, len,
(struct sockaddr*)&remote, sizeof(remote));
}
}
} while(true);
}
#endif
void
Pump()
{
6 years ago
utp_issue_deferred_acks(_utp_ctx);
6 years ago
#ifdef __linux__
ProcessICMP();
#endif
6 years ago
ILinkLayer::Pump();
6 years ago
}
void
Stop()
{
}
6 years ago
llarp_router*
GetRouter();
bool
KeyGen(SecretKey& k)
{
6 years ago
router->crypto.encryption_keygen(k);
return true;
}
6 years ago
void
Tick(llarp_time_t now)
{
utp_check_timeouts(_utp_ctx);
ILinkLayer::Tick(now);
}
6 years ago
ILinkSession*
NewOutboundSession(const RouterContact& rc, const AddressInfo& addr);
utp_socket*
NewSocket()
{
return utp_create_socket(_utp_ctx);
}
const char*
Name() const
{
return "utp";
}
};
std::unique_ptr< ILinkLayer >
NewServer(llarp_router* r)
{
6 years ago
return std::unique_ptr< LinkLayer >(new LinkLayer(r));
}
BaseSession::BaseSession(LinkLayer* p)
{
parent = p;
6 years ago
remoteTransportPubKey.Zero();
6 years ago
recvMsgOffset = 0;
6 years ago
SendKeepAlive = [&]() -> bool {
/*
if(sendq.size() == 0 && state == eSessionReady)
6 years ago
{
DiscardMessage msg;
byte_t tmp[128] = {0};
auto buf = llarp::StackBuffer< decltype(tmp) >(tmp);
if(!msg.BEncode(&buf))
return false;
buf.sz = buf.cur - buf.base;
buf.cur = buf.base;
if(!this->QueueWriteBuffers(buf))
return false;
}
*/
6 years ago
return true;
};
gotLIM = false;
6 years ago
sendBufOffset = 0;
6 years ago
recvBufOffset = 0;
TimedOut = [&](llarp_time_t now) -> bool {
6 years ago
return this->IsTimedOut(now) || this->state == eClose;
6 years ago
};
6 years ago
GetPubKey = std::bind(&BaseSession::RemotePubKey, this);
6 years ago
lastActive = llarp_time_now_ms();
6 years ago
Pump = std::bind(&BaseSession::PumpWrite, this);
6 years ago
Tick = std::bind(&BaseSession::TickImpl, this, std::placeholders::_1);
6 years ago
SendMessageBuffer = std::bind(&BaseSession::QueueWriteBuffers, this,
std::placeholders::_1);
IsEstablished = [=]() {
return this->state == eSessionReady || this->state == eLinkEstablished;
6 years ago
};
6 years ago
SendClose = std::bind(&BaseSession::Close, this);
GetRemoteEndpoint = std::bind(&BaseSession::RemoteEndpoint, this);
6 years ago
}
BaseSession::BaseSession(LinkLayer* p, utp_socket* s,
6 years ago
const RouterContact& rc, const AddressInfo& addr)
: BaseSession(p)
{
6 years ago
remoteRC.Clear();
remoteTransportPubKey = addr.pubkey;
remoteRC = rc;
sock = s;
assert(utp_set_userdata(sock, this) == this);
6 years ago
assert(s == sock);
6 years ago
remoteAddr = addr;
Start = std::bind(&BaseSession::Connect, this);
GotLIM =
std::bind(&BaseSession::OutboundLIM, this, std::placeholders::_1);
6 years ago
}
BaseSession::BaseSession(LinkLayer* p, utp_socket* s, const Addr& addr)
: BaseSession(p)
6 years ago
{
p->router->crypto.shorthash(sessionKey,
InitBuffer(p->router->pubkey(), PUBKEYSIZE));
6 years ago
remoteRC.Clear();
sock = s;
6 years ago
assert(s == sock);
6 years ago
assert(utp_set_userdata(sock, this) == this);
remoteAddr = addr;
Start = []() {};
GotLIM = std::bind(&BaseSession::InboundLIM, this, std::placeholders::_1);
}
bool
BaseSession::InboundLIM(const LinkIntroMessage* msg)
{
if(gotLIM && remoteRC.pubkey != msg->rc.pubkey)
{
return false;
}
remoteRC = msg->rc;
gotLIM = true;
if(!DoKeyExchange(Router()->crypto.transport_dh_server, msg->N,
remoteRC.enckey, parent->TransportSecretKey()))
return false;
EnterState(eSessionReady);
return true;
}
bool
BaseSession::OutboundLIM(const LinkIntroMessage* msg)
{
if(gotLIM && remoteRC.pubkey != msg->rc.pubkey)
{
return false;
}
remoteRC = msg->rc;
gotLIM = true;
// TODO: update address info pubkey
return DoKeyExchange(Router()->crypto.transport_dh_client, msg->N,
remoteTransportPubKey, Router()->encryption);
}
void
BaseSession::OutboundHandshake()
{
// set session key
Router()->crypto.shorthash(sessionKey, ConstBuffer(remoteRC.pubkey));
byte_t tmp[LinkIntroMessage::MaxSize];
auto buf = StackBuffer< decltype(tmp) >(tmp);
// build our RC
LinkIntroMessage msg;
msg.rc = Router()->rc();
if(!msg.rc.VerifySignature(&Router()->crypto))
{
llarp::LogError("our RC is invalid? closing session to", remoteAddr);
Close();
return;
}
msg.N.Randomize();
msg.P = DefaultLinkSessionLifetime;
if(!msg.Sign(&Router()->crypto, Router()->identity))
{
llarp::LogError("failed to sign LIM for outbound handshake to ",
remoteAddr);
Close();
return;
}
// encode
if(!msg.BEncode(&buf))
{
llarp::LogError("failed to encode LIM for handshake to ", remoteAddr);
Close();
return;
}
// rewind
buf.sz = buf.cur - buf.base;
buf.cur = buf.base;
// send
if(!SendMessageBuffer(buf))
{
llarp::LogError("failed to send handshake to ", remoteAddr);
Close();
return;
}
// mix keys
if(!DoKeyExchange(Router()->crypto.transport_dh_client, msg.N,
remoteTransportPubKey, Router()->encryption))
{
llarp::LogError("failed to mix keys for outbound session to ",
remoteAddr);
Close();
return;
}
EnterState(eSessionReady);
6 years ago
}
6 years ago
llarp_router*
BaseSession::Router()
{
6 years ago
return parent->router;
}
BaseSession::~BaseSession()
{
6 years ago
if(sock)
{
utp_set_userdata(sock, nullptr);
6 years ago
sock = nullptr;
6 years ago
}
}
6 years ago
ILinkSession*
LinkLayer::NewOutboundSession(const RouterContact& rc,
const AddressInfo& addr)
{
return new BaseSession(this, utp_create_socket(_utp_ctx), rc, addr);
6 years ago
}
uint64
LinkLayer::OnRead(utp_callback_arguments* arg)
{
BaseSession* self =
static_cast< BaseSession* >(utp_get_userdata(arg->socket));
6 years ago
6 years ago
if(self)
{
6 years ago
if(self->state == BaseSession::eClose)
{
return 0;
}
if(!self->Recv(arg->buf, arg->len))
6 years ago
{
llarp::LogDebug("recv fail for ", self->remoteAddr);
self->Close();
return 0;
6 years ago
}
utp_read_drained(arg->socket);
6 years ago
}
else
{
llarp::LogWarn("utp_socket got data with no underlying session");
}
return 0;
}
6 years ago
uint64
LinkLayer::OnStateChange(utp_callback_arguments* arg)
{
6 years ago
LinkLayer* l =
static_cast< LinkLayer* >(utp_context_get_userdata(arg->context));
BaseSession* session =
static_cast< BaseSession* >(utp_get_userdata(arg->socket));
6 years ago
if(session)
6 years ago
{
6 years ago
if(arg->state == UTP_STATE_CONNECT)
{
if(session->state == BaseSession::eClose)
{
return 0;
}
session->OutboundLinkEstablished(l);
}
6 years ago
else if(arg->state == UTP_STATE_WRITABLE)
{
6 years ago
if(session->IsEstablished())
{
llarp::LogDebug("write resumed for ", session->remoteAddr);
session->stalled = false;
session->PumpWrite();
}
6 years ago
}
6 years ago
else if(arg->state == UTP_STATE_EOF)
{
6 years ago
llarp::LogDebug("got eof from ", session->remoteAddr);
session->Close();
6 years ago
}
6 years ago
}
return 0;
}
uint64
LinkLayer::OnAccept(utp_callback_arguments* arg)
{
LinkLayer* self =
static_cast< LinkLayer* >(utp_context_get_userdata(arg->context));
Addr remote(*arg->address);
llarp::LogDebug("utp accepted from ", remote);
BaseSession* session = new BaseSession(self, arg->socket, remote);
self->PutSession(remote, session);
6 years ago
session->OnLinkEstablished(self);
return 0;
}
6 years ago
void
6 years ago
BaseSession::EncryptThenHash(const byte_t* ptr, uint32_t sz,
bool isLastFragment)
6 years ago
{
sendq.emplace_back();
auto& buf = sendq.back();
6 years ago
llarp::LogDebug("encrypt then hash ", sz, " bytes last=", isLastFragment);
6 years ago
buf.Randomize();
6 years ago
byte_t* nonce = buf.data() + FragmentHashSize;
byte_t* body = nonce + FragmentNonceSize;
byte_t* base = body;
6 years ago
if(isLastFragment)
htobe32buf(body, 0);
6 years ago
else
htobe32buf(body, 1);
6 years ago
body += sizeof(uint32_t);
htobe32buf(body, sz);
body += sizeof(uint32_t);
memcpy(body, ptr, sz);
6 years ago
auto payload =
InitBuffer(base, FragmentBufferSize - FragmentOverheadSize);
// encrypt
6 years ago
Router()->crypto.xchacha20(payload, sessionKey, nonce);
6 years ago
payload.base = nonce;
6 years ago
payload.cur = payload.base;
6 years ago
payload.sz = FragmentBufferSize - FragmentHashSize;
// key'd hash
6 years ago
Router()->crypto.hmac(buf.data(), payload, sessionKey);
6 years ago
}
void
BaseSession::EnterState(State st)
{
6 years ago
state = st;
6 years ago
if(st == eSessionReady)
{
6 years ago
parent->MapAddr(remoteRC.pubkey, this);
6 years ago
Router()->HandleLinkSessionEstablished(remoteRC);
}
6 years ago
Alive();
6 years ago
}
bool
6 years ago
BaseSession::VerifyThenDecrypt(byte_t* buf)
6 years ago
{
6 years ago
llarp::LogDebug("verify then decrypt ", remoteAddr);
6 years ago
ShortHash digest;
6 years ago
6 years ago
auto hbuf = InitBuffer(buf + FragmentHashSize,
FragmentBufferSize - FragmentHashSize);
if(!Router()->crypto.hmac(digest.data(), hbuf, sessionKey))
6 years ago
{
llarp::LogError("keyed hash failed");
return false;
}
6 years ago
ShortHash expected(buf);
if(expected != digest)
6 years ago
{
6 years ago
llarp::LogError("Message Integrity Failed: got ", digest, " from ",
6 years ago
remoteAddr, " instead of ", expected);
llarp::DumpBuffer(InitBuffer(buf, FragmentBufferSize));
6 years ago
return false;
}
6 years ago
auto body = InitBuffer(buf + FragmentOverheadSize,
6 years ago
FragmentBufferSize - FragmentOverheadSize);
6 years ago
Router()->crypto.xchacha20(body, sessionKey, buf + FragmentHashSize);
6 years ago
uint32_t upper, lower;
if(!(llarp_buffer_read_uint32(&body, &upper)
&& llarp_buffer_read_uint32(&body, &lower)))
return false;
bool fragmentEnd = upper == 0;
6 years ago
llarp::LogDebug("fragment size ", lower, " from ", remoteAddr);
if(lower + recvMsgOffset > recvMsg.size())
6 years ago
{
llarp::LogError("Fragment too big: ", lower, " bytes");
return false;
}
6 years ago
memcpy(recvMsg.data() + recvMsgOffset, body.cur, lower);
6 years ago
recvMsgOffset += lower;
6 years ago
if(fragmentEnd)
{
// got a message
6 years ago
llarp::LogDebug("end of message from ", remoteAddr);
6 years ago
auto mbuf = InitBuffer(recvMsg.data(), recvMsgOffset);
if(!Router()->HandleRecvLinkMessageBuffer(this, mbuf))
6 years ago
{
llarp::LogWarn("failed to handle message from ", remoteAddr);
llarp::DumpBuffer(mbuf);
}
recvMsgOffset = 0;
6 years ago
}
return true;
}
6 years ago
void
6 years ago
BaseSession::Close()
6 years ago
{
if(state != eClose)
{
6 years ago
if(sock)
{
utp_shutdown(sock, SHUT_RDWR);
utp_close(sock);
llarp::LogDebug("utp_close ", remoteAddr);
utp_set_userdata(sock, nullptr);
}
6 years ago
}
EnterState(eClose);
sock = nullptr;
}
void
BaseSession::Alive()
{
lastActive = llarp_time_now_ms();
6 years ago
}
} // namespace utp
} // namespace llarp