vfdshutil.cpp

Go to the documentation of this file.

/*
    vfdshutil.cpp
 
    Virtual Floppy Drive for Windows
    Driver control library
    shell extension utility functions
 
    Copyright (c) 2003-2005 Ken Kato
*/
 
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <objbase.h>
#include <stdio.h>
 
#include "vfdtypes.h"
#include "vfdapi.h"
#include "vfdlib.h"
#include "vfdshcfact.h"
 
//=====================================
// Initialize the GUID instance
//=====================================
 
#ifdef _MSC_VER
#pragma data_seg(".text")
#endif
#define INITGUID
#include <initguid.h>
#include <shlguid.h>
#include "vfdshguid.h"
#ifdef _MSC_VER
#pragma data_seg()
#endif
 
//
//  Registry path to the approved shell extensions key
//
#define REGKEY_APPROVED \
    "Software\\Microsoft\\Windows\\CurrentVersion\\Shell Extensions\\Approved"
 
 
//=====================================
//  Shell extension library requirements
//=====================================
 
//
//  Creates a class factory instance
//
STDAPI DllGetClassObject(
    REFCLSID        rclsid,
    REFIID          riid,
    LPVOID          *ppvOut)
{
    VFDTRACE(0,
        ("DllGetClassObject\n"));
 
    *ppvOut = NULL;
 
    if (IsEqualIID(rclsid, CLSID_VfdShellExt)) {
        CVfdFactory *pFactory = new CVfdFactory;
 
        if (!pFactory) {
            return E_OUTOFMEMORY;
        }
 
        return pFactory->QueryInterface(riid, ppvOut);
    }
 
    return CLASS_E_CLASSNOTAVAILABLE;
}
 
//
//  DllCanUnloadNow
//
STDAPI DllCanUnloadNow(void)
{
    VFDTRACE(0,
        ("DllCanUnloadNow - %s\n", (g_cDllRefCnt ? "No" : "Yes")));
 
    return (g_cDllRefCnt ? S_FALSE : S_OK);
}
 
//=====================================
// Shell extension register functions
//=====================================
 
static inline void MakeGuidString(LPTSTR str, const GUID &guid)
{
    sprintf(str, "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
        guid.Data1, guid.Data2, guid.Data3,
        guid.Data4[0], guid.Data4[1],
        guid.Data4[2], guid.Data4[3], guid.Data4[4],
        guid.Data4[5], guid.Data4[6], guid.Data4[7]);
}
 
//
//  Regster this dll as shell extention handlers
//
DWORD WINAPI VfdRegisterHandlers()
{
    TCHAR   buf[MAX_PATH];
    TCHAR   guid_str[40];
    HKEY    hKey;
    DWORD   temp;
    DWORD   ret;
 
    MakeGuidString(guid_str, CLSID_VfdShellExt);
 
    //
    //  Register the GUID in the CLSID subtree
    //
    sprintf(buf, "CLSID\\%s", guid_str);
 
    VFDTRACE(0, ("HKCR\\%s\n", buf));
 
    ret = RegCreateKeyEx(
        HKEY_CLASSES_ROOT, buf, 0, NULL,
        0, KEY_ALL_ACCESS, NULL, &hKey, &temp);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    if (temp == REG_OPENED_EXISTING_KEY) {
        temp = sizeof(buf);
 
        ret = RegQueryValueEx(
            hKey, NULL, NULL, NULL, (PBYTE)buf, &temp);
 
        if (ret != ERROR_SUCCESS) {
            RegCloseKey(hKey);
            return ret;
        }
 
        if (_stricmp(buf, VFDEXT_DESCRIPTION)) {
            RegCloseKey(hKey);
            return ERROR_FILE_EXISTS;
        }
    }
    else {
 
        VFDTRACE(0, ("@=" VFDEXT_DESCRIPTION "\n"));
 
        ret = RegSetValueEx(hKey, NULL, NULL, REG_SZ,
            (PBYTE)VFDEXT_DESCRIPTION, sizeof(VFDEXT_DESCRIPTION));
 
        RegCloseKey(hKey);
 
        if (ret != ERROR_SUCCESS) {
            return ret;
        }
    }
 
    //
    //  Register the executable path
    //
    sprintf(buf, "CLSID\\%s\\InProcServer32", guid_str);
 
    VFDTRACE(0, ("HKCR\\%s\n", buf));
 
    ret = RegCreateKeyEx(
        HKEY_CLASSES_ROOT, buf, 0, NULL,
        0, KEY_ALL_ACCESS, NULL, &hKey, NULL);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    temp = GetModuleFileName(g_hDllModule, buf, sizeof(buf));
 
    VFDTRACE(0, ("@=%s\n", buf));
 
    ret = RegSetValueEx(
        hKey, NULL, NULL, REG_SZ, (PBYTE)buf, temp + 1);
 
    if (ret != ERROR_SUCCESS) {
        RegCloseKey(hKey);
        return ret;
    }
 
    VFDTRACE(0, ("ThreadingModel=Apartment\n"));
 
    ret = RegSetValueEx(hKey, "ThreadingModel", NULL, REG_SZ,
        (PBYTE)"Apartment", sizeof("Apartment"));
 
    RegCloseKey(hKey);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    //
    //  Register context menu handler
    //
    VFDTRACE(0, ("HKCR\\" VFDEXT_MENU_REGKEY "\n"));
 
    ret = RegCreateKeyEx(
        HKEY_CLASSES_ROOT, VFDEXT_MENU_REGKEY, 0, NULL,
        0, KEY_ALL_ACCESS, NULL, &hKey, NULL);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    VFDTRACE(0, ("@=%s\n", guid_str));
 
    ret = RegSetValueEx(hKey, NULL, NULL, REG_SZ,
        (PBYTE)guid_str, strlen(guid_str) + 1);
 
    RegCloseKey(hKey);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    //
    //  Register Drag&Drop handler
    //
    if (!IS_WINDOWS_NT()) {
        //
        //  Windows NT does not support Drag&Drop handlers ???
        //
        VFDTRACE(0, ("HKCR\\" VFDEXT_DND_REGKEY "\n"));
 
        ret = RegCreateKeyEx(
            HKEY_CLASSES_ROOT, VFDEXT_DND_REGKEY, 0, NULL,
            0, KEY_ALL_ACCESS, NULL, &hKey, NULL);
 
        if (ret != ERROR_SUCCESS) {
            return ret;
        }
 
        VFDTRACE(0, ("@=%s\n", guid_str));
 
        ret = RegSetValueEx(hKey, NULL, NULL, REG_SZ,
            (PBYTE)guid_str, strlen(guid_str) + 1);
 
        RegCloseKey(hKey);
 
        if (ret != ERROR_SUCCESS) {
            return ret;
        }
    }
 
    //
    //  Register property sheet handler
    //
    VFDTRACE(0, ("HKCR\\" VFDEXT_PROP_REGKEY "\n"));
 
    ret = RegCreateKeyEx(
        HKEY_CLASSES_ROOT, VFDEXT_PROP_REGKEY, 0, NULL,
        0, KEY_ALL_ACCESS, NULL, &hKey, NULL);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    VFDTRACE(0, ("@=%s\n", guid_str));
 
    ret = RegSetValueEx(hKey, NULL, NULL, REG_SZ,
        (PBYTE)guid_str, strlen(guid_str) + 1);
 
    RegCloseKey(hKey);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    //
    //  Register approved extensions entry
    //
    VFDTRACE(0, ("HKLM\\" REGKEY_APPROVED "\n"));
 
    ret = RegCreateKeyEx(
        HKEY_LOCAL_MACHINE, REGKEY_APPROVED,
        0, NULL, 0, KEY_ALL_ACCESS, NULL, &hKey, NULL);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    VFDTRACE(0,
        ("%s=" VFDEXT_DESCRIPTION "\n", guid_str));
 
    ret = RegSetValueEx(hKey, guid_str, NULL, REG_SZ,
        (PBYTE)VFDEXT_DESCRIPTION, sizeof(VFDEXT_DESCRIPTION));
 
    RegCloseKey(hKey);
 
    return ret;
}
 
//
//  Unregister context menu handler
//
DWORD WINAPI VfdUnregisterHandlers()
{
    TCHAR   buf[MAX_PATH];
    TCHAR   guid_str[40];
    HKEY    hKey;
    DWORD   temp;
    DWORD   ret;
 
    MakeGuidString(guid_str, CLSID_VfdShellExt);
 
    sprintf(buf, "CLSID\\%s", guid_str);
 
    VFDTRACE(0, ("HKCR\\%s\n", buf));
 
    temp = sizeof(buf);
 
    ret = RegQueryValue(HKEY_CLASSES_ROOT, buf, buf, (PLONG)&temp);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    if (_stricmp(buf, VFDEXT_DESCRIPTION)) {
        return ERROR_PATH_NOT_FOUND;
    }
 
    sprintf(buf, "CLSID\\%s\\InProcServer32", guid_str);
 
    VFDTRACE(0, ("HKCR\\%s\n", buf));
 
    ret = RegDeleteKey(HKEY_CLASSES_ROOT, buf);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    sprintf(buf, "CLSID\\%s", guid_str);
 
    VFDTRACE(0, ("HKCR\\%s\n", buf));
 
    ret = RegDeleteKey(HKEY_CLASSES_ROOT, buf);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    VFDTRACE(0, ("HKCR\\" VFDEXT_MENU_REGKEY "\n"));
 
    ret = RegDeleteKey(HKEY_CLASSES_ROOT, VFDEXT_MENU_REGKEY);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    if (!IS_WINDOWS_NT()) {
 
        //  Windows NT doesn't support Drag & Drop handlers ???
 
        VFDTRACE(0, ("HKCR\\" VFDEXT_DND_REGKEY "\n"));
 
        ret = RegDeleteKey(HKEY_CLASSES_ROOT, VFDEXT_DND_REGKEY);
 
        if (ret != ERROR_SUCCESS) {
            return ret;
        }
    }
 
    VFDTRACE(0, ("HKCR\\" VFDEXT_PROP_REGKEY "\n"));
 
    ret = RegDeleteKey(HKEY_CLASSES_ROOT, VFDEXT_PROP_REGKEY);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    VFDTRACE(0, ("HKLM\\" REGKEY_APPROVED "\n"));
 
    ret = RegCreateKeyEx(
        HKEY_LOCAL_MACHINE,
        REGKEY_APPROVED,
        0, NULL, 0, KEY_ALL_ACCESS, NULL, &hKey, NULL);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    ret = RegDeleteValue(hKey, guid_str);
 
    RegCloseKey(hKey);
 
    return ret;
}
 
//
//  Check if context menu handler is registered
//
DWORD WINAPI VfdCheckHandlers()
{
    TCHAR   buf[MAX_PATH];
    TCHAR   guid_str[40];
    DWORD   temp;
    DWORD   ret;
 
    MakeGuidString(guid_str, CLSID_VfdShellExt);
 
    sprintf(buf, "CLSID\\%s", guid_str);
 
    VFDTRACE(0, ("HKCR\\%s\n", buf));
 
    temp = sizeof(buf);
 
    ret = RegQueryValue(HKEY_CLASSES_ROOT, buf, buf, (PLONG)&temp);
 
    if (ret != ERROR_SUCCESS) {
        return ret;
    }
 
    if (_stricmp(buf, VFDEXT_DESCRIPTION)) {
        return ERROR_PATH_NOT_FOUND;
    }
 
    return ERROR_SUCCESS;
}