fcoin/lib/bcoin/pool.js
2016-01-05 02:28:06 -08:00

1545 lines
33 KiB
JavaScript

/**
* pool.js - peer management for bcoin
* Copyright (c) 2014-2015, Fedor Indutny (MIT License)
* https://github.com/indutny/bcoin
*/
var async = require('async');
var inherits = require('inherits');
var EventEmitter = require('events').EventEmitter;
var bcoin = require('../bcoin');
var utils = bcoin.utils;
var assert = utils.assert;
var network = bcoin.protocol.network;
/**
* Pool
*/
function Pool(options) {
var self = this;
var Chain;
if (!(this instanceof Pool))
return new Pool(options);
EventEmitter.call(this);
this.options = options || {};
if (this.options.network)
network.set(this.options.network);
this.options.fullNode = !!this.options.fullNode;
this.options.headers = this.options.headers;
this.options.multiplePeers = this.options.multiplePeers;
this.options.relay = this.options.relay == null
? (this.options.fullNode ? true : false)
: this.options.relay;
this.storage = this.options.storage;
this.destroyed = false;
this.size = options.size || 32;
this.parallel = options.parallel || 2000;
this.redundancy = options.redundancy || 2;
this.seeds = network.seeds.slice();
this._createConnection = options.createConnection;
this._createSocket = options.createSocket;
if (!this.options.fullNode) {
if (this.options.headers == null)
this.options.headers = true;
if (this.options.multiplePeers == null)
this.options.multiplePeers = true;
} else {
if (this.options.headers == null)
this.options.headers = false;
if (this.options.multiplePeers == null)
this.options.multiplePeers = false;
}
if (!this.options.headers)
this.options.multiplePeers = false;
this.backoff = {
delta: options.backoffDelta || 500,
max: options.backoffMax || 5000
};
this.load = {
timeout: options.loadTimeout || 30000,
interval: options.loadInterval || 5000,
window: options.loadWindow || 250,
lastRange: null,
rangeWindow: options.rangeWindow || 1000,
timer: null,
lwm: options.lwm || this.parallel * 2,
hwm: options.hwm || this.parallel * 8,
hiReached: false
};
this.maxRetries = options.maxRetries || 42;
this.requestTimeout = options.requestTimeout || 10000;
this.chain = new bcoin.chain({
storage: this.storage,
fullNode: this.options.fullNode
});
this.watchMap = {};
this.bloom = new bcoin.bloom(
8 * 1024,
10,
(Math.random() * 0xffffffff) | 0
);
this.peers = {
// Peers that are loading blocks themselves
block: [],
// Peers that are still connecting
pending: [],
// Peers that are loading block ids
load: null,
// All peers
all: []
};
this.block = {
bestHeight: 0,
bestHash: null,
type: this.options.fullNode ? 'block' : 'filtered'
};
this.request = {
map: {},
active: 0,
queue: []
};
this.validate = {
// 5 days scan delta for obtaining TXs
delta: 5 * 24 * 3600,
// Minimum verification depth
minDepth: options.minValidateDepth || 0,
// getTX map
map: {}
};
// Currently broadcasted TXs
this.inv = {
list: [],
timeout: options.txTimeout || 60000
};
// Added and watched wallets
this.options.wallets = this.options.wallets || [];
this.wallets = [];
this.chain.on('debug', function() {
var args = Array.prototype.slice.call(arguments);
self.emit.apply(self, ['debug'].concat(args));
});
if (!this.chain.loading) {
this._init();
} else {
this.chain.once('load', function() {
self._init();
});
}
}
inherits(Pool, EventEmitter);
Pool.prototype._init = function _init() {
var self = this;
var i;
this._addLoader();
for (i = 0; i < this.size; i++)
this._addPeer(0);
this.chain.on('missing', function(hash, preload, parent) {
if (!self.options.fullNode) {
self._request(self.block.type, hash, { force: true });
self._scheduleRequests();
// self._loadRange(preload);
}
});
this.chain.on('block', function(block, peer) {
self.emit('block', block, peer);
});
this.chain.on('fork', function(height, hash, checkpoint) {
var peer = self.peers.load;
this.emit('debug', 'Checkpoint %s failed at height %d', hash, height);
if (!peer)
return;
peer.destroy();
});
this.options.wallets.forEach(function(w) {
self.addWallet(w);
});
};
Pool.prototype._startTimer = function _startTimer() {
var self = this;
this._stopTimer();
function destroy() {
// Chain is full and up-to-date
if (self.chain.isFull()) {
self._stopTimer();
self.emit('full');
return;
}
if (self.peers.load)
self.peers.load.destroy();
}
this._timer = setTimeout(destroy, this.load.timeout);
};
Pool.prototype._stopTimer = function _stopTimer() {
if (!this._timer)
return;
clearTimeout(this._timer);
delete this._timer;
};
Pool.prototype._startInterval = function _startInterval() {
var self = this;
this._stopInterval();
function load() {
// self._load();
}
this._interval = setInterval(load, this.load.interval);
};
Pool.prototype._stopInterval = function _stopInterval() {
if (!this._interval)
return;
clearInterval(this._interval);
delete this._interval;
};
Pool.prototype.createConnection = function createConnection(peer, pool) {
var addr, parts, host, net, socket, port;
if (pool._createConnection)
return pool._createConnection(peer, pool);
addr = pool.usableSeed(true);
parts = addr.split(':');
host = parts[0];
port = +parts[1] || network.port;
peer.host = host;
peer.port = port;
if (this._createSocket) {
socket = this._createSocket(port, host);
} else {
net = require('net');
socket = net.connect(port, host);
}
socket.on('connect', function() {
pool.emit('debug', 'Connected to %s:%d', host, port);
});
return socket;
};
Pool.prototype._addLoader = function _addLoader() {
var self = this;
var peer;
if (this.destroyed)
return;
if (this.peers.load != null)
return;
peer = new bcoin.peer(this, this.createConnection, {
backoff: 750 * Math.random(),
startHeight: this.options.startHeight,
relay: this.options.relay
});
this.peers.load = peer;
this.peers.all.push(peer);
peer.on('error', function(err) {
self.emit('error', err, peer);
});
peer.on('debug', function() {
var args = Array.prototype.slice.call(arguments);
self.emit.apply(self, ['debug'].concat(args));
});
peer.once('close', function() {
self._stopInterval();
self._stopTimer();
self._removePeer(peer);
if (self.destroyed)
return;
self._addLoader();
});
peer.once('ack', function() {
peer.updateWatch();
if (!self._load())
self._stopTimer();
});
peer.on('version', function(version) {
if (version.height > self.block.bestHeight)
self.block.bestHeight = version.height;
self.emit('version', version, peer);
});
peer.on('merkleblock', function(block) {
self._handleBlock(block, peer);
});
peer.on('block', function(block) {
self._handleBlock(block, peer);
});
if (self.options.headers) {
peer.on('blocks', function(hashes) {
self._handleInv(hashes, peer);
});
peer.on('headers', function(headers) {
self._handleHeaders(headers, peer);
});
} else {
peer.on('blocks', function(hashes) {
self._handleBlocks(hashes, peer);
});
}
this._startInterval();
this._startTimer();
};
Pool.prototype._handleHeaders = function _handleHeaders(headers, peer) {
var i, header, last, block;
assert(this.options.headers);
if (headers.length === 0)
return;
this.emit('debug',
'Recieved %s headers from %s',
headers.length,
peer.host);
if (headers.length > 2000) {
peer.destroy();
return;
}
this.emit('headers', headers);
for (i = 0; i < headers.length; i++) {
block = bcoin.block(headers[i], 'header');
if (last && block.prevBlock !== last.hash('hex'))
break;
if (!block.verify())
break;
if (!this.chain.has(block))
this._request(this.block.type, block.hash('hex'));
last = block;
}
// Restart the getheaders process
// Technically `last` is not indexed yet so
// the locator hashes will not be entirely
// accurate. However, it shouldn't matter
// that much since FindForkInGlobalIndex
// simply tries to find the latest block in
// the peer's chain.
if (last && headers.length === 2000)
peer.loadHeaders(this.chain.locatorHashes(last), null);
// Push our getdata packet
this._scheduleRequests();
// Reset interval to avoid calling getheaders unnecessarily
this._startInterval();
// Reset timeout to avoid killing the loader
this._startTimer();
this.emit('debug',
'Requesting %s block packets from %s with getdata',
this.request.queue.length,
peer.host
);
};
Pool.prototype._handleBlocks = function _handleBlocks(hashes, peer) {
var i, hash;
assert(!this.options.headers);
if (hashes.length === 0)
return;
this.emit('debug',
'Recieved %s block hashes from %s',
hashes.length,
peer.host);
if (hashes.length > 500) {
peer.destroy();
return;
}
this.emit('blocks', hashes);
for (i = 0; i < hashes.length; i++) {
hash = hashes[i];
// Request block if we don't have it
if (!this.chain.has(hash)) {
this._request(this.block.type, hash);
continue;
}
// Resolve orphan chain
if (this.chain.hasOrphan(hash)) {
peer.loadBlocks(
this.chain.locatorHashes(),
this.chain.getOrphanRoot(hash)
);
continue;
}
}
// Push our getdata packet
this._scheduleRequests();
// Reset interval to avoid calling getblocks unnecessarily
this._startInterval();
// Reset timeout to avoid killing the loader
this._startTimer();
this.emit('debug',
'Requesting %s block packets from %s with getdata',
this.request.queue.length,
peer.host
);
};
Pool.prototype._handleInv = function _handleInv(hashes, peer) {
var i, hash;
for (i = 0; i < hashes.length; i++) {
hash = utils.toHex(hashes[i]);
if (!this.chain.has(hash)) {
if (this.options.headers)
this.peers.load.loadHeaders(this.chain.locatorHashes(), hash);
else
this._request(this.block.type, hash);
}
}
if (!this.options.headers)
this._scheduleRequests();
};
Pool.prototype._handleBlock = function _handleBlock(block, peer) {
var self = this;
var requested = this._response(block);
// Someone is sending us blocks without us requesting them.
if (!requested)
return;
// Emulate bip37 - emit all the "watched" txs
if (this.options.fullNode && this.listeners('watched').length > 0) {
block.txs.forEach(function(tx) {
if (self.isWatched(tx))
self.emit('watched', tx, peer);
});
}
// Ignore if we already have
if (this.chain.has(block))
return;
// Make sure the block is valid
if (!block.verify())
return;
// Resolve orphan chain
if (!this.options.headers) {
if (!this.chain.hasBlock(block.prevBlock)) {
this._addIndex(block, peer);
peer.loadBlocks(
this.chain.locatorHashes(),
this.chain.getOrphanRoot(block)
);
return;
}
}
// Add to index and emit/save
this._addIndex(block, peer);
};
Pool.prototype._addIndex = function _addIndex(block, peer) {
var self = this;
var hash, size, orphans, res;
hash = block.hash('hex');
size = this.chain.size();
orphans = this.chain.orphan.count;
res = this.chain.add(block, peer);
if (res)
this.emit('chain-error', bcoin.chain.msg(res), peer);
// Do not emit if nothing was added to the chain
if (this.chain.size() === size) {
if (this.chain.orphan.count > orphans)
return true;
return false;
}
this.emit('chain-progress', this.chain.fillPercent(), peer);
return true;
};
Pool.prototype.isFull = function isFull() {
return this.chain.isFull();
};
Pool.prototype._loadRange = function _loadRange(hashes, force) {
var now = +new Date();
var last;
if (this.options.fullNode)
return;
if (!hashes)
return;
if (hashes.length <= 1)
return;
// Limit number of requests
if (!force && now - this.load.lastRange < this.load.rangeWindow)
return;
this.load.lastRange = now;
if (!this.peers.load)
this._addLoader();
last = hashes[hashes.length - 1];
hashes.slice(0, -1).forEach(function(hash) {
this.peers.load.loadBlocks([hash], last);
}, this);
};
Pool.prototype._load = function _load() {
var self = this;
var next;
if (this.request.queue.length >= this.load.hwm) {
this.load.hiReached = true;
return false;
}
this.load.hiReached = false;
if (!this.peers.load) {
this._addLoader();
return true;
}
if (this.options.headers)
this.peers.load.loadHeaders(this.chain.locatorHashes(), null);
else
this.peers.load.loadBlocks(this.chain.locatorHashes(), null);
return true;
};
Pool.prototype._addPeer = function _addPeer(backoff) {
var self = this;
var peer;
if (this.destroyed)
return;
if (this.peers.block.length + this.peers.pending.length >= this.size)
return;
if (!this._createConnection && !this.usableSeed()) {
setTimeout(this._addPeer.bind(this, 0), this.backoff.max);
return;
}
peer = new bcoin.peer(this, this.createConnection, {
backoff: backoff,
startHeight: this.options.startHeight,
relay: this.options.relay
});
this.peers.pending.push(peer);
this.peers.all.push(peer);
peer._retry = 0;
// Create new peer on failure
peer.on('error', function(err) {
self.emit('error', err, peer);
});
peer.on('debug', function() {
var args = Array.prototype.slice.call(arguments);
self.emit.apply(self, ['debug'].concat(args));
});
peer.once('close', function() {
self._removePeer(peer);
if (self.destroyed)
return;
self._addPeer(Math.min(backoff + self.backoff.delta, self.backoff.max));
});
peer.once('ack', function() {
var i;
if (self.destroyed)
return;
i = self.peers.pending.indexOf(peer);
if (i !== -1) {
self.peers.pending.splice(i, 1);
self.peers.block.push(peer);
}
peer.updateWatch();
self.inv.list.forEach(function(entry) {
var result = peer.broadcast(entry.msg);
if (!result)
return;
result[0].once('request', function() {
entry.e.emit('ack', peer);
});
});
self._scheduleRequests();
});
peer.on('merkleblock', function(block) {
// Reset backoff, peer seems to be responsive
backoff = 0;
self._handleBlock(block, peer);
});
peer.on('block', function(block) {
// Reset backoff, peer seems to be responsive
backoff = 0;
self._handleBlock(block, peer);
});
// Just FYI
peer.on('reject', function(payload) {
self.emit('reject', payload, peer);
});
peer.on('notfound', function(items) {
items.forEach(function(item) {
var req = self.request.map[utils.toHex(item.hash)];
if (req)
req.finish(null);
});
});
peer.on('tx', function(tx) {
self._response(tx);
self.emit('tx', tx, peer);
if (!self.options.fullNode)
self.emit('watched', tx, peer);
});
peer.on('addr', function(addr) {
var host = addr.ipv4 + ':' + addr.port;
if (self.seeds.length > self.size * 2)
self.seeds = network.seeds.slice();
if (!~self.seeds.indexOf(host)) {
self.emit('debug', 'Found new peer: %s', host);
self.seeds.push(host);
}
self.emit('addr', addr, peer);
});
peer.on('blocks', function(hashes) {
self._handleInv(hashes, peer);
});
peer.on('txs', function(txs) {
self.emit('txs', txs, peer);
});
peer.on('version', function(version) {
if (version.height > self.block.bestHeight)
self.block.bestHeight = version.height;
self.emit('version', version, peer);
});
utils.nextTick(function() {
self.emit('peer', peer);
});
};
Pool.prototype.bestPeer = function bestPeer() {
return this.peers.block.reduce(function(best, peer) {
if (!peer.version || !peer.socket)
return;
if (!best || peer.version.height > best.version.height)
return peer;
return best;
}, null);
};
Pool.prototype._removePeer = function _removePeer(peer) {
var i = this.peers.pending.indexOf(peer);
if (i !== -1)
this.peers.pending.splice(i, 1);
i = this.peers.block.indexOf(peer);
if (i !== -1)
this.peers.block.splice(i, 1);
i = this.peers.all.indexOf(peer);
if (i !== -1)
this.peers.all.splice(i, 1);
if (this.peers.load === peer)
this.peers.load = null;
};
Pool.prototype.watch = function watch(id) {
var hid, i;
if (id instanceof bcoin.wallet) {
this.watchWallet(id);
return;
}
if (id) {
hid = utils.toHex(id);
if (this.watchMap[hid])
this.watchMap[hid]++;
else
this.watchMap[hid] = 1;
if (this.bloom.test(id, 'hex'))
return;
this.bloom.add(id, 'hex');
}
if (this.peers.load)
this.peers.load.updateWatch();
for (i = 0; i < this.peers.block.length; i++)
this.peers.block[i].updateWatch();
};
Pool.prototype.unwatch = function unwatch(id) {
var i;
id = utils.toHex(id);
if (!this.bloom.test(id, 'hex'))
return;
if (!this.watchMap[id] || --this.watchMap[id] !== 0)
return;
delete this.watchMap[id];
// Reset bloom filter
this.bloom.reset();
Object.keys(this.watchMap).forEach(function(id) {
this.bloom.add(id, 'hex');
}, this);
// Resend it to peers
if (this.peers.load)
this.peers.load.updateWatch();
for (i = 0; i < this.peers.block.length; i++)
this.peers.block[i].updateWatch();
};
// See "Filter matching algorithm":
// https://github.com/bitcoin/bips/blob/master/bip-0037.mediawiki
Pool.prototype.isWatched = function(tx, bloom) {
var i, input, output, prev;
if (!bloom)
bloom = this.bloom;
function testScript(script) {
return script.some(function(chunk) {
if (!Array.isArray(chunk) || chunk.length === 0)
return false;
return bloom.test(chunk);
});
}
// 1. Test the tx hash
if (bloom.test(tx.hash()))
return true;
// 2. Test data elements in output scripts
// (may need to update filter on match)
for (i = 0; i < tx.outputs.length; i++) {
output = tx.outputs[i];
// Test the output script
if (testScript(output.script))
return true;
}
// 3. Test prev_out structure
// 4. Test data elements in input scripts
for (i = 0; i < tx.inputs.length; i++) {
input = tx.inputs[i];
prev = input.out.hash;
if (typeof prev === 'string')
prev = utils.toArray(prev, 'hex');
// Test the prev_out tx hash
if (bloom.test(prev))
return true;
// Test the prev_out script
if (input.out.tx) {
prev = input.out.tx.outputs[input.out.index];
if (testScript(prev.script))
return true;
}
// Test the input script
if (testScript(input.script))
return true;
}
// 5. No match
return false;
};
Pool.prototype.addWallet = function addWallet(w, defaultTs) {
var self = this;
var e;
if (this.wallets.indexOf(w) !== -1)
return false;
this.watchWallet(w);
this.wallets.push(w);
e = new EventEmitter();
function search(ts) {
// Relay pending TXs
// NOTE: It is important to do it after search, because search could
// add TS to pending TXs, thus making them confirmed
w.pending().forEach(function(tx) {
self.sendTX(tx);
});
if (self.options.fullNode)
return;
// Search for last week by default
if (!ts)
ts = defaultTs || (utils.now() - 7 * 24 * 3600);
// self.search(false, ts, e);
self.searchWallet(ts);
}
if (w.loaded)
search(w.lastTs);
else
w.once('load', function() { search(w.lastTs) });
return e;
};
Pool.prototype.removeWallet = function removeWallet(w) {
var i = this.wallets.indexOf(w);
if (i == -1)
return;
this.wallets.splice(i, 1);
this.unwatchWallet(w);
};
Pool.prototype.watchWallet = function watchWallet(w) {
if (w.type === 'scripthash') {
// For the redeem script hash in outputs:
this.watch(w.getFullHash());
// For the redeem script in inputs:
this.watch(w.getFullPublicKey());
}
// For the pubkey hash in outputs:
this.watch(w.getOwnHash());
// For the pubkey in inputs:
this.watch(w.getOwnPublicKey());
};
Pool.prototype.unwatchWallet = function unwatchWallet(w) {
if (w.type === 'scripthash') {
// For the redeem script hash in p2sh outputs:
this.unwatch(w.getFullHash());
// For the redeem script in p2sh inputs:
this.unwatch(w.getFullPublicKey());
}
// For the pubkey hash in p2pk/multisig outputs:
this.unwatch(w.getOwnHash());
// For the pubkey in p2pkh inputs:
this.unwatch(w.getOwnPublicKey());
};
Pool.prototype.searchWallet = function(w) {
var self = this;
var ts;
if (this.options.fullNode)
return;
if (!w) {
ts = this.wallets.reduce(function(ts, w) {
if (w.lastTs < ts)
return w.lastTs;
return ts;
}, Infinity);
assert(ts !== Infinity);
} else if (typeof w === 'number') {
ts = w;
} else {
if (!w.loaded) {
w.once('load', function() {
self.searchWallet(w);
});
return;
}
ts = w.lastTs;
if (!ts)
ts = utils.now() - 7 * 24 * 3600;
}
utils.nextTick(function() {
self.emit('debug', 'Wallet time: %s', new Date(ts * 1000));
});
// this.search(ts);
this.chain.resetTime(ts);
};
Pool.prototype.search = function search(id, range, e) {
var self = this;
var hashes, pending, listener, timeout, done, total;
if (this.options.fullNode)
return;
e = e || new EventEmitter();
// Optional id argument
if ((id !== null
&& typeof id === 'object'
&& !Array.isArray(id))
|| typeof id === 'number') {
range = id;
id = null;
}
if (typeof id === 'string')
id = utils.toArray(id, 'hex');
if (typeof range === 'number')
range = { start: range, end: null };
else if (range)
range = { start: range.start, end: range.end };
else
range = { start: 0, end: 0 };
// Last 5 days by default, this covers 1000 blocks that we have in the
// chain by default
if (!range.end)
range.end = utils.now();
if (!range.start)
range.start = utils.now() - 432000;
if (range.start < this.chain.index.lastTs) {
if (id)
this.watch(id);
done = function(res) {
e.emit('end', res);
clearInterval(timeout);
self.removeListener('block', listener);
if (id)
self.unwatch(id);
};
this.on('block', listener = function(block) {
if (block.ts >= range.end)
done();
});
// Estimated number of blocks in time range
total = (range.end - range.start) / network.powTargetSpacing | 0;
if (total === 0)
total = 1;
// 500 blocks every 3 seconds
total = (total / 500 | 0) * 3;
// Add half the total time and convert to ms
total = (total + Math.ceil(total / 2)) * 1000;
timeout = setTimeout(done.bind(null, true), total);
this.chain.resetTime(range.start);
if (this.peers.load)
this.peers.load.destroy();
this._load();
return e;
}
hashes = this.chain.hashRange(range.start, range.end);
pending = hashes.length;
if (id)
this.watch(id);
done = function() {
pending--;
assert(pending >= 0);
e.emit('progress', count - pending, count);
if (pending === 0) {
if (id)
self.unwatch(id);
e.emit('end');
}
};
hashes.forEach(function(hash) {
self._request('filtered', hash, { force: true }, done);
});
if (hashes.length === 0) {
bcoin.utils.nextTick(function() {
e.emit('end', true);
});
}
return e;
};
Pool.prototype._request = function _request(type, hash, options, cb) {
var self = this;
// Optional `force`
if (typeof options === 'function') {
cb = options;
options = {};
}
if (!options)
options = {};
hash = utils.toHex(hash);
if (this.request.map[hash])
return this.request.map[hash].addCallback(cb);
function next(has) {
if (has)
return;
if (self.destroyed)
return;
var req = new LoadRequest(self, type, hash, cb);
req.add(options.noQueue);
}
// Block should be not in chain, or be requested
// Do not use with headers-first
if (!options.force && (type === 'block' || type === 'filtered'))
return this.chain.has(hash, true, next);
return next(false);
};
Pool.prototype._response = function _response(entity) {
var req = this.request.map[entity.hash('hex')];
if (!req)
return false;
req.finish(entity);
return true;
};
Pool.prototype._scheduleRequests = function _scheduleRequests() {
var self = this;
if (this.destroyed)
return;
if (this.request.active > this.parallel / 2)
return;
// No need to wait - already have enough data
if (this.request.queue.length > this.parallel) {
if (this.load.timer !== null)
clearTimeout(this.load.timer);
this.load.timer = null;
return this._doRequests();
}
// Already waiting
if (this.load.timer !== null)
return;
this.load.timer = setTimeout(function() {
self.load.timer = null;
self._doRequests();
}, this.load.window);
};
Pool.prototype._doRequests = function _doRequests() {
var queue, above, items, below;
var red, count, split, i, off, req, j;
var mapReq;
if (this.request.active >= this.parallel)
return;
// No peers so far
if (this.peers.block.length === 0)
return;
queue = this.request.queue;
above = queue.length >= this.load.lwm;
items = queue.slice(0, this.parallel - this.request.active);
this.request.queue = queue.slice(items.length);
below = this.request.queue.length < this.load.lwm;
// Watermark boundary crossed, load more blocks
if (above && below && this.load.hiReached)
this._load();
if (this.options.multiplePeers) {
mapReq = function(item) {
return item.start(this.peers.block[i]);
};
// Split list between peers
red = this.redundancy;
count = this.peers.block.length;
split = Math.ceil(items.length * red / count);
for (i = 0, off = 0; i < count; i += red, off += split) {
req = items.slice(off, off + split).map(mapReq, this);
for (j = 0; j < red && i + j < count; j++)
this.peers.block[i + j].getData(req);
}
return;
}
req = items.map(function(item) {
return item.start(this.peers.load);
}, this);
this.peers.load.getData(req);
};
Pool.prototype.getBlock = function getBlock(hash, cb) {
this._request('block', hash, { force: true }, function(block) {
cb(null, block);
});
this._scheduleRequests();
};
Pool.prototype.sendBlock = function sendBlock(block) {
return this.broadcast(block);
};
Pool.prototype.getTX = function getTX(hash, range, cb) {
var self = this;
var cbs, tx, finished, req, delta;
if (this.options.fullNode)
return cb(new Error('Cannot get tx with full node'));
hash = utils.toHex(hash);
if (typeof range === 'function') {
cb = range;
range = null;
}
// Do not perform duplicate searches
if (this.validate.map[hash])
return this.validate.map[hash].push(cb);
cbs = [cb];
this.validate.map[hash] = cbs;
// Add request without queueing it to get notification at the time of load
tx = null;
finished = false;
req = this._request('tx', hash, { noQueue: true }, function(t) {
finished = true;
tx = t;
});
// Do incremental search until the TX is found
delta = this.validate.delta;
// Start from the existing range if given
if (range)
range = { start: range.start, end: range.end };
else
range = { start: utils.now() - delta, end: 0 };
function doSearch() {
var e = self.search(hash, range);
e.on('end', function(empty) {
if (finished) {
delete self.validate.map[hash];
cbs.forEach(function(cb) {
cb(null, tx, range);
});
return;
}
// Tried everything, but still no matches
if (empty)
return cb(new Error('Not found.'));
// Not found yet, continue scanning
range.end = range.start;
range.start -= delta;
if (range.start < 0)
range.start = 0;
doSearch();
});
}
doSearch();
};
Pool.prototype.sendTX = function sendTX(tx) {
return this.broadcast(tx);
};
Pool.prototype.broadcast = function broadcast(msg) {
var self = this;
var e = new EventEmitter();
var entry = {
msg: msg,
e: e,
timer: setTimeout(function() {
var i = self.inv.list.indexOf(entry);
if (i !== -1)
self.inv.list.splice(i, 1);
}, this.inv.timeout)
};
this.inv.list.push(entry);
this.peers.block.forEach(function(peer) {
var result = peer.broadcast(msg);
if (!result) return;
result[0].once('request', function() {
e.emit('ack', peer);
});
});
return e;
};
Pool.prototype.destroy = function destroy() {
if (this.destroyed)
return;
this.destroyed = true;
if (this.peers.load)
this.peers.load.destroy();
this.request.queue.slice().forEach(function(item) {
item.finish(null);
});
this.inv.list.forEach(function(entry) {
clearTimeout(entry.timer);
entry.timer = null;
});
this.peers.pending.slice().forEach(function(peer) {
peer.destroy();
});
this.peers.block.slice().forEach(function(peer) {
peer.destroy();
});
if (this.load.timer)
clearTimeout(this.load.timer);
this.load.timer = null;
};
Pool.prototype.getPeer = function getPeer(addr) {
var parts, host, port, i, peer;
if (!addr)
return;
parts = addr.split(':');
host = parts[0];
port = +parts[1] || network.port;
for (i = 0; i < this.peers.all.length; i++) {
peer = this.peers.all[i];
if (peer.host === host)
return peer;
}
};
Pool.prototype.usableSeed = function usableSeed(addrs, force) {
var i, addr;
if (typeof addrs === 'boolean') {
force = addrs;
addrs = null;
}
if (!addrs)
addrs = this.seeds;
for (i = 0; i < addrs.length; i++) {
if (!this.getPeer(addrs[i]))
return addrs[i];
}
if (!force)
return;
if (addrs.length === 1)
return addrs[0];
do {
addr = addrs[Math.random() * (addrs.length - 1) | 0];
} while (this.peers.load && this.getPeer(addr) === this.peers.load);
return addr;
};
Pool.prototype.toJSON = function toJSON() {
return {
v: 1,
type: 'pool',
chain: this.chain.toJSON(),
requests: this.request.queue.map(function(item) {
return {
type: item.type,
hash: item.hash
};
})
};
};
Pool.prototype.fromJSON = function fromJSON(json) {
assert.equal(json.v, 1);
assert.equal(json.type, 'pool');
this.chain.fromJSON(json.chain);
json.requests.forEach(function(item) {
this._request(item.type, item.hash);
}, this);
return this;
};
function LoadRequest(pool, type, hash, cb) {
var self = this;
this.pool = pool;
this.type = type;
this.hash = hash;
this.cbs = cb ? [cb] : [];
this.timer = null;
this.peer = null;
this.ts = +new Date();
this.active = false;
this.noQueue = false;
this.onclose = function onclose() {
if (self.pool.destroyed)
self.clear();
else
self.retry();
};
}
LoadRequest.prototype.start = function start(peer) {
var self = this;
var reqType;
assert(!this.active);
this.active = true;
this.pool.request.active++;
assert(!this.timer);
this.timer = setTimeout(function() {
self.timer = null;
self.retry();
}, this.pool.requestTimeout);
assert(!this.peer);
this.peer = peer;
this.peer.once('close', this.onclose);
return {
type: this.type,
hash: this.hash
};
};
LoadRequest.compare = function compare(a, b) {
return a.ts - b.ts;
};
LoadRequest.prototype.add = function add(noQueue) {
this.pool.request.map[this.hash] = this;
if (!noQueue)
this.pool.request.queue.push(this);
else
this.noQueue = true;
};
LoadRequest.prototype.clear = function clear() {
assert(this.active);
this.pool.request.active--;
this.active = false;
this.peer.removeListener('close', this.onclose);
this.peer = null;
clearTimeout(this.timer);
this.timer = null;
};
LoadRequest.prototype.retry = function retry() {
var peer = this.peer;
// Put block into the queue, ensure that the queue is always sorted by ts
utils.binaryInsert(this.pool.request.queue, this, LoadRequest.compare);
this.clear();
// Kill peer, if it misbehaves
if (++peer._retry > this.pool.maxRetries)
peer.destroy();
// And schedule requesting blocks again
this.pool._scheduleRequests();
};
LoadRequest.prototype.finish = function finish(entity) {
var index;
if (this.active) {
this.clear();
} else {
// It could be that request was never sent to the node, remove it from
// queue and forget about it
index = this.pool.request.queue.indexOf(this);
assert(index !== -1 || this.noQueue);
if (!this.noQueue)
this.pool.request.queue.splice(index, 1);
assert(!this.peer);
assert(!this.timer);
}
delete this.pool.request.map[this.hash];
// We may have some free slots in queue now
this.pool._scheduleRequests();
this.cbs.forEach(function(cb) {
cb(entity);
});
};
LoadRequest.prototype.addCallback = function addCallback(cb) {
if (cb)
this.cbs.push(cb);
};
/**
* Expose
*/
module.exports = Pool;