Merge pull request #406 from pq-code-package/feat-code-size-measurement

Feat: port code size measurement feature from mlkem-native

Author: mkannwischer

Makefile

@@ -15,7 +15,9 @@
 	bench_components_44 bench_components_65 bench_components_87 bench_components \
 	run_bench_components_44 run_bench_components_65 run_bench_components_87 run_bench_components \
 	build test all \
-	clean quickcheck check-defined-CYCLES
+	clean quickcheck check-defined-CYCLES \
+	size_44 size_65 size_87 size \
+	run_size_44 run_size_65 run_size_87 run_size 
 
 SHELL := /bin/bash
 .DEFAULT_GOAL := build
@@ -151,6 +153,31 @@ run_bench_components: \
 	run_bench_components_65 .WAIT\
 	run_bench_components_87
 
+
+size_44: $(BUILD_DIR)/libmldsa44.a
+size_65: $(BUILD_DIR)/libmldsa65.a
+size_87: $(BUILD_DIR)/libmldsa87.a
+size: size_44 size_65 size_87
+
+run_size_44: size_44
+	$(Q)echo "size $(BUILD_DIR)/libmldsa44.a"
+	$(Q)$(SIZE) $(BUILD_DIR)/libmldsa44.a | (read header; echo "$$header"; awk '$$5 != 0' | sort -k5 -n -r)
+
+run_size_65: size_65
+	$(Q)echo "size $(BUILD_DIR)/libmldsa65.a"
+	$(Q)$(SIZE) $(BUILD_DIR)/libmldsa65.a | (read header; echo "$$header"; awk '$$5 != 0' | sort -k5 -n -r)
+
+run_size_87: size_87
+	$(Q)echo "size $(BUILD_DIR)/libmldsa87.a"
+	$(Q)$(SIZE) $(BUILD_DIR)/libmldsa87.a | (read header; echo "$$header"; awk '$$5 != 0' | sort -k5 -n -r)
+
+
+run_size: \
+	run_size_44 \
+	run_size_65 \
+	run_size_87
+
+
 clean:
 	-$(RM) -rf *.gcno *.gcda *.lcov *.o *.so
 	-$(RM) -rf $(BUILD_DIR)

scripts/tests

@@ -198,6 +198,7 @@ class TEST_TYPES(Enum):
     BENCH_COMPONENTS = 4
     ACVP = 5
     STACK = 6
+    SIZE = 7
 
     def is_benchmark(self):
         return self in [TEST_TYPES.BENCH, TEST_TYPES.BENCH_COMPONENTS]
@@ -234,6 +235,8 @@ class TEST_TYPES(Enum):
             return "ACVP Test"
         if self == TEST_TYPES.STACK:
             return "Stack Usage Test"
+        if self == TEST_TYPES.SIZE:
+            return "Measurement Code Size"
 
     def make_dir(self):
         return ""
@@ -251,6 +254,8 @@ class TEST_TYPES(Enum):
             return "acvp"
         if self == TEST_TYPES.STACK:
             return "stack"
+        if self == TEST_TYPES.SIZE:
+            return "size"
 
     def make_run_target(self, scheme):
         t = self.make_target()
@@ -604,6 +609,29 @@ class Tests:
 
         self.check_fail()
 
+    def size(self):
+
+        test_type = TEST_TYPES.SIZE
+
+        resultss = None
+
+        if self.do_opt_all():
+            self._compile_schemes(test_type, False)
+            if self.args.run:
+                self._run_schemes(test_type, False, suppress_output=False)
+            self._compile_schemes(test_type, True)
+            if self.args.run:
+                resultss = self._run_schemes(test_type, True, suppress_output=False)
+        else:
+            self._compile_schemes(test_type, self.do_opt())
+            if self.args.run:
+                resultss = self._run_schemes(
+                    test_type, self.do_opt(), suppress_output=False
+                )
+
+        if resultss is None:
+            self.check_fail()
+
     def all(self):
         func = self.args.func
         kat = self.args.kat
@@ -926,6 +954,11 @@ def cli():
         action="store_true",
         default=False,
     )
+    size_parser = cmd_subparsers.add_parser(
+        "size",
+        help="Run the code size measurement for all object file",
+        parents=[common_parser],
+    )
 
     # cbmc arguments
     cbmc_parser = cmd_subparsers.add_parser(
@@ -1040,6 +1073,8 @@ def cli():
         Tests(args).kat()
     elif args.cmd == "stack":
         Tests(args).stack()
+    elif args.cmd == "size":
+        Tests(args).size()
 
 
 if __name__ == "__main__":