mirror of https://github.com/BTCPrivate/copay.git
592 lines
15 KiB
JavaScript
592 lines
15 KiB
JavaScript
'use strict';
|
|
|
|
var imports = require('soop').imports();
|
|
|
|
var bitcore = require('bitcore');
|
|
var coinUtil = bitcore.util;
|
|
var buffertools = bitcore.buffertools;
|
|
var Builder = bitcore.TransactionBuilder;
|
|
var http = require('http');
|
|
var EventEmitter = imports.EventEmitter || require('events').EventEmitter;
|
|
var copay = copay || require('../../../copay');
|
|
var SecureRandom = bitcore.SecureRandom;
|
|
var Base58Check = bitcore.Base58.base58Check;
|
|
|
|
function Wallet(opts) {
|
|
var self = this;
|
|
|
|
//required params
|
|
['storage', 'network', 'blockchain',
|
|
'requiredCopayers', 'totalCopayers', 'spendUnconfirmed',
|
|
'publicKeyRing', 'txProposals', 'privateKey', 'version'
|
|
].forEach(function(k) {
|
|
if (typeof opts[k] === 'undefined') throw new Error('missing key:' + k);
|
|
self[k] = opts[k];
|
|
});
|
|
|
|
this.log('creating ' + opts.requiredCopayers + ' of ' + opts.totalCopayers + ' wallet');
|
|
|
|
this.id = opts.id || Wallet.getRandomId();
|
|
this.name = opts.name;
|
|
this.netKey = opts.netKey || SecureRandom.getRandomBuffer(8).toString('base64');
|
|
|
|
this.verbose = opts.verbose;
|
|
this.publicKeyRing.walletId = this.id;
|
|
this.txProposals.walletId = this.id;
|
|
this.network.maxPeers = this.totalCopayers;
|
|
this.registeredPeerIds = [];
|
|
}
|
|
|
|
Wallet.parent = EventEmitter;
|
|
Wallet.prototype.log = function() {
|
|
if (!this.verbose) return;
|
|
if (console)
|
|
console.log.apply(console, arguments);
|
|
};
|
|
|
|
Wallet.getRandomId = function() {
|
|
var r = bitcore.SecureRandom.getPseudoRandomBuffer(8).toString('hex');
|
|
return r;
|
|
};
|
|
|
|
Wallet.prototype.seedCopayer = function(pubKey) {
|
|
this.seededCopayerId = pubKey;
|
|
};
|
|
|
|
Wallet.prototype.connectToAll = function() {
|
|
var all = this.publicKeyRing.getAllCopayerIds();
|
|
this.network.connectToCopayers(all);
|
|
if (this.seededCopayerId) {
|
|
this.sendWalletReady(this.seededCopayerId);
|
|
this.seededCopayerId = null;
|
|
}
|
|
};
|
|
|
|
Wallet.prototype._handlePublicKeyRing = function(senderId, data, isInbound) {
|
|
this.log('RECV PUBLICKEYRING:', data);
|
|
|
|
var recipients, pkr = this.publicKeyRing;
|
|
var inPKR = copay.PublicKeyRing.fromObj(data.publicKeyRing);
|
|
|
|
var hasChanged = pkr.merge(inPKR, true);
|
|
if (hasChanged) {
|
|
this.connectToAll();
|
|
if (this.publicKeyRing.isComplete()) {
|
|
this._lockIncomming();
|
|
}
|
|
this.log('### BROADCASTING PKR');
|
|
|
|
recipients = null;
|
|
this.sendPublicKeyRing(recipients);
|
|
}
|
|
this.emit('refresh', this.publicKeyRing);
|
|
this.store();
|
|
};
|
|
|
|
|
|
Wallet.prototype._handleTxProposals = function(senderId, data, isInbound) {
|
|
this.log('RECV TXPROPOSAL:', data);
|
|
|
|
var recipients;
|
|
var inTxp = copay.TxProposals.fromObj(data.txProposals);
|
|
var mergeInfo = this.txProposals.merge(inTxp, true);
|
|
var addSeen = this.addSeenToTxProposals();
|
|
if (mergeInfo.hasChanged || addSeen) {
|
|
this.log('### BROADCASTING txProposals. ');
|
|
recipients = null;
|
|
TODO: only diff
|
|
this.sendTxProposals(recipients);
|
|
}
|
|
this.emit('txProposalsUpdated', this.txProposals);
|
|
this.store();
|
|
};
|
|
|
|
Wallet.prototype._handleData = function(senderId, data, isInbound) {
|
|
// TODO check message signature
|
|
if (this.id !== data.walletId) {
|
|
this.emit('badMessage', senderId);
|
|
this.log('badMessage FROM:', senderId); //TODO
|
|
return;
|
|
}
|
|
switch (data.type) {
|
|
// This handler is repeaded on WalletFactory (#join). TODO
|
|
case 'walletId':
|
|
this.sendWalletReady(senderId);
|
|
break;
|
|
case 'walletReady':
|
|
this.sendPublicKeyRing(senderId);
|
|
this.sendTxProposals(senderId); // send old
|
|
break;
|
|
case 'publicKeyRing':
|
|
this._handlePublicKeyRing(senderId, data, isInbound);
|
|
break;
|
|
case 'txProposals':
|
|
this._handleTxProposals(senderId, data, isInbound);
|
|
break;
|
|
}
|
|
};
|
|
|
|
Wallet.prototype._handleConnect = function(newCopayerId) {
|
|
if (newCopayerId) {
|
|
this.log('#### Setting new COPAYER:', newCopayerId);
|
|
this.sendWalletId(newCopayerId);
|
|
}
|
|
var peerID = this.network.peerFromCopayer(newCopayerId)
|
|
this.emit('connect', peerID);
|
|
};
|
|
|
|
Wallet.prototype._handleDisconnect = function(peerID) {
|
|
this.emit('disconnect', peerID);
|
|
};
|
|
|
|
Wallet.prototype._optsToObj = function() {
|
|
var obj = {
|
|
id: this.id,
|
|
spendUnconfirmed: this.spendUnconfirmed,
|
|
requiredCopayers: this.requiredCopayers,
|
|
totalCopayers: this.totalCopayers,
|
|
name: this.name,
|
|
netKey: this.netKey,
|
|
version: this.version,
|
|
};
|
|
|
|
return obj;
|
|
};
|
|
|
|
|
|
Wallet.prototype.getCopayerId = function(index) {
|
|
return this.publicKeyRing.getCopayerId(index || 0);
|
|
};
|
|
|
|
|
|
Wallet.prototype.getMyCopayerId = function() {
|
|
return this.getCopayerId(0);
|
|
};
|
|
|
|
|
|
Wallet.prototype.getSecret = function() {
|
|
var i = new Buffer(this.getMyCopayerId(), 'hex');
|
|
var k = new Buffer(this.netKey, 'base64');
|
|
var b = Buffer.concat([i, k]);
|
|
var str = Base58Check.encode(b);
|
|
return str;
|
|
};
|
|
|
|
|
|
Wallet.decodeSecret = function(secretB) {
|
|
var secret = Base58Check.decode(secretB);
|
|
var netKeyBuf = secret.slice(-8);
|
|
var pubKeyBuf = secret.slice(0, 33);
|
|
return {
|
|
pubKey: pubKeyBuf.toString('hex'),
|
|
netKey: netKeyBuf.toString('base64'),
|
|
}
|
|
};
|
|
|
|
Wallet.prototype._lockIncomming = function() {
|
|
this.network.lockIncommingConnections(this.publicKeyRing.getAllCopayerIds());
|
|
};
|
|
|
|
Wallet.prototype.netStart = function() {
|
|
var self = this;
|
|
var net = this.network;
|
|
net.removeAllListeners();
|
|
net.on('connect', self._handleConnect.bind(self));
|
|
net.on('disconnect', self._handleDisconnect.bind(self));
|
|
net.on('data', self._handleData.bind(self));
|
|
net.on('openError', function() {
|
|
self.log('[Wallet.js.132:openError:] GOT openError'); //TODO
|
|
self.emit('openError');
|
|
});
|
|
net.on('close', function() {
|
|
self.emit('close');
|
|
});
|
|
|
|
var myId = self.getMyCopayerId();
|
|
var startOpts = {
|
|
copayerId: myId,
|
|
maxPeers: self.totalCopayers,
|
|
netKey: this.netKey,
|
|
};
|
|
|
|
if (this.publicKeyRing.isComplete()) {
|
|
this._lockIncomming();
|
|
}
|
|
|
|
net.start(startOpts, function() {
|
|
self.connectToAll();
|
|
self.emit('ready', net.getPeer());
|
|
self.emit('refresh');
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.getOnlinePeerIDs = function() {
|
|
return this.network.getOnlinePeerIDs();
|
|
};
|
|
|
|
Wallet.prototype.getRegisteredCopayerIds = function() {
|
|
var l = this.publicKeyRing.registeredCopayers();
|
|
var copayers = [];
|
|
for (var i = 0; i < l; i++) {
|
|
var cid = this.getCopayerId(i);
|
|
copayers.push(cid);
|
|
}
|
|
return copayers;
|
|
};
|
|
|
|
Wallet.prototype.getRegisteredPeerIds = function() {
|
|
var l = this.publicKeyRing.registeredCopayers();
|
|
if (this.registeredPeerIds.length !== l) {
|
|
this.registeredPeerIds = [];
|
|
var copayers = this.getRegisteredCopayerIds();
|
|
for (var i = 0; i < l; i++) {
|
|
var cid = copayers[i];
|
|
var pid = this.network.peerFromCopayer(cid);
|
|
this.registeredPeerIds.push({
|
|
peerId: pid,
|
|
nick: this.publicKeyRing.nicknameForCopayer(cid)
|
|
});
|
|
}
|
|
}
|
|
return this.registeredPeerIds;
|
|
};
|
|
|
|
Wallet.prototype.store = function() {
|
|
var wallet = this.toObj();
|
|
this.storage.setFromObj(this.id, wallet);
|
|
this.log('Wallet stored');
|
|
};
|
|
|
|
Wallet.prototype.toObj = function() {
|
|
var optsObj = this._optsToObj();
|
|
var walletObj = {
|
|
opts: optsObj,
|
|
publicKeyRing: this.publicKeyRing.toObj(),
|
|
txProposals: this.txProposals.toObj(),
|
|
privateKey: this.privateKey.toObj()
|
|
};
|
|
|
|
return walletObj;
|
|
};
|
|
|
|
Wallet.fromObj = function(o, storage, network, blockchain) {
|
|
var opts = JSON.parse(JSON.stringify(o.opts));
|
|
opts.publicKeyRing = copay.PublicKeyRing.fromObj(o.publicKeyRing);
|
|
opts.txProposals = copay.TxProposals.fromObj(o.txProposals);
|
|
opts.privateKey = copay.PrivateKey.fromObj(o.privateKey);
|
|
|
|
opts.storage = storage;
|
|
opts.network = network;
|
|
opts.blockchain = blockchain;
|
|
var w = new Wallet(opts);
|
|
return w;
|
|
};
|
|
|
|
Wallet.prototype.toEncryptedObj = function() {
|
|
var walletObj = this.toObj();
|
|
return this.storage.export(walletObj);
|
|
};
|
|
|
|
Wallet.prototype.sendTxProposals = function(recipients, ntxid) {
|
|
this.log('### SENDING txProposals TO:', recipients || 'All', this.txProposals);
|
|
|
|
var toSend = ntxid ? [ntxid] : this.getTxProposals.getNtxids();
|
|
for(var i in toSend) {
|
|
var id = toSend[i];
|
|
this.network.send(recipients, {
|
|
type: 'txProposals',
|
|
txProposals: this.txProposals.toObj(id),
|
|
walletId: this.id,
|
|
});
|
|
}
|
|
};
|
|
|
|
Wallet.prototype.sendWalletReady = function(recipients) {
|
|
this.log('### SENDING WalletReady TO:', recipients);
|
|
|
|
this.network.send(recipients, {
|
|
type: 'walletReady',
|
|
walletId: this.id,
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.sendWalletId = function(recipients) {
|
|
this.log('### SENDING walletId TO:', recipients || 'All', this.id);
|
|
|
|
this.network.send(recipients, {
|
|
type: 'walletId',
|
|
walletId: this.id,
|
|
opts: this._optsToObj()
|
|
});
|
|
};
|
|
|
|
|
|
Wallet.prototype.sendPublicKeyRing = function(recipients) {
|
|
this.log('### SENDING publicKeyRing TO:', recipients || 'All', this.publicKeyRing.toObj());
|
|
|
|
this.network.send(recipients, {
|
|
type: 'publicKeyRing',
|
|
publicKeyRing: this.publicKeyRing.toObj(),
|
|
walletId: this.id,
|
|
});
|
|
};
|
|
|
|
|
|
Wallet.prototype.generateAddress = function(isChange) {
|
|
var addr = this.publicKeyRing.generateAddress(isChange);
|
|
this.sendPublicKeyRing();
|
|
this.store();
|
|
return addr;
|
|
};
|
|
|
|
|
|
Wallet.prototype.getTxProposals = function() {
|
|
var ret = [];
|
|
var copayers = this.getRegisteredCopayerIds();
|
|
for (var k in this.txProposals.txps) {
|
|
var i = this.txProposals.getTxProposal(k, copayers);
|
|
i.signedByUs = i.signedBy[this.getMyCopayerId()] ? true : false;
|
|
i.rejectedByUs = i.rejectedBy[this.getMyCopayerId()] ? true : false;
|
|
if (this.totalCopayers - i.rejectCount < this.requiredCopayers)
|
|
i.finallyRejected = true;
|
|
|
|
ret.push(i);
|
|
}
|
|
return ret;
|
|
};
|
|
|
|
|
|
Wallet.prototype.reject = function(ntxid) {
|
|
var myId = this.getMyCopayerId();
|
|
var txp = this.txProposals.txps[ntxid];
|
|
if (!txp || txp.rejectedBy[myId] || txp.signedBy[myId]) return;
|
|
|
|
txp.rejectedBy[myId] = Date.now();
|
|
this.sendTxProposals(null, ntxid);
|
|
this.store();
|
|
this.emit('txProposalsUpdated');
|
|
};
|
|
|
|
|
|
Wallet.prototype.sign = function(ntxid) {
|
|
var self = this;
|
|
var myId = this.getMyCopayerId();
|
|
var txp = self.txProposals.txps[ntxid];
|
|
if (!txp || txp.rejectedBy[myId] || txp.signedBy[myId]) return;
|
|
|
|
var pkr = self.publicKeyRing;
|
|
var keys = self.privateKey.getAll(pkr.addressIndex, pkr.changeAddressIndex);
|
|
|
|
var b = txp.builder;
|
|
var before = b.signaturesAdded;
|
|
b.sign(keys);
|
|
|
|
if (b.signaturesAdded > before) {
|
|
txp.signedBy[myId] = Date.now();
|
|
this.sendTxProposals(null, ntxid);
|
|
this.store();
|
|
this.emit('txProposalsUpdated');
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|
|
|
|
Wallet.prototype.sendTx = function(ntxid, cb) {
|
|
var txp = this.txProposals.txps[ntxid];
|
|
if (!txp) return;
|
|
|
|
var tx = txp.builder.build();
|
|
if (!tx.isComplete()) return;
|
|
this.log('[Wallet.js.231] BROADCASTING TX!!!'); //TODO
|
|
|
|
var txHex = tx.serialize().toString('hex');
|
|
this.log('[Wallet.js.261:txHex:]', txHex); //TODO
|
|
|
|
var self = this;
|
|
|
|
this.blockchain.sendRawTransaction(txHex, function(txid) {
|
|
self.log('BITCOND txid:', txid); //TODO
|
|
if (txid) {
|
|
self.txProposals.setSent(ntxid, txid);
|
|
}
|
|
self.sendTxProposals(null, ntxid);
|
|
self.store();
|
|
return cb(txid);
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.addSeenToTxProposals = function() {
|
|
var ret = false;
|
|
var myId = this.getMyCopayerId();
|
|
|
|
for (var k in this.txProposals.txps) {
|
|
var txp = this.txProposals.txps[k];
|
|
if (!txp.seenBy[myId]) {
|
|
|
|
txp.seenBy[myId] = Date.now();
|
|
ret = true;
|
|
}
|
|
}
|
|
return ret;
|
|
};
|
|
|
|
// TODO: remove this method and use getAddressesInfo everywhere
|
|
Wallet.prototype.getAddresses = function(opts) {
|
|
return this.publicKeyRing.getAddresses(opts);
|
|
};
|
|
|
|
Wallet.prototype.getAddressesStr = function(opts) {
|
|
return this.getAddresses(opts).map(function(a) {
|
|
return a.toString();
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.getAddressesInfo = function(opts) {
|
|
return this.publicKeyRing.getAddressesInfo(opts);
|
|
};
|
|
|
|
Wallet.prototype.addressIsOwn = function(addrStr, opts) {
|
|
var addrList = this.getAddressesStr(opts);
|
|
var l = addrList.length;
|
|
var ret = false;
|
|
|
|
for (var i = 0; i < l; i++) {
|
|
if (addrList[i] === addrStr) {
|
|
ret = true;
|
|
break;
|
|
}
|
|
}
|
|
return ret;
|
|
};
|
|
|
|
Wallet.prototype.getBalance = function(safe, cb) {
|
|
var balance = 0;
|
|
var balanceByAddr = {};
|
|
var isMain = {};
|
|
var COIN = bitcore.util.COIN;
|
|
var f = safe ? this.getSafeUnspent.bind(this) : this.getUnspent.bind(this);
|
|
f(function(utxos) {
|
|
for (var i = 0; i < utxos.length; i++) {
|
|
var u = utxos[i];
|
|
var amt = u.amount * COIN;
|
|
balance += amt;
|
|
balanceByAddr[u.address] = (balanceByAddr[u.address] || 0) + amt;
|
|
}
|
|
|
|
// we multiply and divide by COIN to avoid rounding errors when adding
|
|
for (var a in balanceByAddr) {
|
|
balanceByAddr[a] = balanceByAddr[a] / COIN;
|
|
}
|
|
balance = balance / COIN;
|
|
return cb(balance, balanceByAddr, isMain);
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.getUnspent = function(cb) {
|
|
this.blockchain.getUnspent(this.getAddressesStr(), function(unspentList) {
|
|
return cb(unspentList);
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.getSafeUnspent = function(cb) {
|
|
var self = this;
|
|
this.blockchain.getUnspent(this.getAddressesStr(), function(unspentList) {
|
|
|
|
var ret = [];
|
|
var maxRejectCount = self.totalCopayers - self.requiredCopayers;
|
|
var uu = self.txProposals.getUsedUnspent(maxRejectCount);
|
|
|
|
for (var i in unspentList) {
|
|
if (uu.indexOf(unspentList[i].txid) === -1)
|
|
ret.push(unspentList[i]);
|
|
}
|
|
|
|
return cb(ret);
|
|
});
|
|
};
|
|
|
|
|
|
Wallet.prototype.createTx = function(toAddress, amountSatStr, opts, cb) {
|
|
var self = this;
|
|
if (typeof opts === 'function') {
|
|
cb = opts;
|
|
opts = {};
|
|
}
|
|
opts = opts || {};
|
|
|
|
if (typeof opts.spendUnconfirmed === 'undefined') {
|
|
opts.spendUnconfirmed = this.spendUnconfirmed;
|
|
}
|
|
|
|
self.getSafeUnspent(function(unspentList) {
|
|
var ntxid = self.createTxSync(toAddress, amountSatStr, unspentList, opts);
|
|
if (ntxid) {
|
|
self.sendPublicKeyRing(); // For the new change Address
|
|
self.sendTxProposals(null, ntxid);
|
|
self.store();
|
|
self.emit('txProposalsUpdated');
|
|
}
|
|
return cb();
|
|
});
|
|
};
|
|
|
|
Wallet.prototype.createTxSync = function(toAddress, amountSatStr, utxos, opts) {
|
|
var pkr = this.publicKeyRing;
|
|
var priv = this.privateKey;
|
|
opts = opts || {};
|
|
|
|
var amountSat = bitcore.bignum(amountSatStr);
|
|
|
|
if (!pkr.isComplete()) {
|
|
throw new Error('publicKeyRing is not complete');
|
|
}
|
|
|
|
if (!opts.remainderOut) {
|
|
opts.remainderOut = {
|
|
address: this.generateAddress(true).toString()
|
|
};
|
|
}
|
|
|
|
var b = new Builder(opts)
|
|
.setUnspent(utxos)
|
|
.setHashToScriptMap(pkr.getRedeemScriptMap())
|
|
.setOutputs([{
|
|
address: toAddress,
|
|
amountSat: amountSat
|
|
}]);
|
|
|
|
var signRet;
|
|
if (priv) {
|
|
b.sign(priv.getAll(pkr.addressIndex, pkr.changeAddressIndex));
|
|
}
|
|
var myId = this.getMyCopayerId();
|
|
var now = Date.now();
|
|
|
|
var me = {};
|
|
if (priv && b.signaturesAdded) me[myId] = now;
|
|
|
|
var meSeen = {};
|
|
if (priv) meSeen[myId] = now;
|
|
|
|
var data = {
|
|
signedBy: me,
|
|
seenBy: meSeen,
|
|
creator: myId,
|
|
createdTs: now,
|
|
builder: b,
|
|
};
|
|
|
|
return this.txProposals.add(data);
|
|
};
|
|
|
|
Wallet.prototype.disconnect = function() {
|
|
this.log('## DISCONNECTING');
|
|
this.network.disconnect();
|
|
};
|
|
|
|
Wallet.prototype.getNetwork = function() {
|
|
return this.network;
|
|
};
|
|
|
|
module.exports = require('soop')(Wallet);
|