gears/base/ie/module.cc

   1 // Copyright 2006, Google Inc.
   2 //
   3 // Redistribution and use in source and binary forms, with or without
   4 // modification, are permitted provided that the following conditions are met:
   5 //
   6 //  1. Redistributions of source code must retain the above copyright notice,
   7 //     this list of conditions and the following disclaimer.
   8 //  2. Redistributions in binary form must reproduce the above copyright notice,
   9 //     this list of conditions and the following disclaimer in the documentation
  10 //     and/or other materials provided with the distribution.
  11 //  3. Neither the name of Google Inc. nor the names of its contributors may be
  12 //     used to endorse or promote products derived from this software without
  13 //     specific prior written permission.
  14 //
  15 // THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
  16 // WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  17 // MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
  18 // EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  19 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  20 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  21 // OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  22 // WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  23 // OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  24 // ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25
  26 #include "gears/base/common/thread_locals.h"
  27 #include "gears/base/ie/resource.h" // for .rgs resource ids (IDR_*)
  28 #include "ie/genfiles/interfaces.h" // from OUTDIR
  29
  30 // This is defined in gears/base/common/message_queue_ie.h.  It should only be
  31 // called here.
  32 void ShutdownThreadMessageQueue();
  33
  34 class DllModule : public CAtlDllModuleT< DllModule > {
  35  public:
  36   DECLARE_LIBID(LIBID_GearsTypelib)
  37   DECLARE_REGISTRY_APPID_RESOURCEID(IDR_GEARSIE, \
  38   "{B56936F7-0433-4E0B-921B-D095E7142B6D}")
  39 };
  40
  41 DllModule atl_module;
  42 #ifdef WINCE
  43 int _charmax = 255;
  44 #endif
  45
  46 inline BOOL MyDllMain(HANDLE instance, DWORD reason, LPVOID reserved) {
  47   switch (reason) {
  48     case DLL_THREAD_DETACH:
  49       ShutdownThreadMessageQueue();
  50       ThreadLocals::HandleThreadDetached();
  51       break;
  52     case DLL_PROCESS_DETACH:
  53       ThreadLocals::HandleProcessDetached();
  54       break;
  55   }
  56
  57   BOOL rv = atl_module.DllMain(reason, reserved);
  58
  59   if ((reason == DLL_PROCESS_ATTACH) && rv) {
  60     rv = ThreadLocals::HandleProcessAttached();
  61   }
  62
  63   return rv;
  64 }
  65
  66 extern "C"
  67 BOOL WINAPI DllMain(HANDLE instance, DWORD reason, LPVOID reserved) {
  68  return MyDllMain(instance, reason, reserved);
  69 }
  70
  71
  72 STDAPI DllCanUnloadNow(void) {
  73   return atl_module.DllCanUnloadNow();
  74 }
  75
  76 STDAPI DllGetClassObject(REFCLSID class_id, REFIID riid, LPVOID* ppv) {
  77   return atl_module.DllGetClassObject(class_id, riid, ppv);
  78 }
  79
  80 STDAPI DllRegisterServer(void) {
  81   return atl_module.DllRegisterServer();
  82 }
  83
  84 STDAPI DllUnregisterServer(void) {
  85   return atl_module.DllUnregisterServer();
  86 }