Merge pull request #567 from earlbread/fix-expected-responses

Fix expectedResponses in GetBlocksAsync
planetarium · Oct 8, 2019 · 9885d6f · 9885d6f
2 parents e7ccac6 + 9f1ac42
commit 9885d6f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Libplanet/Net/Swarm.cs b/Libplanet/Net/Swarm.cs
@@ -919,11 +919,17 @@ internal async Task SendMessageAsync(BoundPeer peer, Message message)
                     blockHashes.ToArray();
                 var request = new GetBlocks(blockHashesAsArray);
                 int hashCount = blockHashesAsArray.Count();
+
+                if (hashCount < 1)
+                {
+                    yield.Break();
+                }
+
                 IEnumerable<Message> replies = await SendMessageWithReplyAsync(
                     peer,
                     request,
                     BlockRecvTimeout,
-                    (hashCount / request.ChunkSize) + 1,
+                    ((hashCount - 1) / request.ChunkSize) + 1,
                     yieldToken
                 );