blockchain.go

// Copyright Monax Industries Limited
// SPDX-License-Identifier: Apache-2.0

package bcm

import (
	"bytes"
	"fmt"
	"sync"
	"time"

	"github.com/hyperledger/burrow/encoding"
	"github.com/tendermint/tendermint/types"

	"github.com/hyperledger/burrow/genesis"
	"github.com/hyperledger/burrow/logging"
	dbm "github.com/tendermint/tm-db"
)

var stateKey = []byte("BlockchainState")

type BlockchainInfo interface {
	GenesisHash() []byte
	GenesisDoc() genesis.GenesisDoc
	ChainID() string
	LastBlockHeight() uint64
	LastBlockTime() time.Time
	LastCommitTime() time.Time
	LastCommitDuration() time.Duration
	LastBlockHash() []byte
	AppHashAfterLastBlock() []byte
	// BlockHash gets the hash at a height (or nil if no BlockStore mounted or block could not be found)
	BlockHash(height uint64) ([]byte, error)
	// GetBlockHeader returns the header at the specified height
	GetBlockHeader(blockNumber uint64) (*types.Header, error)
	// GetNumTxs returns the number of transactions included in a particular block
	GetNumTxs(blockNumber uint64) (int, error)
}

type Blockchain struct {
	sync.RWMutex
	persistedState PersistedState
	// Non-persisted state
	db                 dbm.DB
	blockStore         *BlockStore
	genesisDoc         genesis.GenesisDoc
	lastBlockHash      []byte
	lastCommitTime     time.Time
	lastCommitDuration time.Duration
}

var _ BlockchainInfo = &Blockchain{}

// LoadOrNewBlockchain returns true if state already exists
func LoadOrNewBlockchain(db dbm.DB, genesisDoc *genesis.GenesisDoc, logger *logging.Logger) (_ *Blockchain, exists bool, _ error) {
	logger = logger.WithScope("LoadOrNewBlockchain")
	logger.InfoMsg("Trying to load blockchain state from database",
		"database_key", stateKey)
	bc, err := loadBlockchain(db, genesisDoc)
	if err != nil {
		return nil, false, fmt.Errorf("error loading blockchain state from database: %v", err)
	}
	if bc != nil {
		dbHash := bc.GenesisHash()
		argHash := genesisDoc.Hash()
		if !bytes.Equal(dbHash, argHash) {
			return nil, false, fmt.Errorf("GenesisDoc passed to LoadOrNewBlockchain has hash: 0x%X, which does not "+
				"match the one found in database: 0x%X, database genesis:\n%v\npassed genesis:\n%v\n",
				argHash, dbHash, bc.genesisDoc.JSONString(), genesisDoc.JSONString())
		}
		if bc.LastBlockTime().Before(genesisDoc.GenesisTime) {
			return nil, false, fmt.Errorf("LastBlockTime %v from loaded Blockchain is before GenesisTime %v",
				bc.LastBlockTime(), genesisDoc.GenesisTime)
		}
		return bc, true, nil
	}

	logger.InfoMsg("No existing blockchain state found in database, making new blockchain")
	return NewBlockchain(db, genesisDoc), false, nil
}

// NewBlockchain returns a pointer to blockchain state initialised from genesis
func NewBlockchain(db dbm.DB, genesisDoc *genesis.GenesisDoc) *Blockchain {
	bc := &Blockchain{
		db: db,
		persistedState: PersistedState{
			AppHashAfterLastBlock: genesisDoc.Hash(),
			GenesisHash:           genesisDoc.Hash(),
			LastBlockTime:         genesisDoc.GenesisTime,
		},
		genesisDoc: *genesisDoc,
	}
	return bc
}

func GetSyncInfo(blockchain BlockchainInfo) *SyncInfo {
	return &SyncInfo{
		LatestBlockHeight:   blockchain.LastBlockHeight(),
		LatestBlockHash:     blockchain.LastBlockHash(),
		LatestAppHash:       blockchain.AppHashAfterLastBlock(),
		LatestBlockTime:     blockchain.LastBlockTime(),
		LatestBlockSeenTime: blockchain.LastCommitTime(),
		LatestBlockDuration: blockchain.LastCommitDuration(),
	}
}

func loadBlockchain(db dbm.DB, genesisDoc *genesis.GenesisDoc) (*Blockchain, error) {
	buf, err := db.Get(stateKey)
	if err != nil {
		return nil, err
	} else if len(buf) == 0 {
		return nil, nil
	}
	bc, err := decodeBlockchain(buf, genesisDoc)
	if err != nil {
		return nil, err
	}
	bc.db = db
	return bc, nil
}

func (bc *Blockchain) CommitBlock(blockTime time.Time, blockHash, appHash []byte) error {
	return bc.CommitBlockAtHeight(blockTime, blockHash, appHash, bc.persistedState.LastBlockHeight+1)
}

func (bc *Blockchain) CommitBlockAtHeight(blockTime time.Time, blockHash, appHash []byte, height uint64) error {
	bc.Lock()
	defer bc.Unlock()
	// Checkpoint on the _previous_ block. If we die, this is where we will resume since we know all intervening state
	// has been written successfully since we are committing the next block.
	// If we fall over we can resume a safe committed state and Tendermint will catch us up
	err := bc.save()
	if err != nil {
		return err
	}
	bc.lastCommitDuration = blockTime.Sub(bc.persistedState.LastBlockTime)
	bc.lastBlockHash = blockHash
	bc.persistedState.LastBlockHeight = height
	bc.persistedState.LastBlockTime = blockTime
	bc.persistedState.AppHashAfterLastBlock = appHash
	bc.lastCommitTime = time.Now().UTC()
	return nil
}

func (bc *Blockchain) CommitWithAppHash(appHash []byte) error {
	bc.persistedState.AppHashAfterLastBlock = appHash
	bc.Lock()
	defer bc.Unlock()

	return bc.save()
}

func (bc *Blockchain) save() error {
	if bc.db != nil {
		encodedState, err := bc.Encode()
		if err != nil {
			return err
		}
		bc.db.SetSync(stateKey, encodedState)
	}
	return nil
}

func (bc *Blockchain) Encode() ([]byte, error) {
	return encoding.Encode(&bc.persistedState)
}

func decodeBlockchain(encodedState []byte, genesisDoc *genesis.GenesisDoc) (*Blockchain, error) {
	bc := NewBlockchain(nil, genesisDoc)
	err := encoding.Decode(encodedState, &bc.persistedState)
	if err != nil {
		return nil, err
	}
	return bc, nil
}

func (bc *Blockchain) GenesisHash() []byte {
	return bc.persistedState.GenesisHash
}

func (bc *Blockchain) GenesisDoc() genesis.GenesisDoc {
	return bc.genesisDoc
}

func (bc *Blockchain) ChainID() string {
	return bc.genesisDoc.GetChainID()
}

func (bc *Blockchain) LastBlockHeight() uint64 {
	if bc == nil {
		return 0
	}
	bc.RLock()
	defer bc.RUnlock()
	return bc.persistedState.LastBlockHeight
}

func (bc *Blockchain) LastBlockTime() time.Time {
	bc.RLock()
	defer bc.RUnlock()
	return bc.persistedState.LastBlockTime
}

func (bc *Blockchain) LastCommitTime() time.Time {
	bc.RLock()
	defer bc.RUnlock()
	return bc.lastCommitTime
}

func (bc *Blockchain) LastCommitDuration() time.Duration {
	bc.RLock()
	defer bc.RUnlock()
	return bc.lastCommitDuration
}

func (bc *Blockchain) LastBlockHash() []byte {
	bc.RLock()
	defer bc.RUnlock()
	return bc.lastBlockHash
}

func (bc *Blockchain) AppHashAfterLastBlock() []byte {
	bc.RLock()
	defer bc.RUnlock()
	return bc.persistedState.AppHashAfterLastBlock
}

// Tendermint block access

func (bc *Blockchain) SetBlockStore(bs *BlockStore) {
	bc.blockStore = bs
}

func (bc *Blockchain) BlockHash(height uint64) ([]byte, error) {
	header, err := bc.GetBlockHeader(height)
	if err != nil {
		return nil, err
	}
	return header.Hash(), nil
}

func (bc *Blockchain) getBlockMeta(height uint64) (*types.BlockMeta, error) {
	const errHeader = "getBlockMeta():"
	if bc == nil {
		return nil, fmt.Errorf("%s could not get block hash because Blockchain has not been given access to "+
			"tendermint BlockStore", errHeader)
	}
	return bc.blockStore.BlockMeta(int64(height))
}

// GetBlockHeader returns the block header at any given height
func (bc *Blockchain) GetBlockHeader(height uint64) (*types.Header, error) {
	const errHeader = "GetBlockHeader():"
	blockMeta, err := bc.getBlockMeta(height)
	if err != nil {
		return nil, fmt.Errorf("%s could not get BlockMeta: %v", errHeader, err)
	}
	return &blockMeta.Header, nil
}

// GetNumTxs returns the number of transactions included in a block
func (bc *Blockchain) GetNumTxs(height uint64) (int, error) {
	const errHeader = "GetNumTxs():"
	blockMeta, err := bc.getBlockMeta(height)
	if err != nil {
		return 0, fmt.Errorf("%s could not get BlockMeta: %v", errHeader, err)
	}
	return blockMeta.NumTxs, nil
}