ContentTypeAttribute.cpp

Go to the documentation of this file.

#include "ContentTypeAttribute.h"

#include "OID.h"
#include "Buffer.h"

#include "CryptographicMessageSyntax2004.h"

#include "ASN1Helper.h"
#include "ToolkitUtils.h"

#include "PKIFMessageException.h"


struct CPKIFContentTypeAttributeImpl
{
  CPKIFContentTypeAttribute *m_parent;

  CPKIFOIDPtr m_contentType;
  void ParseAttributes (const unsigned char *buf, const unsigned int length);
};
void CPKIFContentTypeAttributeImpl::ParseAttributes (
    const unsigned char *buf, 
    const unsigned int length)
{
    CACASNWRAPPER_CREATE(CACCMSAttributeValues, attrsVals);
    CACCMSAttributeValues* attributesList = attrsVals.Decode(buf, length);

    //set up an ASN.1 context to parse the OID from the attribute
    OOCTXT ctxt;
    initContext (&ctxt);

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

    ASN1OBJID o;

    //parse the OID
    int stat = decodeOID (&ctxt, &o, 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
        {
            //otherwise set the value then clean up
            CPKIFOIDPtr tmp(new CPKIFOID(o.subid,o.numids));
            m_parent->SetContentType(tmp);
        }
        catch(std::bad_alloc& ba)
        {
            //freeEncodeBuffer(&ctxt);
            //memFreeAll(&ctxt);
            freeContext(&ctxt);

            throw ba;
        }

        //freeEncodeBuffer(&ctxt);
        //memFreeAll(&ctxt);
        freeContext(&ctxt);
    }
}


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

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

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

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


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

    delete m_impl;
    m_impl = NULL;
}

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

    values.clear();

    //set up an ASN.1 context to encode the OID value
    OOCTXT ctxt;
    initContext (&ctxt);
    setBEREncBufPtr(&ctxt, NULL, 0);

    CPKIFStringPtr str(new std::string(m_impl->m_contentType->ToString())); 
    ASN1OBJID* tmpOid = ConvertStringToASN1OBJID(str);
    //int len = encodeOID (&ctxt, m_impl->m_contentType->raw(), ASN1EXPL);
    int len = encodeOID (&ctxt, tmpOid, ASN1EXPL);
    if(tmpOid)
        delete tmpOid;
    OSOCTET* buf = getBEREncBufPtr(&ctxt);
    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);
    }
}

//content type specific functions
void CPKIFContentTypeAttribute::SetContentType(
    CPKIFOIDPtr& oid)
{
    LOG_STRING_DEBUG("CPKIFContentTypeAttribute::SetContentType(CPKIFOIDPtr& oid)", TOOLKIT_MESSAGE_ASN, 0, this);

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

    m_impl->m_contentType = oid;
}
CPKIFOIDPtr CPKIFContentTypeAttribute::GetContentType()
{
    return m_impl->m_contentType;
}