arnaucube/circomlib
1
0
Fork 0
mirror of https://github.com/arnaucube/circomlib.git synced 2026-02-06 18:56:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

pass fixes size array arg in solidity poseidon implementation

Browse Source
This commit is contained in:
Alexey
2020-10-26 16:30:10 +03:00
parent 411a7d7576
commit 1f9d0bd37f
2 changed files with 37 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
src/poseidon_gencontract.js
View File

@@ -4,6 +4,7 @@
const Contract = require("./evmasm"); const Contract = require("./evmasm");
const { unstringifyBigInts } = require("ffjavascript").utils; const { unstringifyBigInts } = require("ffjavascript").utils;
const Web3Utils = require("web3-utils");
const { C:K, M } = unstringifyBigInts(require("./poseidon_constants.json")); const { C:K, M } = unstringifyBigInts(require("./poseidon_constants.json"));
@@ -99,7 +100,7 @@ function createCode(nInputs) {
C.push(0); C.push(0);
C.calldataload(); C.calldataload();
C.div(); C.div();
C.push("0xc4420fb4"); // poseidon(uint256[]) C.push(Web3Utils.keccak256(`poseidon(uint256[${nInputs}])`).slice(0, 10)); // poseidon(uint256[])
C.eq(); C.eq();
C.jmpi("start"); C.jmpi("start");
C.invalid(); C.invalid();
@@ -112,11 +113,10 @@ function createCode(nInputs) {
// Load t values from the call data. // Load t values from the call data.
// The function has a single array param param // The function has a single array param param
// [Selector (4)] [Pointer (32)][Length (32)] [data1 (32)] .... // [Selector (4)] [item1 (32)] [item2 (32)] ....
// We ignore the pointer and the length and just load t values to the state // Stack positions 0-nInputs.
// (Stack positions 0-{t-1}) If the array is shorter, we just set zeros.
for (let i=0; i<t; i++) { for (let i=0; i<t; i++) {
C.push(0x44+(0x20*(t-1-i))); C.push(0x04+(0x20*(nInputs-i)));
C.calldataload(); C.calldataload();
} }
@@ -155,28 +155,33 @@ function createCode(nInputs) {
return C.createTxData(); return C.createTxData();
} }
module.exports.abi = [ function generateABI(nInputs) {
{ return [
"constant": true, {
"inputs": [ "constant": true,
{ "inputs": [
"name": "input", {
"type": "uint256[]" "internalType": `uint256[${nInputs}]`,
} "name": "input",
], "type": `uint256[${nInputs}]`
"name": "poseidon", }
"outputs": [ ],
{ "name": "poseidon",
"name": "", "outputs": [
"type": "uint256" {
} "internalType": "uint256",
], "name": "",
"payable": false, "type": "uint256"
"stateMutability": "pure", }
"type": "function" ],
} "payable": false,
]; "stateMutability": "pure",
"type": "function"
}
];
}
module.exports.generateABI = generateABI;
module.exports.createCode = createCode; module.exports.createCode = createCode;

11
test/poseidoncontract.js
View File

@@ -21,15 +21,16 @@ describe("Poseidon Smart contract test", function () {
}); });
it("Should deploy the contract", async () => { it("Should deploy the contract", async () => {
const C = new web3.eth.Contract(poseidonGenContract.abi); const C6 = new web3.eth.Contract(poseidonGenContract.generateABI(5));
const C3 = new web3.eth.Contract(poseidonGenContract.generateABI(2));
poseidon6 = await C.deploy({ poseidon6 = await C6.deploy({
data: poseidonGenContract.createCode(5) data: poseidonGenContract.createCode(5)
}).send({ }).send({
gas: 5000000, gas: 5000000,
from: accounts[0] from: accounts[0]
}); });
poseidon3 = await C.deploy({ poseidon3 = await C3.deploy({
data: poseidonGenContract.createCode(2) data: poseidonGenContract.createCode(2)
}).send({ }).send({
gas: 5000000, gas: 5000000,
@@ -37,7 +38,7 @@ describe("Poseidon Smart contract test", function () {
}); });
}); });
it("Shold calculate the poseidon correctly t=6", async () => { it("Should calculate the poseidon correctly t=6", async () => {
const res = await poseidon6.methods.poseidon([1,2, 0, 0, 0]).call(); const res = await poseidon6.methods.poseidon([1,2, 0, 0, 0]).call();
@@ -48,7 +49,7 @@ describe("Poseidon Smart contract test", function () {
assert.equal(res.toString(), res2.toString()); assert.equal(res.toString(), res2.toString());
}); });
it("Shold calculate the poseidon correctly t=3", async () => { it("Should calculate the poseidon correctly t=3", async () => {
const res = await poseidon3.methods.poseidon([1,2]).call(); const res = await poseidon3.methods.poseidon([1,2]).call();