git.oblomov.eu Git - wine/blob - dlls/advpack/reg.c

   1 /*
   2  * Advpack registry functions
   3  *
   4  * Copyright 2004 Huw D M Davies
   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 <stdarg.h>
  22 #include "windef.h"
  23 #include "winbase.h"
  24 #include "winreg.h"
  25 #include "winerror.h"
  26 #include "winuser.h"
  27 #include "winternl.h"
  28 #include "setupapi.h"
  29 #include "advpub.h"
  30 #include "wine/unicode.h"
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(advpack);
  34
  35 static const WCHAR REGINST[] = {'R','E','G','I','N','S','T',0};
  36 static const WCHAR Strings[] = {'S','t','r','i','n','g','s',0};
  37 static const WCHAR MOD_PATH[] = {'_','M','O','D','_','P','A','T','H',0};
  38 static const WCHAR SYS_MOD_PATH[] = {'_','S','Y','S','_','M','O','D','_','P','A','T','H',0};
  39 static const WCHAR SystemRoot[] = {'S','y','s','t','e','m','R','o','o','t',0};
  40 static const WCHAR escaped_SystemRoot[] = {'%','S','y','s','t','e','m','R','o','o','t','%',0};
  41 static const WCHAR quote[] = {'\"',0};
  42
  43 static BOOL get_temp_ini_path(LPWSTR name)
  44 {
  45     WCHAR tmp_dir[MAX_PATH];
  46     WCHAR prefix[] = {'a','v','p',0};
  47
  48     if(!GetTempPathW(sizeof(tmp_dir)/sizeof(WCHAR), tmp_dir))
  49        return FALSE;
  50
  51     if(!GetTempFileNameW(tmp_dir, prefix, 0, name))
  52         return FALSE;
  53     return TRUE;
  54 }
  55
  56 static BOOL create_tmp_ini_file(HMODULE hm, WCHAR *ini_file)
  57 {
  58     HRSRC hrsrc;
  59     HGLOBAL hmem = NULL;
  60     DWORD rsrc_size, bytes_written;
  61     VOID *rsrc_data;
  62     HANDLE hf = INVALID_HANDLE_VALUE;
  63
  64     if(!get_temp_ini_path(ini_file)) {
  65         ERR("Can't get temp ini file path\n");
  66         goto error;
  67     }
  68
  69     if(!(hrsrc = FindResourceW(hm, REGINST, REGINST))) {
  70         ERR("Can't find REGINST resource\n");
  71         goto error;
  72     }
  73
  74     rsrc_size = SizeofResource(hm, hrsrc);
  75     hmem = LoadResource(hm, hrsrc);
  76     rsrc_data = LockResource(hmem);
  77
  78     if(!rsrc_data || !rsrc_size) {
  79         ERR("Can't load REGINST resource\n");
  80         goto error;
  81     }
  82
  83     if((hf = CreateFileW(ini_file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
  84                          FILE_ATTRIBUTE_NORMAL, NULL)) == INVALID_HANDLE_VALUE) {
  85         ERR("Unable to create temp ini file\n");
  86         goto error;
  87     }
  88     if(!WriteFile(hf, rsrc_data, rsrc_size, &bytes_written, NULL) || rsrc_size != bytes_written) {
  89         ERR("Write failed\n");
  90         goto error;
  91     }
  92     FreeResource(hmem);
  93     CloseHandle(hf);
  94     return TRUE;
  95 error:
  96     if(hmem) FreeResource(hmem);
  97     if(hf != INVALID_HANDLE_VALUE) CloseHandle(hf);
  98     return FALSE;
  99 }
 100
 101 static void strentry_atow(STRENTRYA *aentry, STRENTRYW *wentry)
 102 {
 103     DWORD name_len, val_len;
 104
 105     name_len = MultiByteToWideChar(CP_ACP, 0, aentry->pszName, -1, NULL, 0);
 106     val_len = MultiByteToWideChar(CP_ACP, 0, aentry->pszValue, -1, NULL, 0);
 107
 108     wentry->pszName = HeapAlloc(GetProcessHeap(), 0, name_len * sizeof(WCHAR));
 109     wentry->pszValue = HeapAlloc(GetProcessHeap(), 0, val_len * sizeof(WCHAR));
 110
 111     MultiByteToWideChar(CP_ACP, 0, aentry->pszName, -1, wentry->pszName, name_len);
 112     MultiByteToWideChar(CP_ACP, 0, aentry->pszValue, -1, wentry->pszValue, val_len);
 113 }
 114
 115 static STRTABLEW *strtable_atow(const STRTABLEA *atable)
 116 {
 117     STRTABLEW *wtable;
 118     DWORD j;
 119
 120     wtable = HeapAlloc(GetProcessHeap(), 0, sizeof(STRTABLEW));
 121     wtable->pse = HeapAlloc(GetProcessHeap(), 0, atable->cEntries * sizeof(STRENTRYW));
 122     wtable->cEntries = atable->cEntries;
 123
 124     for (j = 0; j < wtable->cEntries; j++)
 125         strentry_atow(&atable->pse[j], &wtable->pse[j]);
 126
 127     return wtable;
 128 }
 129
 130 static void free_strtable(STRTABLEW *wtable)
 131 {
 132     DWORD j;
 133
 134     for (j = 0; j < wtable->cEntries; j++)
 135     {
 136         HeapFree(GetProcessHeap(), 0, wtable->pse[j].pszName);
 137         HeapFree(GetProcessHeap(), 0, wtable->pse[j].pszValue);
 138     }
 139
 140     HeapFree(GetProcessHeap(), 0, wtable->pse);
 141     HeapFree(GetProcessHeap(), 0, wtable);
 142 }
 143
 144 /***********************************************************************
 145  *          RegInstallA (advpack.@)
 146  *
 147  * See RegInstallW.
 148  */
 149 HRESULT WINAPI RegInstallA(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable)
 150 {
 151     UNICODE_STRING section;
 152     STRTABLEW *wtable;
 153     HRESULT hr;
 154
 155     TRACE("(%p, %s, %p)\n", hm, debugstr_a(pszSection), pstTable);
 156
 157     if (pstTable)
 158         wtable = strtable_atow(pstTable);
 159     else
 160         wtable = NULL;
 161
 162     RtlCreateUnicodeStringFromAsciiz(&section, pszSection);
 163
 164     hr = RegInstallW(hm, section.Buffer, wtable);
 165
 166     if (pstTable)
 167         free_strtable(wtable);
 168
 169     RtlFreeUnicodeString(&section);
 170
 171     return hr;
 172 }
 173
 174 static HRESULT write_predefined_strings(HMODULE hm, LPWSTR ini_path)
 175 {
 176     WCHAR mod_path[MAX_PATH + 2];
 177     WCHAR sys_mod_path[MAX_PATH + 2];
 178     WCHAR sys_root[MAX_PATH];
 179
 180     *mod_path = '\"';
 181     if (!GetModuleFileNameW(hm, mod_path + 1, sizeof(mod_path) / sizeof(WCHAR) - 2))
 182         return E_FAIL;
 183
 184     lstrcatW(mod_path, quote);
 185     WritePrivateProfileStringW(Strings, MOD_PATH, mod_path, ini_path);
 186
 187     *sys_root = '\0';
 188     GetEnvironmentVariableW(SystemRoot, sys_root, sizeof(sys_root) / sizeof(WCHAR));
 189
 190     if(!strncmpiW(sys_root, mod_path + 1, strlenW(sys_root)))
 191     {
 192         *sys_mod_path = '\"';
 193         strcpyW(sys_mod_path + 1, escaped_SystemRoot);
 194         strcatW(sys_mod_path, mod_path + 1 + strlenW(sys_root));
 195     }
 196     else
 197     {
 198         FIXME("SYS_MOD_PATH needs more work\n");
 199         strcpyW(sys_mod_path, mod_path);
 200     }
 201
 202     WritePrivateProfileStringW(Strings, SYS_MOD_PATH, sys_mod_path, ini_path);
 203
 204     return S_OK;
 205 }
 206
 207 /***********************************************************************
 208  *          RegInstallW (advpack.@)
 209  *
 210  * Loads an INF from a string resource, adds entries to the string
 211  * substitution table, and executes the INF.
 212  *
 213  * PARAMS
 214  *   hm         [I] Module that contains the REGINST resouce.
 215  *   pszSection [I] The INF section to execute.
 216  *   pstTable   [I] Table of string substitutions.
 217  *
 218  * RETURNS
 219  *   Success: S_OK.
 220  *   Failure: E_FAIL.
 221  */
 222 HRESULT WINAPI RegInstallW(HMODULE hm, LPCWSTR pszSection, const STRTABLEW* pstTable)
 223 {
 224     int i;
 225     CABINFOW cabinfo;
 226     WCHAR tmp_ini_path[MAX_PATH];
 227     HINF hinf = INVALID_HANDLE_VALUE;
 228     HRESULT hr = E_FAIL;
 229
 230     TRACE("(%p, %s, %p)\n", hm, debugstr_w(pszSection), pstTable);
 231
 232     if(!create_tmp_ini_file(hm, tmp_ini_path))
 233         return E_FAIL;
 234
 235     if (write_predefined_strings(hm, tmp_ini_path))
 236         goto done;
 237
 238     /* Write the additional string table */
 239     if (pstTable)
 240     {
 241         for(i = 0; i < pstTable->cEntries; i++)
 242         {
 243             WCHAR tmp_value[MAX_PATH + 2];
 244
 245             tmp_value[0] = '\"';
 246             lstrcpyW(tmp_value + 1, pstTable->pse[i].pszValue);
 247             lstrcatW(tmp_value, quote);
 248
 249             WritePrivateProfileStringW(Strings, pstTable->pse[i].pszName, tmp_value, tmp_ini_path);
 250         }
 251     }
 252
 253     /* flush cache */
 254     WritePrivateProfileStringW(NULL, NULL, NULL, tmp_ini_path);
 255
 256     /* FIXME: read AdvOptions val for dwFlags */
 257     ZeroMemory(&cabinfo, sizeof(CABINFOW));
 258     cabinfo.pszInf = tmp_ini_path;
 259     cabinfo.pszSection = (LPWSTR)pszSection;
 260     cabinfo.dwFlags = 0;
 261
 262     hr = ExecuteCabW(NULL, &cabinfo, NULL);
 263
 264 done:
 265     if (hinf != INVALID_HANDLE_VALUE)
 266         SetupCloseInfFile(hinf);
 267
 268     DeleteFileW(tmp_ini_path);
 269
 270     return hr;
 271 }
 272
 273 /***********************************************************************
 274  *          RegRestoreAllA (advpack.@)
 275  *
 276  * See RegRestoreAllW.
 277  */
 278 HRESULT WINAPI RegRestoreAllA(HWND hWnd, LPSTR pszTitleString, HKEY hkBackupKey)
 279 {
 280     UNICODE_STRING title;
 281     HRESULT hr;
 282
 283     TRACE("(%p, %s, %p)\n", hWnd, debugstr_a(pszTitleString), hkBackupKey);
 284
 285     RtlCreateUnicodeStringFromAsciiz(&title, pszTitleString);
 286
 287     hr = RegRestoreAllW(hWnd, title.Buffer, hkBackupKey);
 288
 289     RtlFreeUnicodeString(&title);
 290
 291     return hr;
 292 }
 293
 294 /***********************************************************************
 295  *          RegRestoreAllW (advpack.@)
 296  *
 297  * Restores all saved registry entries.
 298  *
 299  * PARAMS
 300  *   hWnd           [I] Handle to the window used for the display.
 301  *   pszTitleString [I] Title of the window.
 302  *   hkBackupKey    [I] Handle to the backup key.
 303  *
 304  * RETURNS
 305  *   Success: S_OK.
 306  *   Failure: E_FAIL.
 307  *
 308  * BUGS
 309  *   Unimplemented.
 310  */
 311 HRESULT WINAPI RegRestoreAllW(HWND hWnd, LPWSTR pszTitleString, HKEY hkBackupKey)
 312 {
 313     FIXME("(%p, %s, %p) stub\n", hWnd, debugstr_w(pszTitleString), hkBackupKey);
 314
 315     return E_FAIL;
 316 }
 317
 318 /***********************************************************************
 319  *          RegSaveRestoreA (advpack.@)
 320  *
 321  * See RegSaveRestoreW.
 322  */
 323 HRESULT WINAPI RegSaveRestoreA(HWND hWnd, LPCSTR pszTitleString, HKEY hkBackupKey,
 324                                LPCSTR pcszRootKey, LPCSTR pcszSubKey,
 325                                LPCSTR pcszValueName, DWORD dwFlags)
 326 {
 327     UNICODE_STRING title, root, subkey, value;
 328     HRESULT hr;
 329
 330     TRACE("(%p, %s, %p, %s, %s, %s, %ld)\n", hWnd, debugstr_a(pszTitleString),
 331           hkBackupKey, debugstr_a(pcszRootKey), debugstr_a(pcszSubKey),
 332           debugstr_a(pcszValueName), dwFlags);
 333
 334     RtlCreateUnicodeStringFromAsciiz(&title, pszTitleString);
 335     RtlCreateUnicodeStringFromAsciiz(&root, pcszRootKey);
 336     RtlCreateUnicodeStringFromAsciiz(&subkey, pcszSubKey);
 337     RtlCreateUnicodeStringFromAsciiz(&value, pcszValueName);
 338
 339     hr = RegSaveRestoreW(hWnd, title.Buffer, hkBackupKey, root.Buffer,
 340                          subkey.Buffer, value.Buffer, dwFlags);
 341
 342     RtlFreeUnicodeString(&title);
 343     RtlFreeUnicodeString(&root);
 344     RtlFreeUnicodeString(&subkey);
 345     RtlFreeUnicodeString(&value);
 346
 347     return hr;
 348 }
 349
 350 /***********************************************************************
 351  *          RegSaveRestoreW (advpack.@)
 352  *
 353  * Saves or restores the specified registry value.
 354  *
 355  * PARAMS
 356  *   hWnd           [I] Handle to the window used for the display.
 357  *   pszTitleString [I] Title of the window.
 358  *   hkBackupKey    [I] Key used to store the backup data.
 359  *   pcszRootKey    [I] Root key of the registry value
 360  *   pcszSubKey     [I] Sub key of the registry value.
 361  *   pcszValueName  [I] Value to save or restore.
 362  *   dwFlags        [I] See advpub.h.
 363  *
 364  * RETURNS
 365  *   Success: S_OK.
 366  *   Failure: E_FAIL.
 367  *
 368  * BUGS
 369  *   Unimplemented.
 370  */
 371 HRESULT WINAPI RegSaveRestoreW(HWND hWnd, LPCWSTR pszTitleString, HKEY hkBackupKey,
 372                                LPCWSTR pcszRootKey, LPCWSTR pcszSubKey,
 373                                LPCWSTR pcszValueName, DWORD dwFlags)
 374 {
 375     FIXME("(%p, %s, %p, %s, %s, %s, %ld): stub\n", hWnd, debugstr_w(pszTitleString),
 376           hkBackupKey, debugstr_w(pcszRootKey), debugstr_w(pcszSubKey),
 377           debugstr_w(pcszValueName), dwFlags);
 378
 379     return E_FAIL;
 380 }
 381
 382 /***********************************************************************
 383  *          RegSaveRestoreOnINFA (advpack.@)
 384  *
 385  * See RegSaveRestoreOnINFW.
 386  */
 387 HRESULT WINAPI RegSaveRestoreOnINFA(HWND hWnd, LPCSTR pszTitle, LPCSTR pszINF,
 388                                     LPCSTR pszSection, HKEY hHKLMBackKey,
 389                                     HKEY hHKCUBackKey, DWORD dwFlags)
 390 {
 391     UNICODE_STRING title, inf, section;
 392     HRESULT hr;
 393
 394     TRACE("(%p, %s, %s, %s, %p, %p, %ld)\n", hWnd, debugstr_a(pszTitle),
 395           debugstr_a(pszINF), debugstr_a(pszSection),
 396           hHKLMBackKey, hHKCUBackKey, dwFlags);
 397
 398     RtlCreateUnicodeStringFromAsciiz(&title, pszTitle);
 399     RtlCreateUnicodeStringFromAsciiz(&inf, pszINF);
 400     RtlCreateUnicodeStringFromAsciiz(&section, pszSection);
 401
 402     hr = RegSaveRestoreOnINFW(hWnd, title.Buffer, inf.Buffer, section.Buffer,
 403                               hHKLMBackKey, hHKCUBackKey, dwFlags);
 404
 405     RtlFreeUnicodeString(&title);
 406     RtlFreeUnicodeString(&inf);
 407     RtlFreeUnicodeString(&section);
 408
 409     return hr;
 410 }
 411
 412 /***********************************************************************
 413  *          RegSaveRestoreOnINFW (advpack.@)
 414  *
 415  * Saves or restores the specified INF Reg section.
 416  *
 417  * PARAMS
 418  *   hWnd         [I] Handle to the window used for the display.
 419  *   pszTitle     [I] Title of the window.
 420  *   pszINF       [I] Filename of the INF.
 421  *   pszSection   [I] Section to save or restore.
 422  *   hHKLMBackKey [I] Opened key in HKLM to store data.
 423  *   hHKCUBackKey [I] Opened key in HKCU to store data.
 424  *   dwFlags      [I] See advpub.h
 425  *
 426  * RETURNS
 427  *   Success: S_OK.
 428  *   Failure: E_FAIL.
 429  *
 430  * BUGS
 431  *   Unimplemented.
 432  */
 433 HRESULT WINAPI RegSaveRestoreOnINFW(HWND hWnd, LPCWSTR pszTitle, LPCWSTR pszINF,
 434                                     LPCWSTR pszSection, HKEY hHKLMBackKey,
 435                                     HKEY hHKCUBackKey, DWORD dwFlags)
 436 {
 437     FIXME("(%p, %s, %s, %s, %p, %p, %ld): stub\n", hWnd, debugstr_w(pszTitle),
 438           debugstr_w(pszINF), debugstr_w(pszSection),
 439           hHKLMBackKey, hHKCUBackKey, dwFlags);
 440
 441     return E_FAIL;
 442 }