poseidon contract now accepts both selectors: for uint256[n] and bytes32[n]

src/poseidon_gencontract.js

@@ -103,8 +103,13 @@ function createCode(nInputs) {
     C.push(0);
     C.calldataload();
     C.div();
-    C.push(Web3Utils.keccak256(`poseidon(uint256[${nInputs}])`).slice(0, 10)); // poseidon(uint256[])
+    C.dup(0);
+    C.push(Web3Utils.keccak256(`poseidon(uint256[${nInputs}])`).slice(0, 10)); // poseidon(uint256[n])
     C.eq();
+    C.swap(1);
+    C.push(Web3Utils.keccak256(`poseidon(bytes32[${nInputs}])`).slice(0, 10)); // poseidon(bytes32[n])
+    C.eq();
+    C.or();
     C.jmpi("start");
     C.invalid();
 
@@ -160,6 +165,27 @@ function createCode(nInputs) {
 
 function generateABI(nInputs) {
     return [
+        {
+            "constant": true,
+            "inputs": [
+                {
+                    "internalType": `bytes32[${nInputs}]`,
+                    "name": "input",
+                    "type": `bytes32[${nInputs}]`
+                }
+            ],
+            "name": "poseidon",
+            "outputs": [
+                {
+                    "internalType": "bytes32",
+                    "name": "",
+                    "type": "bytes32"
+                }
+            ],
+            "payable": false,
+            "stateMutability": "pure",
+            "type": "function"
+        },
         {
             "constant": true,
             "inputs": [
@@ -181,8 +207,8 @@ function generateABI(nInputs) {
             "stateMutability": "pure",
             "type": "function"
         }
-  ]
-};
+    ];
+}
 
 module.exports.generateABI = generateABI;
 module.exports.createCode = createCode;

test/poseidoncontract.js

@@ -14,8 +14,8 @@ describe("Poseidon Smart contract test", function () {
     let poseidon4;
     let accounts;
     this.timeout(100000);
-    let C2
-    let C4
+    let C2;
+    let C4;
 
     before(async () => {
         web3 = new Web3(ganache.provider(), null, { transactionConfirmationBlocks: 1 });