/* Copyright 2003-2006, Voltage Security, all rights reserved.
 */
#include "vibe.h"
#include "environment.h"
#include "base.h"
#include "libctx.h"
#include "algobj.h"
#include "oidlist.h"
#include "algid.h"
#include "errorctx.h"

int VtDerCoderBFType1IBE (
   VtDerCoderInfo *coderInfo,
   Pointer object,
   unsigned int flag
   )
{
  int status;
  unsigned int bufferSize;
  VoltLibCtx *libCtx;
  VoltAlgorithmObject **obj = (VoltAlgorithmObject **)0;
  VoltDerCoderEncodeData *encodeData = &(coderInfo->info.encodeData);
  VoltDerCoderGetAlgData *getAlgData = &(coderInfo->info.getAlgData);
  VoltDerCoderDecodeData *decodeData = &(coderInfo->info.decodeData);
  Asn1AlgorithmId *algId;
  unsigned char ibeEncAlgID[VoltIBET1EncAlgIdBytesLen] =
    { VoltIBET1EncAlgIdBytes };
  unsigned char ibeClientEncOID[VoltIBET1ClientEncOidBytesLen] =
    { VoltIBET1ClientEncOidBytes };
  VOLT_DECLARE_ERROR_TYPE (errorType)
  VOLT_DECLARE_FNCT_LINE (fnctLine)
  
  switch (flag)
  {
    default:
      VOLT_SET_ERROR_TYPE (errorType, VT_ERROR_TYPE_PRIMARY)
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_INVALID_TYPE;
      break;

    case VOLT_DER_TYPE_ENCODE_FLAG:
      /* If the flag is ENCODE, return the IBE Enc algID.
       */

      /* Check the args.
       */
      VOLT_SET_ERROR_TYPE (errorType, VT_ERROR_TYPE_PRIMARY)
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_NON_NULL_ARG;
      if (object != (Pointer)0)
        break;

      /* We need a place to drop the length.
       */
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_NULL_ARG;
      if (encodeData->encodingLen == (unsigned int *)0)
        break;

      /* The IBE info should be NULL.
       */
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_NON_NULL_ARG;
      if (encodeData->info != (Pointer)0)
        break;

      bufferSize = encodeData->bufferSize;
      if (encodeData->encoding == (unsigned char *)0)
        bufferSize = 0;

      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_BUFFER_TOO_SMALL;
      *(encodeData->encodingLen) = VoltIBET1EncAlgIdBytesLen;
      if (bufferSize < VoltIBET1EncAlgIdBytesLen)
        break;

      /* There's only one possible IBE alg ID.
       */
      libCtx = encodeData->libCtx;
      Z2Memcpy (encodeData->encoding, ibeEncAlgID, VoltIBET1EncAlgIdBytesLen);
      status = 0;
      break;

    case VOLT_DER_TYPE_DECODE_FLAG:
      /* If the flag is decode, set the object to perform IBE encryption
       */
      VOLT_SET_ERROR_TYPE (errorType, VT_ERROR_TYPE_PRIMARY)
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_NULL_ARG;
      if (object == (Pointer)0)
        break;
      obj = (VoltAlgorithmObject **)object;

      /* Check the args, the type should be VOLT_DER_TYPE_ALG_ID_FLAG.
       */
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_UNKNOWN_BER;
      if (decodeData->type != VOLT_DER_TYPE_ALG_ID_FLAG)
        break;

      /* Make sure this is the algId for IBE encryption
       */
      algId = (Asn1AlgorithmId *)(decodeData->asn1Object);
      libCtx = (VoltLibCtx *)((*obj)->voltObject.libraryCtx);

      if (algId->oid->base.length == VoltIBET1EncOidBytesLen)
      {
        VOLT_SET_FNCT_LINE (fnctLine)
        if (Z2Memcmp (algId->oid->base.data, 
          ibeEncAlgID + VoltIBET1EncAlgIdOidOffset,
          VoltIBET1EncOidBytesLen) != 0)
          break;
      }
      else 
      {
        if (algId->oid->base.length == VoltIBET1ClientEncOidBytesLen)
        {
          VOLT_SET_FNCT_LINE (fnctLine)
          if (Z2Memcmp (algId->oid->base.data, 
            ibeClientEncOID ,
            VoltIBET1ClientEncOidBytesLen) != 0)
            break;
        }
        else
        {
          VOLT_SET_FNCT_LINE (fnctLine)
          break;
        }
      }

      VtDestroyAlgorithmObject ((VtAlgorithmObject *)object);

      VOLT_SET_ERROR_TYPE (errorType, 0)
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VtCreateAlgorithmObject (
        libCtx, VtAlgorithmImplBFType1IBE, (Pointer)0,
        (VtAlgorithmObject *)object);
      if (status != 0)
        break;
      
      break;

    case VOLT_DER_TYPE_GET_ALG_FLAG:
      /* If the flag is get alg, check the input to see if it's the
       * IBE encryption OID.
       */

      /* Check the args.
       */
      VOLT_SET_ERROR_TYPE (errorType, VT_ERROR_TYPE_PRIMARY)
      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_NON_NULL_ARG;
      if (object != (Pointer)0)
        break;

      VOLT_SET_FNCT_LINE (fnctLine)
      status = VT_ERROR_NULL_ARG;
      if ( (getAlgData->algorithm == (unsigned int *)0) ||
           (getAlgData->oid == (unsigned char *)0) )
        break;

      /* Is this the IBE encryption OID (either true or client)?
       */
      status = VT_ERROR_UNKNOWN_BER;
      libCtx = getAlgData->libCtx;
      if (getAlgData->oidLen == VoltIBET1EncOidBytesLen)
      {
        VOLT_SET_FNCT_LINE (fnctLine)
        if (Z2Memcmp (
          getAlgData->oid, ibeEncAlgID + VoltIBET1EncAlgIdOidOffset,
          VoltIBET1EncOidBytesLen) != 0)
          break;
      }
      else if (getAlgData->oidLen == VoltIBET1ClientEncOidBytesLen)
      {
        VOLT_SET_FNCT_LINE (fnctLine)
        if (Z2Memcmp (
          getAlgData->oid, ibeClientEncOID,
          VoltIBET1ClientEncOidBytesLen) != 0)
          break;
      }
      else
      {
        /* If we reach this point, the length was not recognized.
         */
        VOLT_SET_FNCT_LINE (fnctLine)
        break;
      }

      /* The OID matches, the algorithm is IBE encryption.
       */
      *(getAlgData->algorithm) = VT_ALG_ID_BF_TYPE1_IBE_ENCRYPT;
      getAlgData->SymKeyParam = (VtKeyParam *)0;
      getAlgData->DigestImpl = (VtAlgorithmImpl *)0;

      status = 0;
  }

  VOLT_LOG_ERROR_INFO_COMPARE (
    status, 0, object, status, 0, errorType,
    (char *)0, "VtDerCoderBFType1IBE", fnctLine, (char *)0)

  return (status);
}