Set throttle streams for tcpIncoming peers

lib/peer.js

@@ -143,18 +143,27 @@ Peer.prototype.onConnect = function () {
   })
   self.startHandshakeTimeout()
 
-  if (self.type.indexOf('tcp') !== -1) {
-    conn.pipe(self.swarm.client.throttleGroups.down.throttle())
-      .pipe(wire)
-      .pipe(self.swarm.client.throttleGroups.up.throttle())
-      .pipe(conn)
+  if (self.type === 'tcpOutgoing') {
+    self.setThrottlePipes()
   } else {
     conn.pipe(wire).pipe(conn)
   }
 
   if (self.swarm && !self.sentHandshake) self.handshake()
 }
 
+Peer.prototype.clearPipes = function () {
+  this.wire.unpipe()
+  this.conn.unpipe()
+}
+
+Peer.prototype.setThrottlePipes = function () {
+  this.conn.pipe(this.swarm.client.throttleGroups.down.throttle())
+    .pipe(this.wire)
+    .pipe(this.swarm.client.throttleGroups.up.throttle())
+    .pipe(this.conn)
+}
+
 /**
  * Called when handshake is received from remote peer.
  * @param {string} infoHash

lib/tcp-pool.js

@@ -106,6 +106,8 @@ TCPPool.prototype._onConnection = function (conn) {
     var torrent = self._client.get(infoHash)
     if (torrent) {
       peer.swarm = torrent
+      peer.clearPipes()
+      peer.setThrottlePipes()
       torrent._addIncomingPeer(peer)
       peer.onHandshake(infoHash, peerId)
     } else {