mirror of
				https://git.proxmox.com/git/fwupd
				synced 2025-10-31 07:42:14 +00:00 
			
		
		
		
	 7afd7cba0d
			
		
	
	
		7afd7cba0d
		
	
	
	
	
		
			
			In many plugins we've wanted to use ->prepare_firmware() to parse the firmware ahead of ->detach() and ->write_firmware() but this has the limitation that it can only return a single blob of data. For many devices, multiple binary blobs are required from one parsed image, for instance providing signatures, config and data blobs that have to be pushed to the device in different way. This also means we parse the firmware *before* we ask the user to detach. Break the internal FuDevice API to support these firmware types as they become more popular. This also allows us to move the Intel HEX and SREC parsing out of the dfu plugin as they are used by a few plugins now, and resolving symbols between plugins isn't exactly awesome.
		
			
				
	
	
		
			608 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			608 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Copyright (C) 2015 Richard Hughes <richard@hughsie.com>
 | |
|  *
 | |
|  * SPDX-License-Identifier: LGPL-2.1+
 | |
|  */
 | |
| 
 | |
| #include "config.h"
 | |
| 
 | |
| #include <glib-object.h>
 | |
| #include <stdlib.h>
 | |
| #include <string.h>
 | |
| 
 | |
| #include "dfu-cipher-xtea.h"
 | |
| #include "dfu-common.h"
 | |
| #include "dfu-device-private.h"
 | |
| #include "dfu-firmware.h"
 | |
| #include "dfu-patch.h"
 | |
| #include "dfu-sector-private.h"
 | |
| #include "dfu-target-private.h"
 | |
| 
 | |
| #include "fu-test.h"
 | |
| 
 | |
| #include "fwupd-error.h"
 | |
| 
 | |
| static gchar *
 | |
| dfu_test_get_filename (const gchar *filename)
 | |
| {
 | |
| 	gchar *tmp;
 | |
| 	char full_tmp[PATH_MAX];
 | |
| 	g_autofree gchar *path = NULL;
 | |
| 	path = g_build_filename (TESTDATADIR, filename, NULL);
 | |
| 	tmp = realpath (path, full_tmp);
 | |
| 	if (tmp == NULL)
 | |
| 		return NULL;
 | |
| 	return g_strdup (full_tmp);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_cipher_xtea_func (void)
 | |
| {
 | |
| 	gboolean ret;
 | |
| 	guint8 buf[] = { 'H', 'i', 'y', 'a', 'D', 'a', 'v', 'e' };
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 
 | |
| 	ret = dfu_cipher_encrypt_xtea ("test", buf, sizeof(buf), &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 
 | |
| 	g_assert_cmpint (buf[0], ==, 128);
 | |
| 	g_assert_cmpint (buf[1], ==, 220);
 | |
| 	g_assert_cmpint (buf[2], ==, 23);
 | |
| 	g_assert_cmpint (buf[3], ==, 55);
 | |
| 	g_assert_cmpint (buf[4], ==, 201);
 | |
| 	g_assert_cmpint (buf[5], ==, 207);
 | |
| 	g_assert_cmpint (buf[6], ==, 182);
 | |
| 	g_assert_cmpint (buf[7], ==, 177);
 | |
| 
 | |
| 	ret = dfu_cipher_decrypt_xtea ("test", buf, sizeof(buf), &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 
 | |
| 	g_assert_cmpint (buf[0], ==, 'H');
 | |
| 	g_assert_cmpint (buf[1], ==, 'i');
 | |
| 	g_assert_cmpint (buf[2], ==, 'y');
 | |
| 	g_assert_cmpint (buf[3], ==, 'a');
 | |
| 	g_assert_cmpint (buf[4], ==, 'D');
 | |
| 	g_assert_cmpint (buf[5], ==, 'a');
 | |
| 	g_assert_cmpint (buf[6], ==, 'v');
 | |
| 	g_assert_cmpint (buf[7], ==, 'e');
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_firmware_xdfu_func (void)
 | |
| {
 | |
| 	gboolean ret;
 | |
| 	g_autofree gchar *fn = NULL;
 | |
| 	g_autoptr(DfuFirmware) firmware = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 	g_autoptr(GFile) file = NULL;
 | |
| 
 | |
| 	fn = dfu_test_get_filename ("example.xdfu");
 | |
| 	g_assert (fn != NULL);
 | |
| 	firmware = dfu_firmware_new ();
 | |
| 	file = g_file_new_for_path (fn);
 | |
| 	ret = dfu_firmware_parse_file (firmware, file,
 | |
| 				       DFU_FIRMWARE_PARSE_FLAG_NONE,
 | |
| 				       &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_firmware_get_cipher_kind (firmware), ==, DFU_CIPHER_KIND_XTEA);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_enums_func (void)
 | |
| {
 | |
| 	for (guint i = 0; i < DFU_STATE_LAST; i++)
 | |
| 		g_assert_cmpstr (dfu_state_to_string (i), !=, NULL);
 | |
| 	for (guint i = 0; i < DFU_STATUS_LAST; i++)
 | |
| 		g_assert_cmpstr (dfu_status_to_string (i), !=, NULL);
 | |
| }
 | |
| 
 | |
| static GBytes *
 | |
| dfu_self_test_get_bytes_for_file (GFile *file, GError **error)
 | |
| {
 | |
| 	gchar *contents = NULL;
 | |
| 	gsize length = 0;
 | |
| 	if (!g_file_load_contents (file, NULL, &contents, &length, NULL, error))
 | |
| 		return NULL;
 | |
| 	return g_bytes_new_take (contents, length);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_firmware_raw_func (void)
 | |
| {
 | |
| 	DfuElement *element;
 | |
| 	DfuImage *image_tmp;
 | |
| 	GBytes *no_suffix_contents;
 | |
| 	gchar buf[256];
 | |
| 	gboolean ret;
 | |
| 	g_autoptr(DfuFirmware) firmware = NULL;
 | |
| 	g_autoptr(GBytes) fw = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 
 | |
| 	/* set up some dummy data */
 | |
| 	for (guint i = 0; i < 256; i++)
 | |
| 		buf[i] = (gchar) i;
 | |
| 	fw = g_bytes_new_static (buf, 256);
 | |
| 
 | |
| 	/* load a non DFU firmware */
 | |
| 	firmware = dfu_firmware_new ();
 | |
| 	ret = dfu_firmware_parse_data (firmware, fw, DFU_FIRMWARE_PARSE_FLAG_NONE, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_firmware_get_vid (firmware), ==, 0xffff);
 | |
| 	g_assert_cmpint (dfu_firmware_get_pid (firmware), ==, 0xffff);
 | |
| 	g_assert_cmpint (dfu_firmware_get_release (firmware), ==, 0xffff);
 | |
| 	g_assert_cmpint (dfu_firmware_get_format (firmware), ==, DFU_FIRMWARE_FORMAT_RAW);
 | |
| 	g_assert_cmpint (dfu_firmware_get_cipher_kind (firmware), ==, DFU_CIPHER_KIND_NONE);
 | |
| 	image_tmp = dfu_firmware_get_image (firmware, 0xfe);
 | |
| 	g_assert (image_tmp == NULL);
 | |
| 	image_tmp = dfu_firmware_get_image (firmware, 0);
 | |
| 	g_assert (image_tmp != NULL);
 | |
| 	g_assert_cmpint (dfu_image_get_size (image_tmp), ==, 256);
 | |
| 	element = dfu_image_get_element (image_tmp, 0);
 | |
| 	g_assert (element != NULL);
 | |
| 	no_suffix_contents = dfu_element_get_contents (element);
 | |
| 	g_assert (no_suffix_contents != NULL);
 | |
| 	g_assert_cmpint (g_bytes_compare (no_suffix_contents, fw), ==, 0);
 | |
| 
 | |
| 	/* can we roundtrip without adding data */
 | |
| 	roundtrip = dfu_firmware_write_data (firmware, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip != NULL);
 | |
| 	ret = fu_common_bytes_compare (roundtrip, fw, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert_true (ret);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_firmware_dfu_func (void)
 | |
| {
 | |
| 	gchar buf[256];
 | |
| 	gboolean ret;
 | |
| 	g_autofree gchar *filename = NULL;
 | |
| 	g_autoptr(DfuFirmware) firmware = NULL;
 | |
| 	g_autoptr(DfuImage) image = NULL;
 | |
| 	g_autoptr(DfuElement) element = NULL;
 | |
| 	g_autoptr(GBytes) data = NULL;
 | |
| 	g_autoptr(GBytes) fw = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip_orig = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 	g_autoptr(GFile) file = NULL;
 | |
| 
 | |
| 	/* set up some dummy data */
 | |
| 	for (guint i = 0; i < 256; i++)
 | |
| 		buf[i] = (gchar) i;
 | |
| 	fw = g_bytes_new_static (buf, 256);
 | |
| 
 | |
| 	/* write DFU format */
 | |
| 	firmware = dfu_firmware_new ();
 | |
| 	dfu_firmware_set_format (firmware, DFU_FIRMWARE_FORMAT_DFU);
 | |
| 	dfu_firmware_set_vid (firmware, 0x1234);
 | |
| 	dfu_firmware_set_pid (firmware, 0x5678);
 | |
| 	dfu_firmware_set_release (firmware, 0xfedc);
 | |
| 	image = dfu_image_new ();
 | |
| 	element = dfu_element_new ();
 | |
| 	dfu_element_set_contents (element, fw);
 | |
| 	dfu_image_add_element (image, element);
 | |
| 	dfu_firmware_add_image (firmware, image);
 | |
| 	g_assert_cmpint (dfu_firmware_get_size (firmware), ==, 256);
 | |
| 	data = dfu_firmware_write_data (firmware, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (data != NULL);
 | |
| 
 | |
| 	/* can we load it again? */
 | |
| 	g_ptr_array_set_size (dfu_firmware_get_images (firmware), 0);
 | |
| 	ret = dfu_firmware_parse_data (firmware, data, DFU_FIRMWARE_PARSE_FLAG_NONE, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_firmware_get_vid (firmware), ==, 0x1234);
 | |
| 	g_assert_cmpint (dfu_firmware_get_pid (firmware), ==, 0x5678);
 | |
| 	g_assert_cmpint (dfu_firmware_get_release (firmware), ==, 0xfedc);
 | |
| 	g_assert_cmpint (dfu_firmware_get_format (firmware), ==, DFU_FIRMWARE_FORMAT_DFU);
 | |
| 	g_assert_cmpint (dfu_firmware_get_size (firmware), ==, 256);
 | |
| 
 | |
| 	/* load a real firmware */
 | |
| 	filename = dfu_test_get_filename ("kiibohd.dfu.bin");
 | |
| 	g_assert (filename != NULL);
 | |
| 	file = g_file_new_for_path (filename);
 | |
| 	g_ptr_array_set_size (dfu_firmware_get_images (firmware), 0);
 | |
| 	ret = dfu_firmware_parse_file (firmware, file,
 | |
| 				       DFU_FIRMWARE_PARSE_FLAG_NONE,
 | |
| 				       &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_firmware_get_vid (firmware), ==, 0x1c11);
 | |
| 	g_assert_cmpint (dfu_firmware_get_pid (firmware), ==, 0xb007);
 | |
| 	g_assert_cmpint (dfu_firmware_get_release (firmware), ==, 0xffff);
 | |
| 	g_assert_cmpint (dfu_firmware_get_format (firmware), ==, DFU_FIRMWARE_FORMAT_DFU);
 | |
| 	g_assert_cmpint (dfu_firmware_get_size (firmware), ==, 0x8eB4);
 | |
| 	g_assert_cmpint (dfu_firmware_get_cipher_kind (firmware), ==, DFU_CIPHER_KIND_NONE);
 | |
| 
 | |
| 	/* can we roundtrip without losing data */
 | |
| 	roundtrip_orig = dfu_self_test_get_bytes_for_file (file, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip_orig != NULL);
 | |
| 	roundtrip = dfu_firmware_write_data (firmware, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip != NULL);
 | |
| 	ret = fu_common_bytes_compare (roundtrip, roundtrip_orig, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert_true (ret);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_firmware_dfuse_func (void)
 | |
| {
 | |
| 	gboolean ret;
 | |
| 	g_autofree gchar *filename = NULL;
 | |
| 	g_autoptr(DfuFirmware) firmware = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip_orig = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 	g_autoptr(GFile) file = NULL;
 | |
| 
 | |
| 	/* load a DeFUse firmware */
 | |
| 	g_setenv ("DFU_SELF_TEST_IMAGE_MEMCPY_NAME", "", FALSE);
 | |
| 	filename = dfu_test_get_filename ("dev_VRBRAIN.dfu");
 | |
| 	g_assert (filename != NULL);
 | |
| 	file = g_file_new_for_path (filename);
 | |
| 	firmware = dfu_firmware_new ();
 | |
| 	ret = dfu_firmware_parse_file (firmware, file,
 | |
| 				       DFU_FIRMWARE_PARSE_FLAG_NONE,
 | |
| 				       &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_firmware_get_vid (firmware), ==, 0x0483);
 | |
| 	g_assert_cmpint (dfu_firmware_get_pid (firmware), ==, 0x0000);
 | |
| 	g_assert_cmpint (dfu_firmware_get_release (firmware), ==, 0x0000);
 | |
| 	g_assert_cmpint (dfu_firmware_get_format (firmware), ==, DFU_FIRMWARE_FORMAT_DFUSE);
 | |
| 	g_assert_cmpint (dfu_firmware_get_size (firmware), ==, 0x168d5);
 | |
| 	g_assert_cmpint (dfu_firmware_get_cipher_kind (firmware), ==, DFU_CIPHER_KIND_NONE);
 | |
| 
 | |
| 	/* can we roundtrip without losing data */
 | |
| 	roundtrip_orig = dfu_self_test_get_bytes_for_file (file, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip_orig != NULL);
 | |
| 	roundtrip = dfu_firmware_write_data (firmware, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip != NULL);
 | |
| 
 | |
| //	g_file_set_contents ("/tmp/1.bin",
 | |
| //			     g_bytes_get_data (roundtrip, NULL),
 | |
| //			     g_bytes_get_size (roundtrip), NULL);
 | |
| 
 | |
| 	ret = fu_common_bytes_compare (roundtrip, roundtrip_orig, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert_true (ret);
 | |
| 
 | |
| 	/* use usual image name copying */
 | |
| 	g_unsetenv ("DFU_SELF_TEST_IMAGE_MEMCPY_NAME");
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_firmware_metadata_func (void)
 | |
| {
 | |
| 	gboolean ret;
 | |
| 	g_autofree gchar *filename = NULL;
 | |
| 	g_autoptr(DfuFirmware) firmware = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip_orig = NULL;
 | |
| 	g_autoptr(GBytes) roundtrip = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 	g_autoptr(GFile) file = NULL;
 | |
| 
 | |
| 	/* load a DFU firmware with a metadata table */
 | |
| 	filename = dfu_test_get_filename ("metadata.dfu");
 | |
| 	g_assert (filename != NULL);
 | |
| 	file = g_file_new_for_path (filename);
 | |
| 	firmware = dfu_firmware_new ();
 | |
| 	ret = dfu_firmware_parse_file (firmware, file,
 | |
| 				       DFU_FIRMWARE_PARSE_FLAG_NONE,
 | |
| 				       &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_firmware_get_size (firmware), ==, 6);
 | |
| 	g_assert_cmpstr (dfu_firmware_get_metadata (firmware, "key"), ==, "value");
 | |
| 	g_assert_cmpstr (dfu_firmware_get_metadata (firmware, "???"), ==, NULL);
 | |
| 
 | |
| 	/* can we roundtrip without losing data */
 | |
| 	roundtrip_orig = dfu_self_test_get_bytes_for_file (file, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip_orig != NULL);
 | |
| 	roundtrip = dfu_firmware_write_data (firmware, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (roundtrip != NULL);
 | |
| 	ret = fu_common_bytes_compare (roundtrip, roundtrip_orig, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert_true (ret);
 | |
| }
 | |
| 
 | |
| static gchar *
 | |
| dfu_target_sectors_to_string (DfuTarget *target)
 | |
| {
 | |
| 	GPtrArray *sectors;
 | |
| 	GString *str;
 | |
| 
 | |
| 	str = g_string_new ("");
 | |
| 	sectors = dfu_target_get_sectors (target);
 | |
| 	for (guint i = 0; i < sectors->len; i++) {
 | |
| 		DfuSector *sector = g_ptr_array_index (sectors, i);
 | |
| 		g_autofree gchar *tmp = dfu_sector_to_string (sector);
 | |
| 		g_string_append_printf (str, "%s\n", tmp);
 | |
| 	}
 | |
| 	if (str->len > 0)
 | |
| 		g_string_truncate (str, str->len - 1);
 | |
| 	return g_string_free (str, FALSE);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_target_dfuse_func (void)
 | |
| {
 | |
| 	gboolean ret;
 | |
| 	gchar *tmp;
 | |
| 	g_autoptr(DfuTarget) target = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 
 | |
| 	/* NULL */
 | |
| 	target = g_object_new (DFU_TYPE_TARGET, NULL);
 | |
| 	ret = dfu_target_parse_sectors (target, NULL, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	tmp = dfu_target_sectors_to_string (target);
 | |
| 	g_assert_cmpstr (tmp, ==, "");
 | |
| 	g_free (tmp);
 | |
| 
 | |
| 	/* no addresses */
 | |
| 	ret = dfu_target_parse_sectors (target, "@Flash3", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	tmp = dfu_target_sectors_to_string (target);
 | |
| 	g_assert_cmpstr (tmp, ==, "");
 | |
| 	g_free (tmp);
 | |
| 
 | |
| 	/* one sector, no space */
 | |
| 	ret = dfu_target_parse_sectors (target, "@Internal Flash /0x08000000/2*001Ka", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	tmp = dfu_target_sectors_to_string (target);
 | |
| 	ret = fu_test_compare_lines (tmp,
 | |
| 				      "Zone:0, Sec#:0, Addr:0x08000000, Size:0x0400, Caps:0x1 [R]\n"
 | |
| 				      "Zone:0, Sec#:0, Addr:0x08000400, Size:0x0400, Caps:0x1 [R]",
 | |
| 				      &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_free (tmp);
 | |
| 
 | |
| 	/* multiple sectors */
 | |
| 	ret = dfu_target_parse_sectors (target, "@Flash1   /0x08000000/2*001Ka,4*001Kg", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	tmp = dfu_target_sectors_to_string (target);
 | |
| 	ret = fu_test_compare_lines (tmp,
 | |
| 				      "Zone:0, Sec#:0, Addr:0x08000000, Size:0x0400, Caps:0x1 [R]\n"
 | |
| 				      "Zone:0, Sec#:0, Addr:0x08000400, Size:0x0400, Caps:0x1 [R]\n"
 | |
| 				      "Zone:0, Sec#:1, Addr:0x08000800, Size:0x0400, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:0, Sec#:1, Addr:0x08000c00, Size:0x0400, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:0, Sec#:1, Addr:0x08001000, Size:0x0400, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:0, Sec#:1, Addr:0x08001400, Size:0x0400, Caps:0x7 [REW]",
 | |
| 				      &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_free (tmp);
 | |
| 
 | |
| 	/* non-contiguous */
 | |
| 	ret = dfu_target_parse_sectors (target, "@Flash2 /0xF000/4*100Ba/0xE000/3*8Kg/0x80000/2*24Kg", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	tmp = dfu_target_sectors_to_string (target);
 | |
| 	ret = fu_test_compare_lines (tmp,
 | |
| 				      "Zone:0, Sec#:0, Addr:0x0000f000, Size:0x0064, Caps:0x1 [R]\n"
 | |
| 				      "Zone:0, Sec#:0, Addr:0x0000f064, Size:0x0064, Caps:0x1 [R]\n"
 | |
| 				      "Zone:0, Sec#:0, Addr:0x0000f0c8, Size:0x0064, Caps:0x1 [R]\n"
 | |
| 				      "Zone:0, Sec#:0, Addr:0x0000f12c, Size:0x0064, Caps:0x1 [R]\n"
 | |
| 				      "Zone:1, Sec#:0, Addr:0x0000e000, Size:0x2000, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:1, Sec#:0, Addr:0x00010000, Size:0x2000, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:1, Sec#:0, Addr:0x00012000, Size:0x2000, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:2, Sec#:0, Addr:0x00080000, Size:0x6000, Caps:0x7 [REW]\n"
 | |
| 				      "Zone:2, Sec#:0, Addr:0x00086000, Size:0x6000, Caps:0x7 [REW]",
 | |
| 				      &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_free (tmp);
 | |
| 
 | |
| 	/* invalid */
 | |
| 	ret = dfu_target_parse_sectors (target, "Flash", NULL);
 | |
| 	g_assert (ret);
 | |
| 	ret = dfu_target_parse_sectors (target, "@Internal Flash /0x08000000", NULL);
 | |
| 	g_assert (!ret);
 | |
| 	ret = dfu_target_parse_sectors (target, "@Internal Flash /0x08000000/12*001a", NULL);
 | |
| 	g_assert (!ret);
 | |
| 
 | |
| 	/* indicate a cipher being used */
 | |
| 	g_assert_cmpint (dfu_target_get_cipher_kind (target), ==, DFU_CIPHER_KIND_NONE);
 | |
| 	ret = dfu_target_parse_sectors (target, "@Flash|XTEA", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	g_assert_cmpint (dfu_target_get_cipher_kind (target), ==, DFU_CIPHER_KIND_XTEA);
 | |
| }
 | |
| 
 | |
| static gboolean
 | |
| dfu_patch_create_from_strings (DfuPatch *patch,
 | |
| 			       const gchar *dold,
 | |
| 			       const gchar *dnew,
 | |
| 			       GError **error)
 | |
| {
 | |
| 	guint32 sz1 = strlen (dold);
 | |
| 	guint32 sz2 = strlen (dnew);
 | |
| 	g_autoptr(GBytes) blob1 = g_bytes_new (dold, sz1);
 | |
| 	g_autoptr(GBytes) blob2 = g_bytes_new (dnew, sz2);
 | |
| 	g_debug ("compare:\n%s\n%s", dold, dnew);
 | |
| 	return dfu_patch_create (patch, blob1, blob2, error);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_patch_merges_func (void)
 | |
| {
 | |
| 	const guint8 *data;
 | |
| 	gboolean ret;
 | |
| 	gsize sz;
 | |
| 	g_autoptr(DfuPatch) patch = dfu_patch_new ();
 | |
| 	g_autoptr(GBytes) blob = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 
 | |
| 	/* check merges happen */
 | |
| 	ret = dfu_patch_create_from_strings (patch, "XXX", "YXY", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	blob = dfu_patch_export (patch, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	data = g_bytes_get_data (blob, &sz);
 | |
| 	g_assert_cmpint (data[0x00], ==, 'D');
 | |
| 	g_assert_cmpint (data[0x01], ==, 'f');
 | |
| 	g_assert_cmpint (data[0x02], ==, 'u');
 | |
| 	g_assert_cmpint (data[0x03], ==, 'P');
 | |
| 	g_assert_cmpint (data[0x04], ==, 0x00); /* reserved */
 | |
| 	g_assert_cmpint (data[0x05], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x06], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x07], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x08 + 0x28], ==, 0x00); /* chunk1, offset */
 | |
| 	g_assert_cmpint (data[0x09 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0a + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0b + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0c + 0x28], ==, 0x03); /* chunk1, size */
 | |
| 	g_assert_cmpint (data[0x0d + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0e + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0f + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x10 + 0x28], ==, 0x00); /* reserved */
 | |
| 	g_assert_cmpint (data[0x11 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x12 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x13 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x14 + 0x28], ==, 'Y');
 | |
| 	g_assert_cmpint (data[0x15 + 0x28], ==, 'X');
 | |
| 	g_assert_cmpint (data[0x16 + 0x28], ==, 'Y');
 | |
| 	g_assert_cmpint (sz, ==, 48 /* hdr */ + 12 /* chunk */ + 3 /* data */);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_patch_apply_func (void)
 | |
| {
 | |
| 	gboolean ret;
 | |
| 	g_autoptr(DfuPatch) patch = dfu_patch_new ();
 | |
| 	g_autoptr(GBytes) blob_new2 = NULL;
 | |
| 	g_autoptr(GBytes) blob_new3 = NULL;
 | |
| 	g_autoptr(GBytes) blob_new4 = NULL;
 | |
| 	g_autoptr(GBytes) blob_new = NULL;
 | |
| 	g_autoptr(GBytes) blob_old = NULL;
 | |
| 	g_autoptr(GBytes) blob_wrong = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 
 | |
| 	/* create a patch */
 | |
| 	blob_old = g_bytes_new_static ("helloworldhelloworldhelloworldhelloworld", 40);
 | |
| 	blob_new = g_bytes_new_static ("XelloXorldhelloworldhelloworldhelloworlXXX", 42);
 | |
| 	ret = dfu_patch_create (patch, blob_old, blob_new, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 
 | |
| 	/* apply the patch */
 | |
| 	blob_new2 = dfu_patch_apply (patch, blob_old, DFU_PATCH_APPLY_FLAG_NONE, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (blob_new2 != NULL);
 | |
| 	g_assert_cmpint (g_bytes_compare (blob_new, blob_new2), ==, 0);
 | |
| 
 | |
| 	/* check we force the patch to an unrelated blob */
 | |
| 	blob_wrong = g_bytes_new_static ("wrongwrongwrongwrongwrongwrongwrongwrong", 40);
 | |
| 	blob_new3 = dfu_patch_apply (patch, blob_wrong, DFU_PATCH_APPLY_FLAG_IGNORE_CHECKSUM, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (blob_new3 != NULL);
 | |
| 
 | |
| 	/* check we can't apply the patch to an unrelated blob */
 | |
| 	blob_new4 = dfu_patch_apply (patch, blob_wrong, DFU_PATCH_APPLY_FLAG_NONE, &error);
 | |
| 	g_assert_error (error, FWUPD_ERROR, FWUPD_ERROR_INVALID_FILE);
 | |
| 	g_assert (blob_new4 == NULL);
 | |
| }
 | |
| 
 | |
| static void
 | |
| dfu_patch_func (void)
 | |
| {
 | |
| 	const guint8 *data;
 | |
| 	gboolean ret;
 | |
| 	gsize sz;
 | |
| 	g_autoptr(DfuPatch) patch = dfu_patch_new ();
 | |
| 	g_autoptr(DfuPatch) patch2 = dfu_patch_new ();
 | |
| 	g_autoptr(GBytes) blob = NULL;
 | |
| 	g_autoptr(GError) error = NULL;
 | |
| 	g_autofree gchar *serialized_str = NULL;
 | |
| 
 | |
| 	/* create binary diff */
 | |
| 	ret = dfu_patch_create_from_strings (patch, "XXX", "XYY", &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 
 | |
| 	/* check we can serialize this object to a blob */
 | |
| 	blob = dfu_patch_export (patch, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	data = g_bytes_get_data (blob, &sz);
 | |
| 	g_assert_cmpint (data[0x00], ==, 'D');
 | |
| 	g_assert_cmpint (data[0x01], ==, 'f');
 | |
| 	g_assert_cmpint (data[0x02], ==, 'u');
 | |
| 	g_assert_cmpint (data[0x03], ==, 'P');
 | |
| 	g_assert_cmpint (data[0x04], ==, 0x00); /* reserved */
 | |
| 	g_assert_cmpint (data[0x05], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x06], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x07], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x08 + 0x28], ==, 0x01); /* chunk1, offset */
 | |
| 	g_assert_cmpint (data[0x09 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0a + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0b + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0c + 0x28], ==, 0x02); /* chunk1, size */
 | |
| 	g_assert_cmpint (data[0x0d + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0e + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x0f + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x10 + 0x28], ==, 0x00); /* reserved */
 | |
| 	g_assert_cmpint (data[0x11 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x12 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x13 + 0x28], ==, 0x00);
 | |
| 	g_assert_cmpint (data[0x14 + 0x28], ==, 'Y');
 | |
| 	g_assert_cmpint (data[0x15 + 0x28], ==, 'Y');
 | |
| 	g_assert_cmpint (sz, ==, 48 /* hdr */ + 12 /* chunk */ + 2 /* data */);
 | |
| 
 | |
| 	/* try to load it from the serialized blob */
 | |
| 	ret = dfu_patch_import (patch2, blob, &error);
 | |
| 	g_assert_no_error (error);
 | |
| 	g_assert (ret);
 | |
| 	serialized_str = dfu_patch_to_string (patch2);
 | |
| 	g_debug ("serialized blob %s", serialized_str);
 | |
| }
 | |
| 
 | |
| 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);
 | |
| 
 | |
| 	/* log everything */
 | |
| 	g_setenv ("G_MESSAGES_DEBUG", "all", FALSE);
 | |
| 
 | |
| 	/* tests go here */
 | |
| 	g_test_add_func ("/dfu/patch", dfu_patch_func);
 | |
| 	g_test_add_func ("/dfu/patch{merges}", dfu_patch_merges_func);
 | |
| 	g_test_add_func ("/dfu/patch{apply}", dfu_patch_apply_func);
 | |
| 	g_test_add_func ("/dfu/enums", dfu_enums_func);
 | |
| 	g_test_add_func ("/dfu/target(DfuSe}", dfu_target_dfuse_func);
 | |
| 	g_test_add_func ("/dfu/cipher{xtea}", dfu_cipher_xtea_func);
 | |
| 	g_test_add_func ("/dfu/firmware{raw}", dfu_firmware_raw_func);
 | |
| 	g_test_add_func ("/dfu/firmware{dfu}", dfu_firmware_dfu_func);
 | |
| 	g_test_add_func ("/dfu/firmware{dfuse}", dfu_firmware_dfuse_func);
 | |
| 	g_test_add_func ("/dfu/firmware{xdfu}", dfu_firmware_xdfu_func);
 | |
| 	g_test_add_func ("/dfu/firmware{metadata}", dfu_firmware_metadata_func);
 | |
| 	return g_test_run ();
 | |
| }
 | |
| 
 |