parent
099767d592
commit
97d402eba8
@ -1,44 +1,35 @@
|
||||
from trezor.dispatcher import register
|
||||
from trezor.utils import unimport_func
|
||||
from trezor.wire import register_type, protobuf_handler
|
||||
from trezor.utils import unimport
|
||||
from trezor.messages.wire_types import \
|
||||
LoadDevice, ResetDevice, WipeDevice, RecoveryDevice
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_LoadDevice(mtype, mbuf):
|
||||
from trezor.messages.LoadDevice import LoadDevice
|
||||
message = LoadDevice.loads(mbuf)
|
||||
@unimport
|
||||
def dispatch_LoadDevice(*args, **kwargs):
|
||||
from .layout_load_device import layout_load_device
|
||||
return layout_load_device(message)
|
||||
return layout_load_device(*args, **kwargs)
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_ResetDevice(mtype, mbuf):
|
||||
from trezor.messages.ResetDevice import ResetDevice
|
||||
message = ResetDevice.loads(mbuf)
|
||||
@unimport
|
||||
def dispatch_ResetDevice(*args, **kwargs):
|
||||
from .layout_reset_device import layout_reset_device
|
||||
return layout_reset_device(message)
|
||||
return layout_reset_device(*args, **kwargs)
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_WipeDevice(mtype, mbuf):
|
||||
from trezor.messages.WipeDevice import WipeDevice
|
||||
message = WipeDevice.loads(mbuf)
|
||||
@unimport
|
||||
def dispatch_WipeDevice(*args, **kwargs):
|
||||
from .layout_wipe_device import layout_wipe_device
|
||||
return layout_wipe_device(message)
|
||||
return layout_wipe_device(*args, **kwargs)
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_RecoveryDevice(mtype, mbuf):
|
||||
from trezor.messages.RecoveryDevice import RecoveryDevice
|
||||
message = RecoveryDevice.loads(mbuf)
|
||||
@unimport
|
||||
def dispatch_RecoveryDevice(*args, **kwargs):
|
||||
from .layout_recovery_device import layout_recovery_device
|
||||
return layout_recovery_device(message)
|
||||
return layout_recovery_device(*args, **kwargs)
|
||||
|
||||
|
||||
def boot():
|
||||
LoadDevice = 13
|
||||
register(LoadDevice, dispatch_LoadDevice)
|
||||
ResetDevice = 14
|
||||
register(ResetDevice, dispatch_ResetDevice)
|
||||
WipeDevice = 5
|
||||
register(WipeDevice, dispatch_WipeDevice)
|
||||
RecoveryDevice = 45
|
||||
register(RecoveryDevice, dispatch_RecoveryDevice)
|
||||
register_type(LoadDevice, protobuf_handler, dispatch_LoadDevice)
|
||||
register_type(ResetDevice, protobuf_handler, dispatch_ResetDevice)
|
||||
register_type(WipeDevice, protobuf_handler, dispatch_WipeDevice)
|
||||
register_type(RecoveryDevice, protobuf_handler, dispatch_RecoveryDevice)
|
||||
|
@ -1,42 +1,28 @@
|
||||
from trezor.dispatcher import register
|
||||
from trezor.utils import unimport_func
|
||||
from trezor.wire import register_type, protobuf_handler
|
||||
from trezor.utils import unimport
|
||||
from trezor.messages.wire_types import \
|
||||
GetPublicKey, SignTx, SignMessage
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_GetPublicKey(mtype, mbuf):
|
||||
from trezor.messages.GetPublicKey import GetPublicKey
|
||||
|
||||
message = GetPublicKey.loads(mbuf)
|
||||
|
||||
@unimport
|
||||
def dispatch_GetPublicKey(*args, **kwargs):
|
||||
from .layout_get_public_key import layout_get_public_key
|
||||
return layout_get_public_key(message)
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_SignTx(mtype, mbuf):
|
||||
from trezor.messages.SignTx import SignTx
|
||||
return layout_get_public_key(*args, **kwargs)
|
||||
|
||||
message = SignTx.loads(mbuf)
|
||||
|
||||
@unimport
|
||||
def dispatch_SignTx(*args, **kwargs):
|
||||
from .layout_sign_tx import layout_sign_tx
|
||||
return layout_sign_tx(message)
|
||||
return layout_sign_tx(*args, **kwargs)
|
||||
|
||||
|
||||
@unimport_func
|
||||
def dispatch_SignMessage(mtype, mbuf):
|
||||
from trezor.messages.SignMessage import SignMessage
|
||||
|
||||
message = SignMessage.loads(mbuf)
|
||||
|
||||
@unimport
|
||||
def dispatch_SignMessage(*args, **kwargs):
|
||||
from .layout_sign_message import layout_sign_message
|
||||
return layout_sign_message(message)
|
||||
return layout_sign_message(*args, **kwargs)
|
||||
|
||||
|
||||
def boot():
|
||||
GetPublicKey = 11
|
||||
register(GetPublicKey, dispatch_GetPublicKey)
|
||||
SignTx = 15
|
||||
register(SignTx, dispatch_SignTx)
|
||||
SignMessage = 38
|
||||
register(SignMessage, dispatch_SignMessage)
|
||||
|
||||
register_type(GetPublicKey, protobuf_handler, dispatch_GetPublicKey)
|
||||
register_type(SignTx, protobuf_handler, dispatch_SignTx)
|
||||
register_type(SignMessage, protobuf_handler, dispatch_SignMessage)
|
||||
|
@ -1,8 +1,8 @@
|
||||
from trezor import wire, ui
|
||||
from trezor.utils import unimport_gen
|
||||
from trezor.utils import unimport
|
||||
|
||||
|
||||
@unimport_gen
|
||||
@unimport
|
||||
def layout_sign_tx(message):
|
||||
ui.clear()
|
||||
print('sending')
|
||||
|
Loading…
Reference in new issue