github/self
1
1
Fork 0
mirror of https://github.com/selfxyz/self.git synced 2026-04-05 03:00:53 -04:00
Code Issues Packages Projects Releases Wiki Activity

implement merkle tree check on csca modulus

Browse Source
This commit is contained in:
turnoffthiscomputer
2024-06-20 18:30:56 +02:00
parent c9970b4e9d
commit 07e8d1348d
7 changed files with 96 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
circuits/circuits/certificates/dsc.circom → circuits/circuits/dsc.circom
View File

@@ -5,9 +5,10 @@ include "circomlib/circuits/poseidon.circom";
include "@zk-email/circuits/helpers/rsa.circom";
include "@zk-email/circuits/helpers/extract.circom";
include "@zk-email/circuits/helpers/sha.circom";
include "../utils/splitBytesToWords.circom";
include "binary-merkle-root.circom";
include "./utils/splitBytesToWords.circom";
template DSC(max_cert_bytes, n_dsc, k_dsc, n_csca, k_csca, dsc_mod_len ) {
template DSC(max_cert_bytes, n_dsc, k_dsc, n_csca, k_csca, dsc_mod_len, nLevels ) {
signal input raw_dsc_cert[max_cert_bytes];
signal input raw_dsc_cert_padded_bytes;
signal input csca_modulus[k_csca];
@@ -16,12 +17,36 @@ template DSC(max_cert_bytes, n_dsc, k_dsc, n_csca, k_csca, dsc_mod_len ) {
signal input start_index;
signal input secret;
signal input merkle_root;
signal input path[nLevels];
signal input siblings[nLevels];
signal output blinded_csca_commitment;
// verify the leaf
component poseidon16first = Poseidon(16);
component poseidon16next = Poseidon(16);
component poseidon2last = Poseidon(2);
component poseidonfinal = Poseidon(3);
for (var i = 0; i < 16; i++) {
poseidon16first.inputs[i] <== csca_modulus[i];
poseidon16next.inputs[i] <== csca_modulus[i+16];
}
poseidon2last.inputs[0] <== csca_modulus[32];
poseidon2last.inputs[1] <== csca_modulus[33];
poseidonfinal.inputs[0] <== poseidon16first.out;
poseidonfinal.inputs[1] <== poseidon16next.out;
poseidonfinal.inputs[2] <== poseidon2last.out;
signal leaf <== poseidonfinal.out;
signal computed_merkle_root <== BinaryMerkleRoot(nLevels)(leaf, nLevels, path, siblings);
merkle_root === computed_merkle_root;
// variables verification
assert(max_cert_bytes % 64 == 0);
assert(n_csca * k_csca > max_cert_bytes);
assert(n_csca < (255 \ 2));
assert(n_csca <= (255 \ 2));
// hash raw TBS certificate
signal sha[256] <== Sha256Bytes(max_cert_bytes)(raw_dsc_cert, raw_dsc_cert_padded_bytes);

5
circuits/circuits/tests/certificates/dsc_2048.circom
View File

@@ -1,6 +1,5 @@
pragma circom 2.1.6;
include "../../certificates/dsc.circom";
include "../../dsc.circom";
component main = DSC(1664,121 ,17 ,121, 34, 256);
component main { public [ merkle_root ] } = DSC(1664,121 ,17 ,121, 17, 256, 12);

4
circuits/circuits/tests/certificates/dsc_4096.circom
View File

@@ -1,5 +1,5 @@
pragma circom 2.1.6;
include "../../certificates/dsc.circom";
include "../../dsc.circom";
component main = DSC(1664,121 ,17 ,121, 34, 256);
component main { public [ merkle_root ] } = DSC(1664,121 ,17 ,121, 34, 256, 12);

10
circuits/test/dsc_4096.test.ts
View File

@@ -5,7 +5,7 @@ import path from 'path';
const wasm_tester = require("circom_tester").wasm;
import { splitToWords } from '../../common/src/utils/utils';
import { sha256Pad } from '../../common/src/utils/shaPad';
import { findStartIndex, getCSCAInputs } from '../../common/src/utils/csca';
import { computeLeafFromModulus, findStartIndex, getCSCAInputs } from '../../common/src/utils/csca';
describe('DSC chain certificate', function () {
this.timeout(0); // Disable timeout
@@ -19,10 +19,10 @@ describe('DSC chain certificate', function () {
const csca = fs.readFileSync('../common/src/mock_certificates/sha256_rsa_4096/mock_csca.crt', 'utf8');
const dscCert = forge.pki.certificateFromPem(dsc);
const cscaCert = forge.pki.certificateFromPem(csca);
const inputs = getCSCAInputs(dscCert, cscaCert, n_dsc, k_dsc, n_csca, k_csca, max_cert_bytes, true);
console.log("inputs:", inputs);
console.log("inputs:", JSON.stringify(inputs, null, 2));
fs.writeFileSync('inputs.json', JSON.stringify(inputs, null, 2));
before(async () => {
circuit = await wasm_tester(
@@ -30,6 +30,8 @@ describe('DSC chain certificate', function () {
{
include: [
"node_modules",
"./node_modules/@zk-kit/binary-merkle-root.circom/src",
"./node_modules/circomlib/circuits"
]
}
);