CredentialSelector.cpp

Go to the documentation of this file.

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

using namespace std;

// WDR: class implementations

//----------------------------------------------------------------------------
// CPKIFCredentialSelectorDlg
//----------------------------------------------------------------------------

// WDR: event table for CPKIFCredentialSelectorDlg

BEGIN_EVENT_TABLE(CPKIFCredentialSelectorDlg,wxDialog)
    EVT_BUTTON( wxID_OK, CPKIFCredentialSelectorDlg::OnOK )
    EVT_BUTTON( ID_MOINFO, CPKIFCredentialSelectorDlg::OnViewCertificate )
    EVT_BUTTON( wxID_CANCEL, CPKIFCredentialSelectorDlg::OnClose )
    EVT_INIT_DIALOG( CPKIFCredentialSelectorDlg::OnInitDialog )
    EVT_LISTBOX_DCLICK( ID_LIST_OF_CREDENTIALS, CPKIFCredentialSelectorDlg::OnCredListDoubleClick )
END_EVENT_TABLE()

CPKIFCredentialSelectorDlg::CPKIFCredentialSelectorDlg( wxWindow *parent, wxWindowID id, const wxString &title,
    const wxPoint &position, const wxSize& size, long style ) :
    wxDialog( parent, id, title, position, size, style )
{
#ifdef WIN32
    memset(m_tmpName, 0, MAX_PATH);
#endif

    m_ku = PKIFCRYPTO::DigitalSignature | PKIFCRYPTO::NonRepudiation | PKIFCRYPTO::KeyEncipherment;

    // WDR: dialog function CredentialSelectorDlg for CPKIFCredentialSelectorDlg
    CredentialSelectorDlg( this, TRUE ); 
}

CPKIFCredentialSelectorDlg::~CPKIFCredentialSelectorDlg()
{
#ifdef WIN32
    if(0 != strlen(m_tmpName))
        remove(m_tmpName);
#endif

}

// WDR: handler implementations for CPKIFCredentialSelectorDlg

void CPKIFCredentialSelectorDlg::OnCredListDoubleClick( wxCommandEvent &event )
{
    OnOK(event);
}
void CPKIFCredentialSelectorDlg::OnClose( wxCommandEvent &event )
{
    event.Skip();
}

void CPKIFCredentialSelectorDlg::OnViewCertificate( wxCommandEvent &event )
{
#ifdef WIN32
    wxListBox* credList = GetListOfCredentials();
    wxASSERT(credList);
    CPKIFCredentialPtr curCred;
    size_t sel = credList->GetSelection();
    if(-1 != sel && sel < m_creds.size())
        curCred = m_creds[sel];
    else
    {
        wxMessageBox("You must make selection before viewing the certificate.", _T("Warning"), wxICON_WARNING);
        return;
    }

    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");

    CPKIFCertificatePtr cert = curCred->GetCertificate();
    CPKIFBufferPtr rawCert = cert->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;
    }
#endif    
}

void CPKIFCredentialSelectorDlg::OnOK( wxCommandEvent &event )
{
    wxListBox* credList = GetListOfCredentials();
    wxASSERT(credList);

    long sel = credList->GetSelection();
    if(-1 != sel && static_cast<size_t>(sel) < m_creds.size())
    {
        m_selCred = m_creds[sel];
        EndModal(wxID_OK);
    }
    else
    {
        wxMessageBox(wxT("You must select a credential or click Cancel to dismiss this dialog."));
    }
}

bool CPKIFCredentialSelectorDlg::SetMediator(
    IPKIFMediatorPtr& m)
{
    IPKIFMediatorPtr emptyMediator;
    m_mediator = emptyMediator;

    if(NULL != m)
    {
        IPKIFCryptoKeyIDOperations* iCKIO = m->GetMediator<IPKIFCryptoKeyIDOperations>();
        if(NULL == iCKIO)
            return false;
    }
    m_mediator = m;
    return true;
}

void CPKIFCredentialSelectorDlg::SetKeyUsages(
    bitset<9>& ku)
{
    m_ku = ku;
}

CPKIFCredentialPtr CPKIFCredentialSelectorDlg::GetSelectedCredential() const
{
    return m_selCred;
}

void CPKIFCredentialSelectorDlg::OnInitDialog(wxInitDialogEvent& event)
{
#ifndef WIN32
    wxButton* moInfo = GetMoInfo();
    moInfo->Disable();
#endif

    if(m_mediator == (IPKIFMediator*)NULL)
    {
        m_mediator = MakeDefaultMediator();
    }

    if(m_mediator == (IPKIFMediator*)NULL)
    {
        wxMessageBox(wxT("No cryptographic module has been specified.  Unable to list available credentials."), wxT("Error"), wxICON_ERROR);
        return;
    }

    wxListBox* credList = GetListOfCredentials();
    wxASSERT(credList);

    //get a pointer to the key ID interface from the mediator
    IPKIFCryptoKeyIDOperations* iCKIO = m_mediator->GetMediator<IPKIFCryptoKeyIDOperations>();
    if(NULL == iCKIO)
    {
        wxMessageBox(wxT("No acceptable cryptographic interface was found.  Unable to list available credentials."), wxT("Error"), wxICON_ERROR);
        return;
    }

    //empty anything present in the key ID vector
    m_creds.clear();

    //get the list of keys
    iCKIO->GetKeyList(m_creds, &m_ku);

    //iterate over the list and populate the list box
    int ii = 0;
    CPKIFCredentialList::iterator pos;
    CPKIFCredentialList::iterator end = m_creds.end();
    for(pos = m_creds.begin(); pos != end; ++pos, ++ii)
    {
        wxString credName((*pos)->Name(),wxConvUTF8);
        credList->InsertItems(1, &credName, ii);
    }
}