1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <stdio.h> /* snprintf */
static bool registry_write_string(HKEY reg, const char *name, const char *value, size_t len)
{
HRESULT hr = RegSetValueExA(
reg,
name,
0,
REG_SZ,
reinterpret_cast<const BYTE *> (value),
static_cast<DWORD>((len + 1) * sizeof(char))
);
return hr == ERROR_SUCCESS;
}
static const char *base_key()
{
if (sizeof(void *) == 4)
{
return "Software\\Classes\\WOW6432Node\\CLSID";
}
else
{
return "Software\\Classes\\CLSID";
}
}
static void format_clsid(REFIID clsid, char *value, size_t max)
{
snprintf(value, max, "{%08lX-%04hX-%04hX-%02hhX%02hhX-%02hhX%02hhX%02hhX%02hhX%02hhX%02hhX}",
clsid.Data1, clsid.Data2, clsid.Data3,
clsid.Data4[0], clsid.Data4[1], clsid.Data4[2], clsid.Data4[3],
clsid.Data4[4], clsid.Data4[5], clsid.Data4[6], clsid.Data4[7]);
}
extern "C" HRESULT register_faudio_dll(void *DllHandle, REFIID clsid)
{
// not your typical COM register functions because we're "hijacking" the existing XAudio2 registration
// - COM classes are historically registered under HKEY_CLASSES_ROOT
// - on modern Windows version this is a merged from on HKEY_LOCAL_MACHINE/Software/Classes and HKEY_CURRENT_USER/Software/Classes
// - the system-wide configuration is stored under HKEY_LOCAL_MACHINE
// - we override the registration for the XAudio2 DLLs for the current user
char str_clsid[40];
format_clsid(clsid, str_clsid, sizeof(str_clsid) / sizeof(char));
char key[2048];
snprintf(key, sizeof(key) / sizeof(char), "%s\\%s\\InProcServer32", base_key(), str_clsid);
// open registry (creating key if it does not exist)
HKEY registry_key = NULL;
if (RegCreateKeyExA(HKEY_CURRENT_USER, key, 0, NULL, 0, KEY_WRITE, NULL, ®istry_key, NULL))
{
return E_FAIL;
}
// retrieve path to the dll
char path[2048];
DWORD len = GetModuleFileNameA(reinterpret_cast<HMODULE>(DllHandle), path, sizeof(key) / sizeof(char));
// update registry
registry_write_string(registry_key, NULL, path, len);
RegCloseKey(registry_key);
return S_OK;
}
extern "C" HRESULT unregister_faudio_dll(void *DllHandle, REFIID clsid)
{
// open registry
HKEY registry_key = NULL;
if (RegOpenKeyExA(HKEY_CURRENT_USER, base_key(), 0, DELETE | KEY_SET_VALUE | KEY_ENUMERATE_SUB_KEYS | KEY_QUERY_VALUE, ®istry_key))
{
return E_FAIL;
}
char str_clsid[40];
format_clsid(clsid, str_clsid, sizeof(str_clsid) / sizeof(char));
// delete key
HRESULT hr = RegDeleteTreeA(registry_key, str_clsid);
RegCloseKey(registry_key);
return S_OK;
}
|