Example #1
0
    def callback_ButtonRequest(self, msg):
        log("ButtonRequest code: " + get_buttonrequest_value(msg.code))

        log("Pressing button " + str(self.button))
        if self.button_wait:
            log("Waiting %d seconds " % self.button_wait)
            time.sleep(self.button_wait)
        self.debug.press_button(self.button)
        return proto.ButtonAck()
Example #2
0
 def callback_ButtonRequest(self, msg):
     # log("Sending ButtonAck for %s " % get_buttonrequest_value(msg.code))
     return proto.ButtonAck()
Example #3
0
def Deserialization(msg_id, message):
    if msg_id == 38:
        print('MessageType_SignMessage: ', message)
        signmessage = tz_messages.SignMessage()
        signmessage.ParseFromString(message)
        print(signmessage)
        print('Type:', type(signmessage.message))  # print int class
        print('Len:', len(signmessage.message))
        for val in signmessage.message:
            print(val)
    elif msg_id == 39:
        print('MessageType_VerifyMessage: ', message)
        verify_message = tz_messages.VerifyMessage()
        verify_message.ParseFromString(message)  # SerializeToOstream
        print(verify_message)
    elif msg_id == 40:
        print('MessageType_MessageSignature: ', message)
        signature_message = tz_messages.MessageSignature()
        signature_message.ParseFromString(message)
        print(signature_message)
        hex_list = []
        for val in signature_message.signature:
            hex_list.append(hex(val))
        print('Signature:', hex_list)
    elif msg_id == 0:
        print('MessageType_Initialize: ', message)
        init_message = tz_messages.Initialize()
        init_message.ParseFromString(message)
        print(init_message)
    elif msg_id == 55:
        print('MessageType_GetFeatures: ', message)
        getFeatures_message = tz_messages.GetFeatures()
        getFeatures_message.ParseFromString(message)
        print(getFeatures_message)
    elif msg_id == 17:
        print('MessageType_Features: ', message)
        features_message = tz_messages.Features()
        features_message.ParseFromString(message)
        print(features_message)

    elif msg_id == 11:
        print('MessageType_GetPublicKey: ', message)
        GetPublicKey_message = tz_messages.GetPublicKey()
        GetPublicKey_message.ParseFromString(message)
        print(GetPublicKey_message)
        print('Type:', type(GetPublicKey_message.address_n))
        print('len:', len(GetPublicKey_message.address_n))
    elif msg_id == 12:
        print('MessageType_PublicKey: ', message)
        PublicKey_message = tz_messages.PublicKey()
        PublicKey_message.ParseFromString(message)
        print(PublicKey_message)
    elif msg_id == 29:
        print('MessageType_GetAddress: ', message)
        GetAddress_message = tz_messages.GetAddress()
        GetAddress_message.ParseFromString(message)
        print(GetAddress_message)
    elif msg_id == 30:
        print('MessageType_Address: ', message)
        Address_message = tz_messages.Address()
        Address_message.ParseFromString(message)
        print(Address_message)
    elif msg_id == 34:
        print('MessageType_BackupDevice: ', message)
        BackupDevice_message = tz_messages.BackupDevice()
        BackupDevice_message.ParseFromString(message)
        print(BackupDevice_message)
    elif msg_id == 2:
        print('MessageType_Success: ', message)
        MessageType_Success = tz_messages.Success()
        MessageType_Success.ParseFromString(message)
        print(MessageType_Success)
    elif msg_id == 26:
        print('MessageType_ButtonRequest: ', message)
        MessageType_ButtonRequest = tz_messages.ButtonRequest()
        MessageType_ButtonRequest.ParseFromString(message)
        print(MessageType_ButtonRequest)
    elif msg_id == 27:
        print('MessageType_ButtonAck: ', message)
        MessageType_ButtonAck = tz_messages.ButtonAck()
        MessageType_ButtonAck.ParseFromString(message)
        print(MessageType_ButtonAck)
    elif msg_id == 25:
        print('MessageType_ApplySettings: ', message)
        MessageType_ApplySettings = tz_messages.ApplySettings()
        MessageType_ApplySettings.ParseFromString(message)
        print(MessageType_ApplySettings)
    elif msg_id == 4:
        print('MessageType_ChangePin: ', message)
        MessageType_ChangePin = tz_messages.ChangePin()
        MessageType_ChangePin.ParseFromString(message)
        print(MessageType_ChangePin)
    elif msg_id == 18:
        print('MessageType_PinMatrixRequest: ', message)
        MessageType_PinMatrixRequest = tz_messages.PinMatrixRequest()
        MessageType_PinMatrixRequest.ParseFromString(message)
        print(MessageType_PinMatrixRequest)
    elif msg_id == 19:
        print('MessageType_PinMatrixAck: ', message)
        MessageType_PinMatrixAck = tz_messages.PinMatrixAck()
        MessageType_PinMatrixAck.ParseFromString(message)
        print(MessageType_PinMatrixAck)
    elif msg_id == 5:
        print('MessageType_WipeDevice: ', message)
        MessageType_WipeDevice = tz_messages.WipeDevice()
        MessageType_WipeDevice.ParseFromString(message)
        print(MessageType_WipeDevice)
    elif msg_id == 14:
        print('MessageType_ResetDevice: ', message)
        MessageType_ResetDevice = tz_messages.ResetDevice()
        MessageType_ResetDevice.ParseFromString(message)
        print(MessageType_ResetDevice)
    elif msg_id == 35:
        print('MessageType_EntropyRequest: ', message)
        MessageType_EntropyRequest = tz_messages.EntropyRequest()
        MessageType_EntropyRequest.ParseFromString(message)
        print(MessageType_EntropyRequest)
    elif msg_id == 36:
        print('MessageType_EntropyAck: ', message)
        MessageType_EntropyAck = tz_messages.EntropyAck()
        MessageType_EntropyAck.ParseFromString(message)
        print(MessageType_EntropyAck)

    elif msg_id == 6:
        print('MessageType_FirmwareErase: ', message)
        MessageType_FirmwareErase = tz_messages.FirmwareErase()
        MessageType_FirmwareErase.ParseFromString(message)
        print(MessageType_FirmwareErase)
    elif msg_id == 7:
        print('MessageType_FirmwareUpload: ', message)
        MessageType_FirmwareUpload = tz_messages.FirmwareUpload()
        MessageType_FirmwareUpload.ParseFromString(message)
        print(MessageType_FirmwareUpload)
    elif msg_id == 8:
        print('MessageType_FirmwareRequest: ', message)
        MessageType_FirmwareRequest = tz_messages.FirmwareRequest()
        MessageType_FirmwareRequest.ParseFromString(message)
        print(MessageType_FirmwareRequest)

    elif msg_id == 45:
        print('MessageType_RecoveryDevice: ', message)
        MessageType_RecoveryDevice = tz_messages.RecoveryDevice()
        MessageType_RecoveryDevice.ParseFromString(message)
        print(MessageType_RecoveryDevice)
    elif msg_id == 46:
        print('MessageType_WordRequest: ', message)
        MessageType_WordRequest = tz_messages.WordRequest()
        MessageType_WordRequest.ParseFromString(message)
        print(MessageType_WordRequest)
    elif msg_id == 47:
        print('MessageType_WordAck: ', message)
        MessageType_WordAck = tz_messages.WordAck()
        MessageType_WordAck.ParseFromString(message)
        print(MessageType_WordAck)

    elif msg_id == 41:
        print('MessageType_PassphraseRequest: ', message)
        MessageType_PassphraseRequest = tz_messages.PassphraseRequest()
        MessageType_PassphraseRequest.ParseFromString(message)
        print(MessageType_PassphraseRequest)
    elif msg_id == 42:
        print('MessageType_PassphraseAck: ', message)
        MessageType_PassphraseAck = tz_messages.PassphraseAck()
        MessageType_PassphraseAck.ParseFromString(message)
        print(MessageType_PassphraseAck)

    # Ethereum
    elif msg_id == 58:
        print('MessageType_EthereumSignTx: ', message)
        MessageType_EthereumSignTx = tz_messages.EthereumSignTx()
        MessageType_EthereumSignTx.ParseFromString(message)
        print(MessageType_EthereumSignTx)
    elif msg_id == 59:
        print('MessageType_EthereumTxRequest: ', message)
        MessageType_EthereumTxRequest = tz_messages.EthereumTxRequest()
        MessageType_EthereumTxRequest.ParseFromString(message)
        print(MessageType_EthereumTxRequest)
    else:
        print('Unrecognized Message Id, Please modify the program.')