buffer.c

Go to the documentation of this file.

/*
 *  OpenVPN -- An application to securely tunnel IP networks
 *             over a single UDP port, with support for SSL/TLS-based
 *             session authentication and key exchange,
 *             packet encryption, packet authentication, and
 *             packet compression.
 *
 *  Copyright (C) 2002-2026 OpenVPN Inc <sales@openvpn.net>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2
 *  as published by the Free Software Foundation.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, see <https://www.gnu.org/licenses/>.
 */
 
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
 
#include "syshead.h"
 
#include "common.h"
#include "buffer.h"
#include "error.h"
#include "mtu.h"
#include "misc.h"
 
#include "memdbg.h"
 
#include <wchar.h>
 
size_t
array_mult_safe(const size_t m1, const size_t m2, const size_t extra)
{
    const size_t limit = ALLOC_SIZE_MAX;
    unsigned long long res =
        (unsigned long long)m1 * (unsigned long long)m2 + (unsigned long long)extra;
    if (unlikely(m1 > limit) || unlikely(m2 > limit) || unlikely(extra > limit)
        || unlikely(res > limit))
    {
        msg(M_FATAL, "attempted allocation of excessively large array");
    }
    return (size_t)res;
}
 
void
buf_size_error(const size_t size)
{
    msg(M_FATAL, "fatal buffer size error, size=%zu", size);
}
 
struct buffer
alloc_buf(size_t size)
{
    struct buffer buf;
    CLEAR(buf);
 
    if (!buf_size_valid(size))
    {
        buf_size_error(size);
    }
    buf.data = calloc(1, size);
    buf.capacity = (int)size;
    check_malloc_return(buf.data);
 
    return buf;
}
 
struct buffer
alloc_buf_gc(size_t size, struct gc_arena *gc)
{
    struct buffer buf;
    CLEAR(buf);
 
    if (!buf_size_valid(size))
    {
        buf_size_error(size);
    }
    buf.data = gc_malloc(size, false, gc);
    buf.capacity = (int)size;
    if (size)
    {
        *buf.data = 0;
    }
    return buf;
}
 
struct buffer
#ifdef BUF_INIT_TRACKING
clone_buf_debug(const struct buffer *buf, const char *file, int line)
#else
clone_buf(const struct buffer *buf)
#endif
{
    struct buffer ret;
    ret.data = malloc(buf->capacity);
    ret.capacity = buf->capacity;
    check_malloc_return(ret.data);
    ret.offset = buf->offset;
    ret.len = buf->len;
#ifdef BUF_INIT_TRACKING
    ret.debug_file = buf->debug_file;
    ret.debug_line = buf->debug_line;
#endif
    memcpy(BPTR(&ret), CBPTR(buf), BLENZ(buf));
    return ret;
}
 
#ifdef BUF_INIT_TRACKING
 
bool
buf_init_debug(struct buffer *buf, int offset, const char *file, int line)
{
    buf->debug_file = file;
    buf->debug_line = line;
    return buf_init_dowork(buf, offset);
}
 
#ifdef VERIFY_ALIGNMENT
static inline int
buf_debug_line(const struct buffer *buf)
{
    return buf->debug_line;
}
 
static const char *
buf_debug_file(const struct buffer *buf)
{
    return buf->debug_file;
}
#endif
 
#else /* ifdef BUF_INIT_TRACKING */
 
#define buf_debug_line(buf) 0
#define buf_debug_file(buf) "[UNDEF]"
 
#endif /* ifdef BUF_INIT_TRACKING */
 
void
buf_clear(struct buffer *buf)
{
    if (buf->capacity > 0)
    {
        secure_memzero(buf->data, buf->capacity);
    }
    buf->len = 0;
    buf->offset = 0;
}
 
bool
buf_assign(struct buffer *dest, const struct buffer *src)
{
    if (!buf_init(dest, src->offset))
    {
        return false;
    }
    return buf_write(dest, CBPTR(src), BLENZ(src));
}
 
void
free_buf(struct buffer *buf)
{
    free(buf->data);
    CLEAR(*buf);
}
 
static void
free_buf_gc(struct buffer *buf, struct gc_arena *gc)
{
    if (gc)
    {
        struct gc_entry **e = &gc->list;
 
        while (*e)
        {
            /* check if this object is the one we want to delete */
            if ((uintptr_t)*e + sizeof(struct gc_entry) == (uintptr_t)buf->data)
            {
                struct gc_entry *to_delete = *e;
 
                /* remove element from linked list and free it */
                *e = (*e)->next;
                free(to_delete);
 
                break;
            }
 
            e = &(*e)->next;
        }
    }
 
    CLEAR(*buf);
}
 
/*
 * Return a buffer for write that is a subset of another buffer
 */
struct buffer
buf_sub(struct buffer *buf, int size, bool prepend)
{
    struct buffer ret;
    uint8_t *data;
 
    CLEAR(ret);
    data = prepend ? buf_prepend(buf, size) : buf_write_alloc(buf, size);
    if (data)
    {
        ret.capacity = size;
        ret.data = data;
    }
    return ret;
}
 
/*
 * printf append to a buffer with overflow check
 */
bool
buf_printf(struct buffer *buf, const char *format, ...)
{
    bool ret = false;
    if (buf_defined(buf))
    {
        char *ptr = (char *)BEND(buf);
        int cap = buf_forward_capacity(buf);
 
        if (cap > 1) /* can write more than \0 */
        {
            int stat;
            va_list arglist;
            va_start(arglist, format);
            stat = vsnprintf(ptr, cap, format, arglist);
            va_end(arglist);
            buf->len += (int)strnlen(ptr, cap);
            if (stat >= 0 && stat < cap)
            {
                ret = true;
            }
        }
    }
    return ret;
}
 
void
buf_puts(struct buffer *buf, const char *str)
{
    char *ptr = (char *)BEND(buf);
    int cap = buf_forward_capacity(buf);
    if (cap > 1) /* can write more than \0 */
    {
        int str_len = (int)strlcpy(ptr, str, cap);
        buf->len += (str_len < cap) ? str_len : cap - 1;
    }
}
 
/*
 * write a string to the end of a buffer that was
 * truncated by buf_printf
 */
void
buf_catrunc(struct buffer *buf, const char *str)
{
    if (buf_forward_capacity(buf) <= 1)
    {
        size_t len = strlen(str) + 1;
        if (buf_size_valid(len) && (int)len < buf_forward_capacity_total(buf))
        {
            memcpy(buf->data + buf->capacity - len, str, len);
        }
    }
}
 
bool
buffer_write_file(const char *filename, const struct buffer *buf)
{
    bool ret = false;
    int fd = platform_open(filename, O_CREAT | O_TRUNC | O_WRONLY, S_IRUSR | S_IWUSR);
    if (fd == -1)
    {
        msg(M_ERRNO, "Cannot open file '%s' for write", filename);
        return false;
    }
 
    const ssize_t size = write(fd, CBPTR(buf), (unsigned int)BLEN(buf));
    if (size != BLEN(buf))
    {
        msg(M_ERRNO, "Write error on file '%s'", filename);
        goto cleanup;
    }
 
    ret = true;
cleanup:
    if (close(fd) < 0)
    {
        msg(M_ERRNO, "Close error on file %s", filename);
        ret = false;
    }
    return ret;
}
 
/*
 * Garbage collection
 */
 
void *
gc_malloc(size_t size, bool clear, struct gc_arena *a)
{
    void *ret;
    if (a)
    {
        struct gc_entry *e;
        e = (struct gc_entry *)malloc(size + sizeof(struct gc_entry));
        check_malloc_return(e);
        ret = (char *)e + sizeof(struct gc_entry);
        e->next = a->list;
        a->list = e;
    }
    else
    {
        ret = malloc(size);
        check_malloc_return(ret);
    }
#ifndef ZERO_BUFFER_ON_ALLOC
    if (clear)
#endif
        memset(ret, 0, size);
    return ret;
}
 
void *
gc_realloc(void *ptr, size_t size, struct gc_arena *a)
{
    void *ret = realloc(ptr, size);
    check_malloc_return(ret);
    if (a)
    {
        if (ptr && ptr != ret)
        {
            /* find the old entry and modify it if realloc changed
             * the pointer */
            struct gc_entry_special *e = NULL;
            for (e = a->list_special; e != NULL; e = e->next)
            {
                if (e->addr == ptr)
                {
                    break;
                }
            }
            ASSERT(e);
            ASSERT(e->addr == ptr);
            e->addr = ret;
        }
        else if (!ptr)
        {
            /* sets e->addr to newptr */
            gc_addspecial(ret, free, a);
        }
    }
 
    return ret;
}
 
void
x_gc_free(struct gc_arena *a)
{
    struct gc_entry *e;
    e = a->list;
    a->list = NULL;
 
    while (e != NULL)
    {
        struct gc_entry *next = e->next;
        free(e);
        e = next;
    }
}
 
/*
 * Functions to handle special objects in gc_entries
 */
 
void
x_gc_freespecial(struct gc_arena *a)
{
    struct gc_entry_special *e;
    e = a->list_special;
    a->list_special = NULL;
 
    while (e != NULL)
    {
        struct gc_entry_special *next = e->next;
        e->free_fnc(e->addr);
        free(e);
        e = next;
    }
}
 
void
gc_addspecial(void *addr, void (*free_function)(void *), struct gc_arena *a)
{
    ASSERT(a);
    struct gc_entry_special *e;
    e = (struct gc_entry_special *)malloc(sizeof(struct gc_entry_special));
    check_malloc_return(e);
    e->free_fnc = free_function;
    e->addr = addr;
 
    e->next = a->list_special;
    a->list_special = e;
}
 
 
/*
 * Transfer src arena to dest, resetting src to an empty arena.
 */
void
gc_transfer(struct gc_arena *dest, struct gc_arena *src)
{
    if (dest && src)
    {
        struct gc_entry *e = src->list;
        if (e)
        {
            while (e->next != NULL)
            {
                e = e->next;
            }
            e->next = dest->list;
            dest->list = src->list;
            src->list = NULL;
        }
    }
}
 
/*
 * Hex dump -- Output a binary buffer to a hex string and return it.
 */
 
char *
format_hex_ex(const uint8_t *data, size_t size, size_t maxoutput, unsigned int space_break_flags,
              const char *separator, struct gc_arena *gc)
{
    const size_t bytes_per_hexblock = space_break_flags & FHE_SPACE_BREAK_MASK;
    const size_t separator_len = separator ? strlen(separator) : 0;
    const size_t out_len = maxoutput > 0
                               ? maxoutput
                               : ((size * 2) + ((size / bytes_per_hexblock) * separator_len) + 2);
 
    struct buffer out = alloc_buf_gc(out_len, gc);
    for (size_t i = 0; i < size; ++i)
    {
        if (separator && i && !(i % bytes_per_hexblock))
        {
            buf_printf(&out, "%s", separator);
        }
        if (space_break_flags & FHE_CAPS)
        {
            buf_printf(&out, "%02X", data[i]);
        }
        else
        {
            buf_printf(&out, "%02x", data[i]);
        }
    }
    buf_catrunc(&out, "[more...]");
    return (char *)out.data;
}
 
/*
 * remove specific trailing character
 */
 
void
buf_rmtail(struct buffer *buf, uint8_t remove)
{
    uint8_t *cp = BLAST(buf);
    if (cp && *cp == remove)
    {
        *cp = '\0';
        --buf->len;
    }
}
 
/*
 * force a null termination even it requires
 * truncation of the last char.
 */
void
buf_null_terminate(struct buffer *buf)
{
    const uint8_t *last = BLAST(buf);
    if (last && *last == '\0') /* already terminated? */
    {
        return;
    }
 
    if (!buf_safe(buf, 1)) /* make space for trailing null */
    {
        buf_inc_len(buf, -1);
    }
 
    buf_write_u8(buf, 0);
}
 
/*
 * Remove trailing \r and \n chars and ensure
 * null termination.
 */
void
buf_chomp(struct buffer *buf)
{
    while (true)
    {
        const uint8_t *last = BLAST(buf);
        if (!last)
        {
            break;
        }
        if (char_class(*last, CC_CRLF | CC_NULL))
        {
            if (!buf_inc_len(buf, -1))
            {
                break;
            }
        }
        else
        {
            break;
        }
    }
    buf_null_terminate(buf);
}
 
const char *
skip_leading_whitespace(const char *str)
{
    while (*str)
    {
        const char c = *str;
        if (!(c == ' ' || c == '\t'))
        {
            break;
        }
        ++str;
    }
    return str;
}
 
#ifdef _WIN32
/*
 * like buf_null_terminate, but operate on strings
 */
void
string_null_terminate(char *str, int len, int capacity)
{
    ASSERT(len >= 0 && len <= capacity && capacity > 0);
    if (len < capacity)
    {
        *(str + len) = '\0';
    }
    else if (len == capacity)
    {
        *(str + len - 1) = '\0';
    }
}
#endif
 
/*
 * Remove trailing \r and \n chars.
 */
void
chomp(char *str)
{
    rm_trailing_chars(str, "\r\n");
}
 
/*
 * Remove trailing chars
 */
void
rm_trailing_chars(char *str, const char *what_to_delete)
{
    size_t len = strlen(str);
    while (len > 0 && strchr(what_to_delete, str[len - 1]) != NULL)
    {
        str[--len] = '\0';
    }
}
 
/*
 * Allocate a string
 */
char *
string_alloc(const char *str, struct gc_arena *gc)
{
    if (str)
    {
        const size_t n = strlen(str) + 1;
        char *ret;
 
        if (gc)
        {
            ret = gc_malloc(n, false, gc);
        }
        else
        {
            /* If there are no garbage collector available, it's expected
             * that the caller cleans up afterwards.  This is coherent with the
             * earlier behaviour when gc_malloc() would be called with gc == NULL
             */
            ret = calloc(1, n);
            check_malloc_return(ret);
        }
        memcpy(ret, str, n);
        return ret;
    }
    else
    {
        return NULL;
    }
}
 
/*
 * Erase all characters in a string
 */
void
string_clear(char *str)
{
    if (str)
    {
        secure_memzero(str, strlen(str));
    }
}
 
/*
 * Return the length of a string array
 */
int
string_array_len(const char **array)
{
    int i = 0;
    if (array)
    {
        for (; *array; ++array)
        {
            ++i;
        }
    }
    return i;
}
 
char *
print_argv(const char **p, struct gc_arena *gc, const unsigned int flags)
{
    struct buffer out = alloc_buf_gc(256, gc);
    int i = 0;
    for (; *p; ++p)
    {
        const char *cp = *p;
        if (i)
        {
            buf_printf(&out, " ");
        }
        if (flags & PA_BRACKET)
        {
            buf_printf(&out, "[%s]", cp);
        }
        else
        {
            buf_printf(&out, "%s", cp);
        }
        ++i;
    }
    return BSTR(&out);
}
 
/*
 * Allocate a string inside a buffer
 */
struct buffer
string_alloc_buf(const char *str, struct gc_arena *gc)
{
    struct buffer buf;
 
    ASSERT(str);
 
    const char *copy = string_alloc(str, gc);
    buf_set_read(&buf, (const uint8_t *)copy, strlen(str) + 1);
 
    if (buf.len > 0) /* Don't count trailing '\0' as part of length */
    {
        --buf.len;
    }
 
    return buf;
}
 
/*
 * String comparison
 */
 
bool
buf_string_match_head_str(const struct buffer *src, const char *match)
{
    const size_t size = strlen(match);
    if (!buf_size_valid(size) || (int)size > src->len)
    {
        return false;
    }
    return memcmp(CBPTR(src), match, size) == 0;
}
 
bool
buf_string_compare_advance(struct buffer *src, const char *match)
{
    if (buf_string_match_head_str(src, match))
    {
        buf_advance(src, strlen(match));
        return true;
    }
    else
    {
        return false;
    }
}
 
int
buf_substring_len(const struct buffer *buf, int delim)
{
    int i = 0;
    struct buffer tmp = *buf;
    int c;
 
    while ((c = buf_read_u8(&tmp)) >= 0)
    {
        ++i;
        if (c == delim)
        {
            return i;
        }
    }
    return -1;
}
 
/*
 * String parsing
 */
 
bool
buf_parse(struct buffer *buf, const int delim, char *line, const int size)
{
    bool eol = false;
    int n = 0;
    int c;
 
    ASSERT(size > 0);
 
    do
    {
        c = buf_peek_u8(buf);
        if (c < 0)
        {
            eol = true;
            line[n] = 0;
            break;
        }
        if (c == delim)
        {
            buf_advance(buf, 1);
            line[n] = 0;
            break;
        }
        if (n >= (size - 1))
        {
            break;
        }
        buf_advance(buf, 1);
        line[n++] = (char)c;
    } while (c);
 
    line[size - 1] = '\0';
    return !(eol && line[0] == '\0');
}
 
/*
 * Print a string which might be NULL
 */
const char *
np(const char *str)
{
    if (str)
    {
        return str;
    }
    else
    {
        return "[NULL]";
    }
}
 
/*
 * Classify and mutate strings based on character types.
 */
 
/* Note 1: This functions depends on getting an unsigned
   char. Both the is*() functions and our own checks expect it
   this way.
   Note 2: For CC_PRINT we just accept everything >= 32, so
   if we ingest non-ASCII UTF-8 we will classify it as
   printable since it will be >= 128. Other encodings are
   not officially supported.
*/
bool
char_class(const unsigned char c, const unsigned int flags)
{
    if (!flags)
    {
        return false;
    }
    if (flags & CC_ANY)
    {
        return true;
    }
 
    if ((flags & CC_NULL) && c == '\0')
    {
        return true;
    }
 
    if ((flags & CC_ALNUM) && isalnum(c))
    {
        return true;
    }
    if ((flags & CC_ALPHA) && isalpha(c))
    {
        return true;
    }
    if ((flags & CC_ASCII) && isascii(c))
    {
        return true;
    }
    if ((flags & CC_CNTRL) && iscntrl(c))
    {
        return true;
    }
    if ((flags & CC_DIGIT) && isdigit(c))
    {
        return true;
    }
    /* allow ascii non-control and UTF-8, consider DEL to be a control */
    if ((flags & CC_PRINT) && (c >= 32 && c != 127))
    {
        return true;
    }
    if ((flags & CC_PUNCT) && ispunct(c))
    {
        return true;
    }
    if ((flags & CC_SPACE) && isspace(c))
    {
        return true;
    }
    if ((flags & CC_XDIGIT) && isxdigit(c))
    {
        return true;
    }
 
    if ((flags & CC_BLANK) && (c == ' ' || c == '\t'))
    {
        return true;
    }
    if ((flags & CC_NEWLINE) && c == '\n')
    {
        return true;
    }
    if ((flags & CC_CR) && c == '\r')
    {
        return true;
    }
 
    if ((flags & CC_BACKSLASH) && c == '\\')
    {
        return true;
    }
    if ((flags & CC_UNDERBAR) && c == '_')
    {
        return true;
    }
    if ((flags & CC_DASH) && c == '-')
    {
        return true;
    }
    if ((flags & CC_DOT) && c == '.')
    {
        return true;
    }
    if ((flags & CC_COMMA) && c == ',')
    {
        return true;
    }
    if ((flags & CC_COLON) && c == ':')
    {
        return true;
    }
    if ((flags & CC_SLASH) && c == '/')
    {
        return true;
    }
    if ((flags & CC_SINGLE_QUOTE) && c == '\'')
    {
        return true;
    }
    if ((flags & CC_DOUBLE_QUOTE) && c == '\"')
    {
        return true;
    }
    if ((flags & CC_REVERSE_QUOTE) && c == '`')
    {
        return true;
    }
    if ((flags & CC_PERCENT) && c == '%')
    {
        return true;
    }
    if ((flags & CC_EXCLAMATION) && c == '!')
    {
        return true;
    }
    if ((flags & CC_LESS_THAN) && c == '<')
    {
        return true;
    }
    if ((flags & CC_GREATER_THAN) && c == '>')
    {
        return true;
    }
    if ((flags & CC_PIPE) && c == '|')
    {
        return true;
    }
    if ((flags & CC_QUESTION_MARK) && c == '?')
    {
        return true;
    }
    if ((flags & CC_ASTERISK) && c == '*')
    {
        return true;
    }
 
    return false;
}
 
static inline bool
char_inc_exc(const char c, const unsigned int inclusive, const unsigned int exclusive)
{
    return char_class((unsigned char)c, inclusive)
           && !char_class((unsigned char)c, exclusive);
}
 
bool
string_class(const char *str, const unsigned int inclusive, const unsigned int exclusive)
{
    ASSERT(str);
    for (; *str; ++str)
    {
        if (!char_inc_exc(*str, inclusive, exclusive))
        {
            return false;
        }
    }
    return true;
}
 
/*
 * Modify string in place.
 * Guaranteed to not increase string length.
 */
bool
string_mod(char *str, const unsigned int inclusive, const unsigned int exclusive,
           const char replace)
{
    ASSERT(str);
 
    char *out = str;
    const char *in = out;
    bool ret = true;
 
    while (true)
    {
        char c = *in++;
        if (c)
        {
            if (!char_inc_exc(c, inclusive, exclusive))
            {
                c = replace;
                ret = false;
            }
            if (c)
            {
                *out++ = c;
            }
        }
        else
        {
            *out = '\0';
            break;
        }
    }
    return ret;
}
 
bool
string_check_buf(const struct buffer *buf, const unsigned int inclusive, const unsigned int exclusive)
{
    ASSERT(buf);
 
    for (int i = 0; i < BLEN(buf); i++)
    {
        char c = CBSTR(buf)[i];
 
        if (!char_inc_exc(c, inclusive, exclusive))
        {
            return false;
        }
    }
    return true;
}
 
const char *
string_mod_const(const char *str, const unsigned int inclusive, const unsigned int exclusive,
                 const char replace, struct gc_arena *gc)
{
    if (str)
    {
        char *buf = string_alloc(str, gc);
        string_mod(buf, inclusive, exclusive, replace);
        return buf;
    }
    else
    {
        return NULL;
    }
}
 
void
string_replace_leading(char *str, const char match, const char replace)
{
    ASSERT(match != '\0');
    while (*str)
    {
        if (*str == match)
        {
            *str = replace;
        }
        else
        {
            break;
        }
        ++str;
    }
}
 
bool
string_defined_equal(const char *s1, const char *s2)
{
    if (s1 && s2)
    {
        return !strcmp(s1, s2);
    }
    else
    {
        return false;
    }
}
 
bool
streqnull(const char *a, const char *b)
{
    if (a == NULL && b == NULL)
    {
        return true;
    }
    else if (a == NULL || b == NULL)
    {
        return false;
    }
    else
    {
        return streq(a, b);
    }
}
 
char *
string_substitute(const char *src, char from, char to, struct gc_arena *gc)
{
    char *ret = gc_malloc(strlen(src) + 1, true, gc);
    char *dest = ret;
 
    for (; *src; ++src)
    {
        *dest++ = (*src == from) ? to : *src;
    }
    *dest = '\0';
    return ret;
}
 
bool
checked_snprintf(char *str, size_t size, const char *format, ...)
{
    va_list arglist;
    va_start(arglist, format);
    ASSERT(size < INT_MAX);
    int len = vsnprintf(str, size, format, arglist);
    va_end(arglist);
    return (len >= 0 && len < (ssize_t)size);
}
 
#ifdef VERIFY_ALIGNMENT
void
valign4(const struct buffer *buf, const char *file, const int line)
{
    if (buf && buf->len)
    {
        msglvl_t msglevel = D_ALIGN_DEBUG;
        const uintptr_t u = (uintptr_t)CBPTR(buf);
 
        if (u & (PAYLOAD_ALIGN - 1))
        {
            msglevel = D_ALIGN_ERRORS;
        }
 
        msg(msglevel, "%sAlignment at %s/%d ptr=" ptr_format " OLC=%d/%d/%d I=%s/%d",
            (msglevel == D_ALIGN_ERRORS) ? "ERROR: " : "", file, line, (ptr_type)buf->data,
            buf->offset, buf->len, buf->capacity, buf_debug_file(buf), buf_debug_line(buf));
    }
}
#endif /* ifdef VERIFY_ALIGNMENT */
 
/*
 * struct buffer_list
 */
struct buffer_list *
buffer_list_new(void)
{
    struct buffer_list *ret;
    ALLOC_OBJ_CLEAR(ret, struct buffer_list);
    ret->size = 0;
    return ret;
}
 
void
buffer_list_free(struct buffer_list *ol)
{
    if (ol)
    {
        buffer_list_reset(ol);
        free(ol);
    }
}
 
bool
buffer_list_defined(const struct buffer_list *ol)
{
    return ol && ol->head != NULL && ol->size > 0;
}
 
void
buffer_list_reset(struct buffer_list *ol)
{
    struct buffer_entry *e = ol->head;
    while (e)
    {
        struct buffer_entry *next = e->next;
        free_buf(&e->buf);
        free(e);
        e = next;
    }
    ol->head = ol->tail = NULL;
    ol->size = 0;
}
 
void
buffer_list_push(struct buffer_list *ol, const char *str)
{
    if (str)
    {
        const size_t len = strlen((const char *)str);
        struct buffer_entry *e = buffer_list_push_data(ol, str, len + 1);
        if (e)
        {
            e->buf.len--; /* Don't count trailing '\0' as part of length */
        }
    }
}
 
struct buffer_entry *
buffer_list_push_data(struct buffer_list *ol, const void *data, size_t size)
{
    struct buffer_entry *e = NULL;
    if (data)
    {
        ALLOC_OBJ_CLEAR(e, struct buffer_entry);
 
        ++ol->size;
        if (ol->tail)
        {
            ASSERT(ol->head);
            ol->tail->next = e;
        }
        else
        {
            ASSERT(!ol->head);
            ol->head = e;
        }
        e->buf = alloc_buf(size);
        memcpy(e->buf.data, data, size);
        /* Note: size implicitly checked by alloc_buf */
        e->buf.len = (int)size;
        ol->tail = e;
    }
    return e;
}
 
struct buffer *
buffer_list_peek(struct buffer_list *ol)
{
    if (ol && ol->head)
    {
        return &ol->head->buf;
    }
    else
    {
        return NULL;
    }
}
 
void
buffer_list_aggregate_separator(struct buffer_list *bl, const size_t max_len, const char *sep)
{
    const size_t sep_len = strlen(sep);
    struct buffer_entry *more = bl->head;
    size_t size = 0;
    size_t count = 0;
    for (; more; ++count)
    {
        size_t extra_len = BLENZ(&more->buf) + sep_len;
        if (size + extra_len > max_len)
        {
            break;
        }
 
        size += extra_len;
        more = more->next;
    }
 
    if (count >= 2)
    {
        struct buffer_entry *f;
        ALLOC_OBJ_CLEAR(f, struct buffer_entry);
        f->buf = alloc_buf(size + 1); /* prevent 0-byte malloc */
 
        struct buffer_entry *e = bl->head;
        for (size_t i = 0; e && i < count; ++i)
        {
            struct buffer_entry *next = e->next;
            buf_copy(&f->buf, &e->buf);
            buf_write(&f->buf, sep, sep_len);
            free_buf(&e->buf);
            free(e);
            e = next;
        }
        bl->head = f;
        bl->size -= count - 1;
        f->next = more;
        if (!more)
        {
            bl->tail = f;
        }
    }
}
 
void
buffer_list_aggregate(struct buffer_list *bl, const size_t max)
{
    buffer_list_aggregate_separator(bl, max, "");
}
 
void
buffer_list_pop(struct buffer_list *ol)
{
    if (buffer_list_defined(ol))
    {
        struct buffer_entry *e = ol->head->next;
        free_buf(&ol->head->buf);
        free(ol->head);
        ol->head = e;
        --ol->size;
        if (!e)
        {
            ol->tail = NULL;
        }
    }
}
 
void
buffer_list_advance(struct buffer_list *ol, ssize_t n)
{
    if (ol->head)
    {
        struct buffer *buf = &ol->head->buf;
        ASSERT(buf_advance(buf, n));
        if (!BLEN(buf))
        {
            buffer_list_pop(ol);
        }
    }
}
 
struct buffer_list *
buffer_list_file(const char *fn, int max_line_len)
{
    FILE *fp = platform_fopen(fn, "r");
    struct buffer_list *bl = NULL;
 
    if (fp)
    {
        char *line = (char *)malloc(max_line_len);
        if (line)
        {
            bl = buffer_list_new();
            while (fgets(line, max_line_len, fp) != NULL)
            {
                buffer_list_push(bl, line);
            }
            free(line);
        }
        fclose(fp);
    }
    return bl;
}
 
struct buffer
buffer_read_from_file(const char *filename, struct gc_arena *gc)
{
    struct buffer ret = { 0 };
 
    platform_stat_t file_stat = { 0 };
    if (platform_stat(filename, &file_stat) < 0)
    {
        return ret;
    }
 
    FILE *fp = platform_fopen(filename, "r");
    if (!fp)
    {
        return ret;
    }
 
    const size_t size = file_stat.st_size;
    ret = alloc_buf_gc(size + 1, gc); /* space for trailing \0 */
    size_t read_size = fread(BPTR(&ret), 1, size, fp);
    if (read_size == 0)
    {
        free_buf_gc(&ret, gc);
        goto cleanup;
    }
    ASSERT(buf_inc_len(&ret, (int)read_size));
    buf_null_terminate(&ret);
 
cleanup:
    fclose(fp);
    return ret;
}
 
char *
buf_extract_field(struct buffer *buf, char sep, struct gc_arena *gc)
{
    if (!buf_valid(buf))
    {
        return NULL;
    }
 
    const uint8_t *seppos = memchr(CBPTR(buf), sep, buf_len(buf));
    if (!seppos)
    {
        return NULL;
    }
    size_t field_len = seppos - CBPTR(buf);
 
    char *field = gc_malloc(field_len + 1, false, gc);
 
    memcpy(field, CBPTR(buf), field_len);
    field[field_len] = 0;
 
    buf_advance(buf, field_len + 1);
    return field;
}