Move RemoteAgent to miner pkg

1 files changed, 81 insertions, 0 deletions

diff --git a/miner/remote_agent.go b/miner/remote_agent.go
new file mode 100644
index 000000000..8ea164f34
--- /dev/null
+++ b/miner/remote_agent.go
@@ -0,0 +1,81 @@
+package miner
+
+import (
+	"github.com/ethereum/ethash"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+)
+
+type RemoteAgent struct {
+	work        *types.Block
+	currentWork *types.Block
+
+	quit     chan struct{}
+	workCh   chan *types.Block
+	returnCh chan<- Work
+}
+
+func NewRemoteAgent() *RemoteAgent {
+	agent := &RemoteAgent{}
+	go agent.run()
+
+	return agent
+}
+
+func (a *RemoteAgent) Work() chan<- *types.Block {
+	return a.workCh
+}
+
+func (a *RemoteAgent) SetWorkCh(returnCh chan<- Work) {
+	a.returnCh = returnCh
+}
+
+func (a *RemoteAgent) Start() {
+	a.quit = make(chan struct{})
+	a.workCh = make(chan *types.Block, 1)
+}
+
+func (a *RemoteAgent) Stop() {
+	close(a.quit)
+	close(a.workCh)
+}
+
+func (a *RemoteAgent) GetHashRate() int64 { return 0 }
+
+func (a *RemoteAgent) run() {
+out:
+	for {
+		select {
+		case <-a.quit:
+			break out
+		case work := <-a.workCh:
+			a.work = work
+		}
+	}
+}
+
+func (a *RemoteAgent) GetWork() [3]string {
+	var res [3]string
+
+	// XXX Wait here until work != nil ?
+	if a.work != nil {
+		res[0] = a.work.HashNoNonce().Hex()
+		seedHash, _ := ethash.GetSeedHash(a.currentWork.NumberU64())
+		res[1] = common.Bytes2Hex(seedHash)
+		res[2] = common.Bytes2Hex(a.work.Difficulty().Bytes())
+	}
+
+	return res
+}
+
+func (a *RemoteAgent) SubmitWork(nonce uint64, mixDigest, seedHash common.Hash) bool {
+	// Return true or false, but does not indicate if the PoW was correct
+
+	// Make sure the external miner was working on the right hash
+	if a.currentWork != nil && a.work != nil && a.currentWork.Hash() == a.work.Hash() {
+		a.returnCh <- Work{a.currentWork.Number().Uint64(), nonce, mixDigest.Bytes(), seedHash.Bytes()}
+		return true
+	}
+
+	return false
+}