D:/opendownloadmanager/ODM-1.x/vmsFileExtensionInOs.cpp

Go to the documentation of this file.

/*
  Free Download Manager Copyright (c) 2003-2007 FreeDownloadManager.ORG
  Open Download Manager Copyright (c) 2008-2010 OpenDownloadManager.ORG
*/        

#include "stdafx.h"
#include "FdmApp.h"
#include "vmsFileExtensionInOs.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif        

vmsFileExtensionInOs::vmsFileExtensionInOs()
{

}

vmsFileExtensionInOs::~vmsFileExtensionInOs()
{

}

fsString vmsFileExtensionInOs::GetAssociation(LPCSTR pszExt, LPCSTR pszVerb)
{
        fsString strExt; strExt = "."; strExt += pszExt;

        CRegKey key;
        if (ERROR_SUCCESS != key.Open (HKEY_CLASSES_ROOT, strExt, KEY_READ))
                return "";

        char sz [1000];
        DWORD dw = sizeof (sz);
        if (ERROR_SUCCESS == key.QueryValue (sz, NULL, &dw))
                return GetAssociation_2 (sz, pszVerb);
        else
                return GetAssociation_2 (strExt, pszVerb);
}

fsString vmsFileExtensionInOs::GetAssociation_2(LPCSTR pszProgId, LPCSTR pszVerb)
{
        fsString str = pszProgId;
        str += "\\shell\\"; str += pszVerb; str += "\\command";

        CRegKey key;
        
        if (ERROR_SUCCESS != key.Open (HKEY_CLASSES_ROOT, str, KEY_READ))
                return "";

        char sz [MY_MAX_PATH];
        DWORD dw = sizeof (sz);
        if (ERROR_SUCCESS != key.QueryValue (sz, NULL, &dw))
                return "";
        
        return sz;
}

BOOL vmsFileExtensionInOs::SetAssociation(LPCSTR pszExt, LPCSTR pszVerb, LPCSTR pszValue)
{
        fsString strExt; strExt = "."; strExt += pszExt;

        CRegKey key;
        if (ERROR_SUCCESS != key.Open (HKEY_CLASSES_ROOT, strExt, KEY_READ | KEY_WRITE))
        {
                if (ERROR_SUCCESS != key.Create (HKEY_CLASSES_ROOT, strExt))
                        return FALSE;
        }

        char sz [1000];
        DWORD dw = sizeof (sz);
        if (ERROR_SUCCESS == key.QueryValue (sz, NULL, &dw))
                return SetAssociation_2 (sz, pszVerb, pszValue);
        else
                return SetAssociation_2 (strExt, pszVerb, pszValue);
}

BOOL vmsFileExtensionInOs::SetAssociation_2(LPCSTR pszProgId, LPCSTR pszVerb, LPCSTR pszValue)
{
        fsString str = pszProgId;
        str += "\\shell\\"; str += pszVerb; str += "\\command";

        CRegKey key;
        
        if (ERROR_SUCCESS != key.Open (HKEY_CLASSES_ROOT, str, KEY_WRITE))
        {
                if (ERROR_SUCCESS != key.Create (HKEY_CLASSES_ROOT, str))
                        return FALSE;
        }

        return ERROR_SUCCESS == key.SetValue (pszValue, NULL);
}

BOOL vmsFileExtensionInOs::CreateAssociation(LPCSTR pszExt, LPCSTR pszProgId)
{
        fsString strExt; strExt = "."; strExt += pszExt;

        CRegKey key;

        if (ERROR_SUCCESS != key.Create (HKEY_CLASSES_ROOT, strExt))
                return FALSE;

        if (ERROR_SUCCESS != key.SetValue (pszProgId, NULL))
                return FALSE;

        key.Close ();

        if (ERROR_SUCCESS != key.Create (HKEY_CLASSES_ROOT, pszProgId))
        {
                if (GetLastError () != ERROR_ALREADY_EXISTS)
                        return FALSE;
        }

        return TRUE;
}

---