From ca9cf22ca0a57e881858d1773de64aaeee23a075 Mon Sep 17 00:00:00 2001
From: Newt <guoching130@gmail.com>
Date: Tue, 12 Nov 2024 01:28:35 +0800
Subject: [PATCH] fix: connection.BlockFetch().Client.GetBlock potential
 deadlock

---
 protocol/blockfetch/client.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/protocol/blockfetch/client.go b/protocol/blockfetch/client.go
index e9f688d8..f581692a 100644
--- a/protocol/blockfetch/client.go
+++ b/protocol/blockfetch/client.go
@@ -164,6 +164,7 @@ func (c *Client) GetBlock(point common.Point) (ledger.Block, error) {
 	}
 	err, ok := <-c.startBatchResultChan
 	if !ok {
+		c.busyMutex.Unlock()
 		return nil, protocol.ProtocolShuttingDownError
 	}
 	if err != nil {
@@ -172,6 +173,7 @@ func (c *Client) GetBlock(point common.Point) (ledger.Block, error) {
 	}
 	block, ok := <-c.blockChan
 	if !ok {
+		c.busyMutex.Unlock()
 		return nil, protocol.ProtocolShuttingDownError
 	}
 	return block, nil