change Wallet to PublicKeyRing

This commit is contained in:
Matias Alejo Garcia 2014-04-04 15:57:28 -03:00
parent 88d8ad285c
commit d33cf4a751
3 changed files with 184 additions and 294 deletions

View File

@ -26,48 +26,47 @@ var PUBLIC_BRANCH = 'm/0/';
var CHANGE_BRANCH = 'm/1/';
function Wallet(opts) {
function PublicKeyRing(opts) {
opts = opts || {};
this.network = opts.network === 'livenet' ?
bitcore.networks.livenet : bitcore.networks.testnet;
this.requiredCosigners = opts.requiredCosigners || 3;
this.totalCosigners = opts.totalCosigners || 5;
this.requiredCopayers = opts.requiredCopayers || 3;
this.totalCopayers = opts.totalCopayers || 5;
this.id = opts.id || Wallet.getRandomId();
this.id = opts.id || PublicKeyRing.getRandomId();
this.dirty = 1;
this.cosignersWallets = [];
this.copayersWallets = [];
this.bip32 = new BIP32(opts.bytes || this.network.name);
this.changeAddressIndex=0;
this.addressIndex=0;
}
Wallet.getRandomId = function () {
PublicKeyRing.getRandomId = function () {
return buffertools.toHex(coinUtil.generateNonce());
};
Wallet.decrypt = function (passphrase, encPayload) {
PublicKeyRing.decrypt = function (passphrase, encPayload) {
log('[wallet.js.35] TODO READ: passphrase IGNORED');
return encPayload;
};
Wallet.encrypt = function (passphrase, payload) {
PublicKeyRing.encrypt = function (passphrase, payload) {
log('[wallet.js.92] TODO: passphrase IGNORED');
return payload;
};
Wallet.read = function (id, passphrase) {
PublicKeyRing.read = function (id, passphrase) {
var encPayload = storage.read(id);
if (!encPayload)
throw new Error('Could not find wallet data');
var data;
try {
data = JSON.parse( Wallet.decrypt( passphrase, encPayload ));
data = JSON.parse( PublicKeyRing.decrypt( passphrase, encPayload ));
} catch (e) {
throw new Error('error in storage: '+ e.toString());
return;
@ -80,12 +79,12 @@ Wallet.read = function (id, passphrase) {
bitcore.networks.livenet : bitcore.networks.testnet
};
var w = new Wallet(config);
var w = new PublicKeyRing(config);
w.requiredCosigners = data.neededCosigners;
w.totalCosigners = data.totalCosigners;
w.cosignersWallets = data.cosignersExtPubKeys.map( function (pk) {
return new Wallet({bytes:pk, network: w.network.name});
w.requiredCopayers = data.neededCopayers;
w.totalCopayers = data.totalCopayers;
w.copayersWallets = data.copayersExtPubKeys.map( function (pk) {
return new PublicKeyRing({bytes:pk, network: w.network.name});
});
w.dirty = 0;
@ -93,57 +92,47 @@ Wallet.read = function (id, passphrase) {
return w;
};
Wallet.prototype.serialize = function () {
PublicKeyRing.prototype.serialize = function () {
return JSON.stringify({
id: this.id,
network: this.network.name,
requiredCosigners: this.neededCosigners,
totalCosigners: this.totalCosigners,
cosignersExtPubKeys: this.cosignersWallets.map( function (b) {
requiredCopayers: this.neededCopayers,
totalCopayers: this.totalCopayers,
copayersExtPubKeys: this.copayersWallets.map( function (b) {
return b.getMasterExtendedPubKey();
}),
priv: this.getMasterExtendedPrivKey(),
});
};
Wallet.prototype.store = function (passphrase) {
PublicKeyRing.prototype.store = function (passphrase) {
if (!this.id)
throw new Error('wallet has no id');
storage.save(this.id, Wallet.encrypt(passphrase,this.serialize()));
storage.save(this.id, PublicKeyRing.encrypt(passphrase,this.serialize()));
this.dirty = 0;
return true;
};
Wallet.prototype.registeredCosigners = function () {
if (! this.cosignersWallets) return 1;
PublicKeyRing.prototype.registeredCopayers = function () {
if (! this.copayersWallets) return 1;
// 1 is self.
return 1 + this.cosignersWallets.length;
return 1 + this.copayersWallets.length;
};
Wallet.prototype.getMasterExtendedPrivKey = function () {
if (!this.bip32)
throw new Error('no priv key defined on the wallet');
return this.bip32.extended_private_key_string();
PublicKeyRing.prototype.getMasterExtendedPubKey = function () {
return this.bip32.extendedPublicKeyString();
};
Wallet.prototype.getMasterExtendedPubKey = function () {
return this.bip32.extended_public_key_string();
PublicKeyRing.prototype.haveAllRequiredPubKeys = function () {
return this.registeredCopayers() === this.totalCopayers;
};
Wallet.prototype.haveAllRequiredPubKeys = function () {
return this.registeredCosigners() === this.totalCosigners;
};
Wallet.prototype._checkKeys = function() {
PublicKeyRing.prototype._checkKeys = function() {
if (!this.haveAllRequiredPubKeys())
throw new Error('dont have required keys yet');
@ -151,26 +140,26 @@ Wallet.prototype._checkKeys = function() {
// should receive an array also?
Wallet.prototype.addCosignerExtendedPubKey = function (newEpk) {
PublicKeyRing.prototype.addCopayerExtendedPubKey = function (newEpk) {
if (this.haveAllRequiredPubKeys())
throw new Error('already have all required key:' + this.totalCosigners);
throw new Error('already have all required key:' + this.totalCopayers);
if (this.getMasterExtendedPubKey() === newEpk)
throw new Error('already have that key (self key)');
this.cosignersWallets.forEach(function(b){
this.copayersWallets.forEach(function(b){
if (b.getMasterExtendedPubKey() === newEpk)
throw new Error('already have that key');
});
this.cosignersWallets.push(new Wallet({bytes:newEpk, network: this.network.name } ));
this.copayersWallets.push(new PublicKeyRing({bytes:newEpk, network: this.network.name } ));
this.dirty = 1;
};
Wallet.prototype.getPubKey = function (index,isChange) {
PublicKeyRing.prototype.getPubKey = function (index,isChange) {
var path = (isChange ? CHANGE_BRANCH : PUBLIC_BRANCH) + index;
var bip32 = this.bip32.derive(path);
@ -178,49 +167,19 @@ Wallet.prototype.getPubKey = function (index,isChange) {
return pub;
};
Wallet.prototype.getCosignersPubKeys = function (index, isChange) {
PublicKeyRing.prototype.getCopayersPubKeys = function (index, isChange) {
this._checkKeys();
var pubKeys = [];
var l = this.cosignersWallets.length;
var l = this.copayersWallets.length;
for(var i=0; i<l; i++) {
pubKeys[i] = this.cosignersWallets[i].getPubKey(index, isChange);
pubKeys[i] = this.copayersWallets[i].getPubKey(index, isChange);
}
return pubKeys;
};
Wallet.prototype.getCosignersSortedPubKeys = function(index, isChange) {
var self = this;
var pubKeys = self.getCosignersPubKeys(index, isChange);
//sort lexicographically, i.e. as strings, i.e. alphabetically
// From https://github.com/ryanxcharles/treasure/blob/master/treasure.js
return pubKeys.sort(function(buf1, buf2) {
var len = buf1.length > buf1.length ? buf1.length : buf2.length;
for (var i = 0; i <= len; i++) {
if (buf1[i] === undefined)
return -1; //shorter strings come first
if (buf2[i] === undefined)
return 1;
if (buf1[i] < buf2[i])
return -1;
if (buf1[i] > buf2[i])
return 1;
else
continue;
}
return 0;
});
};
Wallet.prototype.getAddress = function (index, isChange) {
PublicKeyRing.prototype.getAddress = function (index, isChange) {
if ( (isChange && index > this.changeAddressIndex)
|| (!isChange && index > this.addressIndex)) {
@ -228,10 +187,9 @@ Wallet.prototype.getAddress = function (index, isChange) {
throw new Error('index out of bound');
}
var pubKeys = this.getCosignersSortedPubKeys(index, isChange);
var pubKeys = this.getCopayersPubKeys();
var version = this.network.addressScript;
var script = Script.createMultisig(this.requiredCosigners, pubKeys);
var script = Script.createMultisig(this.requiredCopayers, pubKeys);
var buf = script.buffer;
var hash = coinUtil.sha256ripe160(buf);
var addr = new Address(version, hash);
@ -239,11 +197,11 @@ Wallet.prototype.getAddress = function (index, isChange) {
return addrStr;
};
Wallet.prototype.createAddress = function(isChange) {
//generate a new address, update index.
PublicKeyRing.prototype.generateAddress = function(isChange) {
var ret =
this.getAddress(isChange ? this.changeAddressIndex : this.addressIndex, isChange);
if (isChange)
this.addressIndex++;
else
@ -253,7 +211,7 @@ Wallet.prototype.createAddress = function(isChange) {
};
Wallet.prototype.getAddresses = function() {
PublicKeyRing.prototype.getAddresses = function() {
var ret = [];
for (var i=0; i<this.changeAddressIndex; i++) {
@ -266,16 +224,4 @@ Wallet.prototype.getAddresses = function() {
return ret;
};
Wallet.prototype.createTx = function(utxos,outs, changeAddress) {
var opts = {
remainderAddress: changeAddress || this.createAddress(1),
};
return Transaction.create(utxos, outs, opts);
};
// Input: Bitcore's Transaction, sign with ownPK
// return partially signed or fully signed tx
Wallet.prototype.signTx = function (tx) {
};
module.exports = require('soop')(Wallet);
module.exports = require('soop')(PublicKeyRing);

140
test/test.publickeyring.js Normal file
View File

@ -0,0 +1,140 @@
'use strict';
var chai = chai || require('chai');
var should = chai.should();
var bitcore = bitcore || require('../node_modules/bitcore');
var Address = bitcore.Address;
var buffertools = bitcore.buffertools;
var copay = copay || {};
var fakeStorage = require('./FakeStorage');
var PublicKeyRing = copay.PublicKeyRing || require('soop').load('../js/models/PublicKeyRing', {Storage: fakeStorage});
var aMasterPubKey = 'tprv8ZgxMBicQKsPdSVTiWXEqCCzqRaRr9EAQdn5UVMpT9UHX67Dh1FmzEMbavPumpAicsUm2XvC6NTdcWB89yN5DUWx5HQ7z3KByUg7Ht74VRZ';
var config = {
network:'livenet',
};
var createW = function (network, bytes) {
var config = {
network: network || 'livenet',
};
if (bytes) config.bytes = bytes;
var w = new PublicKeyRing(config);
should.exist(w);
var copayers = [];
for(var i=0; i<4; i++) {
delete config['bytes'];
var c = new PublicKeyRing(config);
w.haveAllRequiredPubKeys().should.equal(false);
w.addCopayerExtendedPubKey(c.getMasterExtendedPubKey());
copayers.push(c);
}
return {w:w, copayers: copayers};
};
describe('PublicKeyRing model', function() {
it('should create an instance (livenet)', function () {
var w = new PublicKeyRing({
network: config.network
});
should.exist(w);
w.network.name.should.equal('livenet');
});
it('should create an instance (testnet)', function () {
var w2 = new PublicKeyRing();
should.exist(w2);
w2.network.name.should.equal('testnet');
});
it('should create an master pub key', function () {
var w2 = new PublicKeyRing(config);
should.exist(w2);
should.exist(w2.getMasterExtendedPubKey());
});
it('should fail to generate shared pub keys wo extended key', function () {
var w2 = new PublicKeyRing(config);
should.exist(w2);
w2.registeredCopayers().should.equal(1);
w2.haveAllRequiredPubKeys().should.equal(false);
w2.getAddress.bind(false).should.throw();
});
it('should add and check when adding shared pub keys', function () {
var k = createW();
var w = k.w;
var copayers = k.copayers;
w.haveAllRequiredPubKeys().should.equal(true);
w.addCopayerExtendedPubKey.bind(w.getMasterExtendedPubKey()).should.throw();
w.addCopayerExtendedPubKey.bind(copayers[0].getMasterExtendedPubKey()).should.throw();
w.addCopayerExtendedPubKey.bind((new PublicKeyRing(config)).getMasterExtendedPubKey()).should.throw();
});
it('show be able to store and retrieve', function () {
var k = createW();
var w = k.w;
var copayers = k.copayers;
w.store().should.equal(true);
var ID = w.id;
delete w['id'];
w.store.bind().should.throw();
var w2 = PublicKeyRing.read(ID);
w2.haveAllRequiredPubKeys().should.equal(true);
w2.addCopayerExtendedPubKey.bind(w.getMasterExtendedPubKey()).should.throw();
w2.addCopayerExtendedPubKey.bind(copayers[0].getMasterExtendedPubKey()).should.throw();
w2.addCopayerExtendedPubKey.bind((new PublicKeyRing(config)).getMasterExtendedPubKey()).should.throw();
});
it('should generate some p2sh addresses', function () {
var k = createW();
var w = k.w;
for(var isChange=0; isChange<2; isChange++) {
for(var i=0; i<5; i++) {
var addr = w.generateAddress(isChange);
var a = new Address(addr);
a.isValid().should.equal(true);
a.isScript().should.equal(true);
a.network().name.should.equal('livenet');
}
}
});
it('should return PublicKeyRing addresses', function () {
var k = createW();
var w = k.w;
var a = w.getAddresses();
a.length.should.equal(0);
for(var isChange=0; isChange<2; isChange++)
for(var i=0; i<6; i++)
w.generateAddress(isChange);
var as = w.getAddresses();
as.length.should.equal(12);
for(var i in as) {
var a = new Address(as[i]);
a.isValid().should.equal(true);
}
});
});

View File

@ -1,196 +0,0 @@
'use strict';
var chai = chai || require('chai');
var should = chai.should();
var bitcore = bitcore || require('../node_modules/bitcore');
var Address = bitcore.Address;
var buffertools = bitcore.buffertools;
var cosign = cosign || {};
var fakeStorage = require('./FakeStorage');
var Wallet = cosign.Wallet || require('soop').load('../js/models/Wallet', {Storage: fakeStorage});
var aMasterPrivKey = 'tprv8ZgxMBicQKsPdSVTiWXEqCCzqRaRr9EAQdn5UVMpT9UHX67Dh1FmzEMbavPumpAicsUm2XvC6NTdcWB89yN5DUWx5HQ7z3KByUg7Ht74VRZ';
var config = {
network:'livenet',
};
var createW = function (network, bytes) {
var config = {
network: network || 'livenet',
};
if (bytes) config.bytes = bytes;
var w = new Wallet(config);
should.exist(w);
var cosigners = [];
for(var i=0; i<4; i++) {
delete config['bytes'];
var c = new Wallet(config);
w.haveAllRequiredPubKeys().should.equal(false);
w.addCosignerExtendedPubKey(c.getMasterExtendedPubKey());
cosigners.push(c);
}
return {w:w, cosigners: cosigners};
};
describe('Wallet model', function() {
it('should create an instance (livenet)', function () {
var w = new Wallet({
network: config.network
});
should.exist(w);
w.network.name.should.equal('livenet');
});
it('should create an instance (testnet)', function () {
var w2 = new Wallet();
should.exist(w2);
w2.network.name.should.equal('testnet');
});
it('should throw master priv key', function () {
var w2 = new Wallet(config);
should.exist(w2);
w2.getMasterExtendedPrivKey.bind().should.throw();
});
it('should create an master priv key', function () {
var w2 = new Wallet(config);
should.exist(w2);
should.exist(w2.getMasterExtendedPrivKey());
});
it('should create an master pub key', function () {
var w2 = new Wallet(config);
should.exist(w2);
should.exist(w2.getMasterExtendedPubKey());
});
it('should fail to generate shared pub keys wo extended key', function () {
var w2 = new Wallet(config);
should.exist(w2);
w2.registeredCosigners().should.equal(1);
w2.haveAllRequiredPubKeys().should.equal(false);
w2.getAddress.bind(false).should.throw();
});
it('should add and check when adding shared pub keys', function () {
var k = createW();
var w = k.w;
var cosigners = k.cosigners;
w.haveAllRequiredPubKeys().should.equal(true);
w.addCosignerExtendedPubKey.bind(w.getMasterExtendedPubKey()).should.throw();
w.addCosignerExtendedPubKey.bind(cosigners[0].getMasterExtendedPubKey()).should.throw();
w.addCosignerExtendedPubKey.bind((new Wallet(config)).getMasterExtendedPubKey()).should.throw();
});
it('show be able to store and retrieve', function () {
var k = createW();
var w = k.w;
var cosigners = k.cosigners;
w.store().should.equal(true);
var ID = w.id;
delete w['id'];
w.store.bind().should.throw();
var w2 = Wallet.read(ID);
w2.haveAllRequiredPubKeys().should.equal(true);
w2.addCosignerExtendedPubKey.bind(w.getMasterExtendedPubKey()).should.throw();
w2.addCosignerExtendedPubKey.bind(cosigners[0].getMasterExtendedPubKey()).should.throw();
w2.addCosignerExtendedPubKey.bind((new Wallet(config)).getMasterExtendedPubKey()).should.throw();
});
it('should create some p2sh addresses', function () {
var k = createW();
var w = k.w;
for(var isChange=0; isChange<2; isChange++) {
for(var i=0; i<5; i++) {
var addr = w.createAddress(isChange);
var a = new Address(addr);
a.isValid().should.equal(true);
a.isScript().should.equal(true);
a.network().name.should.equal('livenet');
}
}
});
it('should return wallet addresses', function () {
var k = createW();
var w = k.w;
var a = w.getAddresses();
a.length.should.equal(0);
for(var isChange=0; isChange<2; isChange++)
for(var i=0; i<6; i++)
w.createAddress(isChange);
var as = w.getAddresses();
as.length.should.equal(12);
for(var i in as) {
var a = new Address(as[i]);
a.isValid().should.equal(true);
}
});
it('should create a TX', function () {
var k = createW('testnet', aMasterPrivKey);
var w = k.w;
var as=[], j=0;
for(var isChange=0; isChange<2; isChange++)
for(var i=0; i<6; i++)
as[j++] = w.createAddress(isChange);
var utxos = [
{
address: as[0],
txid: "2ac165fa7a3a2b535d106a0041c7568d03b531e58aeccdd3199d7289ab12cfc1",
scriptPubKey: "76a9146ce4e1163eb18939b1440c42844d5f0261c0338288ac",
vout: 1,
amount: 1,
confirmations: 3
},
{
address: as[1],
txid: "2ac165fa7a3a2b535d106a0041c7568d03b531e58aeccdd3199d7289ab12cfc1",
scriptPubKey: "76a9146ce4e1163eb18939b1440c42844d5f0261c0338288ac",
vout: 1,
amount: 1.01,
confirmations: 7
},
];
var outs=[{
address: 'mfwSGKeLEGdd1YwsZ1TkoEeKNk8TTWqXLC',
amount: 1.5,
}];
var ret = w.createTx(utxos, outs, '2Mu1GwdF9X1LLxXNxtfNChi5ngBMJxi2csv');
var tx=ret.tx;
should.exist(tx);
tx.ins.length.should.equal(2);
tx.outs.length.should.equal(2);
buffertools.toHex(tx.serialize()).should.equal('0100000002c1cf12ab89729d19d3cdec8ae531b5038d56c741006a105d532b3a7afa65c12a0100000000ffffffffc1cf12ab89729d19d3cdec8ae531b5038d56c741006a105d532b3a7afa65c12a0100000000ffffffff0280d1f008000000001976a91404a154d56e5455dc87f15e293b8872f31d03a12c88acb00b0a030000000017a914134ce4e4ea92b6cb0008803cafe25f197ff1dcb98700000000');
});
});