source: trunk/src/setupapi/dirid.c@ 10010

Last change on this file since 10010 was 9403, checked in by sandervl, 23 years ago

Wine resync
File size: 7.3 KB
Line 
1/*
2 * Directory id handling
3 *
4 * Copyright 2002 Alexandre Julliard for CodeWeavers
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21#include "windef.h"
22#include "winbase.h"
23#include "winternl.h"
24#include "winerror.h"
25#include "setupapi.h"
26#include "wine/unicode.h"
27#include "setupapi_private.h"
28#include "wine/debug.h"
29
30WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
31
32#define MAX_SYSTEM_DIRID DIRID_PRINTPROCESSOR
33
34struct user_dirid
35{
36 int id;
37 WCHAR *str;
38};
39
40static int nb_user_dirids; /* number of user dirids in use */
41static int alloc_user_dirids; /* number of allocated user dirids */
42static struct user_dirid *user_dirids;
43static const WCHAR *system_dirids[MAX_SYSTEM_DIRID+1];
44
45/* retrieve the string for unknown dirids */
46static const WCHAR *get_unknown_dirid(void)
47{
48 static WCHAR *unknown_dirid;
49 static const WCHAR unknown_str[] = {'\\','u','n','k','n','o','w','n',0};
50
51 if (!unknown_dirid)
52 {
53 UINT len = GetSystemDirectoryW( NULL, 0 ) + strlenW(unknown_str);
54 if (!(unknown_dirid = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) return NULL;
55 GetSystemDirectoryW( unknown_dirid, len );
56 strcatW( unknown_dirid, unknown_str );
57 }
58 return unknown_dirid;
59}
60
61/* create the string for a system dirid */
62static const WCHAR *create_system_dirid( int dirid )
63{
64 static const WCHAR Null[] = {0};
65 static const WCHAR C_Root[] = {'C',':','\\',0};
66 static const WCHAR Drivers[] = {'\\','d','r','i','v','e','r','s',0};
67 static const WCHAR Inf[] = {'\\','i','n','f',0};
68 static const WCHAR Help[] = {'\\','h','e','l','p',0};
69 static const WCHAR Fonts[] = {'\\','f','o','n','t','s',0};
70 static const WCHAR Viewers[] = {'\\','v','i','e','w','e','r','s',0};
71 static const WCHAR System[] = {'\\','s','y','s','t','e','m',0};
72 static const WCHAR Spool[] = {'\\','s','p','o','o','l',0};
73
74 WCHAR buffer[MAX_PATH+16], *str;
75 int len;
76
77 switch(dirid)
78 {
79 case DIRID_NULL:
80 return Null;
81 case DIRID_WINDOWS:
82 GetWindowsDirectoryW( buffer, MAX_PATH );
83 break;
84 case DIRID_SYSTEM:
85 GetSystemDirectoryW( buffer, MAX_PATH );
86 break;
87 case DIRID_DRIVERS:
88 GetSystemDirectoryW( buffer, MAX_PATH );
89 strcatW( buffer, Drivers );
90 break;
91 case DIRID_INF:
92 GetWindowsDirectoryW( buffer, MAX_PATH );
93 strcatW( buffer, Inf );
94 break;
95 case DIRID_HELP:
96 GetWindowsDirectoryW( buffer, MAX_PATH );
97 strcatW( buffer, Help );
98 break;
99 case DIRID_FONTS:
100 GetWindowsDirectoryW( buffer, MAX_PATH );
101 strcatW( buffer, Fonts );
102 break;
103 case DIRID_VIEWERS:
104 GetSystemDirectoryW( buffer, MAX_PATH );
105 strcatW( buffer, Viewers );
106 break;
107 case DIRID_APPS:
108 return C_Root; /* FIXME */
109 case DIRID_SHARED:
110 GetWindowsDirectoryW( buffer, MAX_PATH );
111 break;
112 case DIRID_BOOT:
113 return C_Root; /* FIXME */
114 case DIRID_SYSTEM16:
115 GetWindowsDirectoryW( buffer, MAX_PATH );
116 strcatW( buffer, System );
117 break;
118 case DIRID_SPOOL:
119 case DIRID_SPOOLDRIVERS: /* FIXME */
120 GetWindowsDirectoryW( buffer, MAX_PATH );
121 strcatW( buffer, Spool );
122 break;
123 case DIRID_LOADER:
124 return C_Root; /* FIXME */
125 case DIRID_USERPROFILE: /* FIXME */
126 case DIRID_COLOR: /* FIXME */
127 case DIRID_PRINTPROCESSOR: /* FIXME */
128 default:
129 FIXME( "unknwon dirid %d\n", dirid );
130 return get_unknown_dirid();
131 }
132 len = (strlenW(buffer) + 1) * sizeof(WCHAR);
133 if ((str = HeapAlloc( GetProcessHeap(), 0, len ))) memcpy( str, buffer, len );
134 return str;
135}
136
137/* retrieve the string corresponding to a dirid, or NULL if none */
138const WCHAR *DIRID_get_string( HINF hinf, int dirid )
139{
140 int i;
141
142 if (dirid == DIRID_ABSOLUTE || dirid == DIRID_ABSOLUTE_16BIT) dirid = DIRID_NULL;
143
144 if (dirid >= DIRID_USER)
145 {
146 for (i = 0; i < nb_user_dirids; i++)
147 if (user_dirids[i].id == dirid) return user_dirids[i].str;
148 ERR("user id %d not found\n", dirid );
149 return NULL;
150 }
151 else
152 {
153 if (dirid > MAX_SYSTEM_DIRID) return get_unknown_dirid();
154 if (dirid == DIRID_SRCPATH) return PARSER_get_src_root( hinf );
155 if (!system_dirids[dirid]) system_dirids[dirid] = create_system_dirid( dirid );
156 return system_dirids[dirid];
157 }
158}
159
160/* store a user dirid string */
161static BOOL store_user_dirid( HINF hinf, int id, WCHAR *str )
162{
163 int i;
164
165 for (i = 0; i < nb_user_dirids; i++) if (user_dirids[i].id == id) break;
166
167 if (i < nb_user_dirids) HeapFree( GetProcessHeap(), 0, user_dirids[i].str );
168 else
169 {
170 if (nb_user_dirids >= alloc_user_dirids)
171 {
172 int new_size = max( 32, alloc_user_dirids * 2 );
173 struct user_dirid *new = HeapReAlloc( GetProcessHeap(), 0, user_dirids,
174 new_size * sizeof(*new) );
175 if (!new) return FALSE;
176 user_dirids = new;
177 alloc_user_dirids = new_size;
178 }
179 nb_user_dirids++;
180 }
181 user_dirids[i].id = id;
182 user_dirids[i].str = str;
183 TRACE("id %d -> %s\n", id, debugstr_w(str) );
184 return TRUE;
185}
186
187
188/***********************************************************************
189 * SetupSetDirectoryIdA (SETUPAPI.@)
190 */
191BOOL WINAPI SetupSetDirectoryIdA( HINF hinf, DWORD id, PCSTR dir )
192{
193 UNICODE_STRING dirW;
194 int i;
195
196 if (!id) /* clear everything */
197 {
198 for (i = 0; i < nb_user_dirids; i++) HeapFree( GetProcessHeap(), 0, user_dirids[i].str );
199 nb_user_dirids = 0;
200 return TRUE;
201 }
202 if (id < DIRID_USER)
203 {
204 SetLastError( ERROR_INVALID_PARAMETER );
205 return FALSE;
206 }
207
208 /* duplicate the string */
209 if (!RtlCreateUnicodeStringFromAsciiz( &dirW, dir ))
210 {
211 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
212 return FALSE;
213 }
214 return store_user_dirid( hinf, id, dirW.Buffer );
215}
216
217
218/***********************************************************************
219 * SetupSetDirectoryIdW (SETUPAPI.@)
220 */
221BOOL WINAPI SetupSetDirectoryIdW( HINF hinf, DWORD id, PCWSTR dir )
222{
223 int i, len;
224 WCHAR *str;
225
226 if (!id) /* clear everything */
227 {
228 for (i = 0; i < nb_user_dirids; i++) HeapFree( GetProcessHeap(), 0, user_dirids[i].str );
229 nb_user_dirids = 0;
230 return TRUE;
231 }
232 if (id < DIRID_USER)
233 {
234 SetLastError( ERROR_INVALID_PARAMETER );
235 return FALSE;
236 }
237
238 /* duplicate the string */
239 len = (strlenW(dir)+1) * sizeof(WCHAR);
240 if (!(str = HeapAlloc( GetProcessHeap(), 0, len ))) return FALSE;
241 memcpy( str, dir, len );
242 return store_user_dirid( hinf, id, str );
243}
Note: See TracBrowser for help on using the repository browser.