ShellExecuteEx.cpp File Reference

#include "shelltest.h"
#include <shlwapi.h>
#include <stdio.h>

Include dependency graph for ShellExecuteEx.cpp:

Go to the source code of this file.

Classes
struct	TEST_ENTRY
struct	OPENWNDS

Macros
#define	ok_ShellExecuteEx   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : TestShellExecuteEx
#define	DONT_CARE   0x0BADF00D

Typedefs
typedef struct TEST_ENTRY	TEST_ENTRY
typedef struct OPENWNDS	OPENWNDS

Functions
static BOOL	CreateAppPathRegKey (const WCHAR *Name)
static VOID	DeleteAppPathRegKey (const WCHAR *Name)
static VOID	TestShellExecuteEx (const WCHAR *Name, BOOL ExpectedResult)
static void	DoAppPathTest (void)
static BOOL CALLBACK	EnumWindowsProc (HWND hwnd, LPARAM lParam)
static VOID	DoTestEntry (const TEST_ENTRY *pEntry)
static BOOL	GetSubProgramPath (void)
static void	DoTestEntries (void)
	START_TEST (ShellExecuteEx)

Variables
static char	s_sub_program [MAX_PATH]
static char	s_win_test_exe [MAX_PATH]
static char	s_sys_test_exe [MAX_PATH]
static char	s_win_bat_file [MAX_PATH]
static char	s_sys_bat_file [MAX_PATH]
static char	s_win_txt_file [MAX_PATH]
static char	s_sys_txt_file [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

DONT_CARE

#define DONT_CARE   0x0BADF00D

Definition at line 125 of file ShellExecuteEx.cpp.

ok_ShellExecuteEx

#define ok_ShellExecuteEx   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : TestShellExecuteEx

Definition at line 13 of file ShellExecuteEx.cpp.

Typedef Documentation

OPENWNDS

typedef struct OPENWNDS OPENWNDS

TEST_ENTRY

typedef struct TEST_ENTRY TEST_ENTRY

Function Documentation

CreateAppPathRegKey()

static BOOL CreateAppPathRegKey ( const WCHAR *  Name )

static

Definition at line 17 of file ShellExecuteEx.cpp.

{
    HKEY RegistryKey;
    LONG Result;
    WCHAR Buffer[1024];
    WCHAR KeyValue[1024];
    DWORD Length = sizeof(KeyValue);
    DWORD Disposition;

    wcscpy(Buffer, L"Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\");
    wcscat(Buffer, L"IEXPLORE.EXE");
    Result = RegOpenKeyExW(HKEY_LOCAL_MACHINE, Buffer, 0, KEY_READ, &RegistryKey);
    if (Result != ERROR_SUCCESS) trace("Could not open iexplore.exe key. Status: %lu\n", Result);
    if (Result) goto end;
    Result = RegQueryValueExW(RegistryKey, NULL, NULL, NULL, (LPBYTE)KeyValue, &Length);
    if (Result != ERROR_SUCCESS) trace("Could not read iexplore.exe key. Status: %lu\n", Result);
    if (Result) goto end;
    RegCloseKey(RegistryKey);

    wcscpy(Buffer, L"Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\");
    wcscat(Buffer, Name);
    Result = RegCreateKeyExW(HKEY_LOCAL_MACHINE, Buffer, 0, NULL,
        0, KEY_WRITE, NULL, &RegistryKey, &Disposition);
    if (Result != ERROR_SUCCESS) trace("Could not create test key. Status: %lu\n", Result);
    if (Result) goto end;
    Result = RegSetValueW(RegistryKey, NULL, REG_SZ, KeyValue, 0);
    if (Result != ERROR_SUCCESS) trace("Could not set value of the test key. Status: %lu\n", Result);
    if (Result) goto end;
    RegCloseKey(RegistryKey);
end:
    if (RegistryKey) RegCloseKey(RegistryKey);
    return Result == ERROR_SUCCESS;
}

Referenced by DoAppPathTest().

DeleteAppPathRegKey()

static VOID DeleteAppPathRegKey ( const WCHAR *  Name )

static

Definition at line 53 of file ShellExecuteEx.cpp.

{
    LONG Result;
    WCHAR Buffer[1024];
    wcscpy(Buffer, L"Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\");
    wcscat(Buffer, Name);
    Result = RegDeleteKeyW(HKEY_LOCAL_MACHINE, Buffer);
    if (Result != ERROR_SUCCESS) trace("Could not remove the test key. Status: %lu\n", Result);
}

Referenced by DoAppPathTest().

DoAppPathTest()

static void DoAppPathTest ( void  )

static

Definition at line 87 of file ShellExecuteEx.cpp.

{
    ok_ShellExecuteEx(L"iexplore", TRUE);
    ok_ShellExecuteEx(L"iexplore.exe", TRUE);

    if (CreateAppPathRegKey(L"iexplore.bat"))
    {
        ok_ShellExecuteEx(L"iexplore.bat", TRUE);
        ok_ShellExecuteEx(L"iexplore.bat.exe", FALSE);
        DeleteAppPathRegKey(L"iexplore.bat");
    }

    if (CreateAppPathRegKey(L"iexplore.bat.exe"))
    {
        ok_ShellExecuteEx(L"iexplore.bat", FALSE);
        ok_ShellExecuteEx(L"iexplore.bat.exe", TRUE);
        DeleteAppPathRegKey(L"iexplore.bat.exe");
    }
}

Referenced by START_TEST().

DoTestEntries()

static void DoTestEntries ( void  )

static

Definition at line 284 of file ShellExecuteEx.cpp.

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

    // s_win_test_exe
    GetWindowsDirectoryA(s_win_test_exe, _countof(s_win_test_exe));
    PathAppendA(s_win_test_exe, "test program.exe");
    BOOL ret = CopyFileA(s_sub_program, s_win_test_exe, FALSE);
    if (!ret)
    {
        skip("Please retry with admin rights\n");
        return;
    }

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

    // s_sys_test_exe
    GetSystemDirectoryA(s_sys_test_exe, _countof(s_sys_test_exe));
    PathAppendA(s_sys_test_exe, "test program.exe");
    ok_int(CopyFileA(s_sub_program, s_sys_test_exe, FALSE), TRUE);

    // s_win_bat_file
    GetWindowsDirectoryA(s_win_bat_file, _countof(s_win_bat_file));
    PathAppendA(s_win_bat_file, "test program.bat");
    FILE *fp = fopen(s_win_bat_file, "wb");
    fprintf(fp, "exit /b 3");
    fclose(fp);
    ok_int(PathFileExistsA(s_win_bat_file), TRUE);

    // s_sys_bat_file
    GetSystemDirectoryA(s_sys_bat_file, _countof(s_sys_bat_file));
    PathAppendA(s_sys_bat_file, "test program.bat");
    fp = fopen(s_sys_bat_file, "wb");
    fprintf(fp, "exit /b 4");
    fclose(fp);
    ok_int(PathFileExistsA(s_sys_bat_file), TRUE);

    // s_win_txt_file
    GetWindowsDirectoryA(s_win_txt_file, _countof(s_win_txt_file));
    PathAppendA(s_win_txt_file, "test_file.txt");
    fp = fopen(s_win_txt_file, "wb");
    fclose(fp);
    ok_int(PathFileExistsA(s_win_txt_file), TRUE);

    // s_sys_txt_file
    GetSystemDirectoryA(s_sys_txt_file, _countof(s_sys_txt_file));
    PathAppendA(s_sys_txt_file, "test_file.txt");
    fp = fopen(s_sys_txt_file, "wb");
    fclose(fp);
    ok_int(PathFileExistsA(s_sys_txt_file), TRUE);

    for (UINT iTest = 0; iTest < _countof(s_entries_1); ++iTest)
    {
        DoTestEntry(&s_entries_1[iTest]);
    }

    DeleteFileA(s_win_bat_file);
    DeleteFileA(s_sys_bat_file);

    for (UINT iTest = 0; iTest < _countof(s_entries_2); ++iTest)
    {
        DoTestEntry(&s_entries_2[iTest]);
    }

    DeleteFileA(s_win_test_exe);
    DeleteFileA(s_sys_test_exe);
    DeleteFileA(s_win_txt_file);
    DeleteFileA(s_sys_txt_file);

    free(s_wi0.phwnd);
}

Referenced by START_TEST().

DoTestEntry()

static VOID DoTestEntry ( const TEST_ENTRY *  pEntry )

static

Definition at line 211 of file ShellExecuteEx.cpp.

{
    SHELLEXECUTEINFOA info = { sizeof(info) };
    info.fMask = SEE_MASK_NOCLOSEPROCESS | SEE_MASK_FLAG_NO_UI;
    info.nShow = SW_SHOWNORMAL;
    info.lpFile = pEntry->file;
    info.lpParameters = pEntry->params;
    info.lpDirectory = pEntry->curdir;
    BOOL ret = ShellExecuteExA(&info);
    ok(ret == pEntry->ret, "Line %u: ret expected %d, got %d\n",
       pEntry->lineno, pEntry->ret, ret);
    if (!pEntry->ret)
        return;

    if ((UINT)pEntry->bProcessHandle != DONT_CARE)
    {
        if (pEntry->bProcessHandle)
        {
            ok(!!info.hProcess, "Line %u: hProcess expected non-NULL\n", pEntry->lineno);
        }
        else
        {
            ok(!info.hProcess, "Line %u: hProcess expected NULL\n", pEntry->lineno);
            return;
        }
    }

    WaitForInputIdle(info.hProcess, INFINITE);

    // close newly opened windows
    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));

    WaitForSingleObject(info.hProcess, INFINITE);
    CloseHandle(info.hProcess);
}

Referenced by DoTestEntries().

EnumWindowsProc()

static BOOL CALLBACK EnumWindowsProc ( HWND  hwnd, LPARAM  lParam  )

static

Definition at line 200 of file ShellExecuteEx.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 DoTestEntries(), and DoTestEntry().

GetSubProgramPath()

static BOOL GetSubProgramPath ( void  )

static

Definition at line 264 of file ShellExecuteEx.cpp.

{
    GetModuleFileNameA(NULL, s_sub_program, _countof(s_sub_program));
    PathRemoveFileSpecA(s_sub_program);
    PathAppendA(s_sub_program, "shell32_apitest_sub.exe");

    if (!PathFileExistsA(s_sub_program))
    {
        PathRemoveFileSpecA(s_sub_program);
        PathAppendA(s_sub_program, "testdata\\shell32_apitest_sub.exe");

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

    return TRUE;
}

Referenced by DoTestEntries().

START_TEST()

START_TEST

(

ShellExecuteEx

)

Definition at line 367 of file ShellExecuteEx.cpp.

{
    DoAppPathTest();
    DoTestEntries();
}

TestShellExecuteEx()

static VOID TestShellExecuteEx ( const WCHAR *  Name, BOOL  ExpectedResult  )

static

Definition at line 65 of file ShellExecuteEx.cpp.

{
    SHELLEXECUTEINFOW ShellExecInfo;
    BOOL Result;
    ZeroMemory(&ShellExecInfo, sizeof(ShellExecInfo));
    ShellExecInfo.cbSize = sizeof(ShellExecInfo);
    ShellExecInfo.fMask = SEE_MASK_NOCLOSEPROCESS | SEE_MASK_FLAG_NO_UI;
    ShellExecInfo.hwnd = NULL;
    ShellExecInfo.nShow = SW_SHOWNORMAL;
    ShellExecInfo.lpFile = Name;
    ShellExecInfo.lpDirectory = NULL;
    Result = ShellExecuteExW(&ShellExecInfo);
    ok(Result == ExpectedResult, "ShellExecuteEx lpFile %s failed. Error: %lu\n", wine_dbgstr_w(Name), GetLastError());
    if (ShellExecInfo.hProcess)
    {
        Result = TerminateProcess(ShellExecInfo.hProcess, 0);
        if (!Result) trace("Terminate process failed. Error: %lu\n", GetLastError());
        WaitForSingleObject(ShellExecInfo.hProcess, INFINITE);
        CloseHandle(ShellExecInfo.hProcess);
    }
}

Variable Documentation

s_entries_1

const TEST_ENTRY s_entries_1[]

static

Definition at line 127 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_entries_2

const TEST_ENTRY s_entries_2[]

static

Initial value:

=
{
    { __LINE__, TRUE, TRUE, "test program" },
    { __LINE__, TRUE, TRUE, "test program", "TEST" },
    { __LINE__, TRUE, TRUE, "\"test program\"" },
    { __LINE__, TRUE, TRUE, s_win_test_exe },
    { __LINE__, TRUE, TRUE, s_sys_test_exe },
    { __LINE__, FALSE, FALSE, s_win_bat_file },
    { __LINE__, FALSE, FALSE, s_sys_bat_file },
    { __LINE__, FALSE, FALSE, s_win_bat_file, "TEST" },
    { __LINE__, FALSE, FALSE, s_sys_bat_file, "TEST" },
}

Definition at line 179 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_sub_program

char s_sub_program[MAX_PATH]

static

Definition at line 117 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries(), and GetSubProgramPath().

s_sys_bat_file

char s_sys_bat_file[MAX_PATH]

static

Definition at line 121 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_sys_test_exe

char s_sys_test_exe[MAX_PATH]

static

Definition at line 119 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_sys_txt_file

char s_sys_txt_file[MAX_PATH]

static

Definition at line 123 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_wi0

OPENWNDS s_wi0 = { 0 }

static

Definition at line 198 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries(), and DoTestEntry().

s_wi1

OPENWNDS s_wi1 = { 0 }

static

Definition at line 198 of file ShellExecuteEx.cpp.

Referenced by DoTestEntry().

s_win_bat_file

char s_win_bat_file[MAX_PATH]

static

Definition at line 120 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_win_test_exe

char s_win_test_exe[MAX_PATH]

static

Definition at line 118 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().

s_win_txt_file

char s_win_txt_file[MAX_PATH]

static

Definition at line 122 of file ShellExecuteEx.cpp.

Referenced by DoTestEntries().