test_oep4_neovm_transfer_from_sign_cmd.py

import pytest

from application_client.boilerplate_transaction import Transaction
from application_client.boilerplate_command_sender import BoilerplateCommandSender, Errors
from application_client.boilerplate_response_unpacker import unpack_get_public_key_response, unpack_sign_tx_response
from ragger.error import ExceptionRAPDU
from ragger.navigator import NavInsID
from utils import check_signature_validity
import hashlib
# In this tests we check the behavior of the device when asked to sign a oep4 transaction


# In this test we send to the device a oep4 transaction to sign and validate it on screen
# The oep4 transaction is short and will be sent in one chunk
# We will ensure that the displayed information is correct by using screenshots comparison
def test_sign_oep4_neo_vm_transfer_from_tx_short_tx(backend, scenario_navigator):
    # Use the app interface instead of raw interface
    client = BoilerplateCommandSender(backend)
    # The path used for this entire test
    path: str = "m/44'/1024'/0'/0/0"

    # First we need to get the public key of the device in order to build the oep4 transaction
    rapdu = client.get_public_key(path=path)
    _, public_key, _, _ = unpack_get_public_key_response(rapdu.data)

    # Create the oep4 transaction that will be sent to the device for signing
    ope4_transaction = Transaction(
        rawtx = "00d284972568c409000000000000204e0000000000009b50e5a049679c32e4660266f8814001bde3e6fc6ecd841ff217c68dc2faf8106368c5577b1204ece9590c7472616e7366657246726f6d9b50e5a049679c32e4660266f8814001bde3e6fcae6393b337e4a8d856ec00d75af134fdcd6bfe29e7f1d24e2d3bb7fa3051c732507ee74753756ff36b53140000000000000000000000000000"
    ).serialize()

    # Send the sign device instruction.
    # As it requires on-screen validation, the function is asynchronous.
    # It will yield the result when the navigation is done
    with client.sign_tx(path=path, transaction=ope4_transaction):
        # Validate the on-screen request by performing the navigation appropriate for this device
        scenario_navigator.review_approve()

    # The device as yielded the result, parse it and ensure that the signature is correct
    response = client.get_async_response().data
    _, der_sig, _ = unpack_sign_tx_response(response)
    first_hash = hashlib.sha256(ope4_transaction).digest()
    second_hash = hashlib.sha256(first_hash).digest()
    assert check_signature_validity(public_key, der_sig, second_hash)


# Oep4 Transaction signature refused test
# The test will ask for a oep4 transaction signature that will be refused on screen
def test_sign_oep4_neo_vm_transfer_from_tx_refused(backend, scenario_navigator):
    # Use the app interface instead of raw interface
    client = BoilerplateCommandSender(backend)
    path: str = "m/44'/1024'/0'/0/0"

    rapdu = client.get_public_key(path=path)
    _, pub_key, _, _ = unpack_get_public_key_response(rapdu.data)

    oep4_transaction = Transaction(
        rawtx = "00d1a01cc36fc409000000000000204e0000000000009b50e5a049679c32e4660266f8814001bde3e6fc67036b531414e7f1d24e2d3bb7fa3051c732507ee74753756ff314ae6393b337e4a8d856ec00d75af134fdcd6bfe29149b50e5a049679c32e4660266f8814001bde3e6fc54c10c7472616e7366657246726f6d676d03a16843e61e5d96d389e178717ed57f9da4e500"
    ).serialize()

    with pytest.raises(ExceptionRAPDU) as e:
        with client.sign_tx(path=path, transaction=oep4_transaction):
            scenario_navigator.review_reject()

    # Assert that we have received a refusal
    assert e.value.status == Errors.SW_DENY
    assert len(e.value.data) == 0