PROMS/DSOFramer/Source2005/mainentry.cpp

@@ -0,0 +1,318 @@
+/***************************************************************************
+ * MAINENTRY.CPP
+ *
+ * Main DLL Entry and Required COM Entry Points.
+ *
+ *  Copyright <20>1999-2004; Microsoft Corporation. All rights reserved.
+ *  Written by Microsoft Developer Support Office Integration (PSS DSOI)
+ * 
+ *  This code is provided via KB 311765 as a sample. It is not a formal
+ *  product and has not been tested with all containers or servers. Use it
+ *  for educational purposes only. See the EULA.TXT file included in the
+ *  KB download for full terms of use and restrictions.
+ *
+ *  THIS CODE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, 
+ *  EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
+ *  WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ ***************************************************************************/
+#define INITGUID // Init the GUIDS for the control...
+#include "dsoframer.h"
+
+HINSTANCE        v_hModule        = NULL;   // DLL module handle
+HANDLE           v_hPrivateHeap   = NULL;   // Private Memory Heap
+ULONG            v_cLocks         = 0;      // Count of server locks
+HICON            v_icoOffDocIcon  = NULL;   // Small office icon (for caption bar)
+BOOL             v_fUnicodeAPI    = FALSE;  // Flag to determine if we should us Unicode API
+BOOL             v_fWindows2KPlus = FALSE;
+CRITICAL_SECTION v_csecThreadSynch;
+
+////////////////////////////////////////////////////////////////////////
+// DllMain -- OCX Main Entry
+//
+//
+extern "C" BOOL APIENTRY DllMain(HINSTANCE hDllHandle, DWORD dwReason, LPVOID /*lpReserved*/)
+{
+	switch (dwReason)
+	{
+	case DLL_PROCESS_ATTACH:
+		v_hModule = hDllHandle;
+        v_hPrivateHeap = HeapCreate(0, 0x1000, 0);
+		v_icoOffDocIcon = (HICON)LoadImage(hDllHandle, MAKEINTRESOURCE(IDI_SMALLOFFDOC), IMAGE_ICON, 16, 16, 0);
+		{
+			DWORD dwVersion = GetVersion();
+			v_fUnicodeAPI = ((dwVersion & 0x80000000) == 0);
+			v_fWindows2KPlus = ((v_fUnicodeAPI) && (LOBYTE(LOWORD(dwVersion)) > 4));
+		}
+		InitializeCriticalSection(&v_csecThreadSynch);
+		DisableThreadLibraryCalls(hDllHandle);
+		break;
+
+	case DLL_PROCESS_DETACH:
+		if (v_icoOffDocIcon) DeleteObject(v_icoOffDocIcon);
+        if (v_hPrivateHeap) HeapDestroy(v_hPrivateHeap);
+        DeleteCriticalSection(&v_csecThreadSynch);
+		break;
+	}
+	return TRUE;
+}
+
+#ifdef DSO_MIN_CRT_STARTUP
+extern "C" BOOL APIENTRY _DllMainCRTStartup(HINSTANCE hDllHandle, DWORD dwReason, LPVOID lpReserved)
+{return DllMain(hDllHandle, dwReason, lpReserved);}
+#endif
+
+////////////////////////////////////////////////////////////////////////
+// Standard COM DLL Entry Points
+//
+//
+////////////////////////////////////////////////////////////////////////
+// DllCanUnloadNow
+//
+//
+STDAPI DllCanUnloadNow()
+{
+	return ((v_cLocks == 0) ? S_OK : S_FALSE);
+}
+
+////////////////////////////////////////////////////////////////////////
+// DllGetClassObject
+//
+//  Returns IClassFactory instance for FramerControl. We only support
+//  this one object for creation.
+//
+STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void** ppv)
+{
+	HRESULT hr;
+	CDsoFramerClassFactory* pcf;
+
+	CHECK_NULL_RETURN(ppv, E_POINTER);
+	*ppv = NULL;
+
+ // The only component we can create is the BinderControl...
+	if (rclsid != CLSID_FramerControl)
+		return CLASS_E_CLASSNOTAVAILABLE;
+
+ // Create the needed class factory...
+	pcf = new CDsoFramerClassFactory();
+	CHECK_NULL_RETURN(pcf, E_OUTOFMEMORY);
+
+ // Get requested interface.
+	if (FAILED(hr = pcf->QueryInterface(riid, ppv)))
+	{
+		*ppv = NULL; delete pcf;
+	}
+	else InterlockedIncrement((LPLONG)&v_cLocks);
+
+	return hr;
+}
+
+////////////////////////////////////////////////////////////////////////
+// DllRegisterServer
+//
+//  Registration of the OCX.
+//
+STDAPI DllRegisterServer()
+{
+    HRESULT hr = S_OK;
+    HKEY    hk, hk2;
+    DWORD   dwret;
+    CHAR    szbuffer[256];
+	LPWSTR  pwszModule;
+    ITypeInfo* pti;
+    
+ // If we can't find the path to the DLL, we can't register...
+	if (!FGetModuleFileName(v_hModule, &pwszModule))
+		return E_FAIL;
+
+ // Setup the CLSID. This is the most important. If there is a critical failure,
+ // we will set HR = GetLastError and return...
+    if ((dwret = RegCreateKeyEx(HKEY_CLASSES_ROOT, 
+		"CLSID\\"DSOFRAMERCTL_CLSIDSTR, 0, NULL, 0, KEY_WRITE, NULL, &hk, NULL)) != ERROR_SUCCESS)
+	{
+		DsoMemFree(pwszModule);
+        return HRESULT_FROM_WIN32(dwret);
+	}
+
+    lstrcpy(szbuffer, DSOFRAMERCTL_SHORTNAME);
+    RegSetValueEx(hk, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+
+ // Setup the InprocServer32 key...
+    dwret = RegCreateKeyEx(hk, "InprocServer32", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+    if (dwret == ERROR_SUCCESS)
+    {
+        lstrcpy(szbuffer, "Apartment");
+        RegSetValueEx(hk2, "ThreadingModel", 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+        
+	 // We call a wrapper function for this setting since the path should be
+	 // stored in Unicode to handle non-ANSI file path names on some systems.
+	 // This wrapper will convert the path to ANSI if we are running on Win9x.
+	 // The rest of the Reg calls should be OK in ANSI since they do not
+	 // contain non-ANSI/Unicode-specific characters...
+		if (!FSetRegKeyValue(hk2, pwszModule))
+            hr = E_ACCESSDENIED;
+
+        RegCloseKey(hk2);
+
+        dwret = RegCreateKeyEx(hk, "ProgID", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+        if (dwret == ERROR_SUCCESS)
+        {
+            lstrcpy(szbuffer, DSOFRAMERCTL_PROGID);
+            RegSetValueEx(hk2, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+            RegCloseKey(hk2);
+        }
+
+    }
+    else hr = HRESULT_FROM_WIN32(dwret);
+
+	if (SUCCEEDED(hr))
+	{
+		dwret = RegCreateKeyEx(hk, "Control", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+		if (dwret == ERROR_SUCCESS)
+		{
+			RegCloseKey(hk2);
+		}
+		else hr = HRESULT_FROM_WIN32(dwret);
+	}
+
+
+ // If we succeeded so far, andle the remaining (non-critical) reg keys...
+	if (SUCCEEDED(hr))
+	{
+		dwret = RegCreateKeyEx(hk, "ToolboxBitmap32", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+		if (dwret == ERROR_SUCCESS)
+		{
+			LPWSTR pwszT = DsoCopyStringCat(pwszModule, L",102");
+			if (pwszT)
+			{
+				FSetRegKeyValue(hk2, pwszT);
+				DsoMemFree(pwszT);
+			}
+			RegCloseKey(hk2);
+		}
+
+		dwret = RegCreateKeyEx(hk, "TypeLib", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+		if (dwret == ERROR_SUCCESS)
+		{
+			lstrcpy(szbuffer, DSOFRAMERCTL_TLIBSTR);
+			RegSetValueEx(hk2, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+			RegCloseKey(hk2);
+		}
+
+		dwret = RegCreateKeyEx(hk, "Version", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+		if (dwret == ERROR_SUCCESS)
+		{
+			lstrcpy(szbuffer, DSOFRAMERCTL_VERSIONSTR);
+			RegSetValueEx(hk2, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+			RegCloseKey(hk2);
+		}
+
+		dwret = RegCreateKeyEx(hk, "MiscStatus", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+		if (dwret == ERROR_SUCCESS)
+		{
+			lstrcpy(szbuffer, "131473");
+			RegSetValueEx(hk2, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+			RegCloseKey(hk2);
+		}
+
+		dwret = RegCreateKeyEx(hk, "DataFormats\\GetSet\\0", 0, NULL, 0, KEY_WRITE, NULL, &hk2, NULL);
+		if (dwret == ERROR_SUCCESS)
+		{
+			lstrcpy(szbuffer, "3,1,32,1");
+			RegSetValueEx(hk2, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+			RegCloseKey(hk2);
+		}
+    }
+
+    RegCloseKey(hk);
+	DsoMemFree(pwszModule);
+
+ // This should catch any critical failures during setup of CLSID...
+	RETURN_ON_FAILURE(hr);
+
+ // Setup the ProgID (non-critical)...
+    if (RegCreateKeyEx(HKEY_CLASSES_ROOT, DSOFRAMERCTL_PROGID, 0,
+            NULL, 0, KEY_WRITE, NULL, &hk, NULL) == ERROR_SUCCESS)
+    {
+        lstrcpy(szbuffer, DSOFRAMERCTL_FULLNAME);
+        RegSetValueEx(hk, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+
+        if (RegCreateKeyEx(hk, "CLSID", 0,
+                NULL, 0, KEY_WRITE, NULL, &hk2, NULL) == ERROR_SUCCESS)
+        {
+            lstrcpy(szbuffer, DSOFRAMERCTL_CLSIDSTR);
+            RegSetValueEx(hk2, NULL, 0, REG_SZ, (BYTE *)szbuffer, lstrlen(szbuffer));
+            RegCloseKey(hk2);
+        }
+        RegCloseKey(hk);
+    }
+
+ // Load the type info (this should register the lib once)...
+    hr = DsoGetTypeInfoEx(LIBID_DSOFramer, 0,
+		DSOFRAMERCTL_VERSION_MAJOR, DSOFRAMERCTL_VERSION_MINOR, v_hModule, CLSID_FramerControl, &pti);
+    if (SUCCEEDED(hr)) pti->Release();
+
+	return hr;
+}
+
+////////////////////////////////////////////////////////////////////////
+// RegRecursiveDeleteKey
+//
+//  Helper function called by DllUnregisterServer for nested key removal.
+//
+static HRESULT RegRecursiveDeleteKey(HKEY hkParent, LPCSTR pszSubKey)
+{
+    HRESULT hr = S_OK;
+    HKEY hk;
+    DWORD dwret, dwsize;
+	FILETIME time ;
+    CHAR szbuffer[512];
+
+    dwret = RegOpenKeyEx(hkParent, pszSubKey, 0, KEY_ALL_ACCESS, &hk);
+	if (dwret != ERROR_SUCCESS)
+		return HRESULT_FROM_WIN32(dwret);
+
+ // Enumerate all of the decendents of this child...
+	dwsize = 510 ;
+	while (RegEnumKeyEx(hk, 0, szbuffer, &dwsize, NULL, NULL, NULL, &time) == ERROR_SUCCESS)
+	{
+      // If there are any sub-folders, delete them first (to make NT happy)...
+		hr = RegRecursiveDeleteKey(hk, szbuffer);
+		if (FAILED(hr)) break;
+		dwsize = 510;
+	}
+
+ // Close the child...
+	RegCloseKey(hk);
+
+	RETURN_ON_FAILURE(hr);
+
+ // Delete this child.
+    dwret = RegDeleteKey(hkParent, pszSubKey);
+    if (dwret != ERROR_SUCCESS)
+        hr = HRESULT_FROM_WIN32(dwret);
+
+	return hr;
+}
+
+////////////////////////////////////////////////////////////////////////
+// DllUnregisterServer
+//
+//  Removal code for the OCX.
+//
+STDAPI DllUnregisterServer()
+{
+    HRESULT hr;
+    hr = RegRecursiveDeleteKey(HKEY_CLASSES_ROOT, "CLSID\\"DSOFRAMERCTL_CLSIDSTR);
+    if (SUCCEEDED(hr))
+    {
+        RegRecursiveDeleteKey(HKEY_CLASSES_ROOT, DSOFRAMERCTL_PROGID);
+        RegRecursiveDeleteKey(HKEY_CLASSES_ROOT, "TypeLib\\"DSOFRAMERCTL_TLIBSTR);
+    }
+
+  // This means the key does not exist (i.e., the DLL 
+  // was alreay unregistered, so return OK)...
+    if (hr == 0x80070002) hr = S_OK;
+
+	return hr;
+}