123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298 |
- // Copyright 2018 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 (
- "encoding/json"
- "io/ioutil"
- "math/big"
- "net/http"
- "net/http/httptest"
- "strconv"
- "testing"
- "time"
- "github.com/ethereum/go-ethereum/common"
- "github.com/ethereum/go-ethereum/core/types"
- "github.com/ethereum/go-ethereum/internal/testlog"
- "github.com/ethereum/go-ethereum/log"
- )
- // Tests whether remote HTTP servers are correctly notified of new work.
- func TestRemoteNotify(t *testing.T) {
- // Start a simple web server to capture notifications.
- sink := make(chan [3]string)
- server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
- blob, err := ioutil.ReadAll(req.Body)
- if err != nil {
- t.Errorf("failed to read miner notification: %v", err)
- }
- var work [3]string
- if err := json.Unmarshal(blob, &work); err != nil {
- t.Errorf("failed to unmarshal miner notification: %v", err)
- }
- sink <- work
- }))
- defer server.Close()
- // Create the custom ethash engine.
- ethash := NewTester([]string{server.URL}, false)
- defer ethash.Close()
- // Stream a work task and ensure the notification bubbles out.
- header := &types.Header{Number: big.NewInt(1), Difficulty: big.NewInt(100)}
- block := types.NewBlockWithHeader(header)
- ethash.Seal(nil, block, nil, nil)
- select {
- case work := <-sink:
- if want := ethash.SealHash(header).Hex(); work[0] != want {
- t.Errorf("work packet hash mismatch: have %s, want %s", work[0], want)
- }
- if want := common.BytesToHash(SeedHash(header.Number.Uint64())).Hex(); work[1] != want {
- t.Errorf("work packet seed mismatch: have %s, want %s", work[1], want)
- }
- target := new(big.Int).Div(new(big.Int).Lsh(big.NewInt(1), 256), header.Difficulty)
- if want := common.BytesToHash(target.Bytes()).Hex(); work[2] != want {
- t.Errorf("work packet target mismatch: have %s, want %s", work[2], want)
- }
- case <-time.After(3 * time.Second):
- t.Fatalf("notification timed out")
- }
- }
- // Tests whether remote HTTP servers are correctly notified of new work. (Full pending block body / --miner.notify.full)
- func TestRemoteNotifyFull(t *testing.T) {
- // Start a simple web server to capture notifications.
- sink := make(chan map[string]interface{})
- server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
- blob, err := ioutil.ReadAll(req.Body)
- if err != nil {
- t.Errorf("failed to read miner notification: %v", err)
- }
- var work map[string]interface{}
- if err := json.Unmarshal(blob, &work); err != nil {
- t.Errorf("failed to unmarshal miner notification: %v", err)
- }
- sink <- work
- }))
- defer server.Close()
- // Create the custom ethash engine.
- config := Config{
- PowMode: ModeTest,
- NotifyFull: true,
- Log: testlog.Logger(t, log.LvlWarn),
- }
- ethash := New(config, []string{server.URL}, false)
- defer ethash.Close()
- // Stream a work task and ensure the notification bubbles out.
- header := &types.Header{Number: big.NewInt(1), Difficulty: big.NewInt(100)}
- block := types.NewBlockWithHeader(header)
- ethash.Seal(nil, block, nil, nil)
- select {
- case work := <-sink:
- if want := "0x" + strconv.FormatUint(header.Number.Uint64(), 16); work["number"] != want {
- t.Errorf("pending block number mismatch: have %v, want %v", work["number"], want)
- }
- if want := "0x" + header.Difficulty.Text(16); work["difficulty"] != want {
- t.Errorf("pending block difficulty mismatch: have %s, want %s", work["difficulty"], want)
- }
- case <-time.After(3 * time.Second):
- t.Fatalf("notification timed out")
- }
- }
- // Tests that pushing work packages fast to the miner doesn't cause any data race
- // issues in the notifications.
- func TestRemoteMultiNotify(t *testing.T) {
- // Start a simple web server to capture notifications.
- sink := make(chan [3]string, 64)
- server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
- blob, err := ioutil.ReadAll(req.Body)
- if err != nil {
- t.Errorf("failed to read miner notification: %v", err)
- }
- var work [3]string
- if err := json.Unmarshal(blob, &work); err != nil {
- t.Errorf("failed to unmarshal miner notification: %v", err)
- }
- sink <- work
- }))
- defer server.Close()
- // Create the custom ethash engine.
- ethash := NewTester([]string{server.URL}, false)
- ethash.config.Log = testlog.Logger(t, log.LvlWarn)
- defer ethash.Close()
- // Provide a results reader.
- // Otherwise the unread results will be logged asynchronously
- // and this can happen after the test is finished, causing a panic.
- results := make(chan *types.Block, cap(sink))
- // Stream a lot of work task and ensure all the notifications bubble out.
- for i := 0; i < cap(sink); i++ {
- header := &types.Header{Number: big.NewInt(int64(i)), Difficulty: big.NewInt(100)}
- block := types.NewBlockWithHeader(header)
- ethash.Seal(nil, block, results, nil)
- }
- for i := 0; i < cap(sink); i++ {
- select {
- case <-sink:
- <-results
- case <-time.After(10 * time.Second):
- t.Fatalf("notification %d timed out", i)
- }
- }
- }
- // Tests that pushing work packages fast to the miner doesn't cause any data race
- // issues in the notifications. Full pending block body / --miner.notify.full)
- func TestRemoteMultiNotifyFull(t *testing.T) {
- // Start a simple web server to capture notifications.
- sink := make(chan map[string]interface{}, 64)
- server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
- blob, err := ioutil.ReadAll(req.Body)
- if err != nil {
- t.Errorf("failed to read miner notification: %v", err)
- }
- var work map[string]interface{}
- if err := json.Unmarshal(blob, &work); err != nil {
- t.Errorf("failed to unmarshal miner notification: %v", err)
- }
- sink <- work
- }))
- defer server.Close()
- // Create the custom ethash engine.
- config := Config{
- PowMode: ModeTest,
- NotifyFull: true,
- Log: testlog.Logger(t, log.LvlWarn),
- }
- ethash := New(config, []string{server.URL}, false)
- defer ethash.Close()
- // Provide a results reader.
- // Otherwise the unread results will be logged asynchronously
- // and this can happen after the test is finished, causing a panic.
- results := make(chan *types.Block, cap(sink))
- // Stream a lot of work task and ensure all the notifications bubble out.
- for i := 0; i < cap(sink); i++ {
- header := &types.Header{Number: big.NewInt(int64(i)), Difficulty: big.NewInt(100)}
- block := types.NewBlockWithHeader(header)
- ethash.Seal(nil, block, results, nil)
- }
- for i := 0; i < cap(sink); i++ {
- select {
- case <-sink:
- <-results
- case <-time.After(10 * time.Second):
- t.Fatalf("notification %d timed out", i)
- }
- }
- }
- // Tests whether stale solutions are correctly processed.
- func TestStaleSubmission(t *testing.T) {
- ethash := NewTester(nil, true)
- defer ethash.Close()
- api := &API{ethash}
- fakeNonce, fakeDigest := types.BlockNonce{0x01, 0x02, 0x03}, common.HexToHash("deadbeef")
- testcases := []struct {
- headers []*types.Header
- submitIndex int
- submitRes bool
- }{
- // Case1: submit solution for the latest mining package
- {
- []*types.Header{
- {ParentHash: common.BytesToHash([]byte{0xa}), Number: big.NewInt(1), Difficulty: big.NewInt(100000000)},
- },
- 0,
- true,
- },
- // Case2: submit solution for the previous package but have same parent.
- {
- []*types.Header{
- {ParentHash: common.BytesToHash([]byte{0xb}), Number: big.NewInt(2), Difficulty: big.NewInt(100000000)},
- {ParentHash: common.BytesToHash([]byte{0xb}), Number: big.NewInt(2), Difficulty: big.NewInt(100000001)},
- },
- 0,
- true,
- },
- // Case3: submit stale but acceptable solution
- {
- []*types.Header{
- {ParentHash: common.BytesToHash([]byte{0xc}), Number: big.NewInt(3), Difficulty: big.NewInt(100000000)},
- {ParentHash: common.BytesToHash([]byte{0xd}), Number: big.NewInt(9), Difficulty: big.NewInt(100000000)},
- },
- 0,
- true,
- },
- // Case4: submit very old solution
- {
- []*types.Header{
- {ParentHash: common.BytesToHash([]byte{0xe}), Number: big.NewInt(10), Difficulty: big.NewInt(100000000)},
- {ParentHash: common.BytesToHash([]byte{0xf}), Number: big.NewInt(17), Difficulty: big.NewInt(100000000)},
- },
- 0,
- false,
- },
- }
- results := make(chan *types.Block, 16)
- for id, c := range testcases {
- for _, h := range c.headers {
- ethash.Seal(nil, types.NewBlockWithHeader(h), results, nil)
- }
- if res := api.SubmitWork(fakeNonce, ethash.SealHash(c.headers[c.submitIndex]), fakeDigest); res != c.submitRes {
- t.Errorf("case %d submit result mismatch, want %t, get %t", id+1, c.submitRes, res)
- }
- if !c.submitRes {
- continue
- }
- select {
- case res := <-results:
- if res.Header().Nonce != fakeNonce {
- t.Errorf("case %d block nonce mismatch, want %x, get %x", id+1, fakeNonce, res.Header().Nonce)
- }
- if res.Header().MixDigest != fakeDigest {
- t.Errorf("case %d block digest mismatch, want %x, get %x", id+1, fakeDigest, res.Header().MixDigest)
- }
- if res.Header().Difficulty.Uint64() != c.headers[c.submitIndex].Difficulty.Uint64() {
- t.Errorf("case %d block difficulty mismatch, want %d, get %d", id+1, c.headers[c.submitIndex].Difficulty, res.Header().Difficulty)
- }
- if res.Header().Number.Uint64() != c.headers[c.submitIndex].Number.Uint64() {
- t.Errorf("case %d block number mismatch, want %d, get %d", id+1, c.headers[c.submitIndex].Number.Uint64(), res.Header().Number.Uint64())
- }
- if res.Header().ParentHash != c.headers[c.submitIndex].ParentHash {
- t.Errorf("case %d block parent hash mismatch, want %s, get %s", id+1, c.headers[c.submitIndex].ParentHash.Hex(), res.Header().ParentHash.Hex())
- }
- case <-time.NewTimer(time.Second).C:
- t.Errorf("case %d fetch ethash result timeout", id+1)
- }
- }
- }
|