From b2a4901aff728e8dde4202bc22dd3de8acda607b Mon Sep 17 00:00:00 2001 From: Nalin Bhardwaj Date: Sat, 24 Mar 2018 14:56:48 +0530 Subject: [PATCH] chainparams: create isTestnet function Tests if passed parameters correspond to a testnet configuration. --- chainparams.go | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/chainparams.go b/chainparams.go index 20510aa3..fa726bb3 100644 --- a/chainparams.go +++ b/chainparams.go @@ -3,10 +3,11 @@ package main import ( "github.com/lightningnetwork/lnd/keychain" litecoinCfg "github.com/ltcsuite/ltcd/chaincfg" + litecoinWire "github.com/ltcsuite/ltcd/wire" "github.com/roasbeef/btcd/chaincfg" bitcoinCfg "github.com/roasbeef/btcd/chaincfg" "github.com/roasbeef/btcd/chaincfg/chainhash" - "github.com/roasbeef/btcd/wire" + bitcoinWire "github.com/roasbeef/btcd/wire" ) // activeNetParams is a pointer to the parameters specific to the currently @@ -82,7 +83,7 @@ var regTestNetParams = bitcoinNetParams{ // abstract over _which_ chain (or fork) the parameters are for. func applyLitecoinParams(params *bitcoinNetParams, litecoinParams *litecoinNetParams) { params.Name = litecoinParams.Name - params.Net = wire.BitcoinNet(litecoinParams.Net) + params.Net = bitcoinWire.BitcoinNet(litecoinParams.Net) params.DefaultPort = litecoinParams.DefaultPort params.CoinbaseMaturity = litecoinParams.CoinbaseMaturity @@ -116,3 +117,14 @@ func applyLitecoinParams(params *bitcoinNetParams, litecoinParams *litecoinNetPa params.rpcPort = litecoinParams.rpcPort params.CoinType = litecoinParams.CoinType } + +// isTestnet tests if the given params correspond to a testnet +// parameter configuration. +func isTestnet(params *bitcoinNetParams) bool { + switch params.Params.Net { + case bitcoinWire.TestNet3, bitcoinWire.BitcoinNet(litecoinWire.TestNet4): + return true + default: + return false + } +}