/*****************************************************************************
*  Copyright 2005 Alt-N Technologies, Ltd.
*
*  Licensed under the Apache License, Version 2.0 (the "License");
*  you may not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
*      http://www.apache.org/licenses/LICENSE-2.0
*
*  This code incorporates intellectual property owned by Yahoo! and licensed
*  pursuant to the Yahoo! DomainKeys Patent License Agreement.
*
*  Unless required by applicable law or agreed to in writing, software
*  distributed under the License is distributed on an "AS IS" BASIS,
*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*  See the License for the specific language governing permissions and
*  limitations under the License.
*
*  Changes done by ¢feh@fehcom.de obeying the above license
*
*****************************************************************************/
#include "dkim.h"

#include <string.h>

#include "dkimsign.h"
#include "dkimverify.h"


#define DKIMID ('D' | 'K' << 8 | 'I' << 16 | 'M' << 24)

/* taken from removed file "ressource.h" */
#ifdef VERSION
  #define VERSION_STRING VERSION
#else
  #define VERSION_STRING "1.4.0"
#endif


static void InitContext(DKIMContext *pContext, bool bSign, void *pObject)
{
  pContext->reserved1 = DKIMID;
  pContext->reserved2 = bSign ? 1 : 0;
  pContext->reserved3 = pObject;
}

static void *ValidateContext(DKIMContext *pContext, bool bSign)
{
  if (pContext->reserved1 != DKIMID) return NULL;

  if (pContext->reserved2 != (unsigned int)(bSign ? 1 : 0)) return NULL;

  return pContext->reserved3;
}

int DKIM_CALL DKIMSignInit(DKIMContext *pSignContext, DKIMSignOptions *pOptions)
{
  int nRet = DKIM_OUT_OF_MEMORY;

  CDKIMSign *pSign = new CDKIMSign;

  if (pSign) {
    nRet = pSign->Init(pOptions);
    if (nRet != DKIM_SUCCESS) delete pSign;
  }

  if (nRet == DKIM_SUCCESS) InitContext(pSignContext, true, pSign);
  return nRet;
}

int DKIM_CALL DKIMSignProcess(DKIMContext *pSignContext, char *szBuffer, int nBufLength)
{
  CDKIMSign *pSign = (CDKIMSign *)ValidateContext(pSignContext, true);

  if (pSign) return pSign->Process(szBuffer, nBufLength, false);
  return DKIM_INVALID_CONTEXT;
}

int DKIM_CALL DKIMSignGetSig2(
    DKIMContext *pSignContext, char *szRSAPrivKey, char *szECCPrivKey, char **pszSignature)
{
  CDKIMSign *pSign = (CDKIMSign *)ValidateContext(pSignContext, true);

  if (pSign) return pSign->GetSig2(szRSAPrivKey, szECCPrivKey, pszSignature);
  return DKIM_INVALID_CONTEXT;
}

void DKIM_CALL DKIMSignFree(DKIMContext *pSignContext)
{
  CDKIMSign *pSign = (CDKIMSign *)ValidateContext(pSignContext, true);

  if (pSign) {
    delete pSign;
    pSignContext->reserved3 = NULL;
  }
}

int DKIM_CALL DKIMVerifyInit(DKIMContext *pVerifyContext, DKIMVerifyOptions *pOptions)
{
  int nRet = DKIM_OUT_OF_MEMORY;

  CDKIMVerify *pVerify = new CDKIMVerify;

  if (pVerify) {
    nRet = pVerify->Init(pOptions);
    if (nRet != DKIM_SUCCESS) delete pVerify;
  }

  if (nRet == DKIM_SUCCESS) InitContext(pVerifyContext, false, pVerify);

  return nRet;
}


int DKIM_CALL DKIMVerifyProcess(DKIMContext *pVerifyContext, const char *const szBuffer, int nBufLength)
{
  CDKIMVerify *pVerify = (CDKIMVerify *)ValidateContext(pVerifyContext, false);

  if (pVerify) return pVerify->Process(szBuffer, nBufLength, false);

  return DKIM_INVALID_CONTEXT;
}

int DKIM_CALL DKIMVerifyResults(DKIMContext *pVerifyContext)
{
  CDKIMVerify *pVerify = (CDKIMVerify *)ValidateContext(pVerifyContext, false);

  if (pVerify) return pVerify->GetResults();
  return DKIM_INVALID_CONTEXT;
}

int DKIM_CALL DKIMVerifyGetDetails(
    DKIMContext *pVerifyContext, int *nSigCount, DKIMVerifyDetails **pDetails, char *szPractices)
{
  szPractices[0] = '\0';

  CDKIMVerify *pVerify = (CDKIMVerify *)ValidateContext(pVerifyContext, false);

  if (pVerify) {
    strcpy(szPractices, pVerify->GetPractices());
    return pVerify->GetDetails(nSigCount, pDetails);
  }

  return DKIM_INVALID_CONTEXT;
}


void DKIM_CALL DKIMVerifyFree(DKIMContext *pVerifyContext)
{
  CDKIMVerify *pVerify = (CDKIMVerify *)ValidateContext(pVerifyContext, false);

  if (pVerify) {
    delete pVerify;
    pVerifyContext->reserved3 = NULL;
  }
}

const char *DKIM_CALL DKIMVersion()
{
  return VERSION_STRING;
}

static const char *DKIMErrorStrings[-1 - DKIM_MAX_ERROR] = {
    "DKIM_FAIL",
    "DKIM_BAD_SYNTAX",
    "DKIM_SIGNATURE_BAD",
    "DKIM_SIGNATURE_BAD_BUT_TESTING",
    "DKIM_SIGNATURE_EXPIRED",
    "DKIM_SELECTOR_INVALID",
    "DKIM_SELECTOR_GRANULARITY_MISMATCH",
    "DKIM_SELECTOR_KEY_REVOKED",
    "DKIM_SELECTOR_DOMAIN_NAME_TOO_LONG",
    "DKIM_SELECTOR_DNS_TEMP_FAILURE",
    "DKIM_SELECTOR_DNS_PERM_FAILURE",
    "DKIM_SELECTOR_PUBLIC_KEY_INVALID",
    "DKIM_NO_SIGNATURES",
    "DKIM_NO_VALID_SIGNATURES",
    "DKIM_BODY_HASH_MISMATCH",
    "DKIM_SELECTOR_ALGORITHM_MISMATCH",
    "DKIM_STAT_INCOMPAT",
    "DKIM_UNSIGNED_FROM",
    "DKIM_OUT_OF_MEMORY",
    "DKIM_INVALID_CONTEXT",
    "DKIM_NO_SENDER",
    "DKIM_BAD_PRIVATE_KEY",
    "DKIM_BUFFER_TOO_SMALL",
};

const char *DKIM_CALL DKIMGetErrorString(int ErrorCode)
{
  if (ErrorCode >= 0 || ErrorCode <= DKIM_MAX_ERROR)
    return "Unknown";
  else
    return DKIMErrorStrings[-1 - ErrorCode];
}