Add lt_donthave

1 files changed, 12 insertions, 1 deletions

diff --git a/lib/torrent.js b/lib/torrent.js
index ca58842..c87c079 100644
--- a/lib/torrent.js
+++ b/lib/torrent.js
@@ -11,6 +11,7 @@ const fs = require('fs')
 const FSChunkStore = require('fs-chunk-store') // browser: `memory-chunk-store`
 const get = require('simple-get')
 const ImmediateChunkStore = require('immediate-chunk-store')
+const ltDontHave = require('lt_donthave')
 const MultiStream = require('multistream')
 const net = require('net') // browser exclude
 const os = require('os') // browser exclude
@@ -1012,7 +1013,9 @@ class Torrent extends EventEmitter {
     })
 
     // Timeout for piece requests to this peer
-    wire.setTimeout(PIECE_TIMEOUT, true)
+    if (wire.type !== 'webSeed') { // webseeds always send 'unhave' on http timeout
+      wire.setTimeout(PIECE_TIMEOUT, true)
+    }
 
     // Send KEEP-ALIVE (every 60s) so peers will not disconnect the wire
     wire.setKeepAlive(true)
@@ -1059,6 +1062,8 @@ class Torrent extends EventEmitter {
       })
     }
 
+    wire.use(ltDontHave())
+
     // Hook to allow user-defined `bittorrent-protocol` extensions
     // More info: https://github.com/webtorrent/bittorrent-protocol#extension-api
     this.emit('wire', wire, addr)
@@ -1109,6 +1114,12 @@ class Torrent extends EventEmitter {
       this._updateWireInterest(wire)
     })
 
+    wire.lt_donthave.on('donthave', () => {
+      updateSeedStatus()
+      this._update()
+      this._updateWireInterest(wire)
+    })
+
     wire.once('interested', () => {
       wire.unchoke()
     })