contrib/issue-tracker-plugins/ExampleAtlPlugin/ExampleAtlPlugin.cpp

   1 // ExampleAtlPlugin.cpp : Implementation of DLL Exports.
   2
   3
   4 #include "stdafx.h"
   5 #include "resource.h"
   6 #include "ExampleAtlPlugin_i.h"
   7 #include "dllmain.h"
   8
   9 // Used to determine whether the DLL can be unloaded by OLE
  10 STDAPI DllCanUnloadNow(void)
  11 {
  12     return _AtlModule.DllCanUnloadNow();
  13 }
  14
  15
  16 // Returns a class factory to create an object of the requested type
  17 STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
  18 {
  19     return _AtlModule.DllGetClassObject(rclsid, riid, ppv);
  20 }
  21
  22
  23 // DllRegisterServer - Adds entries to the system registry
  24 STDAPI DllRegisterServer(void)
  25 {
  26     // registers object, typelib and all interfaces in typelib
  27     HRESULT hr = _AtlModule.DllRegisterServer();
  28         return hr;
  29 }
  30
  31
  32 // DllUnregisterServer - Removes entries from the system registry
  33 STDAPI DllUnregisterServer(void)
  34 {
  35         HRESULT hr = _AtlModule.DllUnregisterServer();
  36         return hr;
  37 }
  38
  39 // DllInstall - Adds/Removes entries to the system registry per user
  40 //              per machine.
  41 STDAPI DllInstall(BOOL bInstall, LPCWSTR pszCmdLine)
  42 {
  43     HRESULT hr = E_FAIL;
  44     static const wchar_t szUserSwitch[] = _T("user");
  45
  46     if (pszCmdLine != NULL)
  47     {
  48         if (_wcsnicmp(pszCmdLine, szUserSwitch, _countof(szUserSwitch)) == 0)
  49         {
  50                 AtlSetPerUserRegistration(true);
  51         }
  52     }
  53
  54     if (bInstall)
  55     {
  56         hr = DllRegisterServer();
  57         if (FAILED(hr))
  58         {
  59                 DllUnregisterServer();
  60         }
  61     }
  62     else
  63     {
  64         hr = DllUnregisterServer();
  65     }
  66
  67     return hr;
  68 }
  69
  70