mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-22 15:38:11 +00:00
83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
# This file is part of the Trezor project.
|
|
#
|
|
# Copyright (C) 2012-2018 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 . import messages as proto
|
|
from .tools import expect
|
|
|
|
|
|
@expect(proto.Entropy, field="entropy")
|
|
def get_entropy(client, size):
|
|
return client.call(proto.GetEntropy(size=size))
|
|
|
|
|
|
@expect(proto.SignedIdentity)
|
|
def sign_identity(
|
|
client, identity, challenge_hidden, challenge_visual, ecdsa_curve_name=None
|
|
):
|
|
return client.call(
|
|
proto.SignIdentity(
|
|
identity=identity,
|
|
challenge_hidden=challenge_hidden,
|
|
challenge_visual=challenge_visual,
|
|
ecdsa_curve_name=ecdsa_curve_name,
|
|
)
|
|
)
|
|
|
|
|
|
@expect(proto.ECDHSessionKey)
|
|
def get_ecdh_session_key(client, identity, peer_public_key, ecdsa_curve_name=None):
|
|
return client.call(
|
|
proto.GetECDHSessionKey(
|
|
identity=identity,
|
|
peer_public_key=peer_public_key,
|
|
ecdsa_curve_name=ecdsa_curve_name,
|
|
)
|
|
)
|
|
|
|
|
|
@expect(proto.CipheredKeyValue, field="value")
|
|
def encrypt_keyvalue(
|
|
client, n, key, value, ask_on_encrypt=True, ask_on_decrypt=True, iv=b""
|
|
):
|
|
return client.call(
|
|
proto.CipherKeyValue(
|
|
address_n=n,
|
|
key=key,
|
|
value=value,
|
|
encrypt=True,
|
|
ask_on_encrypt=ask_on_encrypt,
|
|
ask_on_decrypt=ask_on_decrypt,
|
|
iv=iv,
|
|
)
|
|
)
|
|
|
|
|
|
@expect(proto.CipheredKeyValue, field="value")
|
|
def decrypt_keyvalue(
|
|
client, n, key, value, ask_on_encrypt=True, ask_on_decrypt=True, iv=b""
|
|
):
|
|
return client.call(
|
|
proto.CipherKeyValue(
|
|
address_n=n,
|
|
key=key,
|
|
value=value,
|
|
encrypt=False,
|
|
ask_on_encrypt=ask_on_encrypt,
|
|
ask_on_decrypt=ask_on_decrypt,
|
|
iv=iv,
|
|
)
|
|
)
|