255 lines
7.9 KiB
Python
255 lines
7.9 KiB
Python
from typing import Callable, Optional, Any
|
|
import json
|
|
import pytest
|
|
from web3 import Web3
|
|
|
|
from ragger.error import ExceptionRAPDU
|
|
from ragger.firmware import Firmware
|
|
from ragger.backend import BackendInterface
|
|
from ragger.navigator.navigation_scenario import NavigateWithScenario
|
|
|
|
from constants import ABIS_FOLDER
|
|
|
|
from client.client import EthAppClient, StatusWord
|
|
import client.response_parser as ResponseParser
|
|
from client.utils import get_selector_from_data, recover_transaction
|
|
|
|
|
|
BIP32_PATH = "m/44'/60'/0'/0/0"
|
|
NONCE = 21
|
|
GAS_PRICE = 13
|
|
GAS_LIMIT = 21000
|
|
FROM = bytes.fromhex("1122334455667788990011223344556677889900")
|
|
TO = bytes.fromhex("0099887766554433221100998877665544332211")
|
|
NFTS = [(1, 3), (5, 2), (7, 4)] # tuples of (token_id, amount)
|
|
DATA = "Some data".encode()
|
|
DEVICE_ADDR: Optional[bytes] = None
|
|
|
|
|
|
class NFTCollection:
|
|
addr: bytes
|
|
name: str
|
|
chain_id: int
|
|
|
|
def __init__(self, addr: bytes, name: str, chain_id: int, contract):
|
|
self.addr = addr
|
|
self.name = name
|
|
self.chain_id = chain_id
|
|
self.contract = contract
|
|
|
|
|
|
class Action:
|
|
fn_name: str
|
|
fn_args: list[Any]
|
|
|
|
def __init__(self, fn_name: str, fn_args: list[Any]):
|
|
self.fn_name = fn_name
|
|
self.fn_args = fn_args
|
|
|
|
|
|
def common_test_nft(firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario,
|
|
collec: NFTCollection,
|
|
action: Action,
|
|
reject: bool,
|
|
plugin_name: str):
|
|
global DEVICE_ADDR
|
|
app_client = EthAppClient(backend)
|
|
|
|
if firmware == Firmware.NANOS:
|
|
pytest.skip("Not supported on LNS")
|
|
|
|
if DEVICE_ADDR is None: # to only have to request it once
|
|
with app_client.get_public_addr(display=False):
|
|
pass
|
|
_, DEVICE_ADDR, _ = ResponseParser.pk_addr(app_client.response().data)
|
|
|
|
data = collec.contract.encode_abi(action.fn_name, action.fn_args)
|
|
app_client.set_plugin(plugin_name,
|
|
collec.addr,
|
|
get_selector_from_data(data),
|
|
collec.chain_id)
|
|
|
|
app_client.provide_nft_metadata(collec.name, collec.addr, collec.chain_id)
|
|
tx_params = {
|
|
"nonce": NONCE,
|
|
"gasPrice": Web3.to_wei(GAS_PRICE, "gwei"),
|
|
"gas": GAS_LIMIT,
|
|
"to": collec.addr,
|
|
"value": 0,
|
|
"chainId": collec.chain_id,
|
|
"data": data,
|
|
}
|
|
with app_client.sign(BIP32_PATH, tx_params):
|
|
test_name = f"{plugin_name.lower()}_{action.fn_name}_{str(collec.chain_id)}"
|
|
if reject:
|
|
test_name += "-rejected"
|
|
scenario_navigator.review_reject(test_name=test_name)
|
|
else:
|
|
if firmware.is_nano:
|
|
end_text = "Accept"
|
|
else:
|
|
end_text = "Sign"
|
|
scenario_navigator.review_approve(test_name=test_name, custom_screen_text=end_text)
|
|
|
|
# verify signature
|
|
vrs = ResponseParser.signature(app_client.response().data)
|
|
addr = recover_transaction(tx_params, vrs)
|
|
assert addr == DEVICE_ADDR
|
|
|
|
|
|
def common_test_nft_reject(test_fn: Callable,
|
|
firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario,
|
|
collec: NFTCollection,
|
|
action: Action):
|
|
with pytest.raises(ExceptionRAPDU) as e:
|
|
test_fn(firmware, backend, scenario_navigator, collec, action, True)
|
|
assert e.value.status == StatusWord.CONDITION_NOT_SATISFIED
|
|
|
|
# ERC-721
|
|
|
|
|
|
ERC721_PLUGIN = "ERC721"
|
|
|
|
with open(f"{ABIS_FOLDER}/erc721.json", encoding="utf-8") as file:
|
|
contract_erc721 = Web3().eth.contract(
|
|
abi=json.load(file),
|
|
address=bytes(20)
|
|
)
|
|
|
|
collecs_721 = [
|
|
NFTCollection(bytes.fromhex("bc4ca0eda7647a8ab7c2061c2e118a18a936f13d"),
|
|
"Bored Ape Yacht Club",
|
|
1,
|
|
contract_erc721),
|
|
NFTCollection(bytes.fromhex("670fd103b1a08628e9557cd66b87ded841115190"),
|
|
"y00ts",
|
|
137,
|
|
contract_erc721),
|
|
NFTCollection(bytes.fromhex("2909cf13e458a576cdd9aab6bd6617051a92dacf"),
|
|
"goerlirocks",
|
|
5,
|
|
contract_erc721),
|
|
]
|
|
actions_721 = [
|
|
Action("safeTransferFrom", [FROM, TO, NFTS[0][0], DATA]),
|
|
Action("safeTransferFrom", [FROM, TO, NFTS[0][0]]),
|
|
Action("transferFrom", [FROM, TO, NFTS[0][0]]),
|
|
Action("approve", [TO, NFTS[0][0]]),
|
|
Action("setApprovalForAll", [TO, False]),
|
|
]
|
|
|
|
|
|
@pytest.fixture(name="collec_721", params=collecs_721)
|
|
def collec_721_fixture(request) -> NFTCollection:
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(name="action_721", params=actions_721)
|
|
def action_721_fixture(request) -> Action:
|
|
return request.param
|
|
|
|
|
|
def test_erc721(firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario,
|
|
collec_721: NFTCollection,
|
|
action_721: Action,
|
|
reject: bool = False):
|
|
common_test_nft(firmware,
|
|
backend,
|
|
scenario_navigator,
|
|
collec_721,
|
|
action_721,
|
|
reject,
|
|
ERC721_PLUGIN)
|
|
|
|
|
|
def test_erc721_reject(firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario):
|
|
common_test_nft_reject(test_erc721,
|
|
firmware,
|
|
backend,
|
|
scenario_navigator,
|
|
collecs_721[0],
|
|
actions_721[0])
|
|
|
|
|
|
# ERC-1155
|
|
|
|
ERC1155_PLUGIN = "ERC1155"
|
|
|
|
with open(f"{ABIS_FOLDER}/erc1155.json", encoding="utf-8") as file:
|
|
contract_erc1155 = Web3().eth.contract(
|
|
abi=json.load(file),
|
|
address=bytes(20)
|
|
)
|
|
|
|
|
|
collecs_1155 = [
|
|
NFTCollection(bytes.fromhex("495f947276749ce646f68ac8c248420045cb7b5e"),
|
|
"OpenSea Shared Storefront",
|
|
1,
|
|
contract_erc1155),
|
|
NFTCollection(bytes.fromhex("2953399124f0cbb46d2cbacd8a89cf0599974963"),
|
|
"OpenSea Collections",
|
|
137,
|
|
contract_erc1155),
|
|
NFTCollection(bytes.fromhex("f4910c763ed4e47a585e2d34baa9a4b611ae448c"),
|
|
"OpenSea Collections",
|
|
5,
|
|
contract_erc1155),
|
|
]
|
|
actions_1155 = [
|
|
Action("safeTransferFrom", [FROM, TO, NFTS[0][0], NFTS[0][1], DATA]),
|
|
Action("safeBatchTransferFrom",
|
|
[
|
|
FROM,
|
|
TO,
|
|
list(map(lambda nft: nft[0], NFTS)),
|
|
list(map(lambda nft: nft[1], NFTS)),
|
|
DATA
|
|
]),
|
|
Action("setApprovalForAll", [TO, False]),
|
|
]
|
|
|
|
|
|
@pytest.fixture(name="collec_1155", params=collecs_1155)
|
|
def collec_1155_fixture(request) -> bool:
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(name="action_1155", params=actions_1155)
|
|
def action_1155_fixture(request) -> Action:
|
|
return request.param
|
|
|
|
|
|
def test_erc1155(firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario,
|
|
collec_1155: NFTCollection,
|
|
action_1155: Action,
|
|
reject: bool = False):
|
|
common_test_nft(firmware,
|
|
backend,
|
|
scenario_navigator,
|
|
collec_1155,
|
|
action_1155,
|
|
reject,
|
|
ERC1155_PLUGIN)
|
|
|
|
|
|
def test_erc1155_reject(firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario):
|
|
common_test_nft_reject(test_erc1155,
|
|
firmware,
|
|
backend,
|
|
scenario_navigator,
|
|
collecs_1155[0],
|
|
actions_1155[0])
|