Browse Source
* rebase on top old history * fix dep * more missing deps * updated docs * impl readme api * final tweaks * more tests * revert version * remove dead codev4
Mathias Buus
7 years ago
committed by
GitHub
GitHub
13 changed files with 1410 additions and 1049 deletions
@ -1 +1,2 @@ |
|||||
node_modules |
node_modules |
||||
|
sandbox.js |
||||
|
|||||
@ -1,446 +1,384 @@ |
|||||
var udp = require('udp-request') |
const { EventEmitter } = require('events') |
||||
var KBucket = require('k-bucket') |
const peers = require('ipv4-peers') |
||||
var inherits = require('inherits') |
const dgram = require('dgram') |
||||
var events = require('events') |
const sodium = require('sodium-universal') |
||||
var peers = require('ipv4-peers') |
const KBucket = require('k-bucket') |
||||
var collect = require('stream-collector') |
const tos = require('time-ordered-set') |
||||
var sodium = require('sodium-universal') |
const collect = require('stream-collector') |
||||
var tos = require('time-ordered-set') |
const codecs = require('codecs') |
||||
var nodes = peers.idLength(32) |
const { Message, Holepunch } = require('./lib/messages') |
||||
var messages = require('./messages') |
const IO = require('./lib/io') |
||||
var queryStream = require('./query-stream') |
const QueryStream = require('./lib/query-stream') |
||||
var blake2b = require('./blake2b') |
const blake2b = require('./lib/blake2b') |
||||
|
|
||||
module.exports = DHT |
const UNSUPPORTED_COMMAND = new Error('Unsupported command') |
||||
|
const nodes = peers.idLength(32) |
||||
function DHT (opts) { |
|
||||
if (!(this instanceof DHT)) return new DHT(opts) |
exports = module.exports = opts => new DHT(opts) |
||||
|
|
||||
|
class DHT extends EventEmitter { |
||||
|
constructor (opts) { |
||||
if (!opts) opts = {} |
if (!opts) opts = {} |
||||
|
|
||||
events.EventEmitter.call(this) |
super() |
||||
|
|
||||
var self = this |
this.bootstrapped = false |
||||
|
this.destroyed = false |
||||
this.concurrency = opts.concurrency || 16 |
this.concurrency = 16 |
||||
this.id = opts.id || randomBytes(32) |
this.socket = dgram.createSocket('udp4') |
||||
this.ephemeral = !!opts.ephemeral |
this.id = randomBytes(32) |
||||
this.bucket = new KBucket({localNodeId: this.id, arbiter: arbiter}) |
|
||||
this.bucket.on('ping', onnodeping) |
|
||||
this.inflightQueries = 0 |
this.inflightQueries = 0 |
||||
|
this.ephemeral = !!opts.ephemeral |
||||
|
|
||||
this.socket = udp({ |
this.nodes = tos() |
||||
socket: opts.socket, |
this.bucket = new KBucket({ localNodeId: this.id }) |
||||
requestEncoding: messages.Request, |
this.bucket.on('ping', this._onnodeping.bind(this)) |
||||
responseEncoding: messages.Response |
this.bootstrapNodes = [].concat(opts.bootstrap || []).map(parsePeer) |
||||
}) |
|
||||
|
|
||||
this.socket.on('request', onrequest) |
this.socket.on('listening', this.emit.bind(this, 'listening')) |
||||
this.socket.on('response', onresponse) |
this.socket.on('close', this.emit.bind(this, 'close')) |
||||
this.socket.on('close', onclose) |
|
||||
|
|
||||
this.nodes = tos() |
const queryId = this.ephemeral ? null : this.id |
||||
|
const io = new IO(this.socket, queryId, this) |
||||
|
|
||||
this._bootstrap = [].concat(opts.bootstrap || []).map(parseAddr) |
this._io = io |
||||
this._queryId = this.ephemeral ? null : this.id |
this._commands = new Map() |
||||
this._bootstrapped = false |
|
||||
this._pendingRequests = [] |
|
||||
this._tick = 0 |
this._tick = 0 |
||||
this._secrets = [randomBytes(32), randomBytes(32)] |
this._tickInterval = setInterval(this._ontick.bind(this), 5000) |
||||
this._secretsInterval = setInterval(rotateSecrets, 5 * 60 * 1000) |
|
||||
this._tickInterval = setInterval(tick, 5 * 1000) |
|
||||
|
|
||||
if (opts.nodes) { |
process.nextTick(this.bootstrap.bind(this)) |
||||
for (var i = 0; i < opts.nodes.length; i++) { |
|
||||
this._addNode(opts.nodes[i].id, opts.nodes[i]) |
|
||||
} |
} |
||||
} |
|
||||
|
|
||||
process.nextTick(function () { |
|
||||
self.bootstrap() |
|
||||
}) |
|
||||
|
|
||||
function rotateSecrets () { |
_ontick () { |
||||
self._rotateSecrets() |
this._tick++ |
||||
|
if ((this._tick & 7) === 0) this._pingSome() |
||||
} |
} |
||||
|
|
||||
function onrequest (request, peer) { |
address () { |
||||
self._onrequest(request, peer) |
return this.socket.address() |
||||
} |
} |
||||
|
|
||||
function onresponse (response, peer) { |
command (name, opts) { |
||||
self._onresponse(response, peer) |
this._commands.set(name, { |
||||
|
inputEncoding: codecs(opts.inputEncoding || opts.valueEncoding), |
||||
|
outputEncoding: codecs(opts.outputEncoding || opts.valueEncoding), |
||||
|
query: opts.query || queryNotSupported, |
||||
|
update: opts.update || updateNotSupported |
||||
|
}) |
||||
} |
} |
||||
|
|
||||
function onnodeping (oldContacts, newContact) { |
ready (onready) { |
||||
self._onnodeping(oldContacts, newContact) |
if (!this.bootstrapped) this.once('ready', onready) |
||||
|
else onready() |
||||
} |
} |
||||
|
|
||||
function onclose () { |
onrequest (type, message, peer) { |
||||
while (self._pendingRequests.length) { |
if (validateId(message.id)) { |
||||
self._pendingRequests.shift().callback(new Error('Request cancelled')) |
this._addNode(message.id, peer, null) |
||||
} |
|
||||
self.emit('close') |
|
||||
} |
} |
||||
|
|
||||
function tick () { |
switch (message.command) { |
||||
self._tick++ |
case '_ping': |
||||
if ((self._tick & 7) === 0) self._pingSome() |
return this._onping(message, peer) |
||||
} |
|
||||
} |
|
||||
|
|
||||
inherits(DHT, events.EventEmitter) |
case '_find_node': |
||||
|
return this._onfindnode(message, peer) |
||||
|
|
||||
DHT.prototype.ready = function (cb) { |
case '_holepunch': |
||||
if (!this._bootstrapped) this.once('ready', cb) |
return this._onholepunch(message, peer) |
||||
else cb() |
|
||||
} |
|
||||
|
|
||||
DHT.prototype.query = function (query, opts, cb) { |
default: |
||||
if (typeof opts === 'function') return this.query(query, null, opts) |
return this._oncommand(type, message, peer) |
||||
return collect(queryStream(this, query, opts), cb) |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype.update = function (query, opts, cb) { |
|
||||
if (typeof opts === 'function') return this.update(query, null, opts) |
|
||||
if (!opts) opts = {} |
|
||||
if (opts.query) opts.verbose = true |
|
||||
opts.token = true |
|
||||
return collect(queryStream(this, query, opts), cb) |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype._pingSome = function () { |
_onping (message, peer) { |
||||
var cnt = this.inflightQueries > 2 ? 1 : 3 |
if (message.value && !this.id.equals(message.value)) return |
||||
var oldest = this.nodes.oldest |
this._io.response(message, peers.encode([ peer ]), null, peer) |
||||
|
|
||||
while (cnt--) { |
|
||||
if (!oldest || this._tick - oldest.tick < 3) continue |
|
||||
this._check(oldest) |
|
||||
oldest = oldest.next |
|
||||
} |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype.holepunch = function (peer, referrer, cb) { |
_onholepunch (message, peer) { |
||||
peer = parseAddr(peer) |
const value = decodeHolepunch(message.value) |
||||
referrer = parseAddr(referrer) |
if (!value) return |
||||
|
|
||||
this._ping(peer, noop) |
|
||||
this._holepunch(peer, referrer, cb) |
|
||||
} |
|
||||
|
|
||||
DHT.prototype.ping = function (peer, cb) { |
if (value.to) { |
||||
this._ping(parseAddr(peer), function (err, res, peer) { |
const to = decodePeer(value.to) |
||||
if (err) return cb(err) |
if (!to || samePeer(to, peer)) return |
||||
var rinfo = decodePeer(res.value) |
message.id = this._io.id |
||||
if (!rinfo) return cb(new Error('Invalid pong')) |
message.value = Holepunch.encode({ from: peers.encode([ peer ]) }) |
||||
cb(null, rinfo, {port: peer.port, host: peer.host, id: res.id}) |
this.emit('holepunch', peer, to) |
||||
}) |
this._io.send(Message.encode(message), to) |
||||
|
return |
||||
} |
} |
||||
|
|
||||
DHT.prototype.toArray = function () { |
if (value.from) { |
||||
return this.bucket.toArray() |
const from = decodePeer(value.from) |
||||
|
if (from) peer = from |
||||
} |
} |
||||
|
|
||||
DHT.prototype.destroy = function () { |
this._io.response(message, null, null, peer) |
||||
clearInterval(this._secretsInterval) |
|
||||
clearInterval(this._tickInterval) |
|
||||
this.socket.destroy() |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype.address = function () { |
_onfindnode (message, peer) { |
||||
return this.socket.address() |
if (!validateId(message.target)) return |
||||
} |
|
||||
|
|
||||
DHT.prototype._rotateSecrets = function () { |
const closerNodes = nodes.encode(this.bucket.closest(message.target, 20)) |
||||
var secret = randomBytes(32) |
this._io.response(message, null, closerNodes, peer) |
||||
this._secrets[1] = this._secrets[0] |
|
||||
this._secrets[0] = secret |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype.bootstrap = function (cb) { |
_oncommand (type, message, peer) { |
||||
var self = this |
if (!message.target) return |
||||
|
|
||||
if (!this._bootstrap.length) return process.nextTick(done) |
|
||||
|
|
||||
var backgroundCon = Math.min(self.concurrency, Math.max(2, Math.floor(self.concurrency / 8))) |
|
||||
var qs = this.query({ |
|
||||
command: '_find_node', |
|
||||
target: this.id |
|
||||
}) |
|
||||
|
|
||||
qs.on('data', update) |
const self = this |
||||
qs.on('error', onerror) |
const cmd = this._commands.get(message.command) |
||||
qs.on('end', done) |
|
||||
|
|
||||
update() |
if (!cmd) return reply(UNSUPPORTED_COMMAND) |
||||
|
|
||||
function onerror (err) { |
const query = { |
||||
if (cb) cb(err) |
type, |
||||
|
command: message.command, |
||||
|
node: peer, |
||||
|
target: message.target, |
||||
|
value: cmd.inputEncoding.decode(message.value) |
||||
} |
} |
||||
|
|
||||
function done () { |
if (type === IO.UPDATE) cmd.update(query, reply) |
||||
if (!self._bootstrapped) { |
else cmd.query(query, reply) |
||||
self._bootstrapped = true |
|
||||
self.emit('ready') |
function reply (err, value) { |
||||
|
const closerNodes = nodes.encode(self.bucket.closest(message.target, 20)) |
||||
|
if (err) return self._io.error(message, err, closerNodes, peer) |
||||
|
self._io.response(message, value && cmd.outputEncoding.encode(value), closerNodes, peer) |
||||
} |
} |
||||
if (cb) cb() |
|
||||
} |
} |
||||
|
|
||||
function update () { |
onresponse (message, peer) { |
||||
qs._concurrency = self.inflightQueries === 1 ? self.concurrency : backgroundCon |
if (validateId(message.id)) { |
||||
|
this._addNode(message.id, peer, message.roundtripToken) |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
DHT.prototype._ping = function (peer, cb) { |
holepunch (peer, cb) { |
||||
this._request({command: '_ping', id: this._queryId}, peer, false, cb) |
if (!peer.referrer) throw new Error('peer.referrer is required') |
||||
|
this._io.query('_holepunch', null, null, peer, cb) |
||||
} |
} |
||||
|
|
||||
DHT.prototype._holepunch = function (peer, referrer, cb) { |
destroy () { |
||||
// Expects the caller to have already sent a message to peer to open the firewall session
|
this.destroyed = true |
||||
this._request({command: '_ping', id: this._queryId, forwardRequest: encodePeer(peer)}, referrer, false, cb) |
this._io.destroy() |
||||
|
clearInterval(this._tickInterval) |
||||
} |
} |
||||
|
|
||||
DHT.prototype._request = function (request, peer, important, cb) { |
ping (peer, cb) { |
||||
if (this.socket.inflight >= this.concurrency || this._pendingRequests.length) { |
this._io.query('_ping', null, peer.id, peer, function (err, res) { |
||||
this._pendingRequests.push({request: request, peer: peer, callback: cb}) |
if (err) return cb(err) |
||||
} else { |
if (res.error) return cb(new Error(res.error)) |
||||
this.socket.request(request, peer, cb) |
const pong = decodePeer(res.value) |
||||
} |
if (!pong) return cb(new Error('Invalid pong')) |
||||
|
cb(null, pong) |
||||
|
}) |
||||
} |
} |
||||
|
|
||||
DHT.prototype._onrequest = function (request, peer) { |
_addNode (id, peer, token) { |
||||
if (validateId(request.id)) this._addNode(request.id, peer, request.roundtripToken) |
if (id.equals(this.id)) return |
||||
|
|
||||
if (request.roundtripToken) { |
var node = this.bucket.get(id) |
||||
if (!request.roundtripToken.equals(this._token(peer, 0))) { |
const fresh = !node |
||||
if (!request.roundtripToken.equals(this._token(peer, 1))) { |
|
||||
request.roundtripToken = null |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
if (request.forwardRequest) { |
if (!node) node = {} |
||||
this._forwardRequest(request, peer) |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
if (request.forwardResponse) peer = this._forwardResponse(request, peer) |
node.id = id |
||||
|
node.port = peer.port |
||||
|
node.host = peer.host |
||||
|
if (token) node.roundtripToken = token |
||||
|
node.tick = this._tick |
||||
|
|
||||
switch (request.command) { |
if (!fresh) this.nodes.remove(node) |
||||
case '_ping': return this._onping(request, peer) |
this.nodes.add(node) |
||||
case '_find_node': return this._onfindnode(request, peer) |
this.bucket.add(node) |
||||
|
if (fresh) this.emit('add-node', node) |
||||
} |
} |
||||
|
|
||||
this._onquery(request, peer) |
_removeNode (node) { |
||||
|
this.nodes.remove(node) |
||||
|
this.bucket.remove(node.id) |
||||
|
this.emit('remove-node') |
||||
} |
} |
||||
|
|
||||
DHT.prototype._forwardResponse = function (request, peer) { |
_token (peer, i) { |
||||
if (request.command !== '_ping') return // only allow ping for now
|
return blake2b.batch([ |
||||
|
this._secrets[i], |
||||
try { |
Buffer.from(peer.host) |
||||
var from = peers.decode(request.forwardResponse)[0] |
]) |
||||
if (!from) return |
|
||||
} catch (err) { |
|
||||
return |
|
||||
} |
} |
||||
|
|
||||
from.request = true |
_onnodeping (oldContacts, newContact) { |
||||
from.tid = peer.tid |
// if bootstrapping, we've recently pinged all nodes
|
||||
|
if (!this.bootstrapped) return |
||||
|
|
||||
return from |
const reping = [] |
||||
} |
|
||||
|
|
||||
DHT.prototype._forwardRequest = function (request, peer) { |
for (var i = 0; i < oldContacts.length; i++) { |
||||
if (request.command !== '_ping') return // only allow ping forwards right now
|
const old = oldContacts[i] |
||||
|
|
||||
try { |
// check if we recently talked to this peer ...
|
||||
var to = peers.decode(request.forwardRequest)[0] |
if (this._tick === old.tick) { |
||||
if (!to) return |
this.bucket.add(oldContacts[i]) |
||||
} catch (err) { |
continue |
||||
return |
|
||||
} |
} |
||||
|
|
||||
this.emit('holepunch', peer, to) |
reping.push(old) |
||||
request.forwardRequest = null |
|
||||
request.forwardResponse = encodePeer(peer) |
|
||||
this.socket.forwardRequest(request, peer, to) |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype._onquery = function (request, peer) { |
if (reping.length) this._reping(reping, newContact) |
||||
if (!validateId(request.target)) return |
|
||||
|
|
||||
var self = this |
|
||||
var query = { |
|
||||
node: { |
|
||||
id: request.id, |
|
||||
port: peer.port, |
|
||||
host: peer.host |
|
||||
}, |
|
||||
command: request.command, |
|
||||
target: request.target, |
|
||||
value: request.value, |
|
||||
roundtripToken: request.roundtripToken |
|
||||
} |
} |
||||
|
|
||||
var method = request.roundtripToken ? 'update' : 'query' |
_check (node) { |
||||
|
const self = this |
||||
|
this.ping(node, function (err) { |
||||
|
if (err) self._removeNode(node) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
if (!this.emit(method + ':' + request.command, query, callback) && !this.emit(method, query, callback)) callback() |
_reping (oldContacts, newContact) { |
||||
|
const self = this |
||||
|
|
||||
function callback (err, value) { |
ping() |
||||
if (err) return |
|
||||
|
|
||||
var res = { |
function ping () { |
||||
id: self._queryId, |
const next = oldContacts.shift() |
||||
value: value || null, |
if (!next) return |
||||
nodes: nodes.encode(self.bucket.closest(request.target, 20)), |
self._io.queryImmediately('_ping', null, next.id, next, afterPing) |
||||
roundtripToken: self._token(peer, 0) |
|
||||
} |
} |
||||
|
|
||||
self.socket.response(res, peer) |
function afterPing (err, res, node) { |
||||
|
if (!err) return ping() |
||||
|
self._removeNode(node) |
||||
|
self.bucket.add(newContact) |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
DHT.prototype._onresponse = function (response, peer) { |
_pingSome () { |
||||
if (validateId(response.id)) this._addNode(response.id, peer, response.roundtripToken) |
var cnt = this.inflightQueries > 2 ? 1 : 3 |
||||
|
var oldest = this.nodes.oldest |
||||
|
|
||||
while (this.socket.inflight < this.concurrency && this._pendingRequests.length) { |
while (cnt--) { |
||||
var next = this._pendingRequests.shift() |
if (!oldest || this._tick === oldest.tick) continue |
||||
this.socket.request(next.request, next.peer, next.callback) |
this._check(oldest) |
||||
|
oldest = oldest.next |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
DHT.prototype._onping = function (request, peer) { |
query (command, target, value, cb) { |
||||
var res = { |
if (typeof value === 'function') return this.query(command, target, null, value) |
||||
id: this._queryId, |
return collect(this.runCommand(command, target, value, { query: true, update: false }), cb) |
||||
value: encodePeer(peer), |
|
||||
roundtripToken: this._token(peer, 0) |
|
||||
} |
} |
||||
|
|
||||
this.socket.response(res, peer) |
update (command, target, value, cb) { |
||||
|
if (typeof value === 'function') return this.update(command, target, null, value) |
||||
|
return collect(this.runCommand(command, target, value, { query: false, update: true }), cb) |
||||
} |
} |
||||
|
|
||||
DHT.prototype._onfindnode = function (request, peer) { |
queryAndUpdate (command, target, value, cb) { |
||||
if (!validateId(request.target)) return |
if (typeof value === 'function') return this.queryAndUpdate(command, target, null, value) |
||||
|
return collect(this.runCommand(command, target, value, { query: true, update: true }), cb) |
||||
|
} |
||||
|
|
||||
var res = { |
runCommand (command, target, value, opts) { |
||||
id: this._queryId, |
return new QueryStream(this, command, target, value, opts) |
||||
nodes: nodes.encode(this.bucket.closest(request.target, 20)), |
|
||||
roundtripToken: this._token(peer, 0) |
|
||||
} |
} |
||||
|
|
||||
this.socket.response(res, peer) |
listen (port, addr, cb) { |
||||
|
if (typeof port === 'function') return this.listen(0, null, port) |
||||
|
if (typeof addr === 'function') return this.listen(port, null, addr) |
||||
|
if (cb) this.once('listening', cb) |
||||
|
this.socket.bind(port, addr) |
||||
} |
} |
||||
|
|
||||
DHT.prototype._onnodeping = function (oldContacts, newContact) { |
bootstrap (cb) { |
||||
if (!this._bootstrapped) return // bootstrapping, we've recently pinged all nodes
|
const self = this |
||||
|
const backgroundCon = Math.min(this.concurrency, Math.max(2, Math.floor(this.concurrency / 8))) |
||||
|
|
||||
var reping = [] |
if (!this.bootstrapNodes.length) return process.nextTick(done) |
||||
|
|
||||
for (var i = 0; i < oldContacts.length; i++) { |
const qs = this.query('_find_node', this.id) |
||||
var old = oldContacts[i] |
|
||||
|
|
||||
if (this._tick - old.tick < 3) { // less than 10s since we talked to this peer ...
|
qs.on('data', update) |
||||
this.bucket.add(oldContacts[i]) |
qs.on('error', onerror) |
||||
continue |
qs.on('end', done) |
||||
} |
|
||||
|
|
||||
reping.push(old) |
update() |
||||
} |
|
||||
|
|
||||
if (reping.length) this._reping(reping, newContact) |
function onerror (err) { |
||||
|
if (cb) cb(err) |
||||
} |
} |
||||
|
|
||||
DHT.prototype._check = function (node) { |
function done () { |
||||
var self = this |
if (!self.bootstrapped) { |
||||
this._request({command: '_ping', id: this._queryId}, node, false, function (err) { |
self.bootstrapped = true |
||||
if (err) self._removeNode(node) |
self.emit('ready') |
||||
}) |
|
||||
} |
} |
||||
|
if (cb) cb() |
||||
DHT.prototype._reping = function (oldContacts, newContact) { |
|
||||
var self = this |
|
||||
var next = null |
|
||||
|
|
||||
ping() |
|
||||
|
|
||||
function ping () { |
|
||||
next = oldContacts.shift() |
|
||||
if (next) self._request({command: '_ping', id: self._queryId}, next, true, afterPing) |
|
||||
} |
} |
||||
|
|
||||
function afterPing (err) { |
function update () { |
||||
if (!err) return ping() |
qs._concurrency = self.inflightQueries === 1 ? self.concurrency : backgroundCon |
||||
|
|
||||
self._removeNode(next) |
|
||||
self.bucket.add(newContact) |
|
||||
} |
} |
||||
} |
} |
||||
|
|
||||
DHT.prototype._token = function (peer, i) { |
|
||||
return blake2b.batch([this._secrets[i], Buffer.from(peer.host)]) |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype._addNode = function (id, peer, token) { |
exports.QUERY = DHT.QUERY = IO.QUERY |
||||
if (id.equals(this.id)) return |
exports.UPDATE = DHT.UPDATE = IO.UPDATE |
||||
|
exports.DHT = DHT |
||||
var node = this.bucket.get(id) |
|
||||
var fresh = !node |
|
||||
|
|
||||
if (!node) node = {} |
|
||||
|
|
||||
node.id = id |
|
||||
node.port = peer.port |
|
||||
node.host = peer.host |
|
||||
node.roundtripToken = token |
|
||||
node.tick = this._tick |
|
||||
|
|
||||
if (!fresh) this.nodes.remove(node) |
function validateId (id) { |
||||
this.nodes.add(node) |
return id && id.length === 32 |
||||
|
|
||||
this.bucket.add(node) |
|
||||
if (fresh) this.emit('add-node', node) |
|
||||
} |
} |
||||
|
|
||||
DHT.prototype._removeNode = function (node) { |
function randomBytes (n) { |
||||
this.nodes.remove(node) |
const buf = Buffer.allocUnsafe(n) |
||||
this.bucket.remove(node.id) |
sodium.randombytes_buf(buf) |
||||
this.emit('remove-node', node) |
return buf |
||||
} |
} |
||||
|
|
||||
DHT.prototype.listen = function (port, cb) { |
function decodeHolepunch (buf) { |
||||
this.socket.listen(port, cb) |
try { |
||||
|
return Holepunch.decode(buf) |
||||
|
} catch (err) { |
||||
|
return null |
||||
} |
} |
||||
|
|
||||
function encodePeer (peer) { |
|
||||
return peer && peers.encode([peer]) |
|
||||
} |
} |
||||
|
|
||||
function decodePeer (buf) { |
function decodePeer (buf) { |
||||
try { |
try { |
||||
return buf && peers.decode(buf)[0] |
const p = peers.decode(buf)[0] |
||||
|
if (!p) throw new Error('No peer in buffer') |
||||
|
return p |
||||
} catch (err) { |
} catch (err) { |
||||
return null |
return null |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
function parseAddr (addr) { |
function parsePeer (peer) { |
||||
if (typeof addr === 'object' && addr) return addr |
if (typeof peer === 'object' && peer) return peer |
||||
if (typeof addr === 'number') return parseAddr(':' + addr) |
if (typeof peer === 'number') return parsePeer(':' + peer) |
||||
if (addr[0] === ':') return parseAddr('127.0.0.1' + addr) |
if (peer[0] === ':') return parsePeer('127.0.0.1' + peer) |
||||
return {port: Number(addr.split(':')[1] || 3282), host: addr.split(':')[0]} |
|
||||
} |
|
||||
|
|
||||
function validateId (id) { |
const parts = peer.split(':') |
||||
return id && id.length === 32 |
return { |
||||
|
host: parts[0], |
||||
|
port: parseInt(parts[1], 10) |
||||
|
} |
||||
} |
} |
||||
|
|
||||
function arbiter (incumbant, candidate) { |
function samePeer (a, b) { |
||||
return candidate |
return a.port === b.port && a.host === b.host |
||||
} |
} |
||||
|
|
||||
function randomBytes (n) { |
function updateNotSupported (query, cb) { |
||||
var buf = Buffer.allocUnsafe(n) |
cb(new Error('Update not supported')) |
||||
sodium.randombytes_buf(buf) |
|
||||
return buf |
|
||||
} |
} |
||||
|
|
||||
function noop () {} |
function queryNotSupported (query, cb) { |
||||
|
cb(null, null) |
||||
|
} |
||||
|
|||||
@ -0,0 +1,299 @@ |
|||||
|
const { Message, Holepunch, TYPE } = require('./messages') |
||||
|
const blake2b = require('./blake2b') |
||||
|
const peers = require('ipv4-peers') |
||||
|
const sodium = require('sodium-universal') |
||||
|
|
||||
|
const QUERY = Symbol('QUERY') |
||||
|
const UPDATE = Symbol('UPDATE') |
||||
|
|
||||
|
const ECANCELLED = new Error('Request cancelled') |
||||
|
const ETIMEDOUT = new Error('Request timed out') |
||||
|
|
||||
|
ETIMEDOUT.code = 'ETIMEDOUT' |
||||
|
ECANCELLED.code = 'ECANCELLED' |
||||
|
|
||||
|
const TRIES = 3 |
||||
|
|
||||
|
class IO { |
||||
|
constructor (socket, id, ctx) { |
||||
|
this.id = id |
||||
|
this.socket = socket |
||||
|
this.inflight = [] |
||||
|
|
||||
|
this._ctx = ctx |
||||
|
this._rid = (Math.random() * 65536) | 0 |
||||
|
this._requests = new Array(65536) |
||||
|
this._pending = [] |
||||
|
this._secrets = [ randomBytes(32), randomBytes(32) ] |
||||
|
this._tickInterval = setInterval(this._ontick.bind(this), 250) |
||||
|
this._rotateInterval = setInterval(this._onrotate.bind(this), 300000) |
||||
|
|
||||
|
socket.on('message', this._onmessage.bind(this)) |
||||
|
} |
||||
|
|
||||
|
_token (peer, i) { |
||||
|
return blake2b.batch([ |
||||
|
this._secrets[i], |
||||
|
Buffer.from(peer.host) |
||||
|
]) |
||||
|
} |
||||
|
|
||||
|
_free () { |
||||
|
const rid = this._rid++ |
||||
|
if (this._rid === 65536) this._rid = 0 |
||||
|
return rid |
||||
|
} |
||||
|
|
||||
|
/* |
||||
|
|
||||
|
R |
||||
|
/ \ |
||||
|
A B |
||||
|
|
||||
|
A sent a message to B that failed |
||||
|
|
||||
|
It could be that the message got dropped |
||||
|
or that it needs holepunching |
||||
|
|
||||
|
To retry |
||||
|
|
||||
|
resend(req, A -> B) |
||||
|
fire_and_forget({ _holepunch, to: B }, A -> R) |
||||
|
|
||||
|
R.onholepunch { to: B } => fire_and_forget({ _holepunch, from: A }, R -> B) |
||||
|
B.onholepunch { from: A } => fire_and_forget({ _holepunch }, B -> A) |
||||
|
|
||||
|
A and B is now holepunched and the session has been retried as well |
||||
|
|
||||
|
*/ |
||||
|
|
||||
|
_holepunch (req) { |
||||
|
const rid = req.message.command === '_holepunch' |
||||
|
? req.rid |
||||
|
: this._free() |
||||
|
|
||||
|
const punch = { |
||||
|
type: req.message.type, |
||||
|
rid, |
||||
|
command: '_holepunch', |
||||
|
value: Holepunch.encode({ |
||||
|
to: peers.encode([ req.peer ]) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
this.send(Message.encode(punch), req.peer.referrer) |
||||
|
} |
||||
|
|
||||
|
_retry (req) { |
||||
|
req.timeout = 4 |
||||
|
this.send(req.buffer, req.peer) |
||||
|
// if referrer is avail, try holepunching automatically
|
||||
|
if (req.peer.referrer) this._holepunch(req) |
||||
|
} |
||||
|
|
||||
|
_onmessage (buf, rinfo) { |
||||
|
const message = decodeMessage(buf) |
||||
|
if (!message) return |
||||
|
|
||||
|
const peer = { port: rinfo.port, host: rinfo.address } |
||||
|
|
||||
|
switch (message.type) { |
||||
|
case TYPE.RESPONSE: |
||||
|
this._ctx.onresponse(message, peer) |
||||
|
this._finish(message.rid, null, message, peer) |
||||
|
break |
||||
|
|
||||
|
case TYPE.QUERY: |
||||
|
this._ctx.onrequest(QUERY, message, peer) |
||||
|
break |
||||
|
|
||||
|
case TYPE.UPDATE: |
||||
|
const rt = message.roundtripToken |
||||
|
if (!rt || (!rt.equals(this._token(peer, 0)) && !rt.equals(this._token(peer, 1)))) return |
||||
|
this._ctx.onrequest(UPDATE, message, peer) |
||||
|
break |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
_finish (rid, err, val, peer) { |
||||
|
const req = this._requests[rid] |
||||
|
if (!req) return |
||||
|
|
||||
|
this._requests[rid] = undefined |
||||
|
const top = this.inflight[this.inflight.length - 1] |
||||
|
this.inflight[top.index = req.index] = top |
||||
|
this.inflight.pop() |
||||
|
|
||||
|
const type = req.message.type === TYPE.QUERY |
||||
|
? QUERY |
||||
|
: UPDATE |
||||
|
|
||||
|
req.callback(err, val, peer, req.message, req.peer, type) |
||||
|
|
||||
|
while (this._pending.length && this.inflight.length < this._ctx.concurrency) { |
||||
|
const { message, peer, callback } = this._pending.shift() |
||||
|
this._requestImmediately(message, peer, callback) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
_request (message, peer, callback) { |
||||
|
// Should we wait to send?
|
||||
|
if (this._pending.length || (this.inflight.length >= this._ctx.concurrency)) { |
||||
|
this._pending.push({ message, peer, callback }) |
||||
|
} else { |
||||
|
this._requestImmediately(message, peer, callback) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
_requestImmediately (message, peer, callback) { |
||||
|
const rid = message.rid = this._free() |
||||
|
const buffer = Message.encode(message) |
||||
|
|
||||
|
const req = { |
||||
|
rid, |
||||
|
index: this.inflight.length, |
||||
|
callback, |
||||
|
message, |
||||
|
buffer, |
||||
|
peer, |
||||
|
timeout: 4, |
||||
|
tries: 0 |
||||
|
} |
||||
|
|
||||
|
this._requests[rid] = req |
||||
|
this.inflight.push(req) |
||||
|
this.send(buffer, peer) |
||||
|
|
||||
|
// if sending a holepunch cmd, forward it right away
|
||||
|
if (message.command === '_holepunch') this._holepunch(req) |
||||
|
} |
||||
|
|
||||
|
_cancel (rid, err) { |
||||
|
this._finish(rid, err || ECANCELLED, null, null) |
||||
|
} |
||||
|
|
||||
|
_onrotate () { |
||||
|
this._secrets[1] = this._secrets[0] |
||||
|
this._secrets[0] = randomBytes(32) |
||||
|
} |
||||
|
|
||||
|
_ontick () { |
||||
|
for (var i = this.inflight.length - 1; i >= 0; i--) { |
||||
|
const req = this.inflight[i] |
||||
|
|
||||
|
if (req.timeout === 2 && ++req.tries < TRIES) { |
||||
|
this._retry(req) |
||||
|
continue |
||||
|
} |
||||
|
|
||||
|
if (--req.timeout) { |
||||
|
continue |
||||
|
} |
||||
|
|
||||
|
this._cancel(req.rid, ETIMEDOUT) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
send (buffer, peer) { |
||||
|
this.socket.send(buffer, 0, buffer.length, peer.port, peer.host) |
||||
|
} |
||||
|
|
||||
|
destroy () { |
||||
|
clearInterval(this._rotateInterval) |
||||
|
clearInterval(this._tickInterval) |
||||
|
|
||||
|
this.socket.close() |
||||
|
|
||||
|
const pending = this._pending |
||||
|
this._pending = [] |
||||
|
|
||||
|
for (const req of pending) req.callback(ECANCELLED) |
||||
|
for (const req of this.inflight) this._cancel(req.rid) |
||||
|
} |
||||
|
|
||||
|
response (request, value, closerNodes, peer) { |
||||
|
const message = { |
||||
|
type: TYPE.RESPONSE, |
||||
|
rid: request.rid, |
||||
|
id: this.id, |
||||
|
closerNodes, |
||||
|
roundtripToken: this._token(peer, 0), |
||||
|
value |
||||
|
} |
||||
|
|
||||
|
this.send(Message.encode(message), peer) |
||||
|
} |
||||
|
|
||||
|
error (request, error, closerNodes, peer) { |
||||
|
const message = { |
||||
|
type: TYPE.RESPONSE, |
||||
|
rid: request.rid, |
||||
|
id: this.id, |
||||
|
closerNodes, |
||||
|
error: error.message |
||||
|
} |
||||
|
|
||||
|
this.send(Message.encode(message), peer) |
||||
|
} |
||||
|
|
||||
|
query (command, target, value, peer, callback) { |
||||
|
if (!callback) callback = noop |
||||
|
|
||||
|
this._request({ |
||||
|
type: TYPE.QUERY, |
||||
|
rid: 0, |
||||
|
id: this.id, |
||||
|
target, |
||||
|
command, |
||||
|
value |
||||
|
}, peer, callback) |
||||
|
} |
||||
|
|
||||
|
queryImmediately (command, target, value, peer, callback) { |
||||
|
if (!callback) callback = noop |
||||
|
|
||||
|
this._requestImmediately({ |
||||
|
type: TYPE.QUERY, |
||||
|
rid: 0, |
||||
|
id: this.id, |
||||
|
target, |
||||
|
command, |
||||
|
value |
||||
|
}, peer, callback) |
||||
|
} |
||||
|
|
||||
|
update (command, target, value, peer, callback) { |
||||
|
if (!callback) callback = noop |
||||
|
|
||||
|
this._request({ |
||||
|
type: TYPE.UPDATE, |
||||
|
rid: 0, |
||||
|
id: this.id, |
||||
|
roundtripToken: peer.roundtripToken, |
||||
|
target, |
||||
|
command, |
||||
|
value |
||||
|
}, peer, callback) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
IO.QUERY = QUERY |
||||
|
IO.UPDATE = UPDATE |
||||
|
|
||||
|
module.exports = IO |
||||
|
|
||||
|
function noop () {} |
||||
|
|
||||
|
function decodeMessage (buf) { |
||||
|
try { |
||||
|
return Message.decode(buf) |
||||
|
} catch (err) { |
||||
|
return null |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
function randomBytes (n) { |
||||
|
const buf = Buffer.allocUnsafe(32) |
||||
|
sodium.randombytes_buf(buf) |
||||
|
return buf |
||||
|
} |
||||
@ -0,0 +1,264 @@ |
|||||
|
const { Readable } = require('stream') |
||||
|
const nodes = require('ipv4-peers').idLength(32) |
||||
|
const QueryTable = require('./query-table') |
||||
|
|
||||
|
const BOOTSTRAPPING = Symbol('BOOTSTRAPPING') |
||||
|
const MOVING_CLOSER = Symbol('MOVING_CLOSER') |
||||
|
const UPDATING = Symbol('UPDATING') |
||||
|
const FINALIZED = Symbol('FINALIZED') |
||||
|
|
||||
|
class QueryStream extends Readable { |
||||
|
constructor (node, command, target, value, opts) { |
||||
|
if (!opts) opts = {} |
||||
|
if (!opts.concurrency) opts.concurrency = opts.highWaterMark || node.concurrency |
||||
|
|
||||
|
super({ |
||||
|
objectMode: true, |
||||
|
highWaterMark: opts.concurrency |
||||
|
}) |
||||
|
|
||||
|
const cmd = node._commands.get(command) |
||||
|
|
||||
|
this.command = command |
||||
|
this.target = target |
||||
|
this.value = cmd ? cmd.inputEncoding.encode(value) : (value || null) |
||||
|
this.update = !!opts.update |
||||
|
this.query = !!opts.query || !opts.update |
||||
|
this.destroyed = false |
||||
|
this.inflight = 0 |
||||
|
this.responses = 0 |
||||
|
this.updates = 0 |
||||
|
this.table = opts.table || new QueryTable(node.id, target) |
||||
|
|
||||
|
node.inflightQueries++ |
||||
|
|
||||
|
this._status = opts.table ? MOVING_CLOSER : BOOTSTRAPPING |
||||
|
this._node = node |
||||
|
this._concurrency = opts.concurrency |
||||
|
this._callback = this._onresponse.bind(this) |
||||
|
this._map = identity |
||||
|
this._outputEncoding = cmd ? cmd.outputEncoding : null |
||||
|
} |
||||
|
|
||||
|
map (fn) { |
||||
|
this._map = fn |
||||
|
return this |
||||
|
} |
||||
|
|
||||
|
_onresponse (err, message, peer, request, to, type) { |
||||
|
this.inflight-- |
||||
|
|
||||
|
if (err && to.id) { |
||||
|
// Request, including retries, failed completely
|
||||
|
// Remove the "to" node.
|
||||
|
const node = this._node.bucket.get(to.id) |
||||
|
if (node) this._node._removeNode(node) |
||||
|
} |
||||
|
|
||||
|
if (this._status === FINALIZED) { |
||||
|
if (!this.inflight) { |
||||
|
if (this.destroyed) this.emit('close') |
||||
|
else this.destroy() |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
if (err) { |
||||
|
this.emit('warning', err) |
||||
|
this._readMaybe() |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
this.responses++ |
||||
|
this.table.addVerified(message, peer) |
||||
|
|
||||
|
if (this._status === MOVING_CLOSER) { |
||||
|
const candidates = decodeNodes(message.closerNodes) |
||||
|
for (var i = 0; i < candidates.length; i++) { |
||||
|
this.table.addUnverified(candidates[i], peer) |
||||
|
} |
||||
|
} else if (this._status === UPDATING) { |
||||
|
this.updates++ |
||||
|
} |
||||
|
|
||||
|
if (message.error) { |
||||
|
this.emit('warning', new Error(message.error)) |
||||
|
this._readMaybe() |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
if (!this.query && this._status === MOVING_CLOSER) { |
||||
|
this._readMaybe() |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
const value = this._outputEncoding |
||||
|
? this._decodeOutput(message.value) |
||||
|
: message.value |
||||
|
|
||||
|
const data = this._map({ |
||||
|
type, |
||||
|
node: { |
||||
|
id: message.id, |
||||
|
port: peer.port, |
||||
|
host: peer.host |
||||
|
}, |
||||
|
value |
||||
|
}) |
||||
|
|
||||
|
if (!data) { |
||||
|
this._readMaybe() |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
this.push(data) |
||||
|
} |
||||
|
|
||||
|
_decodeOutput (val) { |
||||
|
try { |
||||
|
return val && this._outputEncoding.decode(val) |
||||
|
} catch (err) { |
||||
|
return null |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
_bootstrap () { |
||||
|
const table = this.table |
||||
|
const bootstrap = this._node.bucket.closest(table.target, table.k) |
||||
|
var i = 0 |
||||
|
|
||||
|
for (; i < bootstrap.length; i++) { |
||||
|
const b = bootstrap[i] |
||||
|
const node = { id: b.id, port: b.port, host: b.host } |
||||
|
table.addUnverified(node, null) |
||||
|
} |
||||
|
|
||||
|
const bootstrapNodes = this._node.bootstrapNodes |
||||
|
if (bootstrap.length < bootstrapNodes.length) { |
||||
|
for (i = 0; i < bootstrapNodes.length; i++) { |
||||
|
this._send(bootstrapNodes[i], true, false) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
this._status = MOVING_CLOSER |
||||
|
this._moveCloser() |
||||
|
} |
||||
|
|
||||
|
_sendAll (nodes, force, sendToken) { |
||||
|
var free = Math.max(0, this._concurrency - this._node._io.inflight.length) |
||||
|
var sent = 0 |
||||
|
|
||||
|
if (!free && !this.inflight) free = 1 |
||||
|
if (!free) return 0 |
||||
|
|
||||
|
for (var i = 0; i < nodes.length; i++) { |
||||
|
if (this._send(nodes[i], force, sendToken)) { |
||||
|
if (++sent === free) break |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return sent |
||||
|
} |
||||
|
|
||||
|
_send (node, force, isUpdate) { |
||||
|
if (!force) { |
||||
|
if (node.queried) return false |
||||
|
node.queried = true |
||||
|
} |
||||
|
|
||||
|
this.inflight++ |
||||
|
const io = this._node._io |
||||
|
|
||||
|
if (isUpdate) { |
||||
|
if (!node.roundtripToken) return this._callback(new Error('Roundtrip token is required')) |
||||
|
io.update(this.command, this.target, this.value, node, this._callback) |
||||
|
} else if (this.query) { |
||||
|
io.query(this.command, this.target, this.value, node, this._callback) |
||||
|
} else { |
||||
|
io.query('_find_node', this.target, null, node, this._callback) |
||||
|
} |
||||
|
|
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
_sendUpdate () { |
||||
|
const sent = this._sendAll(this.table.closest, false, true) |
||||
|
if (sent || this.inflight) return |
||||
|
|
||||
|
this._finalize() |
||||
|
} |
||||
|
|
||||
|
_moveCloser () { |
||||
|
const table = this.table |
||||
|
const sent = this._sendAll(table.unverified, false, false) |
||||
|
if (sent || this.inflight) return |
||||
|
|
||||
|
if (this.update) { |
||||
|
for (var i = 0; i < table.closest.length; i++) { |
||||
|
table.closest[i].queried = false |
||||
|
} |
||||
|
this._status = UPDATING |
||||
|
this._sendUpdate() |
||||
|
} else { |
||||
|
this._finalize() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
_finalize () { |
||||
|
const status = this._status |
||||
|
if (status === FINALIZED) return |
||||
|
|
||||
|
this._status = FINALIZED |
||||
|
this._node.inflightQueries-- |
||||
|
|
||||
|
if (!this.responses && !this.destroyed) { |
||||
|
this.destroy(new Error('No nodes responded')) |
||||
|
} |
||||
|
if (status === UPDATING && !this.updates && !this.destroyed) { |
||||
|
this.destroy(new Error('No close nodes responded')) |
||||
|
} |
||||
|
|
||||
|
this.push(null) |
||||
|
} |
||||
|
|
||||
|
_readMaybe () { |
||||
|
if (!this.inflight || this._readableState.flowing === true) this._read() |
||||
|
} |
||||
|
|
||||
|
_read () { |
||||
|
if (this._node.destroyed) return |
||||
|
|
||||
|
switch (this._status) { |
||||
|
case BOOTSTRAPPING: return this._bootstrap() |
||||
|
case MOVING_CLOSER: return this._moveCloser() |
||||
|
case UPDATING: return this._sendUpdate() |
||||
|
case FINALIZED: return |
||||
|
} |
||||
|
|
||||
|
throw new Error('Unknown status: ' + this._status) |
||||
|
} |
||||
|
|
||||
|
destroy (err) { |
||||
|
if (this.destroyed) return |
||||
|
this.destroyed = true |
||||
|
|
||||
|
if (err) this.emit('error', err) |
||||
|
this._finalize() |
||||
|
if (!this.inflight) this.emit('close') |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
module.exports = QueryStream |
||||
|
|
||||
|
function decodeNodes (buf) { |
||||
|
if (!buf) return [] |
||||
|
try { |
||||
|
return nodes.decode(buf) |
||||
|
} catch (err) { |
||||
|
return [] |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
function identity (a) { |
||||
|
return a |
||||
|
} |
||||
@ -0,0 +1,69 @@ |
|||||
|
const xor = require('xor-distance') |
||||
|
|
||||
|
class QueryTable { |
||||
|
constructor (id, target) { |
||||
|
this.k = 20 |
||||
|
this.id = id |
||||
|
this.target = target |
||||
|
this.closest = [] |
||||
|
this.unverified = [] |
||||
|
} |
||||
|
|
||||
|
addUnverified (node, referrer) { |
||||
|
if (node.id.equals(this.id)) return |
||||
|
|
||||
|
node.distance = xor(this.target, node.id) |
||||
|
node.referrer = referrer |
||||
|
|
||||
|
insertSorted(node, this.k, this.unverified) |
||||
|
} |
||||
|
|
||||
|
addVerified (message, peer) { |
||||
|
if (!message.id || !message.roundtripToken || message.id.equals(this.id)) { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
var prev = getNode(message.id, this.unverified) |
||||
|
|
||||
|
if (!prev) { |
||||
|
prev = { |
||||
|
id: message.id, |
||||
|
host: peer.host, |
||||
|
port: peer.port, |
||||
|
distance: xor(message.id, this.target) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
prev.roundtripToken = message.roundtripToken |
||||
|
insertSorted(prev, this.k, this.closest) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
module.exports = QueryTable |
||||
|
|
||||
|
function getNode (id, list) { |
||||
|
// find id in the list.
|
||||
|
// technically this would be faster with binary search (against distance)
|
||||
|
// but this list is always small, so meh
|
||||
|
|
||||
|
for (var i = 0; i < list.length; i++) { |
||||
|
if (list[i].id.equals(id)) return list[i] |
||||
|
} |
||||
|
|
||||
|
return null |
||||
|
} |
||||
|
|
||||
|
function insertSorted (node, max, list) { |
||||
|
if (list.length === max && !xor.lt(node.distance, list[max - 1].distance)) return |
||||
|
if (getNode(node.id, list)) return |
||||
|
|
||||
|
if (list.length < max) list.push(node) |
||||
|
else list[max - 1] = node |
||||
|
|
||||
|
var pos = list.length - 1 |
||||
|
while (pos && xor.gt(list[pos - 1].distance, node.distance)) { |
||||
|
list[pos] = list[pos - 1] |
||||
|
list[pos - 1] = node |
||||
|
pos-- |
||||
|
} |
||||
|
} |
||||
@ -1,38 +1,35 @@ |
|||||
{ |
{ |
||||
"name": "dht-rpc", |
"name": "dht-rpc", |
||||
"version": "3.0.1", |
"version": "3.0.1", |
||||
"description": "Make RPC calls over a Kademlia based DHT.", |
"description": "Make RPC calls over a Kademlia based DHT", |
||||
"main": "index.js", |
"main": "index.js", |
||||
"dependencies": { |
"scripts": { |
||||
"duplexify": "^3.5.0", |
"test": "standard && tape test.js", |
||||
"inherits": "^2.0.3", |
"protobuf": "protocol-buffers schema.proto -o lib/messages.js" |
||||
"ipv4-peers": "^1.1.1", |
|
||||
"k-bucket": "^5.0.0", |
|
||||
"protocol-buffers-encodings": "^1.1.0", |
|
||||
"readable-stream": "^2.1.5", |
|
||||
"sodium-universal": "^2.0.0", |
|
||||
"stream-collector": "^1.0.1", |
|
||||
"time-ordered-set": "^1.0.1", |
|
||||
"udp-request": "^1.3.0", |
|
||||
"xor-distance": "^1.0.0" |
|
||||
}, |
|
||||
"devDependencies": { |
|
||||
"protocol-buffers": "^3.2.1", |
|
||||
"standard": "^11.0.1", |
|
||||
"tape": "^4.9.0" |
|
||||
}, |
}, |
||||
"repository": { |
"repository": { |
||||
"type": "git", |
"type": "git", |
||||
"url": "https://github.com/mafintosh/dht-rpc.git" |
"url": "git+https://github.com/mafintosh/dht-rpc.git" |
||||
}, |
|
||||
"scripts": { |
|
||||
"test": "standard && tape test.js", |
|
||||
"protobuf": "protocol-buffers schema.proto -o messages.js" |
|
||||
}, |
}, |
||||
"author": "Mathias Buus (@mafintosh)", |
"author": "Mathias Buus (@mafintosh)", |
||||
"license": "MIT", |
"license": "MIT", |
||||
"bugs": { |
"bugs": { |
||||
"url": "https://github.com/mafintosh/dht-rpc/issues" |
"url": "https://github.com/mafintosh/dht-rpc/issues" |
||||
}, |
}, |
||||
"homepage": "https://github.com/mafintosh/dht-rpc" |
"homepage": "https://github.com/mafintosh/dht-rpc#readme", |
||||
|
"devDependencies": { |
||||
|
"protocol-buffers": "^4.1.0", |
||||
|
"standard": "^12.0.1", |
||||
|
"tape": "^4.9.1" |
||||
|
}, |
||||
|
"dependencies": { |
||||
|
"codecs": "^1.2.1", |
||||
|
"ipv4-peers": "^1.1.1", |
||||
|
"k-bucket": "^5.0.0", |
||||
|
"protocol-buffers-encodings": "^1.1.0", |
||||
|
"sodium-universal": "^2.0.0", |
||||
|
"stream-collector": "^1.0.1", |
||||
|
"time-ordered-set": "^1.0.1", |
||||
|
"xor-distance": "^1.0.0" |
||||
|
} |
||||
} |
} |
||||
|
|||||
@ -1,304 +0,0 @@ |
|||||
var stream = require('readable-stream') |
|
||||
var inherits = require('inherits') |
|
||||
var nodes = require('ipv4-peers').idLength(32) |
|
||||
var xor = require('xor-distance') |
|
||||
|
|
||||
var BLANK = Buffer.from([ |
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 |
|
||||
]) |
|
||||
|
|
||||
module.exports = QueryStream |
|
||||
|
|
||||
function QueryStream (dht, query, opts) { |
|
||||
if (!(this instanceof QueryStream)) return new QueryStream(dht, query, opts) |
|
||||
if (!opts) opts = {} |
|
||||
if (!opts.concurrency) opts.concurrency = opts.highWaterMark || dht.concurrency |
|
||||
if (!query.target) throw new Error('query.target is required') |
|
||||
|
|
||||
stream.Readable.call(this, {objectMode: true, highWaterMark: opts.concurrency}) |
|
||||
|
|
||||
var self = this |
|
||||
var nodes = opts.node || opts.nodes |
|
||||
|
|
||||
this.query = query |
|
||||
this.query.id = dht._queryId |
|
||||
this.target = query.target |
|
||||
this.token = !!opts.token |
|
||||
this.holepunching = opts.holepunching !== false |
|
||||
this.commits = 0 |
|
||||
this.responses = 0 |
|
||||
this.errors = 0 |
|
||||
this.destroyed = false |
|
||||
this.verbose = !!opts.verbose |
|
||||
|
|
||||
dht.inflightQueries++ |
|
||||
|
|
||||
this._dht = dht |
|
||||
this._committing = false |
|
||||
this._finalized = false |
|
||||
this._closest = opts.closest || [] |
|
||||
this._concurrency = opts.concurrency |
|
||||
this._updating = false |
|
||||
this._pending = nodes ? [].concat(nodes).map(copyNode) : [] |
|
||||
this._k = nodes ? Infinity : opts.k || 20 |
|
||||
this._inflight = 0 |
|
||||
this._moveCloser = !nodes |
|
||||
this._map = opts.map || echo |
|
||||
this._bootstrapped = !this._moveCloser |
|
||||
this._onresponse = onresponse |
|
||||
this._onresponseholepunch = onresponseholepunch |
|
||||
|
|
||||
function onresponseholepunch (err, res, peer, query) { |
|
||||
if (!err || !peer || !peer.referrer) self._callback(err, res, peer) |
|
||||
else self._holepunch(peer, query) |
|
||||
} |
|
||||
|
|
||||
function onresponse (err, res, peer) { |
|
||||
self._callback(err, res, peer) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
inherits(QueryStream, stream.Readable) |
|
||||
|
|
||||
QueryStream.prototype.destroy = function (err) { |
|
||||
if (this.destroyed) return |
|
||||
this.destroyed = true |
|
||||
this._finalize() |
|
||||
if (err) this.emit('error', err) |
|
||||
this.emit('close') |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._finalize = function () { |
|
||||
if (this._finalized) return |
|
||||
this._finalized = true |
|
||||
this._dht.inflightQueries-- |
|
||||
if (!this.responses && !this.destroyed) this.destroy(new Error('No nodes responded')) |
|
||||
if (!this.commits && this._committing && !this.destroyed) this.destroy(new Error('No close nodes responded')) |
|
||||
this.push(null) |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._bootstrap = function () { |
|
||||
this._bootstrapped = true |
|
||||
|
|
||||
var bootstrap = this._dht.bucket.closest(this.target, this._k) |
|
||||
var i = 0 |
|
||||
|
|
||||
for (i = 0; i < bootstrap.length; i++) { |
|
||||
var b = bootstrap[i] |
|
||||
this._addPending({id: b.id, port: b.port, host: b.host}, null) |
|
||||
} |
|
||||
|
|
||||
if (bootstrap.length < this._dht._bootstrap.length) { |
|
||||
for (i = 0; i < this._dht._bootstrap.length; i++) { |
|
||||
this._send(this._dht._bootstrap[i], true, false) |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._readMaybe = function () { |
|
||||
if (this._readableState.flowing === true) this._read() |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._sendTokens = function () { |
|
||||
if (this.destroyed) return |
|
||||
|
|
||||
var sent = this._sendAll(this._closest, false, true) |
|
||||
if (sent || this._inflight) return |
|
||||
|
|
||||
this._finalize() |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._sendPending = function () { |
|
||||
if (this.destroyed) return |
|
||||
if (!this._bootstrapped) this._bootstrap() |
|
||||
|
|
||||
var sent = this._sendAll(this._pending, false, false) |
|
||||
if (sent || this._inflight) return |
|
||||
|
|
||||
if (this.token) { |
|
||||
for (var i = 0; i < this._closest.length; i++) this._closest[i].queried = false |
|
||||
this._committing = true |
|
||||
this._sendTokens() |
|
||||
} else { |
|
||||
this._finalize() |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._read = function () { |
|
||||
if (this._committing) this._sendTokens() |
|
||||
else this._sendPending() |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._holepunch = function (peer, query) { |
|
||||
var self = this |
|
||||
|
|
||||
this._dht._holepunch(peer, peer.referrer, function (err) { |
|
||||
if (err) return self._callback(err, null, peer) |
|
||||
self._dht._request(query, peer, false, self._onresponse) |
|
||||
}) |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._callback = function (err, res, peer) { |
|
||||
this._inflight-- |
|
||||
if (this.destroyed) return |
|
||||
|
|
||||
if (err) { |
|
||||
if (res && res.id) { |
|
||||
var node = this._dht.bucket.get(res.id) |
|
||||
if (node) this._dht._removeNode(node) |
|
||||
} |
|
||||
this.errors++ |
|
||||
this.emit('warning', err) |
|
||||
this._readMaybe() |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
this.responses++ |
|
||||
if (this._committing) this.commits++ |
|
||||
this._addClosest(res, peer) |
|
||||
|
|
||||
if (this._moveCloser) { |
|
||||
var candidates = decodeNodes(res.nodes) |
|
||||
for (var i = 0; i < candidates.length; i++) this._addPending(candidates[i], peer) |
|
||||
} |
|
||||
|
|
||||
if (!validateId(res.id) || (this.token && !this.verbose && !this._committing)) { |
|
||||
this._readMaybe() |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
var data = this._map({ |
|
||||
node: { |
|
||||
id: res.id, |
|
||||
port: peer.port, |
|
||||
host: peer.host |
|
||||
}, |
|
||||
value: res.value |
|
||||
}) |
|
||||
|
|
||||
if (!data) { |
|
||||
this._readMaybe() |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
this.push(data) |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._sendAll = function (nodes, force, useToken) { |
|
||||
var sent = 0 |
|
||||
var free = Math.max(0, this._concurrency - this._dht.socket.inflight) |
|
||||
|
|
||||
if (!free && !this._inflight) free = 1 |
|
||||
if (!free) return 0 |
|
||||
|
|
||||
for (var i = 0; i < nodes.length; i++) { |
|
||||
if (this._send(nodes[i], force, useToken)) { |
|
||||
if (++sent === free) break |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
return sent |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._send = function (node, force, useToken) { |
|
||||
if (!force) { |
|
||||
if (node.queried) return false |
|
||||
node.queried = true |
|
||||
} |
|
||||
|
|
||||
this._inflight++ |
|
||||
|
|
||||
var query = this.query |
|
||||
|
|
||||
if (useToken && node.roundtripToken) { |
|
||||
query = { |
|
||||
command: this.query.command, |
|
||||
id: this.query.id, |
|
||||
target: this.query.target, |
|
||||
value: this.query.value, |
|
||||
roundtripToken: node.roundtripToken |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
this._dht._request(query, node, false, this.holepunching ? this._onresponseholepunch : this._onresponse) |
|
||||
return true |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._addPending = function (node, ref) { |
|
||||
if (node.id.equals(this._dht.id)) return |
|
||||
node.distance = xor(this.target, node.id) |
|
||||
node.referrer = ref |
|
||||
insertSorted(node, this._k, this._pending) |
|
||||
} |
|
||||
|
|
||||
QueryStream.prototype._addClosest = function (res, peer) { |
|
||||
if (!res.id || !res.roundtripToken || res.id.equals(this._dht.id)) return |
|
||||
|
|
||||
var prev = getNode(res.id, this._pending) |
|
||||
|
|
||||
if (!prev) { |
|
||||
prev = { |
|
||||
id: res.id, |
|
||||
port: peer.port, |
|
||||
host: peer.host, |
|
||||
distance: xor(res.id, this.target) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
prev.roundtripToken = res.roundtripToken |
|
||||
insertSorted(prev, this._k, this._closest) |
|
||||
} |
|
||||
|
|
||||
function decodeNodes (buf) { |
|
||||
if (!buf) return [] |
|
||||
try { |
|
||||
return nodes.decode(buf) |
|
||||
} catch (err) { |
|
||||
return [] |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
function getNode (id, list) { |
|
||||
// find id in the list.
|
|
||||
// technically this would be faster with binary search (against distance)
|
|
||||
// but this list is always small, so meh
|
|
||||
for (var i = 0; i < list.length; i++) { |
|
||||
if (list[i].id.equals(id)) return list[i] |
|
||||
} |
|
||||
|
|
||||
return null |
|
||||
} |
|
||||
|
|
||||
function validateId (id) { |
|
||||
return id && id.length === 32 |
|
||||
} |
|
||||
|
|
||||
function insertSorted (node, max, list) { |
|
||||
if (list.length === max && !xor.lt(node.distance, list[max - 1].distance)) return |
|
||||
if (getNode(node.id, list)) return |
|
||||
|
|
||||
if (list.length < max) list.push(node) |
|
||||
else list[max - 1] = node |
|
||||
|
|
||||
var pos = list.length - 1 |
|
||||
while (pos && xor.gt(list[pos - 1].distance, node.distance)) { |
|
||||
list[pos] = list[pos - 1] |
|
||||
list[pos - 1] = node |
|
||||
pos-- |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
function copyNode (node) { |
|
||||
return { |
|
||||
id: node.id || BLANK, |
|
||||
port: node.port, |
|
||||
host: node.host, |
|
||||
roundtripToken: node.roundtripToken, |
|
||||
referrer: node.referrer || node.referer |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
function echo (a) { |
|
||||
return a |
|
||||
} |
|
||||
@ -1,16 +1,27 @@ |
|||||
message Request { |
message Holepunch { |
||||
required string command = 1; |
optional bytes from = 2; |
||||
optional bytes id = 2; |
optional bytes to = 3; |
||||
optional bytes target = 3; |
} |
||||
optional bytes forwardRequest = 4; |
|
||||
optional bytes forwardResponse = 5; |
enum TYPE { |
||||
optional bytes roundtripToken = 6; |
QUERY = 1; |
||||
optional bytes value = 7; |
UPDATE = 2; |
||||
|
RESPONSE = 3; |
||||
} |
} |
||||
|
|
||||
message Response { |
message Message { |
||||
optional bytes id = 1; |
// request/response type + id |
||||
optional bytes nodes = 2; |
required TYPE type = 1; |
||||
optional bytes value = 3; |
required uint64 rid = 2; |
||||
optional bytes roundtripToken = 4; |
|
||||
|
// kademlia stuff |
||||
|
optional bytes id = 3; |
||||
|
optional bytes target = 4; |
||||
|
optional bytes closerNodes = 5; |
||||
|
optional bytes roundtripToken = 6; |
||||
|
|
||||
|
// rpc stuff |
||||
|
optional string command = 7; |
||||
|
optional string error = 8; |
||||
|
optional bytes value = 9; |
||||
} |
} |
||||
|
|||||
Loading…
Reference in new issue