Add uTP support (BEP29)

1 files changed, 38 insertions, 2 deletions

diff --git a/lib/peer.js b/lib/peer.js
index 0ddf535..b0fea6d 100644
--- a/lib/peer.js
+++ b/lib/peer.js
@@ -5,6 +5,7 @@ const Wire = require('bittorrent-protocol')
 const WebConn = require('./webconn')
 
 const CONNECT_TIMEOUT_TCP = 5000
+const CONNECT_TIMEOUT_UTP = 5000
 const CONNECT_TIMEOUT_WEBRTC = 25000
 const HANDSHAKE_TIMEOUT = 25000
 
@@ -46,6 +47,22 @@ exports.createTCPIncomingPeer = conn => {
 }
 
 /**
+ * Incoming uTP peers start out connected, because the remote peer connected to the
+ * listening port of the uTP server. Until the remote peer sends a handshake, we don't
+ * know what swarm the connection is intended for.
+ */
+exports.createUTPIncomingPeer = conn => {
+  const addr = `${conn.remoteAddress}:${conn.remotePort}`
+  const peer = new Peer(addr, 'utpIncoming')
+  peer.conn = conn
+  peer.addr = addr
+
+  peer.onConnect()
+
+  return peer
+}
+
+/**
  * Outgoing TCP peers start out with just an IP address. At some point (when there is an
  * available connection), the client can attempt to connect to the address.
  */
@@ -58,6 +75,18 @@ exports.createTCPOutgoingPeer = (addr, swarm) => {
 }
 
 /**
+ * Outgoing uTP peers start out with just an IP address. At some point (when there is an
+ * available connection), the client can attempt to connect to the address.
+ */
+exports.createUTPOutgoingPeer = (addr, swarm) => {
+  const peer = new Peer(addr, 'utpOutgoing')
+  peer.addr = addr
+  peer.swarm = swarm
+
+  return peer
+}
+
+/**
  * Peer that represents a Web Seed (BEP17 / BEP19).
  */
 exports.createWebSeedPeer = (url, swarm) => {
@@ -193,9 +222,16 @@ class Peer {
 
   startConnectTimeout () {
     clearTimeout(this.connectTimeout)
+
+    const connectTimeoutValues = {
+      webrtc: CONNECT_TIMEOUT_WEBRTC,
+      tcpOutgoing: CONNECT_TIMEOUT_TCP,
+      utpOutgoing: CONNECT_TIMEOUT_UTP
+    }
+
     this.connectTimeout = setTimeout(() => {
       this.destroy(new Error('connect timeout'))
-    }, this.type === 'webrtc' ? CONNECT_TIMEOUT_WEBRTC : CONNECT_TIMEOUT_TCP)
+    }, connectTimeoutValues[this.type])
     if (this.connectTimeout.unref) this.connectTimeout.unref()
   }
 
@@ -212,7 +248,7 @@ class Peer {
     this.destroyed = true
     this.connected = false
 
-    debug('destroy %s (error: %s)', this.id, err && (err.message || err))
+    debug('destroy %s %s (error: %s)', this.type, this.id, err && (err.message || err))
 
     clearTimeout(this.connectTimeout)
     clearTimeout(this.handshakeTimeout)