mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-17 13:12:05 +00:00
85 lines
3.3 KiB
Python
85 lines
3.3 KiB
Python
# This file is part of the Trezor project.
|
|
#
|
|
# Copyright (C) 2012-2019 SatoshiLabs and contributors
|
|
#
|
|
# This library is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License version 3
|
|
# as published by the Free Software Foundation.
|
|
#
|
|
# This library is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the License along with this library.
|
|
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
|
|
|
import pytest
|
|
|
|
from trezorlib import btc
|
|
from trezorlib.tools import H_, CallException
|
|
|
|
from .common import MNEMONIC12, TrezorTest
|
|
|
|
|
|
class TestMsgGetpublickeyCurve(TrezorTest):
|
|
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
|
def test_default_curve(self, client):
|
|
assert (
|
|
btc.get_public_node(client, [H_(111), 42]).node.public_key.hex()
|
|
== "02e7fcec053f0df94d88c86447970743e8a1979d242d09338dcf8687a9966f7fbc"
|
|
)
|
|
assert (
|
|
btc.get_public_node(client, [H_(111), H_(42)]).node.public_key.hex()
|
|
== "03ce7b690969d773ba9ed212464eb2b534b87b9b8a9383300bddabe1f093f79220"
|
|
)
|
|
|
|
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
|
def test_secp256k1_curve(self, client):
|
|
assert (
|
|
btc.get_public_node(
|
|
client, [H_(111), 42], ecdsa_curve_name="secp256k1"
|
|
).node.public_key.hex()
|
|
== "02e7fcec053f0df94d88c86447970743e8a1979d242d09338dcf8687a9966f7fbc"
|
|
)
|
|
assert (
|
|
btc.get_public_node(
|
|
client, [H_(111), H_(42)], ecdsa_curve_name="secp256k1"
|
|
).node.public_key.hex()
|
|
== "03ce7b690969d773ba9ed212464eb2b534b87b9b8a9383300bddabe1f093f79220"
|
|
)
|
|
|
|
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
|
def test_nist256p1_curve(self, client):
|
|
assert (
|
|
btc.get_public_node(
|
|
client, [H_(111), 42], ecdsa_curve_name="nist256p1"
|
|
).node.public_key.hex()
|
|
== "02a9ce59b32bd64a70bc52aca96e5d09af65c6b9593ba2a60af8fccfe1437f2129"
|
|
)
|
|
assert (
|
|
btc.get_public_node(
|
|
client, [H_(111), H_(42)], ecdsa_curve_name="nist256p1"
|
|
).node.public_key.hex()
|
|
== "026fe35d8afed67dbf0561a1d32922e8ad0cd0d86effbc82be970cbed7d9bab2c2"
|
|
)
|
|
|
|
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
|
def test_ed25519_curve(self, client):
|
|
# ed25519 curve does not support public derivation, so test only private derivation paths
|
|
assert (
|
|
btc.get_public_node(
|
|
client, [H_(111), H_(42)], ecdsa_curve_name="ed25519"
|
|
).node.public_key.hex()
|
|
== "0069a14b478e508eab6e93303f4e6f5c50b8136627830f2ed5c3a835fc6c0ea2b7"
|
|
)
|
|
assert (
|
|
btc.get_public_node(
|
|
client, [H_(111), H_(65535)], ecdsa_curve_name="ed25519"
|
|
).node.public_key.hex()
|
|
== "00514f73a05184458611b14c348fee4fd988d36cf3aee7207737861bac611de991"
|
|
)
|
|
# test failure when using public derivation
|
|
with pytest.raises(CallException):
|
|
btc.get_public_node(client, [H_(111), 42], ecdsa_curve_name="ed25519")
|