AppleCSP/AES/aescspi.h

   1 /*
   2  * Copyright (c) 2000-2001 Apple Computer, Inc. All Rights Reserved.
   3  *
   4  * The contents of this file constitute Original Code as defined in and are
   5  * subject to the Apple Public Source License Version 1.2 (the 'License').
   6  * You may not use this file except in compliance with the License. Please obtain
   7  * a copy of the License at http://www.apple.com/publicsource and read it before
   8  * using this file.
   9  *
  10  * This Original Code and all software distributed under the License are
  11  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESS
  12  * OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, INCLUDING WITHOUT
  13  * LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
  14  * PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. Please see the License for the
  15  * specific language governing rights and limitations under the License.
  16  */
  17
  18
  19 //
  20 // aescspi.h - AES context class
  21 //
  22 #ifndef _H_AESCSPI
  23 #define _H_AESCSPI
  24
  25 #include <Security/CSPsession.h>
  26 #include "AppleCSP.h"
  27 #include "AppleCSPContext.h"
  28 #include "AppleCSPSession.h"
  29 #include "BlockCryptor.h"
  30 #include "rijndaelApi.h"
  31
  32 /* Symmetric encryption context */
  33 class AESContext : public BlockCryptor {
  34 public:
  35         AESContext(AppleCSPSession &session) :
  36                 BlockCryptor(session),
  37                 mAesKey(NULL),
  38                 mBlockSize(0),
  39                 mInitFlag(false),
  40                 mRawKeySize(0)  { }
  41         ~AESContext();
  42
  43         // called by CSPFullPluginSession
  44         void init(const Context &context, bool encoding = true);
  45
  46         // As an optimization, we allow reuse of a modified context. The main thing
  47         // we avoid is a redundant key scheduling. We save the current raw keys bits
  48         // in mRawKey and compare on re-init.
  49         bool changed(const Context &context)     { return true; }
  50
  51         // called by BlockCryptor
  52         void encryptBlock(
  53                 const void              *plainText,                     // length implied (one block)
  54                 size_t                  plainTextLen,
  55                 void                    *cipherText,
  56                 size_t                  &cipherTextLen,         // in/out, throws on overflow
  57                 bool                    final);
  58         void decryptBlock(
  59                 const void              *cipherText,            // length implied (one cipher block)
  60                 void                    *plainText,
  61                 size_t                  &plainTextLen,          // in/out, throws on overflow
  62                 bool                    final);
  63
  64 private:
  65         void aesError(
  66                 int artn,
  67                 const char *errStr);
  68         void deleteKey();
  69
  70         keyInstance                     *mAesKey;                       // keyInstance or keyInstance128
  71         unsigned                        mBlockSize;                     // in BYTES
  72         bool                            mInitFlag;                      // for easy reuse
  73         aesCryptFcn                     mEncryptFcn;
  74         aesCryptFcn                     mDecryptFcn;
  75
  76         /* raw key bits saved here and checked on re-init to avoid extra key schedule */
  77         uint8                           mRawKey[MAX_AES_KEY_BITS / 8];
  78         uint32                          mRawKeySize;
  79 };      /* AESContext */
  80
  81 // symmetric key generation context
  82 class AESKeyGenContext : public AppleCSPContext, private AppleSymmKeyGenContext {
  83 public:
  84         AESKeyGenContext(
  85                 AppleCSPSession &session) :
  86                         AppleCSPContext(session),
  87                         AppleSymmKeyGenContext(
  88                                 MIN_AES_KEY_BITS,
  89                                 MAX_AES_KEY_BITS,
  90                                 true) { }
  91
  92         void init(const Context &context, bool encoding = true) { }
  93
  94         void generate(
  95                 const Context   &context,
  96                 CssmKey                 &symKey,
  97                 CssmKey                 &dummyKey);
  98
  99 };
 100
 101 #endif //_H_AESCSPI