Simplify use of alloc

cryspen · Jan 15, 2025 · 3fc1be5 · 3fc1be5
1 parent b55be22
commit 3fc1be5
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 16 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -34,6 +34,9 @@ extern crate std;
 #[cfg(not(feature = "std"))]
 extern crate alloc;
 
+#[cfg(not(feature = "std"))]
+use alloc as std;
+
 mod tls13formats;
 mod tls13handshake;
 mod tls13record;

diff --git a/src/test_utils.rs b/src/test_utils.rs
@@ -1,9 +1,5 @@
+use crate::std::vec::Vec;
 use rand::{CryptoRng, RngCore};
-#[cfg(feature = "std")]
-use std::vec::Vec;
-
-#[cfg(not(feature = "std"))]
-use alloc::vec::Vec;
 
 pub struct TestRng {
     bytes: Vec<u8>,

diff --git a/src/tls13crypto.rs b/src/tls13crypto.rs
@@ -7,13 +7,9 @@ use libcrux::{
 use libcrux_kem::{Ct, PrivateKey, PublicKey};
 use rand::{CryptoRng, RngCore};
 
-#[cfg(feature = "std")]
-use std::{format, vec::Vec};
+use crate::std::{format, vec::Vec};
 
-#[cfg(not(feature = "std"))]
-use alloc::{format, vec::Vec};
-
-use std::fmt::Display;
+use crate::std::fmt::Display;
 
 use crate::tls13utils::{
     check_mem, eq, length_u16_encoded, tlserr, Bytes, Error, TLSError, CRYPTO_ERROR,

diff --git a/src/tls13utils.rs b/src/tls13utils.rs
@@ -3,11 +3,7 @@ use core::ops::Range;
 #[cfg(feature = "hax-fstar")]
 use hax_lib_macros::{attributes, requires};
 
-#[cfg(feature = "std")]
-use std::{format, string::String, vec, vec::Vec};
-
-#[cfg(not(feature = "std"))]
-use alloc::{format, string::String, vec, vec::Vec};
+use crate::std::{format, string::String, vec, vec::Vec};
 
 // FIXME: NOT HACSPEC | ONLY FOR DEBUGGING
 pub(crate) fn parse_failed() -> TLSError {