284 lines
8.8 KiB
JavaScript
284 lines
8.8 KiB
JavaScript
'use strict';
|
|
|
|
var rlp = require('rlp');
|
|
|
|
var Contract = require('./contract.js');
|
|
var providers = require('./providers.js');
|
|
var SigningKey = require('./signing-key.js');
|
|
|
|
var utils = require('./utils.js');
|
|
|
|
// This ensures we inject a setImmediate into the global space, which
|
|
// dramatically improves the performance of the scrypt PBKDF.
|
|
require('setimmediate');
|
|
|
|
var transactionFields = [
|
|
{name: 'nonce', maxLength: 32, },
|
|
{name: 'gasPrice', maxLength: 32, },
|
|
{name: 'gasLimit', maxLength: 32, },
|
|
{name: 'to', length: 20, },
|
|
{name: 'value', maxLength: 32, },
|
|
{name: 'data'},
|
|
];
|
|
|
|
function Wallet(privateKey, provider) {
|
|
if (!(this instanceof Wallet)) { throw new Error('missing new'); }
|
|
|
|
// Make sure we have a valid signing key
|
|
var signingKey = privateKey;
|
|
if (!(privateKey instanceof SigningKey)) {
|
|
signingKey = new SigningKey(privateKey);
|
|
}
|
|
utils.defineProperty(this, 'privateKey', signingKey.privateKey);
|
|
|
|
Object.defineProperty(this, 'provider', {
|
|
enumerable: true,
|
|
get: function() { return provider; },
|
|
set: function(value) {
|
|
if (value !== null && !providers.isProvider(value)) {
|
|
throw new Error('invalid provider');
|
|
}
|
|
provider = value;
|
|
}
|
|
});
|
|
|
|
Object.defineProperty(this, '_provider', {
|
|
enumerable: true,
|
|
get: function() {
|
|
if (!provider) { throw new Error('missing provider'); }
|
|
return provider;
|
|
},
|
|
});
|
|
|
|
if (provider !== null) {
|
|
|
|
// If no provider was provided, check for metamask or ilk
|
|
if (provider === undefined) {
|
|
if (global.web3 && global.web3.currentProvider && global.web3.currentProvider.sendAsync) {
|
|
this.provider = new providers.Web3Provider(global.web3.currentProvider);
|
|
}
|
|
|
|
// An Ethereum RPC node
|
|
} else if (typeof(provider) === 'string' && provider.match(/^https?:\/\//)) {
|
|
this.provider = new providers.HttpProvider(provider);
|
|
|
|
} else {
|
|
this.provider = provider;
|
|
}
|
|
}
|
|
|
|
utils.defineProperty(this, 'address', signingKey.address);
|
|
|
|
utils.defineProperty(this, 'sign', function(transaction) {
|
|
var raw = [];
|
|
transactionFields.forEach(function(fieldInfo) {
|
|
var value = transaction[fieldInfo.name] || (new Buffer(0));
|
|
value = utils.hexOrBuffer(utils.hexlify(value), fieldInfo.name);
|
|
|
|
// Fixed-width field
|
|
if (fieldInfo.length && value.length !== fieldInfo.length && value.length > 0) {
|
|
var error = new Error('invalid ' + fieldInfo.name);
|
|
error.reason = 'wrong length';
|
|
error.value = value;
|
|
throw error;
|
|
}
|
|
|
|
// Variable-width (with a maximum)
|
|
if (fieldInfo.maxLength) {
|
|
value = utils.stripZeros(value);
|
|
if (value.length > fieldInfo.maxLength) {
|
|
var error = new Error('invalid ' + fieldInfo.name);
|
|
error.reason = 'too long';
|
|
error.value = value;
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
raw.push(value);
|
|
});
|
|
|
|
var digest = utils.sha3(rlp.encode(raw));
|
|
|
|
var signature = signingKey.signDigest(digest);
|
|
var s = signature.s;
|
|
var v = signature.recoveryParam;
|
|
|
|
raw.push(new Buffer([27 + v]));
|
|
raw.push(utils.bnToBuffer(signature.r));
|
|
raw.push(utils.bnToBuffer(s));
|
|
|
|
return ('0x' + rlp.encode(raw).toString('hex'));
|
|
});
|
|
}
|
|
|
|
utils.defineProperty(Wallet, 'parseTransaction', function(rawTransaction) {
|
|
rawTransaction = utils.hexOrBuffer(rawTransaction, 'rawTransaction');
|
|
var signedTransaction = rlp.decode(rawTransaction);
|
|
|
|
var raw = [];
|
|
|
|
var transaction = {};
|
|
transactionFields.forEach(function(fieldInfo, index) {
|
|
transaction[fieldInfo.name] = signedTransaction[index];
|
|
raw.push(signedTransaction[index]);
|
|
});
|
|
|
|
if (transaction.to) {
|
|
if (transaction.to.length === 0) {
|
|
delete transaction.to;
|
|
} else {
|
|
transaction.to = utils.getAddress('0x' + transaction.to.toString('hex'));
|
|
}
|
|
}
|
|
|
|
['gasPrice', 'gasLimit', 'nonce', 'value'].forEach(function(name) {
|
|
if (!transaction[name]) { return; }
|
|
if (transaction[name].length === 0) {
|
|
transaction[name] = new utils.BN(0);
|
|
} else {
|
|
transaction[name] = new utils.BN(transaction[name].toString('hex'), 16);
|
|
}
|
|
});
|
|
|
|
/* @TODO: Maybe? In the future, all nonces stored as numbers? (obviously, major version change)
|
|
if (transaction.nonce) {
|
|
transaction.nonce = transaction.nonce.toNumber()
|
|
}
|
|
*/
|
|
|
|
if (signedTransaction.length > 6 && signedTransaction[6].length === 1 &&
|
|
signedTransaction[7].length >= 1 && signedTransaction[7].length <= 32 &&
|
|
signedTransaction[8].length >= 1 && signedTransaction[7].length <= 32) {
|
|
|
|
transaction.v = signedTransaction[6][0];
|
|
transaction.r = signedTransaction[7];
|
|
transaction.s = signedTransaction[8];
|
|
|
|
var digest = utils.sha3(rlp.encode(raw));
|
|
try {
|
|
transaction.from = SigningKey.recover(digest, transaction.r, transaction.s, transaction.v - 27);
|
|
} catch (error) { }
|
|
}
|
|
|
|
return transaction;
|
|
});
|
|
|
|
utils.defineProperty(Wallet.prototype, 'getBalance', function(blockNumber) {
|
|
var provider = this._provider;
|
|
|
|
var self = this;
|
|
return new Promise(function(resolve, reject) {
|
|
provider.getBalance(self.address, blockNumber).then(function(balance) {
|
|
resolve(balance);
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
});
|
|
});
|
|
|
|
utils.defineProperty(Wallet.prototype, 'getTransactionCount', function(blockNumber) {
|
|
var provider = this._provider;
|
|
|
|
var self = this;
|
|
return new Promise(function(resolve, reject) {
|
|
provider.getTransactionCount(self.address, blockNumber).then(function(transactionCount) {
|
|
resolve(transactionCount);
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
});
|
|
});
|
|
|
|
utils.defineProperty(Wallet.prototype, 'estimateGas', function(transaction) {
|
|
var provider = this._provider;
|
|
|
|
transaction = utils.cloneObject(transaction);
|
|
if (transaction.from == null) { transaction.from = this.address; }
|
|
|
|
return new Promise(function(resolve, reject) {
|
|
provider.estimateGas(transaction).then(function(gasEstimate) {
|
|
resolve(gasEstimate);
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
});
|
|
});
|
|
|
|
utils.defineProperty(Wallet.prototype, 'sendTransaction', function(transaction) {
|
|
var gasLimit = transaction.gasLimit;
|
|
if (gasLimit == null) { gasLimit = 3000000; }
|
|
|
|
var self = this;
|
|
|
|
var provider = this._provider;
|
|
|
|
var gasPrice = new Promise(function(resolve, reject) {
|
|
if (transaction.gasPrice) {
|
|
return resolve(transaction.gasPrice);
|
|
}
|
|
provider.getGasPrice().then(function(gasPrice) {
|
|
resolve(gasPrice);
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
});
|
|
|
|
var nonce = new Promise(function(resolve, reject) {
|
|
if (transaction.nonce) {
|
|
return resolve(transaction.nonce);
|
|
}
|
|
provider.getTransactionCount(self.address, 'pending').then(function(transactionCount) {
|
|
resolve(transactionCount);
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
});
|
|
|
|
return new Promise(function(resolve, reject) {
|
|
Promise.all([gasPrice, nonce]).then(function(results) {
|
|
var signedTransaction = self.sign({
|
|
to: transaction.to,
|
|
gasLimit: gasLimit,
|
|
gasPrice: results[0],
|
|
nonce: results[1],
|
|
value: transaction.value
|
|
});
|
|
|
|
provider.sendTransaction(signedTransaction).then(function(txid) {
|
|
resolve(txid);
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
|
|
}, function(error) {
|
|
reject(error);
|
|
});
|
|
});
|
|
});
|
|
|
|
utils.defineProperty(Wallet.prototype, 'send', function(address, amountWei, options) {
|
|
address = utils.getAddress(address);
|
|
if (utils.BN.isBN(amountWei)) {
|
|
amountWei = '0x' + utils.bnToBuffer(amountWei).toString('hex');
|
|
}
|
|
if (!utils.isHexString(amountWei)) { throw new Error('invalid amountWei'); }
|
|
|
|
if (!options) { options = {}; }
|
|
|
|
return this.sendTransaction({
|
|
to: address,
|
|
gasLimit: options.gasLimit,
|
|
gasPrice: options.gasPrice,
|
|
nonce: options.nonce,
|
|
value: amountWei,
|
|
});
|
|
});
|
|
|
|
utils.defineProperty(Wallet.prototype, 'getContract', function(address, abi) {
|
|
return new Contract(this, address, new Contract.Interface(abi));
|
|
});
|
|
|
|
utils.defineProperty(Wallet, '_Contract', Contract);
|
|
|
|
module.exports = Wallet;
|