summaryrefslogtreecommitdiffstats
path: root/src/test/test-gpt.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/test-gpt.c')
-rw-r--r--src/test/test-gpt.c111
1 files changed, 111 insertions, 0 deletions
diff --git a/src/test/test-gpt.c b/src/test/test-gpt.c
new file mode 100644
index 0000000..fa5923e
--- /dev/null
+++ b/src/test/test-gpt.c
@@ -0,0 +1,111 @@
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+
+#include "architecture.h"
+#include "glyph-util.h"
+#include "gpt.h"
+#include "log.h"
+#include "pretty-print.h"
+#include "strv.h"
+#include "terminal-util.h"
+#include "tests.h"
+
+TEST(gpt_types_against_architectures) {
+ int r;
+
+ /* Dumps a table indicating for which architectures we know we have matching GPT partition
+ * types. Also validates whether we can properly categorize the entries. */
+
+ FOREACH_STRING(prefix, "root-", "usr-")
+ for (Architecture a = 0; a < _ARCHITECTURE_MAX; a++)
+ FOREACH_STRING(suffix, "", "-verity", "-verity-sig") {
+ _cleanup_free_ char *joined = NULL;
+ GptPartitionType type;
+
+ joined = strjoin(prefix, architecture_to_string(a), suffix);
+ if (!joined)
+ return (void) log_oom();
+
+ r = gpt_partition_type_from_string(joined, &type);
+ if (r < 0) {
+ printf("%s %s\n", RED_CROSS_MARK(), joined);
+ continue;
+ }
+
+ printf("%s %s\n", GREEN_CHECK_MARK(), joined);
+
+ if (streq(prefix, "root-") && streq(suffix, ""))
+ assert_se(type.designator == PARTITION_ROOT);
+ if (streq(prefix, "root-") && streq(suffix, "-verity"))
+ assert_se(type.designator == PARTITION_ROOT_VERITY);
+ if (streq(prefix, "usr-") && streq(suffix, ""))
+ assert_se(type.designator == PARTITION_USR);
+ if (streq(prefix, "usr-") && streq(suffix, "-verity"))
+ assert_se(type.designator == PARTITION_USR_VERITY);
+
+ assert_se(type.arch == a);
+ }
+}
+
+TEST(verity_mappings) {
+ for (PartitionDesignator p = 0; p < _PARTITION_DESIGNATOR_MAX; p++) {
+ PartitionDesignator q;
+
+ q = partition_verity_of(p);
+ assert_se(q < 0 || partition_verity_to_data(q) == p);
+
+ q = partition_verity_sig_of(p);
+ assert_se(q < 0 || partition_verity_sig_to_data(q) == p);
+
+ q = partition_verity_to_data(p);
+ assert_se(q < 0 || partition_verity_of(q) == p);
+
+ q = partition_verity_sig_to_data(p);
+ assert_se(q < 0 || partition_verity_sig_of(q) == p);
+ }
+}
+
+TEST(type_alias_same) {
+ /* Check that the partition type table is consistent, i.e. all aliases of the same partition type
+ * carry the same metadata */
+
+ for (const GptPartitionType *t = gpt_partition_type_table; t->name; t++) {
+ GptPartitionType x, y;
+
+ x = gpt_partition_type_from_uuid(t->uuid); /* search first by uuid */
+ assert_se(gpt_partition_type_from_string(t->name, &y) >= 0); /* search first by name */
+
+ assert_se(t->arch == x.arch);
+ assert_se(t->arch == y.arch);
+ assert_se(t->designator == x.designator);
+ assert_se(t->designator == y.designator);
+ }
+}
+
+TEST(override_architecture) {
+ GptPartitionType x, y;
+
+ assert_se(gpt_partition_type_from_string("root-x86-64", &x) >= 0);
+ assert_se(x.arch == ARCHITECTURE_X86_64);
+
+ assert_se(gpt_partition_type_from_string("root-arm64", &y) >= 0);
+ assert(y.arch == ARCHITECTURE_ARM64);
+
+ x = gpt_partition_type_override_architecture(x, ARCHITECTURE_ARM64);
+ assert_se(x.arch == y.arch);
+ assert_se(x.designator == y.designator);
+ assert_se(sd_id128_equal(x.uuid, y.uuid));
+ assert_se(streq(x.name, y.name));
+
+ /* If the partition type does not have an architecture, nothing should change. */
+
+ assert_se(gpt_partition_type_from_string("esp", &x) >= 0);
+ y = x;
+
+ x = gpt_partition_type_override_architecture(x, ARCHITECTURE_ARM64);
+ assert_se(x.arch == y.arch);
+ assert_se(x.designator == y.designator);
+ assert_se(sd_id128_equal(x.uuid, y.uuid));
+ assert_se(streq(x.name, y.name));
+}
+
+DEFINE_TEST_MAIN(LOG_INFO);