ShellExecCmdLine.cpp File Reference

#include "shelltest.h"
#include <shlwapi.h>
#include <strsafe.h>
#include <versionhelpers.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	RETRY_COUNT   5
#define	RETRY_INTERVAL   250

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

Functions
static void	DoEntry (const TEST_ENTRY *pEntry)
	START_TEST (ShellExecCmdLine)

Variables
SHELLEXECCMDLINE	g_pShellExecCmdLine = NULL
static const char	s_testfile1 [] = "Test File.txt"
static const char	s_testfile2 [] = "Test File.bat"
static char	s_notepad [] = "notepad.exe"
static const TEST_ENTRY	s_entries []

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 12 of file ShellExecCmdLine.cpp.

RETRY_COUNT

#define RETRY_COUNT   5

RETRY_INTERVAL

#define RETRY_INTERVAL   250

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.

Typedef Documentation

SHELLEXECCMDLINE

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

Definition at line 240 of file ShellExecCmdLine.cpp.

TEST_ENTRY

typedef struct TEST_ENTRY TEST_ENTRY

Function Documentation

DoEntry()

static void DoEntry ( const TEST_ENTRY *  pEntry )

static

Definition at line 401 of file ShellExecCmdLine.cpp.

{
    HRESULT hr;
    DWORD dwSeclFlags;

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

    _SEH2_TRY
    {
#ifdef ShellExecCmdLine
        hr = ShellExecCmdLine(NULL, pEntry->pwszCommand, pEntry->pwszStartDir,
                              SW_SHOWNORMAL, NULL, dwSeclFlags);
#else
        hr = (*g_pShellExecCmdLine)(NULL, pEntry->pwszCommand, pEntry->pwszStartDir,
                                    SW_SHOWNORMAL, NULL, dwSeclFlags);
#endif
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        hr = 0xDEADFACE;
    }
    _SEH2_END;

    ok(hr == pEntry->hr, "Line %d: hr expected 0x%lX, was 0x%lX\n", pEntry->lineno, pEntry->hr, hr);

#define RETRY_COUNT     5
#define RETRY_INTERVAL  250
    if (SUCCEEDED(hr) && pEntry->pwszWindowClass)
    {
        BOOL bFound = FALSE;
        Sleep(RETRY_INTERVAL / 2);
        for (int i = 0; i < RETRY_COUNT; ++i)
        {
            HWND hwnd = FindWindowW(pEntry->pwszWindowClass, NULL);
            if (hwnd)
            {
                bFound = TRUE;
                SendMessage(hwnd, WM_CLOSE, 0, 0);
                Sleep(RETRY_INTERVAL);
                break;
            }
            Sleep(RETRY_INTERVAL);
        }
        ok(bFound, "Line %d: The window not found\n", pEntry->lineno);
    }
#undef RETRY_COUNT
#undef RETRY_INTERVAL
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

ShellExecCmdLine

)

Definition at line 453 of file ShellExecCmdLine.cpp.

{
    using namespace std;

#ifndef ShellExecCmdLine
    // CHECKME
    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;
    }
#endif

    // s_testfile1
    FILE *fp = fopen(s_testfile1, "wb");
    ok(fp != NULL, "failed to create a test file\n");
    fclose(fp);

    // s_testfile2
    fp = fopen(s_testfile2, "wb");
    ok(fp != NULL, "failed to create a test file\n");
    if (fp)
    {
        fprintf(fp, "echo OK\n");
    }
    fclose(fp);

    for (size_t i = 0; i < _countof(s_entries); ++i)
    {
        DoEntry(&s_entries[i]);
    }

    WCHAR buf0[MAX_PATH];
    WCHAR buf1[MAX_PATH];
    WCHAR buf2[MAX_PATH];
    TEST_ENTRY additionals[] =
    {
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf0, NULL },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf0, L"." },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf0, L"system32" },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf1, NULL },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf1, L"." },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf1, L"system32" },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf2, NULL },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf2, L"." },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, L"Notepad", buf2, L"system32" },
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, L"Notepad", buf0, NULL }, // FIXME
        { __LINE__, S_OK, TRUE, L"Notepad", buf0, L"." },
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, L"Notepad", buf0, L"system32" }, // FIXME
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, L"Notepad", buf1, NULL }, // FIXME
        { __LINE__, S_OK, TRUE, L"Notepad", buf1, L"." },
        { __LINE__, S_OK, TRUE, L"Notepad", buf1, L"system32" },
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, L"Notepad", buf2, NULL }, // FIXME
        { __LINE__, S_OK, TRUE, L"Notepad", buf2, L"." },
        { __LINE__, S_OK, TRUE, L"Notepad", buf2, L"system32" },
    };

    wsprintfW(buf0, L"%hs", s_testfile1);
    wsprintfW(buf1, L"\"%hs\"", s_testfile1);
    wsprintfW(buf2, L"\"%hs\" \"Test File.txt\"", s_testfile1);
    for (size_t i = 0; i < _countof(additionals); ++i)
    {
        DoEntry(&additionals[i]);
    }

    TEST_ENTRY additionals2[] =
    {
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf0, NULL },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf0, L"." },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf0, L"system32" },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf1, NULL },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf1, L"." },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf1, L"system32" },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf2, NULL },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf2, L"." },
        { __LINE__, CO_E_APPNOTFOUND, FALSE, NULL, buf2, L"system32" },
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, NULL, buf0, NULL }, // FIXME
        { __LINE__, S_OK, TRUE, NULL, buf0, L"." },
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, NULL, buf0, L"system32" },  // FIXME
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, NULL, buf1, NULL }, // FIXME
        { __LINE__, S_OK, TRUE, NULL, buf1, L"." },
        { __LINE__, S_OK, TRUE, NULL, buf1, L"system32" },
        { __LINE__, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), TRUE, NULL, buf2, NULL }, // FIXME
        { __LINE__, S_OK, TRUE, NULL, buf2, L"." },
        { __LINE__, S_OK, TRUE, NULL, buf2, L"system32" },
    };

    wsprintfW(buf0, L"%hs", s_testfile2);
    wsprintfW(buf1, L"\"%hs\"", s_testfile2);
    wsprintfW(buf2, L"\"%hs\" \"Test File.txt\"", s_testfile2);
    for (size_t i = 0; i < _countof(additionals2); ++i)
    {
        DoEntry(&additionals2[i]);
    }

    char path[MAX_PATH];
    ok((INT_PTR)FindExecutableA("notepad.exe", NULL, s_notepad) >= 32, "FindExecutableA failed\n");
    ok(GetModuleFileNameA(NULL, path, _countof(path)), "GetModuleFileNameA failed\n");
    char *pch = strrchr(path, '\\');

    if (pch == NULL)
    {
        skip("pch == NULL\n");
    }
    else
    {
        // create "My Directory"
        strcpy(pch, "\\My Directory");
        if (GetFileAttributesA(path) == INVALID_FILE_ATTRIBUTES)
            ok(CreateDirectoryA(path, NULL), "CreateDirectoryA failed\n");

        // create "My Directory\\Notepad.exe" as clone of Notepad.exe
        strcpy(pch, "\\My Directory\\Notepad.exe");
        ok(CopyFileA(s_notepad, path, FALSE), "CopyFileA failed\n");

        wsprintfW(buf0, L"%hs", path);
        wsprintfW(buf1, L"\"%hs\"", path);
        wsprintfW(buf2, L"\"%hs\" \"Test File.txt\"", path);
        TEST_ENTRY additionals3[] =
        {
            { __LINE__, S_OK, FALSE, NULL, buf0, NULL },
            { __LINE__, S_OK, FALSE, NULL, buf0, L"." },
            { __LINE__, S_OK, FALSE, NULL, buf0, L"system32" },
            { __LINE__, S_OK, FALSE, NULL, buf1, NULL },
            { __LINE__, S_OK, FALSE, NULL, buf1, L"." },
            { __LINE__, S_OK, FALSE, NULL, buf1, L"system32" },
            { __LINE__, S_OK, FALSE, NULL, buf2, NULL },
            { __LINE__, S_OK, FALSE, NULL, buf2, L"." },
            { __LINE__, S_OK, FALSE, NULL, buf2, L"system32" },
            { __LINE__, S_OK, TRUE, NULL, buf0, NULL },
            { __LINE__, S_OK, TRUE, NULL, buf0, L"." },
            { __LINE__, S_OK, TRUE, NULL, buf0, L"system32" },
            { __LINE__, S_OK, TRUE, NULL, buf1, NULL },
            { __LINE__, S_OK, TRUE, NULL, buf1, L"." },
            { __LINE__, S_OK, TRUE, NULL, buf1, L"system32" },
            { __LINE__, S_OK, TRUE, NULL, buf2, NULL },
            { __LINE__, S_OK, TRUE, NULL, buf2, L"." },
            { __LINE__, S_OK, TRUE, NULL, buf2, L"system32" },
        };
        for (size_t i = 0; i < _countof(additionals3); ++i)
        {
            DoEntry(&additionals3[i]);
        }

        DeleteFileA(path);

        strcpy(pch, "\\My Directory");
        RemoveDirectory(path);
    }

    // clean up
    ok(DeleteFileA(s_testfile1), "failed to delete the test file\n");
    ok(DeleteFileA(s_testfile2), "failed to delete the test file\n");
}

Variable Documentation

g_pShellExecCmdLine

SHELLEXECCMDLINE g_pShellExecCmdLine = NULL

Definition at line 241 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_entries

const TEST_ENTRY s_entries[]

static

Definition at line 258 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_notepad

char s_notepad[] = "notepad.exe"

static

Definition at line 256 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_testfile1

const char s_testfile1[] = "Test File.txt"

static

Definition at line 254 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().

s_testfile2

const char s_testfile2[] = "Test File.bat"

static

Definition at line 255 of file ShellExecCmdLine.cpp.

Referenced by START_TEST().