#include <stddef.h>
#include <unistd.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include "util.h"
#include "endiantools.h"
#include "buf.h"
sparkey_returncode buf_init(sparkey_buf *buf, ptrdiff_t size) {
buf->start = malloc(size);
if (buf->start == NULL) {
return SPARKEY_INTERNAL_ERROR;
}
buf->cur = buf->start;
buf->end = buf->start + size;
return SPARKEY_SUCCESS;
}
void buf_close(sparkey_buf *buf) {
free(buf->start);
buf->start = NULL;
buf->cur = NULL;
buf->end = NULL;
}
uint64_t buf_size(sparkey_buf *buf) {
return buf->end - buf->start;
}
uint64_t buf_remaining(sparkey_buf *buf) {
return buf->end - buf->cur;
}
uint64_t buf_used(sparkey_buf *buf) {
return buf->cur - buf->start;
}
sparkey_returncode buf_flushfile(sparkey_buf *buf, int fd) {
RETHROW(write_full(fd, buf->start, buf_used(buf)));
buf->cur = buf->start;
return SPARKEY_SUCCESS;
}
sparkey_returncode buf_add(sparkey_buf *buf, int fd, const uint8_t *data, ptrdiff_t len) {
while (1) {
ptrdiff_t remaining = buf_remaining(buf);
if (remaining >= len) {
memcpy(buf->cur, data, len);
buf->cur += len;
return SPARKEY_SUCCESS;
} else {
memcpy(buf->cur, data, remaining);
buf->cur += remaining;
data += remaining;
len -= remaining;
RETHROW(buf_flushfile(buf, fd));
}
}
return SPARKEY_SUCCESS;
}