80 lines
1.4 KiB
Go
80 lines
1.4 KiB
Go
package chain
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"fmt"
|
|
"golang.org/x/crypto/sha3"
|
|
"log"
|
|
"math/rand"
|
|
)
|
|
|
|
type block struct {
|
|
Transactions []transaction
|
|
HashPrevious []byte
|
|
Nonce []byte
|
|
}
|
|
|
|
func NewBlock(hashPrevious []byte) block {
|
|
return block{
|
|
Transactions: nil,
|
|
HashPrevious: hashPrevious,
|
|
}
|
|
}
|
|
|
|
func (b *block) AddTransaction(t transaction) {
|
|
b.Transactions = append(b.Transactions, t)
|
|
}
|
|
|
|
func (b block) GenerateHash(difficulty int) []byte {
|
|
found := false
|
|
var compBytes []byte
|
|
for i := 0; i < difficulty; i++ {
|
|
compBytes = append(compBytes, []byte("0x00")...)
|
|
}
|
|
for !found {
|
|
random := make([]byte, 64)
|
|
rand.Read(random)
|
|
b.Nonce = random
|
|
var buf bytes.Buffer
|
|
err := gob.NewEncoder(&buf).Encode(b.Nonce)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
hasher := sha3.New512()
|
|
hasher.Write(b.ToBytes())
|
|
hash := hasher.Sum(nil)
|
|
fmt.Printf("\033[2K\r%x\n", hash)
|
|
if checkAllZero(XorSlice(hash[:difficulty], compBytes)) {
|
|
return hash
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b block) ToBytes() []byte {
|
|
var bytes []byte
|
|
for i := 0; i < len(b.Transactions); i++ {
|
|
bytes = append(bytes, b.Transactions[i].ToBytes()...)
|
|
}
|
|
bytes = append(bytes, b.Nonce...)
|
|
return bytes
|
|
}
|
|
|
|
func XorSlice(a []byte, b []byte) []byte {
|
|
for i, _ := range a {
|
|
a[i] = a[i] ^ b[i]
|
|
}
|
|
return a
|
|
}
|
|
|
|
func checkAllZero(a []byte) bool {
|
|
fmt.Printf("Checking if all zero: %x\n", a)
|
|
for i, _ := range a {
|
|
if a[i] != 0x00 {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|