attrib: Make command accept several arguments.
[wine] / programs / winecfg / libraries.c
1 /*
2  * WineCfg libraries tabsheet
3  *
4  * Copyright 2004 Robert van Herk
5  * Copyright 2004 Mike Hearn
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  *
21  */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #define NONAMELESSUNION
27 #define WIN32_LEAN_AND_MEAN
28 #include <windows.h>
29 #include <commdlg.h>
30 #include <wine/library.h>
31 #include <wine/debug.h>
32 #include <stdio.h>
33 #include <dirent.h>
34 #include <assert.h>
35 #include <stdlib.h>
36 #ifdef HAVE_SYS_STAT_H
37 #include <sys/stat.h>
38 #endif
39 #ifdef HAVE_UNISTD_H
40 #include <unistd.h>
41 #endif
42
43 #include "winecfg.h"
44 #include "resource.h"
45
46 WINE_DEFAULT_DEBUG_CHANNEL(winecfg);
47
48 /* dlls that shouldn't be configured anything other than builtin; list must be sorted*/
49 static const char * const builtin_only[] =
50 {
51     "advapi32",
52     "capi2032",
53     "dbghelp",
54     "ddraw",
55     "gdi32",
56     "glu32",
57     "gphoto2.ds",
58     "icmp",
59     "iphlpapi",
60     "kernel32",
61     "mountmgr.sys",
62     "mswsock",
63     "ntdll",
64     "ntoskrnl.exe",
65     "opengl32",
66     "sane.ds",
67     "twain_32",
68     "unicows",
69     "user32",
70     "vdmdbg",
71     "w32skrnl",
72     "wined3d",
73     "winedos",
74     "winemp3.acm",
75     "wineps",
76     "winmm",
77     "wintab32",
78     "wnaspi32",
79     "wow32",
80     "ws2_32",
81     "wsock32",
82 };
83
84 enum dllmode
85 {
86         BUILTIN_NATIVE,
87         NATIVE_BUILTIN,
88         BUILTIN,
89         NATIVE,
90         DISABLE,
91         UNKNOWN /* Special value indicating an erroneous DLL override mode */
92 };
93
94 struct dll
95 {
96         char *name;
97         enum dllmode mode;
98 };
99
100 static const WCHAR emptyW[1];
101
102 /* Convert a registry string to a dllmode */
103 static enum dllmode string_to_mode(char *in)
104 {
105     int i, j, len;
106     char *out;
107     enum dllmode res;
108
109     len = strlen(in);
110     out = HeapAlloc(GetProcessHeap(), 0, len);
111
112     /* remove the spaces */
113     for (i = j = 0; i <= len; ++i) {
114         if (in[i] != ' ') {
115             out[j++] = in[i];
116         }
117     }
118
119     /* parse the string */
120     res = UNKNOWN;
121     if (strcmp(out, "builtin,native") == 0) res = BUILTIN_NATIVE;
122     if (strcmp(out, "native,builtin") == 0) res = NATIVE_BUILTIN;
123     if (strcmp(out, "builtin") == 0) res = BUILTIN;
124     if (strcmp(out, "native") == 0) res = NATIVE;
125     if (strcmp(out, "") == 0) res = DISABLE;
126
127     HeapFree(GetProcessHeap(), 0, out);
128     return res;
129 }
130
131 /* Convert a dllmode to a registry string. */
132 static const char* mode_to_string(enum dllmode mode)
133 {
134     switch( mode )
135     {
136         case NATIVE: return "native";
137         case BUILTIN: return "builtin";
138         case NATIVE_BUILTIN: return "native,builtin";
139         case BUILTIN_NATIVE: return "builtin,native";
140         case DISABLE: return "";
141         default: return "";
142     }
143 }
144
145 /* Convert a dllmode to a pretty string for display. TODO: use translations. */
146 static const char* mode_to_label(enum dllmode mode)
147 {
148     static char buffer[256];
149     UINT id = 0;
150
151     switch( mode )
152     {
153     case NATIVE: id = IDS_DLL_NATIVE; break;
154     case BUILTIN: id = IDS_DLL_BUILTIN; break;
155     case NATIVE_BUILTIN: id = IDS_DLL_NATIVE_BUILTIN; break;
156     case BUILTIN_NATIVE: id = IDS_DLL_BUILTIN_NATIVE; break;
157     case DISABLE: id = IDS_DLL_DISABLED; break;
158     default: return "??";
159     }
160     if (!LoadStringA( GetModuleHandleA(NULL), id, buffer, sizeof(buffer) )) buffer[0] = 0;
161     return buffer;
162 }
163
164 /* Convert a control id (IDC_ constant) to a dllmode */
165 static enum dllmode id_to_mode(DWORD id)
166 {
167     switch( id )
168     {
169         case IDC_RAD_BUILTIN: return BUILTIN;
170         case IDC_RAD_NATIVE: return NATIVE;
171         case IDC_RAD_NATIVE_BUILTIN: return NATIVE_BUILTIN;
172         case IDC_RAD_BUILTIN_NATIVE: return BUILTIN_NATIVE;
173         case IDC_RAD_DISABLE: return DISABLE;
174         default: assert( FALSE ); return 0; /* should not be reached  */
175     }
176 }
177
178 /* Convert a dllmode to a control id (IDC_ constant) */
179 static DWORD mode_to_id(enum dllmode mode)
180 {
181     switch( mode )
182     {
183         case BUILTIN: return IDC_RAD_BUILTIN;
184         case NATIVE: return IDC_RAD_NATIVE;
185         case NATIVE_BUILTIN: return IDC_RAD_NATIVE_BUILTIN;
186         case BUILTIN_NATIVE: return IDC_RAD_BUILTIN_NATIVE;
187         case DISABLE: return IDC_RAD_DISABLE;
188         default: return IDC_RAD_BUILTIN_NATIVE;
189     }
190 }
191
192 /* helper for is_builtin_only */
193 static int compare_dll( const void *ptr1, const void *ptr2 )
194 {
195     const char * const *name1 = ptr1;
196     const char * const *name2 = ptr2;
197     return strcmp( *name1, *name2 );
198 }
199
200 /* check if dll is recommended as builtin only */
201 static inline int is_builtin_only( const char *name )
202 {
203     const char *ext = strrchr( name, '.' );
204
205     if (ext)
206     {
207         if (!strcmp( ext, ".vxd" ) ||
208             !strcmp( ext, ".drv" ) ||
209             !strcmp( ext, ".tlb" ))
210             return TRUE;
211     }
212     return bsearch( &name, builtin_only, sizeof(builtin_only)/sizeof(builtin_only[0]),
213                     sizeof(builtin_only[0]), compare_dll ) != NULL;
214 }
215
216 /* check if dll should be offered in the drop-down list */
217 static int show_dll_in_list( const char *name )
218 {
219     const char *ext = strrchr( name, '.' );
220
221     if (ext)
222     {
223         /* skip 16-bit dlls */
224         if (strlen(ext) > 2 && !strcmp( ext + strlen(ext) - 2, "16" )) return FALSE;
225         /* skip exes */
226         if (!strcmp( ext, ".exe" )) return FALSE;
227     }
228     /* skip dlls that should always be builtin */
229     return !is_builtin_only( name );
230 }
231
232 static void set_controls_from_selection(HWND dialog)
233 {
234     /* FIXME: display/update some information about the selected dll (purpose, recommended load order) maybe? */
235 }
236
237 static void clear_settings(HWND dialog)
238 {
239     int count = SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETCOUNT, 0, 0);
240     int i;
241
242     WINE_TRACE("count=%d\n", count);
243     
244     for (i = 0; i < count; i++)
245     {
246         struct dll *dll = (struct dll *) SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETITEMDATA, 0, 0);
247
248         SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_DELETESTRING, 0, 0);
249         HeapFree(GetProcessHeap(), 0, dll->name);
250         HeapFree(GetProcessHeap(), 0, dll);
251     }
252 }
253
254 /* load the list of available libraries from a given dir */
255 static void load_library_list_from_dir( HWND dialog, const char *dir_path, int check_subdirs )
256 {
257     char *buffer = NULL, name[256];
258     struct dirent *de;
259     DIR *dir = opendir( dir_path );
260
261     if (!dir) return;
262
263     if (check_subdirs)
264         buffer = HeapAlloc( GetProcessHeap(), 0, strlen(dir_path) + 2 * sizeof(name) + 10 );
265
266     while ((de = readdir( dir )))
267     {
268         size_t len = strlen(de->d_name);
269         if (len > sizeof(name)) continue;
270         if (len > 3 && !strcmp( de->d_name + len - 3, ".so"))
271         {
272             len -= 3;
273             if (len > 4 && !strcmp( de->d_name + len - 4, ".dll.so")) len -= 4;
274             memcpy( name, de->d_name, len );
275             name[len] = 0;
276             if (!show_dll_in_list( name )) continue;
277             SendDlgItemMessageA( dialog, IDC_DLLCOMBO, CB_ADDSTRING, 0, (LPARAM)name );
278         }
279         else if (check_subdirs)
280         {
281             struct stat st;
282             if (!show_dll_in_list( de->d_name )) continue;
283             sprintf( buffer, "%s/%s/%s.dll.so", dir_path, de->d_name, de->d_name );
284             if (!stat( buffer, &st ))
285             {
286                 SendDlgItemMessageA( dialog, IDC_DLLCOMBO, CB_ADDSTRING, 0, (LPARAM)de->d_name );
287                 continue;
288             }
289             sprintf( buffer, "%s/%s/%s.so", dir_path, de->d_name, de->d_name );
290             if (!stat( buffer, &st ))
291             {
292                 SendDlgItemMessageA( dialog, IDC_DLLCOMBO, CB_ADDSTRING, 0, (LPARAM)de->d_name );
293                 continue;
294             }
295         }
296     }
297     closedir( dir );
298     HeapFree( GetProcessHeap(), 0, buffer );
299 }
300
301 /* load the list of available libraries */
302 static void load_library_list( HWND dialog )
303 {
304     unsigned int i = 0;
305     const char *path, *build_dir = wine_get_build_dir();
306     char item1[256], item2[256];
307     HCURSOR old_cursor = SetCursor( LoadCursorW(0, (LPWSTR)IDC_WAIT) );
308
309     if (build_dir)
310     {
311         char *dir = HeapAlloc( GetProcessHeap(), 0, strlen(build_dir) + sizeof("/dlls") );
312         strcpy( dir, build_dir );
313         strcat( dir, "/dlls" );
314         load_library_list_from_dir( dialog, dir, TRUE );
315         HeapFree( GetProcessHeap(), 0, dir );
316     }
317
318     while ((path = wine_dll_enum_load_path( i++ )))
319         load_library_list_from_dir( dialog, path, FALSE );
320
321     /* get rid of duplicate entries */
322
323     SendDlgItemMessageA( dialog, IDC_DLLCOMBO, CB_GETLBTEXT, 0, (LPARAM)item1 );
324     i = 1;
325     while (SendDlgItemMessageA( dialog, IDC_DLLCOMBO, CB_GETLBTEXT, i, (LPARAM)item2 ) >= 0)
326     {
327         if (!strcmp( item1, item2 ))
328         {
329             SendDlgItemMessageA( dialog, IDC_DLLCOMBO, CB_DELETESTRING, i, 0 );
330         }
331         else
332         {
333             strcpy( item1, item2 );
334             i++;
335         }
336     }
337     SetCursor( old_cursor );
338 }
339
340 static void load_library_settings(HWND dialog)
341 {
342     char **overrides = enumerate_values(config_key, keypath("DllOverrides"));
343     char **p;
344     int sel, count = 0;
345
346     sel = SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETCURSEL, 0, 0);
347
348     WINE_TRACE("sel=%d\n", sel);
349
350     clear_settings(dialog);
351     
352     if (!overrides || *overrides == NULL)
353     {
354         set_controls_from_selection(dialog);
355         disable(IDC_DLLS_EDITDLL);
356         disable(IDC_DLLS_REMOVEDLL);
357         HeapFree(GetProcessHeap(), 0, overrides);
358         return;
359     }
360
361     enable(IDC_DLLS_EDITDLL);
362     enable(IDC_DLLS_REMOVEDLL);
363     
364     for (p = overrides; *p != NULL; p++)
365     {
366         int index;
367         char *str, *value;
368         const char *label;
369         struct dll *dll;
370
371         value = get_reg_key(config_key, keypath("DllOverrides"), *p, NULL);
372
373         label = mode_to_label(string_to_mode(value));
374         
375         str = HeapAlloc(GetProcessHeap(), 0, strlen(*p) + 2 + strlen(label) + 2);
376         strcpy(str, *p);
377         strcat(str, " (");
378         strcat(str, label);
379         strcat(str, ")");
380
381         dll = HeapAlloc(GetProcessHeap(), 0, sizeof(struct dll));
382         dll->name = *p;
383         dll->mode = string_to_mode(value);
384
385         index = SendDlgItemMessageA(dialog, IDC_DLLS_LIST, LB_ADDSTRING, (WPARAM) -1, (LPARAM) str);
386         SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_SETITEMDATA, index, (LPARAM) dll);
387
388         HeapFree(GetProcessHeap(), 0, str);
389
390         count++;
391     }
392
393     HeapFree(GetProcessHeap(), 0, overrides);
394
395     /* restore the previous selection, if possible  */
396     if (sel >= count - 1) sel = count - 1;
397     else if (sel == -1) sel = 0;
398
399     SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_SETCURSEL, sel, 0);
400
401     set_controls_from_selection(dialog);
402 }
403
404 /* Called when the application is initialized (cannot reinit!)  */
405 static void init_libsheet(HWND dialog)
406 {
407     /* clear the add dll controls  */
408     SendDlgItemMessageW(dialog, IDC_DLLCOMBO, WM_SETTEXT, 1, (LPARAM)emptyW);
409     load_library_list( dialog );
410     disable(IDC_DLLS_ADDDLL);
411 }
412
413 static void on_add_combo_change(HWND dialog)
414 {
415     WCHAR buffer[1024];
416     int sel, len;
417
418     SendDlgItemMessageW(dialog, IDC_DLLCOMBO, WM_GETTEXT, sizeof(buffer)/sizeof(WCHAR), (LPARAM) buffer);
419     /* if lib was chosen from combobox, we receive an empty buffer, check manually */
420     sel=SendDlgItemMessageW(dialog, IDC_DLLCOMBO, CB_GETCURSEL, 0, 0);
421     len=SendDlgItemMessageW(dialog, IDC_DLLCOMBO, CB_GETLBTEXTLEN, sel, 0);
422
423     if (buffer[0] || len>0)
424         enable(IDC_DLLS_ADDDLL)
425     else
426         disable(IDC_DLLS_ADDDLL);
427 }
428
429 static void set_dllmode(HWND dialog, DWORD id)
430 {
431     enum dllmode mode;
432     struct dll *dll;
433     int sel;
434     const char *str;
435
436     mode = id_to_mode(id);
437
438     sel = SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETCURSEL, 0, 0);
439     if (sel == -1) return;
440
441     dll = (struct dll *) SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETITEMDATA, sel, 0);
442
443     str = mode_to_string(mode);
444     WINE_TRACE("Setting %s to %s\n", dll->name, str);
445
446     SendMessageW(GetParent(dialog), PSM_CHANGED, 0, 0);
447     set_reg_key(config_key, keypath("DllOverrides"), dll->name, str);
448
449     load_library_settings(dialog);  /* ... and refresh  */
450 }
451
452 static void on_add_click(HWND dialog)
453 {
454     static const char dotDll[] = ".dll";
455     char buffer[1024], *ptr;
456
457     ZeroMemory(buffer, sizeof(buffer));
458
459     SendDlgItemMessageA(dialog, IDC_DLLCOMBO, WM_GETTEXT, sizeof(buffer), (LPARAM) buffer);
460     if (lstrlenA(buffer) >= sizeof(dotDll))
461     {
462         ptr = buffer + lstrlenA(buffer) - sizeof(dotDll) + 1;
463         if (!lstrcmpiA(ptr, dotDll))
464         {
465             WINE_TRACE("Stripping dll extension\n");
466             *ptr = '\0';
467         }
468     }
469
470     /* check if dll is in the builtin-only list */
471     if (!(ptr = strrchr( buffer, '\\' )))
472     {
473         ptr = buffer;
474         if (*ptr == '*') ptr++;
475     }
476     else ptr++;
477     if (is_builtin_only( ptr ))
478     {
479         MSGBOXPARAMSA params;
480         params.cbSize = sizeof(params);
481         params.hwndOwner = dialog;
482         params.hInstance = GetModuleHandleA( NULL );
483         params.lpszText = MAKEINTRESOURCEA( IDS_DLL_WARNING );
484         params.lpszCaption = MAKEINTRESOURCEA( IDS_DLL_WARNING_CAPTION );
485         params.dwStyle = MB_ICONWARNING | MB_YESNO;
486         params.lpszIcon = NULL;
487         params.dwContextHelpId = 0;
488         params.lpfnMsgBoxCallback = NULL;
489         params.dwLanguageId = 0;
490         if (MessageBoxIndirectA( &params ) != IDYES) return;
491     }
492
493     SendDlgItemMessageW(dialog, IDC_DLLCOMBO, WM_SETTEXT, 0, (LPARAM)emptyW);
494     disable(IDC_DLLS_ADDDLL);
495     
496     WINE_TRACE("Adding %s as native, builtin\n", buffer);
497
498     SendMessageW(GetParent(dialog), PSM_CHANGED, 0, 0);
499     set_reg_key(config_key, keypath("DllOverrides"), buffer, "native,builtin");
500
501     load_library_settings(dialog);
502
503     SendDlgItemMessageA(dialog, IDC_DLLS_LIST, LB_SELECTSTRING, 0, (LPARAM) buffer);
504
505     set_controls_from_selection(dialog);
506 }
507
508 static INT_PTR CALLBACK loadorder_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
509 {
510     static WORD sel;
511
512     switch(uMsg) 
513     {
514     case WM_INITDIALOG:
515         CheckRadioButton(hwndDlg, IDC_RAD_BUILTIN, IDC_RAD_DISABLE, lParam);
516         sel = lParam;
517         return TRUE;
518
519     case WM_COMMAND:
520         if(HIWORD(wParam) != BN_CLICKED) break;
521         switch (LOWORD(wParam))
522         {
523         case IDC_RAD_BUILTIN:
524         case IDC_RAD_NATIVE:
525         case IDC_RAD_BUILTIN_NATIVE:
526         case IDC_RAD_NATIVE_BUILTIN:
527         case IDC_RAD_DISABLE:
528             sel = LOWORD(wParam);
529             return TRUE;
530         case IDOK:
531             EndDialog(hwndDlg, sel);
532             return TRUE;
533         case IDCANCEL:
534             EndDialog(hwndDlg, wParam);
535             return TRUE;
536         }
537     }
538     return FALSE;
539 }
540
541 static void on_edit_click(HWND hwnd)
542 {
543     INT_PTR ret;
544     int index = SendDlgItemMessageW(hwnd, IDC_DLLS_LIST, LB_GETCURSEL, 0, 0);
545     struct dll *dll;
546     DWORD id;
547
548     /* if no override is selected the edit button should be disabled... */
549     assert(index != -1);
550
551     dll = (struct dll *) SendDlgItemMessageW(hwnd, IDC_DLLS_LIST, LB_GETITEMDATA, index, 0);
552     id = mode_to_id(dll->mode);
553
554     ret = DialogBoxParamW(0, MAKEINTRESOURCEW(IDD_LOADORDER), hwnd, loadorder_dlgproc, id);
555
556     if(ret != IDCANCEL)
557         set_dllmode(hwnd, ret);
558 }
559
560 static void on_remove_click(HWND dialog)
561 {
562     int sel = SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETCURSEL, 0, 0);
563     struct dll *dll;
564
565     if (sel == LB_ERR) return;
566
567     dll = (struct dll *) SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETITEMDATA, sel, 0);
568
569     SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_DELETESTRING, sel, 0);
570
571     SendMessageW(GetParent(dialog), PSM_CHANGED, 0, 0);
572     set_reg_key(config_key, keypath("DllOverrides"), dll->name, NULL);
573
574     HeapFree(GetProcessHeap(), 0, dll->name);
575     HeapFree(GetProcessHeap(), 0, dll);
576
577     if (SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_GETCOUNT, 0, 0) > 0)
578         SendDlgItemMessageW(dialog, IDC_DLLS_LIST, LB_SETCURSEL, max(sel - 1, 0), 0);
579     else
580     {
581         disable(IDC_DLLS_EDITDLL);
582         disable(IDC_DLLS_REMOVEDLL);
583     }
584
585     set_controls_from_selection(dialog);
586 }
587
588 INT_PTR CALLBACK
589 LibrariesDlgProc (HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
590 {
591         switch (uMsg)
592         {
593         case WM_INITDIALOG:
594                 init_libsheet(hDlg);
595                 break;
596         case WM_SHOWWINDOW:
597                 set_window_title(hDlg);
598                 break;
599         case WM_NOTIFY:
600                 switch (((LPNMHDR)lParam)->code) {
601                 case PSN_SETACTIVE:
602                     load_library_settings(hDlg);
603                     break;
604                 }
605                 break;
606         case WM_COMMAND:
607                 switch(HIWORD(wParam)) {
608
609                     /* FIXME: when the user hits enter in the DLL combo box we should invoke the add
610                      * add button, rather than the propsheet OK button. But I don't know how to do that!
611                      */
612                     
613                 case CBN_EDITCHANGE:
614                         if(LOWORD(wParam) == IDC_DLLCOMBO)
615                         {
616                             on_add_combo_change(hDlg);
617                             break;
618                         }
619
620                 case BN_CLICKED:
621                         switch(LOWORD(wParam)) {
622                         case IDC_DLLS_ADDDLL:
623                             on_add_click(hDlg);
624                             break;
625                         case IDC_DLLS_EDITDLL:
626                             on_edit_click(hDlg);
627                             break;
628                         case IDC_DLLS_REMOVEDLL:
629                             on_remove_click(hDlg);
630                             break;
631                         }
632                         break;
633                 case LBN_SELCHANGE:
634                         if(LOWORD(wParam) == IDC_DLLCOMBO)
635                             on_add_combo_change(hDlg);
636                         else
637                             set_controls_from_selection(hDlg);
638                         break;
639                 }
640                 break;
641         }
642
643         return 0;
644 }