61 lines
1.0 KiB
Go
61 lines
1.0 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"crypto/sha256"
|
||
|
|
"fmt"
|
||
|
|
"math/big"
|
||
|
|
)
|
||
|
|
|
||
|
|
const proofTargetBits = 16
|
||
|
|
|
||
|
|
type ProofOfWork struct {
|
||
|
|
block *Block
|
||
|
|
target *big.Int
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewProofOfWork(b *Block) *ProofOfWork {
|
||
|
|
target := big.NewInt(1)
|
||
|
|
target.Lsh(target, uint(256-proofTargetBits))
|
||
|
|
|
||
|
|
pow := &ProofOfWork{b, target}
|
||
|
|
|
||
|
|
return pow
|
||
|
|
}
|
||
|
|
|
||
|
|
func (pow *ProofOfWork) prepareData(nonce int) []byte {
|
||
|
|
data := bytes.Join(
|
||
|
|
[][]byte{
|
||
|
|
pow.block.PrevBlockHash,
|
||
|
|
pow.block.Data,
|
||
|
|
IntToHex(pow.block.Timestamp),
|
||
|
|
IntToHex(int64(proofTargetBits)),
|
||
|
|
IntToHex(int64(nonce)),
|
||
|
|
},
|
||
|
|
[]byte{},
|
||
|
|
)
|
||
|
|
return data
|
||
|
|
}
|
||
|
|
|
||
|
|
func (pow *ProofOfWork) Run() (int, []byte) {
|
||
|
|
var hashInt big.Int
|
||
|
|
var hash [32]byte
|
||
|
|
nonce := 0
|
||
|
|
|
||
|
|
fmt.Printf("Mining the block containing \"%s\"\n", pow.block.Data)
|
||
|
|
for nonce < maxNonce {
|
||
|
|
data := pow.prepareData(nonce)
|
||
|
|
hash = sha256.Sum256(data)
|
||
|
|
fmt.Printf("\r%x", hash)
|
||
|
|
hashInt.SetBytes(hash[:])
|
||
|
|
|
||
|
|
if hashInt.Cmp(pow.target) == -1 {
|
||
|
|
break
|
||
|
|
} else {
|
||
|
|
nonce++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fmt.Print("\n\n")
|
||
|
|
return nonce, hash[:]
|
||
|
|
}
|