LBRYCommunity/bitcoinjs-lib
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Started work on wallet, and added hex serialization

Browse source
This commit is contained in:
vub 2013-11-20 13:00:49 -05:00
parent 5e2a9f0c26
commit 93e50e9433
3 changed files with 41 additions and 95 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
bitcoinjs-min.js vendored
View file

File diff suppressed because one or more lines are too long

4
src/transaction.js
View file

@ -163,6 +163,10 @@ Transaction.prototype.serialize = function ()
return buffer;
};
Transaction.prototype.serializeHex = function() {
return conv.bytesToHex(this.serialize());
}
var OP_CODESEPARATOR = 171;
var SIGHASH_ALL = 1;

130
src/wallet.js
View file

@ -5,10 +5,15 @@ var util = require('./util');
var BigInteger = require('./jsbn/jsbn');
var BIP32key = require('./bip32');
var Transaction = require('./transaction').Transaction;
var TransactionIn = require('./transaction').TransactionIn;
var TransactionOut = require('./transaction').TransactionOut;
var SecureRandom = require('./jsbn/rng');
var rng = new SecureRandom();
var Wallet = function () {
// Keychain
//
@ -18,6 +23,7 @@ var Wallet = function () {
// Any functions accessing this value therefore have to be defined in
// the closure of this constructor.
var keys = [];
var masterkey = null;
// Public hashes of our keys
this.addressHashes = [];
@ -28,51 +34,19 @@ var Wallet = function () {
// Other fields
this.addressPointer = 0;
this.genMasterkey = function(seed) {
if (!seed) {
var seedBytes = new Array(32);
rng.nextBytes(seedBytes);
seed = conv.bytesToString(seedBytes)
}
masterkey = new BIP32key(seed);
}
/**
* Add a key to the keychain.
*
* The corresponding public key can be provided as a second parameter. This
* adds it to the cache in the ECKey object and avoid the need to
* expensively calculate it later.
*/
this.addKey = function (key, pub) {
if (!(key instanceof ECKey)) {
key = new ECKey(key);
}
keys.push(key);
if (pub) {
if ("string" === typeof pub) {
pub = conv.hexToBytes(pub);
}
key.setPub(pub);
}
this.addressHashes.push(key.getBitcoinAddress().getHash());
};
/**
* Add multiple keys at once.
*/
this.addKeys = function (keys, pubs) {
if ("string" === typeof keys) {
keys = keys.split(',');
}
if ("string" === typeof pubs) {
pubs = pubs.split(',');
}
var i;
if (Array.isArray(pubs) && keys.length == pubs.length) {
for (i = 0; i < keys.length; i++) {
this.addKey(keys[i], pubs[i]);
}
} else {
for (i = 0; i < keys.length; i++) {
this.addKey(keys[i]);
}
}
};
this.generateAddress = function() {
keys.push(masterkey.ckd(keys.length))
}
/**
* Get the key chain.
@ -80,15 +54,22 @@ var Wallet = function () {
* Returns an array of hex-encoded private values.
*/
this.getKeys = function () {
var serializedWallet = [];
var keyExport = [];
for (var i = 0; i < keys.length; i++) {
serializedWallet.push(keys[i].toString());
keyExport.push(keys[i].toString());
}
return serializedWallet;
return keyExport;
};
this.privateSerialize = function() {
return {
masterkey: masterkey,
keys: this.getKeys()
}
}
/**
* Get the public keys.
*
@ -109,6 +90,7 @@ var Wallet = function () {
*/
this.clear = function () {
keys = [];
masterkey = null;
};
/**
@ -132,38 +114,13 @@ var Wallet = function () {
};
this.getCurAddress = function () {
if (keys[this.addressPointer]) {
return keys[this.addressPointer].getBitcoinAddress();
if (keys[keys.length - 1]) {
return keys[keys.length - 1].getBitcoinAddress();
} else {
return null;
}
};
/**
* Go to the next address.
*
* If there are no more new addresses available, one will be generated
* automatically.
*/
this.getNextAddress = function () {
if (keys.length === 0) {
this.generateAddress();
}
else {
}
/*
this.addressPointer++;
if (!keys[this.addressPointer]) {
this.generateAddress();
}
*/
// TODO(shtylman) this shit is trying to be too smart
// it is making a new address when it shouldn't
// it should just stop being so "smart" and just do what it is told
return keys[this.addressPointer].getBitcoinAddress();
};
/**
* Sign a hash with a key.
*
@ -197,10 +154,6 @@ var Wallet = function () {
};
};
Wallet.prototype.generateAddress = function () {
this.addKey(new ECKey());
};
// return unspent transactions
Wallet.prototype.unspentTx = function() {
return this.unspentOuts;
@ -227,7 +180,7 @@ Wallet.prototype.process = function (tx) {
for (k = 0; k < this.addressHashes.length; k++) {
// if our address, then we add the unspent out to a list of unspent outputs
if (this.addressHashes[k] === hash) {
this.unspentOuts.push({tx: tx, index: j, out: txout});
this.unspentOuts.push({tx: tx, index: j, output: txout});
break;
}
}
@ -264,7 +217,7 @@ Wallet.prototype.process = function (tx) {
};
Wallet.prototype.getBalance = function () {
return this.unspentOuts.reduce(function(t,o) { return t + o.out.value },0);
return this.unspentOuts.reduce(function(t,o) { return t + o.output.value },0);
};
Wallet.prototype.createSend = function (address, sendValue, feeValue) {
@ -275,7 +228,7 @@ Wallet.prototype.createSend = function (address, sendValue, feeValue) {
for (i = 0; i < this.unspentOuts.length; i++) {
var txout = this.unspentOuts[i];
selectedOuts.push(txout);
availableValue += txout.out.value;
availableValue += txout.output.value;
if (availableValue >= txValue) break;
}
@ -294,23 +247,12 @@ Wallet.prototype.createSend = function (address, sendValue, feeValue) {
sendTx.addOutput(address, sendValue);
if (changeValue > 0) {
sendTx.addOutput(this.getNextAddress(), changeValue);
sendTx.addOutput(this.getCurAddress(), changeValue);
}
var hashType = 1; // SIGHASH_ALL
for (i = 0; i < sendTx.ins.length; i++) {
var hash = sendTx.hashTransactionForSignature(selectedOuts[i].out.script, i, hashType);
var pubKeyHash = selectedOuts[i].out.script.simpleOutPubKeyHash();
// this changes because signing uses a random number generator
var signature = this.signWithKey(pubKeyHash, hash);
// Append hash type
signature.push(parseInt(hashType, 10));
sendTx.ins[i].script = Script.createInputScript(signature, this.getPubKeyFromHash(pubKeyHash));
}
sendTx.signWithKeys(this.getKeys(), selectedOuts, hashType)
return sendTx;
};