PKIFLdapCrlNode.cpp

Go to the documentation of this file.

#include "PKIFLdapCrlNode.h"
#include "CRL.h"
#include "Name.h"
#include "PKIFLDAPRepository.h"
#include "LDAP_URL_Header.h"
#include "ToolkitUtils.h"
#include "GottaMatch.h"

#include <iterator>

using namespace std;

struct CPKIFLdapCrlNodeImpl
{
    CPKIFNamePtr m_entry;
    CPKIFLDAPRepositoryPtr m_ldap;
    vector<string> m_attributes;

    //CPKIFCRLPtr m_crl;
    CPKIFCRLNodeList m_cnl;
};

CPKIFLdapCrlNode::CPKIFLdapCrlNode(
    CPKIFLDAPRepositoryPtr& ldap,
    CPKIFNamePtr& entry,
    std::vector<std::string>& attributes) 
    : m_impl(new CPKIFLdapCrlNodeImpl)
{
    m_impl->m_ldap = ldap;
    m_impl->m_entry = entry;
    copy(attributes.begin(), attributes.end(), back_inserter(m_impl->m_attributes));
}
CPKIFLdapCrlNode::CPKIFLdapCrlNode()
    : m_impl(new CPKIFLdapCrlNodeImpl)
{
}
CPKIFLdapCrlNode::~CPKIFLdapCrlNode()
{
    if(m_impl) delete m_impl;
}
/*
CPKIFCRLPtr CPKIFLdapCrlNode::GetCRL()const
{
    if(m_impl->m_crl != (CPKIFCRL*)NULL)
        return m_impl->m_crl;
    else if(PAS_UNAVAILABLE == GetState())
    {
        CPKIFCRLPtr empty;
        return empty;
    }
    else
    {
        CPKIFCRLNodeList cnl;
        if(m_impl->m_entry)
        {
            m_impl->m_ldap->GetCRLs(m_impl->m_entry, m_impl->m_attributes, cnl);
            if(cnl.empty())
            {
                SetState(PAS_UNAVAILABLE);
                CPKIFCRLPtr empty;
                return empty;
            }
            else
            {
                m_impl->m_crl = cnl.front()->GetCRL();
                SetState(PAS_AVAILABLE);
                return m_impl->m_crl;
            }
        }
        else
        {
            vector<string> sources;
            GetSources(sources);
            vector<string>::iterator pos;
            vector<string>::iterator end = sources.end();
            for(pos = sources.begin(); pos != end; ++pos)
            {
                if(0 == (*pos).find("ldap"))
                {
                    CPKIFCRLList crlList;
                    GetCRLfromLDAPURL((*pos).c_str(), crlList);
                    if(!crlList.empty())
                    {
                        m_impl->m_crl = crlList.front();
                        SetState(PAS_AVAILABLE);
                        return m_impl->m_crl;
                    }
                }
            }

            return m_impl->m_crl; //no sources, return NULL
        }
    }
}
*/
void CPKIFLdapCrlNode::GetCrls(CPKIFCRLNodeList& crlNodeList)
{
    if(!m_impl->m_cnl.empty())
    {
        copy(m_impl->m_cnl.begin(), m_impl->m_cnl.end(), back_inserter(crlNodeList));
        return;
    }
    else if(PAS_UNAVAILABLE == GetState())
    {
        return;
    }
    else
    {
        if(m_impl->m_entry)
        {
            m_impl->m_ldap->GetCRLs(m_impl->m_entry, m_impl->m_attributes, m_impl->m_cnl);
            if(m_impl->m_cnl.empty())
            {
                SetState(PAS_UNAVAILABLE);
                return;
            }
            else
            {
                copy(m_impl->m_cnl.begin(), m_impl->m_cnl.end(), back_inserter(crlNodeList));
                SetState(PAS_AVAILABLE);
                return;
            }
        }
        else
        {
            bool atLeastOneWorked = false;
            vector<string> sources;
            GetSources(sources);
            vector<string>::iterator pos;
            vector<string>::iterator end = sources.end();
            for(pos = sources.begin(); pos != end; ++pos)
            {
                if(0 == (*pos).find("ldap"))
                {
                    CPKIFCRLList crlList;
                    GetCRLfromLDAPURL((*pos).c_str(), crlList);
                    if(!crlList.empty())
                    {
                        atLeastOneWorked = true;
                        CPKIFCRLList::iterator listPos;
                        CPKIFCRLList::iterator listEnd = crlList.end();
                        for(listPos = crlList.begin(); listPos != listEnd; ++listPos)
                        {
                            CPKIFCRLNodeEntryPtr newNode(new CPKIFCRLNodeEntry);
                            newNode->SetCRL(*listPos);
                            newNode->AddSource(*pos);
                            newNode->SetState(PAS_AVAILABLE);
                            GottaMatch<CPKIFCRLNodeEntryPtr> gm;
                            gm.SetRHS(newNode);
                            if(m_impl->m_cnl.end() == find_if(m_impl->m_cnl.begin(), m_impl->m_cnl.end(), gm))
                                m_impl->m_cnl.push_back(newNode);
                        }

                        if(!m_impl->m_cnl.empty())
                        {
                            copy(m_impl->m_cnl.begin(), m_impl->m_cnl.end(), back_inserter(crlNodeList));
                            SetState(PAS_AVAILABLE);
                        }
                    }
                }
            }

            if(!atLeastOneWorked)
            {
                SetState(PAS_UNAVAILABLE);
            }

            return; 
        }
    }
}