1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 09:52:26 +01:00
metamask-extension/app/scripts/lib/idStore.js

355 lines
9.7 KiB
JavaScript
Raw Normal View History

2016-02-11 02:44:46 +01:00
const EventEmitter = require('events').EventEmitter
const inherits = require('util').inherits
const Transaction = require('ethereumjs-tx')
2016-03-03 07:58:23 +01:00
const LightwalletKeyStore = require('eth-lightwallet').keystore
const LightwalletSigner = require('eth-lightwallet').signing
2016-02-11 02:44:46 +01:00
const async = require('async')
const clone = require('clone')
const extend = require('xtend')
const createId = require('web3-provider-engine/util/random-id')
const autoFaucet = require('./auto-faucet')
const configManager = require('./config-manager-singleton')
const DEFAULT_RPC = 'https://rawtestrpc.metamask.io/'
2016-02-11 02:44:46 +01:00
module.exports = IdentityStore
inherits(IdentityStore, EventEmitter)
function IdentityStore(ethStore) {
EventEmitter.call(this)
2016-02-11 02:44:46 +01:00
// we just use the ethStore to auto-add accounts
this._ethStore = ethStore
2016-03-03 07:58:23 +01:00
// lightwallet key store
this._keyStore = null
2016-03-03 07:58:23 +01:00
// lightwallet wrapper
this._idmgmt = null
this.hdPathString = "m/44'/60'/0'/0"
2016-02-11 02:44:46 +01:00
this._currentState = {
2016-02-11 02:44:46 +01:00
selectedAddress: null,
identities: {},
unconfTxs: {},
}
// not part of serilized metamask state - only kept in memory
this._unconfTxCbs = {}
2016-02-11 02:44:46 +01:00
}
//
// public
//
IdentityStore.prototype.createNewVault = function(password, entropy, cb){
delete this._keyStore
configManager.clearWallet()
this._createIdmgmt(password, null, entropy, (err) => {
2016-03-03 07:58:23 +01:00
if (err) return cb(err)
this._loadIdentities()
this._didUpdate()
this._autoFaucet()
configManager.setShowSeedWords(true)
var seedWords = this._idmgmt.getSeed()
2016-03-03 07:58:23 +01:00
cb(null, seedWords)
})
2016-02-17 09:55:57 +01:00
}
2016-03-15 21:39:12 +01:00
IdentityStore.prototype.recoverFromSeed = function(password, seed, cb){
this._createIdmgmt(password, seed, null, (err) => {
2016-03-15 21:39:12 +01:00
if (err) return cb(err)
this._loadIdentities()
this._didUpdate()
cb(null, this.getState())
2016-03-15 21:39:12 +01:00
})
}
2016-02-17 09:55:57 +01:00
2016-02-11 02:44:46 +01:00
IdentityStore.prototype.setStore = function(store){
this._ethStore = store
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype.clearSeedWordCache = function(cb) {
configManager.setShowSeedWords(false)
cb()
}
2016-02-17 09:55:57 +01:00
2016-02-11 02:44:46 +01:00
IdentityStore.prototype.getState = function(){
var seedWords = this.getSeedIfUnlocked()
var wallet = configManager.getWallet()
return clone(extend(this._currentState, {
isInitialized: !!configManager.getWallet() && !seedWords,
isUnlocked: this._isUnlocked(),
seedWords: seedWords,
2016-02-11 02:44:46 +01:00
}))
}
IdentityStore.prototype.getSeedIfUnlocked = function() {
var showSeed = configManager.getShouldShowSeedWords()
var idmgmt = this._idmgmt
var shouldShow = showSeed && !!idmgmt
var seedWords = shouldShow ? idmgmt.getSeed() : null
return seedWords
}
2016-02-11 02:44:46 +01:00
IdentityStore.prototype.getSelectedAddress = function(){
return this._currentState.selectedAddress
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype.setSelectedAddress = function(address){
if (!address) {
var addresses = this._getAddresses()
address = addresses[0]
}
this._currentState.selectedAddress = address
this._didUpdate()
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype.setLocked = function(cb){
delete this._keyStore
delete this._idmgmt
cb()
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype.submitPassword = function(password, cb){
this._tryPassword(password, (err) => {
2016-02-11 02:44:46 +01:00
if (err) return cb(err)
// load identities before returning...
this._loadIdentities()
2016-02-11 02:44:46 +01:00
cb()
})
}
2016-04-06 21:01:10 +02:00
IdentityStore.prototype.exportAccount = function(address, cb) {
var privateKey = this._idmgmt.exportPrivateKey(address)
cb(null, privateKey)
}
2016-03-03 07:58:23 +01:00
// comes from dapp via zero-client hooked-wallet provider
IdentityStore.prototype.addUnconfirmedTransaction = function(txParams, cb){
2016-02-11 02:44:46 +01:00
2016-03-03 07:58:23 +01:00
// create txData obj with parameters and meta data
var time = (new Date()).getTime()
var txId = createId()
var txData = {
id: txId,
txParams: txParams,
time: time,
status: 'unconfirmed',
}
this._currentState.unconfTxs[txId] = txData
console.log('addUnconfirmedTransaction:', txData)
2016-02-11 02:44:46 +01:00
2016-03-03 07:58:23 +01:00
// keep the cb around for after approval (requires user interaction)
this._unconfTxCbs[txId] = cb
2016-02-11 02:44:46 +01:00
2016-03-03 07:58:23 +01:00
// signal update
this._didUpdate()
return txId
2016-02-11 02:44:46 +01:00
}
2016-03-03 07:58:23 +01:00
// comes from metamask ui
IdentityStore.prototype.approveTransaction = function(txId, cb){
var txData = this._currentState.unconfTxs[txId]
2016-03-03 07:58:23 +01:00
var txParams = txData.txParams
var approvalCb = this._unconfTxCbs[txId] || noop
2016-03-03 07:58:23 +01:00
// accept tx
2016-02-13 02:57:10 +01:00
cb()
2016-03-03 07:58:23 +01:00
approvalCb(null, true)
// clean up
delete this._currentState.unconfTxs[txId]
delete this._unconfTxCbs[txId]
this._didUpdate()
2016-02-13 02:57:10 +01:00
}
2016-03-03 07:58:23 +01:00
// comes from metamask ui
2016-02-12 21:55:20 +01:00
IdentityStore.prototype.cancelTransaction = function(txId){
var txData = this._currentState.unconfTxs[txId]
var approvalCb = this._unconfTxCbs[txId] || noop
2016-03-03 07:58:23 +01:00
// reject tx
approvalCb(null, false)
// clean up
delete this._currentState.unconfTxs[txId]
delete this._unconfTxCbs[txId]
this._didUpdate()
2016-02-12 21:55:20 +01:00
}
2016-03-03 07:58:23 +01:00
// performs the actual signing, no autofill of params
IdentityStore.prototype.signTransaction = function(txParams, cb){
2016-02-11 02:44:46 +01:00
try {
2016-03-03 07:58:23 +01:00
console.log('signing tx...', txParams)
var rawTx = this._idmgmt.signTx(txParams)
2016-03-03 07:58:23 +01:00
cb(null, rawTx)
2016-02-11 02:44:46 +01:00
} catch (err) {
cb(err)
}
}
2016-03-03 07:58:23 +01:00
//
// private
//
2016-02-11 02:44:46 +01:00
IdentityStore.prototype._didUpdate = function(){
this.emit('update', this.getState())
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype._isUnlocked = function(){
var result = Boolean(this._keyStore) && Boolean(this._idmgmt)
2016-02-11 02:44:46 +01:00
return result
}
2016-02-17 09:55:57 +01:00
// load identities from keyStoreet
2016-02-11 02:44:46 +01:00
IdentityStore.prototype._loadIdentities = function(){
if (!this._isUnlocked()) throw new Error('not unlocked')
var addresses = this._getAddresses()
addresses.forEach((address, i) => {
2016-02-17 09:55:57 +01:00
// // add to ethStore
this._ethStore.addAccount(address)
2016-02-11 02:44:46 +01:00
// add to identities
var identity = {
name: 'Wallet ' + (i+1),
2016-02-11 02:44:46 +01:00
img: 'QmW6hcwYzXrNkuHrpvo58YeZvbZxUddv69ATSHY3BHpPdd',
address: address,
mayBeFauceting: this._mayBeFauceting(i),
2016-02-11 02:44:46 +01:00
}
this._currentState.identities[address] = identity
2016-02-11 02:44:46 +01:00
})
this._didUpdate()
2016-02-11 02:44:46 +01:00
}
// mayBeFauceting
// If on testnet, index 0 may be fauceting.
// The UI will have to check the balance to know.
// If there is no balance and it mayBeFauceting,
// then it is in fact fauceting.
IdentityStore.prototype._mayBeFauceting = function(i) {
var config = configManager.getProvider()
if (i === 0 &&
config.type === 'rpc' &&
config.rpcTarget === DEFAULT_RPC) {
return true
}
return false
}
2016-02-11 02:44:46 +01:00
//
// keyStore managment - unlocking + deserialization
//
IdentityStore.prototype._tryPassword = function(password, cb){
this._createIdmgmt(password, null, null, cb)
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype._createIdmgmt = function(password, seed, entropy, cb){
2016-02-11 02:44:46 +01:00
var keyStore = null
LightwalletKeyStore.deriveKeyFromPassword(password, (err, derivedKey) => {
2016-03-03 07:58:23 +01:00
if (err) return cb(err)
var serializedKeystore = configManager.getWallet()
2016-03-15 21:39:12 +01:00
if (seed) {
2016-04-01 23:02:02 +02:00
try {
keyStore = this._restoreFromSeed(password, seed, derivedKey)
} catch (e) {
return cb(e)
}
// returning user, recovering from storage
2016-03-15 21:39:12 +01:00
} else if (serializedKeystore) {
2016-03-31 19:24:39 +02:00
keyStore = LightwalletKeyStore.deserialize(serializedKeystore)
var isCorrect = keyStore.isDerivedKeyCorrect(derivedKey)
2016-03-03 07:58:23 +01:00
if (!isCorrect) return cb(new Error('Lightwallet - password incorrect'))
// first time here
2016-03-03 07:58:23 +01:00
} else {
keyStore = this._createFirstWallet(entropy, derivedKey)
2016-03-03 07:58:23 +01:00
}
this._keyStore = keyStore
this._idmgmt = new IdManagement({
keyStore: keyStore,
derivedKey: derivedKey,
hdPathSTring: this.hdPathString,
})
2016-03-03 07:58:23 +01:00
cb()
})
2016-02-11 02:44:46 +01:00
}
IdentityStore.prototype._restoreFromSeed = function(password, seed, derivedKey) {
var keyStore = new LightwalletKeyStore(seed, derivedKey, this.hdPathString)
keyStore.addHdDerivationPath(this.hdPathString, derivedKey, {curve: 'secp256k1', purpose: 'sign'});
keyStore.setDefaultHdDerivationPath(this.hdPathString)
keyStore.generateNewAddress(derivedKey, 3)
configManager.setWallet(keyStore.serialize())
console.log('restored from seed. saved to keystore')
return keyStore
}
IdentityStore.prototype._createFirstWallet = function(entropy, derivedKey) {
var secretSeed = LightwalletKeyStore.generateRandomSeed(entropy)
var keyStore = new LightwalletKeyStore(secretSeed, derivedKey, this.hdPathString)
keyStore.addHdDerivationPath(this.hdPathString, derivedKey, {curve: 'secp256k1', purpose: 'sign'});
keyStore.setDefaultHdDerivationPath(this.hdPathString)
keyStore.generateNewAddress(derivedKey, 3)
configManager.setWallet(keyStore.serialize())
console.log('saved to keystore')
return keyStore
}
// get addresses and normalize address hexString
IdentityStore.prototype._getAddresses = function() {
return this._keyStore.getAddresses(this.hdPathString).map((address) => { return '0x'+address })
}
IdentityStore.prototype._autoFaucet = function() {
var addresses = this._getAddresses()
autoFaucet(addresses[0])
}
2016-03-29 20:12:07 +02:00
function IdManagement(opts) {
if (!opts) opts = {}
this.keyStore = opts.keyStore
this.derivedKey = opts.derivedKey
2016-03-29 21:50:47 +02:00
this.hdPathString = "m/44'/60'/0'/0"
this.getAddresses = function(){
return keyStore.getAddresses(this.hdPathString).map(function(address){ return '0x'+address })
}
this.signTx = function(txParams){
// normalize values
txParams.to = ethUtil.addHexPrefix(txParams.to)
txParams.from = ethUtil.addHexPrefix(txParams.from)
txParams.value = ethUtil.addHexPrefix(txParams.value)
txParams.data = ethUtil.addHexPrefix(txParams.data)
txParams.gasLimit = ethUtil.addHexPrefix(txParams.gasLimit || txParams.gas)
txParams.nonce = ethUtil.addHexPrefix(txParams.nonce)
var tx = new Transaction(txParams)
var rawTx = '0x'+tx.serialize().toString('hex')
2016-03-29 21:50:47 +02:00
return '0x'+LightwalletSigner.signTx(this.keyStore, this.derivedKey, rawTx, txParams.from, this.hdPathString)
}
this.getSeed = function(){
return this.keyStore.getSeed(this.derivedKey)
}
2016-04-06 21:01:10 +02:00
this.exportPrivateKey = function(address) {
return this.keyStore.exportPrivateKey(address, this.derivedKey, this.hdPathString)
}
}
2016-02-11 02:44:46 +01:00
// util
function noop(){}