From d160a87b2a2d407cb2ed9d4decfa246fca75f75c Mon Sep 17 00:00:00 2001
From: Olaoluwa Osuntokun <laolu32@gmail.com>
Date: Thu, 24 Mar 2016 14:32:55 -0700
Subject: [PATCH] register waddrmgr as the default EncryptorDecryptor on
 startup

---
 lnd.go | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/lnd.go b/lnd.go
index 67aacb40..e85e7ed5 100644
--- a/lnd.go
+++ b/lnd.go
@@ -56,7 +56,7 @@ func main() {
 
 	// Open the channeldb, which is dedicated to storing channel, and
 	// network related meta-data.
-	chanDB, err := channeldb.New(loadedConfig.DataDir, nil)
+	chanDB, err := channeldb.Open(loadedConfig.DataDir)
 	if err != nil {
 		fmt.Println("unable to open channeldb: ", err)
 		os.Exit(1)
@@ -87,25 +87,28 @@ func main() {
 		CACert:      cert,
 		NetParams:   activeNetParams,
 	}
-	lnwallet, err := lnwallet.NewLightningWallet(config, chanDB)
+	wallet, err := lnwallet.NewLightningWallet(config, chanDB)
 	if err != nil {
 		fmt.Printf("unable to create wallet: %v\n", err)
 		os.Exit(1)
 	}
-	if err := lnwallet.Startup(); err != nil {
+	if err := wallet.Startup(); err != nil {
 		fmt.Printf("unable to start wallet: %v\n", err)
 		os.Exit(1)
 	}
 	ltndLog.Info("LightningWallet opened")
 
+	ec := &lnwallet.WaddrmgrEncryptorDecryptor{wallet.Manager}
+	chanDB.RegisterCryptoSystem(ec)
+
 	// Set up the core server which will listen for incoming peer
 	// connections.
 	defaultListenAddrs := []string{
 		net.JoinHostPort("", strconv.Itoa(loadedConfig.PeerPort)),
 	}
-	server, err := newServer(defaultListenAddrs, lnwallet, chanDB)
+	server, err := newServer(defaultListenAddrs, wallet, chanDB)
 	if err != nil {
-		fmt.Printf("unable to create server: %v\n", err)
+		srvrLog.Errorf("unable to create server: %v\n", err)
 		os.Exit(1)
 	}
 	server.Start()