Examples with ArrayDeque kotlin.collections.ArrayDeque used on opensource projects

Example 6 with ArrayDeque

use of kotlin.collections.ArrayDeque in project besu by hyperledger.

the class GetStorageRangeMessage method range.

public StorageRange range(final boolean withRequestId) {
    final ArrayDeque<Bytes32> hashes = new ArrayDeque<>();
    final RLPInput input = new BytesValueRLPInput(data, false);
    input.enterList();
    if (withRequestId)
        input.skipNext();
    final Hash worldStateRootHash = Hash.wrap(Bytes32.wrap(input.readBytes32()));
    final ImmutableStorageRange.Builder range = ImmutableStorageRange.builder().worldStateRootHash(getRootHash().orElse(worldStateRootHash));
    input.enterList();
    while (!input.isEndOfCurrentList()) {
        hashes.add(input.readBytes32());
    }
    range.hashes(hashes);
    input.leaveList();
    if (input.nextIsNull()) {
        input.skipNext();
        range.startKeyHash(Hash.ZERO);
    } else {
        range.startKeyHash(Hash.wrap(Bytes32.wrap(input.readBytes32())));
    }
    if (input.nextIsNull()) {
        input.skipNext();
        range.endKeyHash(Hash.ZERO);
    } else {
        range.endKeyHash(Hash.wrap(Bytes32.wrap(input.readBytes32())));
    }
    range.responseBytes(input.readBigIntegerScalar());
    input.leaveList();
    return range.build();
}

Example 7 with ArrayDeque

use of kotlin.collections.ArrayDeque in project besu by hyperledger.

the class StorageRangeMessage method create.

public static StorageRangeMessage create(final Optional<BigInteger> requestId, final ArrayDeque<TreeMap<Bytes32, Bytes>> slots, final List<Bytes> proof) {
    final BytesValueRLPOutput tmp = new BytesValueRLPOutput();
    tmp.startList();
    requestId.ifPresent(tmp::writeBigIntegerScalar);
    tmp.writeList(slots, (accountList, accountRlpOutput) -> accountRlpOutput.writeList(accountList.entrySet(), (entry, slotRlpOutput) -> {
        slotRlpOutput.startList();
        slotRlpOutput.writeBytes(entry.getKey());
        slotRlpOutput.writeBytes(entry.getValue());
        slotRlpOutput.endList();
    }));
    tmp.writeList(proof, (bytes, rlpOutput) -> rlpOutput.writeBytes(bytes));
    tmp.endList();
    return new StorageRangeMessage(tmp.encoded());
}

Example 8 with ArrayDeque

use of kotlin.collections.ArrayDeque 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;
}

Example 9 with ArrayDeque

use of kotlin.collections.ArrayDeque in project besu by hyperledger.

the class TaskGenerator method createAccountRequest.

public static List<Task<SnapDataRequest>> createAccountRequest(final boolean withData) {
    final WorldStateStorage worldStateStorage = new InMemoryKeyValueStorageProvider().createWorldStateStorage(DataStorageFormat.FOREST);
    final WorldStateProofProvider worldStateProofProvider = new WorldStateProofProvider(worldStateStorage);
    final MerklePatriciaTrie<Bytes32, Bytes> trie = TrieGenerator.generateTrie(worldStateStorage, 1);
    final RangeStorageEntriesCollector collector = RangeStorageEntriesCollector.createCollector(Bytes32.ZERO, RangeManager.MAX_RANGE, 1, Integer.MAX_VALUE);
    final TrieIterator<Bytes> visitor = RangeStorageEntriesCollector.createVisitor(collector);
    final TreeMap<Bytes32, Bytes> accounts = (TreeMap<Bytes32, Bytes>) trie.entriesFrom(root -> RangeStorageEntriesCollector.collectEntries(collector, visitor, root, Bytes32.ZERO));
    final Hash rootHash = Hash.wrap(trie.getRootHash());
    final AccountRangeDataRequest accountRangeDataRequest = SnapDataRequest.createAccountRangeDataRequest(rootHash, RangeManager.MIN_RANGE, RangeManager.MAX_RANGE);
    if (withData) {
        accountRangeDataRequest.addResponse(worldStateProofProvider, accounts, new ArrayDeque<>());
    }
    final StateTrieAccountValue stateTrieAccountValue = StateTrieAccountValue.readFrom(RLP.input(accounts.firstEntry().getValue()));
    final Hash accountHash = Hash.wrap(accounts.firstKey());
    final StorageRangeDataRequest storageRangeDataRequest = createStorageRangeDataRequest(worldStateProofProvider, worldStateStorage, rootHash, accountHash, stateTrieAccountValue.getStorageRoot(), withData);
    final BytecodeRequest bytecodeRequest = createBytecodeDataRequest(worldStateStorage, rootHash, accountHash, stateTrieAccountValue.getCodeHash(), withData);
    return List.of(new StubTask(accountRangeDataRequest), new StubTask(storageRangeDataRequest), new StubTask(bytecodeRequest));
}

Example 10 with ArrayDeque

use of kotlin.collections.ArrayDeque in project besu by hyperledger.

the class StorageRangeMessageTest method roundTripTest.

@Test
public void roundTripTest() {
    final ArrayDeque<TreeMap<Bytes32, Bytes>> keys = new ArrayDeque<>();
    final TreeMap<Bytes32, Bytes> storage = new TreeMap<>();
    storage.put(Hash.wrap(Bytes32.leftPad(Bytes.of(1))), Bytes32.random());
    keys.add(storage);
    final List<Bytes> proofs = new ArrayList<>();
    proofs.add(Bytes32.random());
    // Perform round-trip transformation
    final MessageData initialMessage = StorageRangeMessage.create(keys, proofs);
    final MessageData raw = new RawMessage(SnapV1.STORAGE_RANGE, initialMessage.getData());
    final StorageRangeMessage message = StorageRangeMessage.readFrom(raw);
    // check match originals.
    final StorageRangeMessage.SlotRangeData range = message.slotsData(false);
    Assertions.assertThat(range.slots()).isEqualTo(keys);
    Assertions.assertThat(range.proofs()).isEqualTo(proofs);
}