wininet: Allow a NULL value to be passed into HTTP_ProcessHeader
[wine] / dlls / setupapi / query.c
1 /*
2  * setupapi query functions
3  *
4  * Copyright 2006 James Hawkins
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
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winuser.h"
26 #include "winreg.h"
27 #include "winver.h"
28 #include "setupapi.h"
29 #include "advpub.h"
30 #include "winnls.h"
31 #include "wine/debug.h"
32 #include "wine/unicode.h"
33 #include "setupapi_private.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
36
37 #ifdef __i386__
38 static const WCHAR source_disks_names_platform[] =
39     {'S','o','u','r','c','e','D','i','s','k','s','N','a','m','e','s','.','x','8','6',0};
40 static const WCHAR source_disks_files_platform[] =
41     {'S','o','u','r','c','e','D','i','s','k','s','F','i','l','e','s','.','x','8','6',0};
42 #elif defined(__x86_64)
43 static const WCHAR source_disks_names_platform[] =
44     {'S','o','u','r','c','e','D','i','s','k','s','N','a','m','e','s','.','a','m','d','6','4',0};
45 static const WCHAR source_disks_files_platform[] =
46     {'S','o','u','r','c','e','D','i','s','k','s','F','i','l','e','s','.','a','m','d','6','4',0};
47 #else  /* FIXME: other platforms */
48 static const WCHAR source_disks_names_platform[] =
49     {'S','o','u','r','c','e','D','i','s','k','s','N','a','m','e','s',0};
50 static const WCHAR source_disks_files_platform[] =
51     {'S','o','u','r','c','e','D','i','s','k','s','F','i','l','e','s',0};
52 #endif
53 static const WCHAR source_disks_names[] =
54     {'S','o','u','r','c','e','D','i','s','k','s','N','a','m','e','s',0};
55 static const WCHAR source_disks_files[] =
56     {'S','o','u','r','c','e','D','i','s','k','s','F','i','l','e','s',0};
57
58 /* fills the PSP_INF_INFORMATION struct fill_info is TRUE
59  * always returns the required size of the information
60  */
61 static BOOL fill_inf_info(HINF inf, PSP_INF_INFORMATION buffer, DWORD size, DWORD *required)
62 {
63     LPCWSTR filename = PARSER_get_inf_filename(inf);
64     DWORD total_size = FIELD_OFFSET(SP_INF_INFORMATION, VersionData)
65                         + (lstrlenW(filename) + 1) * sizeof(WCHAR);
66
67     if (required) *required = total_size;
68
69     /* FIXME: we need to parse the INF file to find the correct version info */
70     if (buffer)
71     {
72         if (size < total_size)
73         {
74             SetLastError(ERROR_INSUFFICIENT_BUFFER);
75             return FALSE;
76         }
77         buffer->InfStyle = INF_STYLE_WIN4;
78         buffer->InfCount = 1;
79         /* put the filename in buffer->VersionData */
80         lstrcpyW((LPWSTR)&buffer->VersionData[0], filename);
81     }
82     return TRUE;
83 }
84
85 static HINF search_for_inf(LPCVOID InfSpec, DWORD SearchControl)
86 {
87     HINF hInf = INVALID_HANDLE_VALUE;
88     WCHAR inf_path[MAX_PATH];
89
90     static const WCHAR infW[] = {'\\','i','n','f','\\',0};
91     static const WCHAR system32W[] = {'\\','s','y','s','t','e','m','3','2','\\',0};
92
93     if (SearchControl == INFINFO_REVERSE_DEFAULT_SEARCH)
94     {
95         GetWindowsDirectoryW(inf_path, MAX_PATH);
96         lstrcatW(inf_path, system32W);
97         lstrcatW(inf_path, InfSpec);
98
99         hInf = SetupOpenInfFileW(inf_path, NULL,
100                                  INF_STYLE_OLDNT | INF_STYLE_WIN4, NULL);
101         if (hInf != INVALID_HANDLE_VALUE)
102             return hInf;
103
104         GetWindowsDirectoryW(inf_path, MAX_PATH);
105         lstrcpyW(inf_path, infW);
106         lstrcatW(inf_path, InfSpec);
107
108         return SetupOpenInfFileW(inf_path, NULL,
109                                  INF_STYLE_OLDNT | INF_STYLE_WIN4, NULL);
110     }
111
112     return INVALID_HANDLE_VALUE;
113 }
114
115 /***********************************************************************
116  *      SetupGetInfInformationA    (SETUPAPI.@)
117  *
118  */
119 BOOL WINAPI SetupGetInfInformationA(LPCVOID InfSpec, DWORD SearchControl,
120                                     PSP_INF_INFORMATION ReturnBuffer,
121                                     DWORD ReturnBufferSize, PDWORD RequiredSize)
122 {
123     LPWSTR inf = (LPWSTR)InfSpec;
124     DWORD len;
125     BOOL ret;
126
127     if (InfSpec && SearchControl >= INFINFO_INF_NAME_IS_ABSOLUTE)
128     {
129         len = lstrlenA(InfSpec) + 1;
130         inf = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
131         MultiByteToWideChar(CP_ACP, 0, InfSpec, -1, inf, len);
132     }
133
134     ret = SetupGetInfInformationW(inf, SearchControl, ReturnBuffer,
135                                   ReturnBufferSize, RequiredSize);
136
137     if (SearchControl >= INFINFO_INF_NAME_IS_ABSOLUTE)
138         HeapFree(GetProcessHeap(), 0, inf);
139
140     return ret;
141 }
142
143 /***********************************************************************
144  *      SetupGetInfInformationW    (SETUPAPI.@)
145  * 
146  * BUGS
147  *   Only handles the case when InfSpec is an INF handle.
148  */
149 BOOL WINAPI SetupGetInfInformationW(LPCVOID InfSpec, DWORD SearchControl,
150                                      PSP_INF_INFORMATION ReturnBuffer,
151                                      DWORD ReturnBufferSize, PDWORD RequiredSize)
152 {
153     HINF inf;
154     BOOL ret;
155     DWORD infSize;
156
157     TRACE("(%p, %d, %p, %d, %p)\n", InfSpec, SearchControl, ReturnBuffer,
158            ReturnBufferSize, RequiredSize);
159
160     if (!InfSpec)
161     {
162         if (SearchControl == INFINFO_INF_SPEC_IS_HINF)
163             SetLastError(ERROR_INVALID_HANDLE);
164         else
165             SetLastError(ERROR_INVALID_PARAMETER);
166
167         return FALSE;
168     }
169
170     switch (SearchControl)
171     {
172         case INFINFO_INF_SPEC_IS_HINF:
173             inf = (HINF)InfSpec;
174             break;
175         case INFINFO_INF_NAME_IS_ABSOLUTE:
176         case INFINFO_DEFAULT_SEARCH:
177             inf = SetupOpenInfFileW(InfSpec, NULL,
178                                     INF_STYLE_OLDNT | INF_STYLE_WIN4, NULL);
179             break;
180         case INFINFO_REVERSE_DEFAULT_SEARCH:
181             inf = search_for_inf(InfSpec, SearchControl);
182             break;
183         case INFINFO_INF_PATH_LIST_SEARCH:
184             FIXME("Unhandled search control: %d\n", SearchControl);
185
186             if (RequiredSize)
187                 *RequiredSize = 0;
188
189             return FALSE;
190         default:
191             SetLastError(ERROR_INVALID_PARAMETER);
192             return FALSE;
193     }
194
195     if (inf == INVALID_HANDLE_VALUE)
196     {
197         SetLastError(ERROR_FILE_NOT_FOUND);
198         return FALSE;
199     }
200
201     ret = fill_inf_info(inf, ReturnBuffer, ReturnBufferSize, &infSize);
202     if (!ReturnBuffer && (ReturnBufferSize >= infSize))
203     {
204         SetLastError(ERROR_INVALID_PARAMETER);
205         ret = FALSE;
206     }
207     if (RequiredSize) *RequiredSize = infSize;
208
209     if (SearchControl >= INFINFO_INF_NAME_IS_ABSOLUTE)
210         SetupCloseInfFile(inf);
211
212     return ret;
213 }
214
215 /***********************************************************************
216  *      SetupQueryInfFileInformationA    (SETUPAPI.@)
217  */
218 BOOL WINAPI SetupQueryInfFileInformationA(PSP_INF_INFORMATION InfInformation,
219                                           UINT InfIndex, PSTR ReturnBuffer,
220                                           DWORD ReturnBufferSize, PDWORD RequiredSize)
221 {
222     LPWSTR filenameW;
223     DWORD size;
224     BOOL ret;
225
226     ret = SetupQueryInfFileInformationW(InfInformation, InfIndex, NULL, 0, &size);
227     if (!ret)
228         return FALSE;
229
230     filenameW = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
231
232     ret = SetupQueryInfFileInformationW(InfInformation, InfIndex,
233                                         filenameW, size, &size);
234     if (!ret)
235     {
236         HeapFree(GetProcessHeap(), 0, filenameW);
237         return FALSE;
238     }
239
240     if (RequiredSize)
241         *RequiredSize = size;
242
243     if (!ReturnBuffer)
244     {
245         if (ReturnBufferSize)
246         {
247             SetLastError(ERROR_INVALID_PARAMETER);
248             return FALSE;
249         }
250
251         return TRUE;
252     }
253
254     if (size > ReturnBufferSize)
255     {
256         SetLastError(ERROR_INSUFFICIENT_BUFFER);
257         return FALSE;
258     }
259
260     WideCharToMultiByte(CP_ACP, 0, filenameW, -1, ReturnBuffer, size, NULL, NULL);
261     HeapFree(GetProcessHeap(), 0, filenameW);
262
263     return ret;
264 }
265
266 /***********************************************************************
267  *      SetupQueryInfFileInformationW    (SETUPAPI.@)
268  */
269 BOOL WINAPI SetupQueryInfFileInformationW(PSP_INF_INFORMATION InfInformation,
270                                           UINT InfIndex, PWSTR ReturnBuffer,
271                                           DWORD ReturnBufferSize, PDWORD RequiredSize) 
272 {
273     DWORD len;
274     LPWSTR ptr;
275
276     TRACE("(%p, %u, %p, %d, %p) Stub!\n", InfInformation, InfIndex,
277           ReturnBuffer, ReturnBufferSize, RequiredSize);
278
279     if (!InfInformation)
280     {
281         SetLastError(ERROR_INVALID_PARAMETER);
282         return FALSE;
283     }
284
285     if (InfIndex != 0)
286         FIXME("Appended INF files are not handled\n");
287
288     ptr = (LPWSTR)&InfInformation->VersionData[0];
289     len = lstrlenW(ptr);
290
291     if (RequiredSize)
292         *RequiredSize = len + 1;
293
294     if (!ReturnBuffer)
295         return TRUE;
296
297     if (ReturnBufferSize < len)
298     {
299         SetLastError(ERROR_INSUFFICIENT_BUFFER);
300         return FALSE;
301     }
302
303     lstrcpyW(ReturnBuffer, ptr);
304     return TRUE;
305 }
306
307 /***********************************************************************
308  *            SetupGetSourceFileLocationA   (SETUPAPI.@)
309  */
310
311 BOOL WINAPI SetupGetSourceFileLocationA( HINF hinf, PINFCONTEXT context, PCSTR filename,
312                                          PUINT source_id, PSTR buffer, DWORD buffer_size,
313                                          PDWORD required_size )
314 {
315     BOOL ret = FALSE;
316     WCHAR *filenameW = NULL, *bufferW = NULL;
317     DWORD required;
318     INT size;
319
320     TRACE("%p, %p, %s, %p, %p, 0x%08x, %p\n", hinf, context, debugstr_a(filename), source_id,
321           buffer, buffer_size, required_size);
322
323     if (filename && !(filenameW = strdupAtoW( filename )))
324         return FALSE;
325
326     if (!SetupGetSourceFileLocationW( hinf, context, filenameW, source_id, NULL, 0, &required ))
327         goto done;
328
329     if (!(bufferW = HeapAlloc( GetProcessHeap(), 0, required * sizeof(WCHAR) )))
330         goto done;
331
332     if (!SetupGetSourceFileLocationW( hinf, context, filenameW, source_id, bufferW, required, NULL ))
333         goto done;
334
335     size = WideCharToMultiByte( CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL );
336     if (required_size) *required_size = size;
337
338     if (buffer)
339     {
340         if (buffer_size >= size)
341             WideCharToMultiByte( CP_ACP, 0, bufferW, -1, buffer, buffer_size, NULL, NULL );
342         else
343         {
344             SetLastError( ERROR_INSUFFICIENT_BUFFER );
345             goto done;
346         }
347     }
348     ret = TRUE;
349
350  done:
351     HeapFree( GetProcessHeap(), 0, filenameW );
352     HeapFree( GetProcessHeap(), 0, bufferW );
353     return ret;
354 }
355
356 static LPWSTR get_source_id( HINF hinf, PINFCONTEXT context, PCWSTR filename )
357 {
358     DWORD size;
359     LPWSTR source_id;
360
361     if (!SetupFindFirstLineW( hinf, source_disks_files_platform, filename, context ) &&
362         !SetupFindFirstLineW( hinf, source_disks_files, filename, context ))
363         return NULL;
364
365     if (!SetupGetStringFieldW( context, 1, NULL, 0, &size ))
366         return NULL;
367
368     if (!(source_id = HeapAlloc( GetProcessHeap(), 0, size * sizeof(WCHAR) )))
369         return NULL;
370
371     if (!SetupGetStringFieldW( context, 1, source_id, size, NULL ))
372     {
373         HeapFree( GetProcessHeap(), 0, source_id );
374         return NULL;
375     }
376
377     if (!SetupFindFirstLineW( hinf, source_disks_names_platform, source_id, context ) &&
378         !SetupFindFirstLineW( hinf, source_disks_names, source_id, context ))
379     {
380         HeapFree( GetProcessHeap(), 0, source_id );
381         return NULL;
382     }
383     return source_id;
384 }
385
386 /***********************************************************************
387  *            SetupGetSourceFileLocationW   (SETUPAPI.@)
388  */
389
390 BOOL WINAPI SetupGetSourceFileLocationW( HINF hinf, PINFCONTEXT context, PCWSTR filename,
391                                          PUINT source_id, PWSTR buffer, DWORD buffer_size,
392                                          PDWORD required_size )
393 {
394     INFCONTEXT ctx;
395     WCHAR *end, *source_id_str;
396
397     TRACE("%p, %p, %s, %p, %p, 0x%08x, %p\n", hinf, context, debugstr_w(filename), source_id,
398           buffer, buffer_size, required_size);
399
400     if (!context) context = &ctx;
401
402     if (!(source_id_str = get_source_id( hinf, context, filename )))
403         return FALSE;
404
405     *source_id = strtolW( source_id_str, &end, 10 );
406     if (end == source_id_str || *end)
407     {
408         HeapFree( GetProcessHeap(), 0, source_id_str );
409         return FALSE;
410     }
411     HeapFree( GetProcessHeap(), 0, source_id_str );
412
413     if (SetupGetStringFieldW( context, 4, buffer, buffer_size, required_size ))
414         return TRUE;
415
416     if (required_size) *required_size = 1;
417     if (buffer)
418     {
419         if (buffer_size >= 1) buffer[0] = 0;
420         else
421         {
422             SetLastError( ERROR_INSUFFICIENT_BUFFER );
423             return FALSE;
424         }
425     }
426     return TRUE;
427 }
428
429 /***********************************************************************
430  *            SetupGetSourceInfoA  (SETUPAPI.@)
431  */
432
433 BOOL WINAPI SetupGetSourceInfoA( HINF hinf, UINT source_id, UINT info,
434                                  PSTR buffer, DWORD buffer_size, LPDWORD required_size )
435 {
436     BOOL ret = FALSE;
437     WCHAR *bufferW = NULL;
438     DWORD required;
439     INT size;
440
441     TRACE("%p, %d, %d, %p, %d, %p\n", hinf, source_id, info, buffer, buffer_size,
442           required_size);
443
444     if (!SetupGetSourceInfoW( hinf, source_id, info, NULL, 0, &required ))
445         return FALSE;
446
447     if (!(bufferW = HeapAlloc( GetProcessHeap(), 0, required * sizeof(WCHAR) )))
448         return FALSE;
449
450     if (!SetupGetSourceInfoW( hinf, source_id, info, bufferW, required, NULL ))
451         goto done;
452
453     size = WideCharToMultiByte( CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL );
454     if (required_size) *required_size = size;
455
456     if (buffer)
457     {
458         if (buffer_size >= size)
459             WideCharToMultiByte( CP_ACP, 0, bufferW, -1, buffer, buffer_size, NULL, NULL );
460         else
461         {
462             SetLastError( ERROR_INSUFFICIENT_BUFFER );
463             goto done;
464         }
465     }
466     ret = TRUE;
467
468  done:
469     HeapFree( GetProcessHeap(), 0, bufferW );
470     return ret;
471 }
472
473 /***********************************************************************
474  *            SetupGetSourceInfoW  (SETUPAPI.@)
475  */
476
477 BOOL WINAPI SetupGetSourceInfoW( HINF hinf, UINT source_id, UINT info,
478                                  PWSTR buffer, DWORD buffer_size, LPDWORD required_size )
479 {
480     INFCONTEXT ctx;
481     WCHAR source_id_str[11];
482     static const WCHAR fmt[] = {'%','d',0};
483     DWORD index;
484
485     TRACE("%p, %d, %d, %p, %d, %p\n", hinf, source_id, info, buffer, buffer_size,
486           required_size);
487
488     sprintfW( source_id_str, fmt, source_id );
489
490     if (!SetupFindFirstLineW( hinf, source_disks_names_platform, source_id_str, &ctx ) &&
491         !SetupFindFirstLineW( hinf, source_disks_names, source_id_str, &ctx ))
492         return FALSE;
493
494     switch (info)
495     {
496     case SRCINFO_PATH:          index = 4; break;
497     case SRCINFO_TAGFILE:       index = 2; break;
498     case SRCINFO_DESCRIPTION:   index = 1; break;
499     default:
500         WARN("unknown info level: %d\n", info);
501         return FALSE;
502     }
503
504     if (SetupGetStringFieldW( &ctx, index, buffer, buffer_size, required_size ))
505         return TRUE;
506
507     if (required_size) *required_size = 1;
508     if (buffer)
509     {
510         if (buffer_size >= 1) buffer[0] = 0;
511         else
512         {
513             SetLastError( ERROR_INSUFFICIENT_BUFFER );
514             return FALSE;
515         }
516     }
517     return TRUE;
518 }
519
520 /***********************************************************************
521  *            SetupGetTargetPathA   (SETUPAPI.@)
522  */
523
524 BOOL WINAPI SetupGetTargetPathA( HINF hinf, PINFCONTEXT context, PCSTR section, PSTR buffer,
525                                  DWORD buffer_size, PDWORD required_size )
526 {
527     BOOL ret = FALSE;
528     WCHAR *sectionW = NULL, *bufferW = NULL;
529     DWORD required;
530     INT size;
531
532     TRACE("%p, %p, %s, %p, 0x%08x, %p\n", hinf, context, debugstr_a(section), buffer,
533           buffer_size, required_size);
534
535     if (section && !(sectionW = strdupAtoW( section )))
536         return FALSE;
537
538     if (!SetupGetTargetPathW( hinf, context, sectionW, NULL, 0, &required ))
539         goto done;
540
541     if (!(bufferW = HeapAlloc( GetProcessHeap(), 0, required * sizeof(WCHAR) )))
542         goto done;
543
544     if (!SetupGetTargetPathW( hinf, context, sectionW, bufferW, required, NULL ))
545         goto done;
546
547     size = WideCharToMultiByte( CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL );
548     if (required_size) *required_size = size;
549
550     if (buffer)
551     {
552         if (buffer_size >= size)
553             WideCharToMultiByte( CP_ACP, 0, bufferW, -1, buffer, buffer_size, NULL, NULL );
554         else
555         {
556             SetLastError( ERROR_INSUFFICIENT_BUFFER );
557             goto done;
558         }
559     }
560     ret = TRUE;
561
562  done:
563     HeapFree( GetProcessHeap(), 0, sectionW );
564     HeapFree( GetProcessHeap(), 0, bufferW );
565     return ret;
566 }
567
568 /***********************************************************************
569  *            SetupGetTargetPathW   (SETUPAPI.@)
570  */
571
572 BOOL WINAPI SetupGetTargetPathW( HINF hinf, PINFCONTEXT context, PCWSTR section, PWSTR buffer,
573                                  DWORD buffer_size, PDWORD required_size )
574 {
575     static const WCHAR destination_dirs[] =
576         {'D','e','s','t','i','n','a','t','i','o','n','D','i','r','s',0};
577     static const WCHAR default_dest_dir[]  =
578         {'D','e','f','a','u','l','t','D','e','s','t','D','i','r',0};
579
580     INFCONTEXT ctx;
581     WCHAR *dir;
582     INT size;
583
584     TRACE("%p, %p, %s, %p, 0x%08x, %p\n", hinf, context, debugstr_w(section), buffer,
585           buffer_size, required_size);
586
587     if (context && !SetupFindFirstLineW( hinf, destination_dirs, NULL, context )) return FALSE;
588     else if (section && !SetupFindFirstLineW( hinf, section, NULL, &ctx )) return FALSE;
589     else if (!SetupFindFirstLineW( hinf, destination_dirs, default_dest_dir, &ctx )) return FALSE;
590
591     if (!(dir = PARSER_get_dest_dir( context ? context : &ctx ))) return FALSE;
592
593     size = lstrlenW( dir ) + 1;
594     if (required_size) *required_size = size;
595
596     if (buffer)
597     {
598         if (buffer_size >= size)
599             lstrcpyW( buffer, dir );
600         else
601         {
602             SetLastError( ERROR_INSUFFICIENT_BUFFER );
603             HeapFree( GetProcessHeap(), 0, dir );
604             return FALSE;
605         }
606     }
607     HeapFree( GetProcessHeap(), 0, dir );
608     return TRUE;
609 }