Merge pull request #95 from zama-ai/squashed-proven-input-mechanism

feat: add proven ciphertext
zama-ai · Oct 4, 2024 · ee4a4dc · ee4a4dc
2 parents 19fd3b8 + cb71f63
commit ee4a4dc
Show file tree

Hide file tree

Showing 46 changed files with 1,836 additions and 6,871 deletions.
diff --git a/config/rollup.config.js b/config/rollup.config.js
@@ -1,28 +1,18 @@
 import { createRequire } from 'node:module';
 
+import json from '@rollup/plugin-json';
+import url from '@rollup/plugin-url';
 import { wasm } from '@rollup/plugin-wasm';
 import typescript from '@rollup/plugin-typescript';
 import replace from '@rollup/plugin-replace';
 import resolve from '@rollup/plugin-node-resolve';
 import commonjs from '@rollup/plugin-commonjs';
-import copy from 'rollup-plugin-copy';
 import nodePolyfills from 'rollup-plugin-polyfill-node';
 
 const require = createRequire(import.meta.url);
 
 const nodePlugins = [
-  copy({
-    targets: [
-      {
-        src: './src/kms/node/*',
-        dest: 'lib/kms/node',
-      },
-      {
-        src: './src/kms/node/kms_lib_bg.wasm',
-        dest: 'lib/',
-      },
-    ],
-  }),
+  json(),
   wasm(),
   commonjs(),
   typescript({
@@ -31,19 +21,13 @@ const nodePlugins = [
 ];
 
 const webPlugins = [
-  copy({
-    targets: [
-      {
-        src: './src/kms/web/*',
-        dest: 'lib/kms/web',
-      },
-    ],
-  }),
+  json(),
+  url(),
   nodePolyfills(),
   replace({
     preventAssignment: true,
     'node-tfhe': 'tfhe',
-    'kms/node': 'kms/web',
+    'node-tkms': 'tkms',
   }),
   typescript({
     tsconfig: './tsconfig.rollup.json',
@@ -56,7 +40,7 @@ const webPlugins = [
   commonjs(),
   resolve({
     browser: true,
-    resolveOnly: ['tfhe'],
+    resolveOnly: ['tfhe', 'tkms'],
     extensions: ['.js', '.ts', '.wasm'],
   }),
 ];
@@ -71,6 +55,16 @@ export default [
     },
     plugins: [...webPlugins],
   },
+  {
+    input:
+      './node_modules/tfhe/snippets/wasm-bindgen-rayon-3e04391371ad0a8e/src/workerHelpers.worker.js',
+    output: {
+      file: 'lib/workerHelpers.worker.js',
+      name: 'fhevm',
+      format: 'es',
+    },
+    plugins: [...webPlugins],
+  },
   {
     input: 'src/node.ts',
     output: {

diff --git a/config/webpack.config.cjs b/config/webpack.config.cjs
@@ -59,9 +59,10 @@ const web = {
       'node-tfhe': 'tfhe/tfhe',
     },
     fallback: {
-      'tfhe_bg.wasm': require.resolve('tfhe/tfhe_bg.wasm'),
       'node-tfhe': require.resolve('tfhe/tfhe'),
-      'kms_lib_bg.wasm': require.resolve('../src/kms/web/kms_lib_bg.wasm'),
+      'tfhe_bg.wasm': require.resolve('tfhe/tfhe_bg.wasm'),
+      'node-tkms': require.resolve('tkms/kms_lib'),
+      'kms_lib_bg.wasm': require.resolve('tkms/kms_lib_bg.wasm'),
       buffer: require.resolve('buffer/'),
       crypto: require.resolve('crypto-browserify'),
       stream: require.resolve('stream-browserify'),
@@ -72,9 +73,6 @@ const web = {
     new webpack.ProvidePlugin({
       Buffer: ['buffer', 'Buffer'],
     }),
-    new webpack.optimize.LimitChunkCountPlugin({
-      maxChunks: 1,
-    }),
   ],
 };
 

diff --git a/generateKeys.js b/generateKeys.js
@@ -0,0 +1,58 @@
+#!/usr/bin/env node
+
+import {
+  TfheCompactPublicKey,
+  TfheConfigBuilder,
+  TfheClientKey,
+  ShortintParameters,
+  ShortintParametersName,
+  CompactPkeCrs,
+  ShortintCompactPublicKeyEncryptionParameters,
+  ShortintCompactPublicKeyEncryptionParametersName,
+} from 'node-tfhe';
+
+import fs from 'fs';
+
+const createTfheKeypair = () => {
+  const block_params = new ShortintParameters(
+    ShortintParametersName.PARAM_MESSAGE_2_CARRY_2_COMPACT_PK_PBS_KS,
+  );
+  const casting_params = new ShortintCompactPublicKeyEncryptionParameters(
+    ShortintCompactPublicKeyEncryptionParametersName.SHORTINT_PARAM_PKE_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M64,
+  );
+  const config = TfheConfigBuilder.default()
+    .use_custom_parameters(block_params)
+    .use_dedicated_compact_public_key_parameters(casting_params)
+    .build();
+  let clientKey = TfheClientKey.generate(config);
+  let publicKey = TfheCompactPublicKey.new(clientKey);
+  fs.writeFileSync('src/test/keys/publicKey.bin', publicKey.serialize());
+  fs.writeFileSync('src/test/keys/privateKey.bin', clientKey.serialize());
+  const crs0 = CompactPkeCrs.from_config(config, 4 * 32);
+  fs.writeFileSync(
+    'src/test/keys/crs128.bin',
+    crs0.public_params().serialize(),
+  );
+  const crs1 = CompactPkeCrs.from_config(config, 4 * 64);
+  fs.writeFileSync(
+    'src/test/keys/crs256.bin',
+    crs1.public_params().serialize(),
+  );
+  const crs2 = CompactPkeCrs.from_config(config, 4 * 128);
+  fs.writeFileSync(
+    'src/test/keys/crs512.bin',
+    crs2.public_params().serialize(),
+  );
+  const crs3 = CompactPkeCrs.from_config(config, 4 * 256);
+  fs.writeFileSync(
+    'src/test/keys/crs1024.bin',
+    crs3.public_params().serialize(),
+  );
+  const crs4 = CompactPkeCrs.from_config(config, 4 * 512);
+  fs.writeFileSync(
+    'src/test/keys/crs2048.bin',
+    crs4.public_params().serialize(),
+  );
+};
+
+createTfheKeypair();
diff --git a/jest.config.cjs b/jest.config.cjs
@@ -26,6 +26,7 @@ module.exports = {
     '!src/node.ts',
     '!src/web.ts',
   ],
+  setupFiles: ['./setupJest.cjs'],
   testRegex: '\\.test\\.tsx?$',
   coverageReporters: ['lcov', 'text-summary', 'json'],
   transformIgnorePatterns: ['/node_modules/'],