dlls/wiaservc/wiaservc_main.c

   1 /*
   2  * Main DLL interface to WIA Device Manager
   3  *
   4  * Copyright 2009 Damjan Jovanovic
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdio.h>
  22
  23 #include "objbase.h"
  24 #include "winuser.h"
  25 #include "winreg.h"
  26 #include "advpub.h"
  27 #include "olectl.h"
  28 #include "winsvc.h"
  29
  30 #include "wia_lh.h"
  31 #include "initguid.h"
  32
  33 #include "wiaservc_private.h"
  34
  35 #include "wine/debug.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(wia);
  38
  39 /* Handle to the base address of this DLL */
  40 static HINSTANCE hInst;
  41
  42 /* Entry point for DLL */
  43 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  44 {
  45     TRACE("(%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
  46
  47     switch (fdwReason)
  48     {
  49         case DLL_WINE_PREATTACH:
  50             return FALSE;  /* prefer native version */
  51         case DLL_PROCESS_ATTACH:
  52             DisableThreadLibraryCalls(hinstDLL);
  53             hInst = hinstDLL;
  54             break;
  55         case DLL_PROCESS_DETACH:
  56             break;
  57     }
  58
  59     return TRUE;
  60 }
  61
  62 static HRESULT init_register_strtable(STRTABLEA *strtable)
  63 {
  64 #define CLSID_EXPANSION_ENTRY(id) { "CLSID_" #id, &CLSID_ ## id }
  65     static const struct {
  66         const char *name;
  67         const CLSID *clsid;
  68     } expns[] =  {
  69         CLSID_EXPANSION_ENTRY(WiaDevMgr)
  70     };
  71 #undef CLSID_EXPANSION_ENTRY
  72     static STRENTRYA pse[sizeof expns / sizeof expns[0]];
  73     DWORD i;
  74
  75     strtable->cEntries = sizeof pse / sizeof pse[0];
  76     strtable->pse = pse;
  77     for (i = 0; i < strtable->cEntries; i++) {
  78         static const char dummy_sample[] = "{12345678-1234-1234-1234-123456789012}";
  79         const CLSID *clsid = expns[i].clsid;
  80         pse[i].pszName = wiaservc_strdup(expns[i].name);
  81         pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, sizeof dummy_sample);
  82         if (!pse[i].pszName || !pse[i].pszValue)
  83             return E_OUTOFMEMORY;
  84         sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
  85                 clsid->Data1, clsid->Data2, clsid->Data3, clsid->Data4[0],
  86                 clsid->Data4[1], clsid->Data4[2], clsid->Data4[3], clsid->Data4[4],
  87                 clsid->Data4[5], clsid->Data4[6], clsid->Data4[7]);
  88     }
  89
  90     return S_OK;
  91 }
  92
  93 static void cleanup_register_strtable(STRTABLEA *strtable)
  94 {
  95     DWORD i;
  96     for (i = 0; i < strtable->cEntries; i++) {
  97         HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszName);
  98         HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszValue);
  99         if (!strtable->pse[i].pszName || !strtable->pse[i].pszValue)
 100             return;
 101     }
 102 }
 103
 104 static HRESULT register_service(BOOL do_register)
 105 {
 106     static const WCHAR name[] = { 's','t','i','s','v','c', 0 };
 107     static const WCHAR path[] = { 's','v','c','h','o','s','t','.','e','x','e',
 108                                   ' ','-','k',' ','i','m','g','s','v','c', 0 };
 109     SC_HANDLE scm, service;
 110
 111     scm = OpenSCManagerW(NULL, NULL, SC_MANAGER_ALL_ACCESS);
 112     if (!scm)
 113         return SELFREG_E_CLASS;
 114
 115     if (do_register)
 116         service = CreateServiceW(scm, name, name, SERVICE_ALL_ACCESS,
 117                                  SERVICE_WIN32_OWN_PROCESS,
 118                                  SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
 119                                  path, NULL, NULL, NULL, NULL, NULL);
 120     else
 121         service = OpenServiceW(scm, name, DELETE);
 122
 123
 124     CloseServiceHandle(scm);
 125     if (service)
 126     {
 127         if (!do_register) DeleteService(service);
 128         CloseServiceHandle(service);
 129     }
 130     return S_OK;
 131 }
 132
 133 /* Use an INF file to register or unregister the DLL */
 134 static HRESULT register_server(BOOL do_register)
 135 {
 136     HRESULT hr;
 137     STRTABLEA strtable;
 138     HMODULE hAdvpack;
 139     HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
 140     static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
 141
 142     TRACE("(%x)\n", do_register);
 143
 144     hr = register_service(do_register);
 145     if (FAILED(hr)) {
 146         ERR("register_service failed: %d\n", GetLastError());
 147         return hr;
 148     }
 149
 150     hAdvpack = LoadLibraryW(wszAdvpack);
 151     pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
 152
 153     hr = init_register_strtable(&strtable);
 154     if (SUCCEEDED(hr))
 155         hr = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll",
 156                          &strtable);
 157     cleanup_register_strtable(&strtable);
 158
 159     if (FAILED(hr))
 160         ERR("RegInstall failed: %08x\n", hr);
 161
 162     return hr;
 163 }
 164
 165 HRESULT WINAPI DllRegisterServer(void)
 166 {
 167     return register_server(TRUE);
 168 }
 169
 170 HRESULT WINAPI DllUnregisterServer(void)
 171 {
 172     return register_server(FALSE);
 173 }