GH-29: RunTheNumbers on startup

Author: onyb

bus/infrastructure.go

@@ -6,6 +6,8 @@ import (
 	"sync"
 	"time"
 
+	"github.com/btcsuite/btcutil"
+
 	"github.com/btcsuite/btcd/chaincfg/chainhash"
 
 	"github.com/ledgerhq/satstack/config"
@@ -336,3 +338,47 @@ func (b *Bus) descriptors(account config.Account) ([]descriptor, error) {
 
 	return ret, nil
 }
+
+// RunTheNumbers performs inflation checks against the connected full node.
+//
+// It does NOT perform any equality comparison between expected and actual
+// supply.
+func (b *Bus) RunTheNumbers() error {
+	client := b.getClient()
+	defer b.recycleClient(client)
+
+	log.Info("Running inflation checks")
+
+	info, err := client.GetTxOutSetInfo()
+	if err != nil {
+		return err
+	}
+
+	const halvingBlocks = 210000
+
+	var (
+		subsidy float64 = 50
+		supply  float64 = 0
+	)
+
+	i := int64(0)
+	for ; i < info.Height/halvingBlocks; i++ {
+		supply += halvingBlocks * subsidy
+		subsidy /= 2
+	}
+
+	supply += subsidy * float64(info.Height-(halvingBlocks*i))
+
+	supplyBTC, err := btcutil.NewAmount(supply)
+	if err != nil {
+		return err
+	}
+
+	log.WithFields(log.Fields{
+		"height":         info.Height,
+		"expectedSupply": supplyBTC,
+		"actualSupply":   info.TotalAmount,
+	}).Info("#RunTheNumbers OK️")
+
+	return nil
+}

cmd/lss.go

@@ -68,6 +68,12 @@ func main() {
 
 		b.Status = bus.PendingScan
 
+		if err := b.RunTheNumbers(); err != nil {
+			log.WithFields(log.Fields{
+				"error": err,
+			}).Fatal("Inflation checks failed")
+		}
+
 		// Skip import of descriptors, if no account config found. SatStack
 		// will run in zero-configuration mode.
 		if configuration.Accounts == nil {

go.mod

@@ -19,3 +19,5 @@ require (
 	golang.org/x/net v0.0.0-20200226121028-0de0cce0169b // indirect
 	golang.org/x/sys v0.0.0-20200523222454-059865788121 // indirect
 )
+
+replace github.com/btcsuite/btcd v0.21.0-beta.0.20200925161806-e9a51e8dcd67 => github.com/onyb/btcd v0.20.1-beta.0.20200928132705-7b5f3feb11c0

go.sum

@@ -1,7 +1,5 @@
 github.com/aead/siphash v1.0.1/go.mod h1:Nywa3cDsYNNK3gaciGTWPwHt0wlpNV15vwmswBAUSII=
 github.com/btcsuite/btcd v0.20.1-beta/go.mod h1:wVuoA8VJLEcwgqHBwHmzLRazpKxTv13Px/pDuV7OomQ=
-github.com/btcsuite/btcd v0.21.0-beta.0.20200925161806-e9a51e8dcd67 h1:Dl9SKb/DS8dGvQgpCCgBXbgE05OAsl8fqMgBFwsI2wo=
-github.com/btcsuite/btcd v0.21.0-beta.0.20200925161806-e9a51e8dcd67/go.mod h1:ZSWyehm27aAuS9bvkATT+Xte3hjHZ+MRgMY/8NJ7K94=
 github.com/btcsuite/btclog v0.0.0-20170628155309-84c8d2346e9f h1:bAs4lUbRJpnnkd9VhRV3jjAVU7DJVjMaK+IsvSeZvFo=
 github.com/btcsuite/btclog v0.0.0-20170628155309-84c8d2346e9f/go.mod h1:TdznJufoqS23FtqVCzL0ZqgP5MqXbb4fg/WgDys70nA=
 github.com/btcsuite/btcutil v0.0.0-20190425235716-9e5f4b9a998d/go.mod h1:+5NJ2+qvTyV9exUAL/rxXi3DcLg2Ts+ymUAY5y4NvMg=
@@ -83,6 +81,8 @@ github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+W
 github.com/onsi/gomega v1.4.1/go.mod h1:C1qb7wdrVGGVU+Z6iS04AVkA3Q65CEZX59MT0QO5uiA=
 github.com/onsi/gomega v1.4.3 h1:RE1xgDvH7imwFD45h+u2SgIfERHlS2yNG4DObb5BSKU=
 github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
+github.com/onyb/btcd v0.20.1-beta.0.20200928132705-7b5f3feb11c0 h1:0LJRBFvQaMFjoa6iADdYLrq6SzeyqCp+RTmhy6zuqJU=
+github.com/onyb/btcd v0.20.1-beta.0.20200928132705-7b5f3feb11c0/go.mod h1:ZSWyehm27aAuS9bvkATT+Xte3hjHZ+MRgMY/8NJ7K94=
 github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc=
 github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=