replication factors in prologue length

holepunchto · Oct 10, 2024 · 67c6a79 · 67c6a79
1 parent 9aa5859
commit 67c6a79
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lib/merkle-tree.js b/lib/merkle-tree.js
@@ -428,7 +428,7 @@ module.exports = class MerkleTree {
     this.crypto = crypto
     this.fork = fork
     this.roots = roots
-    this.length = roots.length ? totalSpan(roots) / 2 : prologue ? prologue.length : 0
+    this.length = roots.length ? totalSpan(roots) / 2 : 0
     this.byteLength = totalSize(roots)
     this.signature = signature
     this.prologue = prologue

diff --git a/lib/replicator.js b/lib/replicator.js
@@ -1189,7 +1189,7 @@ class Peer {
   }
 
   _requestBlock (b) {
-    const { length, fork } = this.core.tree
+    const { prologue, length, fork } = this.core.tree
 
     if (this._remoteHasBlock(b.index) === false || fork !== this.remoteFork) {
       this._maybeWant(b.index)
@@ -1200,7 +1200,9 @@ class Peer {
       return false
     }
 
-    const req = this._makeRequest(b.index >= length, b.priority, b.index + 1)
+    const treeLength = (prologue && prologue.length > length) ? prologue.length : length
+
+    const req = this._makeRequest(b.index >= treeLength, b.priority, b.index + 1)
     if (req === null) return false
 
     this._sendBlockRequest(req, b)