ShellExecCmdLine.cpp File Reference

#include "shelltest.h"
#include <shlwapi.h>
#include <strsafe.h>
#include <versionhelpers.h>
#include "shell32_apitest_sub.h"
#include "closewnd.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

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

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 void	DoEntry (const TEST_ENTRY *pEntry)
static VOID	TEST_ShellExecCmdLine (VOID)
	START_TEST (ShellExecCmdLine)

Variables
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 WINDOW_LIST	s_List1
static WINDOW_LIST	s_List2

Macro Definition Documentation

IDS_FILE_NOT_FOUND

#define IDS_FILE_NOT_FOUND   (-1)

Definition at line 29 of file ShellExecCmdLine.cpp.

NDEBUG

#define NDEBUG

Definition at line 14 of file ShellExecCmdLine.cpp.

SECL_ALLOW_NONEXE

#define SECL_ALLOW_NONEXE   0x20

Definition at line 22 of file ShellExecCmdLine.cpp.

SECL_LOG_USAGE

#define SECL_LOG_USAGE   0x8

Definition at line 20 of file ShellExecCmdLine.cpp.

SECL_NO_UI

#define SECL_NO_UI   0x2

Definition at line 19 of file ShellExecCmdLine.cpp.

SECL_RUNAS

#define SECL_RUNAS   0x40

Definition at line 23 of file ShellExecCmdLine.cpp.

SECL_USE_IDLIST

#define SECL_USE_IDLIST   0x10

Definition at line 21 of file ShellExecCmdLine.cpp.

shell32_hInstance

HINSTANCE shell32_hInstance   GetModuleHandle(NULL)

Definition at line 28 of file ShellExecCmdLine.cpp.

Referenced by CFSDropTarget::_GetEffectFromMenu(), CFSDropTarget::_GetUniqueFileName(), _InsertMenuItemW(), _SetOperationTitle(), _SHGetDefaultValue(), AboutDlgProc(), ActionDlg_OnBrowse(), COpenWithMenu::AddChooseProgramItem(), CDefaultContextMenu::AddStaticContextMenusToMenu(), BrFolder_NewFolder(), COpenWithDialog::Browse(), CDefView::CDefView(), ConfirmDialog(), Control_CreateListView(), Control_DoInterface(), Control_WndProc(), Control_WndProc_Create(), CNewMenu::CreateNewFolder(), CShellLink::CreateShortcutIcon(), CreateToolTipForButtons(), do_error_dialog(), CDefaultContextMenu::DoCreateNewFolder(), DrivesContextMenuCallback(), EditTypeDlg_OnCommand(), CDrvDefExt::ExtraPageProc(), FancyLogoffOnInit(), FileTypesDlg_Initialize(), FileTypesDlg_InitListView(), CDefView::FillEditMenu(), CDefView::FillViewMenu(), FolderOptionsFileTypesDlg(), FormatDriveDlg(), FormatExCB(), GeneralDlg_UpdateIcons(), CControlPanelFolder::GetDetailsOf(), CDrivesFolder::GetDisplayNameOf(), GetDriveNameWithLetter(), GetExtensionDefaultDescription(), GetFileTypeIconsEx(), GetFriendlyVerb(), HCR_GetClassNameA(), HCR_GetClassNameW(), COpenWithDialog::Init(), CDrvDefExt::InitGeneralPage(), CFileDefExt::InitGeneralPage(), CDesktopBrowser::Initialize(), CNewMenu::Initialize(), InitializeDefaultIcons(), InitializeFormatDriveDlg(), InitializeRecycleBinDlg(), CFileDefExt::InitMultifilePage(), CFileDefExt::InitMultifilePageThread(), CFileDefExt::InitOpensWithField(), InsertDefaultClusterSizeForFs(), CNewMenu::InsertShellNewItems(), COpenWithDialog::IsNoOpen(), LogoffWindowsDialog(), CShellLink::OnCommand(), CDefView::OnCreate(), CShellLink::OnInitDialog(), PickIconDlg(), PickIconProc(), Printer_LoadIconsW(), PropSheetProc(), CDefViewBckgrndMenu::QueryContextMenu(), CNewMenu::QueryContextMenu(), COpenWithMenu::QueryContextMenu(), RunDlgProc(), RunFileDlg(), SH_CreatePropertySheetPageEx(), SH_FormatByteSize(), SHBrowseForFolderW(), SHCreateDirectoryExW(), SHELL32_AssocGetFSDirectoryDescription(), SHELL32_OpenPropSheet(), SHELL32_PropertySheet(), SHELL_ConfirmDialogW(), SHELL_ConfirmIDs(), SHELL_ConfirmMsgBox(), ShellAboutW(), ShellExecCmdLine(), SHEmptyRecycleBinW(), SHFormatDrive(), SHOpenWithDialog(), ShowFolderOptionsDialogThreadProc(), SHWinHelp(), SIC_Initialize(), CFileDefExt::UpdateDirStatsResults(), CFileDefExt::UpdateFolderIcon(), CDefView::UpdateStatusbar(), and CDefView::UpdateStatusbarLocation().

ShellExecCmdLine

#define ShellExecCmdLine   proxy_ShellExecCmdLine

Definition at line 26 of file ShellExecCmdLine.cpp.

Typedef Documentation

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().

DoEntry()

static void DoEntry ( const TEST_ENTRY *  pEntry )

static

Definition at line 552 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);
}

Referenced by TEST_ShellExecCmdLine().

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 618 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 (!FindSubProgram(s_sub_program, _countof(s_sub_program)))
    {
        skip("shell32_apitest_sub.exe is not found\n");
        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");
        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);
 
    TEST_ShellExecCmdLine();
 
    // Some process can lock the file of s_win_test_exe
    Sleep(1000);
 
    // 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");
}

TEST_ShellExecCmdLine()

static VOID TEST_ShellExecCmdLine ( VOID  )

static

Definition at line 593 of file ShellExecCmdLine.cpp.

{
    GetWindowList(&s_List1);
 
    // 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]);
    }
 
    // Execution can be asynchronous; you have to wait for it to finish.
    Sleep(2000);
 
    // Close newly-opened window(s)
    GetWindowList(&s_List2);
    CloseNewWindows(&s_List1, &s_List2);
    FreeWindowList(&s_List1);
    FreeWindowList(&s_List2);
}

Referenced by START_TEST().

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(), and TEST_ShellExecCmdLine().

s_entries_1

const TEST_ENTRY s_entries_1[]

static

Definition at line 259 of file ShellExecCmdLine.cpp.

Referenced by TEST_ShellExecCmdLine().

s_entries_2

const TEST_ENTRY s_entries_2[]

static

Definition at line 498 of file ShellExecCmdLine.cpp.

Referenced by TEST_ShellExecCmdLine().

s_List1

WINDOW_LIST s_List1

static

Definition at line 591 of file ShellExecCmdLine.cpp.

Referenced by TEST_ShellExecCmdLine().

s_List2

WINDOW_LIST s_List2

static

Definition at line 591 of file ShellExecCmdLine.cpp.

Referenced by TEST_ShellExecCmdLine().

s_sub_program

WCHAR s_sub_program[MAX_PATH]

static

Definition at line 254 of file ShellExecCmdLine.cpp.

Referenced by 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_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().