fwupd/plugins/bcm57xx/fu-self-test.c
Richard Hughes 52441f28a4 Allow objects to deserialize to XML
This makes a lot more sense; we can parse a firmware and export the same XML
we would use in a .builder.xml file. This allows us to two two things:

 * Check we can round trip from XML -> binary -> XML

 * Using a .builder.xml file we can check ->write() is endian safe
2021-03-15 12:07:30 +00:00

147 lines
4.7 KiB
C

/*
* Copyright (C) 2020 Richard Hughes <richard@hughsie.com>
*
* SPDX-License-Identifier: LGPL-2.1+
*/
#include "config.h"
#include <fwupd.h>
#include <string.h>
#include "fu-common.h"
#include "fu-bcm57xx-common.h"
#include "fu-bcm57xx-dict-image.h"
#include "fu-bcm57xx-firmware.h"
#include "fu-bcm57xx-stage1-image.h"
#include "fu-bcm57xx-stage2-image.h"
static void
fu_bcm57xx_create_verbuf (guint8 *bufver, const gchar *version)
{
memcpy (bufver, version, strlen (version) + 1);
}
static void
fu_bcm57xx_common_veritem_func (void)
{
g_autoptr(Bcm57xxVeritem) veritem1 = NULL;
g_autoptr(Bcm57xxVeritem) veritem2 = NULL;
g_autoptr(Bcm57xxVeritem) veritem3 = NULL;
guint8 bufver[16] = { 0x0 };
fu_bcm57xx_create_verbuf (bufver, "5719-v1.43");
veritem1 = fu_bcm57xx_veritem_new (bufver, sizeof(bufver));
g_assert_nonnull (veritem1);
g_assert_cmpstr (veritem1->version, ==, "1.43");
g_assert_cmpstr (veritem1->branch, ==, BCM_FW_BRANCH_UNKNOWN);
g_assert_cmpint (veritem1->verfmt, ==, FWUPD_VERSION_FORMAT_PAIR);
fu_bcm57xx_create_verbuf (bufver, "stage1-0.4.391");
veritem2 = fu_bcm57xx_veritem_new (bufver, sizeof(bufver));
g_assert_nonnull (veritem2);
g_assert_cmpstr (veritem2->version, ==, "0.4.391");
g_assert_cmpstr (veritem2->branch, ==, BCM_FW_BRANCH_OSS_FIRMWARE);
g_assert_cmpint (veritem2->verfmt, ==, FWUPD_VERSION_FORMAT_TRIPLET);
fu_bcm57xx_create_verbuf (bufver, "RANDOM-7");
veritem3 = fu_bcm57xx_veritem_new (bufver, sizeof(bufver));
g_assert_nonnull (veritem3);
g_assert_cmpstr (veritem3->version, ==, "RANDOM-7");
g_assert_cmpstr (veritem3->branch, ==, BCM_FW_BRANCH_UNKNOWN);
g_assert_cmpint (veritem3->verfmt, ==, FWUPD_VERSION_FORMAT_UNKNOWN);
}
static void
fu_bcm57xx_firmware_talos_func (void)
{
gboolean ret;
g_autofree gchar *fn = NULL;
g_autofree gchar *fn_out = NULL;
g_autoptr(GBytes) blob = NULL;
g_autoptr(GBytes) blob_out = NULL;
g_autoptr(GError) error = NULL;
g_autoptr(GPtrArray) images = NULL;
g_autoptr(FuFirmware) firmware = fu_bcm57xx_firmware_new ();
/* load file */
fn = g_test_build_filename (G_TEST_DIST, "tests", "Bcm5719_talos.bin", NULL);
if (!g_file_test (fn, G_FILE_TEST_EXISTS)) {
g_test_skip ("missing file");
return;
}
blob = fu_common_get_contents_bytes (fn, &error);
g_assert_no_error (error);
g_assert_nonnull (blob);
ret = fu_firmware_parse (firmware, blob, FWUPD_INSTALL_FLAG_NONE, &error);
g_assert_no_error (error);
g_assert_true (ret);
images = fu_firmware_get_images (firmware);
g_assert_cmpint (images->len, ==, 6);
blob_out = fu_firmware_write (firmware, &error);
g_assert_no_error (error);
g_assert_nonnull (blob_out);
fn_out = g_test_build_filename (G_TEST_BUILT, "tests", "Bcm5719_talos.bin", NULL);
ret = fu_common_set_contents_bytes (fn_out, blob_out, &error);
g_assert_no_error (error);
g_assert_true (ret);
ret = fu_common_bytes_compare (blob, blob_out, &error);
g_assert_no_error (error);
g_assert_true (ret);
}
static void
fu_bcm57xx_firmware_xml_func (void)
{
gboolean ret;
g_autofree gchar *csum1 = NULL;
g_autofree gchar *csum2 = NULL;
g_autofree gchar *xml_out = NULL;
g_autofree gchar *xml_src = NULL;
g_autoptr(FuFirmware) firmware1 = fu_bcm57xx_firmware_new ();
g_autoptr(FuFirmware) firmware2 = fu_bcm57xx_firmware_new ();
g_autoptr(GError) error = NULL;
/* build and write */
ret = g_file_get_contents (FWUPD_FUZZINGSRCDIR "/bcm57xx.builder.xml",
&xml_src, NULL, &error);
g_assert_no_error (error);
g_assert_true (ret);
ret = fu_firmware_build_from_xml (firmware1, xml_src, &error);
g_assert_no_error (error);
g_assert_true (ret);
csum1 = fu_firmware_get_checksum (firmware1, G_CHECKSUM_SHA1, &error);
g_assert_no_error (error);
g_assert_cmpstr (csum1, ==, "a3ac108905c37857cf48612b707c1c72c582f914");
/* ensure we can round-trip */
xml_out = fu_firmware_export_to_xml (firmware1,
FU_FIRMWARE_EXPORT_FLAG_NONE,
&error);
g_assert_no_error (error);
ret = fu_firmware_build_from_xml (firmware2, xml_out, &error);
g_assert_no_error (error);
g_assert_true (ret);
csum2 = fu_firmware_get_checksum (firmware2, G_CHECKSUM_SHA1, &error);
g_assert_cmpstr (csum1, ==, csum2);
}
int
main (int argc, char **argv)
{
g_test_init (&argc, &argv, NULL);
g_type_ensure (FU_TYPE_BCM57XX_STAGE1_IMAGE);
g_type_ensure (FU_TYPE_BCM57XX_STAGE2_IMAGE);
g_type_ensure (FU_TYPE_BCM57XX_DICT_IMAGE);
/* only critical and error are fatal */
g_log_set_fatal_mask (NULL, G_LOG_LEVEL_ERROR | G_LOG_LEVEL_CRITICAL);
/* tests go here */
g_test_add_func ("/fwupd/bcm57xx/firmware{xml}", fu_bcm57xx_firmware_xml_func);
g_test_add_func ("/fwupd/bcm57xx/firmware{talos}", fu_bcm57xx_firmware_talos_func);
g_test_add_func ("/fwupd/bcm57xx/common{veritem}", fu_bcm57xx_common_veritem_func);
return g_test_run ();
}