Merge pull request #200 from LedgerHQ/y333_20241010/assert_heap_size

Assert Heap Size
LedgerHQ · Oct 10, 2024 · bc52315 · bc52315
2 parents 575631d + 337973c
commit bc52315
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/ledger_device_sdk/Cargo.toml b/ledger_device_sdk/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ledger_device_sdk"
-version = "1.17.3"
+version = "1.17.4"
 authors = ["yhql", "yogh333", "agrojean-ledger", "kingofpayne"]
 edition = "2021"
 license.workspace = true
@@ -21,7 +21,7 @@ rand_core = { version = "0.6.3", default_features = false }
 zeroize = { version = "1.6.0", default_features = false }
 numtoa = "0.2.4"
 const-zero = "0.1.1"
-ledger_secure_sdk_sys = { path = "../ledger_secure_sdk_sys", version = "1.5.0" }
+ledger_secure_sdk_sys = { path = "../ledger_secure_sdk_sys", version = "1.5.1" }
 
 [features]
 speculos = []

diff --git a/ledger_secure_sdk_sys/Cargo.toml b/ledger_secure_sdk_sys/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ledger_secure_sdk_sys"
-version = "1.5.0"
+version = "1.5.1"
 authors = ["yhql", "agrojean-ledger"]
 edition = "2021"
 license.workspace = true

diff --git a/ledger_secure_sdk_sys/build.rs b/ledger_secure_sdk_sys/build.rs
@@ -600,6 +600,14 @@ impl SDKBuilder {
         // Read the HEAP_SIZE environment variable, default to 8192 if not set
         let heap_size = env::var("HEAP_SIZE").unwrap_or_else(|_| "8192".to_string());
 
+        assert!(
+            match heap_size.as_str() {
+                "2048" | "4096" | "8192" | "16384" | "24576" => true,
+                _ => false,
+            },
+            "Invalid heap size: {heap_size}; Authorized sizes are 2048, 4096, 8192, 16384, 24576"
+        );
+
         // Generate the heap_size.rs file with the HEAP_SIZE value
         let out_dir = env::var("OUT_DIR").unwrap();
         let dest_path = Path::new(&out_dir).join("heap_size.rs");