Add support for impersonating a token.
[wine/multimedia.git] / dlls / setupapi / parser.c
blob53c9e3968b815a4e0ceeb0baa6449f9456426891
1 /*
2 * INF file parsing
4 * Copyright 2002 Alexandre Julliard for CodeWeavers
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.
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.
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 #include "config.h"
22 #include "wine/port.h"
24 #include <assert.h>
25 #include <limits.h>
26 #include <stdarg.h>
27 #include <string.h>
28 #include <stdlib.h>
30 #include "windef.h"
31 #include "winbase.h"
32 #include "wingdi.h"
33 #include "winuser.h"
34 #include "winnls.h"
35 #include "winreg.h"
36 #include "winternl.h"
37 #include "winerror.h"
38 #include "setupapi.h"
39 #include "setupapi_private.h"
41 #include "wine/unicode.h"
42 #include "wine/debug.h"
44 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
46 #define CONTROL_Z '\x1a'
47 #define MAX_SECTION_NAME_LEN 255
48 #define MAX_FIELD_LEN 511 /* larger fields get silently truncated */
49 /* actual string limit is MAX_INF_STRING_LENGTH+1 (plus terminating null) under Windows */
50 #define MAX_STRING_LEN (MAX_INF_STRING_LENGTH+1)
52 /* inf file structure definitions */
54 struct field
56 const WCHAR *text; /* field text */
59 struct line
61 int first_field; /* index of first field in field array */
62 int nb_fields; /* number of fields in line */
63 int key_field; /* index of field for key or -1 if no key */
66 struct section
68 const WCHAR *name; /* section name */
69 unsigned int nb_lines; /* number of used lines */
70 unsigned int alloc_lines; /* total number of allocated lines in array below */
71 struct line lines[16]; /* lines information (grown dynamically, 16 is initial size) */
74 struct inf_file
76 struct inf_file *next; /* next appended file */
77 WCHAR *strings; /* buffer for string data (section names and field values) */
78 WCHAR *string_pos; /* position of next available string in buffer */
79 unsigned int nb_sections; /* number of used sections */
80 unsigned int alloc_sections; /* total number of allocated section pointers */
81 struct section **sections; /* section pointers array */
82 unsigned int nb_fields;
83 unsigned int alloc_fields;
84 struct field *fields;
85 int strings_section; /* index of [Strings] section or -1 if none */
86 WCHAR *src_root; /* source root directory */
89 /* parser definitions */
91 enum parser_state
93 LINE_START, /* at beginning of a line */
94 SECTION_NAME, /* parsing a section name */
95 KEY_NAME, /* parsing a key name */
96 VALUE_NAME, /* parsing a value name */
97 EOL_BACKSLASH, /* backslash at end of line */
98 QUOTES, /* inside quotes */
99 LEADING_SPACES, /* leading spaces */
100 TRAILING_SPACES, /* trailing spaces */
101 COMMENT, /* inside a comment */
102 NB_PARSER_STATES
105 struct parser
107 const WCHAR *start; /* start position of item being parsed */
108 const WCHAR *end; /* end of buffer */
109 struct inf_file *file; /* file being built */
110 enum parser_state state; /* current parser state */
111 enum parser_state stack[4]; /* state stack */
112 int stack_pos; /* current pos in stack */
114 int cur_section; /* index of section being parsed*/
115 struct line *line; /* current line */
116 unsigned int line_pos; /* current line position in file */
117 unsigned int error; /* error code */
118 unsigned int token_len; /* current token len */
119 WCHAR token[MAX_FIELD_LEN+1]; /* current token */
122 typedef const WCHAR * (*parser_state_func)( struct parser *parser, const WCHAR *pos );
124 /* parser state machine functions */
125 static const WCHAR *line_start_state( struct parser *parser, const WCHAR *pos );
126 static const WCHAR *section_name_state( struct parser *parser, const WCHAR *pos );
127 static const WCHAR *key_name_state( struct parser *parser, const WCHAR *pos );
128 static const WCHAR *value_name_state( struct parser *parser, const WCHAR *pos );
129 static const WCHAR *eol_backslash_state( struct parser *parser, const WCHAR *pos );
130 static const WCHAR *quotes_state( struct parser *parser, const WCHAR *pos );
131 static const WCHAR *leading_spaces_state( struct parser *parser, const WCHAR *pos );
132 static const WCHAR *trailing_spaces_state( struct parser *parser, const WCHAR *pos );
133 static const WCHAR *comment_state( struct parser *parser, const WCHAR *pos );
135 static const parser_state_func parser_funcs[NB_PARSER_STATES] =
137 line_start_state, /* LINE_START */
138 section_name_state, /* SECTION_NAME */
139 key_name_state, /* KEY_NAME */
140 value_name_state, /* VALUE_NAME */
141 eol_backslash_state, /* EOL_BACKSLASH */
142 quotes_state, /* QUOTES */
143 leading_spaces_state, /* LEADING_SPACES */
144 trailing_spaces_state, /* TRAILING_SPACES */
145 comment_state /* COMMENT */
149 /* Unicode string constants */
150 static const WCHAR Version[] = {'V','e','r','s','i','o','n',0};
151 static const WCHAR Signature[] = {'S','i','g','n','a','t','u','r','e',0};
152 static const WCHAR Chicago[] = {'$','C','h','i','c','a','g','o','$',0};
153 static const WCHAR WindowsNT[] = {'$','W','i','n','d','o','w','s',' ','N','T','$',0};
154 static const WCHAR Windows95[] = {'$','W','i','n','d','o','w','s',' ','9','5','$',0};
155 static const WCHAR LayoutFile[] = {'L','a','y','o','u','t','F','i','l','e',0};
157 /* extend an array, allocating more memory if necessary */
158 static void *grow_array( void *array, unsigned int *count, size_t elem )
160 void *new_array;
161 unsigned int new_count = *count + *count / 2;
162 if (new_count < 32) new_count = 32;
164 if (array)
165 new_array = HeapReAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, array, new_count * elem );
166 else
167 new_array = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, new_count * elem );
169 if (new_array)
170 *count = new_count;
171 else
172 HeapFree( GetProcessHeap(), 0, array );
173 return new_array;
177 /* find a section by name */
178 static int find_section( struct inf_file *file, const WCHAR *name )
180 unsigned int i;
182 for (i = 0; i < file->nb_sections; i++)
183 if (!strcmpiW( name, file->sections[i]->name )) return i;
184 return -1;
188 /* find a line by name */
189 static struct line *find_line( struct inf_file *file, int section_index, const WCHAR *name )
191 struct section *section;
192 struct line *line;
193 int i;
195 if (section_index < 0 || section_index >= file->nb_sections) return NULL;
196 section = file->sections[section_index];
197 for (i = 0, line = section->lines; i < section->nb_lines; i++, line++)
199 if (line->key_field == -1) continue;
200 if (!strcmpiW( name, file->fields[line->key_field].text )) return line;
202 return NULL;
206 /* add a section to the file and return the section index */
207 static int add_section( struct inf_file *file, const WCHAR *name )
209 struct section *section;
211 if (file->nb_sections >= file->alloc_sections)
213 if (!(file->sections = grow_array( file->sections, &file->alloc_sections,
214 sizeof(file->sections[0]) ))) return -1;
216 if (!(section = HeapAlloc( GetProcessHeap(), 0, sizeof(*section) ))) return -1;
217 section->name = name;
218 section->nb_lines = 0;
219 section->alloc_lines = sizeof(section->lines)/sizeof(section->lines[0]);
220 file->sections[file->nb_sections] = section;
221 return file->nb_sections++;
225 /* add a line to a given section */
226 static struct line *add_line( struct inf_file *file, int section_index )
228 struct section *section;
229 struct line *line;
231 assert( section_index >= 0 && section_index < file->nb_sections );
233 section = file->sections[section_index];
234 if (section->nb_lines == section->alloc_lines) /* need to grow the section */
236 int size = sizeof(*section) - sizeof(section->lines) + 2*section->alloc_lines*sizeof(*line);
237 if (!(section = HeapReAlloc( GetProcessHeap(), 0, section, size ))) return NULL;
238 section->alloc_lines *= 2;
239 file->sections[section_index] = section;
241 line = &section->lines[section->nb_lines++];
242 line->first_field = file->nb_fields;
243 line->nb_fields = 0;
244 line->key_field = -1;
245 return line;
249 /* retrieve a given line from section/line index */
250 inline static struct line *get_line( struct inf_file *file, unsigned int section_index,
251 unsigned int line_index )
253 struct section *section;
255 if (section_index >= file->nb_sections) return NULL;
256 section = file->sections[section_index];
257 if (line_index >= section->nb_lines) return NULL;
258 return &section->lines[line_index];
262 /* retrieve a given field from section/line/field index */
263 static struct field *get_field( struct inf_file *file, int section_index, int line_index,
264 int field_index )
266 struct line *line = get_line( file, section_index, line_index );
268 if (!line) return NULL;
269 if (!field_index) /* get the key */
271 if (line->key_field == -1) return NULL;
272 return &file->fields[line->key_field];
274 field_index--;
275 if (field_index >= line->nb_fields) return NULL;
276 return &file->fields[line->first_field + field_index];
280 /* allocate a new field, growing the array if necessary */
281 static struct field *add_field( struct inf_file *file, const WCHAR *text )
283 struct field *field;
285 if (file->nb_fields >= file->alloc_fields)
287 if (!(file->fields = grow_array( file->fields, &file->alloc_fields,
288 sizeof(file->fields[0]) ))) return NULL;
290 field = &file->fields[file->nb_fields++];
291 field->text = text;
292 return field;
296 /* retrieve the string substitution for a directory id */
297 static const WCHAR *get_dirid_subst( int dirid, unsigned int *len )
299 extern const WCHAR *DIRID_get_string( HINF hinf, int dirid );
300 const WCHAR *ret = DIRID_get_string( 0, dirid );
301 if (ret) *len = strlenW(ret);
302 return ret;
306 /* retrieve the string substitution for a given string, or NULL if not found */
307 /* if found, len is set to the substitution length */
308 static const WCHAR *get_string_subst( struct inf_file *file, const WCHAR *str, unsigned int *len )
310 static const WCHAR percent = '%';
312 struct section *strings_section;
313 struct line *line;
314 struct field *field;
315 unsigned int i;
316 int dirid;
317 WCHAR *dirid_str, *end;
318 const WCHAR *ret = NULL;
320 if (!*len) /* empty string (%%) is replaced by single percent */
322 *len = 1;
323 return &percent;
325 if (file->strings_section == -1) goto not_found;
326 strings_section = file->sections[file->strings_section];
327 for (i = 0, line = strings_section->lines; i < strings_section->nb_lines; i++, line++)
329 if (line->key_field == -1) continue;
330 if (strncmpiW( str, file->fields[line->key_field].text, *len )) continue;
331 if (!file->fields[line->key_field].text[*len]) break;
333 if (i == strings_section->nb_lines || !line->nb_fields) goto not_found;
334 field = &file->fields[line->first_field];
335 *len = strlenW( field->text );
336 return field->text;
338 not_found: /* check for integer id */
339 if ((dirid_str = HeapAlloc( GetProcessHeap(), 0, (*len+1) * sizeof(WCHAR) )))
341 memcpy( dirid_str, str, *len * sizeof(WCHAR) );
342 dirid_str[*len] = 0;
343 dirid = strtolW( dirid_str, &end, 10 );
344 if (!*end) ret = get_dirid_subst( dirid, len );
345 HeapFree( GetProcessHeap(), 0, dirid_str );
346 return ret;
348 return NULL;
352 /* do string substitutions on the specified text */
353 /* the buffer is assumed to be large enough */
354 /* returns necessary length not including terminating null */
355 unsigned int PARSER_string_substW( struct inf_file *file, const WCHAR *text, WCHAR *buffer,
356 unsigned int size )
358 const WCHAR *start, *subst, *p;
359 unsigned int len, total = 0;
360 int inside = 0;
362 if (!buffer) size = MAX_STRING_LEN + 1;
363 for (p = start = text; *p; p++)
365 if (*p != '%') continue;
366 inside = !inside;
367 if (inside) /* start of a %xx% string */
369 len = p - start;
370 if (len > size - 1) len = size - 1;
371 if (buffer) memcpy( buffer + total, start, len * sizeof(WCHAR) );
372 total += len;
373 size -= len;
374 start = p;
376 else /* end of the %xx% string, find substitution */
378 len = p - start - 1;
379 subst = get_string_subst( file, start + 1, &len );
380 if (!subst)
382 subst = start;
383 len = p - start + 1;
385 if (len > size - 1) len = size - 1;
386 if (buffer) memcpy( buffer + total, subst, len * sizeof(WCHAR) );
387 total += len;
388 size -= len;
389 start = p + 1;
393 if (start != p) /* unfinished string, copy it */
395 len = p - start;
396 if (len > size - 1) len = size - 1;
397 if (buffer) memcpy( buffer + total, start, len * sizeof(WCHAR) );
398 total += len;
400 if (buffer && size) buffer[total] = 0;
401 return total;
405 /* do string substitutions on the specified text */
406 /* the buffer is assumed to be large enough */
407 /* returns necessary length not including terminating null */
408 unsigned int PARSER_string_substA( struct inf_file *file, const WCHAR *text, char *buffer,
409 unsigned int size )
411 WCHAR buffW[MAX_STRING_LEN+1];
412 DWORD ret;
414 unsigned int len = PARSER_string_substW( file, text, buffW, sizeof(buffW)/sizeof(WCHAR) );
415 if (!buffer) RtlUnicodeToMultiByteSize( &ret, buffW, len * sizeof(WCHAR) );
416 else
418 RtlUnicodeToMultiByteN( buffer, size-1, &ret, buffW, len * sizeof(WCHAR) );
419 buffer[ret] = 0;
421 return ret;
425 /* push some string data into the strings buffer */
426 static WCHAR *push_string( struct inf_file *file, const WCHAR *string )
428 WCHAR *ret = file->string_pos;
429 strcpyW( ret, string );
430 file->string_pos += strlenW( ret ) + 1;
431 return ret;
435 /* push the current state on the parser stack */
436 inline static void push_state( struct parser *parser, enum parser_state state )
438 assert( parser->stack_pos < sizeof(parser->stack)/sizeof(parser->stack[0]) );
439 parser->stack[parser->stack_pos++] = state;
443 /* pop the current state */
444 inline static void pop_state( struct parser *parser )
446 assert( parser->stack_pos );
447 parser->state = parser->stack[--parser->stack_pos];
451 /* set the parser state and return the previous one */
452 inline static enum parser_state set_state( struct parser *parser, enum parser_state state )
454 enum parser_state ret = parser->state;
455 parser->state = state;
456 return ret;
460 /* check if the pointer points to an end of file */
461 inline static int is_eof( struct parser *parser, const WCHAR *ptr )
463 return (ptr >= parser->end || *ptr == CONTROL_Z);
467 /* check if the pointer points to an end of line */
468 inline static int is_eol( struct parser *parser, const WCHAR *ptr )
470 return (ptr >= parser->end || *ptr == CONTROL_Z || *ptr == '\n');
474 /* push data from current token start up to pos into the current token */
475 static int push_token( struct parser *parser, const WCHAR *pos )
477 int len = pos - parser->start;
478 const WCHAR *src = parser->start;
479 WCHAR *dst = parser->token + parser->token_len;
481 if (len > MAX_FIELD_LEN - parser->token_len) len = MAX_FIELD_LEN - parser->token_len;
483 parser->token_len += len;
484 for ( ; len > 0; len--, dst++, src++) *dst = *src ? *src : ' ';
485 *dst = 0;
486 parser->start = pos;
487 return 0;
491 /* add a section with the current token as name */
492 static int add_section_from_token( struct parser *parser )
494 int section_index;
496 if (parser->token_len > MAX_SECTION_NAME_LEN)
498 parser->error = ERROR_SECTION_NAME_TOO_LONG;
499 return -1;
501 if ((section_index = find_section( parser->file, parser->token )) == -1)
503 /* need to create a new one */
504 const WCHAR *name = push_string( parser->file, parser->token );
505 if ((section_index = add_section( parser->file, name )) == -1)
507 parser->error = ERROR_NOT_ENOUGH_MEMORY;
508 return -1;
511 parser->token_len = 0;
512 parser->cur_section = section_index;
513 return section_index;
517 /* add a field containing the current token to the current line */
518 static struct field *add_field_from_token( struct parser *parser, int is_key )
520 struct field *field;
521 WCHAR *text;
523 if (!parser->line) /* need to start a new line */
525 if (parser->cur_section == -1) /* got a line before the first section */
527 parser->error = ERROR_WRONG_INF_STYLE;
528 return NULL;
530 if (!(parser->line = add_line( parser->file, parser->cur_section ))) goto error;
532 else assert(!is_key);
534 text = push_string( parser->file, parser->token );
535 if ((field = add_field( parser->file, text )))
537 if (!is_key) parser->line->nb_fields++;
538 else
540 /* replace first field by key field */
541 parser->line->key_field = parser->line->first_field;
542 parser->line->first_field++;
544 parser->token_len = 0;
545 return field;
547 error:
548 parser->error = ERROR_NOT_ENOUGH_MEMORY;
549 return NULL;
553 /* close the current line and prepare for parsing a new one */
554 static void close_current_line( struct parser *parser )
556 struct line *cur_line = parser->line;
558 if (cur_line)
560 /* if line has a single field and no key, the field is the key too */
561 if (cur_line->nb_fields == 1 && cur_line->key_field == -1)
562 cur_line->key_field = cur_line->first_field;
564 parser->line = NULL;
568 /* handler for parser LINE_START state */
569 static const WCHAR *line_start_state( struct parser *parser, const WCHAR *pos )
571 const WCHAR *p;
573 for (p = pos; !is_eof( parser, p ); p++)
575 switch(*p)
577 case '\n':
578 parser->line_pos++;
579 close_current_line( parser );
580 break;
581 case ';':
582 push_state( parser, LINE_START );
583 set_state( parser, COMMENT );
584 return p + 1;
585 case '[':
586 parser->start = p + 1;
587 set_state( parser, SECTION_NAME );
588 return p + 1;
589 default:
590 if (!isspaceW(*p))
592 parser->start = p;
593 set_state( parser, KEY_NAME );
594 return p;
596 break;
599 close_current_line( parser );
600 return NULL;
604 /* handler for parser SECTION_NAME state */
605 static const WCHAR *section_name_state( struct parser *parser, const WCHAR *pos )
607 const WCHAR *p;
609 for (p = pos; !is_eol( parser, p ); p++)
611 if (*p == ']')
613 push_token( parser, p );
614 if (add_section_from_token( parser ) == -1) return NULL;
615 push_state( parser, LINE_START );
616 set_state( parser, COMMENT ); /* ignore everything else on the line */
617 return p + 1;
620 parser->error = ERROR_BAD_SECTION_NAME_LINE; /* unfinished section name */
621 return NULL;
625 /* handler for parser KEY_NAME state */
626 static const WCHAR *key_name_state( struct parser *parser, const WCHAR *pos )
628 const WCHAR *p, *token_end = parser->start;
630 for (p = pos; !is_eol( parser, p ); p++)
632 if (*p == ',') break;
633 switch(*p)
636 case '=':
637 push_token( parser, token_end );
638 if (!add_field_from_token( parser, 1 )) return NULL;
639 parser->start = p + 1;
640 push_state( parser, VALUE_NAME );
641 set_state( parser, LEADING_SPACES );
642 return p + 1;
643 case ';':
644 push_token( parser, token_end );
645 if (!add_field_from_token( parser, 0 )) return NULL;
646 push_state( parser, LINE_START );
647 set_state( parser, COMMENT );
648 return p + 1;
649 case '"':
650 push_token( parser, token_end );
651 parser->start = p + 1;
652 push_state( parser, KEY_NAME );
653 set_state( parser, QUOTES );
654 return p + 1;
655 case '\\':
656 push_token( parser, token_end );
657 parser->start = p;
658 push_state( parser, KEY_NAME );
659 set_state( parser, EOL_BACKSLASH );
660 return p;
661 default:
662 if (!isspaceW(*p)) token_end = p + 1;
663 else
665 push_token( parser, p );
666 push_state( parser, KEY_NAME );
667 set_state( parser, TRAILING_SPACES );
668 return p;
670 break;
673 push_token( parser, token_end );
674 set_state( parser, VALUE_NAME );
675 return p;
679 /* handler for parser VALUE_NAME state */
680 static const WCHAR *value_name_state( struct parser *parser, const WCHAR *pos )
682 const WCHAR *p, *token_end = parser->start;
684 for (p = pos; !is_eol( parser, p ); p++)
686 switch(*p)
688 case ';':
689 push_token( parser, token_end );
690 if (!add_field_from_token( parser, 0 )) return NULL;
691 push_state( parser, LINE_START );
692 set_state( parser, COMMENT );
693 return p + 1;
694 case ',':
695 push_token( parser, token_end );
696 if (!add_field_from_token( parser, 0 )) return NULL;
697 parser->start = p + 1;
698 push_state( parser, VALUE_NAME );
699 set_state( parser, LEADING_SPACES );
700 return p + 1;
701 case '"':
702 push_token( parser, token_end );
703 parser->start = p + 1;
704 push_state( parser, VALUE_NAME );
705 set_state( parser, QUOTES );
706 return p + 1;
707 case '\\':
708 push_token( parser, token_end );
709 parser->start = p;
710 push_state( parser, VALUE_NAME );
711 set_state( parser, EOL_BACKSLASH );
712 return p;
713 default:
714 if (!isspaceW(*p)) token_end = p + 1;
715 else
717 push_token( parser, p );
718 push_state( parser, VALUE_NAME );
719 set_state( parser, TRAILING_SPACES );
720 return p;
722 break;
725 push_token( parser, token_end );
726 if (!add_field_from_token( parser, 0 )) return NULL;
727 set_state( parser, LINE_START );
728 return p;
732 /* handler for parser EOL_BACKSLASH state */
733 static const WCHAR *eol_backslash_state( struct parser *parser, const WCHAR *pos )
735 const WCHAR *p;
737 for (p = pos; !is_eof( parser, p ); p++)
739 switch(*p)
741 case '\n':
742 parser->line_pos++;
743 parser->start = p + 1;
744 set_state( parser, LEADING_SPACES );
745 return p + 1;
746 case '\\':
747 continue;
748 case ';':
749 push_state( parser, EOL_BACKSLASH );
750 set_state( parser, COMMENT );
751 return p + 1;
752 default:
753 if (isspaceW(*p)) continue;
754 push_token( parser, p );
755 pop_state( parser );
756 return p;
759 parser->start = p;
760 pop_state( parser );
761 return p;
765 /* handler for parser QUOTES state */
766 static const WCHAR *quotes_state( struct parser *parser, const WCHAR *pos )
768 const WCHAR *p, *token_end = parser->start;
770 for (p = pos; !is_eol( parser, p ); p++)
772 if (*p == '"')
774 if (p+1 < parser->end && p[1] == '"') /* double quotes */
776 push_token( parser, p + 1 );
777 parser->start = token_end = p + 2;
778 p++;
780 else /* end of quotes */
782 push_token( parser, p );
783 parser->start = p + 1;
784 pop_state( parser );
785 return p + 1;
789 push_token( parser, p );
790 pop_state( parser );
791 return p;
795 /* handler for parser LEADING_SPACES state */
796 static const WCHAR *leading_spaces_state( struct parser *parser, const WCHAR *pos )
798 const WCHAR *p;
800 for (p = pos; !is_eol( parser, p ); p++)
802 if (*p == '\\')
804 parser->start = p;
805 set_state( parser, EOL_BACKSLASH );
806 return p;
808 if (!isspaceW(*p)) break;
810 parser->start = p;
811 pop_state( parser );
812 return p;
816 /* handler for parser TRAILING_SPACES state */
817 static const WCHAR *trailing_spaces_state( struct parser *parser, const WCHAR *pos )
819 const WCHAR *p;
821 for (p = pos; !is_eol( parser, p ); p++)
823 if (*p == '\\')
825 set_state( parser, EOL_BACKSLASH );
826 return p;
828 if (!isspaceW(*p)) break;
830 pop_state( parser );
831 return p;
835 /* handler for parser COMMENT state */
836 static const WCHAR *comment_state( struct parser *parser, const WCHAR *pos )
838 const WCHAR *p = pos;
840 while (!is_eol( parser, p )) p++;
841 pop_state( parser );
842 return p;
846 /* parse a complete buffer */
847 static DWORD parse_buffer( struct inf_file *file, const WCHAR *buffer, const WCHAR *end,
848 UINT *error_line )
850 static const WCHAR Strings[] = {'S','t','r','i','n','g','s',0};
852 struct parser parser;
853 const WCHAR *pos = buffer;
855 parser.start = buffer;
856 parser.end = end;
857 parser.file = file;
858 parser.line = NULL;
859 parser.state = LINE_START;
860 parser.stack_pos = 0;
861 parser.cur_section = -1;
862 parser.line_pos = 1;
863 parser.error = 0;
864 parser.token_len = 0;
866 /* parser main loop */
867 while (pos) pos = (parser_funcs[parser.state])( &parser, pos );
869 /* trim excess buffer space */
870 if (file->alloc_sections > file->nb_sections)
872 file->sections = HeapReAlloc( GetProcessHeap(), 0, file->sections,
873 file->nb_sections * sizeof(file->sections[0]) );
874 file->alloc_sections = file->nb_sections;
876 if (file->alloc_fields > file->nb_fields)
878 file->fields = HeapReAlloc( GetProcessHeap(), 0, file->fields,
879 file->nb_fields * sizeof(file->fields[0]) );
880 file->alloc_fields = file->nb_fields;
882 file->strings = HeapReAlloc( GetProcessHeap(), HEAP_REALLOC_IN_PLACE_ONLY, file->strings,
883 (file->string_pos - file->strings) * sizeof(WCHAR) );
885 if (parser.error)
887 if (error_line) *error_line = parser.line_pos;
888 return parser.error;
891 /* find the [strings] section */
892 file->strings_section = find_section( file, Strings );
893 return 0;
897 /* append a child INF file to its parent list, in a thread-safe manner */
898 static void append_inf_file( struct inf_file *parent, struct inf_file *child )
900 struct inf_file **ppnext = &parent->next;
901 child->next = NULL;
903 for (;;)
905 struct inf_file *next = InterlockedCompareExchangePointer( (void **)ppnext, child, NULL );
906 if (!next) return;
907 ppnext = &next->next;
912 /***********************************************************************
913 * parse_file
915 * parse an INF file.
917 static struct inf_file *parse_file( HANDLE handle, const WCHAR *class, UINT *error_line )
919 void *buffer;
920 DWORD err = 0;
921 struct inf_file *file;
923 DWORD size = GetFileSize( handle, NULL );
924 HANDLE mapping = CreateFileMappingW( handle, NULL, PAGE_READONLY, 0, size, NULL );
925 if (!mapping) return NULL;
926 buffer = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, size );
927 NtClose( mapping );
928 if (!buffer) return NULL;
930 if (class) FIXME( "class %s not supported yet\n", debugstr_w(class) );
932 if (!(file = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*file) )))
934 err = ERROR_NOT_ENOUGH_MEMORY;
935 goto done;
938 /* we won't need more strings space than the size of the file,
939 * so we can preallocate it here
941 if (!(file->strings = HeapAlloc( GetProcessHeap(), 0, size * sizeof(WCHAR) )))
943 err = ERROR_NOT_ENOUGH_MEMORY;
944 goto done;
946 file->string_pos = file->strings;
947 file->strings_section = -1;
949 if (!RtlIsTextUnicode( buffer, size, NULL ))
951 WCHAR *new_buff = HeapAlloc( GetProcessHeap(), 0, size * sizeof(WCHAR) );
952 if (new_buff)
954 DWORD len = MultiByteToWideChar( CP_ACP, 0, buffer, size, new_buff,
955 size * sizeof(WCHAR) );
956 err = parse_buffer( file, new_buff, new_buff + len, error_line );
957 HeapFree( GetProcessHeap(), 0, new_buff );
960 else err = parse_buffer( file, buffer, (WCHAR *)((char *)buffer + size), error_line );
962 if (!err) /* now check signature */
964 int version_index = find_section( file, Version );
965 if (version_index != -1)
967 struct line *line = find_line( file, version_index, Signature );
968 if (line && line->nb_fields > 0)
970 struct field *field = file->fields + line->first_field;
971 if (!strcmpiW( field->text, Chicago )) goto done;
972 if (!strcmpiW( field->text, WindowsNT )) goto done;
973 if (!strcmpiW( field->text, Windows95 )) goto done;
976 err = ERROR_WRONG_INF_STYLE;
979 done:
980 UnmapViewOfFile( buffer );
981 if (err)
983 HeapFree( GetProcessHeap(), 0, file );
984 SetLastError( err );
985 file = NULL;
987 return file;
991 /***********************************************************************
992 * PARSER_get_src_root
994 * Retrieve the source directory of an inf file.
996 const WCHAR *PARSER_get_src_root( HINF hinf )
998 struct inf_file *file = hinf;
999 return file->src_root;
1003 /***********************************************************************
1004 * PARSER_get_dest_dir
1006 * retrieve a destination dir of the form "dirid,relative_path" in the given entry.
1007 * returned buffer must be freed by caller.
1009 WCHAR *PARSER_get_dest_dir( INFCONTEXT *context )
1011 const WCHAR *dir;
1012 WCHAR *ptr, *ret;
1013 INT dirid;
1014 DWORD len1, len2;
1016 if (!SetupGetIntField( context, 1, &dirid )) return NULL;
1017 if (!(dir = DIRID_get_string( context->Inf, dirid ))) return NULL;
1018 len1 = strlenW(dir) + 1;
1019 if (!SetupGetStringFieldW( context, 2, NULL, 0, &len2 )) len2 = 0;
1020 if (!(ret = HeapAlloc( GetProcessHeap(), 0, (len1+len2) * sizeof(WCHAR) ))) return NULL;
1021 strcpyW( ret, dir );
1022 ptr = ret + strlenW(ret);
1023 if (len2 && ptr > ret && ptr[-1] != '\\') *ptr++ = '\\';
1024 if (!SetupGetStringFieldW( context, 2, ptr, len2, NULL )) *ptr = 0;
1025 return ret;
1029 /***********************************************************************
1030 * SetupOpenInfFileA (SETUPAPI.@)
1032 HINF WINAPI SetupOpenInfFileA( PCSTR name, PCSTR class, DWORD style, UINT *error )
1034 UNICODE_STRING nameW, classW;
1035 HINF ret = (HINF)INVALID_HANDLE_VALUE;
1037 classW.Buffer = NULL;
1038 if (class && !RtlCreateUnicodeStringFromAsciiz( &classW, class ))
1040 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1041 return ret;
1043 if (RtlCreateUnicodeStringFromAsciiz( &nameW, name ))
1045 ret = SetupOpenInfFileW( nameW.Buffer, classW.Buffer, style, error );
1046 RtlFreeUnicodeString( &nameW );
1048 RtlFreeUnicodeString( &classW );
1049 return ret;
1053 /***********************************************************************
1054 * SetupOpenInfFileW (SETUPAPI.@)
1056 HINF WINAPI SetupOpenInfFileW( PCWSTR name, PCWSTR class, DWORD style, UINT *error )
1058 struct inf_file *file = NULL;
1059 HANDLE handle;
1060 WCHAR *path, *p;
1061 UINT len;
1063 if (strchrW( name, '\\' ) || strchrW( name, '/' ))
1065 if (!(len = GetFullPathNameW( name, 0, NULL, NULL ))) return (HINF)INVALID_HANDLE_VALUE;
1066 if (!(path = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
1068 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1069 return (HINF)INVALID_HANDLE_VALUE;
1071 GetFullPathNameW( name, len, path, NULL );
1072 handle = CreateFileW( path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0 );
1074 else /* try Windows directory */
1076 static const WCHAR Inf[] = {'\\','i','n','f','\\',0};
1077 static const WCHAR System32[] = {'\\','s','y','s','t','e','m','3','2','\\',0};
1079 len = GetWindowsDirectoryW( NULL, 0 ) + strlenW(name) + 12;
1080 if (!(path = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
1082 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1083 return (HINF)INVALID_HANDLE_VALUE;
1085 GetWindowsDirectoryW( path, len );
1086 p = path + strlenW(path);
1087 strcpyW( p, Inf );
1088 strcatW( p, name );
1089 handle = CreateFileW( path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0 );
1090 if (handle == INVALID_HANDLE_VALUE)
1092 strcpyW( p, System32 );
1093 strcatW( p, name );
1094 handle = CreateFileW( path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0 );
1098 if (handle != INVALID_HANDLE_VALUE)
1100 file = parse_file( handle, class, error );
1101 CloseHandle( handle );
1103 if (!file)
1105 HeapFree( GetProcessHeap(), 0, path );
1106 return (HINF)INVALID_HANDLE_VALUE;
1108 TRACE( "%s -> %p\n", debugstr_w(path), file );
1109 file->src_root = path;
1110 if ((p = strrchrW( path, '\\' ))) p[1] = 0; /* remove file name */
1111 SetLastError( 0 );
1112 return (HINF)file;
1116 /***********************************************************************
1117 * SetupOpenAppendInfFileA (SETUPAPI.@)
1119 BOOL WINAPI SetupOpenAppendInfFileA( PCSTR name, HINF parent_hinf, UINT *error )
1121 HINF child_hinf;
1123 if (!name) return SetupOpenAppendInfFileW( NULL, parent_hinf, error );
1124 child_hinf = SetupOpenInfFileA( name, NULL, INF_STYLE_WIN4, error );
1125 if (child_hinf == (HINF)INVALID_HANDLE_VALUE) return FALSE;
1126 append_inf_file( parent_hinf, child_hinf );
1127 TRACE( "%p: appended %s (%p)\n", parent_hinf, debugstr_a(name), child_hinf );
1128 return TRUE;
1132 /***********************************************************************
1133 * SetupOpenAppendInfFileW (SETUPAPI.@)
1135 BOOL WINAPI SetupOpenAppendInfFileW( PCWSTR name, HINF parent_hinf, UINT *error )
1137 HINF child_hinf;
1139 if (!name)
1141 INFCONTEXT context;
1142 WCHAR filename[MAX_PATH];
1143 int idx = 1;
1145 if (!SetupFindFirstLineW( parent_hinf, Version, LayoutFile, &context )) return FALSE;
1146 while (SetupGetStringFieldW( &context, idx++, filename,
1147 sizeof(filename)/sizeof(WCHAR), NULL ))
1149 child_hinf = SetupOpenInfFileW( filename, NULL, INF_STYLE_WIN4, error );
1150 if (child_hinf == (HINF)INVALID_HANDLE_VALUE) return FALSE;
1151 append_inf_file( parent_hinf, child_hinf );
1152 TRACE( "%p: appended %s (%p)\n", parent_hinf, debugstr_w(filename), child_hinf );
1154 return TRUE;
1156 child_hinf = SetupOpenInfFileW( name, NULL, INF_STYLE_WIN4, error );
1157 if (child_hinf == (HINF)INVALID_HANDLE_VALUE) return FALSE;
1158 append_inf_file( parent_hinf, child_hinf );
1159 TRACE( "%p: appended %s (%p)\n", parent_hinf, debugstr_w(name), child_hinf );
1160 return TRUE;
1164 /***********************************************************************
1165 * SetupOpenMasterInf (SETUPAPI.@)
1167 HINF WINAPI SetupOpenMasterInf( VOID )
1169 static const WCHAR Layout[] = {'\\','i','n','f','\\', 'l', 'a', 'y', 'o', 'u', 't', '.', 'i', 'n', 'f', 0};
1170 WCHAR Buffer[MAX_PATH];
1172 GetWindowsDirectoryW( Buffer, MAX_PATH );
1173 strcatW( Buffer, Layout );
1174 return SetupOpenInfFileW( Buffer, NULL, INF_STYLE_WIN4, NULL);
1179 /***********************************************************************
1180 * SetupCloseInfFile (SETUPAPI.@)
1182 void WINAPI SetupCloseInfFile( HINF hinf )
1184 struct inf_file *file = hinf;
1185 unsigned int i;
1187 for (i = 0; i < file->nb_sections; i++) HeapFree( GetProcessHeap(), 0, file->sections[i] );
1188 HeapFree( GetProcessHeap(), 0, file->src_root );
1189 HeapFree( GetProcessHeap(), 0, file->sections );
1190 HeapFree( GetProcessHeap(), 0, file->fields );
1191 HeapFree( GetProcessHeap(), 0, file->strings );
1192 HeapFree( GetProcessHeap(), 0, file );
1196 /***********************************************************************
1197 * SetupGetLineCountA (SETUPAPI.@)
1199 LONG WINAPI SetupGetLineCountA( HINF hinf, PCSTR name )
1201 UNICODE_STRING sectionW;
1202 LONG ret = -1;
1204 if (!RtlCreateUnicodeStringFromAsciiz( &sectionW, name ))
1205 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1206 else
1208 ret = SetupGetLineCountW( hinf, sectionW.Buffer );
1209 RtlFreeUnicodeString( &sectionW );
1211 return ret;
1215 /***********************************************************************
1216 * SetupGetLineCountW (SETUPAPI.@)
1218 LONG WINAPI SetupGetLineCountW( HINF hinf, PCWSTR section )
1220 struct inf_file *file = hinf;
1221 int section_index;
1222 LONG ret = -1;
1224 for (file = hinf; file; file = file->next)
1226 if ((section_index = find_section( file, section )) == -1) continue;
1227 if (ret == -1) ret = 0;
1228 ret += file->sections[section_index]->nb_lines;
1230 TRACE( "(%p,%s) returning %ld\n", hinf, debugstr_w(section), ret );
1231 SetLastError( (ret == -1) ? ERROR_SECTION_NOT_FOUND : 0 );
1232 return ret;
1236 /***********************************************************************
1237 * SetupGetLineByIndexA (SETUPAPI.@)
1239 BOOL WINAPI SetupGetLineByIndexA( HINF hinf, PCSTR section, DWORD index, INFCONTEXT *context )
1241 UNICODE_STRING sectionW;
1242 BOOL ret = FALSE;
1244 if (!RtlCreateUnicodeStringFromAsciiz( &sectionW, section ))
1245 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1246 else
1248 ret = SetupGetLineByIndexW( hinf, sectionW.Buffer, index, context );
1249 RtlFreeUnicodeString( &sectionW );
1251 return ret;
1255 /***********************************************************************
1256 * SetupGetLineByIndexW (SETUPAPI.@)
1258 BOOL WINAPI SetupGetLineByIndexW( HINF hinf, PCWSTR section, DWORD index, INFCONTEXT *context )
1260 struct inf_file *file = hinf;
1261 int section_index;
1263 SetLastError( ERROR_SECTION_NOT_FOUND );
1264 for (file = hinf; file; file = file->next)
1266 if ((section_index = find_section( file, section )) == -1) continue;
1267 SetLastError( ERROR_LINE_NOT_FOUND );
1268 if (index < file->sections[section_index]->nb_lines)
1270 context->Inf = hinf;
1271 context->CurrentInf = file;
1272 context->Section = section_index;
1273 context->Line = index;
1274 SetLastError( 0 );
1275 TRACE( "(%p,%s): returning %d/%ld\n",
1276 hinf, debugstr_w(section), section_index, index );
1277 return TRUE;
1279 index -= file->sections[section_index]->nb_lines;
1281 TRACE( "(%p,%s) not found\n", hinf, debugstr_w(section) );
1282 return FALSE;
1286 /***********************************************************************
1287 * SetupFindFirstLineA (SETUPAPI.@)
1289 BOOL WINAPI SetupFindFirstLineA( HINF hinf, PCSTR section, PCSTR key, INFCONTEXT *context )
1291 UNICODE_STRING sectionW, keyW;
1292 BOOL ret = FALSE;
1294 if (!RtlCreateUnicodeStringFromAsciiz( &sectionW, section ))
1296 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1297 return FALSE;
1300 if (!key) ret = SetupFindFirstLineW( hinf, sectionW.Buffer, NULL, context );
1301 else
1303 if (RtlCreateUnicodeStringFromAsciiz( &keyW, key ))
1305 ret = SetupFindFirstLineW( hinf, sectionW.Buffer, keyW.Buffer, context );
1306 RtlFreeUnicodeString( &keyW );
1308 else SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1310 RtlFreeUnicodeString( &sectionW );
1311 return ret;
1315 /***********************************************************************
1316 * SetupFindFirstLineW (SETUPAPI.@)
1318 BOOL WINAPI SetupFindFirstLineW( HINF hinf, PCWSTR section, PCWSTR key, INFCONTEXT *context )
1320 struct inf_file *file;
1321 int section_index;
1323 SetLastError( ERROR_SECTION_NOT_FOUND );
1324 for (file = hinf; file; file = file->next)
1326 if ((section_index = find_section( file, section )) == -1) continue;
1327 if (key)
1329 INFCONTEXT ctx;
1330 ctx.Inf = hinf;
1331 ctx.CurrentInf = file;
1332 ctx.Section = section_index;
1333 ctx.Line = -1;
1334 return SetupFindNextMatchLineW( &ctx, key, context );
1336 SetLastError( ERROR_LINE_NOT_FOUND ); /* found at least one section */
1337 if (file->sections[section_index]->nb_lines)
1339 context->Inf = hinf;
1340 context->CurrentInf = file;
1341 context->Section = section_index;
1342 context->Line = 0;
1343 SetLastError( 0 );
1344 TRACE( "(%p,%s,%s): returning %d/0\n",
1345 hinf, debugstr_w(section), debugstr_w(key), section_index );
1346 return TRUE;
1349 TRACE( "(%p,%s,%s): not found\n", hinf, debugstr_w(section), debugstr_w(key) );
1350 return FALSE;
1354 /***********************************************************************
1355 * SetupFindNextLine (SETUPAPI.@)
1357 BOOL WINAPI SetupFindNextLine( PINFCONTEXT context_in, PINFCONTEXT context_out )
1359 struct inf_file *file = context_in->CurrentInf;
1360 struct section *section;
1362 if (context_in->Section >= file->nb_sections) goto error;
1364 section = file->sections[context_in->Section];
1365 if (context_in->Line+1 < section->nb_lines)
1367 if (context_out != context_in) *context_out = *context_in;
1368 context_out->Line++;
1369 SetLastError( 0 );
1370 return TRUE;
1373 /* now search the appended files */
1375 for (file = file->next; file; file = file->next)
1377 int section_index = find_section( file, section->name );
1378 if (section_index == -1) continue;
1379 if (file->sections[section_index]->nb_lines)
1381 context_out->Inf = context_in->Inf;
1382 context_out->CurrentInf = file;
1383 context_out->Section = section_index;
1384 context_out->Line = 0;
1385 SetLastError( 0 );
1386 return TRUE;
1389 error:
1390 SetLastError( ERROR_LINE_NOT_FOUND );
1391 return FALSE;
1395 /***********************************************************************
1396 * SetupFindNextMatchLineA (SETUPAPI.@)
1398 BOOL WINAPI SetupFindNextMatchLineA( PINFCONTEXT context_in, PCSTR key,
1399 PINFCONTEXT context_out )
1401 UNICODE_STRING keyW;
1402 BOOL ret = FALSE;
1404 if (!key) return SetupFindNextLine( context_in, context_out );
1406 if (!RtlCreateUnicodeStringFromAsciiz( &keyW, key ))
1407 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1408 else
1410 ret = SetupFindNextMatchLineW( context_in, keyW.Buffer, context_out );
1411 RtlFreeUnicodeString( &keyW );
1413 return ret;
1417 /***********************************************************************
1418 * SetupFindNextMatchLineW (SETUPAPI.@)
1420 BOOL WINAPI SetupFindNextMatchLineW( PINFCONTEXT context_in, PCWSTR key,
1421 PINFCONTEXT context_out )
1423 struct inf_file *file = context_in->CurrentInf;
1424 struct section *section;
1425 struct line *line;
1426 unsigned int i;
1428 if (!key) return SetupFindNextLine( context_in, context_out );
1430 if (context_in->Section >= file->nb_sections) goto error;
1432 section = file->sections[context_in->Section];
1434 for (i = context_in->Line+1, line = &section->lines[i]; i < section->nb_lines; i++, line++)
1436 if (line->key_field == -1) continue;
1437 if (!strcmpiW( key, file->fields[line->key_field].text ))
1439 if (context_out != context_in) *context_out = *context_in;
1440 context_out->Line = i;
1441 SetLastError( 0 );
1442 TRACE( "(%p,%s,%s): returning %d\n",
1443 file, debugstr_w(section->name), debugstr_w(key), i );
1444 return TRUE;
1448 /* now search the appended files */
1450 for (file = file->next; file; file = file->next)
1452 int section_index = find_section( file, section->name );
1453 if (section_index == -1) continue;
1454 section = file->sections[section_index];
1455 for (i = 0, line = section->lines; i < section->nb_lines; i++, line++)
1457 if (line->key_field == -1) continue;
1458 if (!strcmpiW( key, file->fields[line->key_field].text ))
1460 context_out->Inf = context_in->Inf;
1461 context_out->CurrentInf = file;
1462 context_out->Section = section_index;
1463 context_out->Line = i;
1464 SetLastError( 0 );
1465 TRACE( "(%p,%s,%s): returning %d/%d\n",
1466 file, debugstr_w(section->name), debugstr_w(key), section_index, i );
1467 return TRUE;
1471 TRACE( "(%p,%s,%s): not found\n",
1472 context_in->CurrentInf, debugstr_w(section->name), debugstr_w(key) );
1473 error:
1474 SetLastError( ERROR_LINE_NOT_FOUND );
1475 return FALSE;
1479 /***********************************************************************
1480 * SetupGetLineTextW (SETUPAPI.@)
1482 BOOL WINAPI SetupGetLineTextW( PINFCONTEXT context, HINF hinf, PCWSTR section_name,
1483 PCWSTR key_name, PWSTR buffer, DWORD size, PDWORD required )
1485 struct inf_file *file;
1486 struct line *line;
1487 struct field *field;
1488 int i;
1489 DWORD total = 0;
1491 if (!context)
1493 INFCONTEXT new_context;
1494 if (!SetupFindFirstLineW( hinf, section_name, key_name, &new_context )) return FALSE;
1495 file = new_context.CurrentInf;
1496 line = get_line( file, new_context.Section, new_context.Line );
1498 else
1500 file = context->CurrentInf;
1501 if (!(line = get_line( file, context->Section, context->Line )))
1503 SetLastError( ERROR_LINE_NOT_FOUND );
1504 return FALSE;
1508 for (i = 0, field = &file->fields[line->first_field]; i < line->nb_fields; i++, field++)
1509 total += PARSER_string_substW( file, field->text, NULL, 0 ) + 1;
1511 if (required) *required = total;
1512 if (buffer)
1514 if (total > size)
1516 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1517 return FALSE;
1519 for (i = 0, field = &file->fields[line->first_field]; i < line->nb_fields; i++, field++)
1521 unsigned int len = PARSER_string_substW( file, field->text, buffer, size );
1522 if (i+1 < line->nb_fields) buffer[len] = ',';
1523 buffer += len + 1;
1526 return TRUE;
1530 /***********************************************************************
1531 * SetupGetLineTextA (SETUPAPI.@)
1533 BOOL WINAPI SetupGetLineTextA( PINFCONTEXT context, HINF hinf, PCSTR section_name,
1534 PCSTR key_name, PSTR buffer, DWORD size, PDWORD required )
1536 struct inf_file *file;
1537 struct line *line;
1538 struct field *field;
1539 int i;
1540 DWORD total = 0;
1542 if (!context)
1544 INFCONTEXT new_context;
1545 if (!SetupFindFirstLineA( hinf, section_name, key_name, &new_context )) return FALSE;
1546 file = new_context.CurrentInf;
1547 line = get_line( file, new_context.Section, new_context.Line );
1549 else
1551 file = context->CurrentInf;
1552 if (!(line = get_line( file, context->Section, context->Line )))
1554 SetLastError( ERROR_LINE_NOT_FOUND );
1555 return FALSE;
1559 for (i = 0, field = &file->fields[line->first_field]; i < line->nb_fields; i++, field++)
1560 total += PARSER_string_substA( file, field->text, NULL, 0 ) + 1;
1562 if (required) *required = total;
1563 if (buffer)
1565 if (total > size)
1567 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1568 return FALSE;
1570 for (i = 0, field = &file->fields[line->first_field]; i < line->nb_fields; i++, field++)
1572 unsigned int len = PARSER_string_substA( file, field->text, buffer, size );
1573 if (i+1 < line->nb_fields) buffer[len] = ',';
1574 buffer += len + 1;
1577 return TRUE;
1581 /***********************************************************************
1582 * SetupGetFieldCount (SETUPAPI.@)
1584 DWORD WINAPI SetupGetFieldCount( PINFCONTEXT context )
1586 struct inf_file *file = context->CurrentInf;
1587 struct line *line = get_line( file, context->Section, context->Line );
1589 if (!line) return 0;
1590 return line->nb_fields;
1594 /***********************************************************************
1595 * SetupGetStringFieldA (SETUPAPI.@)
1597 BOOL WINAPI SetupGetStringFieldA( PINFCONTEXT context, DWORD index, PSTR buffer,
1598 DWORD size, PDWORD required )
1600 struct inf_file *file = context->CurrentInf;
1601 struct field *field = get_field( file, context->Section, context->Line, index );
1602 unsigned int len;
1604 SetLastError(0);
1605 if (!field) return FALSE;
1606 len = PARSER_string_substA( file, field->text, NULL, 0 );
1607 if (required) *required = len + 1;
1608 if (buffer)
1610 if (size <= len)
1612 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1613 return FALSE;
1615 PARSER_string_substA( file, field->text, buffer, size );
1617 TRACE( "context %p/%p/%d/%d index %ld returning %s\n",
1618 context->Inf, context->CurrentInf, context->Section, context->Line,
1619 index, debugstr_a(buffer) );
1621 return TRUE;
1625 /***********************************************************************
1626 * SetupGetStringFieldW (SETUPAPI.@)
1628 BOOL WINAPI SetupGetStringFieldW( PINFCONTEXT context, DWORD index, PWSTR buffer,
1629 DWORD size, PDWORD required )
1631 struct inf_file *file = context->CurrentInf;
1632 struct field *field = get_field( file, context->Section, context->Line, index );
1633 unsigned int len;
1635 SetLastError(0);
1636 if (!field) return FALSE;
1637 len = PARSER_string_substW( file, field->text, NULL, 0 );
1638 if (required) *required = len + 1;
1639 if (buffer)
1641 if (size <= len)
1643 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1644 return FALSE;
1646 PARSER_string_substW( file, field->text, buffer, size );
1648 TRACE( "context %p/%p/%d/%d index %ld returning %s\n",
1649 context->Inf, context->CurrentInf, context->Section, context->Line,
1650 index, debugstr_w(buffer) );
1652 return TRUE;
1656 /***********************************************************************
1657 * SetupGetIntField (SETUPAPI.@)
1659 BOOL WINAPI SetupGetIntField( PINFCONTEXT context, DWORD index, PINT result )
1661 char localbuff[20];
1662 char *end, *buffer = localbuff;
1663 DWORD required;
1664 INT res;
1665 BOOL ret = FALSE;
1667 if (!SetupGetStringFieldA( context, index, localbuff, sizeof(localbuff), &required ))
1669 if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) return FALSE;
1670 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, required ))) return FALSE;
1671 if (!SetupGetStringFieldA( context, index, buffer, required, NULL )) goto done;
1673 res = strtol( buffer, &end, 0 );
1674 if (end != buffer && !*end)
1676 *result = res;
1677 ret = TRUE;
1679 else SetLastError( ERROR_INVALID_DATA );
1681 done:
1682 if (buffer != localbuff) HeapFree( GetProcessHeap(), 0, buffer );
1683 return ret;
1687 /***********************************************************************
1688 * SetupGetBinaryField (SETUPAPI.@)
1690 BOOL WINAPI SetupGetBinaryField( PINFCONTEXT context, DWORD index, BYTE *buffer,
1691 DWORD size, LPDWORD required )
1693 struct inf_file *file = context->CurrentInf;
1694 struct line *line = get_line( file, context->Section, context->Line );
1695 struct field *field;
1696 int i;
1698 if (!line)
1700 SetLastError( ERROR_LINE_NOT_FOUND );
1701 return FALSE;
1703 if (!index || index >= line->nb_fields)
1705 SetLastError( ERROR_INVALID_PARAMETER );
1706 return FALSE;
1708 index--; /* fields start at 0 */
1709 if (required) *required = line->nb_fields - index;
1710 if (!buffer) return TRUE;
1711 if (size < line->nb_fields - index)
1713 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1714 return FALSE;
1716 field = &file->fields[line->first_field + index];
1717 for (i = index; i < line->nb_fields; i++, field++)
1719 const WCHAR *p;
1720 DWORD value = 0;
1721 for (p = field->text; *p && isxdigitW(*p); p++)
1723 if ((value <<= 4) > 255)
1725 SetLastError( ERROR_INVALID_DATA );
1726 return FALSE;
1728 if (*p <= '9') value |= (*p - '0');
1729 else value |= (tolowerW(*p) - 'a' + 10);
1731 buffer[i - index] = value;
1733 if (TRACE_ON(setupapi))
1735 TRACE( "%p/%p/%d/%d index %ld returning",
1736 context->Inf, context->CurrentInf, context->Section, context->Line, index );
1737 for (i = index; i < line->nb_fields; i++) TRACE( " %02x", buffer[i - index] );
1738 TRACE( "\n" );
1740 return TRUE;
1744 /***********************************************************************
1745 * SetupGetMultiSzFieldA (SETUPAPI.@)
1747 BOOL WINAPI SetupGetMultiSzFieldA( PINFCONTEXT context, DWORD index, PSTR buffer,
1748 DWORD size, LPDWORD required )
1750 struct inf_file *file = context->CurrentInf;
1751 struct line *line = get_line( file, context->Section, context->Line );
1752 struct field *field;
1753 unsigned int len;
1754 int i;
1755 DWORD total = 1;
1757 if (!line)
1759 SetLastError( ERROR_LINE_NOT_FOUND );
1760 return FALSE;
1762 if (!index || index >= line->nb_fields)
1764 SetLastError( ERROR_INVALID_PARAMETER );
1765 return FALSE;
1767 index--; /* fields start at 0 */
1768 field = &file->fields[line->first_field + index];
1769 for (i = index; i < line->nb_fields; i++, field++)
1771 if (!(len = PARSER_string_substA( file, field->text, NULL, 0 ))) break;
1772 total += len + 1;
1775 if (required) *required = total;
1776 if (!buffer) return TRUE;
1777 if (total > size)
1779 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1780 return FALSE;
1782 field = &file->fields[line->first_field + index];
1783 for (i = index; i < line->nb_fields; i++, field++)
1785 if (!(len = PARSER_string_substA( file, field->text, buffer, size ))) break;
1786 buffer += len + 1;
1788 *buffer = 0; /* add final null */
1789 return TRUE;
1793 /***********************************************************************
1794 * SetupGetMultiSzFieldW (SETUPAPI.@)
1796 BOOL WINAPI SetupGetMultiSzFieldW( PINFCONTEXT context, DWORD index, PWSTR buffer,
1797 DWORD size, LPDWORD required )
1799 struct inf_file *file = context->CurrentInf;
1800 struct line *line = get_line( file, context->Section, context->Line );
1801 struct field *field;
1802 unsigned int len;
1803 int i;
1804 DWORD total = 1;
1806 if (!line)
1808 SetLastError( ERROR_LINE_NOT_FOUND );
1809 return FALSE;
1811 if (!index || index >= line->nb_fields)
1813 SetLastError( ERROR_INVALID_PARAMETER );
1814 return FALSE;
1816 index--; /* fields start at 0 */
1817 field = &file->fields[line->first_field + index];
1818 for (i = index; i < line->nb_fields; i++, field++)
1820 if (!(len = PARSER_string_substW( file, field->text, NULL, 0 ))) break;
1821 total += len + 1;
1824 if (required) *required = total;
1825 if (!buffer) return TRUE;
1826 if (total > size)
1828 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1829 return FALSE;
1831 field = &file->fields[line->first_field + index];
1832 for (i = index; i < line->nb_fields; i++, field++)
1834 if (!(len = PARSER_string_substW( file, field->text, buffer, size ))) break;
1835 buffer += len + 1;
1837 *buffer = 0; /* add final null */
1838 return TRUE;