force the 32-byte alignment of nonce and balance (#71)

* force the 32-byte alignment of nonce and balance * review feedback: fix endianness in output
gballet · Feb 9, 2022 · 291a947 · 291a947
1 parent 2554974
commit 291a947
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/trie/verkle.go b/trie/verkle.go
@@ -68,11 +68,15 @@ func (t *VerkleTrie) TryUpdateAccount(key []byte, acc *types.StateAccount) error
  return fmt.Errorf("updateStateObject (%x) error: %v", key, err)
  }
  var nonce [32]byte
- binary.BigEndian.PutUint64(nonce[:], acc.Nonce)
+ binary.LittleEndian.PutUint64(nonce[24:], acc.Nonce)
  if err = t.TryUpdate(utils.GetTreeKeyNonce(key), nonce[:]); err != nil {
  return fmt.Errorf("updateStateObject (%x) error: %v", key, err)
  }
- if err = t.TryUpdate(utils.GetTreeKeyBalance(key), acc.Balance.Bytes()); err != nil {
+ var balance [32]byte
+ for i, b := range acc.Balance.Bytes() {
+ balance[31-i] = b
+ }
+ if err = t.TryUpdate(utils.GetTreeKeyBalance(key), balance[:]); err != nil {
  return fmt.Errorf("updateStateObject (%x) error: %v", key, err)
  }
  if err = t.TryUpdate(utils.GetTreeKeyCodeKeccak(key), acc.CodeHash); err != nil {