Category

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

Windows Kernel

Development Platform:

Visual C++

main.cpp：Code Content
  1. /*++
  2.    
  3.     Contains DLLMain and standard OLE COM object creation stuff.
  5. --*/
  8. #include <windows.h>
  9. #include <objbase.h>
  10. #include <shlobj.h>
  11. #include <olectl.h> // Dll[Un]RegisterServer
  13. #include "classfac.h"
  14. #include "psexsup.h"
  16. //   GUID stuff
  18. // this is only done once
  19. // TODO, see if this is appropriate
  21. #pragma data_seg(".text")
  22. #define INITGUID
  23. #include <initguid.h>
  24. #include <shlguid.h>
  25. #include "guid.h"
  26. #pragma data_seg()
  29. HINSTANCE   g_hInst;
  30. LONG        g_DllRefCount = 0;
  31. BOOL        g_bShowIETB;
  32. BOOL        g_bShowISTB;
  33. int         g_nColumn1;
  34. int         g_nColumn2;
  37. extern "C" BOOL WINAPI DllMain(
  38.     HINSTANCE hInstance,
  39.     DWORD dwReason,
  40.     LPVOID lpReserved
  41.     )
  42. {
  43.     switch(dwReason)
  44.     {
  45.         case DLL_PROCESS_ATTACH:
  46.             DisableThreadLibraryCalls(hInstance);
  47.             g_hInst = hInstance;
  49.             //
  50.             // initialize Protected Storage Support routines
  51.             //
  52.             
  53.             if(!InitializePStoreSupport())
  54.                 return FALSE;
  56.     //
  57. // init common controls
  58. //
  60.     INITCOMMONCONTROLSEX iccex;
  61.     iccex.dwSize = sizeof(INITCOMMONCONTROLSEX);
  62.     iccex.dwICC = ICC_LISTVIEW_CLASSES;
  63.     InitCommonControlsEx(&iccex);
  65.             break;
  67.         case DLL_PROCESS_DETACH:
  68.             ShutdownPStoreSupport();
  69.             break;
  70.     }
  72.     return TRUE;
  73. }                                 
  76. STDAPI
  77. DllCanUnloadNow(
  78.     void
  79.     )
  80. {
  81.     return (g_DllRefCount ? S_FALSE : S_OK);
  82. }
  85. STDAPI
  86. DllGetClassObject(
  87.     REFCLSID rclsid,
  88.     REFIID riid,
  89.     LPVOID *ppReturn
  90.     )
  91. {
  93.     //
  94.     // if we don't support this classid, return the proper error code
  95.     //
  97.     if(!IsEqualCLSID(rclsid, CLSID_PStoreNameSpace))
  98.        return CLASS_E_CLASSNOTAVAILABLE;
  100.     //
  101.     // create a CClassFactory object and check it for validity
  102.     //
  104.     CClassFactory *pClassFactory = new CClassFactory();
  105.     if(NULL == pClassFactory)
  106.        return E_OUTOFMEMORY;
  108.     //
  109.     // get the QueryInterface return for our return value
  110.     //
  112.     HRESULT hResult = pClassFactory->QueryInterface(riid, ppReturn);
  114.     //
  115.     // call Release to decrement the ref count - creating the object set it to
  116.     // one and QueryInterface incremented it - since its being used externally
  117.     // (not by us), we only want the ref count to be 1
  118.     //
  120.     pClassFactory->Release();
  122.     return hResult;
  123. }
  127. //
  128. // overload new and delete so we don't need to bring in full CRT
  129. //
  132. void * __cdecl operator new(unsigned int cb)
  133. {
  134.     return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, cb);
  135. }
  137. void __cdecl operator delete(void * pv)
  138. {
  139.     HeapFree(GetProcessHeap(), 0, pv);
  140. }