Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
dll.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 5k
Category:

Windows Kernel

Development Platform:

Visual C++

dll.cpp：Code Content
  1. /*----------------------------------------------------------------------------
  2. / Title;
  3. /   dll.cpp
  4. /
  5. / Authors;
  6. /   David De Vorchik (daviddv)
  7. /
  8. / Notes;
  9. /   Core entry points for the DLL
  10. /----------------------------------------------------------------------------*/
  11. #include "pch.h"
  12. #include "query.h"
  13. #include <advpub.h>         // For REGINSTALL
  14. #pragma hdrstop
  16. #define INITGUID
  17. #include <initguid.h>
  18. #include "cmnquery.h"
  19. #include "dsquery.h"
  20. #include "iids.h"
  23. /*----------------------------------------------------------------------------
  24. / Globals
  25. /----------------------------------------------------------------------------*/
  27. HINSTANCE g_hInstance = 0;
  29. HRESULT CallRegInstall(LPSTR szSection);
  32. /*-----------------------------------------------------------------------------
  33. / DllMain
  34. / -------
  35. /   Main entry point.  We are passed reason codes and assored other
  36. /   information when loaded or closed down.
  37. /
  38. / In:
  39. /   hInstance = our instance handle
  40. /   dwReason = reason code
  41. /   pReserved = depends on the reason code.
  42. /
  43. / Out:
  44. /   -
  45. /----------------------------------------------------------------------------*/
  46. STDAPI_(BOOL) DllMain( HINSTANCE hInstance, DWORD dwReason, LPVOID pReserved )
  47. {
  48.     if ( dwReason == DLL_PROCESS_ATTACH )
  49.     {
  50.         GLOBAL_HINSTANCE = hInstance;
  51.     }
  53.     return TRUE;
  54. }
  57. /*-----------------------------------------------------------------------------
  58. / DllCanUnloadNow
  59. / ---------------
  60. /   Called by the outside world to determine if our DLL can be unloaded. If we
  61. /   have any objects in existance then we must not unload.
  62. /
  63. / In:
  64. /   -
  65. / Out:
  66. /   BOOL inidicate unload state.
  67. /----------------------------------------------------------------------------*/
  68. STDAPI DllCanUnloadNow( void )
  69. {
  70.     return GLOBAL_REFCOUNT ? S_FALSE : S_OK;
  71. }
  74. /*-----------------------------------------------------------------------------
  75. / DllGetClassObject
  76. / -----------------
  77. /   Given a class ID and an interface ID, return the relevant object.  This used
  78. /   by the outside world to access the objects contained here in.
  79. /
  80. / In:
  81. /   rCLISD = class ID required
  82. /   riid = interface within that class required
  83. /   ppVoid -> receives the newly created object.
  84. /
  85. / Out:
  86. /   -
  87. /----------------------------------------------------------------------------*/
  88. STDAPI DllGetClassObject(REFCLSID rCLSID, REFIID riid, LPVOID* ppVoid)
  89. {
  90.     HRESULT hr = E_OUTOFMEMORY;
  92.     *ppVoid = NULL;
  94.     if ( IsEqualIID(rCLSID, CLSID_ExampleQueryForm) )
  95.     {
  96.         CExampleQueryFormClassFactory* pClassFactory = new CExampleQueryFormClassFactory;
  98.         if ( !pClassFactory )
  99.             ExitGracefully(hr, E_OUTOFMEMORY, "Failed to get CExampleQueryForm's class factory");
  101.         hr = pClassFactory->QueryInterface(riid, ppVoid);
  103.         if ( FAILED(hr) )
  104.             delete pClassFactory;
  105.     }
  106.     else
  107.     {
  108.         ExitGracefully(hr, E_NOINTERFACE, "Object doesn't support the given CLISD");
  109.     }
  111. exit_gracefully:
  113.     return hr;
  114. }
  117. /*-----------------------------------------------------------------------------
  118. / Dll[Unregister/Register]Server
  119. / ------------------------------
  120. /   Called to allow us to setup the registry entries that we use, this
  121. /   takes advantage of the ADVPACK APIs and loads our .inf data from
  122. /   our resource block.
  123. /
  124. / In:
  125. /   -
  126. / Out:
  127. /   HRESULT
  128. /----------------------------------------------------------------------------*/
  129. STDAPI DllRegisterServer(VOID)
  130. {
  131.     return CallRegInstall("RegDll");
  132. }
  134. STDAPI DllUnregisterServer(VOID)
  135. {
  136.     return CallRegInstall("UnRegDll");
  137. }    
  140. /*-----------------------------------------------------------------------------
  141. / CallRegInstall
  142. / --------------
  143. /   Call ADVPACK for the given section of our resource based INF>
  144. /
  145. / In:
  146. /   szSection = section name to invoke
  147. /
  148. / Out:
  149. /   HRESULT:
  150. /----------------------------------------------------------------------------*/
  151. HRESULT CallRegInstall(LPSTR szSection)
  152. {
  153.     HRESULT hr = E_FAIL;
  154.     HINSTANCE hinstAdvPack = LoadLibrary(TEXT("ADVPACK.DLL"));
  156.     if (hinstAdvPack)
  157.     {
  158.         REGINSTALL pfnri = (REGINSTALL)GetProcAddress(hinstAdvPack, "RegInstall");
  160. #ifdef UNICODE
  161.         if ( pfnri )
  162.         {
  163.             STRENTRY seReg[] = 
  164.             {
  165.                 // These two NT-specific entries must be at the end
  166.                 { "25", "%SystemRoot%" },
  167.                 { "11", "%SystemRoot%\system32" },
  168.             };
  169.             STRTABLE stReg = { ARRAYSIZE(seReg), seReg };
  170.             
  171.             hr = pfnri(GLOBAL_HINSTANCE, szSection, &stReg);
  172.         }
  173. #else
  174.         if (pfnri)
  175.         {
  176.             hr = pfnri(GLOBAL_HINSTANCE, szSection, NULL);
  177.         }
  179. #endif
  180.         FreeLibrary(hinstAdvPack);
  181.     }
  183.     return hr;
  184. }