/*
 * Copyright (C) 2017 Richard Hughes <richard@hughsie.com>
 *
 * SPDX-License-Identifier: LGPL-2.1+
 */

#include "config.h"

#include <fwupd.h>
#include <string.h>

#include "fu-nitrokey-common.h"

static void
fu_nitrokey_version_test(void)
{
	/* use the Nitrokey Storage v0.53 status response for test, CRC 0xa2762d14 */
	NitrokeyGetDeviceStatusPayload payload;
	NitrokeyHidResponse res;
	guint32 crc_tmp;
	/* 65 bytes of response from HIDAPI; first byte is always 0 */
	const guint8 buf[] = {/*0x00,*/
			      0x00, 0x2e, 0xef, 0xc4, 0x9b, 0x84, 0x00, 0x00, 0x00, 0x00, 0x00,
			      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0c, 0x2e,
			      0x01, 0x00, 0x00, 0x00, 0x03, 0x00, 0x1c, 0x18, 0x33, 0x00, 0x00,
			      0x00, 0x35, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x45, 0x24, 0xf1,
			      0x4c, 0x01, 0x00, 0x03, 0x03, 0xc7, 0x37, 0x00, 0x00, 0x00, 0x00,
			      0x00, 0x00, 0x00, 0x00, 0x00, 0x14, 0x2d, 0x76, 0xa2};

	/* testing the whole path, as in fu_nitrokey_device_setup()*/
	memcpy(&res, buf, sizeof(buf));
	memcpy(&payload, &res.payload, sizeof(payload));

	/* verify the version number */
	g_assert_cmpint(payload.VersionMajor, ==, 0);
	g_assert_cmpint(payload.VersionMinor, ==, 53);
	g_assert_cmpint(buf[34], ==, payload.VersionMinor);
	g_assert_cmpint(payload.VersionBuildIteration, ==, 0);

	/* verify the response checksum */
	crc_tmp = fu_nitrokey_perform_crc32(buf, sizeof(res) - 4);
	g_assert_cmpint(GUINT32_FROM_LE(res.crc), ==, crc_tmp);
}

static void
fu_nitrokey_version_test_static(void)
{
	/* use static response from numbered bytes, to make sure fields occupy
	 * expected bytes */
	NitrokeyGetDeviceStatusPayload payload;
	NitrokeyHidResponse res;

	const guint8 buf[] = {
	    0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C,
	    0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19,
	    0x1A, 0x1B, 0x1C, 0x1D, 0x1E, 0x1F, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26,
	    0x27, 0x28, 0x29, 0x2A, 0x2B, 0x2C, 0x2D, 0x2E, 0x2F, 0x30, 0x31, 0x32, 0x33,
	    0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3A, 0x3B, 0x3C, 0x3D, 0x3E, 0x3F,
	};
	memcpy(&res, buf, sizeof(buf));
	memcpy(&payload, &res.payload, sizeof(payload));
	g_assert_cmpint(payload.VersionMajor, ==, 33); /* 0x1a */
	g_assert_cmpint(payload.VersionMinor, ==, 34); /* 0x1b */
	g_assert_cmpint(buf[34], ==, 34);
}

static void
fu_nitrokey_func(void)
{
	const guint8 buf[] = {0x00,
			      0x01,
			      0x02,
			      0x03,
			      0x04,
			      0x05,
			      0x06,
			      0x07,
			      0x08,
			      0x09,
			      0x0a,
			      0x0b,
			      0x0c,
			      0x0d,
			      0x0e,
			      0x0f};
	g_assert_cmpint(fu_nitrokey_perform_crc32(buf, 16), ==, 0x081B46CA);
	g_assert_cmpint(fu_nitrokey_perform_crc32(buf, 15), ==, 0xED7320AB);
}

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);

	/* tests go here */
	g_test_add_func("/fwupd/nitrokey", fu_nitrokey_func);
	g_test_add_func("/fwupd/nitrokey-version-static", fu_nitrokey_version_test_static);
	g_test_add_func("/fwupd/nitrokey-version", fu_nitrokey_version_test);
	return g_test_run();
}
