brunobar79
6 years ago
parent
2355573340
commit
0b9b892c6b
@ -1,320 +0,0 @@ |
|||||||
const {EventEmitter} = require('events') |
|
||||||
const HDKey = require('hdkey') |
|
||||||
const ethUtil = require('ethereumjs-util') |
|
||||||
const sigUtil = require('eth-sig-util') |
|
||||||
const Transaction = require('ethereumjs-tx') |
|
||||||
|
|
||||||
const hdPathString = `44'/60'/0'` |
|
||||||
const type = 'Ledger Hardware' |
|
||||||
const BRIDGE_URL = 'https://localhost:3000' |
|
||||||
const pathBase = 'm' |
|
||||||
const MAX_INDEX = 1000 |
|
||||||
|
|
||||||
class LedgerKeyring extends EventEmitter { |
|
||||||
constructor (opts = {}) { |
|
||||||
super() |
|
||||||
this.type = type |
|
||||||
this.page = 0 |
|
||||||
this.perPage = 5 |
|
||||||
this.unlockedAccount = 0 |
|
||||||
this.hdk = new HDKey() |
|
||||||
this.paths = {} |
|
||||||
this.iframe = null |
|
||||||
this.setupIframe() |
|
||||||
this.deserialize(opts) |
|
||||||
} |
|
||||||
|
|
||||||
setupIframe () { |
|
||||||
this.iframe = document.createElement('iframe') |
|
||||||
this.iframe.src = BRIDGE_URL |
|
||||||
document.head.appendChild(this.iframe) |
|
||||||
} |
|
||||||
|
|
||||||
sendMessage (msg, cb) { |
|
||||||
this.iframe.contentWindow.postMessage({...msg, target: 'LEDGER-IFRAME'}, '*') |
|
||||||
window.addEventListener('message', ({ origin, data }) => { |
|
||||||
if (origin !== BRIDGE_URL) return false |
|
||||||
if (data && data.action && data.action === `${msg.action}-reply`) { |
|
||||||
cb(data) |
|
||||||
} |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
serialize () { |
|
||||||
return Promise.resolve({hdPath: this.hdPath, accounts: this.accounts}) |
|
||||||
} |
|
||||||
|
|
||||||
deserialize (opts = {}) { |
|
||||||
this.hdPath = opts.hdPath || hdPathString |
|
||||||
this.unlocked = opts.unlocked || false |
|
||||||
this.accounts = opts.accounts || [] |
|
||||||
return Promise.resolve() |
|
||||||
} |
|
||||||
|
|
||||||
isUnlocked () { |
|
||||||
return this.unlocked |
|
||||||
} |
|
||||||
|
|
||||||
setAccountToUnlock (index) { |
|
||||||
this.unlockedAccount = parseInt(index, 10) |
|
||||||
} |
|
||||||
|
|
||||||
unlock () { |
|
||||||
|
|
||||||
if (this.isUnlocked()) return Promise.resolve('already unlocked') |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
this.sendMessage({ |
|
||||||
action: 'ledger-unlock', |
|
||||||
params: { |
|
||||||
hdPath: this.hdPath, |
|
||||||
}, |
|
||||||
}, |
|
||||||
({success, payload}) => { |
|
||||||
if (success) { |
|
||||||
this.hdk.publicKey = new Buffer(payload.publicKey, 'hex') |
|
||||||
this.hdk.chainCode = new Buffer(payload.chainCode, 'hex') |
|
||||||
resolve('just unlocked') |
|
||||||
} else { |
|
||||||
reject(payload.error || 'Unknown error') |
|
||||||
} |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
addAccounts (n = 1) { |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
this.unlock() |
|
||||||
.then(_ => { |
|
||||||
const from = this.unlockedAccount |
|
||||||
const to = from + n |
|
||||||
this.accounts = [] |
|
||||||
|
|
||||||
for (let i = from; i < to; i++) { |
|
||||||
const address = this._addressFromIndex(pathBase, i) |
|
||||||
this.accounts.push(address) |
|
||||||
this.page = 0 |
|
||||||
} |
|
||||||
resolve(this.accounts) |
|
||||||
}) |
|
||||||
.catch(e => { |
|
||||||
reject(e) |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
getFirstPage () { |
|
||||||
this.page = 0 |
|
||||||
return this.__getPage(1) |
|
||||||
} |
|
||||||
|
|
||||||
getNextPage () { |
|
||||||
return this.__getPage(1) |
|
||||||
} |
|
||||||
|
|
||||||
getPreviousPage () { |
|
||||||
return this.__getPage(-1) |
|
||||||
} |
|
||||||
|
|
||||||
__getPage (increment) { |
|
||||||
|
|
||||||
this.page += increment |
|
||||||
|
|
||||||
if (this.page <= 0) { this.page = 1 } |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
this.unlock() |
|
||||||
.then(_ => { |
|
||||||
|
|
||||||
const from = (this.page - 1) * this.perPage |
|
||||||
const to = from + this.perPage |
|
||||||
|
|
||||||
const accounts = [] |
|
||||||
|
|
||||||
for (let i = from; i < to; i++) { |
|
||||||
const address = this._addressFromIndex(pathBase, i) |
|
||||||
accounts.push({ |
|
||||||
address: address, |
|
||||||
balance: null, |
|
||||||
index: i, |
|
||||||
}) |
|
||||||
this.paths[ethUtil.toChecksumAddress(address)] = i |
|
||||||
|
|
||||||
} |
|
||||||
resolve(accounts) |
|
||||||
}) |
|
||||||
.catch(e => { |
|
||||||
reject(e) |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
getAccounts () { |
|
||||||
return Promise.resolve(this.accounts.slice()) |
|
||||||
} |
|
||||||
|
|
||||||
removeAccount (address) { |
|
||||||
if (!this.accounts.map(a => a.toLowerCase()).includes(address.toLowerCase())) { |
|
||||||
throw new Error(`Address ${address} not found in this keyring`) |
|
||||||
} |
|
||||||
this.accounts = this.accounts.filter(a => a.toLowerCase() !== address.toLowerCase()) |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
// tx is an instance of the ethereumjs-transaction class.
|
|
||||||
async signTransaction (address, tx) { |
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
this.unlock() |
|
||||||
.then(_ => { |
|
||||||
|
|
||||||
const newTx = new Transaction({ |
|
||||||
from: this._normalize(address), |
|
||||||
to: this._normalize(tx.to), |
|
||||||
value: this._normalize(tx.value), |
|
||||||
data: this._normalize(tx.data), |
|
||||||
chainId: tx._chainId, |
|
||||||
nonce: this._normalize(tx.nonce), |
|
||||||
gasLimit: this._normalize(tx.gasLimit), |
|
||||||
gasPrice: this._normalize(tx.gasPrice), |
|
||||||
v: ethUtil.bufferToHex(tx.getChainId()), |
|
||||||
r: '0x00', |
|
||||||
s: '0x00', |
|
||||||
}) |
|
||||||
|
|
||||||
this.sendMessage({ |
|
||||||
action: 'ledger-sign-transaction', |
|
||||||
params: { |
|
||||||
tx: newTx.serialize().toString('hex'), |
|
||||||
hdPath: this._pathFromAddress(address), |
|
||||||
}, |
|
||||||
}, |
|
||||||
({success, payload}) => { |
|
||||||
if (success) { |
|
||||||
|
|
||||||
newTx.v = Buffer.from(payload.v, 'hex') |
|
||||||
newTx.r = Buffer.from(payload.r, 'hex') |
|
||||||
newTx.s = Buffer.from(payload.s, 'hex') |
|
||||||
|
|
||||||
const valid = newTx.verifySignature() |
|
||||||
if (valid) { |
|
||||||
resolve(newTx) |
|
||||||
} else { |
|
||||||
reject('The transaction signature is not valid') |
|
||||||
} |
|
||||||
} else { |
|
||||||
reject(payload) |
|
||||||
} |
|
||||||
}) |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
async signMessage (withAccount, data) { |
|
||||||
throw new Error('Not supported on this device') |
|
||||||
} |
|
||||||
|
|
||||||
// For personal_sign, we need to prefix the message:
|
|
||||||
async signPersonalMessage (withAccount, message) { |
|
||||||
const humanReadableMsg = this._toAscii(message) |
|
||||||
const bufferMsg = Buffer.from(humanReadableMsg).toString('hex') |
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
this.unlock() |
|
||||||
.then(_ => { |
|
||||||
this.sendMessage({ |
|
||||||
action: 'ledger-sign-personal-message', |
|
||||||
params: { |
|
||||||
hdPath: this._pathFromAddress(withAccount), |
|
||||||
message: bufferMsg, |
|
||||||
}, |
|
||||||
}, |
|
||||||
({success, payload}) => { |
|
||||||
if (success) { |
|
||||||
let v = payload['v'] - 27 |
|
||||||
v = v.toString(16) |
|
||||||
if (v.length < 2) { |
|
||||||
v = `0${v}` |
|
||||||
} |
|
||||||
const signature = `0x${payload['r']}${payload['s']}${v}` |
|
||||||
const addressSignedWith = sigUtil.recoverPersonalSignature({data: message, sig: signature}) |
|
||||||
if (ethUtil.toChecksumAddress(addressSignedWith) !== ethUtil.toChecksumAddress(withAccount)) { |
|
||||||
reject('signature doesnt match the right address') |
|
||||||
} |
|
||||||
resolve(signature) |
|
||||||
} else { |
|
||||||
reject(payload) |
|
||||||
} |
|
||||||
}) |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
async signTypedData (withAccount, typedData) { |
|
||||||
throw new Error('Not supported on this device') |
|
||||||
} |
|
||||||
|
|
||||||
async exportAccount (address) { |
|
||||||
throw new Error('Not supported on this device') |
|
||||||
} |
|
||||||
|
|
||||||
forgetDevice () { |
|
||||||
this.accounts = [] |
|
||||||
this.unlocked = false |
|
||||||
this.page = 0 |
|
||||||
this.unlockedAccount = 0 |
|
||||||
this.paths = {} |
|
||||||
} |
|
||||||
|
|
||||||
/* PRIVATE METHODS */ |
|
||||||
|
|
||||||
_padLeftEven (hex) { |
|
||||||
return hex.length % 2 !== 0 ? `0${hex}` : hex |
|
||||||
} |
|
||||||
|
|
||||||
_normalize (buf) { |
|
||||||
return this._padLeftEven(ethUtil.bufferToHex(buf).toLowerCase()) |
|
||||||
} |
|
||||||
|
|
||||||
_addressFromIndex (pathBase, i) { |
|
||||||
const dkey = this.hdk.derive(`${pathBase}/${i}`) |
|
||||||
const address = ethUtil |
|
||||||
.publicToAddress(dkey.publicKey, true) |
|
||||||
.toString('hex') |
|
||||||
return ethUtil.toChecksumAddress(address) |
|
||||||
} |
|
||||||
|
|
||||||
_pathFromAddress (address) { |
|
||||||
const checksummedAddress = ethUtil.toChecksumAddress(address) |
|
||||||
let index = this.paths[checksummedAddress] |
|
||||||
if (typeof index === 'undefined') { |
|
||||||
for (let i = 0; i < MAX_INDEX; i++) { |
|
||||||
if (checksummedAddress === this._addressFromIndex(pathBase, i)) { |
|
||||||
index = i |
|
||||||
break |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
if (typeof index === 'undefined') { |
|
||||||
throw new Error('Unknown address') |
|
||||||
} |
|
||||||
return `${this.hdPath}/${index}` |
|
||||||
} |
|
||||||
|
|
||||||
_toAscii (hex) { |
|
||||||
let str = '' |
|
||||||
let i = 0; const l = hex.length |
|
||||||
if (hex.substring(0, 2) === '0x') { |
|
||||||
i = 2 |
|
||||||
} |
|
||||||
for (; i < l; i += 2) { |
|
||||||
const code = parseInt(hex.substr(i, 2), 16) |
|
||||||
str += String.fromCharCode(code) |
|
||||||
} |
|
||||||
|
|
||||||
return str |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
LedgerKeyring.type = type |
|
||||||
module.exports = LedgerKeyring |
|
||||||
Loading…
Reference in new issue