IPKIFHasExtensions.h

Go to the documentation of this file.

#ifndef __HasExtensions_H__
#define __HasExtensions_H__

#include "PKIFdll.h"
#include <vector>

class IPKIFParseExtensions;
FD_SMART_PTR(CPKIFBuffer);
FD_SMART_PTR(CPKIFX509Extension);
FD_SMART_PTR(CPKIFOID);

class CAC_API CAC_NO_VTABLE IPKIFHasExtensions
{
public:
    virtual ~IPKIFHasExtensions ();

    template <typename T> boost::shared_ptr<T> GetExtension();
    virtual void GetExtensions (IPKIFParseExtensions* m, std::vector<CPKIFX509ExtensionPtr>& exts);
    
    virtual void GetExtensionByOID(const CPKIFOID& oid, CPKIFX509ExtensionPtr& ref) = 0;
    virtual void GetEncodedExtensions (CPKIFBufferPtr& buf) = 0;
    
};
DECLARE_SMART_POINTERS(IPKIFHasExtensions);

// needed for the template function
#include "X509Extension.h"
#include "OID.h"
#include "ASN1Errors.h"
#include "PKIFException.h"


template <typename T> 
boost::shared_ptr<T> IPKIFHasExtensions::GetExtension()
{
    CPKIFX509ExtensionPtr ext;
    CPKIFStringPtr oidStr(new std::string(T::extOID));
    CPKIFOID oid(oidStr);
    GetExtensionByOID(oid,ext);
    
    if(ext && 0 != dynamic_cast<T*>(ext.get())) {
        return *((boost::shared_ptr<T>*)&(ext));
    }
    else if(ext)
    {
        //if we get here then an extension failed to parse
        throw CPKIFException(TOOLKIT_ASN, ASN1_DECODE_ERROR);
    }
    boost::shared_ptr<T> nullExt;
    return nullExt;
}

#endif