Bruno
7 years ago
parent
d4201ae1cc
commit
8763ea898e
File diff suppressed because it is too large
Load Diff
@ -1,239 +0,0 @@ |
|||||||
const { EventEmitter } = require('events') |
|
||||||
const ethUtil = require('ethereumjs-util') |
|
||||||
const sigUtil = require('eth-sig-util') |
|
||||||
|
|
||||||
const hdPathString = `m/44'/60'/0'/0` |
|
||||||
const keyringType = 'Trezor Hardware' |
|
||||||
const Transaction = require('ethereumjs-tx') |
|
||||||
const pathBase = 'm' |
|
||||||
const TrezorConnect = require('./trezor-connect.js') |
|
||||||
const HDKey = require('hdkey') |
|
||||||
const TREZOR_MIN_FIRMWARE_VERSION = '1.5.2' |
|
||||||
const log = require('loglevel') |
|
||||||
|
|
||||||
class TrezorKeyring extends EventEmitter { |
|
||||||
constructor (opts = {}) { |
|
||||||
super() |
|
||||||
this.type = keyringType |
|
||||||
this.accounts = [] |
|
||||||
this.hdk = new HDKey() |
|
||||||
this.deserialize(opts) |
|
||||||
this.page = 0 |
|
||||||
this.perPage = 5 |
|
||||||
this.unlockedAccount = 0 |
|
||||||
} |
|
||||||
|
|
||||||
serialize () { |
|
||||||
return Promise.resolve({ |
|
||||||
hdPath: this.hdPath, |
|
||||||
accounts: this.accounts, |
|
||||||
page: this.page, |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
deserialize (opts = {}) { |
|
||||||
this.hdPath = opts.hdPath || hdPathString |
|
||||||
this.accounts = opts.accounts || [] |
|
||||||
this.page = opts.page || 0 |
|
||||||
return Promise.resolve() |
|
||||||
} |
|
||||||
|
|
||||||
unlock () { |
|
||||||
|
|
||||||
if (this.hdk.publicKey) return Promise.resolve() |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
TrezorConnect.getXPubKey( |
|
||||||
this.hdPath, |
|
||||||
response => { |
|
||||||
if (response.success) { |
|
||||||
this.hdk.publicKey = new Buffer(response.publicKey, 'hex') |
|
||||||
this.hdk.chainCode = new Buffer(response.chainCode, 'hex') |
|
||||||
resolve() |
|
||||||
} else { |
|
||||||
reject(response.error || 'Unknown error') |
|
||||||
} |
|
||||||
}, |
|
||||||
TREZOR_MIN_FIRMWARE_VERSION |
|
||||||
) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
setAccountToUnlock (index) { |
|
||||||
this.unlockedAccount = parseInt(index, 10) |
|
||||||
} |
|
||||||
|
|
||||||
addAccounts (n = 1) { |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
return this.unlock() |
|
||||||
.then(_ => { |
|
||||||
const from = this.unlockedAccount |
|
||||||
const to = from + 1 |
|
||||||
this.accounts = [] |
|
||||||
|
|
||||||
for (let i = from; i < to; i++) { |
|
||||||
|
|
||||||
this.accounts.push(this._addressFromId(pathBase, i)) |
|
||||||
this.page = 0 |
|
||||||
} |
|
||||||
resolve(this.accounts) |
|
||||||
}) |
|
||||||
.catch(e => { |
|
||||||
reject(e) |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
getPage () { |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
return this.unlock() |
|
||||||
.then(_ => { |
|
||||||
|
|
||||||
const from = this.page === 0 ? 0 : (this.page - 1) * this.perPage |
|
||||||
const to = from + this.perPage |
|
||||||
|
|
||||||
const accounts = [] |
|
||||||
|
|
||||||
for (let i = from; i < to; i++) { |
|
||||||
|
|
||||||
accounts.push({ |
|
||||||
address: this._addressFromId(pathBase, i), |
|
||||||
balance: 0, |
|
||||||
index: i, |
|
||||||
}) |
|
||||||
} |
|
||||||
log.debug(accounts) |
|
||||||
resolve(accounts) |
|
||||||
}) |
|
||||||
.catch(e => { |
|
||||||
reject(e) |
|
||||||
}) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
async getPrevAccountSet () { |
|
||||||
this.page-- |
|
||||||
return await this.getPage() |
|
||||||
} |
|
||||||
|
|
||||||
async getNextAccountSet () { |
|
||||||
this.page++ |
|
||||||
return await this.getPage() |
|
||||||
} |
|
||||||
|
|
||||||
getAccounts () { |
|
||||||
return Promise.resolve(this.accounts.slice()) |
|
||||||
} |
|
||||||
|
|
||||||
// tx is an instance of the ethereumjs-transaction class.
|
|
||||||
async signTransaction (address, tx) { |
|
||||||
|
|
||||||
return new Promise((resolve, reject) => { |
|
||||||
|
|
||||||
log.debug('sign transaction ', address, tx) |
|
||||||
|
|
||||||
TrezorConnect.ethereumSignTx( |
|
||||||
this._getUnlockedAccount(), |
|
||||||
this._normalize(tx.nonce), |
|
||||||
this._normalize(tx.gasPrice), |
|
||||||
this._normalize(tx.gasLimit), |
|
||||||
this._normalize(tx.to), |
|
||||||
this._normalize(tx.value), |
|
||||||
this._normalize(tx.data), |
|
||||||
tx._chainId, |
|
||||||
response => { |
|
||||||
if (response.success) { |
|
||||||
|
|
||||||
tx.v = `0x${response.v.toString(16)}` |
|
||||||
tx.r = `0x${response.r}` |
|
||||||
tx.s = `0x${response.s}` |
|
||||||
log.debug('about to create new tx with data', tx) |
|
||||||
|
|
||||||
const signedTx = new Transaction(tx) |
|
||||||
|
|
||||||
const addressSignedWith = ethUtil.toChecksumAddress(`0x${signedTx.from.toString('hex')}`) |
|
||||||
const correctAddress = ethUtil.toChecksumAddress(address) |
|
||||||
if (addressSignedWith !== correctAddress) { |
|
||||||
log.error('signature doesnt match the right address', addressSignedWith, correctAddress) |
|
||||||
throw new Error('signature doesnt match the right address') |
|
||||||
} |
|
||||||
|
|
||||||
resolve(signedTx) |
|
||||||
|
|
||||||
} else { |
|
||||||
throw new Error(response.error || 'Unknown error') |
|
||||||
} |
|
||||||
}, |
|
||||||
TREZOR_MIN_FIRMWARE_VERSION) |
|
||||||
}) |
|
||||||
} |
|
||||||
|
|
||||||
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) { |
|
||||||
|
|
||||||
TrezorConnect.ethereumSignMessage(this._getUnlockedAccount(), message, response => { |
|
||||||
if (response.success) { |
|
||||||
|
|
||||||
const signature = this._personalToRawSig(response.signature) |
|
||||||
const addressSignedWith = sigUtil.recoverPersonalSignature({data: message, sig: signature}) |
|
||||||
const correctAddress = ethUtil.toChecksumAddress(withAccount) |
|
||||||
if (addressSignedWith !== correctAddress) { |
|
||||||
log.error('signature doesnt match the right address', addressSignedWith, correctAddress) |
|
||||||
throw new Error('signature doesnt match the right address') |
|
||||||
} |
|
||||||
return signature |
|
||||||
|
|
||||||
} else { |
|
||||||
throw new Error(response.error || 'Unknown error') |
|
||||||
} |
|
||||||
|
|
||||||
}, TREZOR_MIN_FIRMWARE_VERSION) |
|
||||||
} |
|
||||||
|
|
||||||
async signTypedData (withAccount, typedData) { |
|
||||||
// Waiting on trezor to enable this
|
|
||||||
throw new Error('Not supported on this device') |
|
||||||
} |
|
||||||
|
|
||||||
async exportAccount (address) { |
|
||||||
throw new Error('Not supported on this device') |
|
||||||
} |
|
||||||
|
|
||||||
_padLeftEven (hex) { |
|
||||||
return hex.length % 2 !== 0 ? `0${hex}` : hex |
|
||||||
} |
|
||||||
|
|
||||||
_normalize (buf) { |
|
||||||
return this._padLeftEven(ethUtil.bufferToHex(buf).substring(2).toLowerCase()) |
|
||||||
} |
|
||||||
|
|
||||||
_addressFromId (pathBase, i) { |
|
||||||
const dkey = this.hdk.derive(`${pathBase}/${i}`) |
|
||||||
const address = ethUtil |
|
||||||
.publicToAddress(dkey.publicKey, true) |
|
||||||
.toString('hex') |
|
||||||
return ethUtil.toChecksumAddress(address) |
|
||||||
} |
|
||||||
|
|
||||||
_getUnlockedAccount () { |
|
||||||
return `${this.hdPath}/${this.unlockedAccount}` |
|
||||||
} |
|
||||||
|
|
||||||
_personalToRawSig (signature) { |
|
||||||
var v = signature['v'] - 27 |
|
||||||
v = v.toString(16) |
|
||||||
if (v.length < 2) { |
|
||||||
v = '0' + v |
|
||||||
} |
|
||||||
return '0x' + signature['r'] + signature['s'] + v |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
TrezorKeyring.type = keyringType |
|
||||||
module.exports = TrezorKeyring |
|
||||||
Loading…
Reference in new issue