LdapAndOcspPanel.cpp

Go to the documentation of this file.

#if defined(__GNUG__) && !defined(NO_GCC_PRAGMA) && defined(PKIFRESOURCES_ALLOW_GCC_PRAGMA)
    #pragma implementation "LdapAndOcspPanel.cpp"
#endif

//Pick up PKIF windows SDK configuration macros instead of WX's 
#include "PKIFdll.h"
// For compilers that support precompilation, includes "wx/wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
    #pragma hdrstop
#endif

#include "LdapAndOcspPanel.h"
#include "DirectoryEntryDlg.h"
#include "OcspResponderEntryDlg.h"
#include "CSingletonLDAPConnection.h"
#include "HttpServerBlacklist.h"
#include "PRStringUtils.h"
#include "PKIFEnums.h"

using namespace boost;
using namespace std;
// WDR: class implementations

//----------------------------------------------------------------------------
// CPKIFLdapAndOcspPanel
//----------------------------------------------------------------------------

// WDR: event table for CPKIFLdapAndOcspPanel

BEGIN_EVENT_TABLE(CPKIFLdapAndOcspPanel,wxPanel)
    EVT_BUTTON( ID_BUTTON_ADD_DIR, CPKIFLdapAndOcspPanel::OnAddLdap )
    EVT_BUTTON( ID_BUTTON_EDIT_DIR, CPKIFLdapAndOcspPanel::OnEditLdap )
    EVT_BUTTON( ID_BUTTON_REMOVE_DIR, CPKIFLdapAndOcspPanel::OnRemoveLdap )
    EVT_BUTTON( ID_BUTTON_REMOVE_OCSP, CPKIFLdapAndOcspPanel::OnRemoveOcsp )
    EVT_BUTTON( ID_BUTTON_EDIT_OCSP, CPKIFLdapAndOcspPanel::OnEditOcsp )
    EVT_BUTTON( ID_BUTTON_ADD_OCSP, CPKIFLdapAndOcspPanel::OnAddOcsp )
    EVT_BUTTON( ID_BUTTON_ADD_DIR_BL, CPKIFLdapAndOcspPanel::OnBlacklistAdd )
    EVT_BUTTON( ID_BUTTON_REMOVE_DIR_BL, CPKIFLdapAndOcspPanel::OnBlacklistRemove )
    EVT_BUTTON( ID_BUTTON_CLEAR_DIR_BL, CPKIFLdapAndOcspPanel::OnBlacklistClear )
    EVT_BUTTON( ID_BUTTON_ADD_DIR_HTTP_BL, CPKIFLdapAndOcspPanel::OnHttpBlacklistAdd )
    EVT_BUTTON( ID_BUTTON_REMOVE_DIR_HTTP_BL, CPKIFLdapAndOcspPanel::OnHttpBlacklistRemove )
    EVT_BUTTON( ID_BUTTON_CLEAR_DIR_HTTP_BL, CPKIFLdapAndOcspPanel::OnHttpBlacklistClear )
END_EVENT_TABLE()
CPKIFLdapAndOcspPanel::CPKIFLdapAndOcspPanel( wxWindow *parent, wxWindowID id,
    const wxPoint &position, const wxSize& size, long style,bool enableBlacklists ) :
    wxPanel( parent, id, position, size, style )
{
    // WDR: dialog function LdapAndOcspPanel for CPKIFLdapAndOcspPanel
    LdapAndOcspPanel( this, TRUE ); 

    if(!enableBlacklists)
    {
        wxSizer *sz = GetSizer();

        sz->Show(GlobalBLsSizer, false);
        Layout();
    }
}
CPKIFLdapAndOcspPanel::~CPKIFLdapAndOcspPanel()
{
    m_vOcsp.clear();
}
bool CPKIFLdapAndOcspPanel::Validate()
{
    return TRUE;
}

// WDR: handler implementations for CPKIFLdapAndOcspPanel
void CPKIFLdapAndOcspPanel::OnBlacklistClear( wxCommandEvent &event )
{
    m_blacklist.clear();
    UpdateBlacklist();
}
void CPKIFLdapAndOcspPanel::OnBlacklistRemove( wxCommandEvent &event )
{
    int sel = GetListboxLdapBlacklist()->GetSelection();
    if(wxNOT_FOUND == sel) return;
    wxString serverName = GetListboxLdapBlacklist()->GetStringSelection();
    string item(serverName.mb_str());
    vector<string>::iterator found = find(m_blacklist.begin(),m_blacklist.end(),item);
    if(found != m_blacklist.end())
        m_blacklist.erase(found);
    UpdateBlacklist();
}
void CPKIFLdapAndOcspPanel::OnBlacklistAdd( wxCommandEvent &event )
{
    wxTextEntryDialog ted(GetParent(),wxT("Server Name"),wxT("Add LDAP server to blacklist"),
        wxT(""),wxOK|wxCANCEL);
    ted.CentreOnParent();
    if(wxID_OK != ted.ShowModal()) return;
    wxString serverName = ted.GetValue();
    if(serverName == wxT("")) return;
    string item(serverName.mb_str());
    vector<string>::iterator found = find(m_blacklist.begin(),m_blacklist.end(),item);
    if(found == m_blacklist.end())
        m_blacklist.push_back(item);
    UpdateBlacklist();
}
void CPKIFLdapAndOcspPanel::OnAddOcsp( wxCommandEvent &event )
{
    CPKIFOcspResponderEntryDlg ored(this, -1, wxT("Enter OCSP responder information"));
    ored.Centre();
    if(wxID_OK == ored.ShowModal())
    {
        CPKIFOCSPCheckerPtr ocspChecker(new CPKIFOCSPChecker(*ored.GetOCSPChecker()));
        m_vOcsp.push_back(ocspChecker);
        UpdateOcsp();
    }
}
void CPKIFLdapAndOcspPanel::OnEditOcsp( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxOcsp();
    wxASSERT(lb);
    int sel = lb->GetSelection();
    wxASSERT(sel < m_vOcsp.size());
    if(wxNOT_FOUND != sel)
    {
        CPKIFOcspResponderEntryDlg ored(this, -1, wxT("Enter OCSP responder information"));
        ored.Centre();

        CPKIFOCSPCheckerPtr ocsp = m_vOcsp[sel];
        ored.SetInitialOCSPChecker(ocsp);

        if(wxID_OK == ored.ShowModal())
        {
            m_vOcsp[sel] = ored.GetOCSPChecker();
            //delete ocsp;
            UpdateOcsp();
        }
    }
    else
    {
        wxMessageBox(wxT("You must select an entry to edit."));
    }
}
void CPKIFLdapAndOcspPanel::OnRemoveOcsp( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxOcsp();
    wxASSERT(lb);
    int sel = lb->GetSelection();
    if(wxNOT_FOUND != sel)
    {
        CPKIFOCSPCheckerPtr ocsp = m_vOcsp[sel];
        m_vOcsp.erase(m_vOcsp.begin()+sel);
        //delete ocsp;
        UpdateOcsp();
    }    
}
void CPKIFLdapAndOcspPanel::OnRemoveLdap( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxLdap();
    wxASSERT(lb);
    int sel = lb->GetSelection();
    if(wxNOT_FOUND != sel)
    {
        m_vLdap.erase(m_vLdap.begin()+sel);
        UpdateLdap();
    }
}
void CPKIFLdapAndOcspPanel::OnEditLdap( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxLdap();
    wxASSERT(lb);
    int sel = lb->GetSelection();
    wxASSERT(sel < m_vLdap.size());
    if(wxNOT_FOUND != sel)
    {
        CPKIFDirectoryEntryDlg ded(this, -1, wxT("Enter directory server information"));
        ded.Centre();

        ServerInformationPtr sip = m_vLdap[sel];
        ded.SetAddress(wxString(WXCASTCHARS(sip->m_hostOrIPAddress.c_str())));
        ded.SetPort(sip->m_port);
        ded.SetNamespaces(sip->m_namespaces);

        if(wxID_OK == ded.ShowModal())
        {
            wxString tmpAddr;
            ded.GetAddress(tmpAddr);
            sip->m_hostOrIPAddress = wxASCIIHelper(tmpAddr).to_chars();
            sip->m_port = ded.GetPort();
            sip->m_namespaces = ded.GetNamespaces();

            UpdateLdap();
        }
    }
    else
    {
        wxMessageBox(wxT("You must select an entry to edit."));
    }
}
void CPKIFLdapAndOcspPanel::OnAddLdap( wxCommandEvent &event )
{
    CPKIFDirectoryEntryDlg ded(this, -1, wxT("Enter directory server information"));
    ded.Centre();
    if(wxID_OK == ded.ShowModal())
    {
        ServerInformation* si = new ServerInformation();
        wxString tmpAddr;
        ded.GetAddress(tmpAddr);
        si->m_hostOrIPAddress = tmpAddr.mb_str();
        si->m_port = ded.GetPort();
        si->m_namespaces = ded.GetNamespaces();

        ServerInformationPtr sip(si);
        m_vLdap.push_back(sip);
        UpdateLdap();
    }
}
void CPKIFLdapAndOcspPanel::UpdateLdap()
{
    wxListBox* lb = GetListboxLdap();
    lb->Clear();

    vector<ServerInformationPtr>::iterator pos;
    vector<ServerInformationPtr>::iterator end = m_vLdap.end();
    for(pos = m_vLdap.begin(); pos != end; ++pos)
    {
        wxString tmp((*pos)->m_hostOrIPAddress.c_str(),wxConvUTF8);
        lb->InsertItems(1, &tmp, lb->GetCount());   
    }
}
void CPKIFLdapAndOcspPanel::UpdateOcsp()
{
    wxListBox* lb = GetListboxOcsp();
    lb->Clear();

    vector<CPKIFOCSPCheckerPtr>::iterator pos;
    vector<CPKIFOCSPCheckerPtr>::iterator end = m_vOcsp.end();
    for(pos = m_vOcsp.begin(); pos != end; ++pos)
    {
        wxString tmp((*pos)->GetURL()->c_str(),wxConvUTF8);
        lb->InsertItems(1, &tmp, lb->GetCount());   
    }
}
void CPKIFLdapAndOcspPanel::GetDirectories(
    vector<CPKIFLDAPRepositoryPtr>& d) const
{
    vector<ServerInformationPtr>::const_iterator pos;
    vector<ServerInformationPtr>::const_iterator end = m_vLdap.end();
    for(pos = m_vLdap.begin(); pos != end; ++pos)
    {
        CPKIFLDAPRepositoryPtr ldap(new CPKIFLDAPRepository);
        ldap->Set_Port((*pos)->m_port);
        ldap->SetHost((*pos)->m_hostOrIPAddress.c_str());

        CPKIFGeneralSubtreeList::iterator pos2;
        CPKIFGeneralSubtreeList::iterator end2 = (*pos)->m_namespaces.end();
        for(pos2 = (*pos)->m_namespaces.begin(); pos2 != end2; ++pos2)
        {
            ldap->AddNamespace((*pos2));
        }

        d.push_back(ldap);
    }
}
void CPKIFLdapAndOcspPanel::GetResponders(
    vector<CPKIFOCSPCheckerPtr>& r) const
{
    vector<CPKIFOCSPCheckerPtr>::const_iterator pos;
    vector<CPKIFOCSPCheckerPtr>::const_iterator end = m_vOcsp.end();
    for(pos = m_vOcsp.begin(); pos != end; ++pos)
    {
        CPKIFOCSPCheckerPtr newOcsp(new CPKIFOCSPChecker(*(*pos)));
        r.push_back(newOcsp);
    }
}
void CPKIFLdapAndOcspPanel::OnInitDialog(wxInitDialogEvent& event)
{
    if(NULL == m_m)
        return;

    {
        CPKIFCacheMediator2* sm = m_m->GetMediator<CPKIFCacheMediator2>();
        if(NULL == sm) return;

        std::vector<IPKIFColleaguePtr> c;
        sm->GetColleagues(c);

        CPKIFLDAPRepositoryPtr ldap;
        int ldapCount = 0;
        std::vector<IPKIFColleaguePtr>::iterator pos;
        std::vector<IPKIFColleaguePtr>::iterator end = c.end();
        for(pos = c.begin(); pos != end; ++pos, ++ldapCount)
        {
            if(NULL != dynamic_pointer_cast<CPKIFLDAPRepository, IPKIFColleague>(*pos))
            {
                ldap = dynamic_pointer_cast<CPKIFLDAPRepository, IPKIFColleague>(*pos);

                ServerInformationPtr newLdap(new ServerInformation);
                newLdap->m_hostOrIPAddress = ldap->GetHost();
                newLdap->m_port = ldap->GetPort();
                newLdap->m_namespaces = ldap->GetNamespaces();
                m_vLdap.push_back(newLdap);
            }
        }
    }

    {
        CPKIFRevocationStatusMediator2* rm = m_m->GetMediator<CPKIFRevocationStatusMediator2>();
        if(NULL == rm) return;

        std::vector<IPKIFColleaguePtr> c;
        rm->GetColleagues(c);

        CPKIFOCSPCheckerPtr ocsp;
        int ldapCount = 0;
        std::vector<IPKIFColleaguePtr>::iterator pos;
        std::vector<IPKIFColleaguePtr>::iterator end = c.end();
        for(pos = c.begin(); pos != end; ++pos, ++ldapCount)
        {
            if(NULL != dynamic_pointer_cast<CPKIFOCSPChecker, IPKIFColleague>(*pos))
            {
                ocsp = dynamic_pointer_cast<CPKIFOCSPChecker, IPKIFColleague>(*pos);

                if(ocsp->GetURL() != (std::string*)NULL)
                {
                    CPKIFOCSPCheckerPtr newOcsp(new CPKIFOCSPChecker(*ocsp));
                    m_vOcsp.push_back(newOcsp);
                }
            }
        }
    }

    PKIFLDAP::GetServerBlacklist(m_blacklist);
    PKIFHTTP::GetHttpServerBlacklist(m_httpBlackList);

    UpdateLdap();
    UpdateOcsp();
    UpdateBlacklist();
    UpdateHttpBlacklist();
}
void CPKIFLdapAndOcspPanel::SetInitialMediator(IPKIFMediatorPtr& m)
{
    m_m = m;
}
void CPKIFLdapAndOcspPanel::UpdateBlacklist(void)
{
    wxListBox * lb = GetListboxLdapBlacklist();
    lb->Clear();
    vector<string>::iterator end = m_blacklist.end();
    for(vector<string>::iterator i = m_blacklist.begin(); i != end; ++i)
    {
        wxString item((*i).c_str(),wxConvUTF8);
        lb->Append(item);
    }
}

void CPKIFLdapAndOcspPanel::UpdateHttpBlacklist(void)
{
    wxListBox * lb = GetListboHttpBlacklist();
    lb->Clear();
    vector<string>::iterator end = m_httpBlackList.end();
    for(vector<string>::iterator i = m_httpBlackList.begin(); i != end; ++i)
    {
        wxString item((*i).c_str(),wxConvUTF8);
        lb->Append(item);
    }
}
void CPKIFLdapAndOcspPanel::GetBlacklist(
    std::vector<std::string> & l) const
{
    l = m_blacklist;
}

void CPKIFLdapAndOcspPanel::GetHttpBlacklist(
    std::vector<std::string> & l) const
{
    l = m_httpBlackList;
}

void CPKIFLdapAndOcspPanel::OnHttpBlacklistClear( wxCommandEvent &event )
{
    m_httpBlackList.clear();
    UpdateHttpBlacklist();
}
void CPKIFLdapAndOcspPanel::OnHttpBlacklistRemove( wxCommandEvent &event )
{
    int sel = GetListboHttpBlacklist()->GetSelection();
    if(wxNOT_FOUND == sel) return;
    wxString serverName = GetListboHttpBlacklist()->GetStringSelection();
    string item(serverName.mb_str());
    vector<string>::iterator found = find(m_httpBlackList.begin(),m_httpBlackList.end(),item);
    if(found != m_httpBlackList.end())
        m_httpBlackList.erase(found);
    UpdateHttpBlacklist();
}
void CPKIFLdapAndOcspPanel::OnHttpBlacklistAdd( wxCommandEvent &event )
{
    wxTextEntryDialog ted(GetParent(),wxT("Server Name"),wxT("Add HTTP server to blacklist"),
        wxT(""),wxOK|wxCANCEL);
    ted.CentreOnParent();
    if(wxID_OK != ted.ShowModal()) return;
    wxString serverName = ted.GetValue();
    if(serverName == wxT("")) return;
    string item(serverName.mb_str());
    vector<string>::iterator found = find(m_httpBlackList.begin(),m_httpBlackList.end(),item);
    if(found == m_httpBlackList.end())
        m_httpBlackList.push_back(item);
    UpdateHttpBlacklist();
}