vfdshutil.cpp File Reference

#include <windows.h>
#include <objbase.h>
#include <stdio.h>
#include "vfdtypes.h"
#include "vfdapi.h"
#include "vfdlib.h"
#include "vfdshcfact.h"
#include <initguid.h>
#include <shlguid.h>
#include "vfdshguid.h"

Include dependency graph for vfdshutil.cpp:

Go to the source code of this file.

Macros
#define	WIN32_LEAN_AND_MEAN
#define	INITGUID
#define	REGKEY_APPROVED    "Software\\Microsoft\\Windows\\CurrentVersion\\Shell Extensions\\Approved"

Functions
STDAPI	DllGetClassObject (REFCLSID rclsid, REFIID riid, LPVOID *ppvOut)
STDAPI	DllCanUnloadNow (void)
static void	MakeGuidString (LPTSTR str, const GUID &guid)
DWORD WINAPI	VfdRegisterHandlers ()
DWORD WINAPI	VfdUnregisterHandlers ()
DWORD WINAPI	VfdCheckHandlers ()

Macro Definition Documentation

INITGUID

#define INITGUID

Definition at line 28 of file vfdshutil.cpp.

REGKEY_APPROVED

#define REGKEY_APPROVED    "Software\\Microsoft\\Windows\\CurrentVersion\\Shell Extensions\\Approved"

Definition at line 39 of file vfdshutil.cpp.

WIN32_LEAN_AND_MEAN

#define WIN32_LEAN_AND_MEAN

Definition at line 11 of file vfdshutil.cpp.

Function Documentation

DllCanUnloadNow()

STDAPI DllCanUnloadNow

(

void

)

Definition at line 76 of file vfdshutil.cpp.

{
    VFDTRACE(0,
        ("DllCanUnloadNow - %s\n", (g_cDllRefCnt ? "No" : "Yes")));
 
    return (g_cDllRefCnt ? S_FALSE : S_OK);
}

DllGetClassObject()

STDAPI DllGetClassObject	(	REFCLSID	rclsid,
		REFIID	riid,
		LPVOID *	ppvOut
	)

Definition at line 50 of file vfdshutil.cpp.

{
    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;
}

MakeGuidString()

static void MakeGuidString ( LPTSTR  str, const GUID &  guid  )

inlinestatic

Definition at line 88 of file vfdshutil.cpp.

{
    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]);
}

Referenced by VfdCheckHandlers(), VfdRegisterHandlers(), and VfdUnregisterHandlers().

VfdCheckHandlers()

DWORD WINAPI VfdCheckHandlers

(

)

Definition at line 393 of file vfdshutil.cpp.

{
    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;
}

Referenced by Shell(), and Status().

VfdRegisterHandlers()

DWORD WINAPI VfdRegisterHandlers

(

)

Definition at line 100 of file vfdshutil.cpp.

{
    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;
}

Referenced by Shell().

VfdUnregisterHandlers()

DWORD WINAPI VfdUnregisterHandlers

(

)

Definition at line 297 of file vfdshutil.cpp.

{
    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;
}

Referenced by Shell().