diff --git a/peer/peer.go b/peer/peer.go index 44a0f56a..d12ddea2 100644 --- a/peer/peer.go +++ b/peer/peer.go @@ -1227,8 +1227,9 @@ func (p *Peer) maybeAddDeadline(pendingResponses map[string]time.Time, msgCmd st pendingResponses[wire.CmdInv] = deadline case wire.CmdGetData: - // Expects a block, tx, or notfound message. + // Expects a block, merkleblock, tx, or notfound message. pendingResponses[wire.CmdBlock] = deadline + pendingResponses[wire.CmdMerkleBlock] = deadline pendingResponses[wire.CmdTx] = deadline pendingResponses[wire.CmdNotFound] = deadline @@ -1285,10 +1286,13 @@ out: switch msgCmd := msg.message.Command(); msgCmd { case wire.CmdBlock: fallthrough + case wire.CmdMerkleBlock: + fallthrough case wire.CmdTx: fallthrough case wire.CmdNotFound: delete(pendingResponses, wire.CmdBlock) + delete(pendingResponses, wire.CmdMerkleBlock) delete(pendingResponses, wire.CmdTx) delete(pendingResponses, wire.CmdNotFound)