vaseboot/VasEBoot-core/lib/tss2/buffer.c

148 lines
3.6 KiB
C

/*
* VAS_EBOOT -- GRand Unified Bootloader
* Copyright (C) 2022 Microsoft Corporation
* Copyright (C) 2024 Free Software Foundation, Inc.
*
* VAS_EBOOT is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* VAS_EBOOT 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 VAS_EBOOT. If not, see <http://www.gnu.org/licenses/>.
*/
#include <VasEBoot/misc.h>
#include <tss2_buffer.h>
void VasEBoot_tpm2_buffer_init (VasEBoot_tpm2_buffer_t buffer)
{
VasEBoot_memset (buffer->data, 0, sizeof (buffer->data));
buffer->size = 0;
buffer->offset = 0;
buffer->cap = sizeof (buffer->data);
buffer->error = 0;
}
void
VasEBoot_tpm2_buffer_pack (VasEBoot_tpm2_buffer_t buffer, const void *data, VasEBoot_size_t size)
{
VasEBoot_uint32_t r = buffer->cap - buffer->size;
if (buffer->error)
return;
if (size > r)
{
buffer->error = 1;
return;
}
VasEBoot_memcpy (&buffer->data[buffer->size], (void *) data, size);
buffer->size += size;
}
void
VasEBoot_tpm2_buffer_pack_u8 (VasEBoot_tpm2_buffer_t buffer, VasEBoot_uint8_t value)
{
VasEBoot_tpm2_buffer_pack (buffer, (const void *) &value, sizeof (value));
}
void
VasEBoot_tpm2_buffer_pack_u16 (VasEBoot_tpm2_buffer_t buffer, VasEBoot_uint16_t value)
{
VasEBoot_uint16_t tmp = VasEBoot_cpu_to_be16 (value);
VasEBoot_tpm2_buffer_pack (buffer, (const void *) &tmp, sizeof (tmp));
}
void
VasEBoot_tpm2_buffer_pack_u32 (VasEBoot_tpm2_buffer_t buffer, VasEBoot_uint32_t value)
{
VasEBoot_uint32_t tmp = VasEBoot_cpu_to_be32 (value);
VasEBoot_tpm2_buffer_pack (buffer, (const void *) &tmp, sizeof (tmp));
}
void
VasEBoot_tpm2_buffer_unpack (VasEBoot_tpm2_buffer_t buffer, void *data, VasEBoot_size_t size)
{
VasEBoot_uint32_t r = buffer->size - buffer->offset;
if (buffer->error)
return;
if (size > r)
{
buffer->error = 1;
return;
}
VasEBoot_memcpy (data, &buffer->data[buffer->offset], size);
buffer->offset += size;
}
void
VasEBoot_tpm2_buffer_unpack_u8 (VasEBoot_tpm2_buffer_t buffer, VasEBoot_uint8_t *value)
{
VasEBoot_uint32_t r = buffer->size - buffer->offset;
if (buffer->error)
return;
if (sizeof (*value) > r)
{
buffer->error = 1;
return;
}
VasEBoot_memcpy (value, &buffer->data[buffer->offset], sizeof (*value));
buffer->offset += sizeof (*value);
}
void
VasEBoot_tpm2_buffer_unpack_u16 (VasEBoot_tpm2_buffer_t buffer, VasEBoot_uint16_t *value)
{
VasEBoot_uint16_t tmp;
VasEBoot_uint32_t r = buffer->size - buffer->offset;
if (buffer->error)
return;
if (sizeof (tmp) > r)
{
buffer->error = 1;
return;
}
VasEBoot_memcpy (&tmp, &buffer->data[buffer->offset], sizeof (tmp));
buffer->offset += sizeof (tmp);
*value = VasEBoot_be_to_cpu16 (tmp);
}
void
VasEBoot_tpm2_buffer_unpack_u32 (VasEBoot_tpm2_buffer_t buffer, VasEBoot_uint32_t *value)
{
VasEBoot_uint32_t tmp;
VasEBoot_uint32_t r = buffer->size - buffer->offset;
if (buffer->error)
return;
if (sizeof (tmp) > r)
{
buffer->error = 1;
return;
}
VasEBoot_memcpy (&tmp, &buffer->data[buffer->offset], sizeof (tmp));
buffer->offset += sizeof (tmp);
*value = VasEBoot_be_to_cpu32 (tmp);
}