forked from Consensys/teku
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support for slashing interchange format tests (Consensys#8185)
- Loading branch information
1 parent
e9ef718
commit 6a84100
Showing
12 changed files
with
324 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
22 changes: 22 additions & 0 deletions
22
data/dataexchange/src/test/resources/format2_minimal_different_genesis_validators_root.json
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,22 @@ | ||
| { | ||
| "metadata": { | ||
| "interchange_format_version": "5", | ||
| "genesis_validators_root": "0x0000000000000000000000000000000000000000000000000000000000123457" | ||
| }, | ||
| "data": [ | ||
| { | ||
| "pubkey": "0xb845089a1457f811bfc000588fbb4e713669be8ce060ea6be3c6ece09afc3794106c91ca73acda5e5457122d58723bed", | ||
| "signed_blocks": [ | ||
| { | ||
| "slot": "81952" | ||
| } | ||
| ], | ||
| "signed_attestations": [ | ||
| { | ||
| "source_epoch": "2290", | ||
| "target_epoch": "3007" | ||
| } | ||
| ] | ||
| } | ||
| ] | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
161 changes: 161 additions & 0 deletions
161
...nce/phase0/slashing_protection_interchange/SlashingProtectionInterchangeTestExecutor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,161 @@ | ||
| /* | ||
| * Copyright Consensys Software Inc., 2024 | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with | ||
| * the License. You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on | ||
| * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the | ||
| * specific language governing permissions and limitations under the License. | ||
| */ | ||
|
|
||
| package tech.pegasys.teku.reference.phase0.slashing_protection_interchange; | ||
|
|
||
| import static org.assertj.core.api.Assertions.assertThat; | ||
|
|
||
| import com.fasterxml.jackson.annotation.JsonProperty; | ||
| import java.io.IOException; | ||
| import java.io.UncheckedIOException; | ||
| import java.nio.file.DirectoryStream; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Path; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.Optional; | ||
| import org.apache.logging.log4j.LogManager; | ||
| import org.apache.logging.log4j.Logger; | ||
| import org.apache.tuweni.bytes.Bytes32; | ||
| import tech.pegasys.teku.bls.BLSPublicKey; | ||
| import tech.pegasys.teku.data.SlashingProtectionImporter; | ||
| import tech.pegasys.teku.data.slashinginterchange.SlashingProtectionInterchangeFormat; | ||
| import tech.pegasys.teku.ethtests.finder.TestDefinition; | ||
| import tech.pegasys.teku.infrastructure.io.SyncDataAccessor; | ||
| import tech.pegasys.teku.infrastructure.unsigned.UInt64; | ||
| import tech.pegasys.teku.reference.TestDataUtils; | ||
| import tech.pegasys.teku.reference.TestExecutor; | ||
| import tech.pegasys.teku.reference.phase0.slashing_protection_interchange.SlashingProtectionInterchangeTestExecutor.TestData.Step; | ||
| import tech.pegasys.teku.spec.signatures.LocalSlashingProtector; | ||
|
|
||
| public class SlashingProtectionInterchangeTestExecutor implements TestExecutor { | ||
|
|
||
| private static final Logger LOG = LogManager.getLogger(); | ||
|
|
||
| @Override | ||
| public void runTest(final TestDefinition testDefinition) throws Throwable { | ||
| final TestData testData = | ||
| TestDataUtils.loadJson(testDefinition, testDefinition.getTestName(), TestData.class); | ||
|
|
||
| // our implementation fails when importing one of the keys in an interchange, which is already | ||
| // in our slashprotection directory with a different genesis validators root. However, the test | ||
| // does not import any keys. This case is covered by | ||
| // SlashingProtectionImporterTest#shouldFailImportingIfValidatorExistingRecordHasDifferentGenesisValidatorsRoot() | ||
| if (testData.name.startsWith("wrong_genesis_validators_root")) { | ||
| LOG.info("Skipping {}", testData.name); | ||
| return; | ||
| } | ||
|
|
||
| LOG.info("Running {}", testData.name); | ||
|
|
||
| final Path slashingProtectionPath = Files.createTempDirectory("slashprotection"); | ||
| try { | ||
| runTest(testData, slashingProtectionPath); | ||
| } finally { | ||
| deleteDirectory(slashingProtectionPath); | ||
| } | ||
| } | ||
|
|
||
| private void runTest(final TestData testData, final Path slashingProtectionPath) { | ||
| final SlashingProtectionImporter importer = | ||
| new SlashingProtectionImporter(slashingProtectionPath); | ||
| final LocalSlashingProtector slashingProtector = | ||
| new LocalSlashingProtector( | ||
| SyncDataAccessor.create(slashingProtectionPath), slashingProtectionPath); | ||
| testData.steps.forEach(step -> runStep(step, importer, slashingProtector)); | ||
| } | ||
|
|
||
| private void runStep( | ||
| final Step step, | ||
| final SlashingProtectionImporter importer, | ||
| final LocalSlashingProtector slashingProtector) { | ||
| final Map<BLSPublicKey, String> importErrors = importInterchange(importer, step.interchange); | ||
| if (step.shouldSucceed) { | ||
| assertThat(importErrors).isEmpty(); | ||
| } else { | ||
| assertThat(importErrors).isNotEmpty(); | ||
| } | ||
| final Bytes32 genesisValidatorsRoot = step.interchange.metadata.genesisValidatorsRoot; | ||
| step.blocks.forEach( | ||
| block -> | ||
| assertThat( | ||
| slashingProtector.maySignBlock(block.pubkey, genesisValidatorsRoot, block.slot)) | ||
| .isCompletedWithValue(block.shouldSucceed)); | ||
| step.attestations.forEach( | ||
| attestation -> | ||
| assertThat( | ||
| slashingProtector.maySignAttestation( | ||
| attestation.pubkey, | ||
| genesisValidatorsRoot, | ||
| attestation.sourceEpoch, | ||
| attestation.targetEpoch)) | ||
| .isCompletedWithValue(attestation.shouldSucceed)); | ||
| } | ||
|
|
||
| private Map<BLSPublicKey, String> importInterchange( | ||
| final SlashingProtectionImporter importer, | ||
| final SlashingProtectionInterchangeFormat interchange) { | ||
| try { | ||
| final Path importFile = Files.createTempFile("import", ".json"); | ||
| TestDataUtils.writeJsonToFile(interchange, importFile); | ||
| final Optional<String> initialiseError = importer.initialise(importFile.toFile()); | ||
| assertThat(initialiseError).isEmpty(); | ||
| // cleanup | ||
| Files.delete(importFile); | ||
| } catch (IOException ex) { | ||
| throw new UncheckedIOException(ex); | ||
| } | ||
| return importer.updateLocalRecords(status -> LOG.info("Import status: " + status)); | ||
| } | ||
|
|
||
| private void deleteDirectory(final Path dir) { | ||
| try (DirectoryStream<Path> files = Files.newDirectoryStream(dir)) { | ||
| for (Path file : files) { | ||
| if (Files.isRegularFile(file)) { | ||
| Files.delete(file); | ||
| } | ||
| } | ||
| Files.delete(dir); | ||
| } catch (IOException ex) { | ||
| throw new UncheckedIOException(ex); | ||
| } | ||
| } | ||
|
|
||
| public record TestData( | ||
| String name, | ||
| @JsonProperty("genesis_validators_root") Bytes32 genesisValidatorsRoot, | ||
| List<Step> steps) { | ||
|
|
||
| public record Step( | ||
| @JsonProperty("should_succeed") boolean shouldSucceed, | ||
| // we don't fail importing when the interchange contains slashable data, so can safely | ||
| // ignore this field in the tests | ||
| @JsonProperty("contains_slashable_data") boolean containsSlashableData, | ||
| SlashingProtectionInterchangeFormat interchange, | ||
| List<Block> blocks, | ||
| List<Attestation> attestations) {} | ||
|
|
||
| public record Block( | ||
| BLSPublicKey pubkey, | ||
| UInt64 slot, | ||
| @JsonProperty("signing_root") Bytes32 signingRoot, | ||
| @JsonProperty("should_succeed") boolean shouldSucceed) {} | ||
|
|
||
| public record Attestation( | ||
| BLSPublicKey pubkey, | ||
| @JsonProperty("source_epoch") UInt64 sourceEpoch, | ||
| @JsonProperty("target_epoch") UInt64 targetEpoch, | ||
| @JsonProperty("signing_root") Bytes32 signingRoot, | ||
| @JsonProperty("should_succeed") boolean shouldSucceed) {} | ||
| } | ||
| } |
Oops, something went wrong.