CipherLabs
/
besu
mirror of https://github.com/hyperledger/besu
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

(internal) Refactor PrivacyReorgTest to use mock enclave (#3103)

Browse Source 
* init

Signed-off-by: Frank Li <b439988l@gmail.com>

* working test

Signed-off-by: Frank Li <b439988l@gmail.com>

* WIP | 3/4 tests working

Signed-off-by: Frank Li <b439988l@gmail.com>
pull/3116/head
Frank Li 3 years ago committed by GitHub
parent 0b4c03c8ca
commit 78758cc796
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 87 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 124
      besu/src/test/java/org/hyperledger/besu/PrivacyReorgTest.java

124
besu/src/test/java/org/hyperledger/besu/PrivacyReorgTest.java
Unescape View File

@ -32,7 +32,7 @@ import org.hyperledger.besu.datatypes.Wei;
import org.hyperledger.besu.enclave.Enclave;
import org.hyperledger.besu.enclave.EnclaveFactory;
import org.hyperledger.besu.enclave.types.PrivacyGroup;
import org.hyperledger.besu.enclave.types.SendResponse;
import org.hyperledger.besu.enclave.types.ReceiveResponse;
import org.hyperledger.besu.ethereum.GasLimitCalculator;
import org.hyperledger.besu.ethereum.ProtocolContext;
import org.hyperledger.besu.ethereum.chain.DefaultBlockchain;
@ -61,25 +61,19 @@ import org.hyperledger.besu.metrics.noop.NoOpMetricsSystem;
import org.hyperledger.besu.plugin.data.Restriction;
import org.hyperledger.besu.plugin.data.TransactionType;
import org.hyperledger.besu.testutil.TestClock;
import org.hyperledger.enclave.testutil.EnclaveKeyConfiguration;
import org.hyperledger.enclave.testutil.EnclaveTestHarness;
import org.hyperledger.enclave.testutil.OrionTestHarnessFactory;
import java.io.IOException;
import java.math.BigInteger;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import com.google.common.base.Supplier;
import com.google.common.base.Suppliers;
import io.vertx.core.Vertx;
import org.apache.tuweni.bytes.Bytes;
import org.apache.tuweni.bytes.Bytes32;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
@ -136,31 +130,52 @@ public class PrivacyReorgTest {
private final BlockDataGenerator gen = new BlockDataGenerator();
private BesuController besuController;
private EnclaveTestHarness enclave;
private PrivateStateRootResolver privateStateRootResolver;
private PrivacyParameters privacyParameters;
private RestrictedDefaultPrivacyController privacyController;
private Enclave mockEnclave;
private Transaction privacyMarkerTransaction;
@Before
public void setUp() throws IOException {
enclave =
OrionTestHarnessFactory.create(
"orion",
folder.newFolder().toPath(),
new EnclaveKeyConfiguration("enclavePublicKey", "enclavePrivateKey"));
enclave.start();
mockEnclave = mock(Enclave.class);
final BytesValueRLPOutput rlpOutput = new BytesValueRLPOutput();
PRIVATE_TRANSACTION.writeTo(rlpOutput);
when(mockEnclave.receive(any()))
.thenReturn(
new ReceiveResponse(
rlpOutput.encoded().toBase64String().getBytes(StandardCharsets.UTF_8),
PRIVACY_GROUP_BYTES32.toBase64String(),
ENCLAVE_PUBLIC_KEY.toBase64String()));
privacyMarkerTransaction =
Transaction.builder()
.type(TransactionType.FRONTIER)
.chainId(BigInteger.valueOf(1337))
.gasLimit(60000)
.gasPrice(Wei.of(1000))
.nonce(0)
.payload(Bytes32.random())
.to(DEFAULT_PRIVACY)
.value(Wei.ZERO)
.signAndBuild(KEY_PAIR);
// Create Storage
final Path dataDir = folder.newFolder().toPath();
// Configure Privacy
EnclaveFactory enclaveFactory = mock(EnclaveFactory.class);
when(enclaveFactory.createVertxEnclave(any())).thenReturn(mockEnclave);
privacyParameters =
new PrivacyParameters.Builder()
.setEnabled(true)
.setStorageProvider(createKeyValueStorageProvider())
.setEnclaveUrl(enclave.clientUrl())
.setEnclaveFactory(new EnclaveFactory(Vertx.vertx()))
.setEnclaveUrl(URI.create("http//1.1.1.1:1234"))
.setEnclaveFactory(enclaveFactory)
.build();
privacyParameters.setPrivacyUserId(ENCLAVE_PUBLIC_KEY.toBase64String());
privacyController = mock(RestrictedDefaultPrivacyController.class);
when(privacyController.findPrivacyGroupByGroupId(any(), any()))
@ -192,11 +207,6 @@ public class PrivacyReorgTest {
.build();
}
@After
public void tearDown() {
enclave.stop();
}
@Test
public void privacyGroupHeadIsTracked() {
// Setup an initial blockchain with one private transaction
@ -204,13 +214,11 @@ public class PrivacyReorgTest {
final DefaultBlockchain blockchain = (DefaultBlockchain) protocolContext.getBlockchain();
final PrivateStateStorage privateStateStorage = privacyParameters.getPrivateStateStorage();
final Transaction privateMarkerTransaction =
buildMarkerTransaction(getEnclaveKey(enclave.clientUrl()));
final Block firstBlock =
gen.block(
getBlockOptionsWithTransaction(
blockchain.getGenesisBlock(),
privateMarkerTransaction,
privacyMarkerTransaction,
FIRST_BLOCK_WITH_SINGLE_TRANSACTION_STATE_ROOT));
appendBlock(besuController, blockchain, protocolContext, firstBlock);
@ -249,7 +257,7 @@ public class PrivacyReorgTest {
gen.block(
getBlockOptionsWithTransaction(
blockchain.getGenesisBlock(),
buildMarkerTransaction(getEnclaveKey(enclave.clientUrl())),
privacyMarkerTransaction,
FIRST_BLOCK_WITH_SINGLE_TRANSACTION_STATE_ROOT));
appendBlock(besuController, blockchain, protocolContext, firstBlock);
@ -288,9 +296,7 @@ public class PrivacyReorgTest {
final Block secondBlock =
gen.block(
getBlockOptionsWithTransaction(
firstBlock,
buildMarkerTransaction(getEnclaveKey(enclave.clientUrl())),
secondBlockStateRoot));
firstBlock, privacyMarkerTransaction, secondBlockStateRoot));
appendBlock(besuController, blockchain, protocolContext, firstBlock);
appendBlock(besuController, blockchain, protocolContext, secondBlock);
@ -337,7 +343,7 @@ public class PrivacyReorgTest {
gen.block(
getBlockOptionsWithTransaction(
blockchain.getGenesisBlock(),
buildMarkerTransaction(getEnclaveKey(enclave.clientUrl())),
privacyMarkerTransaction,
FIRST_BLOCK_WITH_SINGLE_TRANSACTION_STATE_ROOT));
appendBlock(besuController, blockchain, protocolContext, firstBlock);
@ -384,7 +390,7 @@ public class PrivacyReorgTest {
gen.block(
getBlockOptionsWithTransactionAndDifficulty(
secondForkBlock,
buildMarkerTransaction(getEnclaveKey(enclave.clientUrl())),
privacyMarkerTransaction,
secondForkBlock.getHeader().getDifficulty().plus(10L),
thirdForkBlockStateRoot));
@ -412,62 +418,6 @@ public class PrivacyReorgTest {
return new InMemoryPrivacyStorageProvider();
}
private Bytes getEnclaveKey(final URI enclaveURI) {
final Enclave enclave = new EnclaveFactory(Vertx.vertx()).createVertxEnclave(enclaveURI);
final SendResponse sendResponse =
sendRequest(enclave, PRIVATE_TRANSACTION, ENCLAVE_PUBLIC_KEY.toBase64String());
final Bytes payload = Bytes.fromBase64String(sendResponse.getKey());
// If the key has 0 bytes generate a new key.
// This is to keep the gasUsed constant allowing
// hard-coded receipt roots in the block headers
for (int i = 0; i < payload.size(); i++) {
if (payload.get(i) == 0) {
return getEnclaveKey(enclaveURI);
}
}
return payload;
}
private SendResponse sendRequest(
final Enclave enclave,
final PrivateTransaction privateTransaction,
final String privacyUserId) {
final BytesValueRLPOutput rlpOutput = new BytesValueRLPOutput();
privateTransaction.writeTo(rlpOutput);
final String payload = rlpOutput.encoded().toBase64String();
if (privateTransaction.getPrivacyGroupId().isPresent()) {
return enclave.send(
payload, privacyUserId, privateTransaction.getPrivacyGroupId().get().toBase64String());
} else {
final List<String> privateFor =
privateTransaction.getPrivateFor().get().stream()
.map(Bytes::toBase64String)
.collect(Collectors.toList());
if (privateFor.isEmpty()) {
privateFor.add(privateTransaction.getPrivateFrom().toBase64String());
}
return enclave.send(
payload, privateTransaction.getPrivateFrom().toBase64String(), privateFor);
}
}
private Transaction buildMarkerTransaction(final Bytes payload) {
return Transaction.builder()
.type(TransactionType.FRONTIER)
.chainId(BigInteger.valueOf(1337))
.gasLimit(60000)
.gasPrice(Wei.of(1000))
.nonce(0)
.payload(payload)
.to(DEFAULT_PRIVACY)
.value(Wei.ZERO)
.signAndBuild(KEY_PAIR);
}
private void assertPrivateStateRoot(
final PrivateStateRootResolver privateStateRootResolver,
final DefaultBlockchain blockchain,

Write Preview
Loading…
Cancel
Save