fwupd/libfwupdplugin/fu-efi-firmware-filesystem.c
2021-08-24 11:18:40 -05:00

117 lines
2.7 KiB
C

/*
* Copyright (C) 2021 Richard Hughes <richard@hughsie.com>
*
* SPDX-License-Identifier: LGPL-2.1+
*/
#include "config.h"
#include <fwupdplugin.h>
#include "fu-efi-firmware-file.h"
#include "fu-efi-firmware-filesystem.h"
/**
* FuEfiFirmwareFilesystem:
*
* A UEFI filesystem.
*
* See also: [class@FuFirmware]
*/
G_DEFINE_TYPE(FuEfiFirmwareFilesystem, fu_efi_firmware_filesystem, FU_TYPE_FIRMWARE)
static gboolean
fu_efi_firmware_filesystem_parse(FuFirmware *firmware,
GBytes *fw,
guint64 addr_start,
guint64 addr_end,
FwupdInstallFlags flags,
GError **error)
{
gsize offset = 0;
gsize bufsz = 0x0;
const guint8 *buf = g_bytes_get_data(fw, &bufsz);
while (offset + 0x18 < bufsz) {
g_autoptr(FuFirmware) img = fu_efi_firmware_file_new();
g_autoptr(GBytes) fw_tmp = NULL;
gboolean is_freespace = TRUE;
/* ignore free space */
for (guint i = 0; i < 0x18; i++) {
if (buf[offset + i] != 0xff) {
is_freespace = FALSE;
break;
}
}
if (is_freespace)
break;
fw_tmp = fu_common_bytes_new_offset(fw, offset, bufsz - offset, error);
if (fw_tmp == NULL)
return FALSE;
if (!fu_firmware_parse(img, fw_tmp, flags, error)) {
g_prefix_error(error, "failed to parse EFI file at 0x%x: ", (guint)offset);
return FALSE;
}
fu_firmware_set_offset(firmware, offset);
fu_firmware_add_image(firmware, img);
/* next! */
offset += fu_firmware_get_size(img);
}
/* success */
return TRUE;
}
static GBytes *
fu_efi_firmware_filesystem_write(FuFirmware *firmware, GError **error)
{
g_autoptr(GByteArray) buf = g_byte_array_new();
g_autoptr(GPtrArray) images = fu_firmware_get_images(firmware);
/* add each file */
for (guint i = 0; i < images->len; i++) {
FuFirmware *img = g_ptr_array_index(images, i);
g_autoptr(GBytes) blob = NULL;
fu_firmware_set_offset(img, buf->len);
blob = fu_firmware_write(img, error);
if (blob == NULL)
return NULL;
fu_byte_array_append_bytes(buf, blob);
fu_byte_array_align_up(buf, fu_firmware_get_alignment(firmware), 0xFF);
}
/* success */
return g_byte_array_free_to_bytes(g_steal_pointer(&buf));
}
static void
fu_efi_firmware_filesystem_init(FuEfiFirmwareFilesystem *self)
{
fu_firmware_set_alignment(FU_FIRMWARE(self), FU_FIRMWARE_ALIGNMENT_8);
}
static void
fu_efi_firmware_filesystem_class_init(FuEfiFirmwareFilesystemClass *klass)
{
FuFirmwareClass *klass_firmware = FU_FIRMWARE_CLASS(klass);
klass_firmware->parse = fu_efi_firmware_filesystem_parse;
klass_firmware->write = fu_efi_firmware_filesystem_write;
}
/**
* fu_efi_firmware_filesystem_new:
*
* Creates a new #FuFirmware
*
* Since: 1.6.2
**/
FuFirmware *
fu_efi_firmware_filesystem_new(void)
{
return FU_FIRMWARE(g_object_new(FU_TYPE_EFI_FIRMWARE_FILESYSTEM, NULL));
}