Disable DHT and PEX when the torrent is flagged as private

Add test for private torrent - should disable DHT

Fail public torrent test when using a private torrent

standard

Disable DHT for the torrent's swarm and discovery.

1 files changed, 8 insertions, 4 deletions

diff --git a/lib/torrent.js b/lib/torrent.js
index 9d4ecdf..0275d70 100644
--- a/lib/torrent.js
+++ b/lib/torrent.js
@@ -196,7 +196,9 @@ Torrent.prototype._onParsedTorrent = function (parsedTorrent) {
 
   // create swarm
   self.swarm = new Swarm(self.infoHash, self.client.peerId, {
-    handshake: { dht: !!self.client.dht }
+    handshake: {
+      dht: self.private ? false : !!self.client.dht
+    }
   })
   self.swarm.on('error', self._onError.bind(self))
   self.swarm.on('wire', self._onWire.bind(self))
@@ -263,7 +265,9 @@ Torrent.prototype._onSwarmListening = function () {
   // begin discovering peers via the DHT and tracker servers
   self.discovery = new Discovery({
     announce: self.announce,
-    dht: self.client.dht,
+    dht: self.private
+      ? false
+      : self.client.dht,
     tracker: self.client.tracker,
     peerId: self.client.peerId,
     port: self.client.torrentPort,
@@ -587,8 +591,8 @@ Torrent.prototype._onWire = function (wire, addr) {
     wire.ut_metadata.fetch()
   }
 
-  // use ut_pex extension
-  if (typeof ut_pex === 'function') {
+  // use ut_pex extension if the torrent is not flagged as private
+  if (typeof ut_pex === 'function' && !self.private) {
     wire.use(ut_pex())
 
     // wire.ut_pex.start() // TODO two-way communication