From e7bae846627fcf2763428aa1fe620521bbd629ee Mon Sep 17 00:00:00 2001
From: Alex <alex@akselrod.org>
Date: Thu, 4 May 2017 12:46:54 -0600
Subject: [PATCH] Rescan works and tests pass.

---
 spvsvc/spvchain/query.go     | 29 ++++++++++++++++++++---------
 spvsvc/spvchain/sync_test.go |  5 +++++
 2 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/spvsvc/spvchain/query.go b/spvsvc/spvchain/query.go
index 9de6d95..f23f468 100644
--- a/spvsvc/spvchain/query.go
+++ b/spvsvc/spvchain/query.go
@@ -292,6 +292,9 @@ func (s *ChainService) GetCFilter(blockHash chainhash.Hash,
 	if err == nil && filter != nil {
 		return filter
 	}
+	// We didn't get the filter from the DB, so we'll set it to nil and try
+	// to get it from the network.
+	filter = nil
 	block, _, err := s.GetBlockByHash(blockHash)
 	if err != nil || block.BlockHash() != blockHash {
 		return nil
@@ -314,6 +317,12 @@ func (s *ChainService) GetCFilter(blockHash chainhash.Hash,
 			switch response := resp.(type) {
 			// We're only interested in "cfilter" messages.
 			case *wire.MsgCFilter:
+				// Only keep this going if we haven't already
+				// found a filter, or we risk closing an already
+				// closed channel.
+				if filter != nil {
+					return
+				}
 				if len(response.Data) < 4 {
 					// Filter data is too short.
 					// Ignore this message.
@@ -324,17 +333,15 @@ func (s *ChainService) GetCFilter(blockHash chainhash.Hash,
 					// request. Ignore this message.
 					return
 				}
-				gotFilter, err :=
-					gcs.FromNBytes(builder.DefaultP,
-						response.Data)
+				gotFilter, err := gcs.FromNBytes(
+					builder.DefaultP, response.Data)
 				if err != nil {
 					// Malformed filter data. We
 					// can ignore this message.
 					return
 				}
 				if builder.MakeHeaderForFilter(gotFilter,
-					*prevHeader) !=
-					*curHeader {
+					*prevHeader) != *curHeader {
 					// Filter data doesn't match
 					// the headers we know about.
 					// Ignore this response.
@@ -386,9 +393,14 @@ func (s *ChainService) GetBlockFromNetwork(
 			switch response := resp.(type) {
 			// We're only interested in "block" messages.
 			case *wire.MsgBlock:
+				// Only keep this going if we haven't already
+				// found a block, or we risk closing an already
+				// closed channel.
+				if foundBlock != nil {
+					return
+				}
 				// If this isn't our block, ignore it.
-				if response.BlockHash() !=
-					blockHash {
+				if response.BlockHash() != blockHash {
 					return
 				}
 				block := btcutil.NewBlock(response)
@@ -396,8 +408,7 @@ func (s *ChainService) GetBlockFromNetwork(
 				// automagically put one in.
 				if block.Height() ==
 					btcutil.BlockHeightUnknown {
-					block.SetHeight(
-						int32(height))
+					block.SetHeight(int32(height))
 				}
 				// If this claims our block but doesn't
 				// pass the sanity check, the peer is
diff --git a/spvsvc/spvchain/sync_test.go b/spvsvc/spvchain/sync_test.go
index 94e66bf..c5356d0 100644
--- a/spvsvc/spvchain/sync_test.go
+++ b/spvsvc/spvchain/sync_test.go
@@ -32,6 +32,11 @@ import (
 )
 
 var (
+	// Try btclog.InfoLvl for output like you'd see in normal operation, or
+	// btclog.TraceLvl to help debug code. Anything but btclog.Off turns on
+	// log messages from the tests themselves as well. Keep in mind some
+	// log messages may not appear in order due to use of multiple query
+	// goroutines in the tests.
 	logLevel    = btclog.Off
 	syncTimeout = 30 * time.Second
 	syncUpdate  = time.Second