AppleCSP/AES/gladmanContext.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 // gladmanContext.h - Gladman AES context class
  21 //
  22 #ifndef _H_GLADMAN_CONTEXT
  23 #define _H_GLADMAN_CONTEXT
  24
  25 #include <Security/CSPsession.h>
  26 #include "AppleCSP.h"
  27 #include "AppleCSPContext.h"
  28 #include "AppleCSPSession.h"
  29 #include "BlockCryptor.h"
  30 #include "rijndaelGladman.h"
  31 #include "aesCommon.h"
  32
  33 #define GLADMAN_BLOCK_SIZE_BYTES        DEFAULT_AES_BLOCK_BYTES
  34
  35 /* Symmetric encryption context */
  36 class GAESContext : public BlockCryptor {
  37 public:
  38         GAESContext(AppleCSPSession &session);
  39         ~GAESContext();
  40
  41         // called by CSPFullPluginSession
  42         void init(const Context &context, bool encoding = true);
  43
  44         // As an optimization, we allow reuse of a modified context. The main thing
  45         // we avoid is a redundant key scheduling. We save the current raw keys bits
  46         // in mRawKey and compare on re-init.
  47         bool changed(const Context &context)     { return true; }
  48
  49         // called by BlockCryptor
  50         void encryptBlock(
  51                 const void              *plainText,                     // length implied (one block)
  52                 size_t                  plainTextLen,
  53                 void                    *cipherText,
  54                 size_t                  &cipherTextLen,         // in/out, throws on overflow
  55                 bool                    final);
  56         void decryptBlock(
  57                 const void              *cipherText,            // length implied (one cipher block)
  58                 void                    *plainText,
  59                 size_t                  &plainTextLen,          // in/out, throws on overflow
  60                 bool                    final);
  61
  62 private:
  63         void deleteKey();
  64
  65         /* scheduled key */
  66         GAesKey                         mAesKey;
  67         bool                            mKeyValid;
  68         bool                            mInitFlag;                      // for easy reuse
  69
  70         /* raw key bits saved here and checked on re-init to avoid extra key schedule */
  71         uint8                           mRawKey[MAX_AES_KEY_BITS / 8];
  72         uint32                          mRawKeySize;
  73 };      /* AESContext */
  74
  75 #endif //_H_GLADMAN_CONTEXT