mirror of
https://git.proxmox.com/git/fwupd
synced 2025-05-30 01:13:49 +00:00
107 lines
2.9 KiB
C
107 lines
2.9 KiB
C
/* -*- mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
|
|
*
|
|
* Copyright (C) 2017 Intel Corporation.
|
|
*
|
|
* Licensed under the GNU General Public License Version 2
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <string.h>
|
|
#include <glib.h>
|
|
|
|
#include "fu-thunderbolt-image.h"
|
|
#include "fu-plugin-vfuncs.h"
|
|
#include "fu-test.h"
|
|
|
|
static gsize
|
|
read_farb_pointer (gchar *image)
|
|
{
|
|
gsize ret = 0;
|
|
memcpy (&ret, image, 3);
|
|
ret = GSIZE_FROM_LE (ret);
|
|
if (ret != 0 && ret != 0xFFFFFF)
|
|
return ret;
|
|
ret = 0;
|
|
memcpy (&ret, image + 0x1000, 3);
|
|
ret = GSIZE_FROM_LE (ret);
|
|
g_assert (ret != 0 && ret != 0xFFFFFF);
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
g_autoptr(GError) error = NULL;
|
|
g_autoptr(GFile) fw_file = NULL;
|
|
gboolean ok;
|
|
gsize len;
|
|
gchar *data = NULL;
|
|
g_autoptr(GBytes) image = NULL;
|
|
g_autoptr(GBytes) controller = NULL;
|
|
FuPluginValidation validation;
|
|
|
|
if (argc < 2 || argc > 3) {
|
|
g_print ("Usage: %s <filename> [<controller>]\n", argv[0]);
|
|
g_print ("Runs image validation on 'filename', comparing it to itself\n"
|
|
"after removing the headers or to 'controller' if given\n");
|
|
return 1;
|
|
}
|
|
|
|
fw_file = g_file_new_for_path (argv[1]);
|
|
g_assert_nonnull (fw_file);
|
|
|
|
ok = g_file_load_contents (fw_file, NULL, &data, &len, NULL, &error);
|
|
g_assert_no_error (error);
|
|
g_assert_true (ok);
|
|
|
|
image = g_bytes_new_take (data, len);
|
|
|
|
if (argc == 2) {
|
|
gssize header_size = read_farb_pointer (data);
|
|
g_assert_cmpuint (header_size, !=, 0);
|
|
g_assert_cmpuint (header_size, <, len);
|
|
|
|
controller = g_bytes_new_from_bytes (image, header_size, len - header_size);
|
|
} else {
|
|
g_autoptr(GFile) controller_file = NULL;
|
|
gsize controller_len;
|
|
gchar *controller_data = NULL;
|
|
|
|
controller_file = g_file_new_for_path (argv[2]);
|
|
g_assert_nonnull (controller_file);
|
|
|
|
ok = g_file_load_contents (controller_file,
|
|
NULL,
|
|
&controller_data,
|
|
&controller_len,
|
|
NULL,
|
|
&error);
|
|
g_assert_no_error (error);
|
|
g_assert_true (ok);
|
|
|
|
controller = g_bytes_new_take (controller_data, controller_len);
|
|
}
|
|
|
|
validation = fu_plugin_thunderbolt_validate_image (controller, image, &error);
|
|
g_assert_no_error (error);
|
|
g_assert_cmpint (validation, ==, VALIDATION_PASSED);
|
|
|
|
g_print ("test passed\n");
|
|
return 0;
|
|
}
|