141 lines
3.7 KiB
JavaScript
141 lines
3.7 KiB
JavaScript
'use strict';
|
|
var fs = require('fs');
|
|
var bitcore = require('bitcore');
|
|
var bitcoreUtil = bitcore.util;
|
|
var Sync = require('./Sync');
|
|
var Peer = bitcore.Peer;
|
|
var PeerManager = bitcore.PeerManager;
|
|
var config = require('../config/config');
|
|
var networks = bitcore.networks;
|
|
var sockets = require('../app/controllers/socket.js');
|
|
|
|
var peerdb_fn = 'peerdb.json';
|
|
|
|
function PeerSync(opts) {
|
|
opts = opts|| {};
|
|
this.shouldBroadcast = opts.shouldBroadcast;
|
|
this.connected = false;
|
|
this.peerdb = undefined;
|
|
this.allowReorgs = false;
|
|
var pmConfig = {
|
|
network: config.network
|
|
};
|
|
this.peerman = new PeerManager(pmConfig);
|
|
this.load_peers();
|
|
this.sync = new Sync(opts);
|
|
}
|
|
|
|
PeerSync.prototype.load_peers = function() {
|
|
this.peerdb = [{
|
|
ipv4: config.bitcoind.host,
|
|
port: config.bitcoind.p2pPort
|
|
}];
|
|
|
|
fs.writeFileSync(peerdb_fn, JSON.stringify(this.peerdb));
|
|
};
|
|
|
|
PeerSync.prototype.info = function() {
|
|
return {
|
|
connected: this.connected,
|
|
host: this.peerdb[0].ipv4,
|
|
port: this.peerdb[0].port
|
|
};
|
|
};
|
|
|
|
PeerSync.prototype.handleInv = function(info) {
|
|
var invs = info.message.invs;
|
|
info.conn.sendGetData(invs);
|
|
};
|
|
|
|
PeerSync.prototype.handleTx = function(info) {
|
|
var self =this;
|
|
var tx = this.sync.txDb.getStandardizedTx(info.message.tx);
|
|
console.log('[p2p_sync] Handle tx: ' + tx.txid);
|
|
tx.time = tx.time || Math.round(new Date().getTime() / 1000);
|
|
|
|
this.sync.storeTxs([tx], function(err) {
|
|
if (err) {
|
|
console.log('[p2p_sync] Error in handle TX: ' + JSON.stringify(err));
|
|
}
|
|
else {
|
|
if (self.shouldBroadcast) {
|
|
sockets.broadcastTx(tx);
|
|
if (tx.relatedAddrs) {
|
|
for(var ii in tx.relatedAddrs){
|
|
sockets.broadcastAddressTx(tx.relatedAddrs[ii], tx.txid);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
});
|
|
};
|
|
|
|
|
|
PeerSync.prototype.handleBlock = function(info) {
|
|
var self = this;
|
|
var block = info.message.block;
|
|
var blockHash = bitcoreUtil.formatHashFull(block.calcHash());
|
|
console.log('[p2p_sync] Handle block: %s (allowReorgs: %s)', blockHash, self.allowReorgs);
|
|
|
|
var tx_hashes = block.txs.map(function(tx) {
|
|
return bitcoreUtil.formatHashFull(tx.hash);
|
|
});
|
|
|
|
self.sync.storeTipBlock({
|
|
'hash': blockHash,
|
|
'tx': tx_hashes,
|
|
'previousblockhash': bitcoreUtil.formatHashFull(block.prev_hash),
|
|
}, self.allowReorgs, function(err) {
|
|
if (err && err.message.match(/NEED_SYNC/) && self.historicSync) {
|
|
console.log('[p2p_sync] Orphan block received. Triggering sync');
|
|
self.historicSync.start({}, function(){
|
|
console.log('[p2p_sync] Done resync.');
|
|
});
|
|
}
|
|
else if (err) {
|
|
console.log('[p2p_sync] Error in handle Block: ' + err);
|
|
}
|
|
else {
|
|
if (self.shouldBroadcast) {
|
|
sockets.broadcastBlock(blockHash);
|
|
}
|
|
}
|
|
});
|
|
};
|
|
|
|
PeerSync.prototype.handleConnected = function(data) {
|
|
var peerman = data.pm;
|
|
var peers_n = peerman.peers.length;
|
|
console.log('[p2p_sync] Connected to ' + peers_n + ' peer' + (peers_n !== 1 ? 's' : ''));
|
|
};
|
|
|
|
PeerSync.prototype.run = function() {
|
|
var self = this;
|
|
|
|
this.peerdb.forEach(function(datum) {
|
|
var peer = new Peer(datum.ipv4, datum.port);
|
|
self.peerman.addPeer(peer);
|
|
});
|
|
|
|
this.peerman.on('connection', function(conn) {
|
|
self.connected = true;
|
|
conn.on('inv', self.handleInv.bind(self));
|
|
conn.on('block', self.handleBlock.bind(self));
|
|
conn.on('tx', self.handleTx.bind(self));
|
|
});
|
|
this.peerman.on('connect', self.handleConnected.bind(self));
|
|
|
|
this.peerman.on('netDisconnected', function() {
|
|
self.connected = false;
|
|
});
|
|
|
|
this.peerman.start();
|
|
};
|
|
|
|
PeerSync.prototype.close = function() {
|
|
this.sync.close();
|
|
};
|
|
|
|
|
|
module.exports = require('soop')(PeerSync);
|