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
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
* vim: sw=2 ts=2 et lcs=trail\:.,tab\:>~ :
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "mozilla/RefPtr.h"
#include "RegUtils.h"
#include <windows.h>
#include <objbase.h>
already_AddRefed<IClassFactory> CreateFactory();
// {10A9521E-0205-4CC7-93A1-62F30A9A54B3}
GUID CLSID_TestShellEx = {
0x10a9521e, 0x205, 0x4cc7, {0x93, 0xa1, 0x62, 0xf3, 0xa, 0x9a, 0x54, 0xb3}};
wchar_t kFriendlyName[] = L"Minimum Shell Extension for Firefox testing";
std::wstring gDllPath;
BOOL APIENTRY DllMain(HMODULE aModule, DWORD aReason, LPVOID) {
wchar_t buf[MAX_PATH];
switch (aReason) {
case DLL_PROCESS_ATTACH:
if (!::GetModuleFileNameW(aModule, buf, ARRAYSIZE(buf))) {
return FALSE;
}
gDllPath = buf;
break;
case DLL_THREAD_ATTACH:
case DLL_THREAD_DETACH:
case DLL_PROCESS_DETACH:
break;
}
return TRUE;
}
STDAPI DllGetClassObject(REFCLSID aClsid, REFIID aIid, void** aResult) {
if (!IsEqualCLSID(aClsid, CLSID_TestShellEx) ||
!IsEqualCLSID(aIid, IID_IClassFactory)) {
return CLASS_E_CLASSNOTAVAILABLE;
}
RefPtr<IClassFactory> factory = CreateFactory();
return factory ? factory->QueryInterface(aIid, aResult) : E_OUTOFMEMORY;
}
STDAPI DllCanUnloadNow() { return S_OK; }
// These functions enable us to run "regsvr32 [/u] TestShellEx.dll" manually.
// (No admin privilege is needed because all access is under HKCU.)
// We don't use these functions in the mochitest, but having these makes easier
// to test the module manually.
STDAPI DllRegisterServer() {
ComRegisterer reg(CLSID_TestShellEx, kFriendlyName);
if (!reg.RegisterObject(L"Apartment") || !reg.RegisterExtensions()) {
return E_ACCESSDENIED;
}
return S_OK;
}
STDAPI DllUnregisterServer() {
ComRegisterer reg(CLSID_TestShellEx, kFriendlyName);
if (!reg.UnregisterAll()) {
return E_ACCESSDENIED;
}
return S_OK;
}
|
