mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-20 21:54:30 +00:00
88 lines
No EOL
3.3 KiB
Go
88 lines
No EOL
3.3 KiB
Go
package contracts
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/state"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
)
|
|
|
|
var (
|
|
slotValidatorMapping = map[string]uint64{
|
|
"withdrawsState": 0,
|
|
"validatorsState": 1,
|
|
"voters": 2,
|
|
"candidates": 3,
|
|
"candidateCount": 4,
|
|
"minCandidateCap": 5,
|
|
"minVoterCap": 6,
|
|
"maxValidatorNumber": 7,
|
|
"candidateWithdrawDelay": 8,
|
|
"voterWithdrawDelay": 9,
|
|
}
|
|
)
|
|
|
|
func GetCandidates(statedb *state.StateDB) []common.Address {
|
|
slot := slotValidatorMapping["candidates"]
|
|
slotHash := common.BigToHash(new(big.Int).SetUint64(slot))
|
|
arrLength := statedb.GetState(common.HexToAddress(common.MasternodeVotingSMC), slotHash)
|
|
keys := []common.Hash{}
|
|
for i := uint64(0); i < arrLength.Big().Uint64(); i++ {
|
|
key := getLocDynamicArrAtElement(slotHash, i, 1)
|
|
keys = append(keys, key)
|
|
}
|
|
rets := []common.Address{}
|
|
for _, key := range keys {
|
|
ret := statedb.GetState(common.HexToAddress(common.MasternodeVotingSMC), key)
|
|
rets = append(rets, common.HexToAddress(ret.Hex()))
|
|
}
|
|
return rets
|
|
}
|
|
|
|
func GetCandidateOwner(statedb *state.StateDB, candidate common.Address) common.Address {
|
|
slot := slotValidatorMapping["validatorsState"]
|
|
// validatorsState[_candidate].owner;
|
|
locValidatorsState := getLocMappingAtKey(candidate.Hash(), slot)
|
|
locCandidateOwner := locValidatorsState.Add(locValidatorsState, new(big.Int).SetUint64(uint64(0)))
|
|
ret := statedb.GetState(common.HexToAddress(common.MasternodeVotingSMC), common.BigToHash(locCandidateOwner))
|
|
return common.HexToAddress(ret.Hex())
|
|
}
|
|
|
|
func GetCandidateCap(statedb *state.StateDB, parsed abi.ABI, candidate common.Address) string {
|
|
slot := slotValidatorMapping["validatorsState"]
|
|
// validatorsState[_candidate].cap;
|
|
locValidatorsState := getLocMappingAtKey(candidate.Hash(), slot)
|
|
locCandidateCap := locValidatorsState.Add(locValidatorsState, new(big.Int).SetUint64(uint64(1)))
|
|
ret := statedb.GetState(common.HexToAddress(common.MasternodeVotingSMC), common.BigToHash(locCandidateCap))
|
|
return ret.Hex()
|
|
}
|
|
|
|
func GetVoters(statedb *state.StateDB, candidate common.Address) []common.Address {
|
|
//mapping(address => address[]) voters;
|
|
slot := slotValidatorMapping["voters"]
|
|
locVoters := getLocMappingAtKey(candidate.Hash(), slot)
|
|
arrLength := statedb.GetState(common.HexToAddress(common.MasternodeVotingSMC), common.BigToHash(locVoters))
|
|
keys := []common.Hash{}
|
|
for i := uint64(0); i < arrLength.Big().Uint64(); i++ {
|
|
key := getLocDynamicArrAtElement(common.BigToHash(locVoters), i, 1)
|
|
keys = append(keys, key)
|
|
}
|
|
rets := []common.Address{}
|
|
for _, key := range keys {
|
|
ret := statedb.GetState(common.HexToAddress(common.MasternodeVotingSMC), key)
|
|
rets = append(rets, common.HexToAddress(ret.Hex()))
|
|
}
|
|
|
|
return rets
|
|
}
|
|
|
|
func GetVoterCap(state *state.StateDB, candidate, voter common.Address) *big.Int {
|
|
slot := slotValidatorMapping["validatorsState"]
|
|
locValidatorsState := getLocMappingAtKey(candidate.Hash(), slot)
|
|
locCandidateVoters := locValidatorsState.Add(locValidatorsState, new(big.Int).SetUint64(uint64(2)))
|
|
retByte := crypto.Keccak256(voter.Hash().Bytes(), common.BigToHash(locCandidateVoters).Bytes())
|
|
ret := state.GetState(common.HexToAddress(common.MasternodeVotingSMC), common.BytesToHash(retByte))
|
|
return ret.Big()
|
|
} |