From 89537318c1a98cf972a6fa8c542639a729a96c8c Mon Sep 17 00:00:00 2001
From: Olaoluwa Osuntokun <laolu32@gmail.com>
Date: Sat, 17 Jun 2017 00:12:34 +0200
Subject: [PATCH] chains: add default routing policy for current supported
 blockchains

---
 chainregistry.go | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/chainregistry.go b/chainregistry.go
index f54cafd3..a7d818c8 100644
--- a/chainregistry.go
+++ b/chainregistry.go
@@ -15,6 +15,7 @@ import (
 	"github.com/lightningnetwork/lnd/chainntnfs/btcdnotify"
 	"github.com/lightningnetwork/lnd/chainntnfs/neutrinonotify"
 	"github.com/lightningnetwork/lnd/channeldb"
+	"github.com/lightningnetwork/lnd/htlcswitch"
 	"github.com/lightningnetwork/lnd/lnwallet"
 	"github.com/lightningnetwork/lnd/lnwallet/btcwallet"
 	"github.com/lightningnetwork/lnd/routing/chainview"
@@ -24,6 +25,24 @@ import (
 	"github.com/roasbeef/btcwallet/walletdb"
 )
 
+// defaultBitcoinForwardingPolicy is the default forwarding policy used for
+// Bitcoin channels.
+var defaultBitcoinForwardingPolicy = htlcswitch.ForwardingPolicy{
+	MinHTLC:       0,
+	BaseFee:       1,
+	FeeRate:       1,
+	TimeLockDelta: 1,
+}
+
+// defaultLitecoinForwardingPolicy is the default forwarding policy used for
+// Litecoin channels.
+var defaultLitecoinForwardingPolicy = htlcswitch.ForwardingPolicy{
+	MinHTLC:       0,
+	BaseFee:       1,
+	FeeRate:       1,
+	TimeLockDelta: 1,
+}
+
 // chainCode is an enum-like structure for keeping track of the chains currently
 // supported within lnd.
 type chainCode uint32
@@ -65,6 +84,8 @@ type chainControl struct {
 	chainView chainview.FilteredChainView
 
 	wallet *lnwallet.LightningWallet
+
+	routingPolicy htlcswitch.ForwardingPolicy
 }
 
 // newChainControlFromConfig attempts to create a chainControl instance
@@ -93,6 +114,16 @@ func newChainControlFromConfig(cfg *config, chanDB *channeldb.DB) (*chainControl
 		feeEstimator: estimator,
 	}
 
+	switch registeredChains.PrimaryChain() {
+	case bitcoinChain:
+		cc.routingPolicy = defaultBitcoinForwardingPolicy
+	case litecoinChain:
+		cc.routingPolicy = defaultLitecoinForwardingPolicy
+	default:
+		return nil, nil, fmt.Errorf("Default routing policy for "+
+			"chain %v is unknown", registeredChains.PrimaryChain())
+	}
+
 	var (
 		err     error
 		cleanUp func()