-
Notifications
You must be signed in to change notification settings - Fork 25
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Extend cluster type for testing to allow for adding funded hosts to t…
…he cluster
- Loading branch information
1 parent
ab0488d
commit c5f3edc
Showing
11 changed files
with
550 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| .vscode |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,146 @@ | ||
| // TODO: remove this file when we can import it from hostd | ||
| package node | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/binary" | ||
| "errors" | ||
| "fmt" | ||
| "sync" | ||
|
|
||
| "gitlab.com/NebulousLabs/fastrand" | ||
| "go.sia.tech/siad/crypto" | ||
| "go.sia.tech/siad/modules" | ||
| "go.sia.tech/siad/types" | ||
| ) | ||
|
|
||
| const solveAttempts = 1e4 | ||
|
|
||
| type ( | ||
| // Consensus defines a minimal interface needed by the miner to interact | ||
| // with the consensus set | ||
| Consensus interface { | ||
| AcceptBlock(types.Block) error | ||
| } | ||
|
|
||
| // A Miner is a CPU miner that can mine blocks, sending the reward to a | ||
| // specified address. | ||
| Miner struct { | ||
| consensus Consensus | ||
|
|
||
| mu sync.Mutex | ||
| height types.BlockHeight | ||
| target types.Target | ||
| currentBlockID types.BlockID | ||
| txnsets map[modules.TransactionSetID][]types.TransactionID | ||
| transactions []types.Transaction | ||
| } | ||
| ) | ||
|
|
||
| var errFailedToSolve = errors.New("failed to solve block") | ||
|
|
||
| // ProcessConsensusChange implements modules.ConsensusSetSubscriber. | ||
| func (m *Miner) ProcessConsensusChange(cc modules.ConsensusChange) { | ||
| m.mu.Lock() | ||
| defer m.mu.Unlock() | ||
| m.target = cc.ChildTarget | ||
| m.currentBlockID = cc.AppliedBlocks[len(cc.AppliedBlocks)-1].ID() | ||
| m.height = cc.BlockHeight | ||
| } | ||
|
|
||
| // ReceiveUpdatedUnconfirmedTransactions implements modules.TransactionPoolSubscriber | ||
| func (m *Miner) ReceiveUpdatedUnconfirmedTransactions(diff *modules.TransactionPoolDiff) { | ||
| m.mu.Lock() | ||
| defer m.mu.Unlock() | ||
|
|
||
| reverted := make(map[types.TransactionID]bool) | ||
| for _, setID := range diff.RevertedTransactions { | ||
| for _, txnID := range m.txnsets[setID] { | ||
| reverted[txnID] = true | ||
| } | ||
| } | ||
|
|
||
| filtered := m.transactions[:0] | ||
| for _, txn := range m.transactions { | ||
| if reverted[txn.ID()] { | ||
| continue | ||
| } | ||
| filtered = append(filtered, txn) | ||
| } | ||
|
|
||
| for _, txnset := range diff.AppliedTransactions { | ||
| m.txnsets[txnset.ID] = txnset.IDs | ||
| filtered = append(filtered, txnset.Transactions...) | ||
| } | ||
| m.transactions = filtered | ||
| } | ||
|
|
||
| // mineBlock attempts to mine a block and add it to the consensus set. | ||
| func (m *Miner) mineBlock(addr types.UnlockHash) error { | ||
| m.mu.Lock() | ||
| block := types.Block{ | ||
| ParentID: m.currentBlockID, | ||
| Timestamp: types.CurrentTimestamp(), | ||
| } | ||
|
|
||
| randBytes := fastrand.Bytes(types.SpecifierLen) | ||
| randTxn := types.Transaction{ | ||
| ArbitraryData: [][]byte{append(modules.PrefixNonSia[:], randBytes...)}, | ||
| } | ||
| block.Transactions = append([]types.Transaction{randTxn}, m.transactions...) | ||
| block.MinerPayouts = append(block.MinerPayouts, types.SiacoinOutput{ | ||
| Value: block.CalculateSubsidy(m.height + 1), | ||
| UnlockHash: addr, | ||
| }) | ||
| target := m.target | ||
| m.mu.Unlock() | ||
|
|
||
| merkleRoot := block.MerkleRoot() | ||
| header := make([]byte, 80) | ||
| copy(header, block.ParentID[:]) | ||
| binary.LittleEndian.PutUint64(header[40:48], uint64(block.Timestamp)) | ||
| copy(header[48:], merkleRoot[:]) | ||
|
|
||
| var nonce uint64 | ||
| var solved bool | ||
| for i := 0; i < solveAttempts; i++ { | ||
| id := crypto.HashBytes(header) | ||
| if bytes.Compare(target[:], id[:]) >= 0 { | ||
| block.Nonce = *(*types.BlockNonce)(header[32:40]) | ||
| solved = true | ||
| break | ||
| } | ||
| binary.LittleEndian.PutUint64(header[32:], nonce) | ||
| nonce += types.ASICHardforkFactor | ||
| } | ||
| if !solved { | ||
| return errFailedToSolve | ||
| } | ||
|
|
||
| if err := m.consensus.AcceptBlock(block); err != nil { | ||
| return fmt.Errorf("failed to get block accepted: %w", err) | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // Mine mines n blocks, sending the reward to addr | ||
| func (m *Miner) Mine(addr types.UnlockHash, n int) error { | ||
| var err error | ||
| for mined := 1; mined <= n; { | ||
| // return the error only if the miner failed to solve the block, | ||
| // ignore any consensus related errors | ||
| if err = m.mineBlock(addr); errors.Is(err, errFailedToSolve) { | ||
| return fmt.Errorf("failed to mine block %v: %w", mined, errFailedToSolve) | ||
| } | ||
| mined++ | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // NewMiner initializes a new CPU miner | ||
| func NewMiner(consensus Consensus) *Miner { | ||
| return &Miner{ | ||
| consensus: consensus, | ||
| txnsets: make(map[modules.TransactionSetID][]types.TransactionID), | ||
| } | ||
| } |
Oops, something went wrong.