dlls/setupapi/dialog.c

   1 /*
   2  * SetupAPI dialog functions
   3  *
   4  * Copyright 2009 Ricardo Filipe
   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "wine/debug.h"
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "wingdi.h"
  27 #include "winuser.h"
  28 #include "winreg.h"
  29 #include "commdlg.h"
  30 #include "setupapi.h"
  31 #include "winnls.h"
  32 #include "setupapi_private.h"
  33
  34 #include "wine/unicode.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
  37
  38 struct promptdisk_params {
  39     PCWSTR DialogTitle;
  40     PCWSTR DiskName;
  41     PCWSTR PathToSource;
  42     PCWSTR FileSought;
  43     PCWSTR TagFile;
  44     DWORD DiskPromptStyle;
  45     PWSTR PathBuffer;
  46     DWORD PathBufferSize;
  47     PDWORD PathRequiredSize;
  48 };
  49
  50 /* initiates the fields of the SetupPromptForDisk dialog according to the parameters
  51 */
  52 static void promptdisk_init(HWND hwnd, struct promptdisk_params *params)
  53 {
  54     WCHAR format[256];
  55     WCHAR message[256];
  56
  57     SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)params);
  58
  59     if(params->DialogTitle)
  60         SetWindowTextW(hwnd, params->DialogTitle);
  61     if(params->PathToSource)
  62         SetDlgItemTextW(hwnd, IDC_PATH, params->PathToSource);
  63
  64     if(!(params->DiskPromptStyle & IDF_OEMDISK))
  65     {
  66         LoadStringW(SETUPAPI_hInstance, IDS_PROMPTDISK, format,
  67             sizeof(format)/sizeof(format[0]));
  68
  69         if(params->DiskName)
  70             snprintfW(message, sizeof(message)/sizeof(message[0]), format,
  71                 params->FileSought, params->DiskName);
  72         else
  73         {
  74             WCHAR unknown[256];
  75             LoadStringW(SETUPAPI_hInstance, IDS_UNKNOWN, unknown,
  76                 sizeof(unknown)/sizeof(unknown[0]));
  77             snprintfW(message, sizeof(message)/sizeof(message[0]), format,
  78                 params->FileSought, unknown);
  79         }
  80         SetDlgItemTextW(hwnd, IDC_FILENEEDED, message);
  81
  82         LoadStringW(SETUPAPI_hInstance, IDS_INFO, message,
  83             sizeof(message)/sizeof(message[0]));
  84         SetDlgItemTextW(hwnd, IDC_INFO, message);
  85         LoadStringW(SETUPAPI_hInstance, IDS_COPYFROM, message,
  86             sizeof(message)/sizeof(message[0]));
  87         SetDlgItemTextW(hwnd, IDC_COPYFROM, message);
  88     }
  89     if(params->DiskPromptStyle & IDF_NOBROWSE)
  90         ShowWindow(GetDlgItem(hwnd, IDC_RUNDLG_BROWSE), SW_HIDE);
  91 }
  92
  93 /* When the user clicks in the Ok button in SetupPromptForDisk dialog
  94  * if the parameters are good it copies the path from the dialog to the output buffer
  95  * saves the required size for the buffer if PathRequiredSize is given
  96  * returns NO_ERROR if there is no PathBuffer to copy too
  97  * returns DPROMPT_BUFFERTOOSMALL if the path is too big to fit in PathBuffer
  98  */
  99 static void promptdisk_ok(HWND hwnd, struct promptdisk_params *params)
 100 {
 101     int requiredSize;
 102     WCHAR aux[MAX_PATH];
 103     GetWindowTextW(GetDlgItem(hwnd, IDC_PATH), aux, MAX_PATH);
 104     requiredSize = strlenW(aux)+1;
 105
 106     if(params->PathRequiredSize)
 107     {
 108         *params->PathRequiredSize = requiredSize;
 109         TRACE("returning PathRequiredSize=%d\n",*params->PathRequiredSize);
 110     }
 111     if(!params->PathBuffer)
 112     {
 113         EndDialog(hwnd, NO_ERROR);
 114         return;
 115     }
 116     if(requiredSize > params->PathBufferSize)
 117     {
 118         EndDialog(hwnd, DPROMPT_BUFFERTOOSMALL);
 119         return;
 120     }
 121     strcpyW(params->PathBuffer, aux);
 122     TRACE("returning PathBuffer=%s\n", debugstr_w(params->PathBuffer));
 123     EndDialog(hwnd, DPROMPT_SUCCESS);
 124 }
 125
 126 /* When the user clicks the browse button in SetupPromptForDisk dialog
 127  * it copies the path of the selected file to the dialog path field
 128  */
 129 static void promptdisk_browse(HWND hwnd, struct promptdisk_params *params)
 130 {
 131     OPENFILENAMEW ofn;
 132     ZeroMemory(&ofn, sizeof(ofn));
 133
 134     ofn.lStructSize = sizeof(ofn);
 135     ofn.Flags = OFN_HIDEREADONLY | OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST;
 136     ofn.hwndOwner = hwnd;
 137     ofn.nMaxFile = MAX_PATH;
 138     ofn.lpstrFile = HeapAlloc(GetProcessHeap(), 0, MAX_PATH*sizeof(WCHAR));
 139     strcpyW(ofn.lpstrFile, params->FileSought);
 140
 141     if(GetOpenFileNameW(&ofn))
 142     {
 143         WCHAR* last_slash = strrchrW(ofn.lpstrFile, '\\');
 144         if (last_slash) *last_slash = 0;
 145         SetDlgItemTextW(hwnd, IDC_PATH, ofn.lpstrFile);
 146     }
 147     HeapFree(GetProcessHeap(), 0, ofn.lpstrFile);
 148 }
 149
 150 /* Handles the messages sent to the SetupPromptForDisk dialog
 151 */
 152 static INT_PTR CALLBACK promptdisk_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
 153 {
 154     switch(msg)
 155     {
 156         case WM_INITDIALOG:
 157             promptdisk_init(hwnd, (struct promptdisk_params *)lParam);
 158             return TRUE;
 159         case WM_COMMAND:
 160             switch(wParam)
 161             {
 162                 case IDOK:
 163                 {
 164                     struct promptdisk_params *params =
 165                         (struct promptdisk_params *)GetWindowLongPtrW(hwnd, DWLP_USER);
 166                     promptdisk_ok(hwnd, params);
 167                     return TRUE;
 168                 }
 169                 case IDCANCEL:
 170                     EndDialog(hwnd, DPROMPT_CANCEL);
 171                     return TRUE;
 172                 case IDC_RUNDLG_BROWSE:
 173                 {
 174                     struct promptdisk_params *params =
 175                         (struct promptdisk_params *)GetWindowLongPtrW(hwnd, DWLP_USER);
 176                     promptdisk_browse(hwnd, params);
 177                     return TRUE;
 178                 }
 179             }
 180     }
 181     return FALSE;
 182 }
 183
 184 /***********************************************************************
 185  *      SetupPromptForDiskA (SETUPAPI.@)
 186  */
 187 UINT WINAPI SetupPromptForDiskA(HWND hwndParent, PCSTR DialogTitle, PCSTR DiskName,
 188         PCSTR PathToSource, PCSTR FileSought, PCSTR TagFile, DWORD DiskPromptStyle,
 189         PSTR PathBuffer, DWORD PathBufferSize, PDWORD PathRequiredSize)
 190 {
 191     WCHAR *DialogTitleW, *DiskNameW, *PathToSourceW;
 192     WCHAR *FileSoughtW, *TagFileW, PathBufferW[MAX_PATH];
 193     UINT ret, length;
 194
 195     TRACE("%p, %s, %s, %s, %s, %s, 0x%08x, %p, %d, %p\n", hwndParent, debugstr_a(DialogTitle),
 196           debugstr_a(DiskName), debugstr_a(PathToSource), debugstr_a(FileSought),
 197           debugstr_a(TagFile), DiskPromptStyle, PathBuffer, PathBufferSize,
 198           PathRequiredSize);
 199
 200     DialogTitleW = strdupAtoW(DialogTitle);
 201     DiskNameW = strdupAtoW(DiskName);
 202     PathToSourceW = strdupAtoW(PathToSource);
 203     FileSoughtW = strdupAtoW(FileSought);
 204     TagFileW = strdupAtoW(TagFile);
 205
 206     ret = SetupPromptForDiskW(hwndParent, DialogTitleW, DiskNameW, PathToSourceW,
 207             FileSoughtW, TagFileW, DiskPromptStyle, PathBufferW, MAX_PATH, PathRequiredSize);
 208
 209     HeapFree(GetProcessHeap(), 0, DialogTitleW);
 210     HeapFree(GetProcessHeap(), 0, DiskNameW);
 211     HeapFree(GetProcessHeap(), 0, PathToSourceW);
 212     HeapFree(GetProcessHeap(), 0, FileSoughtW);
 213     HeapFree(GetProcessHeap(), 0, TagFileW);
 214
 215     if(ret == DPROMPT_SUCCESS)
 216     {
 217         length = WideCharToMultiByte(CP_ACP, 0, PathBufferW, -1, NULL, 0, NULL, NULL);
 218         if(PathRequiredSize) *PathRequiredSize = length;
 219         if(PathBuffer)
 220         {
 221             if(length > PathBufferSize)
 222                 return DPROMPT_BUFFERTOOSMALL;
 223             WideCharToMultiByte(CP_ACP, 0, PathBufferW, -1, PathBuffer, length, NULL, NULL);
 224         }
 225     }
 226     return ret;
 227 }
 228
 229 /***********************************************************************
 230  *      SetupPromptForDiskW (SETUPAPI.@)
 231  */
 232 UINT WINAPI SetupPromptForDiskW(HWND hwndParent, PCWSTR DialogTitle, PCWSTR DiskName,
 233         PCWSTR PathToSource, PCWSTR FileSought, PCWSTR TagFile, DWORD DiskPromptStyle,
 234         PWSTR PathBuffer, DWORD PathBufferSize, PDWORD PathRequiredSize)
 235 {
 236     struct promptdisk_params params;
 237     UINT ret;
 238
 239     TRACE("%p, %s, %s, %s, %s, %s, 0x%08x, %p, %d, %p\n", hwndParent, debugstr_w(DialogTitle),
 240           debugstr_w(DiskName), debugstr_w(PathToSource), debugstr_w(FileSought),
 241           debugstr_w(TagFile), DiskPromptStyle, PathBuffer, PathBufferSize,
 242           PathRequiredSize);
 243
 244     if(!FileSought)
 245     {
 246         SetLastError(ERROR_INVALID_PARAMETER);
 247         return DPROMPT_CANCEL;
 248     }
 249     params.DialogTitle = DialogTitle;
 250     params.DiskName = DiskName;
 251     params.PathToSource = PathToSource;
 252     params.FileSought = FileSought;
 253     params.TagFile = TagFile;
 254     params.DiskPromptStyle = DiskPromptStyle;
 255     params.PathBuffer = PathBuffer;
 256     params.PathBufferSize = PathBufferSize;
 257     params.PathRequiredSize = PathRequiredSize;
 258
 259     ret = DialogBoxParamW(SETUPAPI_hInstance, MAKEINTRESOURCEW(IDPROMPTFORDISK),
 260         hwndParent, promptdisk_proc, (LPARAM)&params);
 261
 262     if(ret == DPROMPT_CANCEL)
 263         SetLastError(ERROR_CANCELLED);
 264     return ret;
 265 }