trezor-firmware/tests/device_tests/test_msg_ethereum_signtx_eip155.py

# 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>.

from collections import namedtuple

import pytest

from trezorlib import ethereum
from trezorlib.tools import H_

Vector = namedtuple("Vector", "chain_id nonce sig_v sig_r sig_s value gas_limit data")


VECTORS = [
    Vector(*t)
    for t in [
        (
            3,
            0,
            42,
            "cde31d8ab07d423d5e52aeb148180528ea54974cdb4c5578499c0137ec24d892",
            "41fc58955b3b3e3f3b2aced65e11e8a3cb6339027f943bec3d504d6398b69dd2",
            100000000000000000,
            21000,
            None,
        ),
        Vector(
            3,
            1,
            41,
            "57951fed170f3765dea164d65acd31373799db32ec572e213b1d9a1209956b98",
            "0971f8830c0e2e89919309f217ed2eadb0c63d647e016d220729ce79d27c24a0",
            100000000000000000,
            21000,
            None,
        ),
        (
            3,
            2,
            42,
            "73744f66231690edd9eed2ab3c2b56ec4f6c4b9aabc633ae7f3f4ea94223d52c",
            "7f500afbe2b2b4e4e57f22511e3a42b3596b85cad7fe1eca700cdae1905d3555",
            100000000000000000,
            21004,
            b"\0",
        ),
        (
            3,
            3,
            42,
            "1a4fc1ec5f98bf874d5336aaf1fa9069ce68dc36c3f77e93465c9ac2c8b4b741",
            "13007c9b1df6a0d2f2ffa9d0ebcdec189122a5e781eb64967eb0d6a6def95b7a",
            100000000000000000,
            299732,
            b"ABCDEFGHIJKLMNOP" * 256 + b"!!!",
        ),
        (
            3,
            4,
            42,
            "8da0358d780df542f767d977f99ad034b6d9fa808fe50997141be2a1b93542c0",
            "2dafe1ead8aae1051e6662c5d553b34067bda9c8fa7314ae8693ec61ddfc96d4",
            0,
            21004,
            b"\0",
        ),
        (
            1,
            1,
            38,
            "b72707f0f5a38339c9dd0359720312c739a8ac6554659c7af48456f06ba33374",
            "75a431c046046942f9c1f3305cd08f34302164811c675ac0a0ac0b73cb30a90e",
            0,
            21004,
            b"\0",
        ),
        (
            255,
            1,
            545,
            "529172fb644a6d29b7218fb783f3d666021fc29cc4bf9bffbcfb3b84ab8d6181",
            "30980c6102a12872ef9cd888f2bf90c81bbbdc8878ff7d1d1382f8983b0d0c49",
            0,
            21004,
            b"\0",
        ),
        (
            256,
            1,
            548,
            "db53c05c679bdfdf3ded787ce9607d3f109ae46c87b1dcc9ab34053e5ed0eace",
            "39645dd48118d369b588dbf279f1a8c01051fabf65bf8eaa633c6433ff120cce",
            0,
            21004,
            b"\0",
        ),
        (
            65535,
            1,
            131105,
            "b520fa77767cdf07b6014d4a8fb35eebe5ed7c0edab97132b0dc74e3e1f13ed9",
            "78735b2db4cf95fb651c5c1f5529e60542019e456c6cb7a9f4bd9bbb83418d99",
            0,
            21004,
            b"\0",
        ),
        (
            65536,
            1,
            131107,
            "4b6122ba875b57ce084bd5f08e9ae1944e998726a4056c9b7746432d8f46ba99",
            "6812c2668ac9c9927b69ef7cf9baec54436f7319ccc14f0f664e1e94e6109e06",
            0,
            21004,
            b"\0",
        ),
        (
            16777215,
            1,
            33554465,
            "68a8c6f2336a8e3296f17a307d84a1e6d3ab1383fdcc62611c2e8426f2e2777e",
            "2d4ce900077ab40aac26064945998dbac5a014baadae2d3cb629cdeb9452db61",
            0,
            21004,
            b"\0",
        ),
        (
            16777216,
            1,
            33554468,
            "b6c42c584ef69621a2e5f3e1ab9dad890dbff3c92a599230dd0e394cd29d1c68",
            "497eec05ea52773d0f05e7fdf4f7993b3a06ef958804b39af699ef09ed0f5d7e",
            0,
            21004,
            b"\0",
        ),
        (
            2147483629,
            1,
            4294967294,
            "1a31f886c0bba527e622a731270dc29e62a607ff63558fca38745e5b9a672686",
            "0f3fce8a70598bbb54387cde7e8f957a27e4a816cbc9408717b27d8666222bd9",
            0,
            21004,
            b"\0",
        ),
        (
            2147483630,
            1,
            4294967296,
            "ba6cb6e2ebbac3726db9a3e4a939454009108f6515330e567aeada14ecebe074",
            "2bbfba1154cae32e3e6c6bbf3ce41cba6cc8c6b764245ba6026605506838e690",
            0,
            21004,
            None,
        ),
        (
            2147483631,
            1,
            4294967298,
            "3c743528e9ce315db02e487de93f2b2cfc93421e43f1d519f77a2f05bd2ce190",
            "16c1fec1495fe5da89d1a026f1a575ff354e18ff0fb9d04a6cfb0413267ab2bc",
            100000000000000000,
            21000,
            None,
        ),
        (
            3125659152,
            1,
            6251318340,
            "82cde0c9e1a94c1305791b09e1bcd021a49b036a16d9733acbc1a08bb30f3410",
            "472c8897519ba410b86f80993236d992e18e94d1f59c3d8760d2d7c90914dfc6",
            1,
            21005,
            None,
        ),
        (
            4294967295,
            1,
            8589934625,
            "67788e892fb372bba16823e16d3186f67494d7b1128555248f3661ad87e9d7ef",
            "2faf9f06dfdf23ceca2796cf0d55c88187f199e98a94dfb15722824b244d81a1",
            100000000000000000,
            21000,
            None,
        ),
    ]
]


@pytest.mark.altcoin
@pytest.mark.ethereum
@pytest.mark.parametrize("vector", VECTORS)
def test_ethereum_signtx_eip155(client, vector):
    sig_v, sig_r, sig_s = ethereum.sign_tx(
        client,
        n=[H_(44), H_(60), H_(0), 0, 0],
        nonce=vector.nonce,
        gas_price=20000000000,
        gas_limit=vector.gas_limit,
        to="0x8eA7a3fccC211ED48b763b4164884DDbcF3b0A98",
        value=vector.value,
        chain_id=vector.chain_id,
        data=vector.data,
    )
    assert sig_v == vector.sig_v
    assert sig_r.hex() == vector.sig_r
    assert sig_s.hex() == vector.sig_s