2014-01-13 10:17:34 -08:00
|
|
|
'use strict';
|
|
|
|
require('classtool');
|
|
|
|
|
|
|
|
function spec() {
|
|
|
|
var fs = require('fs');
|
|
|
|
var CoinConst = require('bitcore/const');
|
|
|
|
var coinUtil = require('bitcore/util/util');
|
|
|
|
var Sync = require('./Sync').class();
|
2014-02-12 08:11:27 -08:00
|
|
|
var Script = require('bitcore/Script').class();
|
2014-01-13 10:17:34 -08:00
|
|
|
var Peer = require('bitcore/Peer').class();
|
2014-01-23 10:10:52 -08:00
|
|
|
var config = require('../config/config');
|
2014-02-08 05:57:37 -08:00
|
|
|
var networks = require('bitcore/networks');
|
2014-01-13 10:17:34 -08:00
|
|
|
|
|
|
|
var peerdb_fn = 'peerdb.json';
|
|
|
|
function PeerSync() {}
|
2014-01-16 11:11:20 -08:00
|
|
|
|
|
|
|
|
2014-01-23 10:10:52 -08:00
|
|
|
PeerSync.prototype.init = function(opts, cb) {
|
|
|
|
if (!opts) opts = {};
|
2014-02-12 11:27:18 -08:00
|
|
|
this.connected = false;
|
2014-01-16 16:08:50 -08:00
|
|
|
this.peerdb = undefined;
|
2014-02-08 15:09:54 -08:00
|
|
|
this.allowReorgs = false;
|
2014-01-16 16:08:50 -08:00
|
|
|
|
2014-02-09 14:33:39 -08:00
|
|
|
this.sync = new Sync();
|
2014-01-16 16:08:50 -08:00
|
|
|
this.PeerManager = require('bitcore/PeerManager').createClass({
|
2014-02-09 14:33:39 -08:00
|
|
|
network: (config.network === 'testnet' ? networks.testnet : networks.livenet)
|
2014-01-16 16:08:50 -08:00
|
|
|
});
|
|
|
|
this.peerman = new this.PeerManager();
|
|
|
|
this.load_peers();
|
2014-01-23 10:10:52 -08:00
|
|
|
this.sync.init(opts, function() {
|
2014-02-11 11:03:46 -08:00
|
|
|
if (!cb) return;
|
2014-01-16 11:11:20 -08:00
|
|
|
return cb();
|
2014-01-13 10:17:34 -08:00
|
|
|
});
|
|
|
|
};
|
|
|
|
|
|
|
|
PeerSync.prototype.load_peers = function() {
|
2014-01-23 10:10:52 -08:00
|
|
|
this.peerdb = [{
|
|
|
|
ipv4: config.bitcoind.host,
|
2014-01-29 11:16:56 -08:00
|
|
|
port: config.bitcoind.p2pPort
|
2014-01-23 10:10:52 -08:00
|
|
|
}];
|
|
|
|
|
|
|
|
fs.writeFileSync(peerdb_fn, JSON.stringify(this.peerdb));
|
2014-01-13 10:17:34 -08:00
|
|
|
};
|
|
|
|
|
2014-02-12 11:27:18 -08:00
|
|
|
PeerSync.prototype.info = function() {
|
|
|
|
return {
|
|
|
|
connected: this.connected
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
2014-02-09 14:33:39 -08:00
|
|
|
PeerSync.prototype.handleInv = function(info) {
|
2014-01-13 10:17:34 -08:00
|
|
|
var invs = info.message.invs;
|
|
|
|
invs.forEach(function(inv) {
|
2014-02-09 14:33:39 -08:00
|
|
|
console.log('[p2p_sync] Handle inv for a ' + CoinConst.MSG.to_str(inv.type));
|
2014-01-13 10:17:34 -08:00
|
|
|
});
|
2014-01-15 05:32:18 -08:00
|
|
|
// TODO: should limit the invs to objects we haven't seen yet
|
2014-01-13 10:17:34 -08:00
|
|
|
info.conn.sendGetData(invs);
|
|
|
|
};
|
|
|
|
|
2014-02-09 14:33:39 -08:00
|
|
|
PeerSync.prototype.handleTx = function(info) {
|
2014-02-12 08:11:27 -08:00
|
|
|
var self =this;
|
2014-01-13 10:17:34 -08:00
|
|
|
var tx = info.message.tx.getStandardizedObject();
|
2014-02-09 14:33:39 -08:00
|
|
|
console.log('[p2p_sync] Handle tx: ' + tx.hash);
|
2014-02-12 08:11:27 -08:00
|
|
|
tx.time = tx.time || Math.round(new Date().getTime() / 1000);
|
|
|
|
|
|
|
|
var to=0;
|
|
|
|
info.message.tx.outs.forEach( function(o) {
|
|
|
|
var s = new Script(o.s);
|
|
|
|
var addrs = self.sync.getAddrStr(s);
|
|
|
|
|
|
|
|
// support only for p2pubkey p2pubkeyhash and p2sh
|
|
|
|
if (addrs.length === 1) {
|
|
|
|
tx.out[to].addrStr = addrs[0];
|
|
|
|
tx.out[to].n = to;
|
|
|
|
}
|
|
|
|
to++;
|
|
|
|
});
|
2014-02-08 15:09:54 -08:00
|
|
|
|
2014-02-10 14:11:13 -08:00
|
|
|
this.sync.storeTxs([tx], function(err) {
|
2014-01-13 10:17:34 -08:00
|
|
|
if (err) {
|
2014-01-16 16:08:50 -08:00
|
|
|
console.log('[p2p_sync] Error in handle TX: ' + JSON.stringify(err));
|
2014-01-13 10:17:34 -08:00
|
|
|
}
|
|
|
|
});
|
|
|
|
};
|
|
|
|
|
2014-02-09 14:33:39 -08:00
|
|
|
PeerSync.prototype.handleBlock = function(info) {
|
2014-02-08 15:09:54 -08:00
|
|
|
var self = this;
|
2014-01-13 10:17:34 -08:00
|
|
|
var block = info.message.block;
|
|
|
|
var blockHash = coinUtil.formatHashFull(block.calcHash());
|
2014-02-08 15:09:54 -08:00
|
|
|
|
|
|
|
console.log('[p2p_sync] Handle block: %s (allowReorgs: %s)', blockHash, self.allowReorgs);
|
2014-01-15 12:36:49 -08:00
|
|
|
|
|
|
|
var tx_hashes = block.txs.map(function(tx) {
|
|
|
|
return coinUtil.formatHashFull(tx.hash);
|
|
|
|
});
|
|
|
|
|
2014-02-08 05:57:37 -08:00
|
|
|
this.sync.storeTipBlock({
|
2014-01-13 10:17:34 -08:00
|
|
|
'hash': blockHash,
|
2014-01-15 12:36:49 -08:00
|
|
|
'tx': tx_hashes,
|
2014-02-08 05:57:37 -08:00
|
|
|
'previousblockhash': coinUtil.formatHashFull(block.prev_hash),
|
2014-02-08 15:09:54 -08:00
|
|
|
}, self.allowReorgs, function(err) {
|
2014-01-13 10:17:34 -08:00
|
|
|
if (err) {
|
|
|
|
console.log('[p2p_sync] Error in handle Block: ' + err);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
};
|
|
|
|
|
|
|
|
PeerSync.prototype.handle_connected = function(data) {
|
|
|
|
var peerman = data.pm;
|
|
|
|
var peers_n = peerman.peers.length;
|
2014-02-09 14:33:39 -08:00
|
|
|
console.log('[p2p_sync] Connected to ' + peers_n + ' peer' + (peers_n !== 1 ? 's': ''));
|
2014-01-13 10:17:34 -08:00
|
|
|
};
|
|
|
|
|
|
|
|
PeerSync.prototype.run = function() {
|
|
|
|
var self = this;
|
|
|
|
|
|
|
|
this.peerdb.forEach(function(datum) {
|
|
|
|
var peer = new Peer(datum.ipv4, datum.port);
|
2014-01-16 16:08:50 -08:00
|
|
|
self.peerman.addPeer(peer);
|
2014-01-13 10:17:34 -08:00
|
|
|
});
|
|
|
|
|
2014-01-16 16:08:50 -08:00
|
|
|
this.peerman.on('connection', function(conn) {
|
2014-02-12 11:27:18 -08:00
|
|
|
self.connected = true;
|
2014-02-09 14:33:39 -08:00
|
|
|
conn.on('inv', self.handleInv.bind(self));
|
|
|
|
conn.on('block', self.handleBlock.bind(self));
|
|
|
|
conn.on('tx', self.handleTx.bind(self));
|
2014-01-13 10:17:34 -08:00
|
|
|
});
|
2014-01-16 16:08:50 -08:00
|
|
|
this.peerman.on('connect', self.handle_connected.bind(self));
|
2014-01-13 10:17:34 -08:00
|
|
|
|
2014-02-12 11:27:18 -08:00
|
|
|
this.peerman.on('netDisconnected', function() {
|
|
|
|
self.connected = false;
|
|
|
|
});
|
|
|
|
|
2014-01-16 16:08:50 -08:00
|
|
|
this.peerman.start();
|
2014-01-13 10:17:34 -08:00
|
|
|
};
|
2014-01-16 11:12:32 -08:00
|
|
|
|
|
|
|
PeerSync.prototype.close = function() {
|
|
|
|
this.sync.close();
|
|
|
|
};
|
|
|
|
|
|
|
|
|
2014-01-13 10:17:34 -08:00
|
|
|
return PeerSync;
|
|
|
|
|
|
|
|
}
|
|
|
|
module.defineClass(spec);
|
|
|
|
|