comdlg32: Consolidate file dialog initialization to avoid duplication.

Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Nikolay Sivov 2017-02-21 15:35:16 +03:00 committed by Alexandre Julliard
parent aef84332a2
commit c23a91cb63
1 changed files with 148 additions and 205 deletions

View File

@ -318,236 +318,162 @@ static BOOL GetFileName95(FileOpenDlgInfos *fodInfos)
return lRes; return lRes;
} }
/*********************************************************************** static WCHAR *heap_strdupAtoW(const char *str)
* GetFileDialog95A
*
* Call GetFileName95 with this structure and clean the memory.
*
* IN : The OPENFILENAMEA initialisation structure passed to
* GetOpenFileNameA win api function (see filedlg.c)
*/
static BOOL GetFileDialog95A(LPOPENFILENAMEA ofn,UINT iDlgType)
{ {
BOOL ret; WCHAR *ret;
FileOpenDlgInfos fodInfos; INT len;
LPSTR lpstrSavDir = NULL;
LPWSTR title = NULL;
LPWSTR defext = NULL;
LPWSTR filter = NULL;
LPWSTR customfilter = NULL;
INITCOMMONCONTROLSEX icc;
/* Initialize ComboBoxEx32 */ if (!str)
icc.dwSize = sizeof(icc); return NULL;
icc.dwICC = ICC_USEREX_CLASSES;
InitCommonControlsEx(&icc);
/* Initialize CommDlgExtendedError() */ len = MultiByteToWideChar(CP_ACP, 0, str, -1, 0, 0);
COMDLG32_SetCommDlgExtendedError(0); ret = MemAlloc(len * sizeof(WCHAR));
MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
/* Initialize FileOpenDlgInfos structure */ return ret;
ZeroMemory(&fodInfos, sizeof(FileOpenDlgInfos)); }
/* Pass in the original ofn */ static void init_filedlg_infoW(OPENFILENAMEW *ofn, FileOpenDlgInfos *info)
fodInfos.ofnInfos = (LPOPENFILENAMEW)ofn; {
INITCOMMONCONTROLSEX icc;
/* save current directory */ /* Initialize ComboBoxEx32 */
if (ofn->Flags & OFN_NOCHANGEDIR) icc.dwSize = sizeof(icc);
{ icc.dwICC = ICC_USEREX_CLASSES;
lpstrSavDir = MemAlloc(MAX_PATH); InitCommonControlsEx(&icc);
GetCurrentDirectoryA(MAX_PATH, lpstrSavDir);
}
fodInfos.unicode = FALSE; /* Initialize CommDlgExtendedError() */
COMDLG32_SetCommDlgExtendedError(0);
/* convert all the input strings to unicode */ memset(info, 0, sizeof(*info));
if(ofn->lpstrInitialDir)
{
DWORD len = MultiByteToWideChar( CP_ACP, 0, ofn->lpstrInitialDir, -1, NULL, 0 );
fodInfos.initdir = MemAlloc((len+1)*sizeof(WCHAR));
MultiByteToWideChar( CP_ACP, 0, ofn->lpstrInitialDir, -1, fodInfos.initdir, len);
}
else
fodInfos.initdir = NULL;
if(ofn->lpstrFile) /* Pass in the original ofn */
{ info->ofnInfos = ofn;
fodInfos.filename = MemAlloc(ofn->nMaxFile*sizeof(WCHAR));
MultiByteToWideChar( CP_ACP, 0, ofn->lpstrFile, -1, fodInfos.filename, ofn->nMaxFile);
}
else
fodInfos.filename = NULL;
if(ofn->lpstrDefExt) info->title = ofn->lpstrTitle;
{ info->defext = ofn->lpstrDefExt;
DWORD len = MultiByteToWideChar( CP_ACP, 0, ofn->lpstrDefExt, -1, NULL, 0 ); info->filter = ofn->lpstrFilter;
defext = MemAlloc((len+1)*sizeof(WCHAR)); info->customfilter = ofn->lpstrCustomFilter;
MultiByteToWideChar( CP_ACP, 0, ofn->lpstrDefExt, -1, defext, len);
}
fodInfos.defext = defext;
if(ofn->lpstrTitle) if (ofn->lpstrFile)
{ {
DWORD len = MultiByteToWideChar( CP_ACP, 0, ofn->lpstrTitle, -1, NULL, 0 ); info->filename = MemAlloc(ofn->nMaxFile * sizeof(WCHAR));
title = MemAlloc((len+1)*sizeof(WCHAR)); lstrcpynW(info->filename, ofn->lpstrFile, ofn->nMaxFile);
MultiByteToWideChar( CP_ACP, 0, ofn->lpstrTitle, -1, title, len); }
}
fodInfos.title = title;
if (ofn->lpstrFilter) if (ofn->lpstrInitialDir)
{ {
LPCSTR s; DWORD len = ExpandEnvironmentStringsW(ofn->lpstrInitialDir, NULL, 0);
int n, len; if (len)
{
info->initdir = MemAlloc(len * sizeof(WCHAR));
ExpandEnvironmentStringsW(ofn->lpstrInitialDir, info->initdir, len);
}
}
/* filter is a list... title\0ext\0......\0\0 */ info->unicode = TRUE;
s = ofn->lpstrFilter; }
while (*s) s = s+strlen(s)+1;
s++;
n = s - ofn->lpstrFilter;
len = MultiByteToWideChar( CP_ACP, 0, ofn->lpstrFilter, n, NULL, 0 );
filter = MemAlloc(len*sizeof(WCHAR));
MultiByteToWideChar( CP_ACP, 0, ofn->lpstrFilter, n, filter, len );
}
fodInfos.filter = filter;
/* convert lpstrCustomFilter */ static void init_filedlg_infoA(OPENFILENAMEA *ofn, FileOpenDlgInfos *info)
if (ofn->lpstrCustomFilter) {
{ OPENFILENAMEW ofnW;
LPCSTR s;
int n, len;
/* customfilter contains a pair of strings... title\0ext\0 */ ofnW = *(OPENFILENAMEW *)ofn;
s = ofn->lpstrCustomFilter;
if (*s) s = s+strlen(s)+1;
if (*s) s = s+strlen(s)+1;
n = s - ofn->lpstrCustomFilter;
len = MultiByteToWideChar( CP_ACP, 0, ofn->lpstrCustomFilter, n, NULL, 0 );
customfilter = MemAlloc(len*sizeof(WCHAR));
MultiByteToWideChar( CP_ACP, 0, ofn->lpstrCustomFilter, n, customfilter, len );
}
fodInfos.customfilter = customfilter;
/* Initialize the dialog property */ ofnW.lpstrInitialDir = heap_strdupAtoW(ofn->lpstrInitialDir);
fodInfos.DlgInfos.dwDlgProp = 0; ofnW.lpstrFile = heap_strdupAtoW(ofn->lpstrFile);
fodInfos.DlgInfos.hwndCustomDlg = NULL; ofnW.lpstrDefExt = heap_strdupAtoW(ofn->lpstrDefExt);
ofnW.lpstrTitle = heap_strdupAtoW(ofn->lpstrTitle);
switch(iDlgType) if (ofn->lpstrFilter)
{ {
case OPEN_DIALOG : int n, len;
ret = GetFileName95(&fodInfos); LPCSTR s;
break;
case SAVE_DIALOG :
fodInfos.DlgInfos.dwDlgProp |= FODPROP_SAVEDLG;
ret = GetFileName95(&fodInfos);
break;
default :
ret = FALSE;
}
if (lpstrSavDir) /* filter is a list... title\0ext\0......\0\0 */
{ s = ofn->lpstrFilter;
SetCurrentDirectoryA(lpstrSavDir); while (*s) s = s+strlen(s)+1;
MemFree(lpstrSavDir); s++;
} n = s - ofn->lpstrFilter;
len = MultiByteToWideChar(CP_ACP, 0, ofn->lpstrFilter, n, NULL, 0);
ofnW.lpstrFilter = MemAlloc(len * sizeof(WCHAR));
MultiByteToWideChar(CP_ACP, 0, ofn->lpstrFilter, n, (WCHAR *)ofnW.lpstrFilter, len);
}
MemFree(title); /* convert lpstrCustomFilter */
MemFree(defext); if (ofn->lpstrCustomFilter)
MemFree(filter); {
MemFree(customfilter); int n, len;
MemFree(fodInfos.initdir); LPCSTR s;
MemFree(fodInfos.filename);
TRACE("selected file: %s\n",ofn->lpstrFile); /* customfilter contains a pair of strings... title\0ext\0 */
s = ofn->lpstrCustomFilter;
if (*s) s = s+strlen(s)+1;
if (*s) s = s+strlen(s)+1;
n = s - ofn->lpstrCustomFilter;
len = MultiByteToWideChar(CP_ACP, 0, ofn->lpstrCustomFilter, n, NULL, 0);
ofnW.lpstrCustomFilter = MemAlloc(len * sizeof(WCHAR));
MultiByteToWideChar(CP_ACP, 0, ofn->lpstrCustomFilter, n, ofnW.lpstrCustomFilter, len);
}
return ret; init_filedlg_infoW(&ofnW, info);
/* fixup A-specific fields */
info->ofnInfos = (OPENFILENAMEW *)ofn;
info->unicode = FALSE;
/* free what was duplicated */
MemFree((WCHAR *)ofnW.lpstrInitialDir);
MemFree((WCHAR *)ofnW.lpstrFile);
} }
/*********************************************************************** /***********************************************************************
* GetFileDialog95W * GetFileDialog95
* *
* Copy the OPENFILENAMEW structure in a FileOpenDlgInfos structure.
* Call GetFileName95 with this structure and clean the memory. * Call GetFileName95 with this structure and clean the memory.
*
*/ */
static BOOL GetFileDialog95W(LPOPENFILENAMEW ofn,UINT iDlgType) static BOOL GetFileDialog95(FileOpenDlgInfos *info, UINT dlg_type)
{ {
BOOL ret; WCHAR *current_dir = NULL;
FileOpenDlgInfos fodInfos; BOOL ret;
LPWSTR lpstrSavDir = NULL;
INITCOMMONCONTROLSEX icc;
/* Initialize ComboBoxEx32 */ /* save current directory */
icc.dwSize = sizeof(icc); if (info->ofnInfos->Flags & OFN_NOCHANGEDIR)
icc.dwICC = ICC_USEREX_CLASSES;
InitCommonControlsEx(&icc);
/* Initialize CommDlgExtendedError() */
COMDLG32_SetCommDlgExtendedError(0);
/* Initialize FileOpenDlgInfos structure */
ZeroMemory(&fodInfos, sizeof(FileOpenDlgInfos));
/* Pass in the original ofn */
fodInfos.ofnInfos = ofn;
fodInfos.title = ofn->lpstrTitle;
fodInfos.defext = ofn->lpstrDefExt;
fodInfos.filter = ofn->lpstrFilter;
fodInfos.customfilter = ofn->lpstrCustomFilter;
/* convert string arguments, save others */
if(ofn->lpstrFile)
{
fodInfos.filename = MemAlloc(ofn->nMaxFile*sizeof(WCHAR));
lstrcpynW(fodInfos.filename,ofn->lpstrFile,ofn->nMaxFile);
}
else
fodInfos.filename = NULL;
fodInfos.initdir = NULL;
if(ofn->lpstrInitialDir)
{
/* fodInfos.initdir = strdupW(ofn->lpstrInitialDir); */
DWORD len = ExpandEnvironmentStringsW(ofn->lpstrInitialDir, NULL, 0);
if (len)
{ {
fodInfos.initdir = MemAlloc(len * sizeof(WCHAR)); current_dir = MemAlloc(MAX_PATH * sizeof(WCHAR));
ExpandEnvironmentStringsW(ofn->lpstrInitialDir, fodInfos.initdir, len); GetCurrentDirectoryW(MAX_PATH, current_dir);
} }
}
/* save current directory */ switch (dlg_type)
if (ofn->Flags & OFN_NOCHANGEDIR) {
{ case OPEN_DIALOG:
lpstrSavDir = MemAlloc(MAX_PATH*sizeof(WCHAR)); ret = GetFileName95(info);
GetCurrentDirectoryW(MAX_PATH, lpstrSavDir); break;
} case SAVE_DIALOG:
info->DlgInfos.dwDlgProp |= FODPROP_SAVEDLG;
ret = GetFileName95(info);
break;
default:
ret = FALSE;
}
fodInfos.unicode = TRUE; if (current_dir)
{
SetCurrentDirectoryW(current_dir);
MemFree(current_dir);
}
switch(iDlgType) if (!info->unicode)
{ {
case OPEN_DIALOG : MemFree((WCHAR *)info->defext);
ret = GetFileName95(&fodInfos); MemFree((WCHAR *)info->title);
break; MemFree((WCHAR *)info->filter);
case SAVE_DIALOG : MemFree((WCHAR *)info->customfilter);
fodInfos.DlgInfos.dwDlgProp |= FODPROP_SAVEDLG; }
ret = GetFileName95(&fodInfos);
break;
default :
ret = FALSE;
}
if (lpstrSavDir) MemFree(info->filename);
{ MemFree(info->initdir);
SetCurrentDirectoryW(lpstrSavDir); return ret;
MemFree(lpstrSavDir);
}
/* restore saved IN arguments and convert OUT arguments back */
MemFree(fodInfos.filename);
MemFree(fodInfos.initdir);
return ret;
} }
/****************************************************************************** /******************************************************************************
@ -4080,8 +4006,7 @@ static inline BOOL is_win16_looks(DWORD flags)
* FALSE on cancel, error, close or filename-does-not-fit-in-buffer. * FALSE on cancel, error, close or filename-does-not-fit-in-buffer.
* *
*/ */
BOOL WINAPI GetOpenFileNameA( BOOL WINAPI GetOpenFileNameA(OPENFILENAMEA *ofn)
LPOPENFILENAMEA ofn) /* [in/out] address of init structure */
{ {
TRACE("flags %08x\n", ofn->Flags); TRACE("flags %08x\n", ofn->Flags);
@ -4098,7 +4023,12 @@ BOOL WINAPI GetOpenFileNameA(
if (is_win16_looks(ofn->Flags)) if (is_win16_looks(ofn->Flags))
return GetFileName31A(ofn, OPEN_DIALOG); return GetFileName31A(ofn, OPEN_DIALOG);
else else
return GetFileDialog95A(ofn, OPEN_DIALOG); {
FileOpenDlgInfos info;
init_filedlg_infoA(ofn, &info);
return GetFileDialog95(&info, OPEN_DIALOG);
}
} }
/*********************************************************************** /***********************************************************************
@ -4111,8 +4041,7 @@ BOOL WINAPI GetOpenFileNameA(
* FALSE on cancel, error, close or filename-does-not-fit-in-buffer. * FALSE on cancel, error, close or filename-does-not-fit-in-buffer.
* *
*/ */
BOOL WINAPI GetOpenFileNameW( BOOL WINAPI GetOpenFileNameW(OPENFILENAMEW *ofn)
LPOPENFILENAMEW ofn) /* [in/out] address of init structure */
{ {
TRACE("flags %08x\n", ofn->Flags); TRACE("flags %08x\n", ofn->Flags);
@ -4129,7 +4058,12 @@ BOOL WINAPI GetOpenFileNameW(
if (is_win16_looks(ofn->Flags)) if (is_win16_looks(ofn->Flags))
return GetFileName31W(ofn, OPEN_DIALOG); return GetFileName31W(ofn, OPEN_DIALOG);
else else
return GetFileDialog95W(ofn, OPEN_DIALOG); {
FileOpenDlgInfos info;
init_filedlg_infoW(ofn, &info);
return GetFileDialog95(&info, OPEN_DIALOG);
}
} }
@ -4143,8 +4077,7 @@ BOOL WINAPI GetOpenFileNameW(
* FALSE on cancel, error, close or filename-does-not-fit-in-buffer. * FALSE on cancel, error, close or filename-does-not-fit-in-buffer.
* *
*/ */
BOOL WINAPI GetSaveFileNameA( BOOL WINAPI GetSaveFileNameA(OPENFILENAMEA *ofn)
LPOPENFILENAMEA ofn) /* [in/out] address of init structure */
{ {
if (!valid_struct_size( ofn->lStructSize )) if (!valid_struct_size( ofn->lStructSize ))
{ {
@ -4155,7 +4088,12 @@ BOOL WINAPI GetSaveFileNameA(
if (is_win16_looks(ofn->Flags)) if (is_win16_looks(ofn->Flags))
return GetFileName31A(ofn, SAVE_DIALOG); return GetFileName31A(ofn, SAVE_DIALOG);
else else
return GetFileDialog95A(ofn, SAVE_DIALOG); {
FileOpenDlgInfos info;
init_filedlg_infoA(ofn, &info);
return GetFileDialog95(&info, SAVE_DIALOG);
}
} }
/*********************************************************************** /***********************************************************************
@ -4180,7 +4118,12 @@ BOOL WINAPI GetSaveFileNameW(
if (is_win16_looks(ofn->Flags)) if (is_win16_looks(ofn->Flags))
return GetFileName31W(ofn, SAVE_DIALOG); return GetFileName31W(ofn, SAVE_DIALOG);
else else
return GetFileDialog95W(ofn, SAVE_DIALOG); {
FileOpenDlgInfos info;
init_filedlg_infoW(ofn, &info);
return GetFileDialog95(&info, SAVE_DIALOG);
}
} }
/*********************************************************************** /***********************************************************************