/*****************************************************************************
*  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
*
*****************************************************************************/

#define _strnicmp strncasecmp
#define _stricmp  strcasecmp
#define LOWORD(l) ((unsigned)(l) & 0xffff)
#define HIWORD(l) ((unsigned)(l) >> 16)

#include "dkimsign.h"

#include <string.h>

#include <map>

#include "dkim.h"

/*****************************************************************************
*
* Generating Ed25519 signed message:
*
* 1. RSA SHA1/SHA256 signatures are generated in streaming mode together with
*    their hashes. Two different 'contexts' (ctx) are used here:
*    m_Hdr_shaXctx      => Used for signing (EVP_Sign...) -- covering the header only
*    m_[B,E]dy_shaXctx  => Used for hashing (EVP_Digest..) -- covering the body only
*
* 2. Private keys
*    For hybrid signing we need two distinct keys:
*    - RSAKey
*    - ECCKey
*    These private keys needs to be passed concurrently to the signature functions.
*    Given those keys, the signature operation itself is executed in one step.
*
* 3. Public keys
*    The 'public keys' need to be deployed in the DNS:
*    - The RSA public key is DER-header enriched base64-encoded; thus is 9 byte larger
*      than the 'naked' public key, which size depends on the given parameters. 
*    - The Ed25519 public key is also base64-encoded with a constant length of 60 byte.
*
* 4. DKIM message preparation scheme 
*    According to RFC 6376 Sec. 3.7, we have a conducted hash for
*    - the previously available headers in the message; 
*      selected and given in order by h=...,
*    - any existing DKIM signature fields b=...,
*    - except for previous added 'X-Authentication ...' header fields,
*    - and all (new) synthezised DKIM header tokens; except of course for the 
*      signature itself - treated as 'null string': b="".
*    All this is subject of canonicalization (adding/removing CRLF, whitespaces ...).
+    As a result, the input for further calculations depends on this order given.
*
*    Results following the 'preparation scheme':
*    -  The message body hash is included in the DKIM header   => bh=[m_[B,E]dy_shaXctx].
*    -  The message signature (including the result of bh=...) =>  b=[m_Hdr_shaXctx]
*
*    We consider SHA256 as default hash function and SHA1 as exception (on demand).
*
* 5. Generating (ECC) signatures
*    According to RFC 8032 Sect 4., we have two possible Ed25519 signature schemes:
*
*    a) PureEd25519, as a one shot signature calculation swallowing the 
*       complete message and employing a shortened SHA-512 hash input.
*    b) HashEd25519 working again in 'streaming mode' and permitting a choice 
*        for the hash function - which is in RFC 8463 - defined to be SHA-256.
*
*    RFC 8463 in Sect 3 is a bit ambiguous about the signing function:
*    Ed25519-256 vs. PureEd25519. 
*    In fact (after consulting John Levine), it is PureEd25519.
*
*    In order to allow parallel RSA/Ed25519 processing, we need to generate:
*    m_Hdr_sha256ctx  => Used for RSA signatures
*    m_Bdy_sha256ctx  => The SHA256 hash of selected header parts and body (RSA)
*    m_Edy_sha256ctx  => The SHA256 hash of selected header parts and body (Ed25519)
*    m_Hdr_ed25519ctx => The signature of the messsage header using PureEd25519
*                        following the 'preparation' scheme
*
*    Now, two cryptographic informations are provided in the header:
*    bh=[m_Edy_sha256ctx]  => The SHA256 digest of the message (BodyHash),
*     b=[m_Hdr_ed25519ctx] => The PureED25519 signature.
*                             including the value of bh=... (EmailSignature)
*                             having a length of 512 bits => 64 bytes.
*
* 6. Hybrid signatures (RSA and Ed25519)
*    They involve 
*    m_Hdr_sha256ctx  => Used for RSA signatures
*    m_Hdr_ed25519ctx => PureED25519 signature
*    m_Bdy_sha256ctx  => SHA256 digest of the message (BodyHash) for RSA
*    m_Edy_sha256ctx  => SHA256 digest of the message (BodyHash) for Ed25519
*
*    The EVP_DigestFinal routine has to be replaced by EVP_DigestFinal_ex.
*    However; after the first call, its content seems to be garbeled.
*    A common MD for both RSA and Ed2551 seems to be infeasible.
*
* ------
*
* The particular function and variable names chosen here do not obviously match 
* what they are intended to do. However, in order to keep traceablility of the 
* changes, I left those untouched.
*    
*****************************************************************************/

CDKIMSign::CDKIMSign()
{
  m_EmptyLineCount = 0;
  m_pfnHdrCallback = NULL;

#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
  EVP_SignInit(&m_Hdr_sha1ctx, EVP_sha1());
  EVP_SignInit(&m_Hdr_sha256ctx, EVP_sha256());
  EVP_DigestInit(&m_Bdy_sha1ctx, EVP_sha1());
  EVP_DigestInit(&m_Bdy_sha256ctx, EVP_sha256());
#else
  m_Hdr_sha1ctx = EVP_MD_CTX_create();
  EVP_SignInit_ex(m_Hdr_sha1ctx, EVP_sha1(), NULL);

  m_Hdr_sha256ctx = EVP_MD_CTX_create();
  EVP_SignInit_ex(m_Hdr_sha256ctx, EVP_sha256(), NULL);

  m_Bdy_sha1ctx = EVP_MD_CTX_create();
  EVP_DigestInit_ex(m_Bdy_sha1ctx, EVP_sha1(), NULL);

  m_Bdy_sha256ctx = EVP_MD_CTX_create();
  EVP_DigestInit_ex(m_Bdy_sha256ctx, EVP_sha256(), NULL);

  m_Hdr_ed25519ctx = EVP_MD_CTX_create();

  m_Edy_sha256ctx = EVP_MD_CTX_create();
  EVP_DigestInit_ex(m_Edy_sha256ctx, EVP_sha256(), NULL);
#endif
}

CDKIMSign::~CDKIMSign()
{
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
  EVP_MD_CTX_cleanup(&m_Hdr_sha1ctx);
  EVP_MD_CTX_cleanup(&m_Hdr_sha256ctx);
  EVP_MD_CTX_cleanup(&m_Bdy_sha1ctx);
  EVP_MD_CTX_cleanup(&m_Bdy_sha256ctx);
#else
  EVP_MD_CTX_free(m_Hdr_sha1ctx);
  EVP_MD_CTX_free(m_Hdr_sha256ctx);
  EVP_MD_CTX_free(m_Hdr_ed25519ctx);
  EVP_MD_CTX_free(m_Bdy_sha1ctx);
  EVP_MD_CTX_free(m_Bdy_sha256ctx);
  EVP_MD_CTX_free(m_Edy_sha256ctx);
#endif
}

////////////////////////////////////////////////////////////////////////////////
//
// Init - save the options
//
////////////////////////////////////////////////////////////////////////////////
int CDKIMSign::Init(DKIMSignOptions *pOptions)
{
  int nRet = CDKIMBase::Init();

  m_Canon = pOptions->nCanon;

  // as of draft 01, these are the only allowed signing types:
  if ((m_Canon != DKIM_SIGN_SIMPLE_RELAXED) && (m_Canon != DKIM_SIGN_RELAXED)
      && (m_Canon != DKIM_SIGN_RELAXED_SIMPLE))
  {
    m_Canon = DKIM_SIGN_SIMPLE;
  }

  sSelector.assign(pOptions->szSelector);
  eSelector.assign(pOptions->szSelectorE);

  m_pfnHdrCallback = pOptions->pfnHeaderCallback;

  sDomain.assign(pOptions->szDomain);

  m_IncludeBodyLengthTag = (pOptions->nIncludeBodyLengthTag != 0);

  m_nBodyLength = 0;

  m_ExpireTime = pOptions->expireTime;

  sIdentity.assign(pOptions->szIdentity);

  m_nIncludeTimeStamp = pOptions->nIncludeTimeStamp;
  m_nIncludeQueryMethod = pOptions->nIncludeQueryMethod;
  m_nIncludeCopiedHeaders = pOptions->nIncludeCopiedHeaders;

  // NOTE: the following line is not backwards compatible with MD 8.0.3
  // because the szRequiredHeaders member was added after the release
  //sRequiredHeaders.assign(pOptions->szRequiredHeaders);

  //make sure there is a colon after the last header in the list
  if ((sRequiredHeaders.size() > 0) && sRequiredHeaders.at(sRequiredHeaders.size() - 1) != ':') {
    sRequiredHeaders.append(":");
  }

  m_nHash = pOptions->nHash;
  m_bReturnedSigAssembled = false;
  m_sCopiedHeaders.erase();

  // Initializes ED25519 header fields SigHdrs
#if ((OPENSSL_VERSION_NUMBER > 0x10101000L))
  SigHdrs.assign("");
  m_SigHdrs = 0;
#endif

  return nRet;
}

////////////////////////////////////////////////////////////////////////////////
//
// Hash - update the hash
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::Hash(const char *szBuffer, int nBufLength, bool bHdr)
{

  /** START DEBUG CODE **
  if (nBufLength == 2 && szBuffer[0] == '\r' && szBuffer[1] == '\n') {
    printf("[CRLF]\n");
  } else {
    char *szDbg = new char[nBufLength+1];
    strncpy(szDbg,szBuffer,nBufLength);
    szDbg[nBufLength] = '\0';
    printf("[%s]\n",szDbg);
  } ***

  if (fpdebug == NULL) {
    fpdebug = fopen("canon.msg", "wb");
  }

  fwrite(szBuffer,1,nBufLength,fpdebug);

  ** END DEBUG CODE **/

  if (bHdr) { /* Generate signature: b=... */
    if ((m_nHash == DKIM_HASH_SHA1) || (m_nHash == DKIM_HASH_SHA1_AND_SHA256))
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_SignUpdate(&m_Hdr_sha1ctx, szBuffer, nBufLength);
#else
      EVP_SignUpdate(m_Hdr_sha1ctx, szBuffer, nBufLength);
#endif
    if ((m_nHash == DKIM_HASH_SHA256) || (m_nHash == DKIM_HASH_SHA1_AND_SHA256)
        || (m_nHash == DKIM_HASH_RSA256_AND_ED25519))
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_SignUpdate(&m_Hdr_sha256ctx, szBuffer, nBufLength);
#else
      EVP_SignUpdate(m_Hdr_sha256ctx, szBuffer, nBufLength);
#endif
#if ((OPENSSL_VERSION_NUMBER > 0x10101000L))
    if ((m_nHash == DKIM_HASH_ED25519) || (m_nHash == DKIM_HASH_RSA256_AND_ED25519)) {
      SigHdrs.append(szBuffer, nBufLength);
      m_SigHdrs += nBufLength;
    }
#endif
  } else { /* lets go for body hash values: bh=... (either SHA1 or SHA256) */
    if ((m_nHash == DKIM_HASH_SHA1) || (m_nHash == DKIM_HASH_SHA1_AND_SHA256))
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_DigestUpdate(&m_Bdy_sha1ctx, szBuffer, nBufLength);
#else
      EVP_DigestUpdate(m_Bdy_sha1ctx, szBuffer, nBufLength);
#endif
    if (m_nHash != DKIM_HASH_SHA1)
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_DigestUpdate(&m_Bdy_sha256ctx, szBuffer, nBufLength);
#else
      EVP_DigestUpdate(m_Bdy_sha256ctx, szBuffer, nBufLength);
#endif
#if ((OPENSSL_VERSION_NUMBER > 0x10101000L))
    if ((m_nHash == DKIM_HASH_ED25519) || (m_nHash == DKIM_HASH_RSA256_AND_ED25519))
      EVP_DigestUpdate(m_Edy_sha256ctx, szBuffer, nBufLength);
#endif
  }
}

////////////////////////////////////////////////////////////////////////////////
//
// SignThisTag - return boolean whether or not to sign this tag
//
////////////////////////////////////////////////////////////////////////////////
bool CDKIMSign::SignThisTag(const string& sTag)
{
  bool bRet = true;

  if (_strnicmp(sTag.c_str(), "X-", 2) == 0 || _stricmp(sTag.c_str(), "Authentication-Results:") == 0
      || _stricmp(sTag.c_str(), "Return-Path:") == 0)
  {
    bRet = false;
  }

  return bRet;
}

bool ConvertHeaderToQuotedPrintable(const char *source, char *dest)
{
  bool bConvert = false;

  // do quoted printable
  static unsigned char hexchars[16] = {
      '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'};

  unsigned char *d = (unsigned char *)dest;
  for (const unsigned char *s = (const unsigned char *)source; *s != '\0'; s++) {
    if (*s >= 33 && *s <= 126 && *s != '=' && *s != ':' && *s != ';' && *s != '|') {
      *d++ = *s;
    } else {
      bConvert = true;
      *d++ = '=';
      *d++ = hexchars[*s >> 4];
      *d++ = hexchars[*s & 15];
    }
  }
  *d = '\0';

  return bConvert;
}

////////////////////////////////////////////////////////////////////////////////
//
// GetHeaderParams - Extract any needed header parameters
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::GetHeaderParams(const string& sHdr)
{
  if (_strnicmp(sHdr.c_str(), "X", 1) == 0) return;
  if (_strnicmp(sHdr.c_str(), "From:", 5) == 0) {
    sFrom.assign(sHdr.c_str() + 5);
  }
  if (_strnicmp(sHdr.c_str(), "Sender:", 7) == 0) {
    sSender.assign(sHdr.c_str() + 7);
  }

  if (m_nIncludeCopiedHeaders) {
    string::size_type pos = sHdr.find(':');

    if (pos != string::npos) {
      string sTag, sValue;
      char *workBuffer = new char[sHdr.size() * 3 + 1];

      sTag.assign(sHdr.substr(0, pos));
      sValue.assign(sHdr.substr(pos + 1, string::npos));

      ConvertHeaderToQuotedPrintable(sTag.c_str(), workBuffer);
      if (!m_sCopiedHeaders.empty()) {
        m_sCopiedHeaders.append("|");
      }
      m_sCopiedHeaders.append(workBuffer);
      m_sCopiedHeaders.append(":");
      ConvertHeaderToQuotedPrintable(sValue.c_str(), workBuffer);
      m_sCopiedHeaders.append(workBuffer);

      delete[] workBuffer;
    }
  }
}

////////////////////////////////////////////////////////////////////////////////
//
// ProcessHeaders - sign headers and save needed parameters (this is a lie)
//
////////////////////////////////////////////////////////////////////////////////
int CDKIMSign::ProcessHeaders(void)
{
  map<string, list<string>::reverse_iterator> IterMap;
  map<string, list<string>::reverse_iterator>::iterator IterMapIter;
  list<string>::reverse_iterator riter;
  list<string>::iterator iter;
  string sTag;
  bool bFromHeaderFound = false;

  // walk the header list
  for (iter = HeaderList.begin(); iter != HeaderList.end(); iter++) {
    sTag.assign(*iter);

    // look for a colon
    string::size_type pos = sTag.find(':');

    if (pos != string::npos) {
      int nSignThisTag = 1;

      // hack off anything past the colon
      sTag.erase(pos + 1, string::npos);

      // is this the From: header?
      if (_stricmp(sTag.c_str(), "From:") == 0) {
        bFromHeaderFound = true;
        nSignThisTag = 1;
        IsRequiredHeader(sTag);  // remove from required header list
      }
      // is this in the list of headers that must be signed?
      else if (IsRequiredHeader(sTag))
      {
        nSignThisTag = 1;
      } else {
        if (m_pfnHdrCallback) {
          nSignThisTag = m_pfnHdrCallback(iter->c_str());
        } else {
          nSignThisTag = SignThisTag(sTag) ? 1 : 0;
        }
      }

      // save header parameters
      GetHeaderParams(*iter);

      if (nSignThisTag > 0) {
        // add this tag to h=
        hParam.append(sTag);

        IterMapIter = IterMap.find(sTag);

        riter = (IterMapIter == IterMap.end()) ? HeaderList.rbegin() : IterMapIter->second;

        // walk the list in reverse looking for the last instance of this header
        while (riter != HeaderList.rend()) {
          if (_strnicmp(riter->c_str(), sTag.c_str(), sTag.size()) == 0) {
            ProcessHeader(*riter);

            // save the reverse iterator position for this tag
            riter++;
            IterMap[sTag] = riter;
            break;
          }
          riter++;
        }
      }
    }
  }

  if (!bFromHeaderFound) {
    string sFrom("From:");
    hParam.append(sFrom);
    IsRequiredHeader(sFrom);  // remove from required header list
                              //    Hash("\r\n",2);
  }

  hParam.append(sRequiredHeaders);

  //  string::size_type end = sRequiredHeaders.find(':');
  //  while (end != string::npos)
  //  {
  //    Hash("\r\n",2);
  //    end = sRequiredHeaders.find(':', end+1);
  //  }

  // remove the last colon from h=
  if (hParam.at(hParam.size() - 1) == ':') hParam.erase(hParam.size() - 1, string::npos);

  return DKIM_SUCCESS;
}

void CDKIMSign::ProcessHeader(const string& sHdr)
{
  switch (HIWORD(m_Canon)) {
    case DKIM_CANON_SIMPLE:
      Hash(sHdr.c_str(), sHdr.size(), true);
      Hash("\r\n", 2, true);
      break;

    case DKIM_CANON_NOWSP: {
      string sTemp = sHdr;
      RemoveSWSP(sTemp);

      // convert characters before ':' to lower case
      for (char *s = (char *)sTemp.c_str(); *s != '\0' && *s != ':'; s++) {
        if (*s >= 'A' && *s <= 'Z') *s += 'a' - 'A';
      }

      Hash(sTemp.c_str(), sTemp.size(), true);
      Hash("\r\n", 2, true);
    } break;

    case DKIM_CANON_RELAXED: {
      string sTemp = RelaxHeader(sHdr);
      Hash(sTemp.c_str(), sTemp.length(), true);
      Hash("\r\n", 2, true);
    } break;
  }
}

int CDKIMSign::ProcessBody(char *szBuffer, int nBufLength, bool bEOF)
{
  switch (LOWORD(m_Canon)) {
    case DKIM_CANON_SIMPLE:
      if (nBufLength > 0) {
        while (m_EmptyLineCount > 0) {
          Hash("\r\n", 2, false);
          m_nBodyLength += 2;
          m_EmptyLineCount--;
        }
        Hash(szBuffer, nBufLength, false);
        Hash("\r\n", 2, false);
        m_nBodyLength += nBufLength + 2;
      } else {
        m_EmptyLineCount++;
        if (bEOF) {
          Hash("\r\n", 2, false);
          m_nBodyLength += 2;
        }
      }
      break;
    case DKIM_CANON_NOWSP:
      RemoveSWSP(szBuffer, nBufLength);
      if (nBufLength > 0) {
        Hash(szBuffer, nBufLength, false);
        m_nBodyLength += nBufLength;
      }
      break;
    case DKIM_CANON_RELAXED:
      CompressSWSP(szBuffer, nBufLength);
      if (nBufLength > 0) {
        while (m_EmptyLineCount > 0) {
          Hash("\r\n", 2, false);
          m_nBodyLength += 2;
          m_EmptyLineCount--;
        }
        Hash(szBuffer, nBufLength, false);
        m_nBodyLength += nBufLength;
        if (!bEOF) {
          Hash("\r\n", 2, false);
          m_nBodyLength += 2;
        }
      } else
        m_EmptyLineCount++;
      break;
  }

  return DKIM_SUCCESS;
}

bool CDKIMSign::ParseFromAddress(void)
{
  string::size_type pos;
  string sAddress;

  if (!sFrom.empty()) {
    sAddress.assign(sFrom);
  } else if (!sSender.empty()) {
    sAddress.assign(sSender);
  } else {
    return false;
  }

  // simple for now, beef it up later

  // remove '<' and anything before it
  pos = sAddress.find('<');
  if (pos != string::npos) sAddress.erase(0, pos);

  // remove '>' and anything after it
  pos = sAddress.find('>');
  if (pos != string::npos) sAddress.erase(pos, string::npos);

  // look for '@' symbol
  pos = sAddress.find('@');
  if (pos == string::npos) return false;

  if (sDomain.empty()) {
    sDomain.assign(sAddress.c_str() + pos + 1);
    RemoveSWSP(sDomain);
  }

  return true;
}

////////////////////////////////////////////////////////////////////////////////
//
// InitSig - initialize signature folding algorithm
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::InitSig(void)
{
  m_sSig.reserve(1024);
  m_sSig.assign("DKIM-Signature:");
  m_nSigPos = m_sSig.size();
}

////////////////////////////////////////////////////////////////////////////////
//
// AddTagToSig - add tag and value to signature folding if necessary
//               if bFold, fold at cbrk char
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::AddTagToSig(const char *const Tag, const string& sValue, char cbrk, bool bFold)
{
  int nTagLen = strlen(Tag);

  AddInterTagSpace((!bFold) ? sValue.size() + nTagLen + 2 : nTagLen + 2);

  m_sSig.append(Tag);
  m_sSig.append("=");
  m_nSigPos += 1 + nTagLen;

  if (!bFold) {
    m_sSig.append(sValue);
    m_nSigPos += sValue.size();
  } else {
    AddFoldedValueToSig(sValue, cbrk);
  }
  m_sSig.append(";");
  m_nSigPos++;
}

////////////////////////////////////////////////////////////////////////////////
//
// AddTagToSig - add tag and numeric value to signature folding if necessary
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::AddTagToSig(const char *const Tag, unsigned long nValue)
{
  char szValue[64];
  sprintf(szValue, "%lu", nValue);
  AddTagToSig(Tag, szValue, 0, false);
}

////////////////////////////////////////////////////////////////////////////////
//
// AddInterTagSpace - add space or fold here
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::AddInterTagSpace(int nSizeOfNextTag)
{
  if (m_nSigPos + nSizeOfNextTag + 1 > OptimalHeaderLineLength) {
    //    m_sSig.append("\r\n\t");
    m_sSig.append("\r\n  "); /* s/qmail style */
    m_nSigPos = 1;
  } else {
    m_sSig.append(" ");
    m_nSigPos++;
  }
}

////////////////////////////////////////////////////////////////////////////////
//
// AddTagToSig - add value to signature folding if necessary
//               if cbrk == 0 fold anywhere, otherwise fold only at cbrk
//
////////////////////////////////////////////////////////////////////////////////
void CDKIMSign::AddFoldedValueToSig(const string& sValue, char cbrk)
{
  string::size_type pos = 0;

  if (cbrk == 0) {
    // fold anywhere
    while (pos < sValue.size()) {
      string::size_type len = OptimalHeaderLineLength - m_nSigPos;
      if (len > sValue.size() - pos) len = sValue.size() - pos;
      m_sSig.append(sValue.substr(pos, len));
      m_nSigPos += len;
      pos += len;

      if (pos < sValue.size()) {
        //        m_sSig.append("\r\n\t");
        m_sSig.append("\r\n  "); /* s/qmail style */
        m_nSigPos = 1;
      }
    }
  } else {
    // fold only at cbrk
    while (pos < sValue.size()) {
      string::size_type len = OptimalHeaderLineLength - m_nSigPos;
      string::size_type brkpos;

      if (sValue.size() - pos < len) {
        brkpos = sValue.size();
      } else {
        brkpos = sValue.rfind(cbrk, pos + len);
      }

      if (brkpos == string::npos || brkpos < pos) {
        brkpos = sValue.find(cbrk, pos);
        if (brkpos == string::npos) {
          brkpos = sValue.size();
        }
      }

      len = brkpos - pos + 1;

      m_sSig.append(sValue.substr(pos, len));

      m_nSigPos += len;
      pos += len;

      if (pos < sValue.size()) {
        //        m_sSig.append("\r\n\t");
        m_sSig.append("\r\n  "); /* s/qmail style */
        m_nSigPos = 1;
      }
    }
  }
}

////////////////////////////////////////////////////////////////////////////////
//
// GetSig - compute hash and return signature header in szSignature
//
////////////////////////////////////////////////////////////////////////////////
int CDKIMSign::GetSig2(char *szRSAKey, char *szECCKey, char **pszSignature)
{
  if (szRSAKey == NULL && szECCKey == NULL) {
    return DKIM_BAD_PRIVATE_KEY;
  }

  if (pszSignature == NULL) {
    return DKIM_BUFFER_TOO_SMALL;
  }

  int nRet = AssembleReturnedSig(szRSAKey, szECCKey);

  if (nRet != DKIM_SUCCESS) return nRet;

  *pszSignature = (char *)m_sReturnedSig.c_str();

  return DKIM_SUCCESS;
}


////////////////////////////////////////////////////////////////////////////////
//
// IsRequiredHeader - Check if header in required list. If so, delete
//                    header from list.
//
////////////////////////////////////////////////////////////////////////////////
bool CDKIMSign::IsRequiredHeader(const string& sTag)
{
  string::size_type start = 0;
  string::size_type end = sRequiredHeaders.find(':');

  while (end != string::npos) {
    // check for a zero-length header
    if (start == end) {
      sRequiredHeaders.erase(start, 1);
    } else {
      if (_stricmp(sTag.c_str(), sRequiredHeaders.substr(start, end - start + 1).c_str()) == 0) {
        sRequiredHeaders.erase(start, end - start + 1);
        return true;
      } else {
        start = end + 1;
      }
    }

    end = sRequiredHeaders.find(':', start);
  }

  return false;
}
////////////////////////////////////////////////////////////////////////////////
//
// ConstructSignature
//
//  Here, we don't construct the 'signature' but rather the DKIM header
//  multiply and indidually crafted for each distinct nSigAlg method
//
//  nSigAlg: DKIM_HASH_SHA1, DKIM_HASH_SHA256, DKIM_HASH_ED25519
//
////////////////////////////////////////////////////////////////////////////////
int CDKIMSign::ConstructSignature(char *szPrivKey, int nSigAlg)
{
  string sSignedSig;
  unsigned char *sig;
  EVP_PKEY *pkey = 0;
  BIO *bio, *b64;
  unsigned int siglen;
  int size;
  int len;
  char *buf;
  int nSignRet;

  /* construct the DKIM-Signature: header and add to hash */
  InitSig();

  AddTagToSig("v", "1", 0, false);

  switch (nSigAlg) {
    case DKIM_HASH_SHA1:    AddTagToSig("a", "rsa-sha1", 0, false); break;
    case DKIM_HASH_SHA256:  AddTagToSig("a", "rsa-sha256", 0, false); break;
    case DKIM_HASH_ED25519: AddTagToSig("a", "ed25519-sha256", 0, false); break;
  }

  switch (m_Canon) {
    case DKIM_SIGN_SIMPLE:         AddTagToSig("c", "simple/simple", 0, false); break;
    case DKIM_SIGN_SIMPLE_RELAXED: AddTagToSig("c", "simple/relaxed", 0, false); break;
    case DKIM_SIGN_RELAXED:        AddTagToSig("c", "relaxed/relaxed", 0, false); break;
    case DKIM_SIGN_RELAXED_SIMPLE: AddTagToSig("c", "relaxed/simple", 0, false); break;
  }

  AddTagToSig("d", sDomain, 0, false);
  if (nSigAlg == DKIM_HASH_ED25519)
    AddTagToSig("s", eSelector, 0, false);
  else
    AddTagToSig("s", sSelector, 0, false);
  if (m_IncludeBodyLengthTag) {
    AddTagToSig("l", m_nBodyLength);
  }
  if (m_nIncludeTimeStamp != 0) {
    time_t t;
    time(&t);
    AddTagToSig("t", t);
  }
  if (m_ExpireTime != 0) {
    AddTagToSig("x", m_ExpireTime);
  }
  if (!sIdentity.empty()) {
    AddTagToSig("i", sIdentity, 0, false);
  }
  if (m_nIncludeQueryMethod) {
    AddTagToSig("q", "dns/txt", 0, false);
  }

  AddTagToSig("h", hParam, ':', true);  // copied headers follow the ':'
  if (m_nIncludeCopiedHeaders) {
    AddTagToSig("z", m_sCopiedHeaders, 0, true);
  }

  /* Set up context for (body) hash */

  unsigned char Hash[4096];
  unsigned int nHashLen = 0;

  switch (nSigAlg) {
    case DKIM_HASH_SHA1:
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_DigestFinal(&m_Bdy_sha1ctx, Hash, &nHashLen);
      break;
#else
      EVP_DigestFinal_ex(m_Bdy_sha1ctx, Hash, &nHashLen);
      break;
#endif
    case DKIM_HASH_SHA256:
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_DigestFinal(&m_Bdy_sha256ctx, Hash, &nHashLen);
      break;
#else
      EVP_DigestFinal_ex(m_Bdy_sha256ctx, Hash, &nHashLen);
      break;
#endif
#if (OPENSSL_VERSION_NUMBER > 0x10101000L)
    case DKIM_HASH_ED25519: EVP_DigestFinal_ex(m_Edy_sha256ctx, Hash, &nHashLen); break;
#endif
  }

  bio = BIO_new(BIO_s_mem());
  if (!bio) return DKIM_OUT_OF_MEMORY;

  b64 = BIO_new(BIO_f_base64());
  if (!b64) {
    BIO_free(bio);
    return DKIM_OUT_OF_MEMORY;
  }
  BIO_set_flags(b64, BIO_FLAGS_BASE64_NO_NL);
  BIO_push(b64, bio);
  if (BIO_write(b64, Hash, nHashLen) < (int)nHashLen) {
    BIO_free_all(b64);
    return DKIM_OUT_OF_MEMORY;
  }
  BIO_flush(b64);

  len = nHashLen * 2;
  buf = new char[len];

  if (buf == NULL) {
    BIO_free_all(b64);
    return DKIM_OUT_OF_MEMORY;
  }

  size = BIO_read(bio, buf, len);
  BIO_free_all(b64);

  // this should never happen
  if (size >= len) {
    delete[] buf;
    return DKIM_OUT_OF_MEMORY;
  }

  buf[size] = '\0';
  AddTagToSig("bh", buf, 0, true);
  delete[] buf;

  AddInterTagSpace(3);

  m_sSig.append("b=");
  m_nSigPos += 2;

  // Force a full copy - no reference copies please
  sSignedSig.assign(m_sSig.c_str());

  // note that since we're not calling hash here, need to dump this
  // to the debug file if you want the full canonical form

  string sTemp;

  if (HIWORD(m_Canon) == DKIM_CANON_RELAXED) {
    sTemp = RelaxHeader(sSignedSig);
  } else {
    sTemp = sSignedSig.c_str();
  }

  /* Update streaming signatures */

  switch (nSigAlg) {
    case DKIM_HASH_SHA1:
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_SignUpdate(&m_Hdr_sha1ctx, sTemp.c_str(), sTemp.size());
      break;
#else
      EVP_SignUpdate(m_Hdr_sha1ctx, sTemp.c_str(), sTemp.size());
      break;
#endif
    case DKIM_HASH_SHA256:
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      EVP_SignUpdate(&m_Hdr_sha256ctx, sTemp.c_str(), sTemp.size());
      break;
#else
      EVP_SignUpdate(m_Hdr_sha256ctx, sTemp.c_str(), sTemp.size());
      break;
#endif
#if ((OPENSSL_VERSION_NUMBER > 0x10101000L))
    case DKIM_HASH_ED25519:
      SigHdrs.append(sTemp.c_str(), sTemp.size());
      m_SigHdrs += sTemp.size();
      break;
#endif
  }

  bio = BIO_new_mem_buf(szPrivKey, -1);
  if (bio == NULL) return DKIM_OUT_OF_MEMORY;

  pkey = PEM_read_bio_PrivateKey(bio, NULL, NULL, NULL);  // FIXME - done
  BIO_free(bio);

  if (!pkey) {
    return DKIM_BAD_PRIVATE_KEY;
  }
  siglen = EVP_PKEY_size(pkey);

  sig = (unsigned char *)OPENSSL_malloc(siglen);
  if (sig == NULL) {
    EVP_PKEY_free(pkey);
    return DKIM_OUT_OF_MEMORY;
  }

  /* Finish streaming signature and potentially go for Ed25519 signatures */

  size_t sig_len;
  unsigned char *SignMsg;

  switch (nSigAlg) {
    case DKIM_HASH_SHA1:
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      nSignRet = EVP_SignFinal(&m_Hdr_sha1ctx, sig, &siglen, pkey);
      break;
#else
      nSignRet = EVP_SignFinal(m_Hdr_sha1ctx, sig, &siglen, pkey);
      break;
#endif
    case DKIM_HASH_SHA256:
#if (                                      \
    (OPENSSL_VERSION_NUMBER < 0x10100000L) \
    || (LIBRESSL_VERSION_NUMBER > 0 && LIBRESSL_VERSION_NUMBER < 0x20700000L))
      nSignRet = EVP_SignFinal(&m_Hdr_sha256ctx, sig, &siglen, pkey);
      break;
#else
      nSignRet = EVP_SignFinal(m_Hdr_sha256ctx, sig, &siglen, pkey);
      break;
#endif
#if (OPENSSL_VERSION_NUMBER > 0x10101000L)
    case DKIM_HASH_ED25519:
      EVP_DigestSignInit(m_Hdr_ed25519ctx, NULL, NULL, NULL, pkey);
      SignMsg = (unsigned char *)SigHdrs.c_str();
      EVP_DigestSign(m_Hdr_ed25519ctx, NULL, &sig_len, SignMsg, m_SigHdrs);
      sig = (unsigned char *)OPENSSL_malloc(sig_len);
      nSignRet = EVP_DigestSign(m_Hdr_ed25519ctx, sig, &sig_len, SignMsg, m_SigHdrs);
      siglen = (unsigned int)sig_len;
      break;
#endif
  }
  EVP_PKEY_free(pkey);

  if (!nSignRet) {
    OPENSSL_free(sig);
    return DKIM_BAD_PRIVATE_KEY;  // key too small
  }

  bio = BIO_new(BIO_s_mem());
  if (!bio) {
    return DKIM_OUT_OF_MEMORY;
  }

  b64 = BIO_new(BIO_f_base64());
  if (!b64) {
    BIO_free(bio);
    return DKIM_OUT_OF_MEMORY;
  }

  BIO_set_flags(b64, BIO_FLAGS_BASE64_NO_NL);
  BIO_push(b64, bio);

  if (BIO_write(b64, sig, siglen) < (int)siglen) {
    OPENSSL_free(sig);
    BIO_free_all(b64);
    return DKIM_OUT_OF_MEMORY;
  }
  BIO_flush(b64);
  OPENSSL_free(sig);

  len = siglen * 2;
  buf = new char[len];

  if (buf == NULL) {
    BIO_free_all(b64);
    return DKIM_OUT_OF_MEMORY;
  }

  size = BIO_read(bio, buf, len);
  BIO_free_all(b64);

  // this should never happen
  if (size >= len) {
    delete[] buf;
    return DKIM_OUT_OF_MEMORY;
  }

  buf[size] = '\0';
  AddFoldedValueToSig(buf, 0);
  delete[] buf;
  return DKIM_SUCCESS;
}

////////////////////////////////////////////////////////////////////////////////
//
// AssembleReturnSig
//
//  calls ConstructSignature
//  for all different hashes and signature key files
//
////////////////////////////////////////////////////////////////////////////////
int CDKIMSign::AssembleReturnedSig(char *szRSAKey, char *szECCKey)
{
  int nRet;

  if (m_bReturnedSigAssembled) return DKIM_SUCCESS;

  ProcessFinal();

  if (ParseFromAddress() == false) {
    return DKIM_NO_SENDER;
  }

  string ed25519Sig, sha256Sig, sha1Sig;

  if ((m_nHash == DKIM_HASH_ED25519) || (m_nHash == DKIM_HASH_RSA256_AND_ED25519)) {
    nRet = ConstructSignature(szECCKey, DKIM_HASH_ED25519);
    if (nRet == DKIM_SUCCESS) {
      ed25519Sig.assign(m_sSig);
    } else {
      return nRet;
    }
  }

  if ((m_nHash == DKIM_HASH_SHA256) || (m_nHash == DKIM_HASH_SHA1_AND_SHA256)
      || (m_nHash == DKIM_HASH_RSA256_AND_ED25519))
  {
    nRet = ConstructSignature(szRSAKey, DKIM_HASH_SHA256);
    if (nRet == DKIM_SUCCESS) {
      sha256Sig.assign(m_sSig);
    } else {
      return nRet;
    }
  }

  if ((m_nHash == DKIM_HASH_SHA1) || (m_nHash == DKIM_HASH_SHA1_AND_SHA256)) {
    nRet = ConstructSignature(szRSAKey, DKIM_HASH_SHA1);
    if (nRet == DKIM_SUCCESS) {
      sha1Sig.assign(m_sSig);
    } else {
      return nRet;
    }
  }

  //  fclose(fpdebug);
  //  fpdebug = NULL;

  if (!ed25519Sig.empty()) {
    /*    if (!m_sReturnedSig.empty()) {
      m_sReturnedSig.append("\r\n");
    }
    */
    m_sReturnedSig.assign(ed25519Sig);
  }

  if (!sha1Sig.empty()) {
    if (!m_sReturnedSig.empty()) {
      m_sReturnedSig.append("\r\n");
    }
    m_sReturnedSig.append(sha1Sig);
  }

  if (!sha256Sig.empty()) {
    if (!m_sReturnedSig.empty()) {
      m_sReturnedSig.append("\r\n");
    }
    m_sReturnedSig.append(sha256Sig);
  }

  m_bReturnedSigAssembled = true;
  return DKIM_SUCCESS;
}