1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-17 11:58:13 +00:00
trezor-firmware/tools/keyctl-proxy

87 lines
2.4 KiB
Plaintext
Raw Normal View History

#!/usr/bin/env python3
import binascii
import sys
2019-02-21 15:18:49 +00:00
import traceback
import Pyro4
import serpent
2019-02-21 15:18:49 +00:00
from trezorlib import cosi, tools
PORT = 5001
2018-07-31 09:35:09 +00:00
indexmap = {"bootloader": 0, "vendorheader": 1, "firmware": 2}
def get_trezor():
from trezorlib.client import TrezorClient
from trezorlib.transport import get_transport
2019-02-21 15:18:49 +00:00
from trezorlib.ui import ClickUI
2018-07-31 09:35:09 +00:00
2019-02-21 15:18:49 +00:00
return TrezorClient(get_transport(), ui=ClickUI())
def get_path(index):
return "10018'/%d'" % indexmap[index]
@Pyro4.expose
class KeyctlProxy(object):
def get_commit(self, index, digest):
digest = serpent.tobytes(digest)
path = get_path(index)
2018-01-29 14:28:17 +00:00
commit = None
while commit is None:
try:
t = get_trezor()
2018-07-31 09:35:09 +00:00
print(
"\n\n\nCommiting to hash %s with path %s:"
% (binascii.hexlify(digest).decode(), path)
)
2019-02-21 15:18:49 +00:00
commit = cosi.commit(t, tools.parse_path(path), digest)
2018-01-29 14:28:17 +00:00
except Exception as e:
print(e)
2019-02-21 15:18:49 +00:00
traceback.print_exc()
2018-07-31 09:35:09 +00:00
print("Trying again ...")
pk = commit.pubkey
R = commit.commitment
2018-07-31 09:35:09 +00:00
print("Commitment sent!")
return (pk, R)
def get_signature(self, index, digest, global_R, global_pk):
2018-07-31 09:35:09 +00:00
digest, global_R, global_pk = (
serpent.tobytes(digest),
serpent.tobytes(global_R),
serpent.tobytes(global_pk),
)
path = get_path(index)
2018-01-29 14:28:17 +00:00
signature = None
while signature is None:
try:
t = get_trezor()
2018-07-31 09:35:09 +00:00
print(
"\n\n\nSigning hash %s with path %s:"
% (binascii.hexlify(digest).decode(), path)
)
2019-02-21 15:18:49 +00:00
signature = cosi.sign(
t, tools.parse_path(path), digest, global_R, global_pk
2018-07-31 09:35:09 +00:00
)
2018-01-29 14:28:17 +00:00
except Exception as e:
print(e)
2019-02-21 15:18:49 +00:00
traceback.print_exc()
2018-07-31 09:35:09 +00:00
print("Trying again ...")
sig = signature.signature
2018-07-31 09:35:09 +00:00
print("Signature sent!")
return sig
2018-07-31 09:35:09 +00:00
if __name__ == "__main__":
if len(sys.argv) > 1:
ipaddr = sys.argv[1]
else:
2018-07-31 09:35:09 +00:00
print("Usage: keyctl-proxy ipaddress")
sys.exit(1)
daemon = Pyro4.Daemon(host=ipaddr, port=PORT)
proxy = KeyctlProxy()
2018-07-31 09:35:09 +00:00
uri = daemon.register(proxy, "keyctl")
print('keyctl-proxy running at URI: "%s"' % uri)
daemon.requestLoop()