mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-03 23:15:52 +00:00
7caa87862e
All #ifndef guards on headers have been removed, I think, in favor of #pragma once Headers are now included as `#include "filename"` if the included file resides in the same directory as the file including it, or any subdirectory therein. Otherwise they are included as `#include <project/top/dir/relative/path/filename>` The above does not include system/os headers.
153 lines
2.8 KiB
C++
153 lines
2.8 KiB
C++
#include "buffer.hpp"
|
|
#include "endian.hpp"
|
|
|
|
#include <cstdarg>
|
|
#include <cstdio>
|
|
|
|
size_t
|
|
llarp_buffer_t::size_left() const
|
|
{
|
|
size_t diff = cur - base;
|
|
if (diff > sz)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
return sz - diff;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::writef(const char* fmt, ...)
|
|
{
|
|
int written;
|
|
size_t toWrite = size_left();
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
written = vsnprintf(reinterpret_cast<char*>(cur), toWrite, fmt, args);
|
|
va_end(args);
|
|
if (written <= 0)
|
|
return false;
|
|
if (toWrite < static_cast<size_t>(written))
|
|
return false;
|
|
cur += written;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::put_uint16(uint16_t i)
|
|
{
|
|
if (size_left() < sizeof(uint16_t))
|
|
return false;
|
|
htobe16buf(cur, i);
|
|
cur += sizeof(uint16_t);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::put_uint64(uint64_t i)
|
|
{
|
|
if (size_left() < sizeof(uint64_t))
|
|
return false;
|
|
htobe64buf(cur, i);
|
|
cur += sizeof(uint64_t);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::put_uint32(uint32_t i)
|
|
{
|
|
if (size_left() < sizeof(uint32_t))
|
|
return false;
|
|
htobe32buf(cur, i);
|
|
cur += sizeof(uint32_t);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::read_uint16(uint16_t& i)
|
|
{
|
|
if (size_left() < sizeof(uint16_t))
|
|
return false;
|
|
i = bufbe16toh(cur);
|
|
cur += sizeof(uint16_t);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::read_uint32(uint32_t& i)
|
|
{
|
|
if (size_left() < sizeof(uint32_t))
|
|
return false;
|
|
i = bufbe32toh(cur);
|
|
cur += sizeof(uint32_t);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_t::read_uint64(uint64_t& i)
|
|
{
|
|
if (size_left() < sizeof(uint64_t))
|
|
return false;
|
|
i = bufbe64toh(cur);
|
|
cur += sizeof(uint64_t);
|
|
return true;
|
|
}
|
|
|
|
size_t
|
|
llarp_buffer_t::read_until(char c_delim, byte_t* result, size_t resultsize)
|
|
{
|
|
const auto delim = static_cast<byte_t>(c_delim);
|
|
size_t read = 0;
|
|
|
|
// do the bound check first, to avoid over running
|
|
while ((cur != base + sz) && *cur != delim && resultsize)
|
|
{
|
|
*result = *cur;
|
|
cur++;
|
|
result++;
|
|
resultsize--;
|
|
read++;
|
|
}
|
|
|
|
if (size_left())
|
|
return read;
|
|
|
|
return 0;
|
|
}
|
|
|
|
bool
|
|
operator==(const llarp_buffer_t& buff, const char* c_str)
|
|
{
|
|
const auto* str = reinterpret_cast<const byte_t*>(c_str);
|
|
ManagedBuffer copy{buff};
|
|
while (*str && copy.underlying.cur != (copy.underlying.base + copy.underlying.sz))
|
|
{
|
|
if (*copy.underlying.cur != *str)
|
|
return false;
|
|
copy.underlying.cur++;
|
|
str++;
|
|
}
|
|
return *str == 0;
|
|
}
|
|
|
|
namespace llarp
|
|
{
|
|
OwnedBuffer
|
|
OwnedBuffer::copy_from(const llarp_buffer_t& b)
|
|
{
|
|
auto buf = std::make_unique<byte_t[]>(b.sz);
|
|
std::copy(b.begin(), b.end(), buf.get());
|
|
return {std::move(buf), b.sz};
|
|
}
|
|
|
|
OwnedBuffer
|
|
OwnedBuffer::copy_used(const llarp_buffer_t& b)
|
|
{
|
|
const size_t sz = b.cur - b.base;
|
|
auto buf = std::make_unique<byte_t[]>(sz);
|
|
std::copy(b.base, b.cur, buf.get());
|
|
return {std::move(buf), sz};
|
|
}
|
|
|
|
} // namespace llarp
|