/*
 * Copyright (c) 2018 Yubico AB. All rights reserved.
 * Use of this source code is governed by a BSD-style
 * license that can be found in the LICENSE file.
 */

#include "fido.h"

fido_blob_t *
fido_blob_new(void)
{
	return calloc(1, sizeof(fido_blob_t));
}

void
fido_blob_reset(fido_blob_t *b)
{
	freezero(b->ptr, b->len);
	explicit_bzero(b, sizeof(*b));
}

int
fido_blob_set(fido_blob_t *b, const u_char *ptr, size_t len)
{
	fido_blob_reset(b);

	if (ptr == NULL || len == 0) {
		fido_log_debug("%s: ptr=%p, len=%zu", __func__,
		    (const void *)ptr, len);
		return -1;
	}

	if ((b->ptr = malloc(len)) == NULL) {
		fido_log_debug("%s: malloc", __func__);
		return -1;
	}

	memcpy(b->ptr, ptr, len);
	b->len = len;

	return 0;
}

int
fido_blob_append(fido_blob_t *b, const u_char *ptr, size_t len)
{
	u_char *tmp;

	if (ptr == NULL || len == 0) {
		fido_log_debug("%s: ptr=%p, len=%zu", __func__,
		    (const void *)ptr, len);
		return -1;
	}
	if (SIZE_MAX - b->len < len) {
		fido_log_debug("%s: overflow", __func__);
		return -1;
	}
	if ((tmp = realloc(b->ptr, b->len + len)) == NULL) {
		fido_log_debug("%s: realloc", __func__);
		return -1;
	}
	b->ptr = tmp;
	memcpy(&b->ptr[b->len], ptr, len);
	b->len += len;

	return 0;
}

void
fido_blob_free(fido_blob_t **bp)
{
	fido_blob_t *b;

	if (bp == NULL || (b = *bp) == NULL)
		return;

	fido_blob_reset(b);
	free(b);
	*bp = NULL;
}

void
fido_free_blob_array(fido_blob_array_t *array)
{
	if (array->ptr == NULL)
		return;

	for (size_t i = 0; i < array->len; i++) {
		fido_blob_t *b = &array->ptr[i];
		freezero(b->ptr, b->len);
		b->ptr = NULL;
	}

	free(array->ptr);
	array->ptr = NULL;
	array->len = 0;
}

cbor_item_t *
fido_blob_encode(const fido_blob_t *b)
{
	if (b == NULL || b->ptr == NULL)
		return NULL;

	return cbor_build_bytestring(b->ptr, b->len);
}

int
fido_blob_decode(const cbor_item_t *item, fido_blob_t *b)
{
	return cbor_bytestring_copy(item, &b->ptr, &b->len);
}

int
fido_blob_is_empty(const fido_blob_t *b)
{
	return b->ptr == NULL || b->len == 0;
}

int
fido_blob_serialise(fido_blob_t *b, const cbor_item_t *item)
{
	size_t alloc;

	if (!fido_blob_is_empty(b))
		return -1;
	if ((b->len = cbor_serialize_alloc(item, &b->ptr, &alloc)) == 0) {
		b->ptr = NULL;
		return -1;
	}

	return 0;
}