neatvnc/src/vec.c

/*
 * Copyright (c) 2019 Andri Yngvason
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */

#include "vec.h"
#include "likely.h"

#include <stdlib.h>
#include <string.h>

int vec_init(struct vec* vec, size_t cap)
{
	memset(vec, 0, sizeof(*vec));
	return vec_reserve(vec, cap);
}

void vec_destroy(struct vec* vec)
{
	free(vec->data);
}

int vec_reserve(struct vec* vec, size_t size)
{
	if (likely(size <= vec->cap))
		return 0;

	void* data = realloc(vec->data, size);
	if (unlikely(!data))
		return -1;

	vec->cap = size;
	vec->data = data;

	return 0;
}

static int vec__grow(struct vec* vec, size_t size)
{
	if (likely(vec->len + size < vec->cap))
		return 0;
	
	return vec_reserve(vec, 2 * (vec->len + size));
}

int vec_assign(struct vec* vec, const void* data, size_t size)
{
	vec->len = 0;

	if (unlikely(vec_reserve(vec, size) < 0))
		return -1;

	vec->len = size;
	memcpy(vec->data, data, size);

	return 0;
}

int vec_append(struct vec* vec, const void* data, size_t size)
{
	if (unlikely(vec__grow(vec, size) < 0))
		return -1;

	char* p = vec->data;
	memcpy(&p[vec->len], data, size);
	vec->len += size;

	return 0;
}

void* vec_append_zero(struct vec* vec, size_t size)
{
	if (unlikely(vec__grow(vec, size) < 0))
		return NULL;

	char* p = vec->data;
	void* r = &p[vec->len];
	memset(r, 0, size);
	vec->len += size;

	return r;
}

void vec_bzero(struct vec* vec)
{
	memset(vec->data, 0, vec->cap);
}