mirror of
https://github.com/yuzu-emu/mbedtls
synced 2024-11-24 19:38:21 +00:00
More robust selection of ctx_enc size
This commit is contained in:
parent
cffe4a65bd
commit
c852a68b96
2 changed files with 59 additions and 11 deletions
|
@ -52,6 +52,18 @@
|
|||
#if defined(POLARSSL_AES_C)
|
||||
#include "aes.h"
|
||||
#endif
|
||||
#if defined(POLARSSL_ARC4_C)
|
||||
#include "arc4.h"
|
||||
#endif
|
||||
#if defined(POLARSSL_DES_C)
|
||||
#include "des.h"
|
||||
#endif
|
||||
#if defined(POLARSSL_CAMELLIA_C)
|
||||
#include "camellia.h"
|
||||
#endif
|
||||
#if defined(POLARSSL_GCM_C)
|
||||
#include "gcm.h"
|
||||
#endif
|
||||
|
||||
#if defined(POLARSSL_X509_PARSE_C)
|
||||
#include "x509.h"
|
||||
|
@ -429,6 +441,40 @@ struct _ssl_session
|
|||
#endif /* POLARSSL_SSL_TRUNCATED_HMAC */
|
||||
};
|
||||
|
||||
/*
|
||||
* Helpers to find the correct size of the context in _ssl_transform
|
||||
* (in the long run, we'll use the cipher layer, but for now...)
|
||||
*/
|
||||
#define SSL_MAX(a, b) ( a > b ? a : b )
|
||||
#define SSL_CTX_MAX_0 0
|
||||
#if defined(POLARSSL_AES_C)
|
||||
#define SSL_CTX_MAX_1 SSL_MAX( SSL_CTX_MAX_0, sizeof( aes_context ) )
|
||||
#else
|
||||
#define SSL_CTX_MAX_1 SSL_CTX_MAX_0
|
||||
#endif
|
||||
#if defined(POLARSSL_ARC4_C)
|
||||
#define SSL_CTX_MAX_2 SSL_MAX( SSL_CTX_MAX_1, sizeof( arc4_context ) )
|
||||
#else
|
||||
#define SSL_CTX_MAX_2 SSL_CTX_MAX_1
|
||||
#endif
|
||||
#if defined(POLARSSL_DES_C)
|
||||
#define SSL_CTX_MAX_3 SSL_MAX( SSL_CTX_MAX_2, sizeof( des_context ) )
|
||||
#define SSL_CTX_MAX_4 SSL_MAX( SSL_CTX_MAX_3, sizeof( des3_context ) )
|
||||
#else
|
||||
#define SSL_CTX_MAX_4 SSL_CTX_MAX_2
|
||||
#endif
|
||||
#if defined(POLARSSL_CAMELLIA_C)
|
||||
#define SSL_CTX_MAX_5 SSL_MAX( SSL_CTX_MAX_4, sizeof( camellia_context ) )
|
||||
#else
|
||||
#define SSL_CTX_MAX_5 SSL_CTX_MAX_4
|
||||
#endif
|
||||
#if defined(POLARSSL_GCM_C)
|
||||
#define SSL_CTX_MAX_6 SSL_MAX( SSL_CTX_MAX_5, sizeof( gcm_context ) )
|
||||
#else
|
||||
#define SSL_CTX_MAX_6 SSL_CTX_MAX_5
|
||||
#endif
|
||||
#define SSL_CTX_MAX SSL_CTX_MAX_6
|
||||
|
||||
/*
|
||||
* This structure contains a full set of runtime transform parameters
|
||||
* either in negotiation or active.
|
||||
|
@ -458,9 +504,8 @@ struct _ssl_transform
|
|||
md_context_t md_ctx_enc; /*!< MAC (encryption) */
|
||||
md_context_t md_ctx_dec; /*!< MAC (decryption) */
|
||||
|
||||
/* 154 == 616 bytes is size of gcm_context (largest context in PolarSSL) */
|
||||
uint32_t ctx_enc[154]; /*!< encryption context */
|
||||
uint32_t ctx_dec[154]; /*!< decryption context */
|
||||
uint32_t ctx_enc[SSL_CTX_MAX / 4]; /*!< encryption context */
|
||||
uint32_t ctx_dec[SSL_CTX_MAX / 4]; /*!< decryption context */
|
||||
|
||||
/*
|
||||
* Session specific compression layer
|
||||
|
@ -471,6 +516,17 @@ struct _ssl_transform
|
|||
#endif
|
||||
};
|
||||
|
||||
/* Not needed any more */
|
||||
#undef SSL_MAX
|
||||
#undef SSL_CTX_MAX_0
|
||||
#undef SSL_CTX_MAX_1
|
||||
#undef SSL_CTX_MAX_2
|
||||
#undef SSL_CTX_MAX_3
|
||||
#undef SSL_CTX_MAX_4
|
||||
#undef SSL_CTX_MAX_5
|
||||
#undef SSL_CTX_MAX_6
|
||||
#undef SSL_CTX_MAX
|
||||
|
||||
/*
|
||||
* This structure contains the parameters only needed during handshake.
|
||||
*/
|
||||
|
|
|
@ -38,14 +38,6 @@
|
|||
#include "polarssl/debug.h"
|
||||
#include "polarssl/ssl.h"
|
||||
|
||||
#include "polarssl/arc4.h"
|
||||
#include "polarssl/camellia.h"
|
||||
#include "polarssl/des.h"
|
||||
|
||||
#if defined(POLARSSL_GCM_C)
|
||||
#include "polarssl/gcm.h"
|
||||
#endif
|
||||
|
||||
#if defined(POLARSSL_MEMORY_C)
|
||||
#include "polarssl/memory.h"
|
||||
#else
|
||||
|
|
Loading…
Reference in a new issue