michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: /* Prototypes of the functions defined in the assembler file.  */
michael@0: void intel_aes_encrypt_init_128(const unsigned char *key, PRUint32 *expanded);
michael@0: void intel_aes_encrypt_init_192(const unsigned char *key, PRUint32 *expanded);
michael@0: void intel_aes_encrypt_init_256(const unsigned char *key, PRUint32 *expanded);
michael@0: void intel_aes_decrypt_init_128(const unsigned char *key, PRUint32 *expanded);
michael@0: void intel_aes_decrypt_init_192(const unsigned char *key, PRUint32 *expanded);
michael@0: void intel_aes_decrypt_init_256(const unsigned char *key, PRUint32 *expanded);
michael@0: SECStatus intel_aes_encrypt_ecb_128(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_decrypt_ecb_128(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_cbc_128(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_decrypt_cbc_128(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_ctr_128(CTRContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_ecb_192(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_decrypt_ecb_192(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_cbc_192(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_decrypt_cbc_192(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_ctr_192(CTRContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_ecb_256(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_decrypt_ecb_256(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_cbc_256(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_decrypt_cbc_256(AESContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: SECStatus intel_aes_encrypt_ctr_256(CTRContext *cx, unsigned char *output,
michael@0: 				    unsigned int *outputLen,
michael@0: 				    unsigned int maxOutputLen,
michael@0: 				    const unsigned char *input,
michael@0: 				    unsigned int inputLen,
michael@0: 				    unsigned int blocksize);
michael@0: 
michael@0: 
michael@0: #define intel_aes_ecb_worker(encrypt, keysize) \
michael@0:   ((encrypt)						\
michael@0:    ? ((keysize) == 16 ? intel_aes_encrypt_ecb_128 :	\
michael@0:       (keysize) == 24 ? intel_aes_encrypt_ecb_192 :	\
michael@0:       intel_aes_encrypt_ecb_256)			\
michael@0:    : ((keysize) == 16 ? intel_aes_decrypt_ecb_128 :	\
michael@0:       (keysize) == 24 ? intel_aes_decrypt_ecb_192 :	\
michael@0:       intel_aes_decrypt_ecb_256))
michael@0: 
michael@0: 
michael@0: #define intel_aes_cbc_worker(encrypt, keysize) \
michael@0:   ((encrypt)						\
michael@0:    ? ((keysize) == 16 ? intel_aes_encrypt_cbc_128 :	\
michael@0:       (keysize) == 24 ? intel_aes_encrypt_cbc_192 :	\
michael@0:       intel_aes_encrypt_cbc_256)			\
michael@0:    : ((keysize) == 16 ? intel_aes_decrypt_cbc_128 :	\
michael@0:       (keysize) == 24 ? intel_aes_decrypt_cbc_192 :	\
michael@0:       intel_aes_decrypt_cbc_256))
michael@0: 
michael@0: #define intel_aes_ctr_worker(nr) \
michael@0:    ((nr) == 10 ? intel_aes_encrypt_ctr_128 :	\
michael@0:     (nr) == 12 ? intel_aes_encrypt_ctr_192 :	\
michael@0:     intel_aes_encrypt_ctr_256)
michael@0: 
michael@0: 
michael@0: #define intel_aes_init(encrypt, keysize) \
michael@0:   do {					 			\
michael@0:       if (encrypt) {			 			\
michael@0: 	  if (keysize == 16)					\
michael@0: 	      intel_aes_encrypt_init_128(key, cx->expandedKey);	\
michael@0: 	  else if (keysize == 24)				\
michael@0: 	      intel_aes_encrypt_init_192(key, cx->expandedKey);	\
michael@0: 	  else							\
michael@0: 	      intel_aes_encrypt_init_256(key, cx->expandedKey);	\
michael@0:       } else {							\
michael@0: 	  if (keysize == 16)					\
michael@0: 	      intel_aes_decrypt_init_128(key, cx->expandedKey);	\
michael@0: 	  else if (keysize == 24)				\
michael@0: 	      intel_aes_decrypt_init_192(key, cx->expandedKey);	\
michael@0: 	  else							\
michael@0: 	      intel_aes_decrypt_init_256(key, cx->expandedKey);	\
michael@0:       }								\
michael@0:   } while (0)