Examples with WorldStateProofProvider org.hyperledger.besu.ethereum.proof.WorldStateProofProvider used on opensource projects

Example 1 with WorldStateProofProvider

use of org.hyperledger.besu.ethereum.proof.WorldStateProofProvider in project besu by hyperledger.

the class StackTrieTest method shouldNotSaveTheRootWhenIncomplete.

@Test
public void shouldNotSaveTheRootWhenIncomplete() {
    final int nbAccounts = 15;
    final WorldStateStorage worldStateStorage = new WorldStateKeyValueStorage(new InMemoryKeyValueStorage());
    final WorldStateStorage recreatedWorldStateStorage = new WorldStateKeyValueStorage(new InMemoryKeyValueStorage());
    final MerklePatriciaTrie<Bytes32, Bytes> accountStateTrie = TrieGenerator.generateTrie(worldStateStorage, nbAccounts);
    final StackTrie stackTrie = new StackTrie(Hash.wrap(accountStateTrie.getRootHash()), 0, 256, lastAccount);
    stackTrie.addSegment();
    final RangeStorageEntriesCollector collector = RangeStorageEntriesCollector.createCollector(lastAccount, RangeManager.MAX_RANGE, 5, Integer.MAX_VALUE);
    final TrieIterator<Bytes> visitor = RangeStorageEntriesCollector.createVisitor(collector);
    final TreeMap<Bytes32, Bytes> accounts = (TreeMap<Bytes32, Bytes>) accountStateTrie.entriesFrom(root -> RangeStorageEntriesCollector.collectEntries(collector, visitor, root, lastAccount));
    final WorldStateProofProvider worldStateProofProvider = new WorldStateProofProvider(worldStateStorage);
    // generate the proof
    final List<Bytes> proofs = worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), lastAccount);
    proofs.addAll(worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), accounts.lastKey()));
    stackTrie.addElement(Bytes32.random(), proofs, accounts);
    final WorldStateStorage.Updater updater = recreatedWorldStateStorage.updater();
    stackTrie.commit(updater::putAccountStateTrieNode);
    updater.commit();
    Assertions.assertThat(recreatedWorldStateStorage.getAccountStateTrieNode(Bytes.EMPTY, accountStateTrie.getRootHash())).isEmpty();
}

Example 2 with WorldStateProofProvider

use of org.hyperledger.besu.ethereum.proof.WorldStateProofProvider in project besu by hyperledger.

the class StackTrieTest method shouldSaveTheRootWhenComplete.

@Test
public void shouldSaveTheRootWhenComplete() {
    final int nbAccounts = 15;
    final WorldStateStorage worldStateStorage = new WorldStateKeyValueStorage(new InMemoryKeyValueStorage());
    final WorldStateStorage recreatedWorldStateStorage = new WorldStateKeyValueStorage(new InMemoryKeyValueStorage());
    final MerklePatriciaTrie<Bytes32, Bytes> accountStateTrie = TrieGenerator.generateTrie(worldStateStorage, nbAccounts);
    final StackTrie stackTrie = new StackTrie(Hash.wrap(accountStateTrie.getRootHash()), 0, 256, lastAccount);
    for (int i = 0; i < nbAccounts; i += 5) {
        stackTrie.addSegment();
        final RangeStorageEntriesCollector collector = RangeStorageEntriesCollector.createCollector(lastAccount, RangeManager.MAX_RANGE, 5, Integer.MAX_VALUE);
        final TrieIterator<Bytes> visitor = RangeStorageEntriesCollector.createVisitor(collector);
        final TreeMap<Bytes32, Bytes> accounts = (TreeMap<Bytes32, Bytes>) accountStateTrie.entriesFrom(root -> RangeStorageEntriesCollector.collectEntries(collector, visitor, root, lastAccount));
        final WorldStateProofProvider worldStateProofProvider = new WorldStateProofProvider(worldStateStorage);
        // generate the proof
        final List<Bytes> proofs = worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), lastAccount);
        proofs.addAll(worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), accounts.lastKey()));
        stackTrie.addElement(Bytes32.random(), proofs, accounts);
        final WorldStateStorage.Updater updater = recreatedWorldStateStorage.updater();
        stackTrie.commit(updater::putAccountStateTrieNode);
        updater.commit();
    }
    Assertions.assertThat(worldStateStorage.getAccountStateTrieNode(Bytes.EMPTY, accountStateTrie.getRootHash())).isPresent();
}

Example 3 with WorldStateProofProvider

use of org.hyperledger.besu.ethereum.proof.WorldStateProofProvider in project besu by hyperledger.

the class RangeManagerTest method testFindNewBeginElementWhenNothingIsMissing.

@Test
public void testFindNewBeginElementWhenNothingIsMissing() {
    final WorldStateStorage worldStateStorage = new WorldStateKeyValueStorage(new InMemoryKeyValueStorage());
    final MerklePatriciaTrie<Bytes32, Bytes> accountStateTrie = TrieGenerator.generateTrie(worldStateStorage, 15);
    final RangeStorageEntriesCollector collector = RangeStorageEntriesCollector.createCollector(Hash.ZERO, RangeManager.MAX_RANGE, 15, Integer.MAX_VALUE);
    final TrieIterator<Bytes> visitor = RangeStorageEntriesCollector.createVisitor(collector);
    final TreeMap<Bytes32, Bytes> accounts = (TreeMap<Bytes32, Bytes>) accountStateTrie.entriesFrom(root -> RangeStorageEntriesCollector.collectEntries(collector, visitor, root, Hash.ZERO));
    final WorldStateProofProvider worldStateProofProvider = new WorldStateProofProvider(worldStateStorage);
    // generate the proof
    final List<Bytes> proofs = worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), Hash.ZERO);
    proofs.addAll(worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), accounts.lastKey()));
    final Optional<Bytes32> newBeginElementInRange = RangeManager.findNewBeginElementInRange(accountStateTrie.getRootHash(), proofs, accounts, RangeManager.MAX_RANGE);
    Assertions.assertThat(newBeginElementInRange).isEmpty();
}

Example 4 with WorldStateProofProvider

use of org.hyperledger.besu.ethereum.proof.WorldStateProofProvider in project besu by hyperledger.

the class RangeManagerTest method testFindNewBeginElement.

@Test
public void testFindNewBeginElement() {
    final WorldStateStorage worldStateStorage = new WorldStateKeyValueStorage(new InMemoryKeyValueStorage());
    final MerklePatriciaTrie<Bytes32, Bytes> accountStateTrie = TrieGenerator.generateTrie(worldStateStorage, 15);
    final RangeStorageEntriesCollector collector = RangeStorageEntriesCollector.createCollector(Hash.ZERO, RangeManager.MAX_RANGE, 10, Integer.MAX_VALUE);
    final TrieIterator<Bytes> visitor = RangeStorageEntriesCollector.createVisitor(collector);
    final TreeMap<Bytes32, Bytes> accounts = (TreeMap<Bytes32, Bytes>) accountStateTrie.entriesFrom(root -> RangeStorageEntriesCollector.collectEntries(collector, visitor, root, Hash.ZERO));
    final WorldStateProofProvider worldStateProofProvider = new WorldStateProofProvider(worldStateStorage);
    // generate the proof
    final List<Bytes> proofs = worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), Hash.ZERO);
    proofs.addAll(worldStateProofProvider.getAccountProofRelatedNodes(Hash.wrap(accountStateTrie.getRootHash()), accounts.lastKey()));
    final Optional<Bytes32> newBeginElementInRange = RangeManager.findNewBeginElementInRange(accountStateTrie.getRootHash(), proofs, accounts, RangeManager.MAX_RANGE);
    Assertions.assertThat(newBeginElementInRange).contains(Bytes32.leftPad(Bytes.wrap(Bytes.ofUnsignedShort(0x0b))));
}

Example 5 with WorldStateProofProvider

use of org.hyperledger.besu.ethereum.proof.WorldStateProofProvider in project besu by hyperledger.

the class TaskGenerator method createStorageRangeDataRequest.

private static StorageRangeDataRequest createStorageRangeDataRequest(final WorldStateProofProvider worldStateProofProvider, final WorldStateStorage worldStateStorage, final Hash rootHash, final Hash accountHash, final Bytes32 storageRoot, final boolean withData) {
    final RangeStorageEntriesCollector collector = RangeStorageEntriesCollector.createCollector(Bytes32.ZERO, RangeManager.MAX_RANGE, 100, Integer.MAX_VALUE);
    final StoredMerklePatriciaTrie<Bytes, Bytes> storageTrie = new StoredMerklePatriciaTrie<>((location, hash) -> worldStateStorage.getAccountStorageTrieNode(accountHash, location, hash), storageRoot, b -> b, b -> b);
    final TrieIterator<Bytes> visitor = RangeStorageEntriesCollector.createVisitor(collector);
    final TreeMap<Bytes32, Bytes> slots = (TreeMap<Bytes32, Bytes>) storageTrie.entriesFrom(root -> RangeStorageEntriesCollector.collectEntries(collector, visitor, root, Bytes32.ZERO));
    final StorageRangeDataRequest request = SnapDataRequest.createStorageRangeDataRequest(rootHash, accountHash, storageRoot, RangeManager.MIN_RANGE, RangeManager.MAX_RANGE);
    if (withData) {
        request.setProofValid(true);
        request.addResponse(null, worldStateProofProvider, slots, new ArrayDeque<>());
    }
    return request;
}