@coti-io/coti-contracts 0.1.0

package/contracts/mocks/utils/mpc/ShiftTestsContract.sol

@@ -0,0 +1,311 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.19;
+
+import "../../../utils/mpc/MpcCore.sol";
+
+contract ShiftTestsContract {
+
+    struct AllGTCastingValues {
+        gtUint8 a8_s;
+        gtUint8 b8_s;
+        gtUint16 a16_s;
+        gtUint16 b16_s;
+        gtUint32 a32_s;
+        gtUint32 b32_s;
+        gtUint64 a64_s;
+        gtUint64 b64_s;
+    }
+
+    struct Check16 {
+        gtUint16 res16_16;
+        gtUint16 res8_16;
+        gtUint16 res16_8;
+    }
+
+    struct Check32 {
+        gtUint32 res32_32;
+        gtUint32 res8_32;
+        gtUint32 res32_8;
+        gtUint32 res16_32;
+        gtUint32 res32_16;
+    }
+
+    struct Check64 {
+        gtUint64 res64_64;
+        gtUint64 res8_64;
+        gtUint64 res64_8;
+        gtUint64 res16_64;
+        gtUint64 res64_16;
+        gtUint64 res32_64;
+        gtUint64 res64_32;
+    }
+
+    uint8 result;
+    uint16 result16;
+    uint32 result32;
+    uint64 result64;
+
+    function getResult() public view returns (uint8) {
+        return result;
+    }
+
+    function getAllShiftResults()
+        public
+        view
+        returns (uint8, uint16, uint32, uint64)
+    {
+        return (result, result16, result32, result64);
+    }
+
+    function setPublicValues(
+        AllGTCastingValues memory castingValues,
+        uint8 a,
+        uint8 b
+    ) public {
+        castingValues.a8_s = MpcCore.setPublic8(a);
+        castingValues.b8_s = MpcCore.setPublic8(b);
+        castingValues.a16_s = MpcCore.setPublic16(a);
+        castingValues.b16_s = MpcCore.setPublic16(b);
+        castingValues.a32_s = MpcCore.setPublic32(a);
+        castingValues.b32_s = MpcCore.setPublic32(b);
+        castingValues.a64_s = MpcCore.setPublic64(a);
+        castingValues.b64_s = MpcCore.setPublic64(b);
+    }
+
+    function decryptAndCompareResults16(
+        Check16 memory check16
+    ) public returns (uint16) {
+        // Calculate the result
+        uint16 result = MpcCore.decrypt(check16.res16_16);
+
+        require(
+            result == MpcCore.decrypt(check16.res8_16) &&
+                result == MpcCore.decrypt(check16.res16_8),
+            "decryptAndCompareAllResults: Failed to decrypt and compare all results"
+        );
+
+        return result;
+    }
+
+    function decryptAndCompareResults32(
+        Check32 memory check32
+    ) public returns (uint32) {
+        // Calculate the result
+        uint32 result = MpcCore.decrypt(check32.res32_32);
+
+        require(
+            result == MpcCore.decrypt(check32.res8_32) &&
+                result == MpcCore.decrypt(check32.res32_8) &&
+                result == MpcCore.decrypt(check32.res32_16) &&
+                result == MpcCore.decrypt(check32.res16_32),
+            "decryptAndCompareAllResults: Failed to decrypt and compare all results"
+        );
+
+        return result;
+    }
+
+    function decryptAndCompareResults64(
+        Check64 memory check64
+    ) public returns (uint64) {
+        // Calculate the result
+        uint64 result = MpcCore.decrypt(check64.res64_64);
+
+        require(
+            result == MpcCore.decrypt(check64.res8_64) &&
+                result == MpcCore.decrypt(check64.res64_8) &&
+                result == MpcCore.decrypt(check64.res64_16) &&
+                result == MpcCore.decrypt(check64.res16_64) &&
+                result == MpcCore.decrypt(check64.res64_32) &&
+                result == MpcCore.decrypt(check64.res32_64),
+            "decryptAndCompareAllResults: Failed to decrypt and compare all results"
+        );
+
+        return result;
+    }
+
+    function shlTest(
+        uint8 a,
+        uint8 b
+    ) public returns (uint8, uint16, uint32, uint64) {
+        AllGTCastingValues memory castingValues;
+        Check16 memory check16;
+        Check32 memory check32;
+        Check64 memory check64;
+        setPublicValues(castingValues, a, b);
+
+        // Calculate the expected result
+        result = MpcCore.decrypt(
+            MpcCore.shl(castingValues.a8_s, castingValues.b8_s)
+        );
+
+        // Calculate the result with casting to 16
+        check16.res16_16 = MpcCore.shl(
+            castingValues.a16_s,
+            castingValues.b16_s
+        );
+        check16.res8_16 = MpcCore.shl(castingValues.a8_s, castingValues.b16_s);
+        check16.res16_8 = MpcCore.shl(castingValues.a16_s, castingValues.b8_s);
+        result16 = decryptAndCompareResults16(check16);
+
+        // Calculate the result with casting to 32
+        check32.res32_32 = MpcCore.shl(
+            castingValues.a32_s,
+            castingValues.b32_s
+        );
+        check32.res8_32 = MpcCore.shl(castingValues.a8_s, castingValues.b32_s);
+        check32.res32_8 = MpcCore.shl(castingValues.a32_s, castingValues.b8_s);
+        check32.res16_32 = MpcCore.shl(
+            castingValues.a16_s,
+            castingValues.b32_s
+        );
+        check32.res32_16 = MpcCore.shl(
+            castingValues.a32_s,
+            castingValues.b16_s
+        );
+        result32 = decryptAndCompareResults32(check32);
+
+        // Calculate the result with casting to 64
+        check64.res64_64 = MpcCore.shl(
+            castingValues.a64_s,
+            castingValues.b64_s
+        );
+        check64.res8_64 = MpcCore.shl(castingValues.a8_s, castingValues.b64_s);
+        check64.res64_8 = MpcCore.shl(castingValues.a64_s, castingValues.b8_s);
+        check64.res16_64 = MpcCore.shl(
+            castingValues.a16_s,
+            castingValues.b64_s
+        );
+        check64.res64_16 = MpcCore.shl(
+            castingValues.a64_s,
+            castingValues.b16_s
+        );
+        check64.res32_64 = MpcCore.shl(
+            castingValues.a32_s,
+            castingValues.b64_s
+        );
+        check64.res64_32 = MpcCore.shl(
+            castingValues.a64_s,
+            castingValues.b32_s
+        );
+        result64 = decryptAndCompareResults64(check64);
+
+        // Check the result with scalar
+        require(
+            result == MpcCore.decrypt(MpcCore.shl(a, castingValues.b8_s)) &&
+                result == MpcCore.decrypt(MpcCore.shl(castingValues.a8_s, b)),
+            "shlTest: test 8 bits with scalar failed"
+        );
+        require(
+            result16 == MpcCore.decrypt(MpcCore.shl(a, castingValues.b16_s)) &&
+                result16 ==
+                MpcCore.decrypt(MpcCore.shl(castingValues.a16_s, b)),
+            "shlTest: test 16 bits with scalar failed"
+        );
+        require(
+            result32 == MpcCore.decrypt(MpcCore.shl(a, castingValues.b32_s)) &&
+                result32 ==
+                MpcCore.decrypt(MpcCore.shl(castingValues.a32_s, b)),
+            "shlTest: test 32 bits with scalar failed"
+        );
+        require(
+            result64 == MpcCore.decrypt(MpcCore.shl(a, castingValues.b64_s)) &&
+                result64 ==
+                MpcCore.decrypt(MpcCore.shl(castingValues.a64_s, b)),
+            "shlTest: test 64 bits with scalar failed"
+        );
+
+        return (result, result16, result32, result64);
+    }
+
+    function shrTest(uint8 a, uint8 b) public returns (uint8) {
+        AllGTCastingValues memory castingValues;
+        Check16 memory check16;
+        Check32 memory check32;
+        Check64 memory check64;
+        setPublicValues(castingValues, a, b);
+
+        // Calculate the expected result
+        result = MpcCore.decrypt(
+            MpcCore.shr(castingValues.a8_s, castingValues.b8_s)
+        );
+
+        // Calculate the results with casting to 16
+        check16.res16_16 = MpcCore.shr(
+            castingValues.a16_s,
+            castingValues.b16_s
+        );
+        check16.res8_16 = MpcCore.shr(castingValues.a8_s, castingValues.b16_s);
+        check16.res16_8 = MpcCore.shr(castingValues.a16_s, castingValues.b8_s);
+        uint16 res16 = decryptAndCompareResults16(check16);
+        require(res16 == result, "shrTest: cast 16 failed");
+
+        // Calculate the result with casting to 32
+        check32.res32_32 = MpcCore.shr(
+            castingValues.a32_s,
+            castingValues.b32_s
+        );
+        check32.res8_32 = MpcCore.shr(castingValues.a8_s, castingValues.b32_s);
+        check32.res32_8 = MpcCore.shr(castingValues.a32_s, castingValues.b8_s);
+        check32.res16_32 = MpcCore.shr(
+            castingValues.a16_s,
+            castingValues.b32_s
+        );
+        check32.res32_16 = MpcCore.shr(
+            castingValues.a32_s,
+            castingValues.b16_s
+        );
+        uint32 res32 = decryptAndCompareResults32(check32);
+        require(result == res32, "shrTest: cast 32 failed");
+
+        // Calculate the result with casting to 64
+        check64.res64_64 = MpcCore.shr(
+            castingValues.a64_s,
+            castingValues.b64_s
+        );
+        check64.res8_64 = MpcCore.shr(castingValues.a8_s, castingValues.b64_s);
+        check64.res64_8 = MpcCore.shr(castingValues.a64_s, castingValues.b8_s);
+        check64.res16_64 = MpcCore.shr(
+            castingValues.a16_s,
+            castingValues.b64_s
+        );
+        check64.res64_16 = MpcCore.shr(
+            castingValues.a64_s,
+            castingValues.b16_s
+        );
+        check64.res32_64 = MpcCore.shr(
+            castingValues.a32_s,
+            castingValues.b64_s
+        );
+        check64.res64_32 = MpcCore.shr(
+            castingValues.a64_s,
+            castingValues.b32_s
+        );
+        uint64 res64 = decryptAndCompareResults64(check64);
+        require(result == res64, "shrTest: cast 64 failed");
+
+        // Check the result with scalar
+        require(
+            result == MpcCore.decrypt(MpcCore.shr(a, castingValues.b8_s)) &&
+                result == MpcCore.decrypt(MpcCore.shr(castingValues.a8_s, b)),
+            "shrTest: test 8 bits with scalar failed"
+        );
+        require(
+            result == MpcCore.decrypt(MpcCore.shr(a, castingValues.b16_s)) &&
+                result == MpcCore.decrypt(MpcCore.shr(castingValues.a16_s, b)),
+            "shrTest: test 16 bits with scalar failed"
+        );
+        require(
+            result == MpcCore.decrypt(MpcCore.shr(a, castingValues.b32_s)) &&
+                result == MpcCore.decrypt(MpcCore.shr(castingValues.a32_s, b)),
+            "shrTest: test 32 bits with scalar failed"
+        );
+        require(
+            result == MpcCore.decrypt(MpcCore.shr(a, castingValues.b64_s)) &&
+                result == MpcCore.decrypt(MpcCore.shr(castingValues.a64_s, b)),
+            "shrTest: test 64 bits with scalar failed"
+        );
+
+        return result;
+    }
+}

package/contracts/mocks/utils/mpc/StringTestsContract.sol

@@ -0,0 +1,59 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.19;
+
+import "../../../utils/mpc/MpcCore.sol";
+
+contract StringTestsContract {
+    
+    ctString private userEncryptedString;
+
+    ctString private networkEncryptedString;
+
+    string public plaintext;
+
+    bool public isEqual;
+
+    function setUserEncryptedString(itString calldata it_) public {
+        gtString memory gt_ = MpcCore.validateCiphertext(it_);
+
+        userEncryptedString = MpcCore.offBoardToUser(gt_, msg.sender);
+    }
+
+    function getUserEncryptedString() public view returns (ctString memory) {
+        return userEncryptedString;
+    }
+
+    function setNetworkEncryptedString(itString calldata it_) public {
+        gtString memory gt_ = MpcCore.validateCiphertext(it_);
+
+        networkEncryptedString = MpcCore.offBoard(gt_);
+    }
+
+    function decryptNetworkEncryptedString() public {
+        gtString memory gt_ = MpcCore.onBoard(networkEncryptedString);
+
+        plaintext = MpcCore.decrypt(gt_);
+    }
+
+    function setPublicString(string calldata str) public {
+        gtString memory gt_ = MpcCore.setPublicString(str);
+
+        userEncryptedString = MpcCore.offBoardToUser(gt_, msg.sender);
+    }
+
+    function setIsEqual(itString calldata a_, itString calldata b_, bool useEq) public {
+        gtString memory a = MpcCore.validateCiphertext(a_);
+        gtString memory b = MpcCore.validateCiphertext(b_);
+
+        gtBool isEqual_;
+
+        if (useEq) {
+            isEqual_ = MpcCore.eq(a, b);
+        } else {
+            isEqual_ = MpcCore.not(MpcCore.ne(a, b));
+        }
+
+        isEqual = MpcCore.decrypt(isEqual_);
+    }
+}

package/contracts/mocks/utils/mpc/TransferScalarTestsContract.sol

@@ -0,0 +1,271 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.19;
+
+import "../../../utils/mpc/MpcCore.sol";
+
+contract TransferScalarTestsContract {
+    
+    struct AllGTCastingValues {
+        gtUint8 a8_s;
+        gtUint8 b8_s;
+        gtUint16 a16_s;
+        gtUint16 b16_s;
+        gtUint32 a32_s;
+        gtUint32 b32_s;
+        gtUint64 a64_s;
+        gtUint64 b64_s;
+    }
+
+    uint8 new_a;
+    uint8 new_b;
+    bool res;
+
+    function getResults() public view returns (uint8, uint8, bool) {
+        return (new_a, new_b, res);
+    }
+
+    function computeAndChekTransfer16(
+        AllGTCastingValues memory allGTCastingValues,
+        uint8 new_a,
+        uint8 new_b,
+        bool res,
+        uint8 amount
+    ) public {
+        // Check all options for casting to 16 while amount is scalar
+        (gtUint16 newA_s, gtUint16 newB_s, gtBool res_s) = MpcCore.transfer(
+            allGTCastingValues.a16_s,
+            allGTCastingValues.b16_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a8_s,
+            allGTCastingValues.b16_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a16_s,
+            allGTCastingValues.b8_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+    }
+
+    function computeAndChekTransfer32(
+        AllGTCastingValues memory allGTCastingValues,
+        uint8 new_a,
+        uint8 new_b,
+        bool res,
+        uint8 amount
+    ) public {
+        // Check all options for casting to 32 while amount is scalar
+        (gtUint32 newA_s, gtUint32 newB_s, gtBool res_s) = MpcCore.transfer(
+            allGTCastingValues.a32_s,
+            allGTCastingValues.b32_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a8_s,
+            allGTCastingValues.b32_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a32_s,
+            allGTCastingValues.b8_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a16_s,
+            allGTCastingValues.b32_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a32_s,
+            allGTCastingValues.b16_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+    }
+
+    function computeAndChekTransfer64(
+        AllGTCastingValues memory allGTCastingValues,
+        uint8 new_a,
+        uint8 new_b,
+        bool res,
+        uint8 amount
+    ) public {
+        // Check all options for casting to 64 while amount is scalar
+        (gtUint64 newA_s, gtUint64 newB_s, gtBool res_s) = MpcCore.transfer(
+            allGTCastingValues.a64_s,
+            allGTCastingValues.b64_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a8_s,
+            allGTCastingValues.b64_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a64_s,
+            allGTCastingValues.b8_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a16_s,
+            allGTCastingValues.b64_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a64_s,
+            allGTCastingValues.b16_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a32_s,
+            allGTCastingValues.b64_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+
+        (newA_s, newB_s, res_s) = MpcCore.transfer(
+            allGTCastingValues.a64_s,
+            allGTCastingValues.b32_s,
+            amount
+        );
+        require(
+            new_a == MpcCore.decrypt(newA_s) &&
+                new_b == MpcCore.decrypt(newB_s) &&
+                res == MpcCore.decrypt(res_s),
+            "transferTest: check scalar failed"
+        );
+    }
+
+    function transferScalarTest(
+        uint8 a,
+        uint8 b,
+        uint8 amount
+    ) public returns (uint8, uint8, bool) {
+        AllGTCastingValues memory allGTCastingValues;
+        allGTCastingValues.a8_s = MpcCore.setPublic8(a);
+        allGTCastingValues.b8_s = MpcCore.setPublic8(b);
+        allGTCastingValues.a16_s = MpcCore.setPublic16(a);
+        allGTCastingValues.b16_s = MpcCore.setPublic16(b);
+        allGTCastingValues.a32_s = MpcCore.setPublic32(a);
+        allGTCastingValues.b32_s = MpcCore.setPublic32(b);
+        allGTCastingValues.a64_s = MpcCore.setPublic64(a);
+        allGTCastingValues.b64_s = MpcCore.setPublic64(b);
+
+        // Calculate the expected result
+        (gtUint8 newA_s, gtUint8 newB_s, gtBool res_s) = MpcCore.transfer(
+            allGTCastingValues.a8_s,
+            allGTCastingValues.b8_s,
+            amount
+        );
+        new_a = MpcCore.decrypt(newA_s);
+        new_b = MpcCore.decrypt(newB_s);
+        res = MpcCore.decrypt(res_s);
+
+        // Calculate the result with casting to 16
+        computeAndChekTransfer16(allGTCastingValues, new_a, new_b, res, amount);
+
+        // Calculate the result with casting to 32
+        computeAndChekTransfer32(allGTCastingValues, new_a, new_b, res, amount);
+
+        // Calculate the result with casting to 64
+        computeAndChekTransfer64(allGTCastingValues, new_a, new_b, res, amount);
+
+        return (new_a, new_b, res);
+    }
+}