secur32: Update ntlm_auth version detection to detect new samba4 version numbers.
[wine] / dlls / advpack / advpack.c
1 /*
2  * Advpack main
3  *
4  * Copyright 2004 Huw D M Davies
5  * Copyright 2005 Sami Aario
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21
22 #include <stdarg.h>
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winuser.h"
27 #include "winreg.h"
28 #include "winver.h"
29 #include "winternl.h"
30 #include "winnls.h"
31 #include "setupapi.h"
32 #include "advpub.h"
33 #include "wine/unicode.h"
34 #include "wine/debug.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(advpack);
37
38 typedef HRESULT (WINAPI *DLLREGISTER) (void);
39
40 #define MAX_FIELD_LENGTH    512
41 #define PREFIX_LEN          5
42
43 /* parses the destination directory parameters from pszSection
44  * the parameters are of the form: root,key,value,unknown,fallback
45  * we first read the reg value root\\key\\value and if that fails,
46  * use fallback as the destination directory
47  */
48 static void get_dest_dir(HINF hInf, PCWSTR pszSection, PWSTR pszBuffer, DWORD dwSize)
49 {
50     INFCONTEXT context;
51     WCHAR key[MAX_PATH], value[MAX_PATH];
52     WCHAR prefix[PREFIX_LEN];
53     HKEY root, subkey;
54     DWORD size;
55
56     static const WCHAR hklm[] = {'H','K','L','M',0};
57     static const WCHAR hkcu[] = {'H','K','C','U',0};
58
59     /* load the destination parameters */
60     SetupFindFirstLineW(hInf, pszSection, NULL, &context);
61     SetupGetStringFieldW(&context, 1, prefix, PREFIX_LEN, &size);
62     SetupGetStringFieldW(&context, 2, key, MAX_PATH, &size);
63     SetupGetStringFieldW(&context, 3, value, MAX_PATH, &size);
64
65     if (!lstrcmpW(prefix, hklm))
66         root = HKEY_LOCAL_MACHINE;
67     else if (!lstrcmpW(prefix, hkcu))
68         root = HKEY_CURRENT_USER;
69     else
70         root = NULL;
71
72     size = dwSize * sizeof(WCHAR);
73
74     /* fallback to the default destination dir if reg fails */
75     if (RegOpenKeyW(root, key, &subkey) ||
76         RegQueryValueExW(subkey, value, NULL, NULL, (LPBYTE)pszBuffer, &size))
77     {
78         SetupGetStringFieldW(&context, 6, pszBuffer, dwSize, NULL);
79     }
80
81     RegCloseKey(subkey);
82 }
83
84 /* loads the LDIDs specified in the install section of an INF */
85 static void set_ldids(HINF hInf, LPCWSTR pszInstallSection)
86 {
87     WCHAR field[MAX_FIELD_LENGTH];
88     WCHAR key[MAX_FIELD_LENGTH];
89     WCHAR dest[MAX_PATH];
90     INFCONTEXT context;
91     DWORD size;
92     int ldid;
93
94     static const WCHAR custDestW[] = {
95         'C','u','s','t','o','m','D','e','s','t','i','n','a','t','i','o','n',0
96     };
97
98     if (!SetupGetLineTextW(NULL, hInf, pszInstallSection, custDestW,
99                            field, MAX_FIELD_LENGTH, &size))
100         return;
101
102     if (!SetupFindFirstLineW(hInf, field, NULL, &context))
103         return;
104
105     do
106     {
107         SetupGetIntField(&context, 0, &ldid);
108         SetupGetLineTextW(&context, NULL, NULL, NULL,
109                           key, MAX_FIELD_LENGTH, &size);
110
111         get_dest_dir(hInf, key, dest, MAX_PATH);
112
113         SetupSetDirectoryIdW(hInf, ldid, dest);
114     } while (SetupFindNextLine(&context, &context));
115 }
116
117 /***********************************************************************
118  *           CloseINFEngine (ADVPACK.@)
119  *
120  * Closes a handle to an INF file opened with OpenINFEngine.
121  *
122  * PARAMS
123  *   hInf [I] Handle to the INF file to close.
124  *
125  * RETURNS
126  *   Success: S_OK.
127  *   Failure: E_FAIL.
128  */
129 HRESULT WINAPI CloseINFEngine(HINF hInf)
130 {
131     TRACE("(%p)\n", hInf);
132
133     if (!hInf)
134         return E_INVALIDARG;
135
136     SetupCloseInfFile(hInf);
137     return S_OK;
138 }
139
140 /***********************************************************************
141  *           DllMain (ADVPACK.@)
142  */
143 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
144 {
145     TRACE("(%p, %ld, %p)\n",hinstDLL, fdwReason, lpvReserved);
146
147     if (fdwReason == DLL_PROCESS_ATTACH)
148         DisableThreadLibraryCalls(hinstDLL);
149
150     return TRUE;
151 }
152
153 /***********************************************************************
154  *              IsNTAdmin       (ADVPACK.@)
155  *
156  * Checks if the user has admin privileges.
157  *
158  * PARAMS
159  *   reserved  [I] Reserved.  Must be 0.
160  *   pReserved [I] Reserved.  Must be NULL.
161  *
162  * RETURNS
163  *   TRUE if user has admin rights, FALSE otherwise.
164  */
165 BOOL WINAPI IsNTAdmin( DWORD reserved, LPDWORD pReserved )
166 {
167     SID_IDENTIFIER_AUTHORITY SidAuthority = {SECURITY_NT_AUTHORITY};
168     PTOKEN_GROUPS pTokenGroups;
169     BOOL bSidFound = FALSE;
170     DWORD dwSize, i;
171     HANDLE hToken;
172     PSID pSid;
173
174     TRACE("(0x%08lx, %p)\n", reserved, pReserved);
175
176     if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
177         return FALSE;
178
179     if (!GetTokenInformation(hToken, TokenGroups, NULL, 0, &dwSize))
180     {
181         if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
182         {
183             CloseHandle(hToken);
184             return FALSE;
185         }
186     }
187
188     pTokenGroups = HeapAlloc(GetProcessHeap(), 0, dwSize);
189     if (!pTokenGroups)
190     {
191         CloseHandle(hToken);
192         return FALSE;
193     }
194
195     if (!GetTokenInformation(hToken, TokenGroups, pTokenGroups, dwSize, &dwSize))
196     {
197         HeapFree(GetProcessHeap(), 0, pTokenGroups);
198         CloseHandle(hToken);
199         return FALSE;
200     }
201
202     CloseHandle(hToken);
203
204     if (!AllocateAndInitializeSid(&SidAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID,
205                                   DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &pSid))
206     {
207         HeapFree(GetProcessHeap(), 0, pTokenGroups);
208         return FALSE;
209     }
210
211     for (i = 0; i < pTokenGroups->GroupCount; i++)
212     {
213         if (EqualSid(pSid, pTokenGroups->Groups[i].Sid))
214         {
215             bSidFound = TRUE;
216             break;
217         }
218     }
219
220     HeapFree(GetProcessHeap(), 0, pTokenGroups);
221     FreeSid(pSid);
222
223     return bSidFound;
224 }
225
226 /***********************************************************************
227  *             NeedRebootInit  (ADVPACK.@)
228  *
229  * Sets up conditions for reboot checking.
230  *
231  * RETURNS
232  *   Value required by NeedReboot.
233  */
234 DWORD WINAPI NeedRebootInit(VOID)
235 {
236     FIXME("(): stub\n");
237     return 0;
238 }
239
240 /***********************************************************************
241  *             NeedReboot      (ADVPACK.@)
242  *
243  * Determines whether a reboot is required.
244  *
245  * PARAMS
246  *   dwRebootCheck [I] Value from NeedRebootInit.
247  *
248  * RETURNS
249  *   TRUE if a reboot is needed, FALSE otherwise.
250  *
251  * BUGS
252  *   Unimplemented.
253  */
254 BOOL WINAPI NeedReboot(DWORD dwRebootCheck)
255 {
256     FIXME("(0x%08lx): stub\n", dwRebootCheck);
257     return FALSE;
258 }
259
260 /***********************************************************************
261  *             OpenINFEngineA   (ADVPACK.@)
262  *
263  * See OpenINFEngineW.
264  */
265 HRESULT WINAPI OpenINFEngineA(LPCSTR pszInfFilename, LPCSTR pszInstallSection,
266                               DWORD dwFlags, HINF *phInf, PVOID pvReserved)
267 {
268     UNICODE_STRING filenameW, installW;
269     HRESULT res;
270
271     TRACE("(%p, %p, %ld, %p, %p)\n", pszInfFilename, pszInstallSection,
272           dwFlags, phInf, pvReserved);
273
274     if (!pszInfFilename || !phInf)
275         return E_INVALIDARG;
276
277     RtlCreateUnicodeStringFromAsciiz(&filenameW, pszInfFilename);
278     RtlCreateUnicodeStringFromAsciiz(&installW, pszInstallSection);
279
280     res = OpenINFEngineW(filenameW.Buffer, installW.Buffer,
281                          dwFlags, phInf, pvReserved);
282
283     RtlFreeUnicodeString(&filenameW);
284     RtlFreeUnicodeString(&installW);
285
286     return res;
287 }
288
289 /***********************************************************************
290  *             OpenINFEngineW   (ADVPACK.@)
291  *
292  * Opens and returns a handle to an INF file to be used by
293  * TranslateInfStringEx to continuously translate the INF file.
294  *
295  * PARAMS
296  *   pszInfFilename    [I] Filename of the INF to open.
297  *   pszInstallSection [I] Name of the Install section in the INF.
298  *   dwFlags           [I] See advpub.h.
299  *   phInf             [O] Handle to the loaded INF file.
300  *   pvReserved        [I] Reserved.  Must be NULL.
301  *
302  * RETURNS
303  *   Success: S_OK.
304  *   Failure: E_FAIL.
305  */
306 HRESULT WINAPI OpenINFEngineW(LPCWSTR pszInfFilename, LPCWSTR pszInstallSection,
307                               DWORD dwFlags, HINF *phInf, PVOID pvReserved)
308 {
309     TRACE("(%p, %p, %ld, %p, %p)\n", debugstr_w(pszInfFilename),
310           debugstr_w(pszInstallSection), dwFlags, phInf, pvReserved);
311
312     if (!pszInfFilename || !phInf)
313         return E_INVALIDARG;
314
315     *phInf = SetupOpenInfFileW(pszInfFilename, NULL, INF_STYLE_WIN4, NULL);
316     if (*phInf == INVALID_HANDLE_VALUE)
317         return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
318
319     set_ldids(*phInf, pszInstallSection);
320     
321     return S_OK;
322 }
323
324 /***********************************************************************
325  *             RebootCheckOnInstallA   (ADVPACK.@)
326  *
327  * Checks if a reboot is required for an installed INF section.
328  *
329  * PARAMS
330  *   hWnd       [I] Handle to the window used for messages.
331  *   pszINF     [I] Filename of the INF file.
332  *   pszSec     [I] INF section to check.
333  *   dwReserved [I] Reserved.  Must be 0.
334  *
335  * RETURNS
336  *   Success: S_OK - Reboot is needed if the INF section is installed.
337  *            S_FALSE - Reboot is not needed.
338  *   Failure: HRESULT of GetLastError().
339  *
340  * NOTES
341  *   if pszSec is NULL, RebootCheckOnInstall checks the DefaultInstall
342  *   or DefaultInstall.NT section.
343  *
344  * BUGS
345  *   Unimplemented.
346  */
347 HRESULT WINAPI RebootCheckOnInstallA(HWND hWnd, LPCSTR pszINF,
348                                     LPSTR pszSec, DWORD dwReserved)
349 {
350     FIXME("(%p, %p, %p, %ld) stub\n", hWnd, pszINF, pszSec, dwReserved);
351
352     return E_FAIL;
353 }
354
355 /***********************************************************************
356  *             RegisterOCX    (ADVPACK.@)
357  */
358 void WINAPI RegisterOCX( HWND hWnd, HINSTANCE hInst, LPCSTR cmdline, INT show )
359 {
360     WCHAR wszBuff[MAX_PATH];
361     WCHAR* pwcComma;
362     HMODULE hm;
363     DLLREGISTER pfnRegister;
364     HRESULT hr;
365
366     TRACE("(%s)\n", cmdline);
367
368     MultiByteToWideChar(CP_ACP, 0, cmdline, strlen(cmdline), wszBuff, MAX_PATH);
369     if ((pwcComma = strchrW( wszBuff, ',' ))) *pwcComma = 0;
370
371     TRACE("Parsed DLL name (%s)\n", debugstr_w(wszBuff));
372
373     hm = LoadLibraryExW(wszBuff, 0, LOAD_WITH_ALTERED_SEARCH_PATH);
374     if (!hm)
375     {
376         ERR("Couldn't load DLL: %s\n", debugstr_w(wszBuff));
377         return;
378     }
379
380     pfnRegister = (DLLREGISTER)GetProcAddress(hm, "DllRegisterServer");
381     if (pfnRegister == NULL)
382     {
383         ERR("DllRegisterServer entry point not found\n");
384     }
385     else
386     {
387         hr = pfnRegister();
388         if (hr != S_OK)
389         {
390             ERR("DllRegisterServer entry point returned %08lx\n", hr);
391         }
392     }
393
394     TRACE("Successfully registered OCX\n");
395
396     FreeLibrary(hm);
397 }
398
399 /***********************************************************************
400  *             SetPerUserSecValuesA   (ADVPACK.@)
401  *
402  * Prepares the per-user stub values under IsInstalled\{GUID} that
403  * control the per-user installation.
404  *
405  * PARAMS
406  *   pPerUser [I] Per-user stub values.
407  *
408  * RETURNS
409  *   Success: S_OK.
410  *   Failure: E_FAIL.
411  *
412  * BUGS
413  *   Unimplemented.
414  */
415 HRESULT WINAPI SetPerUserSecValuesA(PERUSERSECTIONA* pPerUser)
416 {
417     FIXME("(%p) stub\n", pPerUser);
418
419     return E_FAIL;
420 }
421
422 /***********************************************************************
423  *             TranslateInfStringA   (ADVPACK.@)
424  *
425  * See TranslateInfStringW.
426  */
427 HRESULT WINAPI TranslateInfStringA(LPCSTR pszInfFilename, LPCSTR pszInstallSection,
428                 LPCSTR pszTranslateSection, LPCSTR pszTranslateKey, LPSTR pszBuffer,
429                 DWORD dwBufferSize, PDWORD pdwRequiredSize, PVOID pvReserved)
430 {
431     UNICODE_STRING filenameW, installW;
432     UNICODE_STRING translateW, keyW;
433     LPWSTR bufferW;
434     HRESULT res;
435     DWORD len = 0;
436
437     TRACE("(%s %s %s %s %p %ld %p %p)\n",
438           debugstr_a(pszInfFilename), debugstr_a(pszInstallSection),
439           debugstr_a(pszTranslateSection), debugstr_a(pszTranslateKey),
440           pszBuffer, dwBufferSize,pdwRequiredSize, pvReserved);
441
442     if (!pszInfFilename || !pszTranslateSection ||
443         !pszTranslateKey || !pdwRequiredSize)
444         return E_INVALIDARG;
445
446     RtlCreateUnicodeStringFromAsciiz(&filenameW, pszInfFilename);
447     RtlCreateUnicodeStringFromAsciiz(&installW, pszInstallSection);
448     RtlCreateUnicodeStringFromAsciiz(&translateW, pszTranslateSection);
449     RtlCreateUnicodeStringFromAsciiz(&keyW, pszTranslateKey);
450
451     res = TranslateInfStringW(filenameW.Buffer, installW.Buffer,
452                               translateW.Buffer, keyW.Buffer, NULL,
453                               dwBufferSize, &len, NULL);
454
455     if (res == S_OK)
456     {
457         bufferW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
458
459         res = TranslateInfStringW(filenameW.Buffer, installW.Buffer,
460                                   translateW.Buffer, keyW.Buffer, bufferW,
461                                   len, &len, NULL);
462         if (res == S_OK)
463         {
464             *pdwRequiredSize = WideCharToMultiByte(CP_ACP, 0, bufferW, -1,
465                                                    NULL, 0, NULL, NULL);
466
467             if (dwBufferSize >= *pdwRequiredSize)
468             {
469                 WideCharToMultiByte(CP_ACP, 0, bufferW, -1, pszBuffer,
470                                     dwBufferSize, NULL, NULL);
471             }
472             else
473                 res = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
474         }
475         
476         HeapFree(GetProcessHeap(), 0, bufferW);
477     }
478
479     RtlFreeUnicodeString(&filenameW);
480     RtlFreeUnicodeString(&installW);
481     RtlFreeUnicodeString(&translateW);
482     RtlFreeUnicodeString(&keyW);
483
484     return res;
485 }
486
487 /***********************************************************************
488  *             TranslateInfStringW   (ADVPACK.@)
489  *
490  * Translates the value of a specified key in an inf file into the
491  * current locale by expanding string macros.
492  *
493  * PARAMS
494  *   pszInfFilename      [I] Filename of the inf file.
495  *   pszInstallSection   [I]
496  *   pszTranslateSection [I] Inf section where the key exists.
497  *   pszTranslateKey     [I] Key to translate.
498  *   pszBuffer           [O] Contains the translated string on exit.
499  *   dwBufferSize        [I] Size on input of pszBuffer.
500  *   pdwRequiredSize     [O] Length of the translated key.
501  *   pvReserved          [I] Reserved, must be NULL.
502  *
503  * RETURNS
504  *   Success: S_OK.
505  *   Failure: An hresult error code.
506  */
507 HRESULT WINAPI TranslateInfStringW(LPCWSTR pszInfFilename, LPCWSTR pszInstallSection,
508                 LPCWSTR pszTranslateSection, LPCWSTR pszTranslateKey, LPWSTR pszBuffer,
509                 DWORD dwBufferSize, PDWORD pdwRequiredSize, PVOID pvReserved)
510 {
511     HINF hInf;
512
513     TRACE("(%s %s %s %s %p %ld %p %p)\n",
514           debugstr_w(pszInfFilename), debugstr_w(pszInstallSection),
515           debugstr_w(pszTranslateSection), debugstr_w(pszTranslateKey),
516           pszBuffer, dwBufferSize,pdwRequiredSize, pvReserved);
517
518     if (!pszInfFilename || !pszTranslateSection ||
519         !pszTranslateKey || !pdwRequiredSize)
520         return E_INVALIDARG;
521
522     hInf = SetupOpenInfFileW(pszInfFilename, NULL, INF_STYLE_WIN4, NULL);
523     if (hInf == INVALID_HANDLE_VALUE)
524         return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
525
526     set_ldids(hInf, pszInstallSection);
527
528     if (!SetupGetLineTextW(NULL, hInf, pszTranslateSection, pszTranslateKey,
529                            pszBuffer, dwBufferSize, pdwRequiredSize))
530     {
531         if (dwBufferSize < *pdwRequiredSize)
532             return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
533
534         return SPAPI_E_LINE_NOT_FOUND;
535     }
536
537     SetupCloseInfFile(hInf);
538     return S_OK;
539 }
540
541 /***********************************************************************
542  *             TranslateInfStringExA   (ADVPACK.@)
543  *
544  * Using a handle to an INF file opened with OpenINFEngine, translates
545  * the value of a specified key in an inf file into the current locale
546  * by expanding string macros.
547  *
548  * PARAMS
549  *   hInf                [I] Handle to the INF file.
550  *   pszInfFilename      [I] Filename of the INF file.
551  *   pszTranslateSection [I] Inf section where the key exists.
552  *   pszTranslateKey     [I] Key to translate.
553  *   pszBuffer           [O] Contains the translated string on exit.
554  *   dwBufferSize        [I] Size on input of pszBuffer.
555  *   pdwRequiredSize     [O] Length of the translated key.
556  *   pvReserved          [I] Reserved.  Must be NULL.
557  *
558  * RETURNS
559  *   Success: S_OK.
560  *   Failure: E_FAIL.
561  *
562  * NOTES
563  *   To use TranslateInfStringEx to translate an INF file continuously,
564  *   open the INF file with OpenINFEngine, call TranslateInfStringEx as
565  *   many times as needed, then release the handle with CloseINFEngine.
566  *   When translating more than one keys, this method is more efficient
567  *   than calling TranslateInfString, because the INF file is only
568  *   opened once.
569  */
570 HRESULT WINAPI TranslateInfStringExA(HINF hInf, LPCSTR pszInfFilename,
571                                     LPCSTR pszTranslateSection, LPCSTR pszTranslateKey,
572                                     LPSTR pszBuffer, DWORD dwBufferSize,
573                                     PDWORD pdwRequiredSize, PVOID pvReserved)
574 {
575     TRACE("(%p, %p, %p, %p, %p, %ld, %p, %p)\n", hInf, pszInfFilename,
576           pszTranslateSection, pszTranslateKey, pszBuffer, dwBufferSize,
577           pdwRequiredSize, pvReserved);
578
579     if (!hInf || !pszInfFilename || !pszTranslateSection || !pszTranslateKey)
580         return E_INVALIDARG;
581
582     if (!SetupGetLineTextA(NULL, hInf, pszTranslateSection, pszTranslateKey,
583                            pszBuffer, dwBufferSize, pdwRequiredSize))
584     {
585         if (dwBufferSize < *pdwRequiredSize)
586             return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
587
588         return SPAPI_E_LINE_NOT_FOUND;
589     }
590
591     return S_OK;   
592 }
593
594 /***********************************************************************
595  *             UserInstStubWrapperA   (ADVPACK.@)
596  */
597 HRESULT WINAPI UserInstStubWrapperA(HWND hWnd, HINSTANCE hInstance,
598                                    LPSTR pszParms, INT nShow)
599 {
600     FIXME("(%p, %p, %p, %i) stub\n", hWnd, hInstance, pszParms, nShow);
601
602     return E_FAIL;
603 }
604
605 /***********************************************************************
606  *             UserUnInstStubWrapperA   (ADVPACK.@)
607  */
608 HRESULT WINAPI UserUnInstStubWrapperA(HWND hWnd, HINSTANCE hInstance,
609                                      LPSTR pszParms, INT nShow)
610 {
611     FIXME("(%p, %p, %p, %i) stub\n", hWnd, hInstance, pszParms, nShow);
612
613     return E_FAIL;
614 }