fwupd/libfwupdplugin/fu-ifd-image.c
2021-08-24 11:18:40 -05:00

129 lines
3.0 KiB
C

/*
* Copyright (C) 2021 Richard Hughes <richard@hughsie.com>
*
* SPDX-License-Identifier: LGPL-2.1+
*/
#include "config.h"
#include <fwupdplugin.h>
#include "fu-ifd-image.h"
typedef struct {
FuIfdAccess access[FU_IFD_REGION_MAX];
} FuIfdImagePrivate;
G_DEFINE_TYPE_WITH_PRIVATE(FuIfdImage, fu_ifd_image, FU_TYPE_FIRMWARE)
#define GET_PRIVATE(o) (fu_ifd_image_get_instance_private(o))
static void
fu_ifd_image_export(FuFirmware *firmware, FuFirmwareExportFlags flags, XbBuilderNode *bn)
{
FuIfdImage *self = FU_IFD_IMAGE(firmware);
FuIfdImagePrivate *priv = GET_PRIVATE(self);
for (guint i = 0; i < FU_IFD_REGION_MAX; i++) {
if (priv->access[i] == FU_IFD_ACCESS_NONE)
continue;
xb_builder_node_insert_text(bn,
"access",
fu_ifd_access_to_string(priv->access[i]),
"region",
fu_ifd_region_to_string(i),
NULL);
}
}
/**
* fu_ifd_image_set_access:
* @self: a #FuIfdImage
* @region: a #FuIfdRegion, e.g. %FU_IFD_REGION_BIOS
* @access: a #FuIfdAccess, e.g. %FU_IFD_ACCESS_NONE
*
* Sets the access control for a specific reason.
*
* Since: 1.6.2
**/
void
fu_ifd_image_set_access(FuIfdImage *self, FuIfdRegion region, FuIfdAccess access)
{
FuIfdImagePrivate *priv = GET_PRIVATE(self);
priv->access[region] = access;
}
/**
* fu_ifd_image_get_access:
* @self: a #FuIfdImage
* @region: a #FuIfdRegion, e.g. %FU_IFD_REGION_BIOS
*
* Gets the access control for a specific reason.
*
* Returns: a #FuIfdAccess, e.g. %FU_IFD_ACCESS_NONE
*
* Since: 1.6.2
**/
FuIfdAccess
fu_ifd_image_get_access(FuIfdImage *self, FuIfdRegion region)
{
FuIfdImagePrivate *priv = GET_PRIVATE(self);
return priv->access[region];
}
static GBytes *
fu_ifd_image_write(FuFirmware *firmware, GError **error)
{
g_autoptr(GByteArray) buf = g_byte_array_new();
g_autoptr(GPtrArray) images = fu_firmware_get_images(firmware);
/* add each volume */
if (images->len > 0) {
for (guint i = 0; i < images->len; i++) {
FuFirmware *img = g_ptr_array_index(images, i);
g_autoptr(GBytes) bytes = fu_firmware_write(img, error);
if (bytes == NULL)
return NULL;
fu_byte_array_append_bytes(buf, bytes);
}
} else {
g_autoptr(GBytes) bytes = NULL;
bytes = fu_firmware_get_bytes(firmware, error);
if (bytes == NULL)
return NULL;
fu_byte_array_append_bytes(buf, bytes);
}
/* align up */
fu_byte_array_set_size(buf,
fu_common_align_up(buf->len, fu_firmware_get_alignment(firmware)));
/* success */
return g_byte_array_free_to_bytes(g_steal_pointer(&buf));
}
static void
fu_ifd_image_init(FuIfdImage *self)
{
fu_firmware_set_alignment(FU_FIRMWARE(self), FU_FIRMWARE_ALIGNMENT_4K);
}
static void
fu_ifd_image_class_init(FuIfdImageClass *klass)
{
FuFirmwareClass *klass_image = FU_FIRMWARE_CLASS(klass);
klass_image->export = fu_ifd_image_export;
klass_image->write = fu_ifd_image_write;
}
/**
* fu_ifd_image_new:
*
* Creates a new #FuFirmware
*
* Since: 1.6.2
**/
FuFirmware *
fu_ifd_image_new(void)
{
return FU_FIRMWARE(g_object_new(FU_TYPE_IFD_IMAGE, NULL));
}