use of org.aion.zero.impl.db.DetailsDataStore.RLPContractDetails in project aion by aionnetwork.
the class FvmContractDetailsTest method testDecode_withExternalStorageAndMultiCode.
@Test
public void testDecode_withExternalStorageAndMultiCode() {
AionAddress address = new AionAddress(RandomUtils.nextBytes(AionAddress.LENGTH));
byte[] storageHash = RandomUtils.nextBytes(32);
RLPElement root = mock(RLPItem.class);
when(root.getRLPData()).thenReturn(storageHash);
byte[] codeBytes1 = RandomUtils.nextBytes(100);
byte[] codeBytes2 = RandomUtils.nextBytes(100);
RLPList code = new RLPList();
code.add(new RLPItem(codeBytes1));
code.add(new RLPItem(codeBytes2));
RLPContractDetails input = new RLPContractDetails(address, true, root, null, code);
FvmContractDetails details = FvmContractDetails.decodeAtRoot(input, mockDatabase, storageHash);
assertThat(details.address).isEqualTo(address);
// because it uses the setCodes method
assertThat(details.isDirty()).isTrue();
assertThat(details.isDeleted()).isFalse();
assertThat(details.getCodes().size()).isEqualTo(2);
assertThat(details.getCodes().values()).contains(ByteArrayWrapper.wrap(codeBytes1));
assertThat(details.getCodes().values()).contains(ByteArrayWrapper.wrap(codeBytes2));
assertThat(details.getCode(h256(codeBytes1))).isEqualTo(codeBytes1);
assertThat(details.getCode(h256(codeBytes2))).isEqualTo(codeBytes2);
assertThat(details.getStorageHash()).isEqualTo(storageHash);
}
use of org.aion.zero.impl.db.DetailsDataStore.RLPContractDetails in project aion by aionnetwork.
the class DetailsDataStoreTest method testDecodingWithSizeFiveInternal.
@Test
public void testDecodingWithSizeFiveInternal() {
AionAddress address = new AionAddress(RandomUtils.nextBytes(AionAddress.LENGTH));
byte[] code = RandomUtils.nextBytes(512);
// create encoding
byte[] rlpAddress = RLP.encodeElement(address.toByteArray());
byte[] rlpIsExternalStorage = RLP.encodeByte((byte) 0);
byte[] rlpStorageRoot = RLP.encodeElement(EMPTY_BYTE_ARRAY);
SecureTrie trie = new SecureTrie(null);
trie.update(RandomUtils.nextBytes(32), RandomUtils.nextBytes(32));
byte[] rlpStorageTrie = RLP.encodeElement(trie.serialize());
byte[] rlpCode = RLP.encodeElement(code);
byte[] encoding = RLP.encodeList(rlpAddress, rlpIsExternalStorage, rlpStorageRoot, rlpStorageTrie, rlpCode);
// decode
RLPContractDetails details = DetailsDataStore.fromEncoding(encoding);
assertThat(details.address).isEqualTo(address);
assertThat(details.isExternalStorage).isFalse();
assertThat(details.storageRoot.getRLPData()).isEqualTo(EMPTY_BYTE_ARRAY);
assertThat(details.storageTrie.getRLPData()).isEqualTo(trie.serialize());
assertThat(details.code.getRLPData()).isEqualTo(code);
}
use of org.aion.zero.impl.db.DetailsDataStore.RLPContractDetails in project aion by aionnetwork.
the class DetailsDataStoreTest method testDecodingWithSizeThree.
@Test
public void testDecodingWithSizeThree() {
AionAddress address = new AionAddress(RandomUtils.nextBytes(AionAddress.LENGTH));
byte[] code1 = RandomUtils.nextBytes(512);
byte[] code2 = RandomUtils.nextBytes(512);
// create encoding
byte[] rlpAddress = RLP.encodeElement(address.toByteArray());
byte[] root = RandomUtils.nextBytes(32);
byte[] rlpStorageRoot = RLP.encodeElement(root);
byte[] rlpCode = RLP.encodeList(RLP.encodeElement(code1), RLP.encodeElement(code2));
byte[] encoding = RLP.encodeList(rlpAddress, rlpStorageRoot, rlpCode);
// decode
RLPContractDetails details = DetailsDataStore.fromEncoding(encoding);
assertThat(details.address).isEqualTo(address);
assertThat(details.isExternalStorage).isTrue();
assertThat(details.storageRoot.getRLPData()).isEqualTo(root);
assertThat(details.storageTrie).isNull();
assertThat(((SharedRLPList) details.code).get(0).getRLPData()).isEqualTo(code1);
assertThat(((SharedRLPList) details.code).get(1).getRLPData()).isEqualTo(code2);
}
use of org.aion.zero.impl.db.DetailsDataStore.RLPContractDetails in project aion by aionnetwork.
the class AvmContractDetailsTest method testDecode_withInLineStorageAndTransition.
@Test
public void testDecode_withInLineStorageAndTransition() {
SecureTrie trie = new SecureTrie(null);
Map<ByteArrayWrapper, ByteArrayWrapper> storage = new HashMap<>();
for (int i = 0; i < 3; i++) {
byte[] key = RandomUtils.nextBytes(32);
byte[] value = RandomUtils.nextBytes(100);
trie.update(key, RLP.encodeElement(value));
storage.put(ByteArrayWrapper.wrap(key), ByteArrayWrapper.wrap(value));
}
RLPElement storageTrie = RLP.decode2SharedList(trie.serialize());
AionAddress address = new AionAddress(RandomUtils.nextBytes(AionAddress.LENGTH));
byte[] codeBytes = RandomUtils.nextBytes(100);
RLPElement code = mock(SharedRLPItem.class);
when(code.getRLPData()).thenReturn(codeBytes);
byte[] rootHash = RandomUtils.nextBytes(32);
RLPElement root = mock(SharedRLPItem.class);
when(root.getRLPData()).thenReturn(rootHash);
byte[] storageHash = RandomUtils.nextBytes(32);
byte[] graphHash = RandomUtils.nextBytes(32);
byte[] graphBytes = RandomUtils.nextBytes(100);
when(mockDatabase.get(rootHash)).thenReturn(Optional.of(RLP.encodeList(RLP.encodeElement(storageHash), RLP.encodeElement(graphHash))));
when(mockDatabase.get(graphHash)).thenReturn(Optional.of(graphBytes));
Logger log = mock(Logger.class);
ByteArrayKeyValueDatabase db = new MockDB("db", log);
db.open();
assertThat(db.isEmpty()).isTrue();
RLPContractDetails input = new RLPContractDetails(address, false, root, storageTrie, code);
AvmContractDetails details = AvmContractDetails.decodeAtRoot(input, db, mockDatabase, rootHash);
assertThat(details.address).isEqualTo(address);
// because it uses the setCodes method
assertThat(details.isDirty()).isTrue();
assertThat(details.isDeleted()).isFalse();
assertThat(details.getObjectGraph()).isEqualTo(graphBytes);
assertThat(details.getCodes().size()).isEqualTo(1);
assertThat(details.getCodes().values()).contains(ByteArrayWrapper.wrap(codeBytes));
assertThat(details.getCode(h256(codeBytes))).isEqualTo(codeBytes);
storageHash = trie.getRootHash();
byte[] concatenated = new byte[storageHash.length + graphHash.length];
System.arraycopy(storageHash, 0, concatenated, 0, storageHash.length);
System.arraycopy(graphHash, 0, concatenated, storageHash.length, graphHash.length);
assertThat(details.getStorageHash()).isEqualTo(h256(concatenated));
for (ByteArrayWrapper key : storage.keySet()) {
assertThat(details.get(key)).isEqualTo(storage.get(key));
}
assertThat(db.isEmpty()).isFalse();
}
use of org.aion.zero.impl.db.DetailsDataStore.RLPContractDetails in project aion by aionnetwork.
the class AvmContractDetailsTest method testDecode_withMissingConcatenatedData.
@Test(expected = IllegalArgumentException.class)
public void testDecode_withMissingConcatenatedData() {
AionAddress address = new AionAddress(RandomUtils.nextBytes(AionAddress.LENGTH));
byte[] rootHash = RandomUtils.nextBytes(32);
RLPElement root = mock(RLPItem.class);
when(root.getRLPData()).thenReturn(rootHash);
byte[] codeBytes1 = RandomUtils.nextBytes(100);
byte[] codeBytes2 = RandomUtils.nextBytes(100);
RLPList code = new RLPList();
code.add(new RLPItem(codeBytes1));
code.add(new RLPItem(codeBytes2));
when(mockDatabase.get(rootHash)).thenReturn(Optional.empty());
RLPContractDetails input = new RLPContractDetails(address, true, root, null, code);
AvmContractDetails.decodeAtRoot(input, mockDatabase, mockDatabase, rootHash);
}
Aggregations