onlineupdate/source/update/inc/nsWindowsHelpers.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef nsWindowsHelpers_h
   8 #define nsWindowsHelpers_h
   9
  10 #include <windows.h>
  11 #include "nsAutoRef.h"
  12
  13 // Critical Section helper class
  14
  15 class AutoCriticalSection
  16 {
  17 public:
  18   AutoCriticalSection(LPCRITICAL_SECTION aSection)
  19     : mSection(aSection)
  20   {
  21     ::EnterCriticalSection(mSection);
  22   }
  23   ~AutoCriticalSection()
  24   {
  25     ::LeaveCriticalSection(mSection);
  26   }
  27 private:
  28   LPCRITICAL_SECTION mSection;
  29 };
  30
  31 template<>
  32 class nsAutoRefTraits<HKEY>
  33 {
  34 public:
  35   typedef HKEY RawRef;
  36   static HKEY Void()
  37   {
  38     return nullptr;
  39   }
  40
  41   static void Release(RawRef aFD)
  42   {
  43     if (aFD != Void()) {
  44       RegCloseKey(aFD);
  45     }
  46   }
  47 };
  48
  49 template<>
  50 class nsAutoRefTraits<SC_HANDLE>
  51 {
  52 public:
  53   typedef SC_HANDLE RawRef;
  54   static SC_HANDLE Void()
  55   {
  56     return nullptr;
  57   }
  58
  59   static void Release(RawRef aFD)
  60   {
  61     if (aFD != Void()) {
  62       CloseServiceHandle(aFD);
  63     }
  64   }
  65 };
  66
  67 template<>
  68 class nsSimpleRef<HANDLE>
  69 {
  70 protected:
  71   typedef HANDLE RawRef;
  72
  73   nsSimpleRef() : mRawRef(nullptr)
  74   {
  75   }
  76
  77   nsSimpleRef(RawRef aRawRef) : mRawRef(aRawRef)
  78   {
  79   }
  80
  81   bool HaveResource() const
  82   {
  83     return mRawRef && mRawRef != INVALID_HANDLE_VALUE;
  84   }
  85
  86 public:
  87   RawRef get() const
  88   {
  89     return mRawRef;
  90   }
  91
  92   static void Release(RawRef aRawRef)
  93   {
  94     if (aRawRef && aRawRef != INVALID_HANDLE_VALUE) {
  95       CloseHandle(aRawRef);
  96     }
  97   }
  98   RawRef mRawRef;
  99 };
 100
 101
 102 template<>
 103 class nsAutoRefTraits<HMODULE>
 104 {
 105 public:
 106   typedef HMODULE RawRef;
 107   static RawRef Void()
 108   {
 109     return nullptr;
 110   }
 111
 112   static void Release(RawRef aFD)
 113   {
 114     if (aFD != Void()) {
 115       FreeLibrary(aFD);
 116     }
 117   }
 118 };
 119
 120
 121 namespace {
 122
 123 HMODULE inline
 124 LoadLibrarySystem32(LPCWSTR aModule)
 125 {
 126   WCHAR systemPath[MAX_PATH + 1] = { L'\0' };
 127
 128   // If GetSystemPath fails we accept that we'll load the DLLs from the
 129   // normal search path.
 130   GetSystemDirectoryW(systemPath, MAX_PATH + 1);
 131   size_t systemDirLen = wcslen(systemPath);
 132
 133   // Make the system directory path terminate with a slash
 134   if (systemDirLen && systemPath[systemDirLen - 1] != L'\\') {
 135     systemPath[systemDirLen] = L'\\';
 136     ++systemDirLen;
 137     // No need to re-nullptr terminate
 138   }
 139
 140   size_t fileLen = wcslen(aModule);
 141   wcsncpy(systemPath + systemDirLen, aModule,
 142           MAX_PATH - systemDirLen);
 143   if (systemDirLen + fileLen <= MAX_PATH) {
 144     systemPath[systemDirLen + fileLen] = L'\0';
 145   } else {
 146     systemPath[MAX_PATH] = L'\0';
 147   }
 148   return LoadLibraryW(systemPath);
 149 }
 150
 151 }
 152
 153 #endif