PathSettingsSerializer.cpp

Go to the documentation of this file.

#if defined(__GNUG__) && !defined(NO_GCC_PRAGMA) && defined(PKIFRESOURCES_ALLOW_GCC_PRAGMA)
    #pragma interface "PathSettingsSerializer.cpp"
#endif


#include "PathSettingsSerializer.h"
#include "wx/config.h" 
#include "wx/fileconf.h"
#include "pkif.h"
#include "PKIFBase64.h"
#ifdef _WIN32
#include "pkifregconf.h"
#endif
#include "wx/log.h"

#include "PRStringUtils.h"

#define PKIFRESOURCES_PATH_SERIALIZER           TOOLKIT_APP + 2


void SaveSettings_wx(CPKIFPathSettingsPtr& settingsToSave, wxConfigBase* conf, const char* localFile, bool useFile, bool bAllUsers, const char* pathRelToCompanyApp);

CPKIFPathSettingsSerializer::CPKIFPathSettingsSerializer()
{
}


CPKIFPathSettingsSerializer::~CPKIFPathSettingsSerializer()
{
}

void CPKIFPathSettingsSerializer::SaveSettings(
    CPKIFPathSettingsPtr& settingsToSave, 
    const char* application,
    const char* company, 
    const char* localFile,
    bool useFile, 
    bool bAllUsers, 
    const char* pathRelToCompanyApp)
{
    const char* pCompany = "Orion Security Solutions";
    const char* pApp = "PKIFv2";
    const char* pLocalFile = "";
    if(company)
        pCompany = company;
    if(application)
        pApp = application;
    if(localFile)
        pLocalFile = localFile;

    wxString empty;

    //wxConfig * conf = 0;
    wxConfigBase * conf = 0;
    if(useFile)
    { 
        conf = new wxFileConfig(wxString(pApp,wxConvUTF8), wxString(pCompany,wxConvUTF8), wxString(pLocalFile,wxConvUTF8), empty, wxCONFIG_USE_LOCAL_FILE);
    }
    else
    {
#ifdef _WIN32
        conf = new wxPkifRegConfig(wxString(pApp,wxConvUTF8),wxString(pCompany,wxConvUTF8), wxEmptyString, wxEmptyString, bAllUsers ? wxCONFIG_USE_GLOBAL_FILE : 0, pathRelToCompanyApp);
#else
        conf = new wxConfig(wxString(pApp,wxConvUTF8),wxString(pCompany,wxConvUTF8));
#endif
    }


    try
    {
        SaveSettings_wx(settingsToSave, conf, localFile, useFile, bAllUsers, pathRelToCompanyApp);
        conf->Flush();
        if(conf) delete conf;
    }
    catch(...)
    {
        if(conf) delete conf;
    }
}
    
void SaveSettings_wx(
    CPKIFPathSettingsPtr& settingsToSave, 
    wxConfigBase* conf,
    const char* localFile,
    bool useFile, 
    bool bAllUsers, 
    const char* pathRelToCompanyApp)
{

    wxString wxOldPath = conf->GetPath();
    if(pathRelToCompanyApp)
    {
        wxString wxPathRelToCompanyApp(pathRelToCompanyApp,wxConvUTF8);
        conf->SetPath(wxPathRelToCompanyApp);
    }

    //conf->DeleteAll();
    conf->DeleteGroup(wxT("Path Settings"));

    if(NULL == settingsToSave)
    {
        conf->SetPath(wxOldPath);
        return;
    }


    try
    {
        if(!conf->Write(wxT("Path Settings/CheckRevocationStatus"), settingsToSave->GetCheckRevocationStatus()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing CheckRevocationStatus");
        if(!conf->Write(wxT("Path Settings/PathDepth"),  settingsToSave->GetDepth()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing PathDepth");
        if(!conf->Write(wxT("Path Settings/RequireExplicitPolicy"), settingsToSave->GetInitialExplicitPolicyIndicator()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing RequireExplicitPolicy");
        if(!conf->Write(wxT("Path Settings/InhibitPolicyMapping"), settingsToSave->GetInitialPolicyMappingInhibitIndicator()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing InhibitPolicyMapping");
        if(!conf->Write(wxT("Path Settings/InhibitAnyPolicy"), settingsToSave->GetInitialInhibitAnyPolicyIndicator()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing InhibitAnyPolicy");
        if(!conf->Write(wxT("Path Settings/RequireFreshRevData"), settingsToSave->GetRequireFreshRevocationData()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing RequireFreshRevData");
        if(!conf->Write(wxT("Path Settings/RequireRecentRevData"), settingsToSave->GetRequireSufficientlyRecent()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing RequireRecentRevData");
        if(!conf->Write(wxT("Path Settings/UseValidatorFilterWhenBuilding"), settingsToSave->GetUseValidatorFilterWhenBuilding()))
            throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing UseValidatorFilterWhenBuilding");

        if(settingsToSave->GetRequireSufficientlyRecent())
        {
            CPKIFDurationPtr duration = settingsToSave->GetDuration();
            if(duration != (CPKIFDuration*)NULL && !conf->Write(wxT("Path Settings/Duration"), duration->asSeconds()))
                throw CPKIFException(PKIFRESOURCES_PATH_SERIALIZER, COMMON_UNKNOWN_ERROR, "Settings serialization failed while writing Duration");
        }

        //policy
        wxString group(wxT("Path Settings/CertificatePolicies"));

        CPKIFPolicyInformationListPtr initPolSet;
        settingsToSave->GetInitialPolicySet(initPolSet);

        CPKIFPolicyInformationList::iterator pos = initPolSet->begin();
        CPKIFPolicyInformationList::iterator end = initPolSet->end();

        for(int ii = 0; pos != end; ++ii, ++pos)
        {
            wxString key;
            key.Printf(wxT("%s/%02d"),WXCASTCHARS(group.c_str()),ii);
            conf->Write(key, WXCASTCHARS((*pos)->PolicyOID()->ToString()));
        }

        //ExcludedNameConstraints
        wxString groupExcludedNameConstraints(wxT("Path Settings/ExcludedNameConstraints"));

        CPKIFGeneralSubtreeListPtr exclNC;
        settingsToSave->GetInitialExclSubtrees(exclNC);

        if(exclNC != (CPKIFGeneralSubtreeList*)NULL)
        {
            CPKIFGeneralSubtreeList::iterator pos2 = exclNC->begin();
            CPKIFGeneralSubtreeList::iterator end2 = exclNC->end();

            for(int ii = 0; pos2 != end2; ++ii, ++pos2)
            {
                CPKIFBufferPtr buff = (*pos2)->GetBase()->Encoded();
                char * encoded = NULL;
                B64Encode(buff->GetBuffer(), buff->GetLength(), &encoded);
                wxString key;
                key.Printf(wxT("%s/%02d"),groupExcludedNameConstraints.c_str(),ii);
                conf->Write(key, WXCASTCHARS(encoded));
                if(encoded) {
                    delete encoded;
                }
            }
        }

        //PermittedNameConstraints
        wxString groupPermittedNameConstraints(wxT("Path Settings/PermittedNameConstraints"));

        CPKIFGeneralSubtreeListPtr permNC;
        settingsToSave->GetInitialPermSubtrees(permNC);

        if(permNC != (CPKIFGeneralSubtreeList*)NULL)
        {
            CPKIFGeneralSubtreeList::iterator pos3 = permNC->begin();
            CPKIFGeneralSubtreeList::iterator end3 = permNC->end();

            for(int ii = 0; pos3 != end3; ++ii, ++pos3)
            {
                CPKIFBufferPtr buff = (*pos3)->GetBase()->Encoded();
                char * encoded = NULL;
                B64Encode(buff->GetBuffer(), buff->GetLength(), &encoded);
                wxString key;
                key.Printf(wxT("%s/%02d"),groupPermittedNameConstraints.c_str(),ii);
                conf->Write(key, WXCASTCHARS(encoded));
                if(encoded) {
                    delete encoded;
                }
            }
        }

        
    }
    catch(...)
    {
        throw;
    }

    conf->SetPath(wxOldPath);
    // on UNIX, sometimes settings don't get saved unless you call this explicitly
    conf->Flush();
}

CPKIFPathSettingsPtr CPKIFPathSettingsSerializer::LoadSettings(
    const char* application,
    const char* company, 
    const char* localFile, 
    bool useFile,
    bool bAllUsers,
    const char* pathRelToCompanyApp)

{
    CPKIFPathSettingsPtr ps(new CPKIFPathSettings);
    bool b = false;
    int i = 0;

    wxString empty;

    const char* pCompany = "Orion Security Solutions";
    const char* pApp = "PKIFv2";
    const char* pLocalFile = "";
    if(company)
        pCompany = company;
    if(application)
        pApp = application;
    if(localFile)
        pLocalFile = localFile;

    wxLogNull ln;

    //wxConfig * conf = 0;
    wxConfigBase * conf = 0;
    if(useFile)
    { 
        conf = new wxFileConfig(wxString(pApp,wxConvUTF8), wxString(pCompany,wxConvUTF8), wxString(pLocalFile,wxConvUTF8), empty, wxCONFIG_USE_LOCAL_FILE);
    }
    else
    {
#ifdef _WIN32
        conf = new wxPkifRegConfig(wxString(pApp,wxConvUTF8),wxString(pCompany,wxConvUTF8), wxEmptyString, wxEmptyString, bAllUsers ? wxCONFIG_USE_GLOBAL_FILE : 0, pathRelToCompanyApp);
#else
        conf = new wxConfig(wxString(pApp,wxConvUTF8),wxString(pCompany,wxConvUTF8));
#endif
    }
    
    if(pathRelToCompanyApp)
    {
        wxString wxPathRelToCompanyApp(pathRelToCompanyApp,wxConvUTF8);
        conf->SetPath(wxPathRelToCompanyApp);
    }

    try
    {
        if(conf->Read(wxT("Path Settings/CheckRevocationStatus"), &b, true))
            ps->SetCheckRevocationStatus(b);

        if(conf->Read(wxT("Path Settings/PathDepth"), &i, true))
            ps->SetDepth(i);

        if(conf->Read(wxT("Path Settings/RequireExplicitPolicy"), &b, true))
            ps->SetInitialExplicitPolicyIndicator(b);

        if(conf->Read(wxT("Path Settings/InhibitPolicyMapping"), &b, true))
            ps->SetInitialPolicyMappingInhibitIndicator(b);

        if(conf->Read(wxT("Path Settings/InhibitAnyPolicy"), &b, true))
            ps->SetInitialInhibitAnyPolicyIndicator(b);

        if(conf->Read(wxT("Path Settings/RequireFreshRevData"), &b, true))
            ps->SetRequireFreshRevocationData(b);

        if(conf->Read(wxT("Path Settings/RequireRecentRevData"), &b, true))
            ps->SetRequireSufficientlyRecent(b);

        if(conf->Read(wxT("Path Settings/UseValidatorFilterWhenBuilding"), &b, true))
            ps->SetUseValidatorFilterWhenBuilding(b);

        if(ps->GetRequireSufficientlyRecent())
        {
            double d = 0.0;
            if(conf->Read(wxT("Path Settings/Duration"), &d))
                ps->SetSufficientlyRecent(d);

            
        }

        //policy
        CPKIFPolicyInformationListPtr polList(new CPKIFPolicyInformationList);
        wxString group(wxT("Path Settings/CertificatePolicies"));
        
        long ii = 0;
        unsigned long curr = 0;
        wxString oldPath = conf->GetPath();
        wxString path(wxT("Path Settings/CertificatePolicies"));
        conf->SetPath(path);
        while(conf->GetNextEntry(path,ii))
        {
            wxString key, val;
            key = wxString::Format(wxT("%02d"),curr);
            //key.Printf(wxT("%s/%02d"),oldPath.c_str(),curr);
            if(!conf->Read(key, &val))
                continue;

            std::string strOID((const char *)val.mb_str());
            CPKIFOIDPtr polOID(new CPKIFOID(strOID));
            CPKIFPolicyInformationPtr pol(new CPKIFPolicyInformation(polOID));
            polList->push_back(pol);
            curr++;
        }
        conf->SetPath(oldPath);
        if(curr > 0)
            ps->SetInitialPolicySet(polList);

        //ExclNameConstraints
        CPKIFGeneralSubtreeListPtr exclNCList(new CPKIFGeneralSubtreeList);
        
        long ii2 = 0;
        unsigned long curr2 = 0;
        wxString path2(wxT("Path Settings/ExcludedNameConstraints"));
        wxString oldPath2 = conf->GetPath();
        conf->SetPath(path2);
        while(conf->GetNextEntry(path2,ii2))
        {
            wxString key, val;
            key = wxString::Format(wxT("%02d"),curr2 );
            if(!conf->Read(key, &val))
                continue;

            if(!val.empty())
            {
                unsigned char * decodedNC = NULL;
                unsigned long decodedNCLen;
                wxASCIIHelper vala(val);
                bool b = B64Decode(vala.to_chars(), &decodedNC, &decodedNCLen);
                if(b)
                {   
                    try
                    {
                        //stick the decoded name in a buffer object (which can handle the cleanup for us)
                        CPKIFBufferPtr buf(new CPKIFBuffer(true, decodedNC, decodedNCLen));

                        CPKIFGeneralNamePtr gn(new CPKIFGeneralName(buf));  
                        CPKIFGeneralSubtreePtr subtree(new CPKIFGeneralSubtree());
                        subtree->SetBase(gn);
                        subtree->SetMin(0);
                        exclNCList->push_back(subtree);
                    }
                    catch(...)
                    {
                    }
                }
            }
            curr2++;
        }
        conf->SetPath(oldPath2);
        if(curr2 > 0)
            ps->SetInitialExclSubtrees(exclNCList);

        //PermittedNameConstraints
        CPKIFGeneralSubtreeListPtr permNCList(new CPKIFGeneralSubtreeList);
        
        long ii3 = 0;
        unsigned long curr3 = 0;
        wxString path3(wxT("Path Settings/PermittedNameConstraints"));
        wxString oldPath3 = conf->GetPath();
        conf->SetPath(path3);
        while(conf->GetNextEntry(path3,ii3))
        {
            wxString key, val;
            key = wxString::Format(wxT("%02d"),curr3);
            if(!conf->Read(key, &val))
                continue;

            if(!val.empty())
            {
                unsigned char * decodedNC = NULL;
                unsigned long decodedNCLen;
                wxASCIIHelper vala(val);
                bool b = B64Decode(vala.to_chars(), &decodedNC, &decodedNCLen);
                if(b)
                {   
                    try
                    {
                        //stick the decoded name in a buffer object (which can handle the cleanup for us)
                        CPKIFBufferPtr buf(new CPKIFBuffer(true, decodedNC, decodedNCLen));

                        CPKIFGeneralNamePtr gn(new CPKIFGeneralName(buf));  
                        CPKIFGeneralSubtreePtr subtree(new CPKIFGeneralSubtree());
                        subtree->SetBase(gn);   
                        subtree->SetMin(0);             
                        permNCList->push_back(subtree);
                    }
                    catch(...)
                    {
                    }
                }
            }
            curr3++;
        }
        conf->SetPath(oldPath3);
        if(curr3 > 0)
            ps->SetInitialPermSubtrees(permNCList);
    }
    catch(...)
    {
        if(conf) delete conf;
        throw;
    }
    if(conf) delete conf;
    return ps;
}