2015-05-12 23:55:39 +00:00
#!/usr/bin/env python
2016-11-25 21:53:55 +00:00
# This file is part of the TREZOR project.
#
# Copyright (C) 2012-2016 Marek Palatinus <slush@satoshilabs.com>
# Copyright (C) 2012-2016 Pavol Rusnak <stick@satoshilabs.com>
# Copyright (C) 2016 Jochen Hoenicke <hoenicke@gmail.com>
#
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this library. If not, see <http://www.gnu.org/licenses/>.
2015-02-22 13:23:39 +00:00
import binascii
import json
import base64
2017-07-03 15:45:56 +00:00
import click
2015-02-22 13:23:39 +00:00
2017-07-01 15:59:11 +00:00
from trezorlib.client import TrezorClient, TrezorClientVerbose, CallException
2016-06-29 20:40:37 +00:00
import trezorlib.types_pb2 as types
2015-02-22 13:23:39 +00:00
2017-07-01 15:59:11 +00:00
def get_transport_class_by_name(name):
2015-02-22 13:23:39 +00:00
2017-07-01 15:59:11 +00:00
if name == 'usb':
from trezorlib.transport_hid import HidTransport
return HidTransport
2015-09-04 18:41:54 +00:00
2017-07-01 15:59:11 +00:00
if name == 'udp':
2016-04-30 00:37:18 +00:00
from trezorlib.transport_udp import UdpTransport
2017-07-01 15:59:11 +00:00
return UdpTransport
2016-04-30 00:37:18 +00:00
2017-07-01 15:59:11 +00:00
if name == 'pipe':
2015-02-22 13:23:39 +00:00
from trezorlib.transport_pipe import PipeTransport
2017-07-01 15:59:11 +00:00
return PipeTransport
2015-09-04 18:41:54 +00:00
2017-07-01 15:59:11 +00:00
if name == 'bridge':
2015-02-22 13:23:39 +00:00
from trezorlib.transport_bridge import BridgeTransport
2017-07-01 15:59:11 +00:00
return BridgeTransport
raise NotImplementedError('Unknown transport: "%s"' % name)
2016-06-30 14:47:17 +00:00
2017-07-01 15:59:11 +00:00
def get_transport(transport_name, path):
transport = get_transport_class_by_name(transport_name)
dev = transport.find_by_path(path)
return dev
2015-09-04 18:41:54 +00:00
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
@click.group()
@click.option('-t', '--transport', type=click.Choice(['usb', 'udp', 'pipe', 'bridge']), default='usb', help='Select transport used for communication.')
@click.option('-p', '--path', help='Select device by transport-specific path.')
@click.option('-v', '--verbose', is_flag=True, help='Show communication messages.')
@click.option('-j', '--json', 'is_json', is_flag=True, help='Print result as JSON object')
@click.pass_context
def cli(ctx, transport, path, verbose, is_json):
if ctx.invoked_subcommand == 'list':
ctx.obj = transport
else:
t = get_transport(transport, path)
if verbose:
ctx.obj = TrezorClientVerbose(t)
else:
ctx.obj = TrezorClient(t)
2015-09-04 18:41:54 +00:00
2017-07-04 11:55:05 +00:00
@cli.resultcallback()
def print_result(res, transport, path, verbose, is_json):
if is_json:
2017-07-10 16:20:32 +00:00
from google.protobuf import json_format, message
if isinstance(res, message.Message):
2017-07-10 13:36:44 +00:00
click.echo(json_format.MessageToJson(res, preserving_proto_field_name=True))
2017-07-04 11:55:05 +00:00
else:
click.echo(json.dumps(res, sort_keys=True, indent=4))
else:
2017-08-08 22:52:52 +00:00
from google.protobuf import text_format, message
if isinstance(res, message.Message):
click.echo('%s {\n%s}' % (res.DESCRIPTOR.name, text_format.MessageToString(res, indent=4)))
elif isinstance(res, list):
2017-07-04 11:55:05 +00:00
for line in res:
click.echo(line)
elif isinstance(res, dict):
for k, v in res.items():
2017-07-12 16:35:54 +00:00
if isinstance(v, dict):
for kk, vv in v.items():
click.echo('%s.%s: %s' % (k, kk, vv))
else:
click.echo('%s: %s' % (k, v))
2017-07-04 11:55:05 +00:00
else:
click.echo(res)
2017-07-03 15:45:56 +00:00
#
# Common functions
#
2015-09-04 18:41:54 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(name='list', help='List connected TREZOR devices.')
@click.pass_obj
def ls(transport_name):
transport_class = get_transport_class_by_name(transport_name)
devices = transport_class.enumerate()
2017-07-10 13:36:44 +00:00
if transport_name == 'usb':
return [dev[0] for dev in devices]
if transport_name == 'bridge':
return devices
return []
2016-05-24 19:32:58 +00:00
2016-05-27 07:05:54 +00:00
2017-07-03 15:45:56 +00:00
#
# Basic device functions
#
2015-09-04 18:41:54 +00:00
2016-05-24 19:32:58 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Send ping message.')
@click.argument('message')
@click.option('-b', '--button-protection', is_flag=True)
@click.option('-p', '--pin-protection', is_flag=True)
@click.option('-r', '--passphrase-protection', is_flag=True)
@click.pass_obj
def ping(client, message, button_protection, pin_protection, passphrase_protection):
2017-07-04 11:55:05 +00:00
return client.ping(message, button_protection=button_protection, pin_protection=pin_protection, passphrase_protection=passphrase_protection)
2016-05-27 07:05:54 +00:00
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Clear session (remove cached PIN, passphrase, etc.).')
@click.pass_obj
def clear_session(client):
2017-07-04 11:55:05 +00:00
return client.clear_session()
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Get example entropy.')
@click.argument('size', type=int)
@click.pass_obj
def get_entropy(client, size):
2017-07-04 11:55:05 +00:00
return binascii.hexlify(client.get_entropy(size))
2015-09-04 18:41:54 +00:00
2017-02-23 10:56:54 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Retrieve device features and settings.')
@click.pass_obj
def get_features(client):
return client.features
2017-02-23 10:56:54 +00:00
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='List all supported coin types by the device.')
@click.pass_obj
def list_coins(client):
2017-07-04 11:55:05 +00:00
return [coin.coin_name for coin in client.features.coins]
2017-07-03 15:45:56 +00:00
#
# Device management functions
#
@cli.command(help='Change new PIN or remove existing.')
@click.option('-r', '--remove', is_flag=True)
@click.pass_obj
def change_pin(client, remove):
2017-07-04 11:55:05 +00:00
return client.change_pin(remove)
2017-07-03 15:45:56 +00:00
@cli.command(help='Enable passphrase.')
@click.pass_obj
def enable_passphrase(client):
2017-07-04 11:55:05 +00:00
return client.apply_settings(use_passphrase=True)
2017-07-03 15:45:56 +00:00
@cli.command(help='Disable passphrase.')
@click.pass_obj
def disable_passphrase(client):
2017-07-04 11:55:05 +00:00
return client.apply_settings(use_passphrase=False)
2017-07-03 15:45:56 +00:00
@cli.command(help='Set new device label.')
@click.option('-l', '--label')
@click.pass_obj
def set_label(client, label):
2017-07-04 11:55:05 +00:00
return client.apply_settings(label=label)
2017-07-03 15:45:56 +00:00
2017-07-17 16:36:53 +00:00
@cli.command(help='Set device flags.')
@click.argument('flags')
@click.pass_obj
def set_flags(client, flags):
flags = flags.lower()
if flags.startswith('0b'):
flags = int(flags, 2)
elif flags.startswith('0x'):
flags = int(flags, 16)
else:
flags = int(flags)
return client.apply_flags(flags=flags)
2017-07-03 15:45:56 +00:00
@cli.command(help='Set new homescreen.')
@click.option('-f', '--filename', default=None)
@click.pass_obj
def set_homescreen(client, filename):
if filename is not None:
from PIL import Image
im = Image.open(filename)
if im.size != (128, 64):
raise CallException(types.Failure_DataError, 'Wrong size of the image')
im = im.convert('1')
pix = im.load()
2017-07-16 13:03:01 +00:00
img = bytearray(1024)
2017-07-03 15:45:56 +00:00
for j in range(64):
for i in range(128):
2017-07-16 13:03:01 +00:00
if pix[i, j]:
o = (i + j * 128)
img[o // 8] |= (1 << (7 - o % 8))
img = bytes(img)
2017-07-03 15:45:56 +00:00
else:
2017-07-16 13:03:01 +00:00
img = b'\x00'
2017-07-04 11:55:05 +00:00
return client.apply_settings(homescreen=img)
2017-07-03 15:45:56 +00:00
@cli.command(help='Set U2F counter.')
@click.argument('counter', type=int)
@click.pass_obj
def set_u2f_counter(client, counter):
2017-07-04 11:55:05 +00:00
return client.set_u2f_counter(counter)
2017-07-03 15:45:56 +00:00
@cli.command(help='Reset device to factory defaults and remove all private data.')
@click.pass_obj
def wipe_device(client):
2017-07-04 11:55:05 +00:00
return client.wipe_device()
2017-07-03 15:45:56 +00:00
@cli.command(help='Load custom configuration to the device.')
@click.option('-m', '--mnemonic')
@click.option('-e', '--expand', is_flag=True)
@click.option('-x', '--xprv')
@click.option('-p', '--pin', default='')
@click.option('-r', '--passphrase-protection', is_flag=True)
@click.option('-l', '--label', default='')
2017-07-16 12:53:28 +00:00
@click.option('-i', '--ignore-checksum', is_flag=True)
@click.option('-s', '--slip0014', is_flag=True)
2017-07-03 15:45:56 +00:00
@click.pass_obj
2017-07-16 12:53:28 +00:00
def load_device(client, mnemonic, expand, xprv, pin, passphrase_protection, label, ignore_checksum, slip0014):
if not mnemonic and not xprv and not slip0014:
2017-07-03 15:45:56 +00:00
raise CallException(types.Failure_DataError, 'Please provide mnemonic or xprv')
if mnemonic:
2017-07-04 11:55:05 +00:00
return client.load_device_by_mnemonic(
2017-07-03 15:45:56 +00:00
mnemonic,
pin,
passphrase_protection,
label,
'english',
2017-07-16 12:53:28 +00:00
ignore_checksum,
2017-07-03 15:45:56 +00:00
expand
)
if xprv:
2017-07-04 11:55:05 +00:00
return client.load_device_by_xprv(
2017-07-03 15:45:56 +00:00
xprv,
pin,
passphrase_protection,
label,
'english'
)
2017-07-16 12:53:28 +00:00
if slip0014:
return client.load_device_by_mnemonic(
' '.join(['all'] * 12),
pin,
passphrase_protection,
'SLIP-0014'
)
2017-07-03 15:45:56 +00:00
@cli.command(help='Start safe recovery workflow.')
2017-07-12 13:45:16 +00:00
@click.option('-w', '--words', type=click.Choice(['12', '18', '24']), default='24')
2017-07-03 15:45:56 +00:00
@click.option('-e', '--expand', is_flag=True)
@click.option('-p', '--pin-protection', is_flag=True)
@click.option('-r', '--passphrase-protection', is_flag=True)
@click.option('-l', '--label')
@click.option('-t', '--type', 'rec_type', type=click.Choice(['scrambled', 'matrix']), default='scrambled')
@click.option('-d', '--dry-run', is_flag=True)
@click.pass_obj
def recovery_device(client, words, expand, pin_protection, passphrase_protection, label, rec_type, dry_run):
typemap = {
'scrambled': types.RecoveryDeviceType_ScrambledWords,
'matrix': types.RecoveryDeviceType_Matrix
}
2017-07-04 11:55:05 +00:00
return client.recovery_device(
2017-07-12 13:45:16 +00:00
int(words),
2017-07-03 15:45:56 +00:00
passphrase_protection,
pin_protection,
label,
'english',
typemap[rec_type],
expand,
dry_run
2015-02-22 13:23:39 +00:00
)
2017-07-03 15:45:56 +00:00
@cli.command(help='Perform device setup and generate new seed.')
2017-07-12 13:45:16 +00:00
@click.option('-t', '--strength', type=click.Choice(['128', '192', '256']), default='256')
2017-07-03 15:45:56 +00:00
@click.option('-p', '--pin-protection', is_flag=True)
@click.option('-r', '--passphrase-protection', is_flag=True)
@click.option('-l', '--label')
@click.option('-u', '--u2f-counter', default=0)
@click.option('-s', '--skip-backup', is_flag=True)
@click.pass_obj
def reset_device(client, strength, pin_protection, passphrase_protection, label, u2f_counter, skip_backup):
2017-07-04 11:55:05 +00:00
return client.reset_device(
2017-07-03 15:45:56 +00:00
True,
2017-07-12 13:45:16 +00:00
int(strength),
2017-07-03 15:45:56 +00:00
passphrase_protection,
pin_protection,
label,
'english',
u2f_counter,
skip_backup
2015-02-22 13:23:39 +00:00
)
2015-09-04 18:41:54 +00:00
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Perform device seed backup.')
@click.pass_obj
def backup_device(client):
2017-07-04 11:55:05 +00:00
return client.backup_device()
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
#
# Firmware update
#
2015-02-22 13:23:39 +00:00
2017-06-23 17:23:08 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Upload new firmware to device (must be in bootloader mode).')
@click.option('-f', '--filename')
@click.option('-u', '--url')
@click.option('-v', '--version')
@click.option('-s', '--skip-check', is_flag=True)
@click.pass_obj
def firmware_update(client, filename, url, version, skip_check):
if filename:
fp = open(filename, 'rb').read()
elif url:
import requests
click.echo('Downloading from', url)
r = requests.get(url)
fp = r.content
else:
import requests
r = requests.get('https://wallet.trezor.io/data/firmware/releases.json')
releases = r.json()
def version_func(r):
return r['version']
def version_string(r):
return '.'.join(map(str, version_func(r)))
if version:
release = next((r for r in releases if version_string(r) == version))
else:
release = max(releases, key=version_func)
click.echo('Fetching version: %s' % version_string(release))
click.echo('Firmware fingerprint: %s' % release['fingerprint'])
url = 'https://wallet.trezor.io/' + release['url']
click.echo('Downloading from %s' % url)
r = requests.get(url)
fp = r.content
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
if not skip_check:
if fp[:8] == b'54525a52' or fp[:8] == b'54525a56':
fp = binascii.unhexlify(fp)
if fp[:4] != b'TRZR' and fp[:4] != b'TRZV':
raise CallException(types.Failure_FirmwareError, 'TREZOR firmware header expected')
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
click.echo('Please confirm action on device...')
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
from io import BytesIO
2017-07-04 11:55:05 +00:00
return client.firmware_update(fp=BytesIO(fp))
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Perform a self-test.')
@click.pass_obj
def self_test(client):
2017-07-04 11:55:05 +00:00
return client.self_test()
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
#
# Basic coin functions
#
2016-06-12 21:49:52 +00:00
2017-07-03 16:49:03 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Get address for specified path.')
@click.option('-c', '--coin', default='Bitcoin')
@click.option('-n', '--address')
@click.option('-t', '--script-type', type=click.Choice(['address', 'segwit', 'p2shsegwit']), default='address')
@click.option('-d', '--show-display', is_flag=True)
@click.pass_obj
def get_address(client, coin, address, script_type, show_display):
address_n = client.expand_path(address)
typemap = {
'address': types.SPENDADDRESS,
'segwit': types.SPENDWITNESS,
'p2shsegwit': types.SPENDP2SHWITNESS,
}
script_type = typemap[script_type]
2017-07-04 11:55:05 +00:00
return client.get_address(coin, address_n, show_display, script_type=script_type)
2017-07-03 15:45:56 +00:00
@cli.command(help='Get public node of given path.')
@click.option('-c', '--coin', default='Bitcoin')
@click.option('-n', '-address')
@click.option('-e', '--curve')
@click.option('-d', '--show-display', is_flag=True)
@click.pass_obj
def get_public_node(client, coin, address, curve, show_display):
address_n = client.expand_path(address)
2017-07-12 16:35:54 +00:00
result = client.get_public_node(address_n, ecdsa_curve_name=curve, show_display=show_display, coin_name=coin)
return {
'node': {
'depth': result.node.depth,
'fingerprint': "%08x" % result.node.fingerprint,
'child_num': result.node.child_num,
'chain_code': binascii.hexlify(result.node.chain_code),
'public_key': binascii.hexlify(result.node.public_key),
},
'xpub': result.xpub
}
2017-07-03 15:45:56 +00:00
#
# Message functions
#
@cli.command(help='Sign message using address of given path.')
@click.option('-c', '--coin', default='Bitcoin')
@click.option('-n', '-address')
2017-07-24 14:11:38 +00:00
@click.option('-t', '--script-type', type=click.Choice(['address', 'segwit', 'p2shsegwit']), default='address')
2017-07-03 15:45:56 +00:00
@click.argument('message')
@click.pass_obj
2017-07-24 14:11:38 +00:00
def sign_message(client, coin, address, message, script_type):
2017-07-03 15:45:56 +00:00
address_n = client.expand_path(address)
2017-07-24 14:11:38 +00:00
typemap = {
'address': types.SPENDADDRESS,
'segwit': types.SPENDWITNESS,
'p2shsegwit': types.SPENDP2SHWITNESS,
}
script_type = typemap[script_type]
res = client.sign_message(coin, address_n, message, script_type)
2017-07-04 11:55:05 +00:00
return {
2017-07-03 15:45:56 +00:00
'message': message,
'address': res.address,
'signature': base64.b64encode(res.signature)
}
@cli.command(help='Verify message.')
@click.option('-c', '--coin', default='Bitcoin')
@click.argument('address')
@click.argument('signature')
@click.argument('message')
@click.pass_obj
def verify_message(client, coin, address, signature, message):
signature = base64.b64decode(signature)
2017-07-04 11:55:05 +00:00
return client.verify_message(coin, address, signature, message)
2017-07-03 15:45:56 +00:00
2017-07-12 16:35:54 +00:00
@cli.command(help='Sign message with Ethereum address.')
@click.option('-n', '-address')
@click.argument('message')
@click.pass_obj
def ethereum_sign_message(client, address, message):
address_n = client.expand_path(address)
ret = client.ethereum_sign_message(address_n, message)
output = {
'message': message,
2017-08-17 19:18:49 +00:00
'address': '0x%s' % binascii.hexlify(ret.address).decode(),
'signature': '0x%s' % binascii.hexlify(ret.signature).decode()
2017-07-12 16:35:54 +00:00
}
return output
def ethereum_decode_hex(value):
if value.startswith('0x') or value.startswith('0X'):
return value[2:].decode('hex')
else:
return value.decode('hex')
@cli.command(help='Verify message signed with Ethereum address.')
@click.argument('address')
@click.argument('signature')
@click.argument('message')
@click.pass_obj
def ethereum_verify_message(client, address, signature, message):
address = ethereum_decode_hex(address)
signature = ethereum_decode_hex(signature)
return client.ethereum_verify_message(address, signature, message)
2017-07-03 15:45:56 +00:00
@cli.command(help='Encrypt value by given key and path.')
@click.option('-n', '-address')
@click.argument('key')
@click.argument('value')
@click.pass_obj
def encrypt_keyvalue(client, address, key, value):
address_n = client.expand_path(address)
res = client.encrypt_keyvalue(address_n, key, value)
2017-07-04 11:55:05 +00:00
return binascii.hexlify(res)
2017-07-03 15:45:56 +00:00
@cli.command(help='Decrypt value by given key and path.')
@click.option('-n', '-address')
@click.argument('key')
@click.argument('value')
@click.pass_obj
def decrypt_keyvalue(client, address, key, value):
address_n = client.expand_path(address)
2017-07-04 11:55:05 +00:00
return client.decrypt_keyvalue(address_n, key, value.decode('hex'))
2017-07-03 15:45:56 +00:00
@cli.command(help='Encrypt message.')
@click.option('-c', '--coin', default='Bitcoin')
@click.option('-d', '--display-only', is_flag=True)
@click.option('-n', '-address')
@click.argument('pubkey')
@click.argument('message')
@click.pass_obj
def encrypt_message(client, coin, display_only, address, pubkey, message):
pubkey = binascii.unhexlify(pubkey)
address_n = client.expand_path(address)
res = client.encrypt_message(pubkey, message, display_only, coin, address_n)
2017-07-04 11:55:05 +00:00
return {
2017-07-03 15:45:56 +00:00
'nonce': binascii.hexlify(res.nonce),
'message': binascii.hexlify(res.message),
'hmac': binascii.hexlify(res.hmac),
'payload': base64.b64encode(res.nonce + res.message + res.hmac),
}
@cli.command(help='Decrypt message.')
@click.option('-n', '-address')
@click.argument('payload')
@click.pass_obj
def decrypt_message(client, address, payload):
address_n = client.expand_path(address)
payload = base64.b64decode(payload)
nonce, message, msg_hmac = payload[:33], payload[33:-8], payload[-8:]
2017-07-04 11:55:05 +00:00
return client.decrypt_message(address_n, nonce, message, msg_hmac)
2015-02-22 13:23:39 +00:00
2017-07-03 15:45:56 +00:00
#
# Ethereum functions
#
2015-09-04 18:41:54 +00:00
2016-09-27 22:01:32 +00:00
2017-07-03 15:45:56 +00:00
@cli.command(help='Get Ethereum address in hex encoding.')
@click.option('-n', '-address')
@click.option('-d', '--show-display', is_flag=True)
@click.pass_obj
def ethereum_get_address(client, address, show_display):
address_n = client.expand_path(address)
address = client.ethereum_get_address(address_n, show_display)
2017-08-17 19:18:49 +00:00
return '0x%s' % binascii.hexlify(address).decode()
2017-07-03 15:45:56 +00:00
@cli.command(help='Sign (and optionally publish) Ethereum transaction. Use TO as destination address or set TO to "" for contract creation.')
@click.option('-a', '--host', default='localhost:8545', help='RPC port of ethereum node for automatic gas/nonce estimation')
@click.option('-c', '--chain-id', type=int, help='EIP-155 chain id (replay protection)')
@click.option('-n', '-address', help='BIP-32 path to signing key')
@click.option('-v', '--value', default='0', help='Ether amount to transfer, e.g. "100 milliether"')
@click.option('-g', '--gas-limit', type=int, help='Gas limit - Required for offline signing')
@click.option('-t', '--gas-price', help='Gas price, e.g. "20 nanoether" - Required for offline signing')
@click.option('-i', '--nonce', type=int, help='Transaction counter - Required for offline signing')
@click.option('-d', '--data', default='', help='Data as hex string, e.g. 0x12345678')
@click.option('-p', '--publish', is_flag=True, help='Publish transaction via RPC')
@click.argument('to')
@click.pass_obj
def ethereum_sign_tx(client, host, chain_id, address, value, gas_limit, gas_price, nonce, data, publish, to):
from ethjsonrpc import EthJsonRpc
import rlp
ether_units = {
'wei': 1,
'kwei': 1000,
'babbage': 1000,
'femtoether': 1000,
'mwei': 1000000,
'lovelace': 1000000,
'picoether': 1000000,
'gwei': 1000000000,
'shannon': 1000000000,
'nanoether': 1000000000,
'nano': 1000000000,
'szabo': 1000000000000,
'microether': 1000000000000,
'micro': 1000000000000,
'finney': 1000000000000000,
'milliether': 1000000000000000,
'milli': 1000000000000000,
'ether': 1000000000000000000,
'eth': 1000000000000000000,
}
if ' ' in value:
value, unit = value.split(' ', 1)
if unit.lower() not in ether_units:
raise CallException(types.Failure_DataError, 'Unrecognized ether unit %r' % unit)
value = int(value) * ether_units[unit.lower()]
else:
value = int(value)
if gas_price is not None:
if ' ' in gas_price:
gas_price, unit = gas_price.split(' ', 1)
if unit.lower() not in ether_units:
raise CallException(types.Failure_DataError, 'Unrecognized gas price unit %r' % unit)
gas_price = int(gas_price) * ether_units[unit.lower()]
2016-09-27 22:01:32 +00:00
else:
2017-07-03 15:45:56 +00:00
gas_price = int(gas_price)
if gas_limit is not None:
gas_limit = int(gas_limit)
2017-07-12 16:35:54 +00:00
to_address = ethereum_decode_hex(to)
2017-07-03 15:45:56 +00:00
address_n = client.expand_path(address)
address = '0x%s' % (binascii.hexlify(client.ethereum_get_address(address_n)),)
if gas_price is None or gas_limit is None or nonce is None:
host, port = host.split(':')
eth = EthJsonRpc(host, int(port))
if not data:
data = ''
2017-07-12 16:35:54 +00:00
data = ethereum_decode_hex(data)
2017-07-03 15:45:56 +00:00
if gas_price is None:
gas_price = eth.eth_gasPrice()
if gas_limit is None:
gas_limit = eth.eth_estimateGas(
to_address=to,
from_address=address,
value=('0x%x' % value),
data='0x' + data)
if nonce is None:
nonce = eth.eth_getTransactionCount(address)
sig = client.ethereum_sign_tx(
n=address_n,
nonce=nonce,
gas_price=gas_price,
gas_limit=gas_limit,
to=to_address,
value=value,
data=data,
chain_id=chain_id)
transaction = rlp.encode(
(nonce, gas_price, gas_limit, to_address, value, data) + sig)
tx_hex = '0x%s' % binascii.hexlify(transaction)
if publish:
tx_hash = eth.eth_sendRawTransaction(tx_hex)
2017-07-04 11:55:05 +00:00
return 'Transaction published with ID: %s' % tx_hash
2017-07-03 15:45:56 +00:00
else:
2017-07-04 11:55:05 +00:00
return 'Signed raw transaction: %s' % tx_hex
2017-07-03 15:45:56 +00:00
#
# Main
#
2015-09-04 18:41:54 +00:00
2015-02-22 13:23:39 +00:00
if __name__ == '__main__':
2017-07-03 15:45:56 +00:00
cli()