ShellExecCmdLine.cpp File Reference

#include "shelltest.h"
#include <shlwapi.h>
#include <strsafe.h>
#include <versionhelpers.h>
#include "shell32_apitest_sub.h"
#include <debug.h>
#include <stdio.h>

Include dependency graph for ShellExecCmdLine.cpp:

Go to the source code of this file.

Classes
struct	TEST_ENTRY
struct	OPENWNDS

Macros
#define	NDEBUG
#define	SECL_NO_UI   0x2
#define	SECL_LOG_USAGE   0x8
#define	SECL_USE_IDLIST   0x10
#define	SECL_ALLOW_NONEXE   0x20
#define	SECL_RUNAS   0x40
#define	ShellExecCmdLine   proxy_ShellExecCmdLine
#define	shell32_hInstance   GetModuleHandle(NULL)
#define	IDS_FILE_NOT_FOUND   (-1)

Typedefs
typedef HRESULT(WINAPI *	SHELLEXECCMDLINE) (HWND, LPCWSTR, LPCWSTR, INT, LPVOID, DWORD)
typedef struct TEST_ENTRY	TEST_ENTRY
typedef struct OPENWNDS	OPENWNDS

Functions
static __inline void	__SHCloneStrW (WCHAR **target, const WCHAR *source)
static LPCWSTR	SplitParams (LPCWSTR psz, LPWSTR pszArg0, size_t cchArg0)
HRESULT WINAPI	ShellExecCmdLine (HWND hwnd, LPCWSTR pwszCommand, LPCWSTR pwszStartDir, int nShow, LPVOID pUnused, DWORD dwSeclFlags)
static BOOL	GetSubProgramPath (void)
static BOOL CALLBACK	EnumWindowsProc (HWND hwnd, LPARAM lParam)
static void	CleanupNewlyCreatedWindows (void)
static void	DoEntry (const TEST_ENTRY *pEntry)
	START_TEST (ShellExecCmdLine)

Variables
static const WCHAR	wszOpen [] = L"open"
static const WCHAR	wszExe [] = L".exe"
static const WCHAR	wszCom [] = L".com"
SHELLEXECCMDLINE	g_pShellExecCmdLine = NULL
static WCHAR	s_sub_program [MAX_PATH]
static WCHAR	s_win_test_exe [MAX_PATH]
static WCHAR	s_sys_bat_file [MAX_PATH]
static WCHAR	s_cur_dir [MAX_PATH]
static const TEST_ENTRY	s_entries_1 []
static const TEST_ENTRY	s_entries_2 []
static OPENWNDS	s_wi0 = { 0 }
static OPENWNDS	s_wi1 = { 0 }

Macro Definition Documentation

IDS_FILE_NOT_FOUND

#define IDS_FILE_NOT_FOUND   (-1)

Definition at line 28 of file ShellExecCmdLine.cpp.

NDEBUG

#define NDEBUG

Definition at line 13 of file ShellExecCmdLine.cpp.

SECL_ALLOW_NONEXE

#define SECL_ALLOW_NONEXE   0x20

Definition at line 21 of file ShellExecCmdLine.cpp.

SECL_LOG_USAGE

#define SECL_LOG_USAGE   0x8

Definition at line 19 of file ShellExecCmdLine.cpp.

SECL_NO_UI

#define SECL_NO_UI   0x2

Definition at line 18 of file ShellExecCmdLine.cpp.

SECL_RUNAS

#define SECL_RUNAS   0x40

Definition at line 22 of file ShellExecCmdLine.cpp.

SECL_USE_IDLIST

#define SECL_USE_IDLIST   0x10

Definition at line 20 of file ShellExecCmdLine.cpp.

shell32_hInstance

HINSTANCE shell32_hInstance   GetModuleHandle(NULL)

Definition at line 27 of file ShellExecCmdLine.cpp.

Referenced by CFSDropTarget::_GetEffectFromMenu(), CFSDropTarget::_GetUniqueFileName(), _ILGetFileType(), _InsertMenuItemW(), _SetOperationTitle(), _SHGetDefaultValue(), AboutDlgProc(), ActionDlg_OnBrowse(), COpenWithMenu::AddChooseProgramItem(), CDefaultContextMenu::AddStaticContextMenusToMenu(), COpenWithDialog::Browse(), BrsFolder_NewFolder(), CDefView::CDefView(), ConfirmDialog(), Control_CreateListView(), Control_DoInterface(), Control_WndProc(), Control_WndProc_Create(), CFileDefExt::CountFolderAndFiles(), CNewMenu::CreateNewFolder(), CShellLink::CreateShortcutIcon(), CreateToolTipForButtons(), do_error_dialog(), CDefaultContextMenu::DoCreateNewFolder(), DrivesContextMenuCallback(), EditTypeDlg_OnCommand(), CDrvDefExt::ExtraPageProc(), FancyLogoffOnInit(), FileTypesDlg_DoList(), FileTypesDlg_InitListView(), CDefView::FillEditMenu(), CDefView::FillViewMenu(), FolderOptionsFileTypesDlg(), FormatDriveDlg(), FormatExCB(), GeneralDlg_UpdateIcons(), CControlPanelFolder::GetDetailsOf(), CRecycleBin::GetDetailsOf(), CDrivesFolder::GetDisplayNameOf(), GetDriveNameWithLetter(), GetFileTypeIconsEx(), HCR_GetClassNameA(), HCR_GetClassNameW(), COpenWithDialog::Init(), CFileDefExt::InitFileAttr(), CDrvDefExt::InitGeneralPage(), CFileDefExt::InitGeneralPage(), CDesktopBrowser::Initialize(), CNewMenu::Initialize(), InitializeFormatDriveDlg(), InitializeRecycleBinDlg(), CFileDefExt::InitOpensWithField(), InsertDefaultClusterSizeForFs(), CNewMenu::InsertShellNewItems(), COpenWithDialog::IsNoOpen(), LogoffWindowsDialog(), CNewMenu::NewItemByNonCommand(), CShellLink::OnCommand(), CDefView::OnCreate(), PickIconDlg(), PickIconProc(), Printer_LoadIconsW(), PropSheetProc(), CDefViewBckgrndMenu::QueryContextMenu(), COpenWithMenu::QueryContextMenu(), CRecycleBin::QueryContextMenu(), CNewMenu::QueryContextMenu(), CRecycleBinItemContextMenu::QueryContextMenu(), RunDlgProc(), RunFileDlg(), SetFileTypeEntryDefaultIcon(), SH_CreatePropertySheetPage(), SH_FormatByteSize(), SH_ShowRecycleBinProperties(), SHBrowseForFolderW(), SHCreateDirectoryExW(), SHELL_ConfirmDialogW(), SHELL_ConfirmIDs(), SHELL_ConfirmMsgBox(), ShellAboutW(), ShellExecCmdLine(), SHEmptyRecycleBinW(), SHFormatDrive(), SHOpenWithDialog(), ShowFolderOptionsDialog(), SHWinHelp(), SIC_Initialize(), CFileDefExt::UpdateFolderIcon(), and CDefView::UpdateStatusbar().

ShellExecCmdLine

#define ShellExecCmdLine   proxy_ShellExecCmdLine

Definition at line 25 of file ShellExecCmdLine.cpp.

Typedef Documentation

OPENWNDS

typedef struct OPENWNDS OPENWNDS

SHELLEXECCMDLINE

typedef HRESULT(WINAPI * SHELLEXECCMDLINE) (HWND, LPCWSTR, LPCWSTR, INT, LPVOID, DWORD)

Definition at line 242 of file ShellExecCmdLine.cpp.

TEST_ENTRY

typedef struct TEST_ENTRY TEST_ENTRY

Function Documentation

__SHCloneStrW()

static __inline void __SHCloneStrW ( WCHAR **  target, const WCHAR *  source  )

static

Definition at line 34 of file ShellExecCmdLine.cpp.

{
    *target = (WCHAR *)SHAlloc((lstrlenW(source) + 1) * sizeof(WCHAR) );
    lstrcpyW(*target, source);
}

Referenced by ShellExecCmdLine().

CleanupNewlyCreatedWindows()

static void CleanupNewlyCreatedWindows ( void  )

static

Definition at line 597 of file ShellExecCmdLine.cpp.

{
    EnumWindows(EnumWindowsProc, (LPARAM)&s_wi1);
    for (UINT i1 = 0; i1 < s_wi1.count; ++i1)
    {
        BOOL bFound = FALSE;
        for (UINT i0 = 0; i0 < s_wi0.count; ++i0)
        {
            if (s_wi1.phwnd[i1] == s_wi0.phwnd[i0])
            {
                bFound = TRUE;
                break;
            }
        }
        if (!bFound)
            PostMessageW(s_wi1.phwnd[i1], WM_CLOSE, 0, 0);
    }
    free(s_wi1.phwnd);
    ZeroMemory(&s_wi1, sizeof(s_wi1));
}

Referenced by DoEntry(), and START_TEST().

DoEntry()

static void DoEntry ( const TEST_ENTRY *  pEntry )

static

Definition at line 618 of file ShellExecCmdLine.cpp.

{
    HRESULT hr;
    DWORD dwSeclFlags;
    BOOL result;

    if (pEntry->bAllowNonExe)
        dwSeclFlags = SECL_NO_UI | SECL_ALLOW_NONEXE;
    else
        dwSeclFlags = SECL_NO_UI;

    _SEH2_TRY
    {
        if (IsReactOS())
        {
            hr = proxy_ShellExecCmdLine(NULL, pEntry->pwszCommand, pEntry->pwszStartDir,
                                        SW_SHOWNORMAL, NULL, dwSeclFlags);
        }
        else
        {
            hr = (*g_pShellExecCmdLine)(NULL, pEntry->pwszCommand, pEntry->pwszStartDir,
                                        SW_SHOWNORMAL, NULL, dwSeclFlags);
        }
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        hr = 0xBADFACE;
    }
    _SEH2_END;

    if (hr == 0xBADFACE)
        result = hr;
    else
        result = (hr == S_OK);

    ok(result == pEntry->result, "Line %d: result expected %d, was %d\n",
       pEntry->lineno, pEntry->result, result);

    CleanupNewlyCreatedWindows();
}

Referenced by START_TEST().

EnumWindowsProc()

static BOOL CALLBACK EnumWindowsProc ( HWND  hwnd, LPARAM  lParam  )

static

Definition at line 586 of file ShellExecCmdLine.cpp.

{
    OPENWNDS *info = (OPENWNDS *)lParam;
    info->phwnd = (HWND *)realloc(info->phwnd, (info->count + 1) * sizeof(HWND));
    if (!info->phwnd)
        return FALSE;
    info->phwnd[info->count] = hwnd;
    ++(info->count);
    return TRUE;
}

Referenced by CleanupNewlyCreatedWindows(), and START_TEST().

GetSubProgramPath()

static BOOL GetSubProgramPath ( void  )

static

Definition at line 260 of file ShellExecCmdLine.cpp.

{
    GetModuleFileNameW(NULL, s_sub_program, _countof(s_sub_program));
    PathRemoveFileSpecW(s_sub_program);
    PathAppendW(s_sub_program, L"shell32_apitest_sub.exe");

    if (!PathFileExistsW(s_sub_program))
    {
        PathRemoveFileSpecW(s_sub_program);
        PathAppendW(s_sub_program, L"testdata\\shell32_apitest_sub.exe");

        if (!PathFileExistsW(s_sub_program))
        {
            return FALSE;
        }
    }

    return TRUE;
}

Referenced by START_TEST().

ShellExecCmdLine()

HRESULT WINAPI ShellExecCmdLine	(	HWND	hwnd,
		LPCWSTR	pwszCommand,
		LPCWSTR	pwszStartDir,
		int	nShow,
		LPVOID	pUnused,
		DWORD	dwSeclFlags
	)

Definition at line 88 of file ShellExecCmdLine.cpp.

{
    SHELLEXECUTEINFOW info;
    DWORD dwSize, dwError, dwType, dwFlags = SEE_MASK_DOENVSUBST | SEE_MASK_NOASYNC;
    LPCWSTR pszVerb = NULL;
    WCHAR szFile[MAX_PATH], szFile2[MAX_PATH];
    HRESULT hr;
    LPCWSTR pchParams;
    LPWSTR lpCommand = NULL;

    if (pwszCommand == NULL)
        RaiseException(EXCEPTION_ACCESS_VIOLATION, EXCEPTION_NONCONTINUABLE,
                       1, (ULONG_PTR*)pwszCommand);

    __SHCloneStrW(&lpCommand, pwszCommand);
    StrTrimW(lpCommand, L" \t");

    if (dwSeclFlags & SECL_NO_UI)
        dwFlags |= SEE_MASK_FLAG_NO_UI;
    if (dwSeclFlags & SECL_LOG_USAGE)
        dwFlags |= SEE_MASK_FLAG_LOG_USAGE;
    if (dwSeclFlags & SECL_USE_IDLIST)
        dwFlags |= SEE_MASK_INVOKEIDLIST;

    if (dwSeclFlags & SECL_RUNAS)
    {
        dwSize = 0;
        hr = AssocQueryStringW(0, ASSOCSTR_COMMAND, lpCommand, L"RunAs", NULL, &dwSize);
        if (SUCCEEDED(hr) && dwSize != 0)
        {
            pszVerb = L"runas";
        }
    }

    if (UrlIsFileUrlW(lpCommand))
    {
        StringCchCopyW(szFile, _countof(szFile), lpCommand);
        pchParams = NULL;
    }
    else
    {
        pchParams = SplitParams(lpCommand, szFile, _countof(szFile));
        if (szFile[0] != UNICODE_NULL && szFile[1] == L':' &&
            szFile[2] == UNICODE_NULL)
        {
            PathAddBackslashW(szFile);
        }

        WCHAR szCurDir[MAX_PATH];
        GetCurrentDirectoryW(_countof(szCurDir), szCurDir);
        if (pwszStartDir)
        {
            SetCurrentDirectoryW(pwszStartDir);
        }

        if (PathIsRelativeW(szFile) &&
            GetFullPathNameW(szFile, _countof(szFile2), szFile2, NULL) &&
            PathFileExistsW(szFile2))
        {
            StringCchCopyW(szFile, _countof(szFile), szFile2);
        }
        else if (SearchPathW(NULL, szFile, NULL, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(NULL, szFile, wszExe, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(NULL, szFile, wszCom, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(pwszStartDir, szFile, NULL, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(pwszStartDir, szFile, wszExe, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(pwszStartDir, szFile, wszCom, _countof(szFile2), szFile2, NULL))
        {
            StringCchCopyW(szFile, _countof(szFile), szFile2);
        }
        else if (SearchPathW(NULL, lpCommand, NULL, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(NULL, lpCommand, wszExe, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(NULL, lpCommand, wszCom, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(pwszStartDir, lpCommand, NULL, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(pwszStartDir, lpCommand, wszExe, _countof(szFile2), szFile2, NULL) ||
                 SearchPathW(pwszStartDir, lpCommand, wszCom, _countof(szFile2), szFile2, NULL))
        {
            StringCchCopyW(szFile, _countof(szFile), szFile2);
            pchParams = NULL;
        }

        if (pwszStartDir)
        {
            SetCurrentDirectoryW(szCurDir);
        }

        if (!(dwSeclFlags & SECL_ALLOW_NONEXE))
        {
            if (!GetBinaryTypeW(szFile, &dwType))
            {
                SHFree(lpCommand);

                if (!(dwSeclFlags & SECL_NO_UI))
                {
                    WCHAR szText[128 + MAX_PATH], szFormat[128];
                    LoadStringW(shell32_hInstance, IDS_FILE_NOT_FOUND, szFormat, _countof(szFormat));
                    StringCchPrintfW(szText, _countof(szText), szFormat, szFile);
                    MessageBoxW(hwnd, szText, NULL, MB_ICONERROR);
                }
                return CO_E_APPNOTFOUND;
            }
        }
        else
        {
            if (GetFileAttributesW(szFile) == INVALID_FILE_ATTRIBUTES)
            {
                SHFree(lpCommand);

                if (!(dwSeclFlags & SECL_NO_UI))
                {
                    WCHAR szText[128 + MAX_PATH], szFormat[128];
                    LoadStringW(shell32_hInstance, IDS_FILE_NOT_FOUND, szFormat, _countof(szFormat));
                    StringCchPrintfW(szText, _countof(szText), szFormat, szFile);
                    MessageBoxW(hwnd, szText, NULL, MB_ICONERROR);
                }
                return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
            }
        }
    }

    ZeroMemory(&info, sizeof(info));
    info.cbSize = sizeof(info);
    info.fMask = dwFlags;
    info.hwnd = hwnd;
    info.lpVerb = pszVerb;
    info.lpFile = szFile;
    info.lpParameters = (pchParams && *pchParams) ? pchParams : NULL;
    info.lpDirectory = pwszStartDir;
    info.nShow = nShow;
    if (ShellExecuteExW(&info))
    {
        if (info.lpIDList)
            CoTaskMemFree(info.lpIDList);

        SHFree(lpCommand);

        return S_OK;
    }

    dwError = GetLastError();

    SHFree(lpCommand);

    return HRESULT_FROM_WIN32(dwError);
}

SplitParams()

static LPCWSTR SplitParams ( LPCWSTR  psz, LPWSTR  pszArg0, size_t  cchArg0  )

static

Definition at line 42 of file ShellExecCmdLine.cpp.

{
    LPCWSTR pch;
    size_t ich = 0;
    if (*psz == L'"')
    {
        // 1st argument is quoted. the string in quotes is quoted 1st argument.
        // [pch] --> [pszArg0+ich]
        for (pch = psz + 1; *pch && ich + 1 < cchArg0; ++ich, ++pch)
        {
            if (*pch == L'"' && pch[1] == L'"')
            {
                // doubled double quotations found!
                pszArg0[ich] = L'"';
            }
            else if (*pch == L'"')
            {
                // single double quotation found!
                ++pch;
                break;
            }
            else
            {
                // otherwise
                pszArg0[ich] = *pch;
            }
        }
    }
    else
    {
        // 1st argument is unquoted. non-space sequence is 1st argument.
        // [pch] --> [pszArg0+ich]
        for (pch = psz; *pch && !iswspace(*pch) && ich + 1 < cchArg0; ++ich, ++pch)
        {
            pszArg0[ich] = *pch;
        }
    }
    pszArg0[ich] = 0;

    // skip space
    while (iswspace(*pch))
        ++pch;

    return pch;
}

Referenced by ShellExecCmdLine().

START_TEST()

START_TEST

(

ShellExecCmdLine

)

Definition at line 659 of file ShellExecCmdLine.cpp.

{
    using namespace std;

    if (!IsReactOS())
    {
        if (!IsWindowsVistaOrGreater())
        {
            skip("ShellExecCmdLine is not available on this platform\n");
            return;
        }

        HMODULE hShell32 = GetModuleHandleA("shell32");
        g_pShellExecCmdLine = (SHELLEXECCMDLINE)GetProcAddress(hShell32, (LPCSTR)(INT_PTR)265);
        if (!g_pShellExecCmdLine)
        {
            skip("ShellExecCmdLine is not found\n");
            return;
        }
    }

    if (!GetSubProgramPath())
    {
        skip("shell32_apitest_sub.exe is not found\n");
        return;
    }

    // record open windows
    if (!EnumWindows(EnumWindowsProc, (LPARAM)&s_wi0))
    {
        skip("EnumWindows failed\n");
        free(s_wi0.phwnd);
        return;
    }

    // s_win_test_exe
    GetWindowsDirectoryW(s_win_test_exe, _countof(s_win_test_exe));
    PathAppendW(s_win_test_exe, L"test program.exe");
    BOOL ret = CopyFileW(s_sub_program, s_win_test_exe, FALSE);
    if (!ret)
    {
        skip("Please retry with admin rights\n");
        free(s_wi0.phwnd);
        return;
    }

    FILE *fp;

    // s_sys_bat_file
    GetSystemDirectoryW(s_sys_bat_file, _countof(s_sys_bat_file));
    PathAppendW(s_sys_bat_file, L"test program.bat");
    fp = _wfopen(s_sys_bat_file, L"wb");
    fclose(fp);
    ok_int(PathFileExistsW(s_sys_bat_file), TRUE);

    // "Test File 1.txt"
    fp = fopen("Test File 1.txt", "wb");
    ok(fp != NULL, "failed to create a test file\n");
    fclose(fp);
    ok_int(PathFileExistsA("Test File 1.txt"), TRUE);

    // "Test File 2.bat"
    fp = fopen("Test File 2.bat", "wb");
    ok(fp != NULL, "failed to create a test file\n");
    fclose(fp);
    ok_int(PathFileExistsA("Test File 2.bat"), TRUE);

    // s_cur_dir
    GetCurrentDirectoryW(_countof(s_cur_dir), s_cur_dir);

    // do tests
    for (size_t i = 0; i < _countof(s_entries_1); ++i)
    {
        DoEntry(&s_entries_1[i]);
    }
    SetEnvironmentVariableW(L"PATH", s_cur_dir);
    for (size_t i = 0; i < _countof(s_entries_2); ++i)
    {
        DoEntry(&s_entries_2[i]);
    }

    Sleep(2000);
    CleanupNewlyCreatedWindows();

    // clean up
    ok(DeleteFileW(s_win_test_exe), "failed to delete the test file\n");
    ok(DeleteFileW(s_sys_bat_file), "failed to delete the test file\n");
    ok(DeleteFileA("Test File 1.txt"), "failed to delete the test file\n");
    ok(DeleteFileA("Test File 2.bat"), "failed to delete the test file\n");
    free(s_wi0.phwnd);

    DoWaitForWindow(CLASSNAME, CLASSNAME, TRUE, TRUE);
    Sleep(100);
}

Variable Documentation

g_pShellExecCmdLine

SHELLEXECCMDLINE g_pShellExecCmdLine = NULL

Definition at line 243 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_cur_dir

WCHAR s_cur_dir[MAX_PATH]

static

Definition at line 257 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_entries_1

const TEST_ENTRY s_entries_1[]

static

Definition at line 280 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_entries_2

const TEST_ENTRY s_entries_2[]

static

Definition at line 524 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_sub_program

WCHAR s_sub_program[MAX_PATH]

static

Definition at line 254 of file ShellExecCmdLine.cpp.

Referenced by GetSubProgramPath(), and START_TEST().

s_sys_bat_file

WCHAR s_sys_bat_file[MAX_PATH]

static

Definition at line 256 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_wi0

OPENWNDS s_wi0 = { 0 }

static

Definition at line 584 of file ShellExecCmdLine.cpp.

Referenced by CleanupNewlyCreatedWindows(), and START_TEST().

s_wi1

OPENWNDS s_wi1 = { 0 }

static

Definition at line 584 of file ShellExecCmdLine.cpp.

Referenced by CleanupNewlyCreatedWindows().

s_win_test_exe

WCHAR s_win_test_exe[MAX_PATH]

static

Definition at line 255 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

wszCom

const WCHAR wszCom[] = L".com"

static

Definition at line 32 of file ShellExecCmdLine.cpp.

Referenced by ShellExecCmdLine().

wszExe

const WCHAR wszExe[] = L".exe"

static

Definition at line 31 of file ShellExecCmdLine.cpp.

Referenced by ShellExecCmdLine().

wszOpen

const WCHAR wszOpen[] = L"open"

static

Definition at line 30 of file ShellExecCmdLine.cpp.