-
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.
Merge branch 'master' into dev_mackun
* master: implements disputes protocol (#92)
- Loading branch information
Showing
13 changed files
with
363 additions
and
88 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,218 @@ | ||
| import Utils | ||
|
|
||
| public enum DisputeError: Error { | ||
| case invalidEpoch | ||
| case invalidValidatorIndex | ||
| case invalidJudgementSignature | ||
| case invalidCulpritSigner | ||
| case invalidCulpritSignature | ||
| case invalidFaultSigner | ||
| case invalidFaultSignature | ||
| case verdictsNotSorted | ||
| case culpritsNotSorted | ||
| case faultsNotSorted | ||
| case duplicatedReport | ||
| case judgementsNotSorted | ||
| case invalidJudgementsCount | ||
| case expectInFaults | ||
| case expectInCulprits | ||
| } | ||
|
|
||
| public struct ReportItem: Sendable, Equatable, Codable { | ||
| public var workReport: WorkReport | ||
| public var timeslot: TimeslotIndex | ||
|
|
||
| public init( | ||
| workReport: WorkReport, | ||
| timeslot: TimeslotIndex | ||
| ) { | ||
| self.workReport = workReport | ||
| self.timeslot = timeslot | ||
| } | ||
| } | ||
|
|
||
| public struct DisputePostState: Sendable, Equatable { | ||
| public var judgements: JudgementsState | ||
| public var reports: ConfigFixedSizeArray< | ||
| ReportItem?, | ||
| ProtocolConfig.TotalNumberOfCores | ||
| > | ||
|
|
||
| public init( | ||
| judgements: JudgementsState, | ||
| reports: ConfigFixedSizeArray< | ||
| ReportItem?, | ||
| ProtocolConfig.TotalNumberOfCores | ||
| > | ||
| ) { | ||
| self.judgements = judgements | ||
| self.reports = reports | ||
| } | ||
| } | ||
|
|
||
| public protocol Disputes { | ||
| var judgements: JudgementsState { get } | ||
| var reports: ConfigFixedSizeArray< | ||
| ReportItem?, | ||
| ProtocolConfig.TotalNumberOfCores | ||
| > { get } | ||
| var timeslot: TimeslotIndex { get } | ||
| var currentValidators: ConfigFixedSizeArray< | ||
| ValidatorKey, ProtocolConfig.TotalNumberOfValidators | ||
| > { get } | ||
| var previousValidators: ConfigFixedSizeArray< | ||
| ValidatorKey, ProtocolConfig.TotalNumberOfValidators | ||
| > { get } | ||
|
|
||
| func update(config: ProtocolConfigRef, disputes: ExtrinsicDisputes) throws(DisputeError) -> ( | ||
| state: DisputePostState, | ||
| offenders: [Ed25519PublicKey] | ||
| ) | ||
|
|
||
| mutating func mergeWith(postState: DisputePostState) | ||
| } | ||
|
|
||
| extension Disputes { | ||
| public func update(config: ProtocolConfigRef, disputes: ExtrinsicDisputes) throws(DisputeError) -> ( | ||
| state: DisputePostState, | ||
| offenders: [Ed25519PublicKey] | ||
| ) { | ||
| var newJudgements = judgements | ||
| var newReports = reports | ||
| var offenders: [Ed25519PublicKey] = [] | ||
|
|
||
| let epochLength = UInt32(config.value.epochLength) | ||
| let currentEpoch = timeslot / epochLength | ||
| let lastEpoch = currentEpoch == 0 ? nil : currentEpoch - 1 | ||
|
|
||
| for verdict in disputes.verdicts { | ||
| let isCurrent = verdict.epoch == currentEpoch | ||
| let isLast = verdict.epoch == lastEpoch | ||
| guard isCurrent || isLast else { | ||
| throw .invalidEpoch | ||
| } | ||
|
|
||
| let validators = isCurrent ? currentValidators : previousValidators | ||
|
|
||
| for judgement in verdict.judgements { | ||
| guard let signer = validators[safe: Int(judgement.validatorIndex)]?.ed25519 else { | ||
| throw .invalidValidatorIndex | ||
| } | ||
|
|
||
| let prefix = judgement.isValid ? SigningContext.valid : SigningContext.invalid | ||
| let payload = prefix + verdict.reportHash.data | ||
| guard Ed25519.verify(signature: judgement.signature, message: payload, publicKey: signer) else { | ||
| throw .invalidJudgementSignature | ||
| } | ||
| } | ||
|
|
||
| guard verdict.judgements.isSortedAndUnique(by: { $0.validatorIndex < $1.validatorIndex }) else { | ||
| throw .judgementsNotSorted | ||
| } | ||
| } | ||
|
|
||
| var validSigners = Set<Ed25519PublicKey>(currentValidators.map(\.ed25519)) | ||
| validSigners.formUnion(previousValidators.map(\.ed25519)) | ||
| validSigners.subtract(judgements.punishSet) | ||
|
|
||
| for culprit in disputes.culprits { | ||
| guard validSigners.contains(culprit.validatorKey) else { | ||
| throw .invalidCulpritSigner | ||
| } | ||
|
|
||
| let payload = SigningContext.guarantee + culprit.reportHash.data | ||
| guard Ed25519.verify(signature: culprit.signature, message: payload, publicKey: culprit.validatorKey) else { | ||
| throw .invalidCulpritSignature | ||
| } | ||
|
|
||
| newJudgements.punishSet.insert(culprit.validatorKey) | ||
| offenders.append(culprit.validatorKey) | ||
| } | ||
|
|
||
| for fault in disputes.faults { | ||
| guard validSigners.contains(fault.validatorKey) else { | ||
| throw .invalidFaultSigner | ||
| } | ||
|
|
||
| let prefix = fault.vote ? SigningContext.valid : SigningContext.invalid | ||
| let payload = prefix + fault.reportHash.data | ||
| guard Ed25519.verify(signature: fault.signature, message: payload, publicKey: fault.validatorKey) else { | ||
| throw .invalidFaultSignature | ||
| } | ||
|
|
||
| newJudgements.punishSet.insert(fault.validatorKey) | ||
| offenders.append(fault.validatorKey) | ||
| } | ||
|
|
||
| guard disputes.verdicts.isSortedAndUnique(by: { $0.reportHash < $1.reportHash }) else { | ||
| throw .verdictsNotSorted | ||
| } | ||
|
|
||
| guard disputes.culprits.isSortedAndUnique(by: { $0.validatorKey < $1.validatorKey }) else { | ||
| throw .culpritsNotSorted | ||
| } | ||
|
|
||
| guard disputes.faults.isSortedAndUnique(by: { $0.validatorKey < $1.validatorKey }) else { | ||
| throw .faultsNotSorted | ||
| } | ||
|
|
||
| var allReports = Set(disputes.verdicts.map(\.reportHash)) | ||
| allReports.formUnion(judgements.goodSet) | ||
| allReports.formUnion(judgements.banSet) | ||
| allReports.formUnion(judgements.wonkySet) | ||
|
|
||
| let expectedReportCount = disputes.verdicts.count + judgements.goodSet.count + judgements.banSet.count + judgements.wonkySet.count | ||
|
|
||
| guard allReports.count == expectedReportCount else { | ||
| throw .duplicatedReport | ||
| } | ||
|
|
||
| let votes = disputes.verdicts.map { | ||
| (hash: $0.reportHash, vote: $0.judgements.reduce(into: 0) { $0 += $1.isValid ? 1 : 0 }) | ||
| } | ||
|
|
||
| var tobeRemoved = Set<Data32>() | ||
| let third_validators = config.value.totalNumberOfValidators / 3 | ||
| let two_third_plus_one_validators = config.value.totalNumberOfValidators * 2 / 3 + 1 | ||
| for (hash, vote) in votes { | ||
| if vote == 0 { | ||
| // any verdict containing solely valid judgements | ||
| // implies the same report having at least one valid entry in the faults sequence f | ||
| guard disputes.faults.contains(where: { $0.reportHash == hash }) else { | ||
| throw .expectInFaults | ||
| } | ||
|
|
||
| tobeRemoved.insert(hash) | ||
| newJudgements.banSet.insert(hash) | ||
| } else if vote == third_validators { | ||
| // wonky | ||
| tobeRemoved.insert(hash) | ||
| newJudgements.wonkySet.insert(hash) | ||
| } else if vote == two_third_plus_one_validators { | ||
| // Any verdict containing solely invalid judgements | ||
| // implies the same report having at least two valid entries in the culprits sequence c | ||
| guard disputes.culprits.count(where: { $0.reportHash == hash }) >= 2 else { | ||
| throw .expectInCulprits | ||
| } | ||
|
|
||
| newJudgements.goodSet.insert(hash) | ||
| } else { | ||
| throw .invalidJudgementsCount | ||
| } | ||
| } | ||
|
|
||
| for i in 0 ..< newReports.count { | ||
| if let report = newReports[i]?.workReport { | ||
| let hash = report.hash() | ||
| if tobeRemoved.contains(hash) { | ||
| newReports[i] = nil | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return (state: DisputePostState( | ||
| judgements: newJudgements, | ||
| reports: newReports | ||
| ), offenders: offenders) | ||
| } | ||
| } |
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
Oops, something went wrong.