2 * Directory id handling
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
26 #include "wine/unicode.h"
27 #include "setupapi_private.h"
28 #include "wine/debug.h"
30 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
32 #define MAX_SYSTEM_DIRID DIRID_PRINTPROCESSOR
40 static int nb_user_dirids; /* number of user dirids in use */
41 static int alloc_user_dirids; /* number of allocated user dirids */
42 static struct user_dirid *user_dirids;
43 static const WCHAR *system_dirids[MAX_SYSTEM_DIRID+1];
45 /* retrieve the string for unknown dirids */
46 static const WCHAR *get_unknown_dirid(void)
48 static WCHAR *unknown_dirid;
49 static const WCHAR unknown_str[] = {'\\','u','n','k','n','o','w','n',0};
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 );
61 /* create the string for a system dirid */
62 static const WCHAR *create_system_dirid( int dirid )
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};
74 WCHAR buffer[MAX_PATH+16], *str;
82 GetWindowsDirectoryW( buffer, MAX_PATH );
85 GetSystemDirectoryW( buffer, MAX_PATH );
88 GetSystemDirectoryW( buffer, MAX_PATH );
89 strcatW( buffer, Drivers );
92 GetWindowsDirectoryW( buffer, MAX_PATH );
93 strcatW( buffer, Inf );
96 GetWindowsDirectoryW( buffer, MAX_PATH );
97 strcatW( buffer, Help );
100 GetWindowsDirectoryW( buffer, MAX_PATH );
101 strcatW( buffer, Fonts );
104 GetSystemDirectoryW( buffer, MAX_PATH );
105 strcatW( buffer, Viewers );
108 return C_Root; /* FIXME */
110 GetWindowsDirectoryW( buffer, MAX_PATH );
113 return C_Root; /* FIXME */
115 GetWindowsDirectoryW( buffer, MAX_PATH );
116 strcatW( buffer, System );
119 case DIRID_SPOOLDRIVERS: /* FIXME */
120 GetWindowsDirectoryW( buffer, MAX_PATH );
121 strcatW( buffer, Spool );
124 return C_Root; /* FIXME */
125 case DIRID_USERPROFILE: /* FIXME */
126 case DIRID_COLOR: /* FIXME */
127 case DIRID_PRINTPROCESSOR: /* FIXME */
129 FIXME( "unknown dirid %d\n", dirid );
130 return get_unknown_dirid();
132 len = (strlenW(buffer) + 1) * sizeof(WCHAR);
133 if ((str = HeapAlloc( GetProcessHeap(), 0, len ))) memcpy( str, buffer, len );
137 /* retrieve the string corresponding to a dirid, or NULL if none */
138 const WCHAR *DIRID_get_string( HINF hinf, int dirid )
142 if (dirid == DIRID_ABSOLUTE || dirid == DIRID_ABSOLUTE_16BIT) dirid = DIRID_NULL;
144 if (dirid >= DIRID_USER)
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 );
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];
160 /* store a user dirid string */
161 static BOOL store_user_dirid( HINF hinf, int id, WCHAR *str )
165 for (i = 0; i < nb_user_dirids; i++) if (user_dirids[i].id == id) break;
167 if (i < nb_user_dirids) HeapFree( GetProcessHeap(), 0, user_dirids[i].str );
170 if (nb_user_dirids >= alloc_user_dirids)
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;
177 alloc_user_dirids = new_size;
181 user_dirids[i].id = id;
182 user_dirids[i].str = str;
183 TRACE("id %d -> %s\n", id, debugstr_w(str) );
188 /***********************************************************************
189 * SetupSetDirectoryIdA (SETUPAPI.@)
191 BOOL WINAPI SetupSetDirectoryIdA( HINF hinf, DWORD id, PCSTR dir )
196 if (!id) /* clear everything */
198 for (i = 0; i < nb_user_dirids; i++) HeapFree( GetProcessHeap(), 0, user_dirids[i].str );
204 SetLastError( ERROR_INVALID_PARAMETER );
208 /* duplicate the string */
209 if (!RtlCreateUnicodeStringFromAsciiz( &dirW, dir ))
211 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
214 return store_user_dirid( hinf, id, dirW.Buffer );
218 /***********************************************************************
219 * SetupSetDirectoryIdW (SETUPAPI.@)
221 BOOL WINAPI SetupSetDirectoryIdW( HINF hinf, DWORD id, PCWSTR dir )
226 if (!id) /* clear everything */
228 for (i = 0; i < nb_user_dirids; i++) HeapFree( GetProcessHeap(), 0, user_dirids[i].str );
234 SetLastError( ERROR_INVALID_PARAMETER );
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 );