mirror of
https://git.proxmox.com/git/fwupd
synced 2025-05-22 16:34:42 +00:00
1abb32c623
To do this mount all ESP partitions and check all the binaries there to see if they match any entries in the new dbx. If we applied the update when a hash matched, we would unintentially 'brick' the users machine, as the grub and shim binaries *have* to be updated first. This functionality does reimplement the PE hashing functionality found in sbsigntools and pesign. This was done for 4 main reasons: * There were some memory safety issues found when fuzzing random binaries * Executing the tools hundreds of times was a lot of overhead * Operating from a blob of immutable mmap'd memory is much faster * We only need a very small amount of functionality from both tools
94 lines
2.5 KiB
C
94 lines
2.5 KiB
C
/*
|
|
* Copyright (C) 2020 Richard Hughes <richard@hughsie.com>
|
|
*
|
|
* SPDX-License-Identifier: LGPL-2.1+
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <fwupd.h>
|
|
|
|
#include "fu-common.h"
|
|
#include "fu-uefi-dbx-common.h"
|
|
#include "fu-efi-image.h"
|
|
#include "fu-efi-signature-parser.h"
|
|
|
|
static gchar *
|
|
fu_test_get_filename (const gchar *filename)
|
|
{
|
|
g_autofree gchar *path = NULL;
|
|
path = g_build_filename (TESTDATADIR, filename, NULL);
|
|
return fu_common_realpath (path, NULL);
|
|
}
|
|
|
|
static void
|
|
fu_efi_image_func (void)
|
|
{
|
|
const gchar *csum = NULL;
|
|
g_autofree gchar *fn = NULL;
|
|
g_autoptr(FuEfiImage) img = NULL;
|
|
g_autoptr(GBytes) bytes = NULL;
|
|
g_autoptr(GError) error = NULL;
|
|
|
|
fn = fu_test_get_filename ("fwupdx64.efi");
|
|
g_assert_nonnull (fn);
|
|
bytes = fu_common_get_contents_bytes (fn, &error);
|
|
g_assert_no_error (error);
|
|
g_assert_nonnull (bytes);
|
|
|
|
img = fu_efi_image_new (bytes, &error);
|
|
g_assert_no_error (error);
|
|
g_assert_nonnull (img);
|
|
csum = fu_efi_image_get_checksum (img);
|
|
g_assert_cmpstr (csum, ==, "e99707d4378140c01eb3f867240d5cc9e237b126d3db0c3b4bbcd3da1720ddff");
|
|
}
|
|
|
|
static void
|
|
fu_efi_signature_list_parse_func (void)
|
|
{
|
|
FuEfiSignatureList *siglist;
|
|
gboolean ret;
|
|
gsize bufsz = 0;
|
|
g_autofree gchar *fn = NULL;
|
|
g_autofree guint8 *buf = NULL;
|
|
g_autoptr(GPtrArray) siglists = NULL;
|
|
g_autoptr(GError) error = NULL;
|
|
|
|
/* load file */
|
|
fn = fu_uefi_dbx_get_dbxupdate (NULL);
|
|
if (fn == NULL) {
|
|
g_test_skip ("no dbx file, use -Defi_dbxdir=");
|
|
return;
|
|
}
|
|
ret = g_file_get_contents (fn, (gchar **) &buf, &bufsz, &error);
|
|
g_assert_no_error (error);
|
|
g_assert_true (ret);
|
|
|
|
/* parse the update */
|
|
siglists = fu_efi_signature_parser_new (buf, bufsz,
|
|
FU_EFI_SIGNATURE_PARSER_FLAGS_IGNORE_HEADER,
|
|
&error);
|
|
g_assert_no_error (error);
|
|
g_assert_nonnull (siglists);
|
|
g_assert_cmpint (siglists->len, ==, 1);
|
|
siglist = g_ptr_array_index (siglists, 0);
|
|
g_assert_cmpint (fu_efi_signature_list_get_all(siglist)->len, ==, 77);
|
|
g_assert_true (fu_efi_signature_list_has_checksum (siglist, "72e0bd1867cf5d9d56ab158adf3bddbc82bf32a8d8aa1d8c5e2f6df29428d6d8"));
|
|
g_assert_false (fu_efi_signature_list_has_checksum (siglist, "dave"));
|
|
}
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
g_test_init (&argc, &argv, NULL);
|
|
|
|
/* only critical and error are fatal */
|
|
g_log_set_fatal_mask (NULL, G_LOG_LEVEL_ERROR | G_LOG_LEVEL_CRITICAL);
|
|
g_setenv ("G_MESSAGES_DEBUG", "all", TRUE);
|
|
|
|
/* tests go here */
|
|
g_test_add_func ("/uefi-dbx/image", fu_efi_image_func);
|
|
g_test_add_func ("/uefi-dbx/file-parse", fu_efi_signature_list_parse_func);
|
|
return g_test_run ();
|
|
}
|