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.
101 lines
1.8 KiB
C++
101 lines
1.8 KiB
C++
#include <llarp/buffer.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
|
|
extern "C" {
|
|
|
|
size_t
|
|
llarp_buffer_size_left(llarp_buffer_t* buff)
|
|
{
|
|
size_t diff = buff->cur - buff->base;
|
|
if(diff > buff->sz)
|
|
return 0;
|
|
else
|
|
return buff->sz - diff;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_writef(llarp_buffer_t* buff, const char* fmt, ...)
|
|
{
|
|
int written;
|
|
ssize_t sz = llarp_buffer_size_left(buff);
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
written = vsnprintf((char*)buff->cur, sz, fmt, args);
|
|
va_end(args);
|
|
if(written <= 0)
|
|
return false;
|
|
if(sz < written)
|
|
return false;
|
|
buff->cur += written;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_write(llarp_buffer_t* buff, const void* data, size_t sz)
|
|
{
|
|
size_t left = llarp_buffer_size_left(buff);
|
|
if(left >= sz)
|
|
{
|
|
memcpy(buff->cur, data, sz);
|
|
buff->cur += sz;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_readfile(llarp_buffer_t* buff, FILE* f, llarp_alloc* mem)
|
|
{
|
|
ssize_t len;
|
|
fseek(f, 0, SEEK_END);
|
|
len = ftell(f);
|
|
rewind(f);
|
|
if(len > 0)
|
|
{
|
|
buff->base = static_cast< uint8_t* >(mem->alloc(mem, len, 8));
|
|
buff->cur = buff->base;
|
|
buff->sz = len;
|
|
ssize_t sz = fread(buff->base, len, 1, f);
|
|
rewind(f);
|
|
return sz == len;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
size_t
|
|
llarp_buffer_read_until(llarp_buffer_t* buff, char delim, byte_t* result,
|
|
size_t resultsize)
|
|
{
|
|
size_t read = 0;
|
|
|
|
while(*buff->cur != delim && resultsize
|
|
&& (buff->cur != buff->base + buff->sz))
|
|
{
|
|
*result = *buff->cur;
|
|
buff->cur++;
|
|
result++;
|
|
resultsize--;
|
|
read++;
|
|
}
|
|
|
|
if(llarp_buffer_size_left(buff))
|
|
return read;
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
bool
|
|
llarp_buffer_eq(llarp_buffer_t buf, const char* str)
|
|
{
|
|
while(*str && buf.cur != (buf.base + buf.sz))
|
|
{
|
|
if(*buf.cur != *str)
|
|
return false;
|
|
buf.cur++;
|
|
str++;
|
|
}
|
|
return *str == 0;
|
|
}
|
|
}
|