83 lines
3.0 KiB
Python
83 lines
3.0 KiB
Python
import pytest
|
|
from typing import Optional
|
|
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
|
|
import ledger_app_clients.ethereum.response_parser as ResponseParser
|
|
from ragger.bip import calculate_public_key_and_chaincode, CurveChoice
|
|
from constants import ROOT_SNAPSHOT_PATH
|
|
|
|
|
|
@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_SNAPSHOT_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_SNAPSHOT_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
|