MessageImprint.cpp

Go to the documentation of this file.

#include "MessageImprint.h"
#include "PKIFTSP.h"

#ifdef _WIN32
#include "PKIFCAPIRaw.h"
#endif

#include "ASN1Helper.h"
#include "ToolkitUtils.h"
#include "PKIXTSP.h"
#include "Buffer.h"
#include "AlgorithmIdentifier.h"
#include "PKIFMediators.h"
#include "IPKIFHashContext.h"
#include "IPKIFCryptoMisc.h"
#include "PKIFCryptUtils.h"


struct CPKIFMessageImprintImpl 
{
    CPKIFAlgorithmIdentifierPtr m_hashAlgorithm;
    CPKIFBufferPtr m_hashedMessage;
};


CPKIFMessageImprint::CPKIFMessageImprint() :m_impl(new CPKIFMessageImprintImpl)
{
    LOG_STRING_DEBUG("CPKIFMessageImprint::CPKIFMessageImprint()", TOOLKIT_TSP_ASN, 0, this);

    m_impl->m_hashAlgorithm = g_sha1AI; //default to SHA1
}
CPKIFMessageImprint::CPKIFMessageImprint(
    //PKIFTSPMessageImprint& mi)
    const CPKIFBufferPtr& miBuf)
  :m_impl (new CPKIFMessageImprintImpl)
{
    LOG_STRING_DEBUG("CPKIFMessageImprint::CPKIFMessageImprint(PKIFTSPMessageImprint& mi)", TOOLKIT_TSP_ASN, 0, this);

    CACASNWRAPPER_CREATE(PKIFTSPMessageImprint, MsgWrapper);
    PKIFTSPMessageImprint* mi = MsgWrapper.Decode(miBuf->GetBuffer(), miBuf->GetLength());

    //CPKIFAlgorithmIdentifier creation
    CPKIFOIDPtr algOID(new CPKIFOID(mi->hashAlgorithm.algorithm.subid, mi->hashAlgorithm.algorithm.numids));
    
    CPKIFBufferPtr paramBuf;
    if(mi->hashAlgorithm.m.parametersPresent)
    {
        paramBuf = CPKIFBufferPtr(new CPKIFBuffer(mi->hashAlgorithm.parameters.data, mi->hashAlgorithm.parameters.numocts));
    
    
    }
    
    CPKIFAlgorithmIdentifierPtr ai(new CPKIFAlgorithmIdentifier(algOID, paramBuf)); 
    //CPKIFAlgorithmIdentifierPtr ai(new CPKIFAlgorithmIdentifier(mi->hashAlgorithm));
    m_impl->m_hashAlgorithm = ai;

    CPKIFBufferPtr bp(new CPKIFBuffer(mi->hashedMessage.data, mi->hashedMessage.numocts));
    m_impl->m_hashedMessage = bp;
}
CPKIFMessageImprint::~CPKIFMessageImprint()
{
    LOG_STRING_DEBUG("CPKIFMessageImprint::~CPKIFMessageImprint()", TOOLKIT_TSP_ASN, 0, this);

    if (m_impl) {
      delete m_impl;
    }   
}
void CPKIFMessageImprint::SetHashAlgorithm(
    CPKIFAlgorithmIdentifierPtr& hashAlgorithm)
{
    m_impl->m_hashAlgorithm = hashAlgorithm;
}
CPKIFAlgorithmIdentifierPtr CPKIFMessageImprint::GetHashAlgorithm() const
{
    return m_impl->m_hashAlgorithm;
}
void CPKIFMessageImprint::SetHashedMessage(
    CPKIFBufferPtr& hashedMessage)
{
    m_impl->m_hashedMessage = hashedMessage;
}
CPKIFBufferPtr CPKIFMessageImprint::GetHashedMessage() const
{
    return m_impl->m_hashedMessage;
}
CPKIFBufferPtr CPKIFMessageImprint::HashAndSet(
    PKIFCRYPTO::HASH_ALG hashAlgorithm, 
    CPKIFBufferPtr& dataToHash, 
    IPKIFMediator* m)
{
    LOG_STRING_DEBUG("CPKIFMessageImprint::HashAndSet(HASH_ALG hashAlgorithm, CPKIFBufferPtr& dataToHash, IPKIFMediator* m)", TOOLKIT_TSP_ASN, 0, this);

    IPKIFCryptoMisc* iMisc = NULL;
    if(NULL != m)
        iMisc = m->GetMediator<IPKIFCryptoMisc>();

    unsigned char pResult[MAXHASH];
    int nResultLen = MAXHASH;

    if(NULL != iMisc)
    {
        //if we can get a pointer to an externally provided
        //interface use it
        IPKIFHashContext* hash = iMisc->HashInit(hashAlgorithm);
        iMisc->HashUpdate(hash, (unsigned char*)dataToHash->GetBuffer(), dataToHash->GetLength());
        iMisc->HashFinal(hash, pResult, &nResultLen);
        delete hash;
    }
    else
    {
        //otherwise create a temp object to generate the nonce

        IPKIFCryptoMisc* raw = GetPlatformCryptoMisc(); // #include "PKIFCryptUtils.h"
        IPKIFHashContext* hash = raw->HashInit(hashAlgorithm);
        raw->HashUpdate(hash, (unsigned char*)dataToHash->GetBuffer(), dataToHash->GetLength());
        raw->HashFinal(hash, pResult, &nResultLen);
        delete hash;

        /*  CPKIFCAPIRaw raw;
        IPKIFHashContext* hash = raw.HashInit(hashAlgorithm);
        raw.HashUpdate(hash, (unsigned char*)dataToHash->GetBuffer(), dataToHash->GetLength());
        raw.HashFinal(hash, pResult, &nResultLen);
        delete hash;
        */
    }

    CPKIFBufferPtr buf(new CPKIFBuffer(pResult, nResultLen));
    SetHashedMessage(buf);

    CPKIFAlgorithmIdentifierPtr ai = GetHashAlgAI(hashAlgorithm);
    SetHashAlgorithm(ai);

    return buf;
}