tfhe 0.7.2 → 0.7.4

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "tfhe",
   "description": "TFHE-rs is a fully homomorphic encryption (FHE) library that implements Zama's variant of TFHE.",
-  "version": "0.7.2",
+  "version": "0.7.4",
   "license": "BSD-3-Clause-Clear",
   "repository": {
     "type": "git",
@@ -10,7 +10,8 @@
   "files": [
     "tfhe_bg.wasm",
     "tfhe.js",
-    "tfhe.d.ts"
+    "tfhe.d.ts",
+    "snippets"
   ],
   "module": "tfhe.js",
   "homepage": "https://zama.ai/",
@@ -25,4 +26,4 @@
     "fhe",
     "cryptography"
   ]
-}
+}

package/snippets/wasm-bindgen-rayon-3e04391371ad0a8e/src/workerHelpers.js

@@ -0,0 +1,56 @@
+/*
+ * Copyright 2022 Google Inc. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Note: this is never used, but necessary to prevent a bug in Firefox
+// (https://bugzilla.mozilla.org/show_bug.cgi?id=1702191) where it collects
+// Web Workers that have a shared WebAssembly memory with the main thread,
+// but are not explicitly rooted via a `Worker` instance.
+//
+// By storing them in a variable, we can keep `Worker` objects around and
+// prevent them from getting GC-d.
+let _workers;
+
+export async function startWorkers(module, memory, builder) {
+  if (builder.numThreads() === 0) {
+    throw new Error(`num_threads must be > 0.`);
+  }
+
+  const workerInit = {
+    module,
+    memory,
+    receiver: builder.receiver()
+  };
+
+  _workers = await Promise.all(
+    Array.from({ length: builder.numThreads() }, async () => {
+      // Self-spawn into a new Worker.
+      //
+      // TODO: while `new URL('...', import.meta.url) becomes a semi-standard
+      // way to get asset URLs relative to the module across various bundlers
+      // and browser, ideally we should switch to `import.meta.resolve`
+      // once it becomes a standard.
+      const worker = new Worker(
+        new URL('./workerHelpers.worker.js', import.meta.url),
+        {
+          type: 'module'
+        }
+      );
+      worker.postMessage(workerInit);
+      await new Promise(resolve =>
+        worker.addEventListener('message', resolve, { once: true })
+      );
+      return worker;
+    })
+  );
+  builder.build();
+}

package/snippets/wasm-bindgen-rayon-3e04391371ad0a8e/src/workerHelpers.worker.js

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2022 Google Inc. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Note: our JS should have been generated in
+// `[out-dir]/snippets/wasm-bindgen-rayon-[hash]/workerHelpers.worker.js`,
+// resolve the main module via `../../..`.
+//
+// This might need updating if the generated structure changes on wasm-bindgen
+// side ever in the future, but works well with bundlers today. The whole
+// point of this crate, after all, is to abstract away unstable features
+// and temporary bugs so that you don't need to deal with them in your code.
+import initWbg, { wbg_rayon_start_worker } from '../../../';
+
+onmessage = async ({ data: { module, memory, receiver } }) => {
+  await initWbg(module, memory);
+  postMessage(true);
+  wbg_rayon_start_worker(receiver);
+};