Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

119 lines
3.4 KiB

  1. // GetRids.cpp : Implementation of DLL Exports.
  2. // Note: Proxy/Stub Information
  3. // To merge the proxy/stub code into the object DLL, add the file
  4. // dlldatax.c to the project. Make sure precompiled headers
  5. // are turned off for this file, and add _MERGE_PROXYSTUB to the
  6. // defines for the project.
  7. //
  8. // If you are not running WinNT4.0 or Win95 with DCOM, then you
  9. // need to remove the following define from dlldatax.c
  10. // #define _WIN32_WINNT 0x0400
  11. //
  12. // Further, if you are running MIDL without /Oicf switch, you also
  13. // need to remove the following define from dlldatax.c.
  14. // #define USE_STUBLESS_PROXY
  15. //
  16. // Modify the custom build rule for GetRids.idl by adding the following
  17. // files to the Outputs.
  18. // GetRids_p.c
  19. // dlldata.c
  20. // To build a separate proxy/stub DLL,
  21. // run nmake -f GetRidsps.mk in the project directory.
  22. #include "stdafx.h"
  23. #include "resource.h"
  24. #include <initguid.h>
  25. #include "GetRids.h"
  26. #include "dlldatax.h"
  27. #include "GetRids_i.c"
  28. #include "RidSave.h"
  29. #ifdef _MERGE_PROXYSTUB
  30. extern "C" HINSTANCE hProxyDll;
  31. #endif
  32. CComModule _Module;
  33. BEGIN_OBJECT_MAP(ObjectMap)
  34. OBJECT_ENTRY(CLSID_RidSave, RidSave)
  35. END_OBJECT_MAP()
  36. /////////////////////////////////////////////////////////////////////////////
  37. // DLL Entry Point
  38. extern "C"
  39. BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
  40. {
  41. lpReserved;
  42. #ifdef _MERGE_PROXYSTUB
  43. if (!PrxDllMain(hInstance, dwReason, lpReserved))
  44. return FALSE;
  45. #endif
  46. if (dwReason == DLL_PROCESS_ATTACH)
  47. {
  48. ATLTRACE(_T("{GetRids.dll}DllMain(hInstance=0x%08lX, dwReason=DLL_PROCESS_ATTACH,...)\n"), hInstance);
  49. _Module.Init(ObjectMap, hInstance, &LIBID_GETRIDSLib);
  50. DisableThreadLibraryCalls(hInstance);
  51. _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF|_CRTDBG_LEAK_CHECK_DF);
  52. }
  53. else if (dwReason == DLL_PROCESS_DETACH)
  54. {
  55. ATLTRACE(_T("{GetRids.dll}DllMain(hInstance=0x%08lX, dwReason=DLL_PROCESS_DETACH,...)\n"), hInstance);
  56. _Module.Term();
  57. }
  58. return TRUE; // ok
  59. }
  60. /////////////////////////////////////////////////////////////////////////////
  61. // Used to determine whether the DLL can be unloaded by OLE
  62. STDAPI DllCanUnloadNow(void)
  63. {
  64. #ifdef _MERGE_PROXYSTUB
  65. if (PrxDllCanUnloadNow() != S_OK)
  66. return S_FALSE;
  67. #endif
  68. return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
  69. }
  70. /////////////////////////////////////////////////////////////////////////////
  71. // Returns a class factory to create an object of the requested type
  72. STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
  73. {
  74. #ifdef _MERGE_PROXYSTUB
  75. if (PrxDllGetClassObject(rclsid, riid, ppv) == S_OK)
  76. return S_OK;
  77. #endif
  78. return _Module.GetClassObject(rclsid, riid, ppv);
  79. }
  80. /////////////////////////////////////////////////////////////////////////////
  81. // DllRegisterServer - Adds entries to the system registry
  82. STDAPI DllRegisterServer(void)
  83. {
  84. #ifdef _MERGE_PROXYSTUB
  85. HRESULT hRes = PrxDllRegisterServer();
  86. if (FAILED(hRes))
  87. return hRes;
  88. #endif
  89. // registers object, typelib and all interfaces in typelib
  90. return _Module.RegisterServer(TRUE);
  91. }
  92. /////////////////////////////////////////////////////////////////////////////
  93. // DllUnregisterServer - Removes entries from the system registry
  94. STDAPI DllUnregisterServer(void)
  95. {
  96. #ifdef _MERGE_PROXYSTUB
  97. PrxDllUnregisterServer();
  98. #endif
  99. return _Module.UnregisterServer(TRUE);
  100. }