feat(wasm): added support for wasm compilation without Javascript API

PedroMBB · PedroMBB · commit 6254fbec6805 · 2024-01-23T14:50:16.000Z
diff --git a/tfhe/Cargo.toml b/tfhe/Cargo.toml
@@ -81,6 +81,9 @@ serde-wasm-bindgen = { version = "0.6.0", optional = true }
 getrandom = { version = "0.2.8", optional = true }
 bytemuck = "1.13.1"
 
+[target.'cfg(target_arch = "wasm32")'.dependencies.getrandom]
+optional = false
+
 [features]
 boolean = []
 shortint = []
diff --git a/tfhe/src/boolean/engine/bootstrapping.rs b/tfhe/src/boolean/engine/bootstrapping.rs
@@ -336,7 +336,7 @@ impl Bootstrapper {
     }
 
     pub(crate) fn new_compressed_server_key(&mut self, cks: &ClientKey) -> CompressedServerKey {
-        #[cfg(not(feature = "__wasm_api"))]
+        #[cfg(not(target_arch = "wasm32"))]
         let bootstrapping_key = par_allocate_and_generate_new_seeded_lwe_bootstrap_key(
             &cks.lwe_secret_key,
             &cks.glwe_secret_key,
@@ -347,7 +347,7 @@ impl Bootstrapper {
             &mut self.seeder,
         );
 
-        #[cfg(feature = "__wasm_api")]
+        #[cfg(target_arch = "wasm32")]
         let bootstrapping_key = allocate_and_generate_new_seeded_lwe_bootstrap_key(
             &cks.lwe_secret_key,
             &cks.glwe_secret_key,
diff --git a/tfhe/src/boolean/engine/mod.rs b/tfhe/src/boolean/engine/mod.rs
@@ -129,7 +129,7 @@ impl BooleanEngine {
             lwe_sk.lwe_dimension().to_lwe_size().0 * LOG2_Q_32 + 128,
         );
 
-        #[cfg(not(feature = "__wasm_api"))]
+        #[cfg(not(target_arch = "wasm32"))]
         let lwe_public_key: LwePublicKeyOwned<u32> = par_allocate_and_generate_new_lwe_public_key(
             &lwe_sk,
             zero_encryption_count,
@@ -138,7 +138,7 @@ impl BooleanEngine {
             &mut self.encryption_generator,
         );
 
-        #[cfg(feature = "__wasm_api")]
+        #[cfg(target_arch = "wasm32")]
         let lwe_public_key: LwePublicKeyOwned<u32> = allocate_and_generate_new_lwe_public_key(
             &lwe_sk,
             zero_encryption_count,
@@ -172,7 +172,7 @@ impl BooleanEngine {
             lwe_sk.lwe_dimension().to_lwe_size().0 * LOG2_Q_32 + 128,
         );
 
-        #[cfg(not(feature = "__wasm_api"))]
+        #[cfg(not(target_arch = "wasm32"))]
         let compressed_lwe_public_key = par_allocate_and_generate_new_seeded_lwe_public_key(
             &lwe_sk,
             zero_encryption_count,
@@ -181,7 +181,7 @@ impl BooleanEngine {
             &mut self.bootstrapper.seeder,
         );
 
-        #[cfg(feature = "__wasm_api")]
+        #[cfg(target_arch = "wasm32")]
         let compressed_lwe_public_key = allocate_and_generate_new_seeded_lwe_public_key(
             &lwe_sk,
             zero_encryption_count,
diff --git a/tfhe/src/core_crypto/seeders.rs b/tfhe/src/core_crypto/seeders.rs
@@ -5,14 +5,14 @@
 //! for cryptographically secure pseudo random number generators.
 
 pub use crate::core_crypto::commons::math::random::Seeder;
-#[cfg(all(target_os = "macos", not(feature = "__wasm_api")))]
+#[cfg(all(target_os = "macos", not(target_arch = "wasm32")))]
 pub use concrete_csprng::seeders::AppleSecureEnclaveSeeder;
 #[cfg(feature = "seeder_x86_64_rdseed")]
 pub use concrete_csprng::seeders::RdseedSeeder;
 #[cfg(feature = "seeder_unix")]
 pub use concrete_csprng::seeders::UnixSeeder;
 
-#[cfg(feature = "__wasm_api")]
+#[cfg(target_arch = "wasm32")]
 mod wasm_seeder {
     use crate::core_crypto::commons::math::random::{Seed, Seeder};
     // This is used for web interfaces
@@ -73,7 +73,7 @@ pub fn new_seeder() -> Box<dyn Seeder> {
 
     let err_msg;
 
-    #[cfg(not(feature = "__wasm_api"))]
+    #[cfg(not(target_arch = "wasm32"))]
     {
         #[cfg(feature = "seeder_x86_64_rdseed")]
         {
@@ -110,7 +110,7 @@ pub fn new_seeder() -> Box<dyn Seeder> {
         }
     }
 
-    #[cfg(feature = "__wasm_api")]
+    #[cfg(target_arch = "wasm32")]
     {
         if seeder.is_none() && wasm_seeder::WasmSeeder::is_available() {
             seeder = Some(Box::new(wasm_seeder::WasmSeeder {}));
diff --git a/tfhe/src/shortint/ciphertext/mod.rs b/tfhe/src/shortint/ciphertext/mod.rs
@@ -600,14 +600,14 @@ impl CompactCiphertextList {
         );
 
         // No parallelism allowed
-        #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]
+        #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]
         {
             use crate::core_crypto::prelude::expand_lwe_compact_ciphertext_list;
             expand_lwe_compact_ciphertext_list(&mut output_lwe_ciphertext_list, &self.ct_list);
         }
 
         // Parallelism allowed
-        #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]
+        #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]
         {
             use crate::core_crypto::prelude::par_expand_lwe_compact_ciphertext_list;
             par_expand_lwe_compact_ciphertext_list(&mut output_lwe_ciphertext_list, &self.ct_list);
diff --git a/tfhe/src/shortint/engine/public_side.rs b/tfhe/src/shortint/engine/public_side.rs
@@ -39,7 +39,7 @@ impl ShortintEngine {
             secret_encryption_key.lwe_dimension().to_lwe_size(),
         );
 
-        #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]
+        #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]
         let lwe_public_key = par_allocate_and_generate_new_lwe_public_key(
             &secret_encryption_key,
             zero_encryption_count,
@@ -48,7 +48,7 @@ impl ShortintEngine {
             &mut self.encryption_generator,
         );
 
-        #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]
+        #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]
         let lwe_public_key = allocate_and_generate_new_lwe_public_key(
             &secret_encryption_key,
             zero_encryption_count,
@@ -85,7 +85,7 @@ impl ShortintEngine {
             secret_encryption_key.lwe_dimension().to_lwe_size(),
         );
 
-        #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]
+        #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]
         let compressed_public_key = par_allocate_and_generate_new_seeded_lwe_public_key(
             &secret_encryption_key,
             zero_encryption_count,
@@ -94,7 +94,7 @@ impl ShortintEngine {
             &mut self.seeder,
         );
 
-        #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]
+        #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]
         let compressed_public_key = allocate_and_generate_new_seeded_lwe_public_key(
             &secret_encryption_key,
             zero_encryption_count,
diff --git a/tfhe/src/shortint/engine/server_side.rs b/tfhe/src/shortint/engine/server_side.rs
@@ -193,7 +193,7 @@ impl ShortintEngine {
     ) -> CompressedServerKey {
         let bootstrapping_key = match cks.parameters.pbs_parameters().unwrap() {
             crate::shortint::PBSParameters::PBS(pbs_params) => {
-                #[cfg(not(feature = "__wasm_api"))]
+                #[cfg(not(target_arch = "wasm32"))]
                 let bootstrapping_key = par_allocate_and_generate_new_seeded_lwe_bootstrap_key(
                     &cks.small_lwe_secret_key(),
                     &cks.glwe_secret_key,
@@ -204,7 +204,7 @@ impl ShortintEngine {
                     &mut self.seeder,
                 );
 
-                #[cfg(feature = "__wasm_api")]
+                #[cfg(target_arch = "wasm32")]
                 let bootstrapping_key = allocate_and_generate_new_seeded_lwe_bootstrap_key(
                     &cks.small_lwe_secret_key(),
                     &cks.glwe_secret_key,
@@ -218,7 +218,7 @@ impl ShortintEngine {
                 ShortintCompressedBootstrappingKey::Classic(bootstrapping_key)
             }
             crate::shortint::PBSParameters::MultiBitPBS(pbs_params) => {
-                #[cfg(not(feature = "__wasm_api"))]
+                #[cfg(not(target_arch = "wasm32"))]
                 let bootstrapping_key =
                     par_allocate_and_generate_new_seeded_lwe_multi_bit_bootstrap_key(
                         &cks.small_lwe_secret_key(),
@@ -231,7 +231,7 @@ impl ShortintEngine {
                         &mut self.seeder,
                     );
 
-                #[cfg(feature = "__wasm_api")]
+                #[cfg(target_arch = "wasm32")]
                 let bootstrapping_key =
                     allocate_and_generate_new_seeded_lwe_multi_bit_bootstrap_key(
                         &cks.small_lwe_secret_key(),
diff --git a/tfhe/src/shortint/public_key/compact.rs b/tfhe/src/shortint/public_key/compact.rs
@@ -212,7 +212,7 @@ impl CompactPublicKey {
         );
 
         // No parallelism allowed
-        #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]
+        #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]
         {
             use crate::core_crypto::prelude::encrypt_lwe_compact_ciphertext_list_with_compact_public_key;
             ShortintEngine::with_thread_local_mut(|engine| {
@@ -229,7 +229,7 @@ impl CompactPublicKey {
         }
 
         // Parallelism allowed
-        #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]
+        #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]
         {
             use crate::core_crypto::prelude::par_encrypt_lwe_compact_ciphertext_list_with_compact_public_key;
             ShortintEngine::with_thread_local_mut(|engine| {
diff --git a/tfhe/src/shortint/public_key/standard.rs b/tfhe/src/shortint/public_key/standard.rs
@@ -256,12 +256,12 @@ impl From<CompressedPublicKey> for PublicKey {
     fn from(compressed_public_key: CompressedPublicKey) -> Self {
         let parameters = compressed_public_key.parameters;
 
-        #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]
+        #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]
         let decompressed_public_key = compressed_public_key
             .lwe_public_key
             .par_decompress_into_lwe_public_key();
 
-        #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]
+        #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]
         let decompressed_public_key = compressed_public_key
             .lwe_public_key
             .decompress_into_lwe_public_key();

Original file line number	Diff line number	Diff line change
`@@ -5,14 +5,14 @@`
`5`	`5`	`//! for cryptographically secure pseudo random number generators.`
`6`	`6`
`7`	`7`	`pub use crate::core_crypto::commons::math::random::Seeder;`
`8`		`-#[cfg(all(target_os = "macos", not(feature = "__wasm_api")))]`
	`8`	`+#[cfg(all(target_os = "macos", not(target_arch = "wasm32")))]`
`9`	`9`	`pub use concrete_csprng::seeders::AppleSecureEnclaveSeeder;`
`10`	`10`	`#[cfg(feature = "seeder_x86_64_rdseed")]`
`11`	`11`	`pub use concrete_csprng::seeders::RdseedSeeder;`
`12`	`12`	`#[cfg(feature = "seeder_unix")]`
`13`	`13`	`pub use concrete_csprng::seeders::UnixSeeder;`
`14`	`14`
`15`		`-#[cfg(feature = "__wasm_api")]`
	`15`	`+#[cfg(target_arch = "wasm32")]`
`16`	`16`	`mod wasm_seeder {`
`17`	`17`	`use crate::core_crypto::commons::math::random::{Seed, Seeder};`
`18`	`18`	`// This is used for web interfaces`
`@@ -73,7 +73,7 @@ pub fn new_seeder() -> Box<dyn Seeder> {`
`73`	`73`
`74`	`74`	`let err_msg;`
`75`	`75`
`76`		`- #[cfg(not(feature = "__wasm_api"))]`
	`76`	`+ #[cfg(not(target_arch = "wasm32"))]`
`77`	`77`	`{`
`78`	`78`	`#[cfg(feature = "seeder_x86_64_rdseed")]`
`79`	`79`	`{`
`@@ -110,7 +110,7 @@ pub fn new_seeder() -> Box<dyn Seeder> {`
`110`	`110`	`}`
`111`	`111`	`}`
`112`	`112`
`113`		`- #[cfg(feature = "__wasm_api")]`
	`113`	`+ #[cfg(target_arch = "wasm32")]`
`114`	`114`	`{`
`115`	`115`	`if seeder.is_none() && wasm_seeder::WasmSeeder::is_available() {`
`116`	`116`	`seeder = Some(Box::new(wasm_seeder::WasmSeeder {}));`
Original file line number	Diff line number	Diff line change
`@@ -600,14 +600,14 @@ impl CompactCiphertextList {`
`600`	`600`	`);`
`601`	`601`
`602`	`602`	`// No parallelism allowed`
`603`		`- #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]`
	`603`	`+ #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]`
`604`	`604`	`{`
`605`	`605`	`use crate::core_crypto::prelude::expand_lwe_compact_ciphertext_list;`
`606`	`606`	`expand_lwe_compact_ciphertext_list(&mut output_lwe_ciphertext_list, &self.ct_list);`
`607`	`607`	`}`
`608`	`608`
`609`	`609`	`// Parallelism allowed`
`610`		`- #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]`
	`610`	`+ #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]`
`611`	`611`	`{`
`612`	`612`	`use crate::core_crypto::prelude::par_expand_lwe_compact_ciphertext_list;`
`613`	`613`	`par_expand_lwe_compact_ciphertext_list(&mut output_lwe_ciphertext_list, &self.ct_list);`
Original file line number	Diff line number	Diff line change
`@@ -212,7 +212,7 @@ impl CompactPublicKey {`
`212`	`212`	`);`
`213`	`213`
`214`	`214`	`// No parallelism allowed`
`215`		`- #[cfg(all(feature = "__wasm_api", not(feature = "parallel-wasm-api")))]`
	`215`	`+ #[cfg(all(target_arch = "wasm32", not(feature = "parallel-wasm-api")))]`
`216`	`216`	`{`
`217`	`217`	`use crate::core_crypto::prelude::encrypt_lwe_compact_ciphertext_list_with_compact_public_key;`
`218`	`218`	`ShortintEngine::with_thread_local_mut(\|engine\| {`
`@@ -229,7 +229,7 @@ impl CompactPublicKey {`
`229`	`229`	`}`
`230`	`230`
`231`	`231`	`// Parallelism allowed`
`232`		`- #[cfg(any(not(feature = "__wasm_api"), feature = "parallel-wasm-api"))]`
	`232`	`+ #[cfg(any(not(target_arch = "wasm32"), feature = "parallel-wasm-api"))]`
`233`	`233`	`{`
`234`	`234`	`use crate::core_crypto::prelude::par_encrypt_lwe_compact_ciphertext_list_with_compact_public_key;`
`235`	`235`	`ShortintEngine::with_thread_local_mut(\|engine\| {`