68 lines
2.1 KiB
C++
68 lines
2.1 KiB
C++
/* -*- 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";
|
|
|
|
MOZ_RUNINIT 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;
|
|
}
|