/*
 * 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;
}