From 8297b81cd58a734132c84f4c1c0933e286f74615 Mon Sep 17 00:00:00 2001
From: Andrew Kozlik <andrew.kozlik@gmail.com>
Date: Thu, 8 Aug 2019 15:14:53 +0200
Subject: [PATCH] core/crypto: Add AES mode constants.

---
 core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h | 5 +++++
 core/mocks/generated/trezorcrypto/__init__.pyi          | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h b/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h
index 34c84cfe9..ed1f009d8 100644
--- a/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h
+++ b/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h
@@ -36,6 +36,11 @@ enum AESMode {
 ///     """
 ///     AES context.
 ///     """
+///     ECB: int
+///     CBC: int
+///     CFB: int
+///     OFB: int
+///     CTR: int
 typedef struct _mp_obj_AES_t {
   mp_obj_base_t base;
   aes_encrypt_ctx encrypt_ctx;
diff --git a/core/mocks/generated/trezorcrypto/__init__.pyi b/core/mocks/generated/trezorcrypto/__init__.pyi
index 446b897ab..4f29d66e0 100644
--- a/core/mocks/generated/trezorcrypto/__init__.pyi
+++ b/core/mocks/generated/trezorcrypto/__init__.pyi
@@ -6,6 +6,11 @@ class aes:
     """
     AES context.
     """
+    ECB: int
+    CBC: int
+    CFB: int
+    OFB: int
+    CTR: int
 
     def __init__(self, mode: int, key: bytes, iv: bytes = None) -> None:
         """