nextcloud-desktop/shell_integration/windows/OCShellExtensions/OCOverlays/DllMain.cpp

/**
 * Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

#include "OCOverlayRegistrationHandler.h"
#include "OCOverlayFactory.h"
#include "stdafx.h"

HINSTANCE instanceHandle = NULL;

long dllReferenceCount = 0;

BOOL APIENTRY DllMain(HMODULE hModule, DWORD dwReason, LPVOID lpReserved)
{
	switch (dwReason)
	{
		case DLL_PROCESS_ATTACH:
			instanceHandle = hModule;
			DisableThreadLibraryCalls(hModule);
			break;
		case DLL_THREAD_ATTACH:
		case DLL_THREAD_DETACH:
		case DLL_PROCESS_DETACH:
			break;
	}

	return TRUE;
}

HRESULT CreateFactory(REFIID riid, void **ppv, int state)
{
	HRESULT hResult = E_OUTOFMEMORY;

	OCOverlayFactory* ocOverlayFactory = new OCOverlayFactory(state);

	if (ocOverlayFactory) {
		hResult = ocOverlayFactory->QueryInterface(riid, ppv);
		ocOverlayFactory->Release();
	}
	return hResult;
}

STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
{
    HRESULT hResult = CLASS_E_CLASSNOTAVAILABLE;
	GUID guid;
 
	hResult = CLSIDFromString(OVERLAY_GUID_ERROR, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_Error); }

	hResult = CLSIDFromString(OVERLAY_GUID_ERROR_SHARED, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_ErrorShared); }

	hResult = CLSIDFromString(OVERLAY_GUID_OK, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_OK); }

	hResult = CLSIDFromString(OVERLAY_GUID_OK_SHARED, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_OKShared); }

	hResult = CLSIDFromString(OVERLAY_GUID_SYNC, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_Sync); }

	hResult = CLSIDFromString(OVERLAY_GUID_SYNC_SHARED, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_SyncShared); }

	hResult = CLSIDFromString(OVERLAY_GUID_WARNING, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_Warning); }

	hResult = CLSIDFromString(OVERLAY_GUID_WARNING_SHARED, (LPCLSID)&guid);
	if (!SUCCEEDED(hResult)) { return hResult; }
	if (IsEqualCLSID(guid, rclsid)) { return CreateFactory(riid, ppv, State_WarningShared); }
	
	return CLASS_E_CLASSNOTAVAILABLE;
}

STDAPI DllCanUnloadNow(void)
{
    return dllReferenceCount > 0 ? S_FALSE : S_OK;
	
	return S_FALSE;
}

HRESULT RegisterCLSID(LPCOLESTR guidStr, PCWSTR overlayStr, PCWSTR szModule)
{
	HRESULT hResult = S_OK;

	GUID guid;
	hResult = CLSIDFromString(guidStr, (LPCLSID)&guid);

	if (hResult != S_OK) {
		return hResult;
	}

	hResult = OCOverlayRegistrationHandler::RegisterCOMObject(szModule, guid);

	if (!SUCCEEDED(hResult)) {
		return hResult;
	}

	hResult = OCOverlayRegistrationHandler::MakeRegistryEntries(guid, overlayStr);

	return hResult;
}

HRESULT _stdcall DllRegisterServer(void)
{
	HRESULT hResult = S_OK;

	wchar_t szModule[MAX_PATH];

	if (GetModuleFileName(instanceHandle, szModule, ARRAYSIZE(szModule)) == 0) {	
		hResult = HRESULT_FROM_WIN32(GetLastError());
		return hResult;
	}

	hResult = RegisterCLSID(OVERLAY_GUID_ERROR, OVERLAY_NAME_ERROR, szModule);
	if (!SUCCEEDED(hResult)) {	return hResult;	}
	hResult = RegisterCLSID(OVERLAY_GUID_ERROR_SHARED, OVERLAY_NAME_ERROR_SHARED, szModule);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = RegisterCLSID(OVERLAY_GUID_OK, OVERLAY_NAME_OK, szModule);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = RegisterCLSID(OVERLAY_GUID_OK_SHARED, OVERLAY_NAME_OK_SHARED, szModule);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = RegisterCLSID(OVERLAY_GUID_SYNC, OVERLAY_NAME_SYNC, szModule);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = RegisterCLSID(OVERLAY_GUID_SYNC_SHARED, OVERLAY_NAME_SYNC_SHARED,szModule);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = RegisterCLSID(OVERLAY_GUID_WARNING, OVERLAY_NAME_WARNING, szModule);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = RegisterCLSID(OVERLAY_GUID_WARNING_SHARED, OVERLAY_NAME_WARNING_SHARED, szModule);

    return hResult;
}


HRESULT UnregisterCLSID(LPCOLESTR guidStr, PCWSTR overlayStr)
{
	HRESULT hResult = S_OK;
	GUID guid;

	hResult = CLSIDFromString(guidStr, (LPCLSID)&guid);

	if (hResult != S_OK) {
		return hResult;
	}

	hResult = OCOverlayRegistrationHandler::UnregisterCOMObject(guid);

	if (!SUCCEEDED(hResult)) {
		return hResult;
	}

	hResult = OCOverlayRegistrationHandler::RemoveRegistryEntries(overlayStr);

	return hResult;
}

STDAPI DllUnregisterServer(void)
{
    HRESULT hResult = S_OK;

    wchar_t szModule[MAX_PATH];
    
	if (GetModuleFileNameW(instanceHandle, szModule, ARRAYSIZE(szModule)) == 0)
    {
        hResult = HRESULT_FROM_WIN32(GetLastError());
        return hResult;
    }

	hResult = UnregisterCLSID(OVERLAY_GUID_ERROR, OVERLAY_NAME_ERROR);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_ERROR_SHARED, OVERLAY_NAME_ERROR_SHARED);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_OK, OVERLAY_NAME_OK);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_OK_SHARED, OVERLAY_NAME_OK_SHARED);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_SYNC, OVERLAY_NAME_SYNC);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_SYNC_SHARED, OVERLAY_NAME_SYNC_SHARED);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_WARNING, OVERLAY_NAME_WARNING);
	if (!SUCCEEDED(hResult)) { return hResult; }
	hResult = UnregisterCLSID(OVERLAY_GUID_WARNING_SHARED, OVERLAY_NAME_WARNING_SHARED);

    return hResult;
}