test: unit test for hash command

Provide a unit test testing the hash command.

Signed-off-by: Heinrich Schuchardt <heinrich.schuchardt@canonical.com>

diff --git a/test/cmd/Makefile b/test/cmd/Makefile
index fe7a2165af22591da5c6bb5739cb051f5a772f32..1fd3e89efffb8685b6b2a302d578af810e399cd6 100644 (file)
--- a/test/cmd/Makefile
+++ b/test/cmd/Makefile
@@ -17,6 +17,7 @@ obj-$(CONFIG_CMD_ADDRMAP) += addrmap.o
 obj-$(CONFIG_CMD_BDI) += bdinfo.o
 obj-$(CONFIG_CMD_FDT) += fdt.o
 obj-$(CONFIG_CONSOLE_TRUETYPE) += font.o
+obj-$(CONFIG_CMD_HASH) += hash.o
 obj-$(CONFIG_CMD_HISTORY) += history.o
 obj-$(CONFIG_CMD_LOADM) += loadm.o
 obj-$(CONFIG_CMD_MEMINFO) += meminfo.o

diff --git a/test/cmd/hash.c b/test/cmd/hash.c
new file mode 100644 (file)
index 0000000..2fcec9c
--- /dev/null
+++ b/test/cmd/hash.c
@@ -0,0 +1,101 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Tests for hash command
+ *
+ * Copyright 2024, Heinrich Schuchardt <heinrich.schuchardt@canoncal.com>
+ */
+
+#include <command.h>
+#include <dm.h>
+#include <dm/test.h>
+#include <test/test.h>
+#include <test/ut.h>
+
+static int dm_test_cmd_hash_md5(struct unit_test_state *uts)
+{
+       if (!CONFIG_IS_ENABLED(MD5)) {
+               ut_assert(run_command("hash md5 $loadaddr 0", 0));
+
+               return 0;
+       }
+
+       ut_assertok(run_command("hash md5 $loadaddr 0", 0));
+       console_record_readline(uts->actual_str, sizeof(uts->actual_str));
+       ut_asserteq_ptr(uts->actual_str,
+                       strstr(uts->actual_str, "md5 for "));
+       ut_assert(strstr(uts->actual_str,
+                        "d41d8cd98f00b204e9800998ecf8427e"));
+       ut_assert_console_end();
+
+       ut_assertok(run_command("hash md5 $loadaddr 0 foo; echo $foo", 0));
+       console_record_readline(uts->actual_str, sizeof(uts->actual_str));
+       ut_asserteq_ptr(uts->actual_str,
+                       strstr(uts->actual_str, "md5 for "));
+       ut_assert(strstr(uts->actual_str,
+                        "d41d8cd98f00b204e9800998ecf8427e"));
+       ut_check_console_line(uts, "d41d8cd98f00b204e9800998ecf8427e");
+
+       if (!CONFIG_IS_ENABLED(HASH_VERIFY)) {
+               ut_assert(run_command("hash -v sha256 $loadaddr 0 foo", 0));
+               ut_check_console_line(uts, "hash - compute hash message digest");
+
+               return 0;
+       }
+
+       ut_assertok(run_command("hash -v md5 $loadaddr 0 foo", 0));
+       ut_assert_console_end();
+
+       env_set("foo", "ffffffffffffffffffffffffffffffff");
+       ut_assert(run_command("hash -v md5 $loadaddr 0 foo", 0));
+       console_record_readline(uts->actual_str, sizeof(uts->actual_str));
+       ut_assert(strstr(uts->actual_str, "!="));
+       ut_assert_console_end();
+
+       return 0;
+}
+DM_TEST(dm_test_cmd_hash_md5, UTF_CONSOLE);
+
+static int dm_test_cmd_hash_sha256(struct unit_test_state *uts)
+{
+       if (!CONFIG_IS_ENABLED(SHA256)) {
+               ut_assert(run_command("hash sha256 $loadaddr 0", 0));
+
+               return 0;
+       }
+
+       ut_assertok(run_command("hash sha256 $loadaddr 0", 0));
+       console_record_readline(uts->actual_str, sizeof(uts->actual_str));
+       ut_asserteq_ptr(uts->actual_str,
+                       strstr(uts->actual_str, "sha256 for "));
+       ut_assert(strstr(uts->actual_str,
+                        "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"));
+       ut_assert_console_end();
+
+       ut_assertok(run_command("hash sha256 $loadaddr 0 foo; echo $foo", 0));
+       console_record_readline(uts->actual_str, sizeof(uts->actual_str));
+       ut_asserteq_ptr(uts->actual_str,
+                       strstr(uts->actual_str, "sha256 for "));
+       ut_assert(strstr(uts->actual_str,
+                        "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"));
+       ut_check_console_line(uts,
+                             "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855");
+
+       if (!CONFIG_IS_ENABLED(HASH_VERIFY)) {
+               ut_assert(run_command("hash -v sha256 $loadaddr 0 foo", 0));
+               ut_check_console_line(uts, "hash - compute hash message digest");
+
+               return 0;
+       }
+
+       ut_assertok(run_command("hash -v sha256 $loadaddr 0 foo", 0));
+       ut_assert_console_end();
+
+       env_set("foo", "ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff");
+       ut_assert(run_command("hash -v sha256 $loadaddr 0 foo", 0));
+       console_record_readline(uts->actual_str, sizeof(uts->actual_str));
+       ut_assert(strstr(uts->actual_str, "!="));
+       ut_assert_console_end();
+
+       return 0;
+}
+DM_TEST(dm_test_cmd_hash_sha256, UTF_CONSOLE);