feat(rust): to reduce memory fragmentation use mimalloc in command an…

…d stress-test
build-trust · Nov 26, 2024 · a545869 · a545869
1 parent a8acbe9
commit a545869
Show file tree

Hide file tree

Showing 5 changed files with 32 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/implementations/rust/ockam/ockam_command/Cargo.toml b/implementations/rust/ockam/ockam_command/Cargo.toml
@@ -63,6 +63,7 @@ hex = "0.4"
 indicatif = "0.17.8"
 indoc = "2.0.5"
 miette = { version = "7.2.0", features = ["fancy-no-backtrace"] }
+mimalloc = { version = "0.1", features = ["secure"] }
 minicbor = { version = "0.25.1", default-features = false, features = ["alloc", "derive"] }
 nix = { version = "0.29", features = ["signal"] }
 ockam = { path = "../ockam", version = "^0.141.0", features = ["software_vault"] }

diff --git a/implementations/rust/ockam/ockam_command/src/bin/ockam.rs b/implementations/rust/ockam/ockam_command/src/bin/ockam.rs
@@ -2,8 +2,12 @@
 // binary names. The issue is that we need to avoid the `ockam` binary colliding
 // with the `ockam` crate.
 
+use mimalloc::MiMalloc;
 use ockam_command::util::exitcode;
 
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 fn main() {
     if let Err(e) = ockam_command::entry_point::run() {
         // initialization errors are displayed here

diff --git a/tools/stress-test/Cargo.toml b/tools/stress-test/Cargo.toml
@@ -13,6 +13,7 @@ rust-crypto = ["ockam/rust-crypto"]
 
 [dependencies]
 clap = { version = "4.5", features = ["derive", "cargo"] }
+mimalloc = { version = "0.1", features = ["secure"] }
 ockam = { path = "../../implementations/rust/ockam/ockam" }
 ockam_api = { path = "../../implementations/rust/ockam/ockam_api" }
 ockam_core = { path = "../../implementations/rust/ockam/ockam_core" }

diff --git a/tools/stress-test/src/main.rs b/tools/stress-test/src/main.rs
@@ -23,6 +23,11 @@ mod execution;
 mod portal_simulator;
 mod stats;
 
+use mimalloc::MiMalloc;
+
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 #[derive(Debug, Args, Clone)]
 struct RunCommand {
     config: PathBuf,