Merge pull request ethereum#15 from zama-ai/feature/add_fhe_operation…

…s_add_encrypt_decrypt Feature/add fhe operations add encrypt decrypt
sbellem · Dec 15, 2022 · 01e32db · 01e32db
2 parents c4d0e16 + 0765df3
commit 01e32db
Show file tree

Hide file tree

Showing 4 changed files with 317 additions and 5 deletions.
diff --git a/clean_tfhe_rs_api.sh b/clean_tfhe_rs_api.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+
+rm -rf tfhe-rs
+rm core/vm/lib/libtfhe.*
+rm core/vm/tfhe.h
diff --git a/core/vm/contracts.go b/core/vm/contracts.go
diff --git a/install_thfe_rs_api.sh b/install_thfe_rs_api.sh
@@ -0,0 +1,8 @@
+#!/bin/bash
+
+git clone git@github.com:zama-ai/tfhe-rs.git
+mkdir -p core/vm/lib
+cd tfhe-rs
+make build_c_api
+cp target/release/libtfhe.* ../core/vm/lib
+cp target/release/tfhe.h ../core/vm
diff --git a/tests/solidity/zama/handles.sol b/tests/solidity/zama/handles.sol
@@ -3,6 +3,45 @@
 pragma solidity >=0.7.0 <0.9.0;
 
 contract Precompiles {
+
+    function precompile_add(uint256 handle1, uint256 handle2) internal view returns (uint256 out_handle) {
+        bytes32[2] memory input;
+        input[0] = bytes32(handle1);
+        input[1] = bytes32(handle2);
+        bytes32[1] memory output;
+        assembly {
+            if iszero(staticcall(gas(), 65, input, 64, output, 32)) {
+                revert(0, 0)
+            }
+        }
+        out_handle = uint256(output[0]);
+    }
+
+    function precompile_decrypt(uint256 handle1) internal view returns (uint256 out_handle) {
+        bytes32[1] memory input;
+        input[0] = bytes32(handle1);
+        bytes32[1] memory output;
+        assembly {
+            if iszero(staticcall(gas(), 69, input, 32, output, 32)) {
+                revert(0, 0)
+            }
+        }
+        out_handle = uint256(output[0]);
+
+    }
+
+    function precompile_encrypt(uint256  to_be_encrypted) internal view returns (uint256 out_handle) {
+        bytes32[1] memory input;
+        input[0] = bytes32(to_be_encrypted);
+        bytes32[1] memory output;
+        assembly {
+            if iszero(staticcall(gas(), 70, input, 32, output, 32)) {
+                revert(0, 0)
+            }
+        }
+         out_handle = uint256(output[0]);
+    }
+
     function precompile_reencrypt(uint256 in_handle) internal view returns (uint256 out_handle) {
         bytes32[1] memory input;
         input[0] = bytes32(in_handle);
@@ -41,6 +80,7 @@ contract Precompiles {
 
 contract HandleOwner is Precompiles {
     uint256 public handle;
+    uint256 public handle2;
     uint256 public bogus_handle = 42;
     Callee callee;
     address payable owner;
@@ -54,6 +94,22 @@ contract HandleOwner is Precompiles {
         handle = precompile_verify(ciphertext);
     }
 
+     function store2(bytes memory ciphertext) public {
+        handle2 = precompile_verify(ciphertext);
+    }
+
+    function add() public view returns (uint256) {
+        return precompile_add(handle, handle2);
+    }
+
+    function decrypt() public view returns (uint256) {
+        return precompile_decrypt(handle);
+    }
+
+    function encrypt(uint256  input) public view returns (uint256) {
+        return precompile_encrypt(input);
+    }
+
     // If called before `ovewrite_handle()`, `reencrypt()` must suceed.
     function reencrypt() public view returns (uint256) {
         return precompile_reencrypt(handle);
@@ -126,4 +182,4 @@ contract Caller is Precompiles {
         owner.load_handle_without_returning_it();
         return precompile_reencrypt(handle);
     }
-}
+}