Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282/* * SetupAPI dialog functions * * Copyright 2009 Ricardo Filipe * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#include "setupapi_private.h"
struct promptdisk_params { PCWSTR DialogTitle; PCWSTR DiskName; PCWSTR PathToSource; PCWSTR FileSought; PCWSTR TagFile; DWORD DiskPromptStyle; PWSTR PathBuffer; DWORD PathBufferSize; PDWORD PathRequiredSize;};
/* initiates the fields of the SetupPromptForDisk dialog according to the parameters*/static void promptdisk_init(HWND hwnd, struct promptdisk_params *params){ SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)params);
if(params->DialogTitle) SetWindowTextW(hwnd, params->DialogTitle); if(params->PathToSource) SetDlgItemTextW(hwnd, IDC_PATH, params->PathToSource);
if(!(params->DiskPromptStyle & IDF_OEMDISK)) { WCHAR message[256+2*MAX_PATH]; WCHAR format[256]; WCHAR unknown[256]; DWORD_PTR args[2]; LoadStringW(hInstance, IDS_PROMPTDISK, format, sizeof(format)/sizeof(format[0]));
args[0] = (DWORD_PTR)params->FileSought; if(params->DiskName) args[1] = (DWORD_PTR)params->DiskName; else { LoadStringW(hInstance, IDS_UNKNOWN, unknown, sizeof(unknown)/sizeof(unknown[0])); args[1] = (DWORD_PTR)unknown; } FormatMessageW(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY, format, 0, 0, message, sizeof(message)/sizeof(*message), (__ms_va_list*)args); SetDlgItemTextW(hwnd, IDC_FILENEEDED, message);
LoadStringW(hInstance, IDS_INFO, message, sizeof(message)/sizeof(message[0])); SetDlgItemTextW(hwnd, IDC_INFO, message); LoadStringW(hInstance, IDS_COPYFROM, message, sizeof(message)/sizeof(message[0])); SetDlgItemTextW(hwnd, IDC_COPYFROM, message); } if(params->DiskPromptStyle & IDF_NOBROWSE) ShowWindow(GetDlgItem(hwnd, IDC_RUNDLG_BROWSE), SW_HIDE);}
/* When the user clicks in the Ok button in SetupPromptForDisk dialog * if the parameters are good it copies the path from the dialog to the output buffer * saves the required size for the buffer if PathRequiredSize is given * returns NO_ERROR if there is no PathBuffer to copy too * returns DPROMPT_BUFFERTOOSMALL if the path is too big to fit in PathBuffer */static void promptdisk_ok(HWND hwnd, struct promptdisk_params *params){ int requiredSize; WCHAR aux[MAX_PATH]; GetWindowTextW(GetDlgItem(hwnd, IDC_PATH), aux, MAX_PATH); requiredSize = strlenW(aux)+1;
if(params->PathRequiredSize) { *params->PathRequiredSize = requiredSize; TRACE("returning PathRequiredSize=%d\n",*params->PathRequiredSize); } if(!params->PathBuffer) { EndDialog(hwnd, NO_ERROR); return; } if(requiredSize > params->PathBufferSize) { EndDialog(hwnd, DPROMPT_BUFFERTOOSMALL); return; } strcpyW(params->PathBuffer, aux); TRACE("returning PathBuffer=%s\n", debugstr_w(params->PathBuffer)); EndDialog(hwnd, DPROMPT_SUCCESS);}
/* When the user clicks the browse button in SetupPromptForDisk dialog * it copies the path of the selected file to the dialog path field */static void promptdisk_browse(HWND hwnd, struct promptdisk_params *params){ OPENFILENAMEW ofn; ZeroMemory(&ofn, sizeof(ofn));
ofn.lStructSize = sizeof(ofn); ofn.Flags = OFN_EXPLORER | OFN_HIDEREADONLY | OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST; ofn.hwndOwner = hwnd; ofn.nMaxFile = MAX_PATH; ofn.lpstrFile = HeapAlloc(GetProcessHeap(), 0, MAX_PATH*sizeof(WCHAR)); strcpyW(ofn.lpstrFile, params->FileSought);
if(GetOpenFileNameW(&ofn)) { WCHAR* last_slash = strrchrW(ofn.lpstrFile, '\\'); if (last_slash) *last_slash = 0; SetDlgItemTextW(hwnd, IDC_PATH, ofn.lpstrFile); } HeapFree(GetProcessHeap(), 0, ofn.lpstrFile);}
/* Handles the messages sent to the SetupPromptForDisk dialog*/static INT_PTR CALLBACK promptdisk_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam){ switch(msg) { case WM_INITDIALOG: promptdisk_init(hwnd, (struct promptdisk_params *)lParam); return TRUE; case WM_COMMAND: switch(wParam) { case IDOK: { struct promptdisk_params *params = (struct promptdisk_params *)GetWindowLongPtrW(hwnd, DWLP_USER); promptdisk_ok(hwnd, params); return TRUE; } case IDCANCEL: EndDialog(hwnd, DPROMPT_CANCEL); return TRUE; case IDC_RUNDLG_BROWSE: { struct promptdisk_params *params = (struct promptdisk_params *)GetWindowLongPtrW(hwnd, DWLP_USER); promptdisk_browse(hwnd, params); return TRUE; } } } return FALSE;}
/*********************************************************************** * SetupPromptForDiskA (SETUPAPI.@) */UINT WINAPI SetupPromptForDiskA(HWND hwndParent, PCSTR DialogTitle, PCSTR DiskName, PCSTR PathToSource, PCSTR FileSought, PCSTR TagFile, DWORD DiskPromptStyle, PSTR PathBuffer, DWORD PathBufferSize, PDWORD PathRequiredSize){ WCHAR *DialogTitleW, *DiskNameW, *PathToSourceW; WCHAR *FileSoughtW, *TagFileW, PathBufferW[MAX_PATH]; UINT ret, length;
TRACE("%p, %s, %s, %s, %s, %s, 0x%08x, %p, %d, %p\n", hwndParent, debugstr_a(DialogTitle), debugstr_a(DiskName), debugstr_a(PathToSource), debugstr_a(FileSought), debugstr_a(TagFile), DiskPromptStyle, PathBuffer, PathBufferSize, PathRequiredSize);
DialogTitleW = strdupAtoW(DialogTitle); DiskNameW = strdupAtoW(DiskName); PathToSourceW = strdupAtoW(PathToSource); FileSoughtW = strdupAtoW(FileSought); TagFileW = strdupAtoW(TagFile);
ret = SetupPromptForDiskW(hwndParent, DialogTitleW, DiskNameW, PathToSourceW, FileSoughtW, TagFileW, DiskPromptStyle, PathBufferW, MAX_PATH, PathRequiredSize);
HeapFree(GetProcessHeap(), 0, DialogTitleW); HeapFree(GetProcessHeap(), 0, DiskNameW); HeapFree(GetProcessHeap(), 0, PathToSourceW); HeapFree(GetProcessHeap(), 0, FileSoughtW); HeapFree(GetProcessHeap(), 0, TagFileW);
if(ret == DPROMPT_SUCCESS) { length = WideCharToMultiByte(CP_ACP, 0, PathBufferW, -1, NULL, 0, NULL, NULL); if(PathRequiredSize) *PathRequiredSize = length; if(PathBuffer) { if(length > PathBufferSize) return DPROMPT_BUFFERTOOSMALL; WideCharToMultiByte(CP_ACP, 0, PathBufferW, -1, PathBuffer, length, NULL, NULL); } } return ret;}
/*********************************************************************** * SetupPromptForDiskW (SETUPAPI.@) */UINT WINAPI SetupPromptForDiskW(HWND hwndParent, PCWSTR DialogTitle, PCWSTR DiskName, PCWSTR PathToSource, PCWSTR FileSought, PCWSTR TagFile, DWORD DiskPromptStyle, PWSTR PathBuffer, DWORD PathBufferSize, PDWORD PathRequiredSize){ struct promptdisk_params params; UINT ret;
TRACE("%p, %s, %s, %s, %s, %s, 0x%08x, %p, %d, %p\n", hwndParent, debugstr_w(DialogTitle), debugstr_w(DiskName), debugstr_w(PathToSource), debugstr_w(FileSought), debugstr_w(TagFile), DiskPromptStyle, PathBuffer, PathBufferSize, PathRequiredSize);
if(!FileSought) { SetLastError(ERROR_INVALID_PARAMETER); return DPROMPT_CANCEL; }
if (PathToSource && (DiskPromptStyle & IDF_CHECKFIRST)) { static const WCHAR format[] = {'%', 's', '\\', '%', 's', '\0'}; WCHAR filepath[MAX_PATH];
if (strlenW(PathToSource) + 1 + strlenW(FileSought) < sizeof(filepath)) { snprintfW(filepath, MAX_PATH, format, PathToSource, FileSought);
if (GetFileAttributesW(filepath) != INVALID_FILE_ATTRIBUTES) { if (PathRequiredSize) *PathRequiredSize = strlenW(PathToSource) + 1;
if (!PathBuffer) return DPROMPT_SUCCESS;
if (PathBufferSize >= strlenW(PathToSource) + 1) { strcpyW(PathBuffer, PathToSource); return DPROMPT_SUCCESS; } else return DPROMPT_BUFFERTOOSMALL; } } }
params.DialogTitle = DialogTitle; params.DiskName = DiskName; params.PathToSource = PathToSource; params.FileSought = FileSought; params.TagFile = TagFile; params.DiskPromptStyle = DiskPromptStyle; params.PathBuffer = PathBuffer; params.PathBufferSize = PathBufferSize; params.PathRequiredSize = PathRequiredSize;
ret = DialogBoxParamW(hInstance, MAKEINTRESOURCEW(IDPROMPTFORDISK), hwndParent, promptdisk_proc, (LPARAM)¶ms);
if(ret == DPROMPT_CANCEL) SetLastError(ERROR_CANCELLED); return ret;}