924065e19d
* consensus/ethash: add maxEpoch constant * consensus/ethash: improve cache/dataset handling There are two fixes in this commit: Unmap the memory through a finalizer like the libethash wrapper did. The release logic was incorrect and freed the memory while it was being used, leading to crashes like in #14495 or #14943. Track caches and datasets using simplelru instead of reinventing LRU logic. This should make it easier to see whether it's correct. * consensus/ethash: restore 'future item' logic in lru * consensus/ethash: use mmap even in test mode This makes it possible to shorten the time taken for TestCacheFileEvict. * consensus/ethash: shuffle func calc*Size comments around * consensus/ethash: ensure future cache/dataset is in the lru cache * consensus/ethash: add issue link to the new test * consensus/ethash: fix vet * consensus/ethash: fix test * consensus: tiny issue + nitpick fixes
80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
// Copyright 2017 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package ethash
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"math/big"
|
|
"math/rand"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
)
|
|
|
|
// Tests that ethash works correctly in test mode.
|
|
func TestTestMode(t *testing.T) {
|
|
head := &types.Header{Number: big.NewInt(1), Difficulty: big.NewInt(100)}
|
|
|
|
ethash := NewTester()
|
|
block, err := ethash.Seal(nil, types.NewBlockWithHeader(head), nil)
|
|
if err != nil {
|
|
t.Fatalf("failed to seal block: %v", err)
|
|
}
|
|
head.Nonce = types.EncodeNonce(block.Nonce())
|
|
head.MixDigest = block.MixDigest()
|
|
if err := ethash.VerifySeal(nil, head); err != nil {
|
|
t.Fatalf("unexpected verification error: %v", err)
|
|
}
|
|
}
|
|
|
|
// This test checks that cache lru logic doesn't crash under load.
|
|
// It reproduces https://github.com/ethereum/go-ethereum/issues/14943
|
|
func TestCacheFileEvict(t *testing.T) {
|
|
tmpdir, err := ioutil.TempDir("", "ethash-test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(tmpdir)
|
|
e := New(Config{CachesInMem: 3, CachesOnDisk: 10, CacheDir: tmpdir, PowMode: ModeTest})
|
|
|
|
workers := 8
|
|
epochs := 100
|
|
var wg sync.WaitGroup
|
|
wg.Add(workers)
|
|
for i := 0; i < workers; i++ {
|
|
go verifyTest(&wg, e, i, epochs)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func verifyTest(wg *sync.WaitGroup, e *Ethash, workerIndex, epochs int) {
|
|
defer wg.Done()
|
|
|
|
const wiggle = 4 * epochLength
|
|
r := rand.New(rand.NewSource(int64(workerIndex)))
|
|
for epoch := 0; epoch < epochs; epoch++ {
|
|
block := int64(epoch)*epochLength - wiggle/2 + r.Int63n(wiggle)
|
|
if block < 0 {
|
|
block = 0
|
|
}
|
|
head := &types.Header{Number: big.NewInt(block), Difficulty: big.NewInt(100)}
|
|
e.VerifySeal(nil, head)
|
|
}
|
|
}
|