- introduce keypath() function
[wine] / programs / winecfg / winecfg.c
1 /*
2  * WineCfg configuration management
3  *
4  * Copyright 2002 Jaco Greeff
5  * Copyright 2003 Dimitrie O. Paun
6  * Copyright 2003-2004 Mike Hearn
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  *
22  * TODO:   (in rough order of priority)
23  *   - A mind bogglingly vast amount of stuff
24  *
25  *   - Implement autodetect for drive configuration
26  *   - Figure out whether we need the virtual vs real drive selection stuff at the top of the property page
27  *   - Implement explicit mode vs instant-apply mode
28  *   - DLL editing
29  *   - Multimedia page
30  *   - Settings migration code (from old configs)
31  *   - Clean up resource.h, it's a bog
32  *
33  *   Minor things that should be done someday:
34  *   - Make the desktop size UI a combo box, with a Custom option, so it's more obvious what you might want to choose here
35  *
36  * BUGS:
37  *   - x11drv page triggers key writes on entry
38  *
39  */
40
41 #include <assert.h>
42 #include <stdio.h>
43 #include <limits.h>
44 #include <windows.h>
45 #include <winreg.h>
46 #include <wine/debug.h>
47
48 WINE_DEFAULT_DEBUG_CHANNEL(winecfg);
49
50 #include "winecfg.h"
51
52 HKEY configKey = NULL;
53
54
55 int initialize(void) {
56     DWORD res = RegCreateKey(HKEY_LOCAL_MACHINE, WINE_KEY_ROOT, &configKey);
57     if (res != ERROR_SUCCESS) {
58         WINE_ERR("RegOpenKey failed on wine config key (%ld)\n", res);
59         return 1;
60     }
61     return 0;
62 }
63
64
65 /*****************************************************************************
66  * getConfigValue: Retrieves a configuration value from the registry
67  *
68  * const char *subKey : the name of the config section
69  * const char *valueName : the name of the config value
70  * const char *defaultResult : if the key isn't found, return this value instead
71  *
72  * Returns a buffer holding the value if successful, NULL if not. Caller is responsible for freeing the result.
73  *
74  */
75 char *getConfigValue (const char *subkey, const char *valueName, const char *defaultResult)
76 {
77     char *buffer = NULL;
78     DWORD dataLength;
79     HKEY hSubKey = NULL;
80     DWORD res;
81
82     WINE_TRACE("subkey=%s, valueName=%s, defaultResult=%s\n", subkey, valueName, defaultResult);
83
84     res = RegOpenKeyEx( configKey, subkey, 0, KEY_ALL_ACCESS, &hSubKey );
85     if(res != ERROR_SUCCESS)  {
86         if( res==ERROR_FILE_NOT_FOUND )
87         {
88             WINE_TRACE("Section key not present - using default\n");
89             return defaultResult ? strdup(defaultResult) : NULL;
90         }
91         else
92         {
93             WINE_ERR("RegOpenKey failed on wine config key (res=%ld)\n", res);
94         }
95         goto end;
96     }
97
98     res = RegQueryValueExA( hSubKey, valueName, NULL, NULL, NULL, &dataLength);
99     if( res == ERROR_FILE_NOT_FOUND ) {
100         WINE_TRACE("Value not present - using default\n");
101         buffer = defaultResult ? strdup(defaultResult) : NULL;
102         goto end;
103     } else if( res!=ERROR_SUCCESS )  {
104         WINE_ERR("Couldn't query value's length (res=%ld)\n", res );
105         goto end;
106     }
107
108     buffer = malloc(dataLength);
109     if( buffer==NULL )
110     {
111         WINE_ERR("Couldn't allocate %lu bytes for the value\n", dataLength );
112         goto end;
113     }
114     
115     RegQueryValueEx(hSubKey, valueName, NULL, NULL, (LPBYTE)buffer, &dataLength);
116     
117 end:
118     if( hSubKey!=NULL )
119         RegCloseKey( hSubKey );
120
121     return buffer;
122     
123 }
124
125 /*****************************************************************************
126  * setConfigValue : Sets a configuration key in the registry. Section
127  * will be created if it doesn't already exist
128  *
129  * HKEY  hCurrent : the registry key that the configuration is rooted at
130  * const char *subKey : the name of the config section
131  * const char *valueName : the name of the config value
132  * const char *value : the value to set the configuration key to
133  *
134  * Returns 0 on success, non-zero otherwise
135  * 
136  * If *valueName or *value is NULL, an empty section will be created
137  */
138 int setConfigValue (const char *subkey, const char *valueName, const char *value) {
139     DWORD res = 1;
140     HKEY key = NULL;
141
142     WINE_TRACE("subkey=%s, valueName=%s, value=%s\n", subkey, valueName, value);
143
144     assert( subkey != NULL );
145     
146     res = RegCreateKey(configKey, subkey, &key);
147     if (res != ERROR_SUCCESS) goto end;
148     if (value == NULL || valueName == NULL) goto end;
149
150     res = RegSetValueEx(key, valueName, 0, REG_SZ, value, strlen(value) + 1);
151     if (res != ERROR_SUCCESS) goto end;
152
153     res = 0;
154 end:
155     if (key) RegCloseKey(key);
156     if (res != 0) WINE_ERR("Unable to set configuration key %s in section %s to %s, res=%ld\n", valueName, subkey, value, res);
157     return res;
158 }
159
160 /* returns 0 on success, an HRESULT from the registry funtions otherwise */
161 HRESULT doesConfigValueExist(const char *subkey, const char *valueName) {
162     HRESULT hr;
163     HKEY key;
164
165     WINE_TRACE("subkey=%s, valueName=%s - ", subkey, valueName);
166     
167     hr = RegOpenKeyEx(configKey, subkey, 0, KEY_READ, &key);
168     if (hr != S_OK) {
169         WINE_TRACE("no: subkey does not exist\n");
170         return hr;
171     }
172
173     hr = RegQueryValueEx(key, valueName, NULL, NULL, NULL, NULL);
174     if (hr != S_OK) {
175         WINE_TRACE("no: key does not exist\n");
176         return hr;
177     }
178
179     RegCloseKey(key);
180     WINE_TRACE("yes\n");
181     return S_OK;
182 }
183
184 /* removes the requested value from the registry, however, does not remove the section if empty. Returns S_OK (0) on success. */
185 HRESULT removeConfigValue(const char *subkey, const char *valueName) {
186     HRESULT hr;
187     HKEY key;
188     WINE_TRACE("subkey=%s, valueName=%s\n", subkey, valueName);
189     
190     hr = RegOpenKeyEx(configKey, subkey, 0, KEY_READ, &key);
191     if (hr != S_OK) return hr;
192
193     hr = RegDeleteValue(key, valueName);
194     if (hr != ERROR_SUCCESS) return hr;
195
196     return S_OK;
197 }
198
199 /* removes the requested configuration section (subkey) from the registry, assuming it exists */
200 /* this function might be slightly pointless, but in future we may wish to treat recursion specially etc, so we'll keep it for now */
201 HRESULT removeConfigSection(char *section) {
202     HRESULT hr;
203     WINE_TRACE("section=%s\n", section);
204
205     return hr = RegDeleteKey(configKey, section);
206 }
207
208
209 /* ========================================================================= */
210 /* Transaction management code */
211
212 struct transaction *tqhead, *tqtail;
213 int instantApply = 1;
214
215 void destroyTransaction(struct transaction *trans) {
216     assert( trans != NULL );
217     
218     WINE_TRACE("destroying %p\n", trans);
219     
220     free(trans->section);
221     if (trans->key) free(trans->key);
222     if (trans->newValue) free(trans->newValue);
223     
224     if (trans->next) trans->next->prev = trans->prev;
225     if (trans->prev) trans->prev->next = trans->next;
226     if (trans == tqhead) tqhead = NULL;
227     if (trans == tqtail) tqtail = NULL;
228     
229     free(trans);
230 }
231
232 void addTransaction(const char *section, const char *key, enum transaction_action action, const char *newValue) {
233     struct transaction *trans = calloc(sizeof(struct transaction),1);
234     
235     assert( section != NULL );
236     if (action == ACTION_SET) assert( newValue != NULL );
237     if (action == ACTION_SET) assert( key != NULL );
238                                      
239     trans->section = strdup(section);
240     if (key) trans->key = strdup(key);
241     if (newValue) trans->newValue = strdup(newValue);
242     trans->action = action;
243     
244     if (tqtail == NULL) {
245         tqtail = trans;
246         tqhead = tqtail;
247     } else {
248         tqhead->next = trans;
249         trans->prev = tqhead;
250         tqhead = trans;
251     }
252
253     if (instantApply) {
254         processTransaction(trans);
255         destroyTransaction(trans);
256     }
257 }
258
259 void processTransaction(struct transaction *trans) {
260     if (trans->action == ACTION_SET) {
261         WINE_TRACE("Setting %s\\%s to '%s'\n", trans->section, trans->key, trans->newValue);
262         setConfigValue(trans->section, trans->key, trans->newValue);
263     } else if (trans->action == ACTION_REMOVE) {
264         if (trans->key) {
265             WINE_TRACE("Removing %s\\%s\n", trans->section, trans->key);
266             removeConfigValue(trans->section, trans->key);
267         } else {
268             /* NULL key means remove that section entirely */
269             WINE_TRACE("Removing section %s\n", trans->section);
270             removeConfigSection(trans->section);
271         }
272     }
273     /* TODO: implement notifications here */
274 }
275
276 void processTransQueue(void)
277 {
278     WINE_TRACE("\n");
279     while (tqtail != NULL) {
280         struct transaction *next = tqtail->next;
281         processTransaction(tqtail);
282         destroyTransaction(tqtail);
283         tqtail = next;  
284     }
285 }
286
287 /* ================================== utility functions ============================ */
288
289 char *currentApp = NULL; /* the app we are currently editing, or NULL if editing global */
290
291 /* returns a registry key path suitable for passing to addTransaction  */
292 char *keypath(char *section)
293 {
294     static char *result = NULL;
295     
296     if (result) release(result);
297
298     if (currentApp)
299     {
300         result = alloc(strlen("AppDefaults\\") + strlen(currentApp) + 2 /* \\ */ + strlen(section) + 1 /* terminator */);
301         sprintf(result, "AppDefaults\\%s\\%s", currentApp, section);
302     }
303     else
304     {
305         result = strdupA(section);
306     }
307     
308     return result;
309 }
310
311 /* returns a string with the window text of the dialog item. user is responsible for freeing the result */
312 char *getDialogItemText(HWND hDlg, WORD controlID) {
313     HWND item = GetDlgItem(hDlg, controlID);
314     int len = GetWindowTextLength(item) + 1;
315     char *result = malloc(len);
316     if (GetWindowText(item, result, len) == 0) return NULL;
317     return result;
318 }
319
320 void PRINTERROR(void)
321 {
322         LPSTR msg;
323
324         FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM,
325                        0, GetLastError(), MAKELANGID(LANG_NEUTRAL,SUBLANG_DEFAULT),
326                        (LPSTR)&msg, 0, NULL);
327         WINE_TRACE("error: '%s'\n", msg);
328 }