Update assert on heap size

LedgerHQ · Oct 31, 2024 · 6ebdd16 · 6ebdd16
1 parent dcab944
commit 6ebdd16
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 10 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.18.1"
+version = "1.18.2"
 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.1" }
+ledger_secure_sdk_sys = { path = "../ledger_secure_sdk_sys", version = "1.5.2" }
 
 [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.1"
+version = "1.5.2"
 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,12 +600,11 @@ 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());
 
+        let heap_size_value = heap_size.parse::<u32>().unwrap();
+
         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"
+            heap_size_value >= 2048 && heap_size_value <= 24576,
+            "Invalid heap size: {heap_size}; Shall be included in [2048, 24576]"
         );
 
         // Generate the heap_size.rs file with the HEAP_SIZE value