mirror of https://github.com/oxen-io/lokinet
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.
158 lines
3.1 KiB
C++
158 lines
3.1 KiB
C++
6 years ago
|
#include <util/bencode.h>
|
||
|
|
||
|
#include <util/logger.hpp>
|
||
6 years ago
|
|
||
6 years ago
|
#include <cstdlib>
|
||
6 years ago
|
|
||
|
bool
|
||
|
bencode_read_integer(struct llarp_buffer_t* buffer, uint64_t* result)
|
||
|
{
|
||
|
size_t len;
|
||
|
if(*buffer->cur != 'i')
|
||
|
return false;
|
||
|
|
||
|
char numbuf[32];
|
||
|
|
||
|
buffer->cur++;
|
||
|
|
||
|
len =
|
||
|
llarp_buffer_read_until(buffer, 'e', (byte_t*)numbuf, sizeof(numbuf) - 1);
|
||
|
if(!len)
|
||
6 years ago
|
{
|
||
6 years ago
|
return false;
|
||
6 years ago
|
}
|
||
6 years ago
|
|
||
|
buffer->cur++;
|
||
|
|
||
6 years ago
|
numbuf[len] = '\0';
|
||
6 years ago
|
*result = std::strtoull(numbuf, nullptr, 10);
|
||
6 years ago
|
return true;
|
||
6 years ago
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_read_string(llarp_buffer_t* buffer, llarp_buffer_t* result)
|
||
|
{
|
||
|
char numbuf[10];
|
||
|
|
||
6 years ago
|
size_t len =
|
||
6 years ago
|
llarp_buffer_read_until(buffer, ':', (byte_t*)numbuf, sizeof(numbuf) - 1);
|
||
|
if(!len)
|
||
|
return false;
|
||
|
|
||
6 years ago
|
numbuf[len] = '\0';
|
||
|
const int num = atoi(numbuf);
|
||
6 years ago
|
if(num < 0)
|
||
6 years ago
|
{
|
||
6 years ago
|
return false;
|
||
6 years ago
|
}
|
||
6 years ago
|
|
||
6 years ago
|
const size_t slen = num;
|
||
6 years ago
|
|
||
|
buffer->cur++;
|
||
|
|
||
|
len = llarp_buffer_size_left(*buffer);
|
||
|
if(len < slen)
|
||
6 years ago
|
{
|
||
6 years ago
|
return false;
|
||
6 years ago
|
}
|
||
6 years ago
|
|
||
|
result->base = buffer->cur;
|
||
|
result->cur = buffer->cur;
|
||
|
result->sz = slen;
|
||
|
buffer->cur += slen;
|
||
|
return true;
|
||
|
}
|
||
|
|
||
6 years ago
|
bool
|
||
|
bencode_write_bytestring(llarp_buffer_t* buff, const void* data, size_t sz)
|
||
|
{
|
||
|
if(!llarp_buffer_writef(buff, "%zu:", sz))
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
return llarp_buffer_write(buff, data, sz);
|
||
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_write_uint64(llarp_buffer_t* buff, uint64_t i)
|
||
|
{
|
||
|
return llarp_buffer_writef(buff, "i%llu", i)
|
||
|
&& llarp_buffer_write(buff, "e", 1);
|
||
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_write_version_entry(llarp_buffer_t* buff)
|
||
|
{
|
||
|
return llarp_buffer_writef(buff, "1:vi%de", LLARP_PROTO_VERSION);
|
||
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_start_list(llarp_buffer_t* buff)
|
||
|
{
|
||
|
return llarp_buffer_write(buff, "l", 1);
|
||
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_start_dict(llarp_buffer_t* buff)
|
||
|
{
|
||
|
return llarp_buffer_write(buff, "d", 1);
|
||
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_end(llarp_buffer_t* buff)
|
||
|
{
|
||
|
return llarp_buffer_write(buff, "e", 1);
|
||
|
}
|
||
|
|
||
6 years ago
|
bool
|
||
|
bencode_read_dict(llarp_buffer_t* buff, struct dict_reader* r)
|
||
|
{
|
||
|
llarp_buffer_t strbuf; // temporary buffer for current element
|
||
|
r->buffer = buff; // set up dict_reader
|
||
|
if(*r->buffer->cur != 'd') // ensure is a dictionary
|
||
|
return false;
|
||
|
r->buffer->cur++;
|
||
|
while(llarp_buffer_size_left(*r->buffer) && *r->buffer->cur != 'e')
|
||
|
{
|
||
|
if(bencode_read_string(r->buffer, &strbuf))
|
||
|
{
|
||
|
if(!r->on_key(r, &strbuf)) // check for early abort
|
||
|
return false;
|
||
|
}
|
||
6 years ago
|
else
|
||
|
return false;
|
||
6 years ago
|
}
|
||
|
|
||
6 years ago
|
if(*r->buffer->cur != 'e')
|
||
|
{
|
||
|
llarp::LogWarn("reading dict not ending on 'e'");
|
||
|
// make sure we're at dictionary end
|
||
6 years ago
|
return false;
|
||
6 years ago
|
}
|
||
6 years ago
|
r->buffer->cur++;
|
||
6 years ago
|
return r->on_key(r, nullptr);
|
||
6 years ago
|
}
|
||
|
|
||
|
bool
|
||
|
bencode_read_list(llarp_buffer_t* buff, struct list_reader* r)
|
||
|
{
|
||
|
r->buffer = buff;
|
||
|
if(*r->buffer->cur != 'l') // ensure is a list
|
||
6 years ago
|
{
|
||
|
llarp::LogWarn("bencode::bencode_read_list - expecting list got ",
|
||
|
*r->buffer->cur);
|
||
6 years ago
|
return false;
|
||
6 years ago
|
}
|
||
6 years ago
|
|
||
|
r->buffer->cur++;
|
||
|
while(llarp_buffer_size_left(*r->buffer) && *r->buffer->cur != 'e')
|
||
|
{
|
||
|
if(!r->on_item(r, true)) // check for early abort
|
||
|
return false;
|
||
|
}
|
||
|
if(*r->buffer->cur != 'e') // make sure we're at a list end
|
||
|
return false;
|
||
|
r->buffer->cur++;
|
||
|
return r->on_item(r, false);
|
||
|
}
|