MessageDigestAttribute.cpp

Go to the documentation of this file.

#include "MessageDigestAttribute.h"

#include "ASN1Helper.h"
#include "CryptographicMessageSyntax2004.h"
#include "Buffer.h"
#include "ToolkitUtils.h"
#include "PKIFMessageException.h"

struct CPKIFMessageDigestAttributeImpl
{
  CPKIFMessageDigestAttribute *m_parent;

  CPKIFBufferPtr m_messageDigest;
  void ParseAttributes (const unsigned char *buf, const unsigned int length);
};
void CPKIFMessageDigestAttributeImpl::ParseAttributes (
    const unsigned char *buf, 
    const unsigned int length)
{
    CACASNWRAPPER_CREATE(CACCMSAttributeValues, attrsVals);
    CACCMSAttributeValues* attributesList = attrsVals.Decode(buf, length);

    OOCTXT ctxt;
    initContext (&ctxt);
    ASN1OpenType* ot = (ASN1OpenType*)((DListNode*)attributesList->head)->data;
    setBERDecBufPtr(&ctxt, ot->data, ot->numocts, NULL, NULL);

    unsigned char* iv = NULL;
    unsigned int ivLen = ot->numocts;

    int stat = decodeDynOctStr (&ctxt, (const OSOCTET* *)&iv, (OSUINT32 *)&ivLen, ASN1EXPL, ot->numocts);
    if(0 != stat)
    {
        //upon failure - clean up and throw an exception
        //freeEncodeBuffer(&ctxt);
        //memFreeAll(&ctxt);
        freeContext(&ctxt);

        throw CPKIFException(TOOLKIT_MESSAGE, ASN1_DECODE_ERROR);
    }
    else
    {
        try
        {
            CPKIFBufferPtr tmp(new CPKIFBuffer(iv, ivLen));
            m_parent->SetMessageDigest(tmp);
        }
        catch(std::bad_alloc& ba)
        {
            //freeEncodeBuffer(&ctxt);
            //memFreeAll(&ctxt);
            freeContext(&ctxt);

            throw ba;
        }

        //clean up the context
        //freeEncodeBuffer(&ctxt);
        //memFreeAll(&ctxt);
        freeContext(&ctxt);
    }
}



char CPKIFMessageDigestAttribute::extOID[] = "1.2.840.113549.1.9.4";
CPKIFMessageDigestAttribute::CPKIFMessageDigestAttribute()
    :m_impl (new CPKIFMessageDigestAttributeImpl)
{
    LOG_STRING_DEBUG("CPKIFMessageDigestAttribute::CPKIFMessageDigestAttribute()", TOOLKIT_MESSAGE_ASN, 0, this);
    
    m_impl->m_parent = this;
}
/*CPKIFMessageDigestAttribute::CPKIFMessageDigestAttribute(
    const CACCMSAttribute& ext)
    :m_impl (new CPKIFMessageDigestAttributeImpl)
{
    LOG_STRING_DEBUG("CPKIFMessageDigestAttribute::CPKIFMessageDigestAttribute(const CACCMSAttribute& ext)", TOOLKIT_MESSAGE_ASN, 0, this);

    m_impl->m_parent = this;
    m_impl->ParseAttributes (ext.attrValues.data, ext.attrValues.numocts);
}
*/

CPKIFMessageDigestAttribute::CPKIFMessageDigestAttribute(
    const CPKIFBufferPtr& buf)
    :m_impl (new CPKIFMessageDigestAttributeImpl)
{
    LOG_STRING_DEBUG("CPKIFMessageDigestAttribute::CPKIFMessageDigestAttribute(const CACCMSAttribute& ext)", TOOLKIT_MESSAGE_ASN, 0, this);

    m_impl->m_parent = this;
    m_impl->ParseAttributes (buf->GetBuffer(), buf->GetLength());
}

CPKIFMessageDigestAttribute::~CPKIFMessageDigestAttribute()
{
    LOG_STRING_DEBUG("CPKIFMessageDigestAttribute::~CPKIFMessageDigestAttribute()", TOOLKIT_MESSAGE_ASN, 0, this);

    delete m_impl;
    m_impl = NULL;
}

//generic attribute functions
CPKIFOIDPtr CPKIFMessageDigestAttribute::GetOID() const
{
    return g_messageDigestAttribute;
}
void CPKIFMessageDigestAttribute::GetValues(
    CPKIFBufferList& values) const
{
    LOG_STRING_DEBUG("CPKIFMessageDigestAttribute::GetValues(CPKIFBufferList& values)", TOOLKIT_MESSAGE_ASN, 0, this);

    values.clear();

    OOCTXT ctxt;
    initContext (&ctxt);
    setBEREncBufPtr(&ctxt, NULL, 0);
    OOCTXT* pctxt = &ctxt;

    if(m_impl->m_messageDigest != (CPKIFBuffer*)NULL)
    {
        int len = encodeOctStr (pctxt, m_impl->m_messageDigest->GetBuffer(), m_impl->m_messageDigest->GetLength(), ASN1EXPL);
        OSOCTET* buf = getBEREncBufPtr(pctxt);
        if(0 < len && NULL != buf)
        {
            //success
            try
            {
                CPKIFBufferPtr tmp(new CPKIFBuffer(buf, len));
                values.push_back(tmp);
            }
            catch(std::bad_alloc& ba)
            {
                //freeEncodeBuffer(&ctxt);
                //memFreeAll(&ctxt);
                freeContext(&ctxt);

                throw ba;
            }

            //freeEncodeBuffer(&ctxt);
            //memFreeAll(&ctxt);
            freeContext(&ctxt);
        }
        else
        {
            //failure - clean up and throw an exception
            //freeEncodeBuffer(&ctxt);
            //memFreeAll(&ctxt);
            freeContext(&ctxt);
            throw CPKIFException(TOOLKIT_MESSAGE, ASN1_ENCODE_ERROR);
        }
    }
}
void CPKIFMessageDigestAttribute::SetMessageDigest(
    CPKIFBufferPtr& md)
{
    LOG_STRING_DEBUG("CPKIFMessageDigestAttribute::SetMessageDigest(CPKIFBufferPtr& md)", TOOLKIT_MESSAGE_ASN, 0, this);

    //accept no NULLs
    if(md == (CPKIFBuffer*)NULL)
        throw CPKIFMessageException(TOOLKIT_MESSAGE, COMMON_INVALID_INPUT);

    m_impl->m_messageDigest = md;
}
CPKIFBufferPtr CPKIFMessageDigestAttribute::GetMessageDigest()
{
    return m_impl->m_messageDigest;
}