81 lines
3.1 KiB
Python
81 lines
3.1 KiB
Python
|
import pytest
|
||
|
|
from typing import Optional
|
||
|
|
from pathlib import Path
|
||
|
|
from ragger.error import ExceptionRAPDU
|
||
|
|
from ragger.firmware import Firmware
|
||
|
|
from ragger.backend import BackendInterface
|
||
|
|
from ragger.navigator import Navigator, NavInsID
|
||
|
|
from ledger_app_clients.ethereum.client import EthAppClient, StatusWord
|
||
|
|
from ledger_app_clients.ethereum.settings import SettingID, settings_toggle
|
||
|
|
import ledger_app_clients.ethereum.response_parser as ResponseParser
|
||
|
|
from ragger.bip import calculate_public_key_and_chaincode, CurveChoice
|
||
|
|
|
||
|
|
ROOT_SCREENSHOT_PATH = Path(__file__).parent
|
||
|
|
|
||
|
|
@pytest.fixture(params=[True, False])
|
||
|
|
def with_chaincode(request) -> bool:
|
||
|
|
return request.param
|
||
|
|
|
||
|
|
@pytest.fixture(params=[None, 1, 2, 5, 137])
|
||
|
|
def chain(request) -> Optional[int]:
|
||
|
|
return request.param
|
||
|
|
|
||
|
|
def get_moves(firmware: Firmware,
|
||
|
|
navigator: BackendInterface,
|
||
|
|
chain: Optional[int] = None,
|
||
|
|
reject: bool = False):
|
||
|
|
moves = list()
|
||
|
|
|
||
|
|
if firmware.is_nano:
|
||
|
|
moves += [ NavInsID.RIGHT_CLICK ]
|
||
|
|
if firmware.device == "nanos":
|
||
|
|
moves += [ NavInsID.RIGHT_CLICK ] * 3
|
||
|
|
else:
|
||
|
|
moves += [ NavInsID.RIGHT_CLICK ]
|
||
|
|
if reject:
|
||
|
|
moves += [ NavInsID.RIGHT_CLICK ]
|
||
|
|
moves += [ NavInsID.BOTH_CLICK ]
|
||
|
|
else:
|
||
|
|
moves += [ NavInsID.USE_CASE_REVIEW_TAP ]
|
||
|
|
if chain is not None and chain > 1:
|
||
|
|
moves += [ NavInsID.USE_CASE_ADDRESS_CONFIRMATION_TAP ]
|
||
|
|
if reject:
|
||
|
|
moves += [ NavInsID.USE_CASE_ADDRESS_CONFIRMATION_CANCEL ]
|
||
|
|
else:
|
||
|
|
moves += [ NavInsID.USE_CASE_ADDRESS_CONFIRMATION_CONFIRM ]
|
||
|
|
|
||
|
|
return moves
|
||
|
|
|
||
|
|
def test_get_pk_rejected(firmware: Firmware,
|
||
|
|
backend: BackendInterface,
|
||
|
|
navigator: Navigator):
|
||
|
|
app_client = EthAppClient(backend)
|
||
|
|
|
||
|
|
try:
|
||
|
|
with app_client.get_public_addr():
|
||
|
|
navigator.navigate_and_compare(ROOT_SCREENSHOT_PATH,
|
||
|
|
"get_pk_rejected",
|
||
|
|
get_moves(firmware, navigator, reject=True))
|
||
|
|
except ExceptionRAPDU as e:
|
||
|
|
assert e.status == StatusWord.CONDITION_NOT_SATISFIED
|
||
|
|
else:
|
||
|
|
assert False # An exception should have been raised
|
||
|
|
|
||
|
|
def test_get_pk(firmware: Firmware,
|
||
|
|
backend: BackendInterface,
|
||
|
|
navigator: Navigator,
|
||
|
|
with_chaincode: bool,
|
||
|
|
chain: Optional[int]):
|
||
|
|
app_client = EthAppClient(backend)
|
||
|
|
|
||
|
|
with app_client.get_public_addr(chaincode=with_chaincode, chain_id=chain):
|
||
|
|
navigator.navigate_and_compare(ROOT_SCREENSHOT_PATH,
|
||
|
|
"get_pk_%s" % (chain),
|
||
|
|
get_moves(firmware, navigator, chain=chain))
|
||
|
|
pk, addr, chaincode = ResponseParser.pk_addr(app_client.response().data, with_chaincode)
|
||
|
|
ref_pk, ref_chaincode = calculate_public_key_and_chaincode(curve=CurveChoice.Secp256k1,
|
||
|
|
path="m/44'/60'/0'/0/0")
|
||
|
|
assert pk.hex() == ref_pk
|
||
|
|
if with_chaincode:
|
||
|
|
assert chaincode.hex() == ref_chaincode
|