SimpleCertAndCrlPanel.cpp

Go to the documentation of this file.

#if defined(__GNUG__) && !defined(NO_GCC_PRAGMA) && defined(PKIFRESOURCES_ALLOW_GCC_PRAGMA)
    #pragma implementation "SimpleCertAndCrlPanel.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

#if defined(WIN32) || defined(_WIN32)
    #include <shellapi.h>
#endif

#include "SimpleCertAndCrlPanel.h"
#include "CertificateDisplayNotebook.h"
#include "CRLDisplayNotebook.h"
#include "ResourceUtils.h"

#include <iterator>
using namespace boost;
using namespace std;

// WDR: class implementations

//----------------------------------------------------------------------------
// CPKIFSimpleCertAndCrlPanel
//----------------------------------------------------------------------------

// WDR: event table for CPKIFSimpleCertAndCrlPanel

BEGIN_EVENT_TABLE(CPKIFSimpleCertAndCrlPanel,wxPanel)
    EVT_BUTTON( ID_BUTTON_ADD_TA, CPKIFSimpleCertAndCrlPanel::OnAddTA )
    EVT_BUTTON( ID_BUTTON_VIEW_TA, CPKIFSimpleCertAndCrlPanel::OnViewTA )
    EVT_BUTTON( ID_BUTTON_REMOVE_TA, CPKIFSimpleCertAndCrlPanel::OnRemoveTA )
    EVT_BUTTON( ID_BUTTON_ADD_CRL, CPKIFSimpleCertAndCrlPanel::OnAddCrl )
    EVT_BUTTON( ID_BUTTON_VIEW_CRL, CPKIFSimpleCertAndCrlPanel::OnViewCrl )
    EVT_BUTTON( ID_BUTTON_REMOVE_CRL, CPKIFSimpleCertAndCrlPanel::OnRemoveCrl )
    EVT_BUTTON( ID_BUTTON_ADD_CERT, CPKIFSimpleCertAndCrlPanel::OnAddCert )
    EVT_BUTTON( ID_BUTTON_VIEW_CERT, CPKIFSimpleCertAndCrlPanel::OnViewCert )
    EVT_BUTTON( ID_BUTTON_REMOVE_CERT, CPKIFSimpleCertAndCrlPanel::OnRemoveCert )
END_EVENT_TABLE()
CPKIFSimpleCertAndCrlPanel::CPKIFSimpleCertAndCrlPanel( wxWindow *parent, wxWindowID id,
    const wxPoint &position, const wxSize& size, long style ) :
    wxPanel( parent, id, position, size, style )
{
#ifdef WIN32
    memset(m_tmpName, 0, MAX_PATH);
#endif

    // WDR: dialog function SimpleCertAndCrlPanel for CPKIFSimpleCertAndCrlPanel
    SimpleCertAndCrlPanel( this, TRUE ); 
//
//#ifndef WIN32
//  GetButtonViewCrl()->Enable(false);
//  GetButtonViewCert()->Enable(false);
//  GetButtonViewTa()->Enable(false);
//#endif 

    wxListBox* lb1 = GetListboxCerts();
    wxASSERT(lb1);
    if(lb1->GetCount() == 0)
    {
        GetButtonViewCert()->Enable(false);
    }

    wxListBox* lb2 = GetListboxCrls();
    wxASSERT(lb2);

    if(lb2->GetCount() == 0)
    {
        GetButtonViewCrl()->Enable(false);
    }

    wxListBox* lb3 = GetListboxTrustAnchors();
    wxASSERT(lb3);

    if(lb3->GetCount() == 0)
    {
        GetButtonViewTa()->Enable(false);
    }
}
CPKIFSimpleCertAndCrlPanel::~CPKIFSimpleCertAndCrlPanel()
{
    #ifdef WIN32
    if(0 != strlen(m_tmpName))
        remove(m_tmpName);
    #endif
}
bool CPKIFSimpleCertAndCrlPanel::Validate()
{
    return TRUE;
}

// WDR: handler implementations for CPKIFSimpleCertAndCrlPanel
void CPKIFSimpleCertAndCrlPanel::OnRemoveCert( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxCerts();
    wxASSERT(lb);
    wxArrayInt selections;

    int count = lb->GetSelections(selections);

    for(int jj = 0; jj < count; jj++)
    {
        int sel = selections.Item(jj);
        vector<CPKIFCertificatePtr>::iterator i;
        vector<CPKIFCertificatePtr>::iterator end = m_vCerts.end();
        for(i = m_vCerts.begin(); i != end; i++) 
        {
 
            wxString  s = lb->GetString(sel);
            if(wxString((*i)->Subject()->ToString(),wxConvUTF8) == s)
            {
                m_vCerts.erase(i);
                break;
            }
        }
    }
    UpdateCerts();      
}
void CPKIFSimpleCertAndCrlPanel::OnViewCert( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxCerts();
    wxASSERT(lb);
    wxArrayInt selections;

    int count = lb->GetSelections(selections);

    if(count == 0)
    {
        wxMessageBox("You must select a certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }
    if(count > 1)
    {
        wxMessageBox("You must only select one certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }

    int sel = selections.Item(0);
    CPKIFCertificatePtr curCertDetails;
    if(-1 != sel && sel < m_vCerts.size())
        curCertDetails = m_vCerts[sel];
    else
    {
        wxMessageBox("You must make selection before viewing the certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }

    #if defined(WIN32) || defined(_WIN32)
        char tmppath[MAX_PATH];
        GetTempPath(MAX_PATH, tmppath);

        //if we've already done this - remove the previously created file
        if(0 != strlen(m_tmpName))
            remove(m_tmpName);

        GetTempFileName(tmppath, "pkif_", 0, m_tmpName);
        
        //we just wanted a unique file name - not a file
        remove(m_tmpName);

        //we want .cer not .tmp
        char* p = m_tmpName + (strlen(m_tmpName) - 3);
        strcpy(p, "cer");

        CPKIFBufferPtr rawCert = curCertDetails->Encoded();
        FILE* f = fopen(m_tmpName, "wb+");
        fwrite(rawCert->GetBuffer(), rawCert->GetLength(), 1, f);
        fclose(f);

        HINSTANCE h = ShellExecute(NULL, "open", m_tmpName, NULL, NULL, SW_SHOWNORMAL);
        if((int)h <= 32)
        {
            wxMessageBox("There is no default certificate viewer installed.  Unable to display additional information.", _T("Error"), wxICON_ERROR);
            return;
        }
    #else
        CPKIFCertificateDisplayNotebook cdn(NULL,-1,wxT("Certificate"),wxDefaultPosition,wxDefaultSize,
            wxDEFAULT_DIALOG_STYLE);
        cdn.SetCertificate(curCertDetails);
        cdn.Center();
        cdn.ShowModal();
    #endif
}
void CPKIFSimpleCertAndCrlPanel::OnAddCert( wxCommandEvent &event )
{
    static wxString filter(wxT("Certificate files (*.cer, *.der, *.crt)|*.cer;*.der;*.crt|All files (*.*)|*.*"));
    wxFileDialog dlg(this, wxString(wxT("Select one or more certiticates")), wxEmptyString, wxEmptyString, filter, wxOPEN | wxFILE_MUST_EXIST | wxMULTIPLE);
    if(wxID_OK == dlg.ShowModal())
    {
        wxArrayString filenames;
        dlg.GetPaths(filenames);

        wxArrayString::iterator pos;
        wxArrayString::iterator end = filenames.end();
        for(pos = filenames.begin(); pos != end; ++pos)
        {
            CPKIFCertificatePtr cert = GetCert(*pos);

            if(cert != (CPKIFCertificate*)NULL)
            {
                if(!cert->IsSelfSigned())
                    m_vCerts.push_back(cert);
                else
                    wxMessageBox("Self-signed certificates cannot be added to this store.", "ERROR", wxICON_ERROR);
            }
        }           
    }

    UpdateCerts();
}
void CPKIFSimpleCertAndCrlPanel::OnRemoveCrl( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxCrls();
    wxASSERT(lb);
    wxArrayInt selections;

    int count = lb->GetSelections(selections);

    for(int jj = 0; jj < count; jj++)
    {
        int sel = selections.Item(jj);
        vector<CPKIFCRLPtr>::iterator i;
        vector<CPKIFCRLPtr>::iterator end = m_vCrls.end();
        for(i = m_vCrls.begin(); i != end; i++) 
        {
 
            wxString  s = lb->GetString(sel);
            if(wxString((*i)->Issuer()->ToString(),wxConvUTF8) == s)
            {
                m_vCrls.erase(i);
                break;
            }
        }
    }
    UpdateCrls();   
}
void CPKIFSimpleCertAndCrlPanel::OnViewCrl( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxCrls();
    wxASSERT(lb);
    wxArrayInt selections;

    int count = lb->GetSelections(selections);

    if(count == 0)
    {
        wxMessageBox("You must select a CRL.", _T("Warning"), wxICON_WARNING);
        return;
    }

    if(count > 1)
    {
        wxMessageBox("You must only select one CRL.", _T("Warning"), wxICON_WARNING);
        return;
    }

    int sel = selections.Item(0);
    CPKIFCRLPtr curCrlDetails;
    if(-1 != sel && sel < m_vCrls.size())
        curCrlDetails = m_vCrls[sel];
    else
    {
        wxMessageBox("You must make selection before viewing the CRL.", _T("Warning"), wxICON_WARNING);
        return;
    }

    #if defined(WIN32) || defined(_WIN32)
        char tmppath[MAX_PATH];
        GetTempPath(MAX_PATH, tmppath);

        //if we've already done this - remove the previously created file
        if(0 != strlen(m_tmpName))
            remove(m_tmpName);

        GetTempFileName(tmppath, "pkif_", 0, m_tmpName);
        
        //we just wanted a unique file name - not a file
        remove(m_tmpName);

        //we want .cer not .tmp
        char* p = m_tmpName + (strlen(m_tmpName) - 3);
        strcpy(p, "crl");

        CPKIFBufferPtr rawCert = curCrlDetails->Encoded();
        FILE* f = fopen(m_tmpName, "wb+");
        fwrite(rawCert->GetBuffer(), rawCert->GetLength(), 1, f);
        fclose(f);

        HINSTANCE h = ShellExecute(NULL, "open", m_tmpName, NULL, NULL, SW_SHOWNORMAL);
        if((int)h <= 32)
        {
            wxMessageBox("There is no default CRL viewer installed.  Unable to display additional information.", _T("Error"), wxICON_ERROR);
            return;
        }
    #else
        CPKIFCRLDisplayNotebook cdn(NULL,-1,wxT("Certificate Revocation List"),wxDefaultPosition,wxDefaultSize,
        wxDEFAULT_DIALOG_STYLE);
        cdn.SetCRL(curCrlDetails);
        cdn.Center();
        cdn.ShowModal();
    #endif
}
void CPKIFSimpleCertAndCrlPanel::OnAddCrl( wxCommandEvent &event )
{
    static wxString filter(wxT("CRL files (*.crl)|*.crl|All files (*.*)|*.*"));
    wxFileDialog dlg(this, wxString(wxT("Select one or more CRLs")), wxEmptyString, wxEmptyString, filter, wxOPEN | wxFILE_MUST_EXIST | wxMULTIPLE);
    if(wxID_OK == dlg.ShowModal())
    {
        wxArrayString filenames;
        dlg.GetPaths(filenames);
        
        wxArrayString::iterator pos;
        wxArrayString::iterator end = filenames.end();
        for(pos = filenames.begin(); pos != end; ++pos)
        {
            CPKIFCRLPtr crl = GetCrl(*pos);
            if(crl != (CPKIFCRL*)NULL)
            m_vCrls.push_back(crl);
        }           
    }

    UpdateCrls();
}
void CPKIFSimpleCertAndCrlPanel::OnRemoveTA( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxTrustAnchors();
    wxASSERT(lb);
    wxArrayInt selections;

    int count = lb->GetSelections(selections);

    for(int jj = 0; jj < count; jj++)
    {
        int sel = selections.Item(jj);
        vector<CPKIFCertificatePtr>::iterator i;
        vector<CPKIFCertificatePtr>::iterator end = m_vTrustAnchors.end();
        for(i = m_vTrustAnchors.begin(); i != end; i++) 
        {
 
            wxString  s = lb->GetString(sel);
            if(wxString((*i)->Subject()->ToString(),wxConvUTF8) == s)
            {
                m_vTrustAnchors.erase(i);
                break;
            }
        }
    }
    UpdateTrustAnchors();
}
void CPKIFSimpleCertAndCrlPanel::OnViewTA( wxCommandEvent &event )
{
    wxListBox* lb = GetListboxTrustAnchors();
    wxASSERT(lb);
    wxArrayInt selections;

    int count = lb->GetSelections(selections);

    if(count == 0)
    {
        wxMessageBox("You must select a certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }

    if(count > 1)
    {
        wxMessageBox("You must only select one certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }

    int sel = selections.Item(0);
    CPKIFCertificatePtr curCertDetails;
    if(-1 != sel && sel < m_vTrustAnchors.size())
        curCertDetails = m_vTrustAnchors[sel];
    else
    {
        wxMessageBox("You must make selection before viewing the certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }

    #if defined(WIN32) || defined(_WIN32)
        char tmppath[MAX_PATH];
        GetTempPath(MAX_PATH, tmppath);

        //if we've already done this - remove the previously created file
        if(0 != strlen(m_tmpName))
            remove(m_tmpName);

        GetTempFileName(tmppath, "pkif_", 0, m_tmpName);
        
        //we just wanted a unique file name - not a file
        remove(m_tmpName);

        //we want .cer not .tmp
        char* p = m_tmpName + (strlen(m_tmpName) - 3);
        strcpy(p, "cer");

        
        CPKIFBufferPtr rawCert = curCertDetails->Encoded();
        FILE* f = fopen(m_tmpName, "wb+");
        fwrite(rawCert->GetBuffer(), rawCert->GetLength(), 1, f);
        fclose(f);

        HINSTANCE h = ShellExecute(NULL, "open", m_tmpName, NULL, NULL, SW_SHOWNORMAL);
        if((int)h <= 32)
        {
            wxMessageBox("There is no default certificate viewer installed.  Unable to display additional information.", _T("Error"), wxICON_ERROR);
            return;
        }
    #else
        CPKIFCertificateDisplayNotebook cdn(NULL,-1,wxT("Certificate"),wxDefaultPosition,wxDefaultSize,
        wxDEFAULT_DIALOG_STYLE);
        cdn.SetCertificate(curCertDetails);
        cdn.Center();
        cdn.ShowModal();
    #endif
}
void CPKIFSimpleCertAndCrlPanel::OnAddTA( wxCommandEvent &event )
{
    static wxString filter(wxT("Certificate files (*.cer, *.der, *.crt)|*.cer;*.der;*.crt|All files (*.*)|*.*"));
    wxFileDialog dlg(this, wxString(wxT("Select one or more trust anchors")), wxEmptyString, wxEmptyString, filter, wxOPEN | wxFILE_MUST_EXIST | wxMULTIPLE);
    if(wxID_OK == dlg.ShowModal())
    {
        wxArrayString filenames;
        dlg.GetPaths(filenames);
        
        wxArrayString::iterator pos;
        wxArrayString::iterator end = filenames.end();
        for(pos = filenames.begin(); pos != end; ++pos)
        {
            CPKIFCertificatePtr cert = GetCert(*pos);
            if(cert != (CPKIFCertificate*)NULL)
                m_vTrustAnchors.push_back(cert);
        }           
    }

    UpdateTrustAnchors();
}
void CPKIFSimpleCertAndCrlPanel::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);

    SimpleCertCachePtr simpleCert;
    SimpleCRLCachePtr simpleCRL;
    SimpleRootStorePtr simpleRoot;

    std::vector<IPKIFColleaguePtr>::iterator pos;
    std::vector<IPKIFColleaguePtr>::iterator end = c.end();
    for(pos = c.begin(); pos != end; ++pos)
    {
        if(NULL != dynamic_pointer_cast<SimpleRootStore, IPKIFColleague>(*pos))
        {
            simpleRoot = dynamic_pointer_cast<SimpleRootStore, IPKIFColleague>(*pos);
        }
        else if(NULL != dynamic_pointer_cast<SimpleCertCache, IPKIFColleague>(*pos))
        {
            simpleCert = dynamic_pointer_cast<SimpleCertCache, IPKIFColleague>(*pos);
        }
        else if(NULL != dynamic_pointer_cast<SimpleCRLCache, IPKIFColleague>(*pos))
        {
            simpleCRL = dynamic_pointer_cast<SimpleCRLCache, IPKIFColleague>(*pos);
        }
    }

    if(simpleCert)
    {
        std::vector<CPKIFCertificatePtr> certs;
        simpleCert->GetCertificates(certs);

        int certCount = 0;
        std::vector<CPKIFCertificatePtr>::iterator pos;
        std::vector<CPKIFCertificatePtr>::iterator end = certs.end();
        for(pos = certs.begin(); pos != end; ++pos, ++certCount)
        {
            m_vCerts.push_back(*pos);
        }
    }
    if(simpleCRL)
    {
        std::vector<CPKIFCRLPtr> crls;
        simpleCRL->GetCRLs(crls);

        int crlCount = 0;
        std::vector<CPKIFCRLPtr>::iterator pos;
        std::vector<CPKIFCRLPtr>::iterator end = crls.end();
        for(pos = crls.begin(); pos != end; ++pos, ++crlCount)
        {
            m_vCrls.push_back(*pos);
        }
    }
    if(simpleRoot)
    {
        IPKIFTrustAnchorList roots;
        simpleRoot->GetTrustRoots(roots);

        int rootCount = 0;
        IPKIFTrustAnchorList::iterator pos;
        IPKIFTrustAnchorList::iterator end = roots.end();
        for(pos = roots.begin(); pos != end; ++pos, ++rootCount)
        {
            //XXX***TA TEMP WORKAROUND
            CPKIFTrustRoot* ta = dynamic_cast<CPKIFTrustRoot*>(&(*(*pos)));
            if(ta != NULL)
            {
                CPKIFCertificatePtr cert;
                ta->GetCert(cert);
                m_vTrustAnchors.push_back(cert);
            }
        }
    }

    UpdateCerts();
    UpdateTrustAnchors();
    UpdateCrls();
}
void CPKIFSimpleCertAndCrlPanel::SetInitialMediator(
    IPKIFMediatorPtr& m)
{
    m_m = m;
}
void CPKIFSimpleCertAndCrlPanel::UpdateCerts()
{
    wxListBox* lb = GetListboxCerts();
    wxASSERT(lb);
    lb->Clear();

    std::vector<CPKIFCertificatePtr>::iterator pos;
    std::vector<CPKIFCertificatePtr>::iterator end = m_vCerts.end();
    for(pos = m_vCerts.begin(); pos != end; ++pos)
    {
        CPKIFNamePtr subName = (*pos)->Subject();
        wxString wxSubName(subName->ToString(),wxConvUTF8);
        lb->InsertItems(1, &wxSubName, lb->GetCount());
    }

    if(lb->GetCount() > 0)
    {
        GetButtonViewCert()->Enable(true);
    }
    else
    {
        GetButtonViewCert()->Enable(false);
    }
}
void CPKIFSimpleCertAndCrlPanel::UpdateCrls()
{
    wxListBox* lb = GetListboxCrls();
    wxASSERT(lb);
    lb->Clear();

    std::vector<CPKIFCRLPtr>::iterator pos;
    std::vector<CPKIFCRLPtr>::iterator end = m_vCrls.end();
    for(pos = m_vCrls.begin(); pos != end; ++pos)
    {
        CPKIFNamePtr issName = (*pos)->Issuer();
        wxString wxIssName(issName->ToString(),wxConvUTF8);
        lb->InsertItems(1, &wxIssName, lb->GetCount());
    }

    if(lb->GetCount() > 0)
    {
        GetButtonViewCrl()->Enable(true);
    }
    else
    {
        GetButtonViewCrl()->Enable(false);
    }
}
void CPKIFSimpleCertAndCrlPanel::UpdateTrustAnchors()
{
    wxListBox* lb = GetListboxTrustAnchors();
    wxASSERT(lb);
    lb->Clear();

    std::vector<CPKIFCertificatePtr>::iterator pos;
    std::vector<CPKIFCertificatePtr>::iterator end = m_vTrustAnchors.end();
    for(pos = m_vTrustAnchors.begin(); pos != end; ++pos)
    {
        CPKIFNamePtr subName = (*pos)->Subject();
        wxString wxSubName(subName->ToString(),wxConvUTF8);
        lb->InsertItems(1, &wxSubName, lb->GetCount());
    }

    if(lb->GetCount() > 0)
    {
        GetButtonViewTa()->Enable(true);
    }
    else
    {
        GetButtonViewTa()->Enable(false);
    }
}
void CPKIFSimpleCertAndCrlPanel::GetTrustAnchors(
    std::vector<CPKIFCertificatePtr>& v)
{
    copy(m_vTrustAnchors.begin(), m_vTrustAnchors.end(), back_inserter(v));
}
void CPKIFSimpleCertAndCrlPanel::GetCertificates(
    std::vector<CPKIFCertificatePtr>& v)
{
    copy(m_vCerts.begin(), m_vCerts.end(), back_inserter(v));
}
void CPKIFSimpleCertAndCrlPanel::GetCrls(
    std::vector<CPKIFCRLPtr>& v)
{
    copy(m_vCrls.begin(), m_vCrls.end(), back_inserter(v));
}