arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-26 22:32:33 +00:00
Code Issues Releases Wiki Activity

update docu strings in aes

Browse Source
This commit is contained in:
Pavol Rusnak 2016-09-23 18:33:10 +02:00
parent 8f80749870
commit dd78f83576
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
2 changed files with 40 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
extmod/modtrezorcrypto/modtrezorcrypto-aes.h
View File

@ -9,9 +9,6 @@
#include "trezor-crypto/aes.h" #include "trezor-crypto/aes.h"
/*
*/
typedef struct _mp_obj_AES_t { typedef struct _mp_obj_AES_t {
mp_obj_base_t base; mp_obj_base_t base;
union { union {
@ -23,10 +20,6 @@ typedef struct _mp_obj_AES_t {
uint8_t ctr[AES_BLOCK_SIZE]; uint8_t ctr[AES_BLOCK_SIZE];
} mp_obj_AES_t; } mp_obj_AES_t;
/// def trezor.crypto.aes.AES(mode:int, key: bytes, iv: bytes=None) -> AES:
/// '''
/// Create AES context
/// '''
STATIC mp_obj_t mod_TrezorCrypto_AES_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) { STATIC mp_obj_t mod_TrezorCrypto_AES_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
mp_arg_check_num(n_args, n_kw, 2, 3, false); mp_arg_check_num(n_args, n_kw, 2, 3, false);
mp_obj_AES_t *o = m_new_obj(mp_obj_AES_t); mp_obj_AES_t *o = m_new_obj(mp_obj_AES_t);

50
src/trezor/crypto/aes.py
View File

@ -1,31 +1,61 @@
from TrezorCrypto import AES as _AES from TrezorCrypto import AES as _AES
def AES_ECB_Encrypt(key): def AES_ECB_Encrypt(key: bytes):
'''
Create AES encryption context in ECB mode
'''
return _AES(0x00, key) return _AES(0x00, key)
def AES_ECB_Decrypt(key): def AES_ECB_Decrypt(key: bytes):
'''
Create AES decryption context in ECB mode
'''
return _AES(0x80, key) return _AES(0x80, key)
def AES_CBC_Encrypt(key, iv): def AES_CBC_Encrypt(key: bytes, iv: bytes):
'''
Create AES encryption context in CBC mode
'''
return _AES(0x01, key, iv) return _AES(0x01, key, iv)
def AES_CBC_Decrypt(key, iv): def AES_CBC_Decrypt(key: bytes, iv: bytes):
'''
Create AES decryption context in CBC mode
'''
return _AES(0x81, key, iv) return _AES(0x81, key, iv)
def AES_CFB_Encrypt(key, iv): def AES_CFB_Encrypt(key: bytes, iv: bytes):
'''
Create AES encryption context in CFB mode
'''
return _AES(0x02, key, iv) return _AES(0x02, key, iv)
def AES_CFB_Decrypt(key, iv): def AES_CFB_Decrypt(key: bytes, iv: bytes):
'''
Create AES decryption context in CFB mode
'''
return _AES(0x82, key, iv) return _AES(0x82, key, iv)
def AES_OFB_Encrypt(key, iv): def AES_OFB_Encrypt(key: bytes, iv: bytes):
'''
Create AES encryption context in OFB mode
'''
return _AES(0x03, key, iv) return _AES(0x03, key, iv)
def AES_OFB_Decrypt(key, iv): def AES_OFB_Decrypt(key: bytes, iv: bytes):
'''
Create AES decryption context in OFB mode
'''
return _AES(0x83, key, iv) return _AES(0x83, key, iv)
def AES_CTR_Encrypt(key): def AES_CTR_Encrypt(key: bytes):
'''
Create AES encryption context in CTR mode
'''
return _AES(0x04, key) return _AES(0x04, key)
def AES_CTR_Decrypt(key): def AES_CTR_Decrypt(key: bytes):
'''
Create AES decryption context in CTR mode
'''
return _AES(0x84, key) return _AES(0x84, key)