mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-02-13 16:12:18 +00:00
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from trezor import ui
|
|
from trezor.crypto.hashlib import sha3_256
|
|
from trezor.utils import HashWriter
|
|
from trezor.messages.EthereumMessageSignature import EthereumMessageSignature
|
|
from trezor.ui.text import Text
|
|
from trezor.crypto.curve import secp256k1
|
|
from apps.common import seed
|
|
from apps.common.confirm import require_confirm
|
|
from apps.common.signverify import split_message
|
|
|
|
|
|
def message_digest(message):
|
|
|
|
h = HashWriter(sha3_256)
|
|
signed_message_header = '\x19Ethereum Signed Message:\n'
|
|
h.extend(signed_message_header)
|
|
h.extend(str(len(message)))
|
|
h.extend(message)
|
|
|
|
return h.get_digest(True)
|
|
|
|
|
|
async def ethereum_sign_message(ctx, msg):
|
|
|
|
await require_confirm_sign_message(ctx, msg.message)
|
|
|
|
address_n = msg.address_n or ()
|
|
node = await seed.derive_node(ctx, address_n)
|
|
|
|
signature = secp256k1.sign(node.private_key(), message_digest(msg.message), False)
|
|
|
|
sig = EthereumMessageSignature()
|
|
sig.address = node.ethereum_pubkeyhash()
|
|
sig.signature = signature[1:] + bytearray([signature[0]])
|
|
return sig
|
|
|
|
|
|
async def require_confirm_sign_message(ctx, message):
|
|
message = split_message(message)
|
|
content = Text('Sign ETH message', ui.ICON_DEFAULT, max_lines=5, *message)
|
|
await require_confirm(ctx, content)
|