forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-19 02:12:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 5f4ef4fda2 into fbeb4f20f9

Browse source
This commit is contained in:
Christopher Purta 2018-07-09 08:41:43 +00:00 committed by GitHub
commit 4738034b49
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
consensus/ethash/consensus.go
View file

@ -297,6 +297,8 @@ func (ethash *Ethash) CalcDifficulty(chain consensus.ChainReader, time uint64, p
func CalcDifficulty(config *params.ChainConfig, time uint64, parent *types.Header) *big.Int {
next := new(big.Int).Add(parent.Number, big1)
switch {
case isPrivateNetwork(config):
return parent.Difficulty
case config.IsByzantium(next):
return calcDifficultyByzantium(time, parent)
case config.IsHomestead(next):
@ -306,6 +308,15 @@ func CalcDifficulty(config *params.ChainConfig, time uint64, parent *types.Heade
}
}
// isPrivateNetwork is a helper function to determine if the current
// chain is a private network or not based on the configurations
// ChainID.
func isPrivateNetwork(config *params.ChainConfig) bool {
return config.ChainID != params.MainnetChainConfig.ChainID &&
config.ChainID != params.TestChainConfig.ChainID &&
config.ChainID != params.RinkebyChainConfig.ChainID
}
// Some weird constants to avoid constant memory allocs for them.
var (
expDiffPeriod = big.NewInt(100000)

18
consensus/ethash/consensus_test.go
View file

@ -84,3 +84,21 @@ func TestCalcDifficulty(t *testing.T) {
}
}
}
func TestIsPrivateNetwork(t *testing.T) {
tests := make(map[*big.Int]bool)
tests[big.NewInt(314)] = true
tests[params.MainnetChainConfig.ChainID] = false
tests[params.TestChainConfig.ChainID] = false
tests[params.RinkebyChainConfig.ChainID] = false
for chainID, expected := range tests {
config := &params.ChainConfig{ChainID: chainID}
isPrivate := isPrivateNetwork(config)
if isPrivate != expected {
t.Error("Private chain test failed. Expected", expected, "but determined", isPrivate)
}
}
}