def process_deposit(state: BeaconState, deposit: Deposit, slots_per_epoch: int,
deposit_contract_tree_depth: int) -> BeaconState:
"""
Process a deposit from Ethereum 1.0.
"""
validate_deposit(state, deposit, deposit_contract_tree_depth)
# Increment the next deposit index we are expecting. Note that this
# needs to be done here because while the deposit contract will never
# create an invalid Merkle branch, it may admit an invalid deposit
# object, and we need to be able to skip over it
state = state.copy(deposit_index=state.deposit_index + 1, )
validator_pubkeys = tuple(v.pubkey for v in state.validator_registry)
deposit_input = deposit.deposit_data.deposit_input
pubkey = deposit_input.pubkey
amount = deposit.deposit_data.amount
withdrawal_credentials = deposit_input.withdrawal_credentials
if pubkey not in validator_pubkeys:
# Verify the proof of possession
proof_is_valid = bls.verify(
pubkey=pubkey,
message_hash=deposit_input.signing_root,
signature=deposit_input.signature,
domain=get_domain(
state.fork,
state.current_epoch(slots_per_epoch),
SignatureDomain.DOMAIN_DEPOSIT,
),
)
if not proof_is_valid:
return state
validator = Validator.create_pending_validator(
pubkey=pubkey,
withdrawal_credentials=withdrawal_credentials,
)
# Note: In phase 2 registry indices that has been withdrawn for a long time
# will be recycled.
state = add_pending_validator(
state,
validator,
amount,
)
else:
# Top-up - increase balance by deposit
index = ValidatorIndex(validator_pubkeys.index(pubkey))
validator = state.validator_registry[index]
# Update validator's balance and state
state = state.update_validator_balance(
validator_index=index,
balance=state.validator_balances[index] + amount,
)
return state
def _settle_penality_to_validator_and_whistleblower(
*, state: BeaconState, validator_index: ValidatorIndex,
latest_penalized_exit_length: int, whistleblower_reward_quotient: int,
epoch_length: int, max_deposit: Ether) -> BeaconState:
"""
Apply penality/reward to validator and whistleblower and update the meta data
More intuitive pseudo-code:
current_epoch_penalization_index = (state.slot // EPOCH_LENGTH) % LATEST_PENALIZED_EXIT_LENGTH
state.latest_penalized_exit_balances[current_epoch_penalization_index] += (
get_effective_balance(state, index)
)
whistleblower_index = get_beacon_proposer_index(state, state.slot)
whistleblower_reward = get_effective_balance(state, index) // WHISTLEBLOWER_REWARD_QUOTIENT
state.validator_balances[whistleblower_index] += whistleblower_reward
state.validator_balances[index] -= whistleblower_reward
validator.penalized_slot = state.slot
"""
# Update `state.latest_penalized_exit_balances`
current_epoch_penalization_index = (
state.slot // epoch_length) % latest_penalized_exit_length
effective_balance = get_effective_balance(
state.validator_balances,
validator_index,
max_deposit,
)
penalized_exit_balance = (
state.latest_penalized_exit_balances[current_epoch_penalization_index]
+ effective_balance)
latest_penalized_exit_balances = update_tuple_item(
tuple_data=state.latest_penalized_exit_balances,
index=current_epoch_penalization_index,
new_value=penalized_exit_balance,
)
state = state.copy(
latest_penalized_exit_balances=latest_penalized_exit_balances, )
# Update whistleblower's balance
whistleblower_reward = (effective_balance // whistleblower_reward_quotient)
whistleblower_index = get_beacon_proposer_index(state, state.slot,
epoch_length)
state = state.update_validator_balance(
whistleblower_index,
state.validator_balances[whistleblower_index] + whistleblower_reward,
)
# Update validator's balance and `penalized_slot` field
validator = state.validator_registry[validator_index]
validator = validator.copy(penalized_slot=state.slot, )
state = state.update_validator(
validator_index,
validator,
state.validator_balances[validator_index] - whistleblower_reward,
)
return state
def process_deposit(*, state: BeaconState, pubkey: BLSPubkey, amount: Gwei,
proof_of_possession: BLSSignature,
withdrawal_credentials: Hash32, randao_commitment: Hash32,
custody_commitment: Hash32,
epoch_length: int) -> BeaconState:
"""
Process a deposit from Ethereum 1.0.
"""
validate_proof_of_possession(
state=state,
pubkey=pubkey,
proof_of_possession=proof_of_possession,
withdrawal_credentials=withdrawal_credentials,
randao_commitment=randao_commitment,
custody_commitment=custody_commitment,
epoch_length=epoch_length,
)
validator_pubkeys = tuple(v.pubkey for v in state.validator_registry)
if pubkey not in validator_pubkeys:
validator = ValidatorRecord.create_pending_validator(
pubkey=pubkey,
withdrawal_credentials=withdrawal_credentials,
randao_commitment=randao_commitment,
custody_commitment=custody_commitment,
)
# Note: In phase 2 registry indices that has been withdrawn for a long time
# will be recycled.
state = add_pending_validator(
state,
validator,
amount,
)
else:
# Top-up - increase balance by deposit
index = ValidatorIndex(validator_pubkeys.index(pubkey))
validator = state.validator_registry[index]
if validator.withdrawal_credentials != withdrawal_credentials:
raise ValidationError("`withdrawal_credentials` are incorrect:\n"
"\texpected: %s, found: %s" % (
validator.withdrawal_credentials,
validator.withdrawal_credentials,
))
# Update validator's balance and state
state = state.update_validator_balance(
validator_index=index,
balance=state.validator_balances[index] + amount,
)
return state
def process_slashings(state: BeaconState, config: BeaconConfig) -> BeaconState:
"""
Process the slashings.
"""
latest_slashed_exit_length = config.LATEST_SLASHED_EXIT_LENGTH
max_deposit_amount = config.MAX_DEPOSIT_AMOUNT
current_epoch = state.current_epoch(config.SLOTS_PER_EPOCH)
active_validator_indices = get_active_validator_indices(
state.validator_registry, current_epoch)
total_balance = Gwei(
sum(
get_effective_balance(state.validator_balances, i,
max_deposit_amount)
for i in active_validator_indices))
total_penalties = _compute_total_penalties(
state,
config,
current_epoch,
)
for validator_index, validator in enumerate(state.validator_registry):
validator_index = ValidatorIndex(validator_index)
is_halfway_to_withdrawable_epoch = (
current_epoch == validator.withdrawable_epoch -
latest_slashed_exit_length // 2)
if validator.slashed and is_halfway_to_withdrawable_epoch:
penalty = _compute_individual_penalty(
state=state,
config=config,
validator_index=validator_index,
total_penalties=total_penalties,
total_balance=total_balance,
)
state = state.update_validator_balance(
validator_index=validator_index,
balance=state.validator_balances[validator_index] - penalty,
)
return state
def _settle_penality_to_validator_and_whistleblower(
*, state: BeaconState, validator_index: ValidatorIndex,
latest_slashed_exit_length: int, whistleblower_reward_quotient: int,
max_deposit_amount: Gwei,
committee_config: CommitteeConfig) -> BeaconState:
"""
Apply penality/reward to validator and whistleblower and update the meta data
More intuitive pseudo-code:
current_epoch_penalization_index = (state.slot // SLOTS_PER_EPOCH) % LATEST_SLASHED_EXIT_LENGTH
state.latest_slashed_balances[current_epoch_penalization_index] += (
get_effective_balance(state, index)
)
whistleblower_index = get_beacon_proposer_index(state, state.slot)
whistleblower_reward = get_effective_balance(state, index) // WHISTLEBLOWER_REWARD_QUOTIENT
state.validator_balances[whistleblower_index] += whistleblower_reward
state.validator_balances[index] -= whistleblower_reward
validator.slashed = True
validator.withdrawable_epoch = get_current_epoch(state) + LATEST_SLASHED_EXIT_LENGTH
"""
slots_per_epoch = committee_config.SLOTS_PER_EPOCH
# Update `state.latest_slashed_balances`
current_epoch_penalization_index = state.current_epoch(
slots_per_epoch) % latest_slashed_exit_length
effective_balance = get_effective_balance(
state.validator_balances,
validator_index,
max_deposit_amount,
)
slashed_exit_balance = (
state.latest_slashed_balances[current_epoch_penalization_index] +
effective_balance)
latest_slashed_balances = update_tuple_item(
tuple_data=state.latest_slashed_balances,
index=current_epoch_penalization_index,
new_value=slashed_exit_balance,
)
state = state.copy(latest_slashed_balances=latest_slashed_balances, )
# Update whistleblower's balance
whistleblower_reward = (effective_balance // whistleblower_reward_quotient)
whistleblower_index = get_beacon_proposer_index(
state,
state.slot,
committee_config,
)
state = state.update_validator_balance(
whistleblower_index,
state.validator_balances[whistleblower_index] + whistleblower_reward,
)
# Update validator's balance and `slashed`, `withdrawable_epoch` field
validator = state.validator_registry[validator_index].copy(
slashed=True,
withdrawable_epoch=state.current_epoch(slots_per_epoch) +
latest_slashed_exit_length,
)
state = state.update_validator(
validator_index,
validator,
state.validator_balances[validator_index] - whistleblower_reward,
)
return state
def process_rewards_and_penalties(state: BeaconState,
config: BeaconConfig) -> BeaconState:
# Compute previous epoch active validator indices and the total balance they account for
# for later use.
previous_epoch_active_validator_indices = set(
get_active_validator_indices(
state.validator_registry,
state.previous_epoch(config.SLOTS_PER_EPOCH,
config.GENESIS_EPOCH)))
previous_total_balance: Gwei = get_total_balance(
state.validator_balances,
tuple(previous_epoch_active_validator_indices),
config.MAX_DEPOSIT_AMOUNT,
)
# Compute previous epoch attester indices and the total balance they account for
# for later use.
previous_epoch_attestations = get_previous_epoch_attestations(
state,
config.SLOTS_PER_EPOCH,
config.GENESIS_EPOCH,
)
previous_epoch_attester_indices = get_attester_indices_from_attesttion(
state=state,
attestations=previous_epoch_attestations,
committee_config=CommitteeConfig(config),
)
# Compute inclusion slot/distance of previous attestations for later use.
inclusion_infos = get_inclusion_infos(
state=state,
attestations=previous_epoch_attestations,
committee_config=CommitteeConfig(config),
)
# Compute effective balance of each previous epoch active validator for later use
effective_balances = {
index: get_effective_balance(
state.validator_balances,
index,
config.MAX_DEPOSIT_AMOUNT,
)
for index in previous_epoch_active_validator_indices
}
# Compute base reward of each previous epoch active validator for later use
_base_reward_quotient = (integer_squareroot(previous_total_balance) //
config.BASE_REWARD_QUOTIENT)
base_rewards = {
index: get_base_reward(
state=state,
index=index,
base_reward_quotient=_base_reward_quotient,
max_deposit_amount=config.MAX_DEPOSIT_AMOUNT,
)
for index in previous_epoch_active_validator_indices
}
# Initialize the reward (validator) received map
rewards_received = {
index: SignedGwei(0)
for index in previous_epoch_active_validator_indices
}
# 1. Process rewards and penalties for justification and finalization
rewards_received = pipe(
rewards_received,
_process_rewards_and_penalties_for_finality(
state,
config,
previous_epoch_active_validator_indices,
previous_total_balance,
previous_epoch_attestations,
previous_epoch_attester_indices,
inclusion_infos,
effective_balances,
base_rewards,
),
_process_rewards_and_penalties_for_attestation_inclusion(
state,
config,
previous_epoch_attester_indices,
inclusion_infos,
base_rewards,
),
_process_rewards_and_penalties_for_crosslinks(
state,
config,
previous_epoch_attestations,
effective_balances,
base_rewards,
))
# Apply the overall rewards/penalties
for index in previous_epoch_active_validator_indices:
state = state.update_validator_balance(
index,
# Prevent validator balance under flow
max(state.validator_balances[index] + rewards_received[index], 0),
)
return state
def process_rewards_and_penalties(state: BeaconState,
config: Eth2Config) -> BeaconState:
# Compute previous epoch active validator indices and the total balance they account for
# for later use.
previous_epoch_active_validator_indices = set(
get_active_validator_indices(
state.validator_registry,
state.previous_epoch(config.SLOTS_PER_EPOCH)))
previous_total_balance: Gwei = get_total_balance(
state.validator_balances,
tuple(previous_epoch_active_validator_indices),
config.MAX_DEPOSIT_AMOUNT,
)
# Compute previous epoch attester indices and the total balance they account for
# for later use.
previous_epoch_attestations = state.previous_epoch_attestations
previous_epoch_attester_indices = get_attester_indices_from_attestations(
state=state,
attestations=previous_epoch_attestations,
committee_config=CommitteeConfig(config),
)
# Compute inclusion slot/distance of previous attestations for later use.
inclusion_infos = get_inclusion_infos(
state=state,
attestations=previous_epoch_attestations,
committee_config=CommitteeConfig(config),
)
# Compute effective balance of each previous epoch active validator for later use
effective_balances = {
ValidatorIndex(index): get_effective_balance(
state.validator_balances,
ValidatorIndex(index),
config.MAX_DEPOSIT_AMOUNT,
)
for index in range(len(state.validator_registry))
}
# Compute base reward of each previous epoch active validator for later use
base_rewards = {
ValidatorIndex(index): get_base_reward(
state=state,
index=ValidatorIndex(index),
base_reward_quotient=config.BASE_REWARD_QUOTIENT,
previous_total_balance=previous_total_balance,
max_deposit_amount=config.MAX_DEPOSIT_AMOUNT,
)
for index in range(len(state.validator_registry))
}
# 1. Process rewards and penalties for justification and finalization
finality_rewards, finality_penalties = _process_rewards_and_penalties_for_finality(
state,
config,
previous_epoch_active_validator_indices,
previous_total_balance,
previous_epoch_attestations,
previous_epoch_attester_indices,
inclusion_infos,
effective_balances,
base_rewards,
)
# 2. Process rewards and penalties for crosslinks
crosslinks_rewards, crosslinks_penalties = _process_rewards_and_penalties_for_crosslinks(
state,
config,
effective_balances,
base_rewards,
)
# Apply the overall rewards/penalties
for index in range(len(state.validator_registry)):
state = state.update_validator_balance(
ValidatorIndex(index),
# Prevent validator balance under flow
max(
(state.validator_balances[index] + finality_rewards[index] +
crosslinks_rewards[index] - finality_penalties[index] -
crosslinks_penalties[index]),
0,
),
)
return state
