add HashMultiPoseidon and HashMiMC_BN254

go.mod

@@ -1,29 +1,30 @@
 module github.com/vocdoni/arbo
 
-go 1.23.0
+go 1.23.2
 
-toolchain go1.23.1
+toolchain go1.23.4
 
 require (
+	github.com/consensys/gnark-crypto v0.14.1-0.20241010154951-6638408a49f3
 	github.com/frankban/quicktest v1.14.6
 	github.com/iden3/go-iden3-crypto v0.0.17
+	github.com/vocdoni/vocdoni-z-sandbox v0.0.0-20241203073504-acbb1b626579
 	go.vocdoni.io/dvote v1.10.2-0.20241024102542-c1ce6d744bc5
 	golang.org/x/crypto v0.28.0
 )
 
 require (
 	github.com/DataDog/zstd v1.5.2 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
-	github.com/bits-and-blooms/bitset v1.14.2 // indirect
+	github.com/bits-and-blooms/bitset v1.14.3 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/cockroachdb/errors v1.11.3 // indirect
 	github.com/cockroachdb/fifo v0.0.0-20240606204812-0bbfbd93a7ce // indirect
 	github.com/cockroachdb/logtags v0.0.0-20230118201751-21c54148d20b // indirect
 	github.com/cockroachdb/pebble v1.1.2 // indirect
 	github.com/cockroachdb/redact v1.1.5 // indirect
 	github.com/cockroachdb/tokenbucket v0.0.0-20230807174530-cc333fc44b06 // indirect
-	github.com/consensys/bavard v0.1.13 // indirect
-	github.com/consensys/gnark-crypto v0.14.0 // indirect
+	github.com/consensys/bavard v0.1.22 // indirect
 	github.com/getsentry/sentry-go v0.27.0 // indirect
 	github.com/gogo/protobuf v1.3.2 // indirect
 	github.com/golang/snappy v0.0.5-0.20220116011046-fa5810519dcb // indirect
@@ -38,8 +39,8 @@ require (
 	github.com/prometheus/client_model v0.6.1 // indirect
 	github.com/prometheus/common v0.55.0 // indirect
 	github.com/prometheus/procfs v0.15.1 // indirect
-	github.com/rogpeppe/go-internal v1.11.0 // indirect
-	golang.org/x/exp v0.0.0-20240808152545-0cdaa3abc0fa // indirect
+	github.com/rogpeppe/go-internal v1.12.0 // indirect
+	golang.org/x/exp v0.0.0-20240823005443-9b4947da3948 // indirect
 	golang.org/x/sys v0.26.0 // indirect
 	golang.org/x/text v0.19.0 // indirect
 	google.golang.org/protobuf v1.34.2 // indirect

go.sum

@@ -2,8 +2,8 @@ github.com/DataDog/zstd v1.5.2 h1:vUG4lAyuPCXO0TLbXvPv7EB7cNK1QV/luu55UHLrrn8=
 github.com/DataDog/zstd v1.5.2/go.mod h1:g4AWEaM3yOg3HYfnJ3YIawPnVdXJh9QME85blwSAmyw=
 github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
 github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
-github.com/bits-and-blooms/bitset v1.14.2 h1:YXVoyPndbdvcEVcseEovVfp0qjJp7S+i5+xgp/Nfbdc=
-github.com/bits-and-blooms/bitset v1.14.2/go.mod h1:7hO7Gc7Pp1vODcmWvKMRA9BNmbv6a/7QIWpPxHddWR8=
+github.com/bits-and-blooms/bitset v1.14.3 h1:Gd2c8lSNf9pKXom5JtD7AaKO8o7fGQ2LtFj1436qilA=
+github.com/bits-and-blooms/bitset v1.14.3/go.mod h1:7hO7Gc7Pp1vODcmWvKMRA9BNmbv6a/7QIWpPxHddWR8=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
 github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
 github.com/cockroachdb/datadriven v1.0.3-0.20230413201302-be42291fc80f h1:otljaYPt5hWxV3MUfO5dFPFiOXg9CyG5/kCfayTqsJ4=
@@ -20,10 +20,10 @@ github.com/cockroachdb/redact v1.1.5 h1:u1PMllDkdFfPWaNGMyLD1+so+aq3uUItthCFqzwP
 github.com/cockroachdb/redact v1.1.5/go.mod h1:BVNblN9mBWFyMyqK1k3AAiSxhvhfK2oOZZ2lK+dpvRg=
 github.com/cockroachdb/tokenbucket v0.0.0-20230807174530-cc333fc44b06 h1:zuQyyAKVxetITBuuhv3BI9cMrmStnpT18zmgmTxunpo=
 github.com/cockroachdb/tokenbucket v0.0.0-20230807174530-cc333fc44b06/go.mod h1:7nc4anLGjupUW/PeY5qiNYsdNXj7zopG+eqsS7To5IQ=
-github.com/consensys/bavard v0.1.13 h1:oLhMLOFGTLdlda/kma4VOJazblc7IM5y5QPd2A/YjhQ=
-github.com/consensys/bavard v0.1.13/go.mod h1:9ItSMtA/dXMAiL7BG6bqW2m3NdSEObYWoH223nGHukI=
-github.com/consensys/gnark-crypto v0.14.0 h1:DDBdl4HaBtdQsq/wfMwJvZNE80sHidrK3Nfrefatm0E=
-github.com/consensys/gnark-crypto v0.14.0/go.mod h1:CU4UijNPsHawiVGNxe9co07FkzCeWHHrb1li/n1XoU0=
+github.com/consensys/bavard v0.1.22 h1:Uw2CGvbXSZWhqK59X0VG/zOjpTFuOMcPLStrp1ihI0A=
+github.com/consensys/bavard v0.1.22/go.mod h1:k/zVjHHC4B+PQy1Pg7fgvG3ALicQw540Crag8qx+dZs=
+github.com/consensys/gnark-crypto v0.14.1-0.20241010154951-6638408a49f3 h1:jVatckGR1s3OHs4QnGsppX+w2P3eedlWxi7ZFq56rjA=
+github.com/consensys/gnark-crypto v0.14.1-0.20241010154951-6638408a49f3/go.mod h1:F/hJyWBcTr1sWeifAKfEN3aVb3G4U5zheEC8IbWQun4=
 github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@@ -51,8 +51,8 @@ github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
 github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
 github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
 github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
-github.com/leanovate/gopter v0.2.9 h1:fQjYxZaynp97ozCzfOyOuAGOU4aU/z37zf/tOujFk7c=
-github.com/leanovate/gopter v0.2.9/go.mod h1:U2L/78B+KVFIx2VmW6onHJQzXtFb+p5y3y2Sh+Jxxv8=
+github.com/leanovate/gopter v0.2.11 h1:vRjThO1EKPb/1NsDXuDrzldR28RLkBflWYcU9CvzWu4=
+github.com/leanovate/gopter v0.2.11/go.mod h1:aK3tzZP/C+p1m3SPRE4SYZFGP7jjkuSI4f7Xvpt0S9c=
 github.com/mmcloughlin/addchain v0.4.0 h1:SobOdjm2xLj1KkXN5/n0xTIWyZA2+s99UCY1iPfkHRY=
 github.com/mmcloughlin/addchain v0.4.0/go.mod h1:A86O+tHqZLMNO4w6ZZ4FlVQEadcoqkyU72HC5wJ4RlU=
 github.com/mmcloughlin/profile v0.1.1/go.mod h1:IhHD7q1ooxgwTgjxQYkACGA77oFTDdFVejUS1/tS/qU=
@@ -74,10 +74,12 @@ github.com/prometheus/common v0.55.0/go.mod h1:2SECS4xJG1kd8XF9IcM1gMX6510RAEL65
 github.com/prometheus/procfs v0.15.1 h1:YagwOFzUgYfKKHX6Dr+sHT7km/hxC76UB0learggepc=
 github.com/prometheus/procfs v0.15.1/go.mod h1:fB45yRUv8NstnjriLhBQLuOUt+WW4BsoGhij/e3PBqk=
 github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
-github.com/rogpeppe/go-internal v1.11.0 h1:cWPaGQEPrBb5/AsnsZesgZZ9yb1OQ+GOISoDNXVBh4M=
-github.com/rogpeppe/go-internal v1.11.0/go.mod h1:ddIwULY96R17DhadqLgMfk9H9tvdUzkipdSkR5nkCZA=
+github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8=
+github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4=
 github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/vocdoni/vocdoni-z-sandbox v0.0.0-20241203073504-acbb1b626579 h1:v2SQztawwX7nto+nHXw3ed4r8WYa+suQByzz4LPn41k=
+github.com/vocdoni/vocdoni-z-sandbox v0.0.0-20241203073504-acbb1b626579/go.mod h1:zKgPxmIDwNK7+6EJu22WIR6CV4zL8Umh+tESWnP9+kg=
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 go.vocdoni.io/dvote v1.10.2-0.20241024102542-c1ce6d744bc5 h1:22esW3YedMfoEOx0Chc2qhrooMBDRHsasJfGytESxo4=
@@ -87,8 +89,8 @@ golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8U
 golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
 golang.org/x/crypto v0.28.0 h1:GBDwsMXVQi34v5CCYUm2jkJvu4cbtru2U4TN2PSyQnw=
 golang.org/x/crypto v0.28.0/go.mod h1:rmgy+3RHxRZMyY0jjAJShp2zgEdOqj2AO7U0pYmeQ7U=
-golang.org/x/exp v0.0.0-20240808152545-0cdaa3abc0fa h1:ELnwvuAXPNtPk1TJRuGkI9fDTwym6AYBu0qzT8AcHdI=
-golang.org/x/exp v0.0.0-20240808152545-0cdaa3abc0fa/go.mod h1:akd2r19cwCdwSwWeIdzYQGa/EZZyqcOdwWiwj5L5eKQ=
+golang.org/x/exp v0.0.0-20240823005443-9b4947da3948 h1:kx6Ds3MlpiUHKj7syVnbp57++8WpuKPcR5yjLBjvLEA=
+golang.org/x/exp v0.0.0-20240823005443-9b4947da3948/go.mod h1:akd2r19cwCdwSwWeIdzYQGa/EZZyqcOdwWiwj5L5eKQ=
 golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=

hash.go

@@ -4,8 +4,10 @@ import (
 	"crypto/sha256"
 	"math/big"
 
-	"github.com/consensys/gnark-crypto/ecc/bls12-377/fr/mimc"
+	mimc_bls12_377 "github.com/consensys/gnark-crypto/ecc/bls12-377/fr/mimc"
+	mimc_bn254 "github.com/consensys/gnark-crypto/ecc/bn254/fr/mimc"
 	"github.com/iden3/go-iden3-crypto/poseidon"
+	multiposeidon "github.com/vocdoni/vocdoni-z-sandbox/hash/poseidon"
 	"golang.org/x/crypto/blake2b"
 )
 
@@ -15,24 +17,36 @@ var (
 	// TypeHashPoseidon represents the label for the HashFunction of
 	// Poseidon
 	TypeHashPoseidon = []byte("poseidon")
+	// TypeHashPoseidon represents the label for the HashFunction of
+	// Poseidon
+	TypeHashMultiPoseidon = []byte("multiposeidon")
 	// TypeHashBlake2b represents the label for the HashFunction of Blake2b
 	TypeHashBlake2b = []byte("blake2b")
 	// TypeHashMiMC_BLS12_377 represents the label for the HashFunction of MiMC
 	// over BLS12-377 curve
 	TypeHashMiMC_BLS12_377 = []byte("mimc_bls12_377")
+	// TypeHashMiMC_BN254 represents the label for the HashFunction of MiMC
+	// over BN254 curve
+	TypeHashMiMC_BN254 = []byte("mimc_bn254")
 
 	// HashFunctionSha256 contains the HashSha256 struct which implements
 	// the HashFunction interface
 	HashFunctionSha256 HashSha256
 	// HashFunctionPoseidon contains the HashPoseidon struct which implements
 	// the HashFunction interface
 	HashFunctionPoseidon HashPoseidon
+	// HashFunctionMultiPoseidon contains the HashMultiPoseidon struct which implements
+	// the HashFunction interface
+	HashFunctionMultiPoseidon HashMultiPoseidon
 	// HashFunctionBlake2b contains the HashBlake2b struct which implements
 	// the HashFunction interface
 	HashFunctionBlake2b HashBlake2b
 	// HashFunctionMiMC_BLS12_377 contains the HashMiMC_BLS12_377 struct which
 	// implements the HashFunction interface
 	HashFunctionMiMC_BLS12_377 HashMiMC_BLS12_377
+	// HashFunctionMiMC_BN254 contains the HashMiMC_BN254 struct which
+	// implements the HashFunction interface
+	HashFunctionMiMC_BN254 HashMiMC_BN254
 )
 
 // Once Generics are at Go, this will be updated (August 2021
@@ -99,6 +113,44 @@ func (f HashPoseidon) Hash(b ...[]byte) ([]byte, error) {
 	return hB, nil
 }
 
+// HashMultiPoseidon implements the HashFunction interface for the MultiPoseidon hash
+type HashMultiPoseidon struct{}
+
+// Type returns the type of HashFunction for the HashMultiPoseidon
+func (f HashMultiPoseidon) Type() []byte {
+	return TypeHashMultiPoseidon
+}
+
+// Len returns the length of the Hash output
+func (f HashMultiPoseidon) Len() int {
+	return 32 //nolint:gomnd
+}
+
+// Hash implements the hash method for the HashFunction HashMultiPoseidon. It
+// expects the byte arrays to be little-endian representations of big.Int
+// values. Notably, if any input is longer than 32 bytes (f.Len()), it will split it
+// into 32 bytes chunks and interpret each of them as a big.Int value.
+// so Hash({[64]byte}) and Hash({[32]byte, [32]byte}) will yield the same result.
+func (f HashMultiPoseidon) Hash(b ...[]byte) ([]byte, error) {
+	var bigints []*big.Int
+	for _, input := range b {
+		// Split input into chunks of 32 bytes
+		for start := 0; start < len(input); start += f.Len() {
+			end := start + f.Len()
+			if end > len(input) {
+				end = len(input)
+			}
+			// Convert each chunk into a big.Int
+			bigints = append(bigints, BytesToBigInt(input[start:end]))
+		}
+	}
+	h, err := multiposeidon.MultiPoseidon(bigints...)
+	if err != nil {
+		return nil, err
+	}
+	return BigIntToBytes(f.Len(), h), nil
+}
+
 // HashBlake2b implements the HashFunction interface for the Blake2b hash
 type HashBlake2b struct{}
 
@@ -137,16 +189,49 @@ func (f HashMiMC_BLS12_377) Type() []byte {
 
 // Len returns the length of the Hash output for the HashMiMC_BLS12_377
 func (f HashMiMC_BLS12_377) Len() int {
-	return mimc.BlockSize
+	return mimc_bls12_377.BlockSize
 }
 
 // Hash implements the hash method for the HashFunction HashMiMC_BLS12_377
 func (f HashMiMC_BLS12_377) Hash(b ...[]byte) ([]byte, error) {
-	h := mimc.NewMiMC()
+	h := mimc_bls12_377.NewMiMC()
 	for i := 0; i < len(b); i++ {
 		if _, err := h.Write(SwapEndianness(b[i])); err != nil {
 			return nil, err
 		}
 	}
 	return SwapEndianness(h.Sum(nil)), nil
 }
+
+// HashMiMC_BN254 implements the HashFunction interface for the MiMC hash
+// over the BN254 curve
+type HashMiMC_BN254 struct{}
+
+// Type returns the type of HashFunction for the HashMiMC_BN254
+func (f HashMiMC_BN254) Type() []byte {
+	return TypeHashMiMC_BN254
+}
+
+// Len returns the length of the Hash output for the HashMiMC_BN254
+func (f HashMiMC_BN254) Len() int {
+	return mimc_bn254.BlockSize
+}
+
+// Hash implements the hash method for the HashFunction HashMiMC_BN254
+func (f HashMiMC_BN254) Hash(b ...[]byte) ([]byte, error) {
+	h := mimc_bn254.NewMiMC()
+	for _, input := range b {
+		// Since arbo uses little-endian but MiMC expects inputs as big-endian
+		// we need to split long inputs and SwapEndianness of each chunk.
+		for start := 0; start < len(input); start += h.BlockSize() {
+			end := start + h.BlockSize()
+			if end > len(input) {
+				end = len(input)
+			}
+			if _, err := h.Write(SwapEndianness(input[start:end])); err != nil {
+				return nil, err
+			}
+		}
+	}
+	return SwapEndianness(h.Sum(nil)), nil
+}