Add common> Idx parsers

This commit is contained in:
arnaucube
2020-08-04 17:53:19 +02:00
parent f3cfba1bbe
commit 37cefb0057
4 changed files with 63 additions and 8 deletions

View File

@@ -66,10 +66,13 @@ func (l *Leaf) BigInts() ([5]*big.Int, error) {
return e, nil
}
// Value returns the value of the Leaf, which is the Poseidon hash of its *big.Int representation
func (l *Leaf) Value() (*big.Int, error) {
// HashValue returns the value of the Leaf, which is the Poseidon hash of its *big.Int representation
func (l *Leaf) HashValue() (*big.Int, error) {
toHash := [poseidon.T]*big.Int{}
lBI := l.BigInts()
lBI, err := l.BigInts()
if err != nil {
return nil, err
}
copy(toHash[:], lBI[:])
v, err := poseidon.Hash(toHash)

View File

@@ -1,14 +1,38 @@
package common
import (
"encoding/binary"
"math/big"
)
// Idx represents the account Index in the MerkleTree
type Idx uint32
// Bytes returns a byte array representing the Idx
func (idx Idx) Bytes() []byte {
var b [4]byte
binary.LittleEndian.PutUint32(b[:], uint32(idx))
return b[:]
}
// BigInt returns a *big.Int representing the Idx
func (idx Idx) BigInt() *big.Int {
return big.NewInt(int64(idx))
}
// IdxFromBigInt converts a *big.Int to Idx type
func IdxFromBigInt(b *big.Int) (Idx, error) {
if b.Int64() > 4294967295 { // 2**32-1
return 0, ErrNumOverflow
}
return Idx(uint32(b.Int64())), nil
}
// Tx is a struct that represents a Hermez network transaction
type Tx struct {
TxID TxID
FromIdx uint32
ToIdx uint32
FromIdx Idx // FromIdx is used by L1Tx/Deposit to indicate the Idx receiver of the L1Tx.LoadAmount (deposit)
ToIdx Idx // ToIdx is ignored in L1Tx/Deposit, but used in the L1Tx/DepositAndTransfer
TokenID TokenID
Amount *big.Int
Nonce uint64 // effective 48 bits used

23
common/tx_test.go Normal file
View File

@@ -0,0 +1,23 @@
package common
import (
"math/big"
"testing"
"github.com/stretchr/testify/assert"
)
func TestIdx(t *testing.T) {
i := Idx(100)
assert.Equal(t, big.NewInt(100), i.BigInt())
i = Idx(uint32(4294967295))
assert.Equal(t, "4294967295", i.BigInt().String())
b := big.NewInt(4294967296)
i, err := IdxFromBigInt(b)
assert.NotNil(t, err)
assert.Equal(t, ErrNumOverflow, err)
assert.Equal(t, Idx(0), i)
}