mirror of
https://github.com/webtorrent/bittorrent-tracker.git
synced 2025-02-22 21:29:07 +00:00
split client and server into separate files
This commit is contained in:
parent
d1fc89cf1d
commit
c403e5c29f
505
client.js
Normal file
505
client.js
Normal file
@ -0,0 +1,505 @@
|
|||||||
|
module.exports = Client
|
||||||
|
|
||||||
|
var BN = require('bn.js')
|
||||||
|
var bncode = require('bncode')
|
||||||
|
var common = require('./lib/common')
|
||||||
|
var compact2string = require('compact2string')
|
||||||
|
var concat = require('concat-stream')
|
||||||
|
var debug = require('debug')('bittorrent-tracker:client')
|
||||||
|
var dgram = require('dgram')
|
||||||
|
var EventEmitter = require('events').EventEmitter
|
||||||
|
var extend = require('extend.js')
|
||||||
|
var hat = require('hat')
|
||||||
|
var http = require('http')
|
||||||
|
var inherits = require('inherits')
|
||||||
|
var querystring = require('querystring')
|
||||||
|
var url = require('url')
|
||||||
|
|
||||||
|
inherits(Client, EventEmitter)
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A Client manages tracker connections for a torrent.
|
||||||
|
*
|
||||||
|
* @param {string} peerId this peer's id
|
||||||
|
* @param {Number} port port number that the client is listening on
|
||||||
|
* @param {Object} torrent parsed torrent
|
||||||
|
* @param {Object} opts optional options
|
||||||
|
* @param {Number} opts.numWant number of peers to request
|
||||||
|
* @param {Number} opts.interval interval in ms to send announce requests to the tracker
|
||||||
|
*/
|
||||||
|
function Client (peerId, port, torrent, opts) {
|
||||||
|
|
||||||
|
var self = this
|
||||||
|
if (!(self instanceof Client)) return new Client(peerId, port, torrent, opts)
|
||||||
|
EventEmitter.call(self)
|
||||||
|
self._opts = opts || {}
|
||||||
|
|
||||||
|
// required
|
||||||
|
self._peerId = Buffer.isBuffer(peerId)
|
||||||
|
? peerId
|
||||||
|
: new Buffer(peerId, 'utf8')
|
||||||
|
self._port = port
|
||||||
|
self._infoHash = Buffer.isBuffer(torrent.infoHash)
|
||||||
|
? torrent.infoHash
|
||||||
|
: new Buffer(torrent.infoHash, 'hex')
|
||||||
|
self.torrentLength = torrent.length
|
||||||
|
|
||||||
|
// optional
|
||||||
|
self._numWant = self._opts.numWant || 80
|
||||||
|
self._intervalMs = self._opts.interval || (30 * 60 * 1000) // default: 30 minutes
|
||||||
|
|
||||||
|
debug('new tracker client for ' + self._infoHash.toString('hex'))
|
||||||
|
|
||||||
|
if (typeof torrent.announce === 'string') {
|
||||||
|
// magnet-uri returns a string if the magnet uri only contains one 'tr' parameter
|
||||||
|
torrent.announce = [torrent.announce]
|
||||||
|
}
|
||||||
|
self._trackers = torrent.announce.map(function (announceUrl) {
|
||||||
|
return new Tracker(self, announceUrl, self._opts)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Client.prototype.start = function (opts) {
|
||||||
|
var self = this
|
||||||
|
self._trackers.forEach(function (tracker) {
|
||||||
|
tracker.start(opts)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Client.prototype.stop = function (opts) {
|
||||||
|
var self = this
|
||||||
|
self._trackers.forEach(function (tracker) {
|
||||||
|
tracker.stop(opts)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Client.prototype.complete = function (opts) {
|
||||||
|
var self = this
|
||||||
|
self._trackers.forEach(function (tracker) {
|
||||||
|
tracker.complete(opts)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Client.prototype.update = function (opts) {
|
||||||
|
var self = this
|
||||||
|
self._trackers.forEach(function (tracker) {
|
||||||
|
tracker.update(opts)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Client.prototype.scrape = function (opts) {
|
||||||
|
var self = this
|
||||||
|
self._trackers.forEach(function (tracker) {
|
||||||
|
tracker.scrape(opts)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Client.prototype.setInterval = function (intervalMs) {
|
||||||
|
var self = this
|
||||||
|
self._intervalMs = intervalMs
|
||||||
|
|
||||||
|
self._trackers.forEach(function (tracker) {
|
||||||
|
tracker.setInterval(intervalMs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
inherits(Tracker, EventEmitter)
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An individual torrent tracker (used by Client)
|
||||||
|
*
|
||||||
|
* @param {Client} client parent bittorrent tracker client
|
||||||
|
* @param {string} announceUrl announce url of tracker
|
||||||
|
* @param {Object} opts optional options
|
||||||
|
*/
|
||||||
|
function Tracker (client, announceUrl, opts) {
|
||||||
|
var self = this
|
||||||
|
EventEmitter.call(self)
|
||||||
|
self._opts = opts || {}
|
||||||
|
|
||||||
|
self.client = client
|
||||||
|
|
||||||
|
debug('new tracker for ' + announceUrl)
|
||||||
|
|
||||||
|
self._announceUrl = announceUrl
|
||||||
|
self._intervalMs = self.client._intervalMs // use client interval initially
|
||||||
|
self._interval = null
|
||||||
|
|
||||||
|
if (self._announceUrl.indexOf('udp:') === 0) {
|
||||||
|
self._requestImpl = self._requestUdp
|
||||||
|
} else {
|
||||||
|
self._requestImpl = self._requestHttp
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype.start = function (opts) {
|
||||||
|
var self = this
|
||||||
|
opts = opts || {}
|
||||||
|
opts.event = 'started'
|
||||||
|
|
||||||
|
debug('sent `start` to ' + self._announceUrl)
|
||||||
|
self._request(opts)
|
||||||
|
self.setInterval(self._intervalMs) // start announcing on intervals
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype.stop = function (opts) {
|
||||||
|
var self = this
|
||||||
|
opts = opts || {}
|
||||||
|
opts.event = 'stopped'
|
||||||
|
|
||||||
|
debug('sent `stop` to ' + self._announceUrl)
|
||||||
|
self._request(opts)
|
||||||
|
self.setInterval(0) // stop announcing on intervals
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype.complete = function (opts) {
|
||||||
|
var self = this
|
||||||
|
opts = opts || {}
|
||||||
|
opts.event = 'completed'
|
||||||
|
opts.downloaded = opts.downloaded || self.torrentLength || 0
|
||||||
|
|
||||||
|
debug('sent `complete` to ' + self._announceUrl)
|
||||||
|
self._request(opts)
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype.update = function (opts) {
|
||||||
|
var self = this
|
||||||
|
opts = opts || {}
|
||||||
|
|
||||||
|
debug('sent `update` to ' + self._announceUrl)
|
||||||
|
self._request(opts)
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype.scrape = function (opts) {
|
||||||
|
var self = this
|
||||||
|
|
||||||
|
if (!self._scrapeUrl) {
|
||||||
|
var announce = 'announce'
|
||||||
|
var i = self._announceUrl.lastIndexOf('/') + 1
|
||||||
|
|
||||||
|
if (i >= 1 && self._announceUrl.slice(i, i + announce.length) === announce) {
|
||||||
|
self._scrapeUrl = self._announceUrl.slice(0, i) + 'scrape' + self._announceUrl.slice(i + announce.length)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!self._scrapeUrl) {
|
||||||
|
debug('scrape not supported by ' + self._announceUrl)
|
||||||
|
self.client.emit('error', new Error('scrape not supported for announceUrl ' + self._announceUrl))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
debug('sent `scrape` to ' + self._announceUrl)
|
||||||
|
|
||||||
|
opts = extend({
|
||||||
|
info_hash: bytewiseEncodeURIComponent(self.client._infoHash)
|
||||||
|
}, opts)
|
||||||
|
|
||||||
|
self._requestImpl(self._scrapeUrl, opts)
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype.setInterval = function (intervalMs) {
|
||||||
|
var self = this
|
||||||
|
clearInterval(self._interval)
|
||||||
|
|
||||||
|
self._intervalMs = intervalMs
|
||||||
|
if (intervalMs) {
|
||||||
|
self._interval = setInterval(self.update.bind(self), self._intervalMs)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Send an announce request to the tracker
|
||||||
|
*/
|
||||||
|
Tracker.prototype._request = function (opts) {
|
||||||
|
var self = this
|
||||||
|
opts = extend({
|
||||||
|
info_hash: bytewiseEncodeURIComponent(self.client._infoHash),
|
||||||
|
peer_id: bytewiseEncodeURIComponent(self.client._peerId),
|
||||||
|
port: self.client._port,
|
||||||
|
compact: 1,
|
||||||
|
numwant: self.client._numWant,
|
||||||
|
uploaded: 0, // default, user should provide real value
|
||||||
|
downloaded: 0 // default, user should provide real value
|
||||||
|
}, opts)
|
||||||
|
|
||||||
|
if (self.client.torrentLength !== undefined) {
|
||||||
|
opts.left = self.client.torrentLength - (opts.downloaded || 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (self._trackerId) {
|
||||||
|
opts.trackerid = self._trackerId
|
||||||
|
}
|
||||||
|
|
||||||
|
self._requestImpl(self._announceUrl, opts)
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype._requestHttp = function (requestUrl, opts) {
|
||||||
|
var self = this
|
||||||
|
var fullUrl = requestUrl + '?' + querystring.stringify(opts)
|
||||||
|
|
||||||
|
var req = http.get(fullUrl, function (res) {
|
||||||
|
if (res.statusCode !== 200) {
|
||||||
|
res.resume() // consume the whole stream
|
||||||
|
self.client.emit('error', new Error('Invalid response code ' + res.statusCode + ' from tracker ' + requestUrl))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
res.pipe(concat(function (data) {
|
||||||
|
if (data && data.length) self._handleResponse(requestUrl, data)
|
||||||
|
}))
|
||||||
|
})
|
||||||
|
|
||||||
|
req.on('error', function (err) {
|
||||||
|
self.client.emit('error', err)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype._requestUdp = function (requestUrl, opts) {
|
||||||
|
var self = this
|
||||||
|
var parsedUrl = url.parse(requestUrl)
|
||||||
|
var socket = dgram.createSocket('udp4')
|
||||||
|
var transactionId = new Buffer(hat(32), 'hex')
|
||||||
|
|
||||||
|
var stopped = opts.event === 'stopped'
|
||||||
|
// if we're sending a stopped message, we don't really care if it arrives, so set
|
||||||
|
// a short timer and don't call error
|
||||||
|
var timeout = setTimeout(function () {
|
||||||
|
timeout = null
|
||||||
|
cleanup()
|
||||||
|
if (!stopped) {
|
||||||
|
error('tracker request timed out')
|
||||||
|
}
|
||||||
|
}, stopped ? 1500 : 15000)
|
||||||
|
|
||||||
|
if (timeout && timeout.unref) {
|
||||||
|
timeout.unref()
|
||||||
|
}
|
||||||
|
|
||||||
|
send(Buffer.concat([
|
||||||
|
common.CONNECTION_ID,
|
||||||
|
common.toUInt32(common.ACTIONS.CONNECT),
|
||||||
|
transactionId
|
||||||
|
]))
|
||||||
|
|
||||||
|
socket.on('error', error)
|
||||||
|
|
||||||
|
socket.on('message', function (msg, rinfo) {
|
||||||
|
if (msg.length < 8 || msg.readUInt32BE(4) !== transactionId.readUInt32BE(0)) {
|
||||||
|
return error('tracker sent back invalid transaction id')
|
||||||
|
}
|
||||||
|
|
||||||
|
var action = msg.readUInt32BE(0)
|
||||||
|
switch (action) {
|
||||||
|
case 0: // handshake
|
||||||
|
if (msg.length < 16) {
|
||||||
|
return error('invalid udp handshake')
|
||||||
|
}
|
||||||
|
|
||||||
|
var scrapeStr = 'scrape'
|
||||||
|
if (requestUrl.substr(requestUrl.lastIndexOf('/') + 1, scrapeStr.length) === scrapeStr) {
|
||||||
|
scrape(msg.slice(8, 16), opts)
|
||||||
|
} else {
|
||||||
|
announce(msg.slice(8, 16), opts)
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
|
||||||
|
case 1: // announce
|
||||||
|
cleanup()
|
||||||
|
if (msg.length < 20) {
|
||||||
|
return error('invalid announce message')
|
||||||
|
}
|
||||||
|
|
||||||
|
var interval = msg.readUInt32BE(8)
|
||||||
|
if (interval && !self._opts.interval && self._intervalMs !== 0) {
|
||||||
|
// use the interval the tracker recommends, UNLESS the user manually specifies an
|
||||||
|
// interval they want to use
|
||||||
|
self.setInterval(interval * 1000)
|
||||||
|
}
|
||||||
|
|
||||||
|
self.client.emit('update', {
|
||||||
|
announce: self._announceUrl,
|
||||||
|
complete: msg.readUInt32BE(16),
|
||||||
|
incomplete: msg.readUInt32BE(12)
|
||||||
|
})
|
||||||
|
|
||||||
|
compact2string.multi(msg.slice(20)).forEach(function (addr) {
|
||||||
|
self.client.emit('peer', addr)
|
||||||
|
})
|
||||||
|
break
|
||||||
|
|
||||||
|
case 2: // scrape
|
||||||
|
cleanup()
|
||||||
|
if (msg.length < 20) {
|
||||||
|
return error('invalid scrape message')
|
||||||
|
}
|
||||||
|
self.client.emit('scrape', {
|
||||||
|
announce: self._announceUrl,
|
||||||
|
complete: msg.readUInt32BE(8),
|
||||||
|
downloaded: msg.readUInt32BE(12),
|
||||||
|
incomplete: msg.readUInt32BE(16)
|
||||||
|
})
|
||||||
|
break
|
||||||
|
|
||||||
|
case 3: // error
|
||||||
|
cleanup()
|
||||||
|
if (msg.length < 8) {
|
||||||
|
return error('invalid error message')
|
||||||
|
}
|
||||||
|
self.client.emit('error', new Error(msg.slice(8).toString()))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
function send (message) {
|
||||||
|
if (!parsedUrl.port) {
|
||||||
|
parsedUrl.port = 80
|
||||||
|
}
|
||||||
|
socket.send(message, 0, message.length, parsedUrl.port, parsedUrl.hostname)
|
||||||
|
}
|
||||||
|
|
||||||
|
function error (message) {
|
||||||
|
self.client.emit('error', new Error(message + ' (connecting to tracker ' + requestUrl + ')'))
|
||||||
|
cleanup()
|
||||||
|
}
|
||||||
|
|
||||||
|
function cleanup () {
|
||||||
|
if (timeout) {
|
||||||
|
clearTimeout(timeout)
|
||||||
|
timeout = null
|
||||||
|
}
|
||||||
|
try { socket.close() } catch (err) {}
|
||||||
|
}
|
||||||
|
|
||||||
|
function genTransactionId () {
|
||||||
|
transactionId = new Buffer(hat(32), 'hex')
|
||||||
|
}
|
||||||
|
|
||||||
|
function announce (connectionId, opts) {
|
||||||
|
opts = opts || {}
|
||||||
|
genTransactionId()
|
||||||
|
|
||||||
|
send(Buffer.concat([
|
||||||
|
connectionId,
|
||||||
|
common.toUInt32(common.ACTIONS.ANNOUNCE),
|
||||||
|
transactionId,
|
||||||
|
self.client._infoHash,
|
||||||
|
self.client._peerId,
|
||||||
|
toUInt64(opts.downloaded || 0),
|
||||||
|
opts.left ? toUInt64(opts.left) : new Buffer('FFFFFFFFFFFFFFFF', 'hex'),
|
||||||
|
toUInt64(opts.uploaded || 0),
|
||||||
|
common.toUInt32(common.EVENTS[opts.event] || 0),
|
||||||
|
common.toUInt32(0), // ip address (optional)
|
||||||
|
common.toUInt32(0), // key (optional)
|
||||||
|
common.toUInt32(self.client._numWant),
|
||||||
|
toUInt16(self.client._port || 0)
|
||||||
|
]))
|
||||||
|
}
|
||||||
|
|
||||||
|
function scrape (connectionId, opts) {
|
||||||
|
genTransactionId()
|
||||||
|
|
||||||
|
send(Buffer.concat([
|
||||||
|
connectionId,
|
||||||
|
common.toUInt32(common.ACTIONS.SCRAPE),
|
||||||
|
transactionId,
|
||||||
|
self.client._infoHash
|
||||||
|
]))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Tracker.prototype._handleResponse = function (requestUrl, data) {
|
||||||
|
var self = this
|
||||||
|
|
||||||
|
try {
|
||||||
|
data = bncode.decode(data)
|
||||||
|
} catch (err) {
|
||||||
|
return self.client.emit('error', new Error('Error decoding tracker response: ' + err.message))
|
||||||
|
}
|
||||||
|
var failure = data['failure reason']
|
||||||
|
if (failure) {
|
||||||
|
return self.client.emit('error', new Error(failure))
|
||||||
|
}
|
||||||
|
|
||||||
|
var warning = data['warning message']
|
||||||
|
if (warning) {
|
||||||
|
self.client.emit('warning', warning)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (requestUrl === self._announceUrl) {
|
||||||
|
var interval = data.interval || data['min interval']
|
||||||
|
if (interval && !self._opts.interval && self._intervalMs !== 0) {
|
||||||
|
// use the interval the tracker recommends, UNLESS the user manually specifies an
|
||||||
|
// interval they want to use
|
||||||
|
self.setInterval(interval * 1000)
|
||||||
|
}
|
||||||
|
|
||||||
|
var trackerId = data['tracker id']
|
||||||
|
if (trackerId) {
|
||||||
|
// If absent, do not discard previous trackerId value
|
||||||
|
self._trackerId = trackerId
|
||||||
|
}
|
||||||
|
|
||||||
|
self.client.emit('update', {
|
||||||
|
announce: self._announceUrl,
|
||||||
|
complete: data.complete,
|
||||||
|
incomplete: data.incomplete
|
||||||
|
})
|
||||||
|
|
||||||
|
if (Buffer.isBuffer(data.peers)) {
|
||||||
|
// tracker returned compact response
|
||||||
|
compact2string.multi(data.peers).forEach(function (addr) {
|
||||||
|
self.client.emit('peer', addr)
|
||||||
|
})
|
||||||
|
} else if (Array.isArray(data.peers)) {
|
||||||
|
// tracker returned normal response
|
||||||
|
data.peers.forEach(function (peer) {
|
||||||
|
var ip = peer.ip
|
||||||
|
self.client.emit('peer', ip[0] + '.' + ip[1] + '.' + ip[2] + '.' + ip[3] + ':' + peer.port)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} else if (requestUrl === self._scrapeUrl) {
|
||||||
|
// NOTE: the unofficial spec says to use the 'files' key but i've seen 'host' in practice
|
||||||
|
data = data.files || data.host || {}
|
||||||
|
data = data[bytewiseEncodeURIComponent(self.client._infoHash)]
|
||||||
|
|
||||||
|
if (!data) {
|
||||||
|
self.client.emit('error', new Error('invalid scrape response'))
|
||||||
|
} else {
|
||||||
|
// TODO: optionally handle data.flags.min_request_interval (separate from announce interval)
|
||||||
|
self.client.emit('scrape', {
|
||||||
|
announce: self._announceUrl,
|
||||||
|
complete: data.complete,
|
||||||
|
incomplete: data.incomplete,
|
||||||
|
downloaded: data.downloaded
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// HELPERS
|
||||||
|
//
|
||||||
|
|
||||||
|
function toUInt16 (n) {
|
||||||
|
var buf = new Buffer(2)
|
||||||
|
buf.writeUInt16BE(n, 0)
|
||||||
|
return buf
|
||||||
|
}
|
||||||
|
|
||||||
|
var MAX_UINT = 4294967295
|
||||||
|
|
||||||
|
function toUInt64 (n) {
|
||||||
|
if (n > MAX_UINT || typeof n === 'string') {
|
||||||
|
var bytes = new BN(n).toArray()
|
||||||
|
while (bytes.length < 8) {
|
||||||
|
bytes.unshift(0)
|
||||||
|
}
|
||||||
|
return new Buffer(bytes)
|
||||||
|
}
|
||||||
|
return Buffer.concat([common.toUInt32(0), common.toUInt32(n)])
|
||||||
|
}
|
||||||
|
|
||||||
|
function bytewiseEncodeURIComponent (buf) {
|
||||||
|
return encodeURIComponent(buf.toString('binary'))
|
||||||
|
}
|
968
index.js
968
index.js
@ -1,966 +1,2 @@
|
|||||||
exports.Client = Client
|
exports.Client = require('./client')
|
||||||
exports.Server = Server
|
exports.Server = require('./server')
|
||||||
|
|
||||||
var BN = require('bn.js')
|
|
||||||
var bncode = require('bncode')
|
|
||||||
var bufferEqual = require('buffer-equal')
|
|
||||||
var compact2string = require('compact2string')
|
|
||||||
var concat = require('concat-stream')
|
|
||||||
var debug = require('debug')('bittorrent-tracker')
|
|
||||||
var dgram = require('dgram')
|
|
||||||
var EventEmitter = require('events').EventEmitter
|
|
||||||
var extend = require('extend.js')
|
|
||||||
var hat = require('hat')
|
|
||||||
var http = require('http')
|
|
||||||
var inherits = require('inherits')
|
|
||||||
var ipLib = require('ip')
|
|
||||||
var parallel = require('run-parallel')
|
|
||||||
var querystring = require('querystring')
|
|
||||||
var string2compact = require('string2compact')
|
|
||||||
var url = require('url')
|
|
||||||
|
|
||||||
var CONNECTION_ID = Buffer.concat([ toUInt32(0x417), toUInt32(0x27101980) ])
|
|
||||||
var ACTIONS = { CONNECT: 0, ANNOUNCE: 1, SCRAPE: 2, ERROR: 3 }
|
|
||||||
var EVENTS = { update: 0, completed: 1, started: 2, stopped: 3 }
|
|
||||||
var MAX_UINT = 4294967295
|
|
||||||
|
|
||||||
var REMOVE_IPV6_RE = /^::ffff:/
|
|
||||||
|
|
||||||
inherits(Tracker, EventEmitter)
|
|
||||||
|
|
||||||
/**
|
|
||||||
* An individual torrent tracker (used by Client)
|
|
||||||
*
|
|
||||||
* @param {Client} client parent bittorrent tracker client
|
|
||||||
* @param {string} announceUrl announce url of tracker
|
|
||||||
* @param {Object} opts optional options
|
|
||||||
*/
|
|
||||||
function Tracker (client, announceUrl, opts) {
|
|
||||||
var self = this
|
|
||||||
EventEmitter.call(self)
|
|
||||||
self._opts = opts || {}
|
|
||||||
|
|
||||||
self.client = client
|
|
||||||
|
|
||||||
debug('new tracker for ' + announceUrl)
|
|
||||||
|
|
||||||
self._announceUrl = announceUrl
|
|
||||||
self._intervalMs = self.client._intervalMs // use client interval initially
|
|
||||||
self._interval = null
|
|
||||||
|
|
||||||
if (self._announceUrl.indexOf('udp:') === 0) {
|
|
||||||
self._requestImpl = self._requestUdp
|
|
||||||
} else {
|
|
||||||
self._requestImpl = self._requestHttp
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype.start = function (opts) {
|
|
||||||
var self = this
|
|
||||||
opts = opts || {}
|
|
||||||
opts.event = 'started'
|
|
||||||
|
|
||||||
debug('sent `start` to ' + self._announceUrl)
|
|
||||||
self._request(opts)
|
|
||||||
self.setInterval(self._intervalMs) // start announcing on intervals
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype.stop = function (opts) {
|
|
||||||
var self = this
|
|
||||||
opts = opts || {}
|
|
||||||
opts.event = 'stopped'
|
|
||||||
|
|
||||||
debug('sent `stop` to ' + self._announceUrl)
|
|
||||||
self._request(opts)
|
|
||||||
self.setInterval(0) // stop announcing on intervals
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype.complete = function (opts) {
|
|
||||||
var self = this
|
|
||||||
opts = opts || {}
|
|
||||||
opts.event = 'completed'
|
|
||||||
opts.downloaded = opts.downloaded || self.torrentLength || 0
|
|
||||||
|
|
||||||
debug('sent `complete` to ' + self._announceUrl)
|
|
||||||
self._request(opts)
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype.update = function (opts) {
|
|
||||||
var self = this
|
|
||||||
opts = opts || {}
|
|
||||||
|
|
||||||
debug('sent `update` to ' + self._announceUrl)
|
|
||||||
self._request(opts)
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype.scrape = function (opts) {
|
|
||||||
var self = this
|
|
||||||
|
|
||||||
if (!self._scrapeUrl) {
|
|
||||||
var announce = 'announce'
|
|
||||||
var i = self._announceUrl.lastIndexOf('/') + 1
|
|
||||||
|
|
||||||
if (i >= 1 && self._announceUrl.slice(i, i + announce.length) === announce) {
|
|
||||||
self._scrapeUrl = self._announceUrl.slice(0, i) + 'scrape' + self._announceUrl.slice(i + announce.length)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!self._scrapeUrl) {
|
|
||||||
debug('scrape not supported by ' + self._announceUrl)
|
|
||||||
self.client.emit('error', new Error('scrape not supported for announceUrl ' + self._announceUrl))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
debug('sent `scrape` to ' + self._announceUrl)
|
|
||||||
|
|
||||||
opts = extend({
|
|
||||||
info_hash: bytewiseEncodeURIComponent(self.client._infoHash)
|
|
||||||
}, opts)
|
|
||||||
|
|
||||||
self._requestImpl(self._scrapeUrl, opts)
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype.setInterval = function (intervalMs) {
|
|
||||||
var self = this
|
|
||||||
clearInterval(self._interval)
|
|
||||||
|
|
||||||
self._intervalMs = intervalMs
|
|
||||||
if (intervalMs) {
|
|
||||||
self._interval = setInterval(self.update.bind(self), self._intervalMs)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Send an announce request to the tracker
|
|
||||||
*/
|
|
||||||
Tracker.prototype._request = function (opts) {
|
|
||||||
var self = this
|
|
||||||
opts = extend({
|
|
||||||
info_hash: bytewiseEncodeURIComponent(self.client._infoHash),
|
|
||||||
peer_id: bytewiseEncodeURIComponent(self.client._peerId),
|
|
||||||
port: self.client._port,
|
|
||||||
compact: 1,
|
|
||||||
numwant: self.client._numWant,
|
|
||||||
uploaded: 0, // default, user should provide real value
|
|
||||||
downloaded: 0 // default, user should provide real value
|
|
||||||
}, opts)
|
|
||||||
|
|
||||||
if (self.client.torrentLength !== undefined) {
|
|
||||||
opts.left = self.client.torrentLength - (opts.downloaded || 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
if (self._trackerId) {
|
|
||||||
opts.trackerid = self._trackerId
|
|
||||||
}
|
|
||||||
|
|
||||||
self._requestImpl(self._announceUrl, opts)
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype._requestHttp = function (requestUrl, opts) {
|
|
||||||
var self = this
|
|
||||||
var fullUrl = requestUrl + '?' + querystring.stringify(opts)
|
|
||||||
|
|
||||||
var req = http.get(fullUrl, function (res) {
|
|
||||||
if (res.statusCode !== 200) {
|
|
||||||
res.resume() // consume the whole stream
|
|
||||||
self.client.emit('error', new Error('Invalid response code ' + res.statusCode + ' from tracker ' + requestUrl))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
res.pipe(concat(function (data) {
|
|
||||||
if (data && data.length) self._handleResponse(requestUrl, data)
|
|
||||||
}))
|
|
||||||
})
|
|
||||||
|
|
||||||
req.on('error', function (err) {
|
|
||||||
self.client.emit('error', err)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype._requestUdp = function (requestUrl, opts) {
|
|
||||||
var self = this
|
|
||||||
var parsedUrl = url.parse(requestUrl)
|
|
||||||
var socket = dgram.createSocket('udp4')
|
|
||||||
var transactionId = new Buffer(hat(32), 'hex')
|
|
||||||
|
|
||||||
var stopped = opts.event === 'stopped'
|
|
||||||
// if we're sending a stopped message, we don't really care if it arrives, so set
|
|
||||||
// a short timer and don't call error
|
|
||||||
var timeout = setTimeout(function () {
|
|
||||||
timeout = null
|
|
||||||
cleanup()
|
|
||||||
if (!stopped) {
|
|
||||||
error('tracker request timed out')
|
|
||||||
}
|
|
||||||
}, stopped ? 1500 : 15000)
|
|
||||||
|
|
||||||
if (timeout && timeout.unref) {
|
|
||||||
timeout.unref()
|
|
||||||
}
|
|
||||||
|
|
||||||
send(Buffer.concat([
|
|
||||||
CONNECTION_ID,
|
|
||||||
toUInt32(ACTIONS.CONNECT),
|
|
||||||
transactionId
|
|
||||||
]))
|
|
||||||
|
|
||||||
socket.on('error', error)
|
|
||||||
|
|
||||||
socket.on('message', function (msg, rinfo) {
|
|
||||||
if (msg.length < 8 || msg.readUInt32BE(4) !== transactionId.readUInt32BE(0)) {
|
|
||||||
return error('tracker sent back invalid transaction id')
|
|
||||||
}
|
|
||||||
|
|
||||||
var action = msg.readUInt32BE(0)
|
|
||||||
switch (action) {
|
|
||||||
case 0: // handshake
|
|
||||||
if (msg.length < 16) {
|
|
||||||
return error('invalid udp handshake')
|
|
||||||
}
|
|
||||||
|
|
||||||
var scrapeStr = 'scrape'
|
|
||||||
if (requestUrl.substr(requestUrl.lastIndexOf('/') + 1, scrapeStr.length) === scrapeStr) {
|
|
||||||
scrape(msg.slice(8, 16), opts)
|
|
||||||
} else {
|
|
||||||
announce(msg.slice(8, 16), opts)
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
|
|
||||||
case 1: // announce
|
|
||||||
cleanup()
|
|
||||||
if (msg.length < 20) {
|
|
||||||
return error('invalid announce message')
|
|
||||||
}
|
|
||||||
|
|
||||||
var interval = msg.readUInt32BE(8)
|
|
||||||
if (interval && !self._opts.interval && self._intervalMs !== 0) {
|
|
||||||
// use the interval the tracker recommends, UNLESS the user manually specifies an
|
|
||||||
// interval they want to use
|
|
||||||
self.setInterval(interval * 1000)
|
|
||||||
}
|
|
||||||
|
|
||||||
self.client.emit('update', {
|
|
||||||
announce: self._announceUrl,
|
|
||||||
complete: msg.readUInt32BE(16),
|
|
||||||
incomplete: msg.readUInt32BE(12)
|
|
||||||
})
|
|
||||||
|
|
||||||
compact2string.multi(msg.slice(20)).forEach(function (addr) {
|
|
||||||
self.client.emit('peer', addr)
|
|
||||||
})
|
|
||||||
break
|
|
||||||
|
|
||||||
case 2: // scrape
|
|
||||||
cleanup()
|
|
||||||
if (msg.length < 20) {
|
|
||||||
return error('invalid scrape message')
|
|
||||||
}
|
|
||||||
self.client.emit('scrape', {
|
|
||||||
announce: self._announceUrl,
|
|
||||||
complete: msg.readUInt32BE(8),
|
|
||||||
downloaded: msg.readUInt32BE(12),
|
|
||||||
incomplete: msg.readUInt32BE(16)
|
|
||||||
})
|
|
||||||
break
|
|
||||||
|
|
||||||
case 3: // error
|
|
||||||
cleanup()
|
|
||||||
if (msg.length < 8) {
|
|
||||||
return error('invalid error message')
|
|
||||||
}
|
|
||||||
self.client.emit('error', new Error(msg.slice(8).toString()))
|
|
||||||
break
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
function send (message) {
|
|
||||||
if (!parsedUrl.port) {
|
|
||||||
parsedUrl.port = 80
|
|
||||||
}
|
|
||||||
socket.send(message, 0, message.length, parsedUrl.port, parsedUrl.hostname)
|
|
||||||
}
|
|
||||||
|
|
||||||
function error (message) {
|
|
||||||
self.client.emit('error', new Error(message + ' (connecting to tracker ' + requestUrl + ')'))
|
|
||||||
cleanup()
|
|
||||||
}
|
|
||||||
|
|
||||||
function cleanup () {
|
|
||||||
if (timeout) {
|
|
||||||
clearTimeout(timeout)
|
|
||||||
timeout = null
|
|
||||||
}
|
|
||||||
try { socket.close() } catch (err) {}
|
|
||||||
}
|
|
||||||
|
|
||||||
function genTransactionId () {
|
|
||||||
transactionId = new Buffer(hat(32), 'hex')
|
|
||||||
}
|
|
||||||
|
|
||||||
function announce (connectionId, opts) {
|
|
||||||
opts = opts || {}
|
|
||||||
genTransactionId()
|
|
||||||
|
|
||||||
send(Buffer.concat([
|
|
||||||
connectionId,
|
|
||||||
toUInt32(ACTIONS.ANNOUNCE),
|
|
||||||
transactionId,
|
|
||||||
self.client._infoHash,
|
|
||||||
self.client._peerId,
|
|
||||||
toUInt64(opts.downloaded || 0),
|
|
||||||
opts.left ? toUInt64(opts.left) : new Buffer('FFFFFFFFFFFFFFFF', 'hex'),
|
|
||||||
toUInt64(opts.uploaded || 0),
|
|
||||||
toUInt32(EVENTS[opts.event] || 0),
|
|
||||||
toUInt32(0), // ip address (optional)
|
|
||||||
toUInt32(0), // key (optional)
|
|
||||||
toUInt32(self.client._numWant),
|
|
||||||
toUInt16(self.client._port || 0)
|
|
||||||
]))
|
|
||||||
}
|
|
||||||
|
|
||||||
function scrape (connectionId, opts) {
|
|
||||||
genTransactionId()
|
|
||||||
|
|
||||||
send(Buffer.concat([
|
|
||||||
connectionId,
|
|
||||||
toUInt32(ACTIONS.SCRAPE),
|
|
||||||
transactionId,
|
|
||||||
self.client._infoHash
|
|
||||||
]))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Tracker.prototype._handleResponse = function (requestUrl, data) {
|
|
||||||
var self = this
|
|
||||||
|
|
||||||
try {
|
|
||||||
data = bncode.decode(data)
|
|
||||||
} catch (err) {
|
|
||||||
return self.client.emit('error', new Error('Error decoding tracker response: ' + err.message))
|
|
||||||
}
|
|
||||||
var failure = data['failure reason']
|
|
||||||
if (failure) {
|
|
||||||
return self.client.emit('error', new Error(failure))
|
|
||||||
}
|
|
||||||
|
|
||||||
var warning = data['warning message']
|
|
||||||
if (warning) {
|
|
||||||
self.client.emit('warning', warning)
|
|
||||||
}
|
|
||||||
|
|
||||||
if (requestUrl === self._announceUrl) {
|
|
||||||
var interval = data.interval || data['min interval']
|
|
||||||
if (interval && !self._opts.interval && self._intervalMs !== 0) {
|
|
||||||
// use the interval the tracker recommends, UNLESS the user manually specifies an
|
|
||||||
// interval they want to use
|
|
||||||
self.setInterval(interval * 1000)
|
|
||||||
}
|
|
||||||
|
|
||||||
var trackerId = data['tracker id']
|
|
||||||
if (trackerId) {
|
|
||||||
// If absent, do not discard previous trackerId value
|
|
||||||
self._trackerId = trackerId
|
|
||||||
}
|
|
||||||
|
|
||||||
self.client.emit('update', {
|
|
||||||
announce: self._announceUrl,
|
|
||||||
complete: data.complete,
|
|
||||||
incomplete: data.incomplete
|
|
||||||
})
|
|
||||||
|
|
||||||
if (Buffer.isBuffer(data.peers)) {
|
|
||||||
// tracker returned compact response
|
|
||||||
compact2string.multi(data.peers).forEach(function (addr) {
|
|
||||||
self.client.emit('peer', addr)
|
|
||||||
})
|
|
||||||
} else if (Array.isArray(data.peers)) {
|
|
||||||
// tracker returned normal response
|
|
||||||
data.peers.forEach(function (peer) {
|
|
||||||
var ip = peer.ip
|
|
||||||
self.client.emit('peer', ip[0] + '.' + ip[1] + '.' + ip[2] + '.' + ip[3] + ':' + peer.port)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
} else if (requestUrl === self._scrapeUrl) {
|
|
||||||
// NOTE: the unofficial spec says to use the 'files' key but i've seen 'host' in practice
|
|
||||||
data = data.files || data.host || {}
|
|
||||||
data = data[bytewiseEncodeURIComponent(self.client._infoHash)]
|
|
||||||
|
|
||||||
if (!data) {
|
|
||||||
self.client.emit('error', new Error('invalid scrape response'))
|
|
||||||
} else {
|
|
||||||
// TODO: optionally handle data.flags.min_request_interval (separate from announce interval)
|
|
||||||
self.client.emit('scrape', {
|
|
||||||
announce: self._announceUrl,
|
|
||||||
complete: data.complete,
|
|
||||||
incomplete: data.incomplete,
|
|
||||||
downloaded: data.downloaded
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
inherits(Client, EventEmitter)
|
|
||||||
|
|
||||||
/**
|
|
||||||
* A Client manages tracker connections for a torrent.
|
|
||||||
*
|
|
||||||
* @param {string} peerId this peer's id
|
|
||||||
* @param {Number} port port number that the client is listening on
|
|
||||||
* @param {Object} torrent parsed torrent
|
|
||||||
* @param {Object} opts optional options
|
|
||||||
* @param {Number} opts.numWant number of peers to request
|
|
||||||
* @param {Number} opts.interval interval in ms to send announce requests to the tracker
|
|
||||||
*/
|
|
||||||
function Client (peerId, port, torrent, opts) {
|
|
||||||
|
|
||||||
var self = this
|
|
||||||
if (!(self instanceof Client)) return new Client(peerId, port, torrent, opts)
|
|
||||||
EventEmitter.call(self)
|
|
||||||
self._opts = opts || {}
|
|
||||||
|
|
||||||
// required
|
|
||||||
self._peerId = Buffer.isBuffer(peerId)
|
|
||||||
? peerId
|
|
||||||
: new Buffer(peerId, 'utf8')
|
|
||||||
self._port = port
|
|
||||||
self._infoHash = Buffer.isBuffer(torrent.infoHash)
|
|
||||||
? torrent.infoHash
|
|
||||||
: new Buffer(torrent.infoHash, 'hex')
|
|
||||||
self.torrentLength = torrent.length
|
|
||||||
|
|
||||||
// optional
|
|
||||||
self._numWant = self._opts.numWant || 80
|
|
||||||
self._intervalMs = self._opts.interval || (30 * 60 * 1000) // default: 30 minutes
|
|
||||||
|
|
||||||
debug('new tracker client for ' + self._infoHash.toString('hex'))
|
|
||||||
|
|
||||||
if (typeof torrent.announce === 'string') {
|
|
||||||
// magnet-uri returns a string if the magnet uri only contains one 'tr' parameter
|
|
||||||
torrent.announce = [torrent.announce]
|
|
||||||
}
|
|
||||||
self._trackers = torrent.announce.map(function (announceUrl) {
|
|
||||||
return new Tracker(self, announceUrl, self._opts)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Client.prototype.start = function (opts) {
|
|
||||||
var self = this
|
|
||||||
self._trackers.forEach(function (tracker) {
|
|
||||||
tracker.start(opts)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Client.prototype.stop = function (opts) {
|
|
||||||
var self = this
|
|
||||||
self._trackers.forEach(function (tracker) {
|
|
||||||
tracker.stop(opts)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Client.prototype.complete = function (opts) {
|
|
||||||
var self = this
|
|
||||||
self._trackers.forEach(function (tracker) {
|
|
||||||
tracker.complete(opts)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Client.prototype.update = function (opts) {
|
|
||||||
var self = this
|
|
||||||
self._trackers.forEach(function (tracker) {
|
|
||||||
tracker.update(opts)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Client.prototype.scrape = function (opts) {
|
|
||||||
var self = this
|
|
||||||
self._trackers.forEach(function (tracker) {
|
|
||||||
tracker.scrape(opts)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Client.prototype.setInterval = function (intervalMs) {
|
|
||||||
var self = this
|
|
||||||
self._intervalMs = intervalMs
|
|
||||||
|
|
||||||
self._trackers.forEach(function (tracker) {
|
|
||||||
tracker.setInterval(intervalMs)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
inherits(Server, EventEmitter)
|
|
||||||
|
|
||||||
/**
|
|
||||||
* A BitTorrent tracker server.
|
|
||||||
*
|
|
||||||
* A "BitTorrent tracker" is an HTTP service which responds to GET requests from
|
|
||||||
* BitTorrent clients. The requests include metrics from clients that help the tracker
|
|
||||||
* keep overall statistics about the torrent. The response includes a peer list that
|
|
||||||
* helps the client participate in the torrent.
|
|
||||||
*
|
|
||||||
* @param {Object} opts options
|
|
||||||
* @param {Number} opts.interval interval in ms that clients should announce on
|
|
||||||
* @param {Number} opts.trustProxy Trust 'x-forwarded-for' header from reverse proxy
|
|
||||||
* @param {boolean} opts.http Start an http server? (default: true)
|
|
||||||
* @param {boolean} opts.udp Start a udp server? (default: true)
|
|
||||||
*/
|
|
||||||
function Server (opts) {
|
|
||||||
var self = this
|
|
||||||
if (!(self instanceof Server)) return new Server(opts)
|
|
||||||
EventEmitter.call(self)
|
|
||||||
opts = opts || {}
|
|
||||||
|
|
||||||
self._intervalMs = opts.interval
|
|
||||||
? opts.interval / 1000
|
|
||||||
: 10 * 60 // 10 min (in secs)
|
|
||||||
|
|
||||||
self._trustProxy = !!opts.trustProxy
|
|
||||||
|
|
||||||
self.torrents = {}
|
|
||||||
|
|
||||||
// default to starting an http server unless the user explictly says no
|
|
||||||
if (opts.http !== false) {
|
|
||||||
self._httpServer = http.createServer()
|
|
||||||
self._httpServer.on('request', self._onHttpRequest.bind(self))
|
|
||||||
self._httpServer.on('error', function (err) {
|
|
||||||
self.emit('error', err)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// default to starting a udp server unless the user explicitly says no
|
|
||||||
if (opts.udp !== false) {
|
|
||||||
self._udpServer = dgram.createSocket('udp4')
|
|
||||||
self._udpServer.on('message', self._onUdpRequest.bind(self))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype.listen = function (port, onlistening) {
|
|
||||||
var self = this
|
|
||||||
var tasks = []
|
|
||||||
|
|
||||||
if (onlistening) {
|
|
||||||
self.once('listening', onlistening)
|
|
||||||
}
|
|
||||||
|
|
||||||
self._httpServer && tasks.push(function (cb) {
|
|
||||||
self._httpServer.listen(port, cb)
|
|
||||||
})
|
|
||||||
self._udpServer && tasks.push(function (cb) {
|
|
||||||
self._udpServer.bind(port, cb)
|
|
||||||
})
|
|
||||||
|
|
||||||
parallel(tasks, function (err) {
|
|
||||||
if (err) return self.emit('error', err)
|
|
||||||
self.emit('listening')
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype.close = function (cb) {
|
|
||||||
var self = this
|
|
||||||
cb = cb || function () {}
|
|
||||||
if (self._udpServer) {
|
|
||||||
self._udpServer.close()
|
|
||||||
}
|
|
||||||
if (self._httpServer) {
|
|
||||||
self._httpServer.close(cb)
|
|
||||||
} else {
|
|
||||||
cb(null)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype._getSwarm = function (infoHash) {
|
|
||||||
var self = this
|
|
||||||
var swarm = self.torrents[infoHash]
|
|
||||||
if (!swarm) {
|
|
||||||
swarm = self.torrents[infoHash] = {
|
|
||||||
complete: 0,
|
|
||||||
incomplete: 0,
|
|
||||||
peers: {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return swarm
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype._onHttpRequest = function (req, res) {
|
|
||||||
var self = this
|
|
||||||
var warning
|
|
||||||
var s = req.url.split('?')
|
|
||||||
var params = querystring.parse(s[1])
|
|
||||||
|
|
||||||
// TODO: detect when required params are missing
|
|
||||||
// TODO: support multiple info_hash parameters as a concatenation of individual requests
|
|
||||||
var infoHash = bytewiseDecodeURIComponent(params.info_hash).toString('hex')
|
|
||||||
|
|
||||||
if (!infoHash) {
|
|
||||||
return error('bittorrent-tracker server only supports announcing one torrent at a time')
|
|
||||||
}
|
|
||||||
|
|
||||||
if (s[0] === '/announce' || s[0] === '/') {
|
|
||||||
var ip = self._trustProxy
|
|
||||||
? req.headers['x-forwarded-for'] || req.connection.remoteAddress
|
|
||||||
: req.connection.remoteAddress.replace(REMOVE_IPV6_RE, '') // force ipv4
|
|
||||||
var port = Number(params.port)
|
|
||||||
var addr = ip + ':' + port
|
|
||||||
var peerId = bytewiseDecodeURIComponent(params.peer_id).toString('utf8')
|
|
||||||
|
|
||||||
var swarm = self._getSwarm(infoHash)
|
|
||||||
var peer = swarm.peers[addr]
|
|
||||||
|
|
||||||
switch (params.event) {
|
|
||||||
case 'started':
|
|
||||||
if (peer) {
|
|
||||||
warning = 'unexpected `started` event from peer that is already in swarm'
|
|
||||||
} else {
|
|
||||||
var left = Number(params.left)
|
|
||||||
|
|
||||||
if (left === 0) {
|
|
||||||
swarm.complete += 1
|
|
||||||
} else {
|
|
||||||
swarm.incomplete += 1
|
|
||||||
}
|
|
||||||
|
|
||||||
peer = swarm.peers[addr] = {
|
|
||||||
ip: ip,
|
|
||||||
port: port,
|
|
||||||
peerId: peerId
|
|
||||||
}
|
|
||||||
self.emit('start', addr)
|
|
||||||
}
|
|
||||||
|
|
||||||
break
|
|
||||||
|
|
||||||
case 'stopped':
|
|
||||||
if (!peer) {
|
|
||||||
return error('unexpected `stopped` event from peer that is not in swarm')
|
|
||||||
}
|
|
||||||
|
|
||||||
if (peer.complete) {
|
|
||||||
swarm.complete -= 1
|
|
||||||
} else {
|
|
||||||
swarm.incomplete -= 1
|
|
||||||
}
|
|
||||||
|
|
||||||
swarm.peers[addr] = null
|
|
||||||
|
|
||||||
self.emit('stop', addr)
|
|
||||||
break
|
|
||||||
|
|
||||||
case 'completed':
|
|
||||||
if (!peer) {
|
|
||||||
return error('unexpected `completed` event from peer that is not in swarm')
|
|
||||||
}
|
|
||||||
if (peer.complete) {
|
|
||||||
warning = 'unexpected `completed` event from peer that is already marked as completed'
|
|
||||||
}
|
|
||||||
peer.complete = true
|
|
||||||
|
|
||||||
swarm.complete += 1
|
|
||||||
swarm.incomplete -= 1
|
|
||||||
|
|
||||||
self.emit('complete', addr)
|
|
||||||
break
|
|
||||||
|
|
||||||
case '': // update
|
|
||||||
case undefined:
|
|
||||||
if (!peer) {
|
|
||||||
return error('unexpected `update` event from peer that is not in swarm')
|
|
||||||
}
|
|
||||||
|
|
||||||
self.emit('update', addr)
|
|
||||||
break
|
|
||||||
|
|
||||||
default:
|
|
||||||
return error('unexpected event: ' + params.event) // early return
|
|
||||||
}
|
|
||||||
|
|
||||||
// send peers
|
|
||||||
var peers = Number(params.compact) === 1
|
|
||||||
? self._getPeersCompact(swarm)
|
|
||||||
: self._getPeers(swarm)
|
|
||||||
|
|
||||||
var response = {
|
|
||||||
complete: swarm.complete,
|
|
||||||
incomplete: swarm.incomplete,
|
|
||||||
peers: peers,
|
|
||||||
interval: self._intervalMs
|
|
||||||
}
|
|
||||||
|
|
||||||
if (warning) {
|
|
||||||
response['warning message'] = warning
|
|
||||||
}
|
|
||||||
res.end(bncode.encode(response))
|
|
||||||
|
|
||||||
} else if (s[0] === '/scrape') { // unofficial scrape message
|
|
||||||
var swarm = self._getSwarm(infoHash)
|
|
||||||
var response = { files : { } }
|
|
||||||
|
|
||||||
response.files[params.info_hash] = {
|
|
||||||
complete: swarm.complete,
|
|
||||||
incomplete: swarm.incomplete,
|
|
||||||
downloaded: swarm.complete, // TODO: this only provides a lower-bound
|
|
||||||
flags: {
|
|
||||||
min_request_interval: self._intervalMs
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
res.end(bncode.encode(response))
|
|
||||||
}
|
|
||||||
|
|
||||||
function error (message) {
|
|
||||||
res.end(bncode.encode({
|
|
||||||
'failure reason': message
|
|
||||||
}))
|
|
||||||
|
|
||||||
// even though it's an error for the client, it's just a warning for the server.
|
|
||||||
// don't crash the server because a client sent bad data :)
|
|
||||||
self.emit('warning', new Error(message))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype._onUdpRequest = function (msg, rinfo) {
|
|
||||||
var self = this
|
|
||||||
|
|
||||||
if (msg.length < 16) {
|
|
||||||
return error('received packet is too short')
|
|
||||||
}
|
|
||||||
|
|
||||||
if (rinfo.family !== 'IPv4') {
|
|
||||||
return error('udp tracker does not support IPv6')
|
|
||||||
}
|
|
||||||
|
|
||||||
var connectionId = msg.slice(0, 8) // 64-bit
|
|
||||||
var action = msg.readUInt32BE(8)
|
|
||||||
var transactionId = msg.readUInt32BE(12)
|
|
||||||
|
|
||||||
if (!bufferEqual(connectionId, CONNECTION_ID)) {
|
|
||||||
return error('received packet with invalid connection id')
|
|
||||||
}
|
|
||||||
|
|
||||||
var socket = dgram.createSocket('udp4')
|
|
||||||
|
|
||||||
if (action === ACTIONS.CONNECT) {
|
|
||||||
send(Buffer.concat([
|
|
||||||
toUInt32(ACTIONS.CONNECT),
|
|
||||||
toUInt32(transactionId),
|
|
||||||
connectionId
|
|
||||||
]))
|
|
||||||
} else if (action === ACTIONS.ANNOUNCE) {
|
|
||||||
var infoHash = msg.slice(16, 36).toString('hex') // 20 bytes
|
|
||||||
var peerId = msg.slice(36, 56).toString('utf8') // 20 bytes
|
|
||||||
var downloaded = fromUInt64(msg.slice(56, 64))
|
|
||||||
var left = fromUInt64(msg.slice(64, 72))
|
|
||||||
var uploaded = fromUInt64(msg.slice(72, 80))
|
|
||||||
var event = msg.readUInt32BE(80)
|
|
||||||
var ip = msg.readUInt32BE(84) // optional
|
|
||||||
var key = msg.readUInt32BE(88)
|
|
||||||
var numWant = msg.readUInt32BE(92) // optional
|
|
||||||
var port = msg.readUInt16BE(96) // optional
|
|
||||||
|
|
||||||
if (ip) {
|
|
||||||
ip = ipLib.toString(ip)
|
|
||||||
} else {
|
|
||||||
ip = rinfo.address
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!port) {
|
|
||||||
port = rinfo.port
|
|
||||||
}
|
|
||||||
|
|
||||||
var addr = ip + ':' + port
|
|
||||||
|
|
||||||
var swarm = self._getSwarm(infoHash)
|
|
||||||
var peer = swarm.peers[addr]
|
|
||||||
|
|
||||||
switch (event) {
|
|
||||||
case EVENTS.started:
|
|
||||||
if (peer) {
|
|
||||||
return error('unexpected `started` event from peer that is already in swarm')
|
|
||||||
}
|
|
||||||
|
|
||||||
if (left === 0) {
|
|
||||||
swarm.complete += 1
|
|
||||||
} else {
|
|
||||||
swarm.incomplete += 1
|
|
||||||
}
|
|
||||||
|
|
||||||
peer = swarm.peers[addr] = {
|
|
||||||
ip: ip,
|
|
||||||
port: port,
|
|
||||||
peerId: peerId
|
|
||||||
}
|
|
||||||
self.emit('start', addr)
|
|
||||||
|
|
||||||
break
|
|
||||||
|
|
||||||
case EVENTS.stopped:
|
|
||||||
if (!peer) {
|
|
||||||
return error('unexpected `stopped` event from peer that is not in swarm')
|
|
||||||
}
|
|
||||||
|
|
||||||
if (peer.complete) {
|
|
||||||
swarm.complete -= 1
|
|
||||||
} else {
|
|
||||||
swarm.incomplete -= 1
|
|
||||||
}
|
|
||||||
|
|
||||||
swarm.peers[addr] = null
|
|
||||||
|
|
||||||
self.emit('stop', addr)
|
|
||||||
break
|
|
||||||
|
|
||||||
case EVENTS.completed:
|
|
||||||
if (!peer) {
|
|
||||||
return error('unexpected `completed` event from peer that is not in swarm')
|
|
||||||
}
|
|
||||||
if (peer.complete) {
|
|
||||||
warning = 'unexpected `completed` event from peer that is already marked as completed'
|
|
||||||
}
|
|
||||||
peer.complete = true
|
|
||||||
|
|
||||||
swarm.complete += 1
|
|
||||||
swarm.incomplete -= 1
|
|
||||||
|
|
||||||
self.emit('complete', addr)
|
|
||||||
break
|
|
||||||
|
|
||||||
case EVENTS.update: // update
|
|
||||||
if (!peer) {
|
|
||||||
return error('unexpected `update` event from peer that is not in swarm')
|
|
||||||
}
|
|
||||||
|
|
||||||
self.emit('update', addr)
|
|
||||||
break
|
|
||||||
|
|
||||||
default:
|
|
||||||
return error('unexpected event: ' + event) // early return
|
|
||||||
}
|
|
||||||
|
|
||||||
// send peers
|
|
||||||
var peers = self._getPeersCompact(swarm)
|
|
||||||
|
|
||||||
// never send more than 70 peers or else the UDP packet will get too big
|
|
||||||
if (peers.length >= 70 * 6) {
|
|
||||||
peers = peers.slice(0, 70 * 6)
|
|
||||||
}
|
|
||||||
|
|
||||||
send(Buffer.concat([
|
|
||||||
toUInt32(ACTIONS.ANNOUNCE),
|
|
||||||
toUInt32(transactionId),
|
|
||||||
toUInt32(self._intervalMs),
|
|
||||||
toUInt32(swarm.incomplete),
|
|
||||||
toUInt32(swarm.complete),
|
|
||||||
peers
|
|
||||||
]))
|
|
||||||
|
|
||||||
} else if (action === ACTIONS.SCRAPE) { // scrape message
|
|
||||||
var infoHash = msg.slice(16, 36).toString('hex') // 20 bytes
|
|
||||||
|
|
||||||
// TODO: support multiple info_hash scrape
|
|
||||||
if (msg.length > 36) {
|
|
||||||
error('multiple info_hash scrape not supported')
|
|
||||||
}
|
|
||||||
|
|
||||||
var swarm = self._getSwarm(infoHash)
|
|
||||||
|
|
||||||
send(Buffer.concat([
|
|
||||||
toUInt32(ACTIONS.SCRAPE),
|
|
||||||
toUInt32(transactionId),
|
|
||||||
toUInt32(swarm.complete),
|
|
||||||
toUInt32(swarm.complete), // TODO: this only provides a lower-bound
|
|
||||||
toUInt32(swarm.incomplete)
|
|
||||||
]))
|
|
||||||
}
|
|
||||||
|
|
||||||
function send (buf) {
|
|
||||||
socket.send(buf, 0, buf.length, rinfo.port, rinfo.address, function () {
|
|
||||||
try { socket.close() } catch (err) {}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
function error (message) {
|
|
||||||
send(Buffer.concat([
|
|
||||||
toUInt32(ACTIONS.ERROR),
|
|
||||||
toUInt32(transactionId || 0),
|
|
||||||
new Buffer(message, 'utf8')
|
|
||||||
]))
|
|
||||||
self.emit('warning', new Error(message))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype._getPeers = function (swarm) {
|
|
||||||
var self = this
|
|
||||||
var peers = []
|
|
||||||
for (var peerId in swarm.peers) {
|
|
||||||
var peer = swarm.peers[peerId]
|
|
||||||
peers.push({
|
|
||||||
'peer id': peer.peerId,
|
|
||||||
ip: peer.ip,
|
|
||||||
port: peer.port
|
|
||||||
})
|
|
||||||
}
|
|
||||||
return peers
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.prototype._getPeersCompact = function (swarm) {
|
|
||||||
var self = this
|
|
||||||
var addrs = []
|
|
||||||
|
|
||||||
Object.keys(swarm.peers).forEach(function (peerId) {
|
|
||||||
var peer = swarm.peers[peerId]
|
|
||||||
if (peer) {
|
|
||||||
addrs.push(peer.ip + ':' + peer.port)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
return string2compact(addrs)
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
|
||||||
// HELPERS
|
|
||||||
//
|
|
||||||
|
|
||||||
function toUInt16 (n) {
|
|
||||||
var buf = new Buffer(2)
|
|
||||||
buf.writeUInt16BE(n, 0)
|
|
||||||
return buf
|
|
||||||
}
|
|
||||||
|
|
||||||
function toUInt32 (n) {
|
|
||||||
var buf = new Buffer(4)
|
|
||||||
buf.writeUInt32BE(n, 0)
|
|
||||||
return buf
|
|
||||||
}
|
|
||||||
|
|
||||||
function toUInt64 (n) {
|
|
||||||
if (n > MAX_UINT || typeof n === 'string') {
|
|
||||||
var bytes = new BN(n).toArray()
|
|
||||||
while (bytes.length < 8) {
|
|
||||||
bytes.unshift(0)
|
|
||||||
}
|
|
||||||
return new Buffer(bytes)
|
|
||||||
}
|
|
||||||
return Buffer.concat([toUInt32(0), toUInt32(n)])
|
|
||||||
}
|
|
||||||
|
|
||||||
var TWO_PWR_32 = (1 << 16) * 2
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Return the closest floating-point representation to the buffer value. Precision will be
|
|
||||||
* lost for big numbers.
|
|
||||||
*/
|
|
||||||
function fromUInt64 (buf) {
|
|
||||||
var high = buf.readUInt32BE(0) | 0 // force
|
|
||||||
var low = buf.readUInt32BE(4) | 0
|
|
||||||
var lowUnsigned = (low >= 0) ? low : TWO_PWR_32 + low
|
|
||||||
|
|
||||||
return high * TWO_PWR_32 + lowUnsigned
|
|
||||||
}
|
|
||||||
|
|
||||||
function bytewiseEncodeURIComponent (buf) {
|
|
||||||
return encodeURIComponent(buf.toString('binary'))
|
|
||||||
}
|
|
||||||
|
|
||||||
function bytewiseDecodeURIComponent (str) {
|
|
||||||
return new Buffer(decodeURIComponent(str), 'binary')
|
|
||||||
}
|
|
||||||
|
14
lib/common.js
Normal file
14
lib/common.js
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
/**
|
||||||
|
* Functions and constants needed by both the tracker client and server.
|
||||||
|
*/
|
||||||
|
|
||||||
|
function toUInt32 (n) {
|
||||||
|
var buf = new Buffer(4)
|
||||||
|
buf.writeUInt32BE(n, 0)
|
||||||
|
return buf
|
||||||
|
}
|
||||||
|
exports.toUInt32 = toUInt32
|
||||||
|
|
||||||
|
exports.CONNECTION_ID = Buffer.concat([ toUInt32(0x417), toUInt32(0x27101980) ])
|
||||||
|
exports.ACTIONS = { CONNECT: 0, ANNOUNCE: 1, SCRAPE: 2, ERROR: 3 }
|
||||||
|
exports.EVENTS = { update: 0, completed: 1, started: 2, stopped: 3 }
|
464
server.js
Normal file
464
server.js
Normal file
@ -0,0 +1,464 @@
|
|||||||
|
module.exports = Server
|
||||||
|
|
||||||
|
var bncode = require('bncode')
|
||||||
|
var bufferEqual = require('buffer-equal')
|
||||||
|
var common = require('./lib/common')
|
||||||
|
var dgram = require('dgram')
|
||||||
|
var EventEmitter = require('events').EventEmitter
|
||||||
|
var http = require('http')
|
||||||
|
var inherits = require('inherits')
|
||||||
|
var ipLib = require('ip')
|
||||||
|
var parallel = require('run-parallel')
|
||||||
|
var querystring = require('querystring')
|
||||||
|
var string2compact = require('string2compact')
|
||||||
|
|
||||||
|
|
||||||
|
var REMOVE_IPV6_RE = /^::ffff:/
|
||||||
|
|
||||||
|
inherits(Server, EventEmitter)
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A BitTorrent tracker server.
|
||||||
|
*
|
||||||
|
* A "BitTorrent tracker" is an HTTP service which responds to GET requests from
|
||||||
|
* BitTorrent clients. The requests include metrics from clients that help the tracker
|
||||||
|
* keep overall statistics about the torrent. The response includes a peer list that
|
||||||
|
* helps the client participate in the torrent.
|
||||||
|
*
|
||||||
|
* @param {Object} opts options
|
||||||
|
* @param {Number} opts.interval interval in ms that clients should announce on
|
||||||
|
* @param {Number} opts.trustProxy Trust 'x-forwarded-for' header from reverse proxy
|
||||||
|
* @param {boolean} opts.http Start an http server? (default: true)
|
||||||
|
* @param {boolean} opts.udp Start a udp server? (default: true)
|
||||||
|
*/
|
||||||
|
function Server (opts) {
|
||||||
|
var self = this
|
||||||
|
if (!(self instanceof Server)) return new Server(opts)
|
||||||
|
EventEmitter.call(self)
|
||||||
|
opts = opts || {}
|
||||||
|
|
||||||
|
self._intervalMs = opts.interval
|
||||||
|
? opts.interval / 1000
|
||||||
|
: 10 * 60 // 10 min (in secs)
|
||||||
|
|
||||||
|
self._trustProxy = !!opts.trustProxy
|
||||||
|
|
||||||
|
self.torrents = {}
|
||||||
|
|
||||||
|
// default to starting an http server unless the user explictly says no
|
||||||
|
if (opts.http !== false) {
|
||||||
|
self._httpServer = http.createServer()
|
||||||
|
self._httpServer.on('request', self._onHttpRequest.bind(self))
|
||||||
|
self._httpServer.on('error', function (err) {
|
||||||
|
self.emit('error', err)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// default to starting a udp server unless the user explicitly says no
|
||||||
|
if (opts.udp !== false) {
|
||||||
|
self._udpServer = dgram.createSocket('udp4')
|
||||||
|
self._udpServer.on('message', self._onUdpRequest.bind(self))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype.listen = function (port, onlistening) {
|
||||||
|
var self = this
|
||||||
|
var tasks = []
|
||||||
|
|
||||||
|
if (onlistening) {
|
||||||
|
self.once('listening', onlistening)
|
||||||
|
}
|
||||||
|
|
||||||
|
self._httpServer && tasks.push(function (cb) {
|
||||||
|
self._httpServer.listen(port, cb)
|
||||||
|
})
|
||||||
|
self._udpServer && tasks.push(function (cb) {
|
||||||
|
self._udpServer.bind(port, cb)
|
||||||
|
})
|
||||||
|
|
||||||
|
parallel(tasks, function (err) {
|
||||||
|
if (err) return self.emit('error', err)
|
||||||
|
self.emit('listening')
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype.close = function (cb) {
|
||||||
|
var self = this
|
||||||
|
cb = cb || function () {}
|
||||||
|
if (self._udpServer) {
|
||||||
|
self._udpServer.close()
|
||||||
|
}
|
||||||
|
if (self._httpServer) {
|
||||||
|
self._httpServer.close(cb)
|
||||||
|
} else {
|
||||||
|
cb(null)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype._getSwarm = function (infoHash) {
|
||||||
|
var self = this
|
||||||
|
var swarm = self.torrents[infoHash]
|
||||||
|
if (!swarm) {
|
||||||
|
swarm = self.torrents[infoHash] = {
|
||||||
|
complete: 0,
|
||||||
|
incomplete: 0,
|
||||||
|
peers: {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return swarm
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype._onHttpRequest = function (req, res) {
|
||||||
|
var self = this
|
||||||
|
var warning
|
||||||
|
var s = req.url.split('?')
|
||||||
|
var params = querystring.parse(s[1])
|
||||||
|
|
||||||
|
// TODO: detect when required params are missing
|
||||||
|
// TODO: support multiple info_hash parameters as a concatenation of individual requests
|
||||||
|
var infoHash = bytewiseDecodeURIComponent(params.info_hash).toString('hex')
|
||||||
|
|
||||||
|
if (!infoHash) {
|
||||||
|
return error('bittorrent-tracker server only supports announcing one torrent at a time')
|
||||||
|
}
|
||||||
|
|
||||||
|
if (s[0] === '/announce' || s[0] === '/') {
|
||||||
|
var ip = self._trustProxy
|
||||||
|
? req.headers['x-forwarded-for'] || req.connection.remoteAddress
|
||||||
|
: req.connection.remoteAddress.replace(REMOVE_IPV6_RE, '') // force ipv4
|
||||||
|
var port = Number(params.port)
|
||||||
|
var addr = ip + ':' + port
|
||||||
|
var peerId = bytewiseDecodeURIComponent(params.peer_id).toString('utf8')
|
||||||
|
|
||||||
|
var swarm = self._getSwarm(infoHash)
|
||||||
|
var peer = swarm.peers[addr]
|
||||||
|
|
||||||
|
switch (params.event) {
|
||||||
|
case 'started':
|
||||||
|
if (peer) {
|
||||||
|
warning = 'unexpected `started` event from peer that is already in swarm'
|
||||||
|
} else {
|
||||||
|
var left = Number(params.left)
|
||||||
|
|
||||||
|
if (left === 0) {
|
||||||
|
swarm.complete += 1
|
||||||
|
} else {
|
||||||
|
swarm.incomplete += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
peer = swarm.peers[addr] = {
|
||||||
|
ip: ip,
|
||||||
|
port: port,
|
||||||
|
peerId: peerId
|
||||||
|
}
|
||||||
|
self.emit('start', addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
case 'stopped':
|
||||||
|
if (!peer) {
|
||||||
|
return error('unexpected `stopped` event from peer that is not in swarm')
|
||||||
|
}
|
||||||
|
|
||||||
|
if (peer.complete) {
|
||||||
|
swarm.complete -= 1
|
||||||
|
} else {
|
||||||
|
swarm.incomplete -= 1
|
||||||
|
}
|
||||||
|
|
||||||
|
swarm.peers[addr] = null
|
||||||
|
|
||||||
|
self.emit('stop', addr)
|
||||||
|
break
|
||||||
|
|
||||||
|
case 'completed':
|
||||||
|
if (!peer) {
|
||||||
|
return error('unexpected `completed` event from peer that is not in swarm')
|
||||||
|
}
|
||||||
|
if (peer.complete) {
|
||||||
|
warning = 'unexpected `completed` event from peer that is already marked as completed'
|
||||||
|
}
|
||||||
|
peer.complete = true
|
||||||
|
|
||||||
|
swarm.complete += 1
|
||||||
|
swarm.incomplete -= 1
|
||||||
|
|
||||||
|
self.emit('complete', addr)
|
||||||
|
break
|
||||||
|
|
||||||
|
case '': // update
|
||||||
|
case undefined:
|
||||||
|
if (!peer) {
|
||||||
|
return error('unexpected `update` event from peer that is not in swarm')
|
||||||
|
}
|
||||||
|
|
||||||
|
self.emit('update', addr)
|
||||||
|
break
|
||||||
|
|
||||||
|
default:
|
||||||
|
return error('unexpected event: ' + params.event) // early return
|
||||||
|
}
|
||||||
|
|
||||||
|
// send peers
|
||||||
|
var peers = Number(params.compact) === 1
|
||||||
|
? self._getPeersCompact(swarm)
|
||||||
|
: self._getPeers(swarm)
|
||||||
|
|
||||||
|
var response = {
|
||||||
|
complete: swarm.complete,
|
||||||
|
incomplete: swarm.incomplete,
|
||||||
|
peers: peers,
|
||||||
|
interval: self._intervalMs
|
||||||
|
}
|
||||||
|
|
||||||
|
if (warning) {
|
||||||
|
response['warning message'] = warning
|
||||||
|
}
|
||||||
|
res.end(bncode.encode(response))
|
||||||
|
|
||||||
|
} else if (s[0] === '/scrape') { // unofficial scrape message
|
||||||
|
var swarm = self._getSwarm(infoHash)
|
||||||
|
var response = { files : { } }
|
||||||
|
|
||||||
|
response.files[params.info_hash] = {
|
||||||
|
complete: swarm.complete,
|
||||||
|
incomplete: swarm.incomplete,
|
||||||
|
downloaded: swarm.complete, // TODO: this only provides a lower-bound
|
||||||
|
flags: {
|
||||||
|
min_request_interval: self._intervalMs
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
res.end(bncode.encode(response))
|
||||||
|
}
|
||||||
|
|
||||||
|
function error (message) {
|
||||||
|
res.end(bncode.encode({
|
||||||
|
'failure reason': message
|
||||||
|
}))
|
||||||
|
|
||||||
|
// even though it's an error for the client, it's just a warning for the server.
|
||||||
|
// don't crash the server because a client sent bad data :)
|
||||||
|
self.emit('warning', new Error(message))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype._onUdpRequest = function (msg, rinfo) {
|
||||||
|
var self = this
|
||||||
|
|
||||||
|
if (msg.length < 16) {
|
||||||
|
return error('received packet is too short')
|
||||||
|
}
|
||||||
|
|
||||||
|
if (rinfo.family !== 'IPv4') {
|
||||||
|
return error('udp tracker does not support IPv6')
|
||||||
|
}
|
||||||
|
|
||||||
|
var connectionId = msg.slice(0, 8) // 64-bit
|
||||||
|
var action = msg.readUInt32BE(8)
|
||||||
|
var transactionId = msg.readUInt32BE(12)
|
||||||
|
|
||||||
|
if (!bufferEqual(connectionId, common.CONNECTION_ID)) {
|
||||||
|
return error('received packet with invalid connection id')
|
||||||
|
}
|
||||||
|
|
||||||
|
var socket = dgram.createSocket('udp4')
|
||||||
|
|
||||||
|
if (action === common.ACTIONS.CONNECT) {
|
||||||
|
send(Buffer.concat([
|
||||||
|
common.toUInt32(common.ACTIONS.CONNECT),
|
||||||
|
common.toUInt32(transactionId),
|
||||||
|
connectionId
|
||||||
|
]))
|
||||||
|
} else if (action === common.ACTIONS.ANNOUNCE) {
|
||||||
|
var infoHash = msg.slice(16, 36).toString('hex') // 20 bytes
|
||||||
|
var peerId = msg.slice(36, 56).toString('utf8') // 20 bytes
|
||||||
|
var downloaded = fromUInt64(msg.slice(56, 64))
|
||||||
|
var left = fromUInt64(msg.slice(64, 72))
|
||||||
|
var uploaded = fromUInt64(msg.slice(72, 80))
|
||||||
|
var event = msg.readUInt32BE(80)
|
||||||
|
var ip = msg.readUInt32BE(84) // optional
|
||||||
|
var key = msg.readUInt32BE(88)
|
||||||
|
var numWant = msg.readUInt32BE(92) // optional
|
||||||
|
var port = msg.readUInt16BE(96) // optional
|
||||||
|
|
||||||
|
if (ip) {
|
||||||
|
ip = ipLib.toString(ip)
|
||||||
|
} else {
|
||||||
|
ip = rinfo.address
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!port) {
|
||||||
|
port = rinfo.port
|
||||||
|
}
|
||||||
|
|
||||||
|
var addr = ip + ':' + port
|
||||||
|
|
||||||
|
var swarm = self._getSwarm(infoHash)
|
||||||
|
var peer = swarm.peers[addr]
|
||||||
|
|
||||||
|
switch (event) {
|
||||||
|
case common.EVENTS.started:
|
||||||
|
if (peer) {
|
||||||
|
return error('unexpected `started` event from peer that is already in swarm')
|
||||||
|
}
|
||||||
|
|
||||||
|
if (left === 0) {
|
||||||
|
swarm.complete += 1
|
||||||
|
} else {
|
||||||
|
swarm.incomplete += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
peer = swarm.peers[addr] = {
|
||||||
|
ip: ip,
|
||||||
|
port: port,
|
||||||
|
peerId: peerId
|
||||||
|
}
|
||||||
|
self.emit('start', addr)
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
case common.EVENTS.stopped:
|
||||||
|
if (!peer) {
|
||||||
|
return error('unexpected `stopped` event from peer that is not in swarm')
|
||||||
|
}
|
||||||
|
|
||||||
|
if (peer.complete) {
|
||||||
|
swarm.complete -= 1
|
||||||
|
} else {
|
||||||
|
swarm.incomplete -= 1
|
||||||
|
}
|
||||||
|
|
||||||
|
swarm.peers[addr] = null
|
||||||
|
|
||||||
|
self.emit('stop', addr)
|
||||||
|
break
|
||||||
|
|
||||||
|
case common.EVENTS.completed:
|
||||||
|
if (!peer) {
|
||||||
|
return error('unexpected `completed` event from peer that is not in swarm')
|
||||||
|
}
|
||||||
|
if (peer.complete) {
|
||||||
|
warning = 'unexpected `completed` event from peer that is already marked as completed'
|
||||||
|
}
|
||||||
|
peer.complete = true
|
||||||
|
|
||||||
|
swarm.complete += 1
|
||||||
|
swarm.incomplete -= 1
|
||||||
|
|
||||||
|
self.emit('complete', addr)
|
||||||
|
break
|
||||||
|
|
||||||
|
case common.EVENTS.update: // update
|
||||||
|
if (!peer) {
|
||||||
|
return error('unexpected `update` event from peer that is not in swarm')
|
||||||
|
}
|
||||||
|
|
||||||
|
self.emit('update', addr)
|
||||||
|
break
|
||||||
|
|
||||||
|
default:
|
||||||
|
return error('unexpected event: ' + event) // early return
|
||||||
|
}
|
||||||
|
|
||||||
|
// send peers
|
||||||
|
var peers = self._getPeersCompact(swarm)
|
||||||
|
|
||||||
|
// never send more than 70 peers or else the UDP packet will get too big
|
||||||
|
if (peers.length >= 70 * 6) {
|
||||||
|
peers = peers.slice(0, 70 * 6)
|
||||||
|
}
|
||||||
|
|
||||||
|
send(Buffer.concat([
|
||||||
|
common.toUInt32(common.ACTIONS.ANNOUNCE),
|
||||||
|
common.toUInt32(transactionId),
|
||||||
|
common.toUInt32(self._intervalMs),
|
||||||
|
common.toUInt32(swarm.incomplete),
|
||||||
|
common.toUInt32(swarm.complete),
|
||||||
|
peers
|
||||||
|
]))
|
||||||
|
|
||||||
|
} else if (action === common.ACTIONS.SCRAPE) { // scrape message
|
||||||
|
var infoHash = msg.slice(16, 36).toString('hex') // 20 bytes
|
||||||
|
|
||||||
|
// TODO: support multiple info_hash scrape
|
||||||
|
if (msg.length > 36) {
|
||||||
|
error('multiple info_hash scrape not supported')
|
||||||
|
}
|
||||||
|
|
||||||
|
var swarm = self._getSwarm(infoHash)
|
||||||
|
|
||||||
|
send(Buffer.concat([
|
||||||
|
common.toUInt32(common.ACTIONS.SCRAPE),
|
||||||
|
common.toUInt32(transactionId),
|
||||||
|
common.toUInt32(swarm.complete),
|
||||||
|
common.toUInt32(swarm.complete), // TODO: this only provides a lower-bound
|
||||||
|
common.toUInt32(swarm.incomplete)
|
||||||
|
]))
|
||||||
|
}
|
||||||
|
|
||||||
|
function send (buf) {
|
||||||
|
socket.send(buf, 0, buf.length, rinfo.port, rinfo.address, function () {
|
||||||
|
try { socket.close() } catch (err) {}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
function error (message) {
|
||||||
|
send(Buffer.concat([
|
||||||
|
common.toUInt32(common.ACTIONS.ERROR),
|
||||||
|
common.toUInt32(transactionId || 0),
|
||||||
|
new Buffer(message, 'utf8')
|
||||||
|
]))
|
||||||
|
self.emit('warning', new Error(message))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype._getPeers = function (swarm) {
|
||||||
|
var self = this
|
||||||
|
var peers = []
|
||||||
|
for (var peerId in swarm.peers) {
|
||||||
|
var peer = swarm.peers[peerId]
|
||||||
|
peers.push({
|
||||||
|
'peer id': peer.peerId,
|
||||||
|
ip: peer.ip,
|
||||||
|
port: peer.port
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return peers
|
||||||
|
}
|
||||||
|
|
||||||
|
Server.prototype._getPeersCompact = function (swarm) {
|
||||||
|
var self = this
|
||||||
|
var addrs = []
|
||||||
|
|
||||||
|
Object.keys(swarm.peers).forEach(function (peerId) {
|
||||||
|
var peer = swarm.peers[peerId]
|
||||||
|
if (peer) {
|
||||||
|
addrs.push(peer.ip + ':' + peer.port)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
return string2compact(addrs)
|
||||||
|
}
|
||||||
|
|
||||||
|
// HELPER FUNCTIONS
|
||||||
|
|
||||||
|
var TWO_PWR_32 = (1 << 16) * 2
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the closest floating-point representation to the buffer value. Precision will be
|
||||||
|
* lost for big numbers.
|
||||||
|
*/
|
||||||
|
function fromUInt64 (buf) {
|
||||||
|
var high = buf.readUInt32BE(0) | 0 // force
|
||||||
|
var low = buf.readUInt32BE(4) | 0
|
||||||
|
var lowUnsigned = (low >= 0) ? low : TWO_PWR_32 + low
|
||||||
|
|
||||||
|
return high * TWO_PWR_32 + lowUnsigned
|
||||||
|
}
|
||||||
|
|
||||||
|
function bytewiseDecodeURIComponent (str) {
|
||||||
|
return new Buffer(decodeURIComponent(str), 'binary')
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user