PKIFLdapCertNode.cpp

Go to the documentation of this file.

#include "PKIFLdapCertNode.h"

#include "Certificate.h"
#include "GottaMatch.h"
#include "LDAP_URL_Header.h"
#include "Name.h"
#include "PKIFLDAPRepository.h"
#include "ToolkitUtils.h"
#include "GottaMatch.h"

#include <iterator>
using namespace std;

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

    CPKIFCertificateNodeList m_certNodeList;
};

CPKIFLdapCertNode::CPKIFLdapCertNode(
    CPKIFLDAPRepositoryPtr& ldap,
    CPKIFNamePtr& entry, 
    std::vector<std::string>& attributes) 
    : m_impl(new CPKIFLdapCertNodeImpl)
{
    m_impl->m_ldap = ldap;
    m_impl->m_entry = entry;
    copy(attributes.begin(), attributes.end(), back_inserter(m_impl->m_attributes));
}
CPKIFLdapCertNode::CPKIFLdapCertNode()
    : m_impl(new CPKIFLdapCertNodeImpl)
{
}
CPKIFLdapCertNode::~CPKIFLdapCertNode()
{
    if(m_impl) delete m_impl;
}

bool CPKIFLdapCertNode::operator==(const CPKIFLdapCertNode& rhs) const
{
    if(!m_impl->m_entry || !rhs.m_impl->m_entry || //return false if either entry is NULL
        !(*m_impl->m_entry == *rhs.m_impl->m_entry)) //return false if entries do not match
        return false;

    if(!m_impl->m_ldap || !rhs.m_impl->m_ldap ||    //return false if with ldap object is NULL
        (0 != stricmp(m_impl->m_ldap->GetHost(), rhs.m_impl->m_ldap->GetHost())) || //return false if hosts don't match
        (m_impl->m_ldap->GetPort() != rhs.m_impl->m_ldap->GetPort()))   //return false if ports don't match
        return false;

    return true;
}

void CPKIFLdapCertNode::GetCertificates(
    CPKIFCertificateNodeList& certNodeList,
    PathBuildingDirection pbd)
{
    if(m_impl->m_entry)
    {
        m_impl->m_ldap->GetCertificates(m_impl->m_entry, m_impl->m_attributes, m_impl->m_certNodeList, pbd);
    }
    else if(m_impl->m_certNodeList.empty())
    {
        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"))
            {
                GetCertfromLDAPURL((*pos).c_str(), m_impl->m_certNodeList, pbd);
            }
        }
    }

    if(!m_impl->m_certNodeList.empty())
    {
        CPKIFCertificateNodeList::iterator pos;
        CPKIFCertificateNodeList::iterator end = m_impl->m_certNodeList.end();
        for(pos = m_impl->m_certNodeList.begin(); pos != end; ++pos)
        {
            GottaMatch<CPKIFCertificateNodeEntryPtr> gm;
            gm.SetRHS(*pos);
            if(certNodeList.end() == find_if(certNodeList.begin(), certNodeList.end(), gm))
            {
                certNodeList.push_back(*pos);
            }
        }
    }
}