Merge pull request #23 from maraoz/feature/commander-p2p
added commander support for p2p.js
This commit is contained in:
commit
5855efa566
96
p2p.js
96
p2p.js
|
@ -1,3 +1,4 @@
|
||||||
|
#! /usr/bin/env node
|
||||||
'use strict';
|
'use strict';
|
||||||
|
|
||||||
process.env.NODE_ENV = process.env.NODE_ENV || 'development';
|
process.env.NODE_ENV = process.env.NODE_ENV || 'development';
|
||||||
|
@ -9,21 +10,30 @@ var CoinConst = require('bitcore/const');
|
||||||
var coinUtil = require('bitcore/util/util');
|
var coinUtil = require('bitcore/util/util');
|
||||||
var networks = require('bitcore/networks');
|
var networks = require('bitcore/networks');
|
||||||
var Parser = require('bitcore/util/BinaryParser').class();
|
var Parser = require('bitcore/util/BinaryParser').class();
|
||||||
var async = require('async');
|
|
||||||
var Sync = require('./lib/Sync').class();
|
var Sync = require('./lib/Sync').class();
|
||||||
|
var Peer = require('bitcore/Peer').class();
|
||||||
|
|
||||||
var peerdb_fn = 'peerdb.json';
|
var peerdb_fn = 'peerdb.json';
|
||||||
|
|
||||||
var peerdb = undefined;
|
var peerdb = undefined;
|
||||||
var hdrdb = undefined;
|
|
||||||
var network = networks.testnet;
|
var PROGRAM_VERSION = '0.1';
|
||||||
var config = {
|
var program = require('commander');
|
||||||
network: network.name
|
|
||||||
};
|
program
|
||||||
var PeerManager = require('bitcore/PeerManager').createClass({
|
.version(PROGRAM_VERSION)
|
||||||
config: config
|
.option('-N --network [testnet]', 'Set bitcoin network [testnet]', 'testnet')
|
||||||
|
.parse(process.argv);
|
||||||
|
|
||||||
|
var sync = new Sync({
|
||||||
|
networkName: program.network
|
||||||
|
});
|
||||||
|
sync.init();
|
||||||
|
|
||||||
|
var PeerManager = require('bitcore/PeerManager').createClass({
|
||||||
|
config: {
|
||||||
|
network: program.network
|
||||||
|
}
|
||||||
});
|
});
|
||||||
var Peer = require('bitcore/Peer').class();
|
|
||||||
|
|
||||||
function peerdb_load() {
|
function peerdb_load() {
|
||||||
try {
|
try {
|
||||||
|
@ -40,74 +50,23 @@ function peerdb_load() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function hdrdb_load() {
|
|
||||||
hdrdb = new HeaderDB({
|
|
||||||
network: network
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function get_more_headers(info) {
|
|
||||||
var conn = info.conn;
|
|
||||||
var loc = hdrdb.locator();
|
|
||||||
conn.sendGetHeaders(loc, coinUtil.NULL_HASH);
|
|
||||||
}
|
|
||||||
|
|
||||||
function add_header(info, block) {
|
|
||||||
var hashStr = coinUtil.formatHashFull(block.calcHash());
|
|
||||||
|
|
||||||
try {
|
|
||||||
hdrdb.add(block);
|
|
||||||
} catch(e) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function handle_headers(info) {
|
|
||||||
console.log('handle headers');
|
|
||||||
var headers = info.message.headers;
|
|
||||||
|
|
||||||
headers.forEach(function(hdr) {
|
|
||||||
add_header(info, hdr);
|
|
||||||
});
|
|
||||||
|
|
||||||
// We persist the header DB after each batch
|
|
||||||
//hdrdb.writeFile(hdrdb_fn);
|
|
||||||
// Only one request per batch of headers we receive.
|
|
||||||
get_more_headers(info);
|
|
||||||
}
|
|
||||||
|
|
||||||
function handle_verack(info) {
|
|
||||||
var inv = {
|
|
||||||
type: CoinConst.MSG.BLOCK,
|
|
||||||
hash: network.genesisBlock.hash,
|
|
||||||
};
|
|
||||||
var invs = [inv];
|
|
||||||
|
|
||||||
// Asks for the genesis block
|
|
||||||
// console.log('p2psync: Asking for the genesis block');
|
|
||||||
// info.conn.sendGetData(invs);
|
|
||||||
}
|
|
||||||
|
|
||||||
function handle_inv(info) {
|
function handle_inv(info) {
|
||||||
// TODO: should limit the invs to objects we haven't seen yet
|
// TODO: should limit the invs to objects we haven't seen yet
|
||||||
var invs = info.message.invs;
|
var invs = info.message.invs;
|
||||||
invs.forEach(function(inv) {
|
invs.forEach(function(inv) {
|
||||||
console.log('Handle inv for a ' + CoinConst.MSG.to_str(inv.type));
|
console.log('Handle inv for a ' + CoinConst.MSG.to_str(inv.type));
|
||||||
});
|
});
|
||||||
|
// this is not needed right now, but it's left in case
|
||||||
|
// we need to store more info in the future
|
||||||
info.conn.sendGetData(invs);
|
info.conn.sendGetData(invs);
|
||||||
}
|
}
|
||||||
|
|
||||||
var sync = new Sync({
|
|
||||||
networkName: networks.testnet
|
|
||||||
});
|
|
||||||
sync.init();
|
|
||||||
|
|
||||||
function handle_tx(info) {
|
function handle_tx(info) {
|
||||||
var tx = info.message.tx.getStandardizedObject();
|
var tx = info.message.tx.getStandardizedObject();
|
||||||
console.log('Handle tx: ' + tx.hash);
|
console.log('Handle tx: ' + tx.hash);
|
||||||
sync.storeTxs([tx.hash], function(err) {
|
sync.storeTxs([tx.hash], function(err) {
|
||||||
if (err) {
|
if (err) {
|
||||||
console.log('error in handle TX: ' + err);
|
console.log('Error in handle TX: ' + err);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
@ -123,8 +82,9 @@ function handle_block(info) {
|
||||||
},
|
},
|
||||||
function(err) {
|
function(err) {
|
||||||
if (err) {
|
if (err) {
|
||||||
console.log('error in handle Block: ' + err);
|
console.log('Error in handle Block: ' + err);
|
||||||
} else {
|
} else {
|
||||||
|
// if no errors importing block, import the transactions
|
||||||
var hashes = block.txs.map(function(tx) {
|
var hashes = block.txs.map(function(tx) {
|
||||||
return coinUtil.formatHashFull(tx.hash);
|
return coinUtil.formatHashFull(tx.hash);
|
||||||
});
|
});
|
||||||
|
@ -149,10 +109,8 @@ function p2psync() {
|
||||||
});
|
});
|
||||||
|
|
||||||
peerman.on('connection', function(conn) {
|
peerman.on('connection', function(conn) {
|
||||||
conn.on('verack', handle_verack);
|
|
||||||
conn.on('block', handle_block);
|
|
||||||
conn.on('headers', handle_headers);
|
|
||||||
conn.on('inv', handle_inv);
|
conn.on('inv', handle_inv);
|
||||||
|
conn.on('block', handle_block);
|
||||||
conn.on('tx', handle_tx);
|
conn.on('tx', handle_tx);
|
||||||
});
|
});
|
||||||
peerman.on('connect', handle_connected);
|
peerman.on('connect', handle_connected);
|
||||||
|
@ -162,8 +120,6 @@ function p2psync() {
|
||||||
|
|
||||||
function main() {
|
function main() {
|
||||||
peerdb_load();
|
peerdb_load();
|
||||||
hdrdb_load();
|
|
||||||
|
|
||||||
p2psync();
|
p2psync();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -11,7 +11,14 @@ var program = require('commander');
|
||||||
var Sync = require('../lib/Sync').class();
|
var Sync = require('../lib/Sync').class();
|
||||||
var async = require('async');
|
var async = require('async');
|
||||||
|
|
||||||
program.version(SYNC_VERSION).option('-N --network [livenet]', 'Set bitcoin network [testnet]', 'testnet').option('-R --reindex', 'Force reindexing', '0').option('-D --destroy', 'Remove current DB', '0').option('--skip_blocks', 'Sync blocks').option('--skip_txs', 'Sync transactions').parse(process.argv);
|
program
|
||||||
|
.version(SYNC_VERSION)
|
||||||
|
.option('-N --network [livenet]', 'Set bitcoin network [testnet]', 'testnet')
|
||||||
|
.option('-R --reindex', 'Force reindexing', '0')
|
||||||
|
.option('-D --destroy', 'Remove current DB', '0')
|
||||||
|
.option('--skip_blocks', 'Sync blocks')
|
||||||
|
.option('--skip_txs', 'Sync transactions')
|
||||||
|
.parse(process.argv);
|
||||||
|
|
||||||
var sync = new Sync({
|
var sync = new Sync({
|
||||||
networkName: program.network
|
networkName: program.network
|
||||||
|
|
Loading…
Reference in New Issue