cli: show timestamp of the block

.gitignore

@@ -1,2 +1,5 @@
 *.db
 *.dat
+cscope.*
+.*.swp
+blockchain_go

base58.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"bytes"
@@ -40,8 +40,10 @@ func Base58Decode(input []byte) []byte {
 	zeroBytes := 0
 
 	for _, b := range input {
-		if b == 0x00 {
+		if b == b58Alphabet[0] {
 			zeroBytes++
+		} else {
+			break
 		}
 	}
 

base58_test.go

@@ -0,0 +1,27 @@
+package bc
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBase58(t *testing.T) {
+	for i := 0; i < 100; i++ {
+		_, public := newKeyPair()
+		pubKeyHash := HashPubKey(public)
+
+		versionedPayload := append([]byte{version}, pubKeyHash...)
+		checksum := checksum(versionedPayload)
+
+		fullPayload := append(versionedPayload, checksum...)
+		address := Base58Encode(fullPayload)
+
+		assert.Equal(
+			t,
+			ValidateAddress(string(address[:])),
+			true,
+			"Address: %s is invalid", address,
+		)
+	}
+}

block.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"bytes"

blockchain.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"bytes"
@@ -82,7 +82,7 @@ func NewBlockchain(nodeID string) *Blockchain {
 		log.Panic(err)
 	}
 
-	err = db.Update(func(tx *bolt.Tx) error {
+	err = db.View(func(tx *bolt.Tx) error {
 		b := tx.Bucket([]byte(blocksBucket))
 		tip = b.Get([]byte("l"))
 

blockchain_iterator.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"log"

cli.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"flag"
@@ -21,6 +21,14 @@ func (cli *CLI) printUsage() {
 	fmt.Println("  reindexutxo - Rebuilds the UTXO set")
 	fmt.Println("  send -from FROM -to TO -amount AMOUNT -mine - Send AMOUNT of coins from FROM address to TO. Mine on the same node, when -mine is set.")
 	fmt.Println("  startnode -miner ADDRESS - Start a node with ID specified in NODE_ID env. var. -miner enables mining")
+	fmt.Println()
+	fmt.Println("Exploring cmds:")
+	fmt.Println("  generatePrivKey - generate KeyPair for exploring")
+	fmt.Println("  getPubKey -privKey PRIKEY - generate PubKey from privateKey")
+	fmt.Println("  getAddress -pubKey PUBKEY - convert pubKey to address")
+	fmt.Println("  getPubKeyHash -address Address - get pubKeyHash of an address")
+	fmt.Println("  validateAddress -addr Address - validate an address")
+	fmt.Println("  getBlock -hash BlockHash - get a block with BlockHash")
 }
 
 func (cli *CLI) validateArgs() {
@@ -48,6 +56,12 @@ func (cli *CLI) Run() {
 	reindexUTXOCmd := flag.NewFlagSet("reindexutxo", flag.ExitOnError)
 	sendCmd := flag.NewFlagSet("send", flag.ExitOnError)
 	startNodeCmd := flag.NewFlagSet("startnode", flag.ExitOnError)
+	generatePrivKeyCmd := flag.NewFlagSet("generatePrivKey", flag.ExitOnError)
+	getPubKeyCmd := flag.NewFlagSet("getPubKey", flag.ExitOnError)
+	getAddressCmd := flag.NewFlagSet("getAddress", flag.ExitOnError)
+	getPubKeyHashCmd := flag.NewFlagSet("getPubKeyHash", flag.ExitOnError)
+	validateAddrCmd := flag.NewFlagSet("validateAddress", flag.ExitOnError)
+	getBlockCmd := flag.NewFlagSet("getBlock", flag.ExitOnError)
 
 	getBalanceAddress := getBalanceCmd.String("address", "", "The address to get balance for")
 	createBlockchainAddress := createBlockchainCmd.String("address", "", "The address to send genesis block reward to")
@@ -56,6 +70,11 @@ func (cli *CLI) Run() {
 	sendAmount := sendCmd.Int("amount", 0, "Amount to send")
 	sendMine := sendCmd.Bool("mine", false, "Mine immediately on the same node")
 	startNodeMiner := startNodeCmd.String("miner", "", "Enable mining mode and send reward to ADDRESS")
+	privateKey := getPubKeyCmd.String("privKey", "", "generate PubKey based on this")
+	pubKey := getAddressCmd.String("pubKey", "", "the key where address generated")
+	pubKeyAddress := getPubKeyHashCmd.String("address", "", "the pub address")
+	address := validateAddrCmd.String("addr", "", "the public address")
+	blockHash := getBlockCmd.String("hash", "", "the block hash")
 
 	switch os.Args[1] {
 	case "getbalance":
@@ -98,6 +117,36 @@ func (cli *CLI) Run() {
 		if err != nil {
 			log.Panic(err)
 		}
+	case "validateAddress":
+		err := validateAddrCmd.Parse(os.Args[2:])
+		if err != nil {
+			log.Panic(err)
+		}
+	case "generatePrivKey":
+		err := generatePrivKeyCmd.Parse(os.Args[2:])
+		if err != nil {
+			log.Panic(err)
+		}
+	case "getPubKey":
+		err := getPubKeyCmd.Parse(os.Args[2:])
+		if err != nil {
+			log.Panic(err)
+		}
+	case "getPubKeyHash":
+		err := getPubKeyHashCmd.Parse(os.Args[2:])
+		if err != nil {
+			log.Panic(err)
+		}
+	case "getAddress":
+		err := getAddressCmd.Parse(os.Args[2:])
+		if err != nil {
+			log.Panic(err)
+		}
+	case "getBlock":
+		err := getBlockCmd.Parse(os.Args[2:])
+		if err != nil {
+			log.Panic(err)
+		}
 	default:
 		cli.printUsage()
 		os.Exit(1)
@@ -152,4 +201,53 @@ func (cli *CLI) Run() {
 		}
 		cli.startNode(nodeID, *startNodeMiner)
 	}
+
+	if generatePrivKeyCmd.Parsed() {
+		cli.generatePrivKey()
+	}
+
+	if getPubKeyCmd.Parsed() {
+		if *privateKey == "" {
+			getPubKeyCmd.Usage()
+			os.Exit(1)
+		}
+		cli.getPubKey(*privateKey)
+	}
+
+	if getAddressCmd.Parsed() {
+		if *pubKey == "" {
+			getAddressCmd.Usage()
+			os.Exit(1)
+		}
+
+		cli.getAddress(*pubKey)
+	}
+
+	if getPubKeyHashCmd.Parsed() {
+		if *pubKeyAddress == "" {
+			getPubKeyHashCmd.Usage()
+			os.Exit(1)
+		}
+
+		cli.getPubKeyHash(*pubKeyAddress)
+	}
+
+	if validateAddrCmd.Parsed() {
+		if *address == "" {
+			validateAddrCmd.Usage()
+			os.Exit(1)
+		}
+
+		cli.validateAddr(*address)
+	}
+
+	if getBlockCmd.Parsed() {
+		if *blockHash == "" {
+			getBlockCmd.Usage()
+			os.Exit(1)
+		}
+
+		cli.printBlock(*blockHash, nodeID)
+	}
+
 }

cli_createblockchain.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"fmt"

cli_createwallet.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import "fmt"
 

cli_explore.go

@@ -0,0 +1,48 @@
+package bc
+
+import (
+	"crypto/elliptic"
+	"encoding/hex"
+	"fmt"
+)
+
+func (cli *CLI) getPubKey(privateKey string) {
+	curve := elliptic.P256()
+	priv_key, _ := hex.DecodeString(privateKey)
+	x, y := curve.ScalarBaseMult(priv_key)
+	pubKey := append(x.Bytes(), y.Bytes()...)
+	fmt.Println(hex.EncodeToString(pubKey))
+}
+
+func (cli *CLI) generatePrivKey() {
+	private, _ := newKeyPair()
+	fmt.Println(hex.EncodeToString(private.D.Bytes()))
+}
+
+func (cli *CLI) getAddress(pubKey string) {
+	public, _ := hex.DecodeString(pubKey)
+
+	pubKeyHash := HashPubKey(public)
+
+	versionedPayload := append([]byte{version}, pubKeyHash...)
+	fullPayload := append(versionedPayload, checksum(versionedPayload)...)
+
+	fmt.Println()
+	fmt.Printf("PubKey     : %s\n", pubKey)
+	fmt.Printf("PubKeyHash : %x\n", pubKeyHash)
+	fmt.Printf("Address    : %s\n", Base58Encode(fullPayload))
+}
+
+func (cli *CLI) getPubKeyHash(address string) {
+	pubKeyHash := Base58Decode([]byte(address))
+	fmt.Printf("%x\n", pubKeyHash[1:len(pubKeyHash)-4])
+}
+
+func (cli *CLI) validateAddr(address string) {
+	fmt.Printf("Address: %s\n", address)
+	if !ValidateAddress(address) {
+		fmt.Println("Not valid!")
+	} else {
+		fmt.Println("Valid!")
+	}
+}

cli_getbalance.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"fmt"

cli_listaddress.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"fmt"

cli_printchain.go

@@ -1,8 +1,9 @@
-package main
+package bc
 
 import (
 	"fmt"
 	"strconv"
+	"time"
 )
 
 func (cli *CLI) printChain(nodeID string) {
@@ -17,6 +18,7 @@ func (cli *CLI) printChain(nodeID string) {
 		fmt.Printf("============ Block %x ============\n", block.Hash)
 		fmt.Printf("Height: %d\n", block.Height)
 		fmt.Printf("Prev. block: %x\n", block.PrevBlockHash)
+		fmt.Printf("Created at : %s\n", time.Unix(block.Timestamp, 0))
 		pow := NewProofOfWork(block)
 		fmt.Printf("PoW: %s\n\n", strconv.FormatBool(pow.Validate()))
 		for _, tx := range block.Transactions {
@@ -29,3 +31,32 @@ func (cli *CLI) printChain(nodeID string) {
 		}
 	}
 }
+
+func (cli *CLI) printBlock(blockHash, nodeID string) {
+	bc := NewBlockchain(nodeID)
+	defer bc.db.Close()
+
+	bci := bc.Iterator()
+
+	for {
+		block := bci.Next()
+
+		hash := fmt.Sprintf("%x", block.Hash)
+		if hash == blockHash {
+			fmt.Printf("============ Block %x ============\n", block.Hash)
+			fmt.Printf("Height: %d\n", block.Height)
+			fmt.Printf("Prev. block: %x\n", block.PrevBlockHash)
+			fmt.Printf("Created at : %s\n", time.Unix(block.Timestamp, 0))
+			pow := NewProofOfWork(block)
+			fmt.Printf("PoW: %s\n\n", strconv.FormatBool(pow.Validate()))
+			for _, tx := range block.Transactions {
+				fmt.Println(tx)
+			}
+			fmt.Printf("\n\n")
+		}
+
+		if len(block.PrevBlockHash) == 0 {
+			break
+		}
+	}
+}

cli_reindexutxo.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import "fmt"
 

cli_send.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"fmt"
@@ -22,8 +22,12 @@ func (cli *CLI) send(from, to string, amount int, nodeID string, mineNow bool) {
 		log.Panic(err)
 	}
 	wallet := wallets.GetWallet(from)
+	if wallet == nil {
+		fmt.Println("The Address doesn't belongs to you!")
+		return
+	}
 
-	tx := NewUTXOTransaction(&wallet, to, amount, &UTXOSet)
+	tx := NewUTXOTransaction(wallet, to, amount, &UTXOSet)
 
 	if mineNow {
 		cbTx := NewCoinbaseTX(from, "")

cli_startnode.go

@@ -1,4 +1,4 @@
-package main
+package bc
 
 import (
 	"fmt"