Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420/* 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//=====================================
#if !defined(__REACTOS__) || defined(_MSC_VER)#pragma data_seg(".text")#endif#define INITGUID#include <initguid.h>#include <shlguid.h>#include "vfdshguid.h"#if !defined(__REACTOS__) || defined(_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;}