80 lines
1.5 KiB
Go
80 lines
1.5 KiB
Go
|
package types
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/wealdtech/go-merkletree"
|
||
|
|
"github.com/wealdtech/go-merkletree/keccak256"
|
||
|
|
|
||
|
|
"github.com/Secured-Finance/dione/wallet"
|
||
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Block struct {
|
||
|
|
Header *BlockHeader
|
||
|
|
Data []*Transaction
|
||
|
|
}
|
||
|
|
|
||
|
|
type BlockHeader struct {
|
||
|
|
Timestamp int64
|
||
|
|
SeqNum uint64
|
||
|
|
Hash []byte
|
||
|
|
LastHash []byte
|
||
|
|
Proposer peer.ID
|
||
|
|
Signature []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func GenesisBlock() *Block {
|
||
|
|
return &Block{
|
||
|
|
Header: &BlockHeader{
|
||
|
|
Timestamp: 1620845070,
|
||
|
|
SeqNum: 0,
|
||
|
|
Hash: []byte("DIMICANDUM"),
|
||
|
|
},
|
||
|
|
Data: []*Transaction{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateBlock(lastBlockHeader *BlockHeader, txs []*Transaction, wallet *wallet.LocalWallet) (*Block, error) {
|
||
|
|
timestamp := time.Now().Unix()
|
||
|
|
proposer, err := wallet.GetDefault()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// extract hashes from transactions
|
||
|
|
var txHashes [][]byte
|
||
|
|
for _, tx := range txs {
|
||
|
|
txHashes = append(txHashes, tx.Hash)
|
||
|
|
}
|
||
|
|
txHashes = append(txHashes, lastBlockHeader.Hash)
|
||
|
|
|
||
|
|
tree, err := merkletree.NewUsing(txHashes, keccak256.New(), false)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// fetch merkle tree root hash (block hash)
|
||
|
|
blockHash := tree.Root()
|
||
|
|
|
||
|
|
// sign this block hash
|
||
|
|
s, err := wallet.Sign(proposer, blockHash)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
block := &Block{
|
||
|
|
Header: &BlockHeader{
|
||
|
|
Timestamp: timestamp,
|
||
|
|
SeqNum: lastBlockHeader.SeqNum + 1,
|
||
|
|
Proposer: proposer,
|
||
|
|
Signature: s.Data,
|
||
|
|
Hash: blockHash,
|
||
|
|
LastHash: lastBlockHeader.Hash,
|
||
|
|
},
|
||
|
|
Data: txs,
|
||
|
|
}
|
||
|
|
|
||
|
|
return block, nil
|
||
|
|
}
|