factom-walletd.go

// Copyright 2017 Factom Foundation
// Use of this source code is governed by the MIT
// license that can be found in the LICENSE file.

package main

import (
	"flag"
	"fmt"
	"log"
	"os"
	"os/signal"
	"strings"
	"syscall"

	"github.com/FactomProject/factom"
	"github.com/FactomProject/factomd/database/securedb"
	"github.com/FactomProject/factomd/util"
	"github.com/FactomProject/wallet/v2"
	"github.com/FactomProject/wallet/v2/wsapi"
)

func main() {
	// configure the server
	var (
		pflag = flag.Int("p", 8089, "set the port to host the wsapi")
		wflag = flag.String("w", "", "set the default wallet location")
		iflag = flag.String("i", "", "Import a version 1 wallet. Set as path to factoid_wallet_bolt.db")
		mflag = flag.String("m", "", "import a wallet from 12 word mnemonic")
		eflag = flag.Bool("e", false, "export a wallet for backup")
		lflag = flag.Bool("l", false, "Create or use an LDB database")

		configPath = flag.String("config", "", "Override the config file location (factomd.conf)")

		// Use TLS for the wallet "factom-walletd -wallettls=true"
		walletTLSflag = flag.Bool("wallettls", false, "Set to true to require encrypted connections to the wallet")
		walletTLSKey  = flag.String("walletkey", "", "This file is the PRIVATE TLS key encrypting connections to the wallet. (default ~/.factom/walletAPIpriv.key)")
		walletTLSCert = flag.String("walletcert", "", "This file is the PUBLIC TLS certificate wallet API users will need to connect. (default ~/.factom/walletAPIpub.cert)")

		factomdTLSflag = flag.Bool("factomdtls", false, "Set to true when the factomd API is encrypted")
		factomdTLSCert = flag.String("factomdcert", "", "This file is the TLS certificate provided by the factomd API. (default ~/.factom/m2/factomdAPIpub.cert)")

		walletRpcUser      = flag.String("walletuser", "", "Username to expect before allowing connections")
		walletRpcPassword  = flag.String("walletpassword", "", "Password to expect before allowing connections")
		factomdRpcUser     = flag.String("factomduser", "", "Username for API connections to factomd")
		factomdRpcPassword = flag.String("factomdpassword", "", "Password for API connections to factomd")
		corsDomains        = flag.String("corsdomains", "", "CORS Domains")

		factomdLocation = flag.String("s", "", "IPAddr:port# of factomd API to use to access blockchain (default localhost:8088)")
		walletdLocation = flag.String("selfaddr", "", "comma seperated IPAddresses and DNS names of this factom-walletd to use when creating a cert file")
		encryptedDB     = flag.Bool("encrypted", false, "Option to enable encryption for database when not in use.")
		passphrase      = flag.String("passphrase", "", "Passphrase used to encrypt/decrypt the wallet")
		remote          = flag.Bool("remote", false, "Allow remote connections to walletd")
	)
	flag.Parse()

	// see if the config file has values which should be used instead of null strings
	filename := util.ConfigFilename()
	if *configPath != "" {
		filename = *configPath
	}
	cfg := util.ReadConfig(filename)

	if !*encryptedDB {
		if cfg.Walletd.WalletEncrypted {
			fmt.Printf("WalletEncryption enabled in config file at: %s\n", filename)
			*encryptedDB = true
		}
	}

	// set the wallet path to the wflag or to the default
	walletPath := util.GetHomeDir() + "/.factom/wallet/factom_wallet.db"
	if *lflag {
		walletPath = util.GetHomeDir() + "/.factom/wallet/factom_wallet.ldb"
		if *encryptedDB {
			fmt.Println("WalletEncryption option is enabled, but is incompatible with levelDB")
			os.Exit(1)
		}
	}

	encryptedPath := util.GetHomeDir() + "/.factom/wallet/factom_wallet_encrypted.db"
	isEncryptedFirstBoot := false

	if *wflag != "" {
		walletPath = *wflag
	}

	// Conditions around using the encrypted wallet
	if *encryptedDB {
		// Check if regular wallet exists, exit if the wrong wallet exists, assuming an unsafe configuration issue.
		// ignore the check if a custom path were specified.  Assume more competence in that case
		if *wflag == "" {
			_, err := os.Stat(walletPath)
			if !os.IsNotExist(err) {
				// Regular wallet exists, exit
				fmt.Printf("Encrypted Wallet option was selected, however an unencrypted wallet already exists.\n"+
					"Remove or rename the wallet file at '%s' to launch factom-walletd with encryption. "+
					"(Back it up before deleting!)\n", walletPath)
				os.Exit(1)
			}
		}

		// Change the path to encrypted path, or use the specified path
		if *wflag != "" {
			walletPath = *wflag
		} else {
			walletPath = encryptedPath
		}

		// If the wallet doesn't already exist, make sure that a passphrase was specified when making a new wallet.
		_, err := os.Stat(walletPath)
		if os.IsNotExist(err) {
			if *passphrase == "" {
				fmt.Println("WalletEncryption option is enabled, but no encrypted database exists yet.\n" +
					"When starting an encrypted wallet for the first time, you must also specifiy a '-passphrase'")
				os.Exit(1)
			}
			// At this point, encrypted wallet database does not yet exist, and we provided a password to bootstrap it.
			isEncryptedFirstBoot = true
		} else if *passphrase != "" {
			fmt.Printf("An encrypted database already exists at %s. But a '-passphrase' was also provided.\n"+
				"To start the existing encrypted wallet, do not supply the '-passphrase' argument. You will then be able "+
				"to unlock the wallet at any point in the future using the 'unlock-wallet' RPC method.\n"+
				"If you would like to start a new encrypted wallet, rename the above file or specify a new path with "+
				"the '-w' flag.\n", walletPath)
			os.Exit(1)
		}
	} else {
		_, err := os.Stat(encryptedPath)
		if !os.IsNotExist(err) {
			// Encrypted wallet exists, but trying to start unencrypted. Exit assuming an unsafe configuration issue.
			fmt.Printf("The wallet is being launched without database encryption, however an encrypted wallet already exists.\n"+
				"Remove or rename the wallet file at '%s' to launch factom-walletd without encryption. "+
				"(Back it up before deleting!)\n", encryptedPath)

			os.Exit(1)
		}

		fmt.Printf("Warning, factom-walletd database is not encrypted. Private keys are stored plainly.\n")

		// Check if encrypted wallet exists
	}

	if *walletRpcUser == "" {
		if cfg.Walletd.WalletRpcUser != "" {
			fmt.Printf("using factom-walletd API user and password specified in \"%s\" at WalletRpcUser & WalletRpcPass\n", filename)
			*walletRpcUser = cfg.Walletd.WalletRpcUser
			*walletRpcPassword = cfg.Walletd.WalletRpcPass
		} else {
			fmt.Printf("Warning, factom-walletd API is not password protected. Factoids can be stolen remotely.\n")
		}
	} else {
		fmt.Printf("wallet access protected by user/password specified on command line\n")
	}

	if *factomdRpcUser == "" {
		if cfg.App.FactomdRpcUser != "" {
			fmt.Printf("using factomd API user and password specified in \"%s\" at FactomdRpcUser & FactomdRpcPass\n", filename)
			*factomdRpcUser = cfg.App.FactomdRpcUser
			*factomdRpcPassword = cfg.App.FactomdRpcPass
		}
	}

	if *factomdLocation == "" {
		if cfg.Walletd.FactomdLocation != "localhost:8088" {
			fmt.Printf("using factomd location specified in \"%s\" at FactomdLocation = \"%s\"\n", filename, cfg.Walletd.FactomdLocation)
			*factomdLocation = cfg.Walletd.FactomdLocation
		} else {
			*factomdLocation = "localhost:8088"
		}
	}

	if cfg.Walletd.WalletTlsEnabled == true {
		*walletTLSflag = true
	}
	if *walletTLSflag == true {
		if *walletTLSKey == "" { //if specified, instead use what was on the command line
			if cfg.Walletd.WalletTlsPrivateKey != "/full/path/to/walletAPIpriv.key" { //otherwise check if the the config file has something new
				fmt.Printf("using wallet TLS key file specified in \"%s\" at WalletTlsPrivateKey = \"%s\"\n", filename, cfg.Walletd.WalletTlsPrivateKey)
				*walletTLSKey = cfg.Walletd.WalletTlsPrivateKey
			} else { //if none were specified, use the default file
				*walletTLSKey = fmt.Sprint(util.GetHomeDir(), "/.factom/walletAPIpriv.key")
				fmt.Printf("using default wallet TLS key file \"%s\"\n", *walletTLSKey)
			}
		} else {
			fmt.Printf("using specified wallet TLS key file \"%s\"\n", *walletTLSKey)
		}
		if *walletTLSCert == "" { //if specified, instead use what was on the command line
			if cfg.Walletd.WalletTlsPublicCert != "/full/path/to/walletAPIpub.cert" { //otherwise check if the the config file has something new
				fmt.Printf("using wallet TLS certificate file specified in \"%s\" at WalletTlsPublicCert = \"%s\"\n", filename, cfg.Walletd.WalletTlsPublicCert)
				*walletTLSCert = cfg.Walletd.WalletTlsPublicCert
			} else { //if none were specified, use the default file
				*walletTLSCert = fmt.Sprint(util.GetHomeDir(), "/.factom/walletAPIpub.cert")
				fmt.Printf("using default wallet TLS certificate file \"%s\"\n", *walletTLSCert)
			}
		} else {
			fmt.Printf("using specified wallet TLS certificate file \"%s\"\n", *walletTLSCert)
		}
	} else {
		fmt.Printf("Warning, factom-walletd API connection is unencrypted. RPC password is unprotected over the network.\n")
	}

	if cfg.Walletd.WalletdLocation != "localhost:8089" {
		//fmt.Printf("using factom-walletd location specified in \"%s\" as WalletdLocation = \"%s\"\n", filename, cfg.Walletd.WalletdLocation)
		var externalIP string
		externalIP += strings.Split(cfg.Walletd.WalletdLocation, ":")[0]
		if *walletdLocation != "" {
			*walletdLocation += ","
		}
		*walletdLocation += externalIP
	}
	if *walletdLocation != "" {
		fmt.Printf("external IP and DNS name to use if making a new TLS keypair = %s\n", *walletdLocation)
	}

	if cfg.App.FactomdTlsEnabled == true {
		*factomdTLSflag = true
	}
	if *factomdTLSflag == true {
		if *factomdTLSCert == "" { //if specified on the command line, don't use the config file
			if cfg.App.FactomdTlsPublicCert != "/full/path/to/factomdAPIpub.cert" { //otherwise check if the the config file has something new
				fmt.Printf("using wallet TLS certificate file specified in \"%s\" at FactomdTlsPublicCert = \"%s\"\n", filename, cfg.App.FactomdTlsPublicCert)
				*factomdTLSCert = cfg.App.FactomdTlsPublicCert
			} else { //if none were specified, use the default file
				*factomdTLSCert = fmt.Sprint(util.GetHomeDir(), "/.factom/m2/factomdAPIpub.cert")
				fmt.Printf("using default factomd TLS certificate file \"%s\"\n", *factomdTLSCert)
			}
		} else {
			fmt.Printf("using specified factomd TLS certificate file \"%s\"\n", *factomdTLSCert)
		}
	}

	if *corsDomains == "" {
		if cfg.App.CorsDomains != "" {
			*corsDomains = cfg.App.CorsDomains
		}
	}

	port := *pflag
	RPCConfig := factom.RPCConfig{
		WalletTLSEnable:   *walletTLSflag,
		WalletTLSKeyFile:  *walletTLSKey,
		WalletTLSCertFile: *walletTLSCert,
		WalletRPCUser:     *walletRpcUser,
		WalletRPCPassword: *walletRpcPassword,
		WalletServer:      *walletdLocation,
		WalletCORSDomains: *corsDomains,
	}
	factom.SetFactomdRpcConfig(*factomdRpcUser, *factomdRpcPassword)
	factom.SetFactomdServer(*factomdLocation)
	factom.SetFactomdEncryption(*factomdTLSflag, *factomdTLSCert)

	if *mflag != "" {
		log.Printf("Creating new wallet with mnemonic")
		w, err := func() (*wallet.Wallet, error) {
			if *encryptedDB {
				return wallet.ImportEncryptedWalletFromMnemonic(*mflag, walletPath, *passphrase)
			}
			if *lflag {
				return wallet.ImportLDBWalletFromMnemonic(*mflag, walletPath)
			}
			return wallet.ImportWalletFromMnemonic(*mflag, walletPath)
		}()
		if err != nil {
			log.Fatal(err)
		}

		fmt.Printf("SUCCESS: re-run factom-walletd without the -m flag to start the wallet from the newly imported seed")
		w.Close()
		os.Exit(0)
	}

	if *iflag != "" {
		if *encryptedDB {
			fmt.Printf("EXIT: v1 wallet imports are not officially supported for encrypted wallets")
			os.Exit(1)
		}

		log.Printf("Importing version 1 wallet %s into %s", *iflag, walletPath)
		w, err := func() (*wallet.Wallet, error) {
			if *encryptedDB {
				return wallet.ImportV1WalletToEncryptedDB(*iflag, walletPath, *passphrase)
			}
			if *lflag {
				return wallet.ImportV1WalletToLDB(*iflag, walletPath)
			}
			return wallet.ImportV1Wallet(*iflag, walletPath)
		}()
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("SUCCESS: re-run factom-walletd without the -i flag to start the wallet from the newly created wallet")
		w.Close()
		os.Exit(0)
	}

	if *eflag {
		if *encryptedDB {
			fmt.Printf("EXIT: Wallet exports from the cli are not allowed for encrypted wallets.\n" +
				"However, you can perform a back up by performing the following:\n" +
				"1. boot the encrypted wallet normally\n" +
				"2. unlock it via the 'unlock-wallet' RPC call\n" +
				"3. issue a 'wallet-backup' RPC call before the unlock timeout period\n")
			os.Exit(1)
		}
		m, fs, es, err := func() (string, []*factom.FactoidAddress, []*factom.ECAddress, error) {
			if *lflag {
				return wallet.ExportLDBWallet(walletPath)
			}
			return wallet.ExportWallet(walletPath)
		}()
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("%q\n", m)
		for _, f := range fs {
			fmt.Println(f.SecString())
		}
		for _, e := range es {
			fmt.Println(e.SecString())
		}
		os.Exit(0)
	}

	// setup handling for os signals and stop the server gracefully
	go func() {
		c := make(chan os.Signal, 1)
		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
		for sig := range c {
			log.Printf("Captured %v, stopping web server and exiting", sig)
			wsapi.Stop()
		}
	}()

	// open or create a new wallet file
	fctWallet, err := func() (*wallet.Wallet, error) {
		if *encryptedDB {
			if isEncryptedFirstBoot {
				// At this point: encrypted wallet db file does not exist, a bootstrap passphrase was provided
				// So now create the wallet object with a new encrypted boltdb
				return wallet.NewEncryptedBoltDBWallet(walletPath, *passphrase)
			}
			// Not the first boot for our encrypted wallet: an encrypted boltdb must have been created with a password previously
			// So now start the server without a dboverlay and wait for passphrase to be sent over wsapi
			return wallet.NewEncryptedBoltDBWalletAwaitingPassphrase(walletPath)
		}
		if *lflag {
			return wallet.NewOrOpenLevelDBWallet(walletPath)
		}
		return wallet.NewOrOpenBoltDBWallet(walletPath)
	}()
	if err != nil {
		log.Fatal(err)
	}

	// open and add a transaction database to the wallet object.
	txdb, err := wallet.NewTXBoltDB(walletPath + ".cache")
	if err != nil {
		log.Println("Could not add transaction database to wallet:", err)
	} else {
		txdb.Update()
		fctWallet.AddTXDB(txdb)
	}

	// If it is encrypted, we need to start the wallet as locked.
	// However, if it is not the first boot, the DBO does not exist yet. So we can't lock what doesn't exist
	if *encryptedDB && isEncryptedFirstBoot {
		fctWallet.DBO.DB.(*securedb.EncryptedDB).Lock()
	}

	// start the wsapi server
	bindTo := "127.0.0.1"
	if *remote {
		fmt.Println("Remote connections to this wallet are enabled")
		bindTo = "0.0.0.0"
	} else {
		fmt.Println("Remote connections to this wallet are disabled")
	}
	wsapi.Start(fctWallet, fmt.Sprintf("%s:%d", bindTo, port), RPCConfig)
}