From c0255959518e0bcb1dafda495ea46f8b1c067a81 Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Thu, 24 Dec 2015 17:31:09 +0100 Subject: [PATCH] add iv parameter to encrypt_keyvalue/decrypt_keyvalue --- trezorlib/client.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/trezorlib/client.py b/trezorlib/client.py index c97dcb9..173bb54 100644 --- a/trezorlib/client.py +++ b/trezorlib/client.py @@ -505,25 +505,27 @@ class ProtocolMixin(object): @field('value') @expect(proto.CipheredKeyValue) - def encrypt_keyvalue(self, n, key, value, ask_on_encrypt=True, ask_on_decrypt=True): + def encrypt_keyvalue(self, n, key, value, ask_on_encrypt=True, ask_on_decrypt=True, iv=None): n = self._convert_prime(n) return self.call(proto.CipherKeyValue(address_n=n, key=key, value=value, encrypt=True, ask_on_encrypt=ask_on_encrypt, - ask_on_decrypt=ask_on_decrypt)) + ask_on_decrypt=ask_on_decrypt, + iv=iv if iv is not None else '')) @field('value') @expect(proto.CipheredKeyValue) - def decrypt_keyvalue(self, n, key, value, ask_on_encrypt=True, ask_on_decrypt=True): + def decrypt_keyvalue(self, n, key, value, ask_on_encrypt=True, ask_on_decrypt=True, iv=None): n = self._convert_prime(n) return self.call(proto.CipherKeyValue(address_n=n, key=key, value=value, encrypt=False, ask_on_encrypt=ask_on_encrypt, - ask_on_decrypt=ask_on_decrypt)) + ask_on_decrypt=ask_on_decrypt, + iv=iv if iv is not None else '')) @field('tx_size') @expect(proto.TxSize)