reactos/modules/rostests/apitests/shell32/SHChangeNotify.cpp
Katayama Hirofumi MZ fccea84dd4
[SHELL32_APITEST] Add FindExecutable testcase (#3723)
Investigate the implementation of shell32!FindExecutable function.
CORE-17351, CORE-16898, CORE-17612
2021-06-06 06:51:42 +09:00

434 lines
14 KiB
C++

/*
* PROJECT: ReactOS api tests
* LICENSE: LGPL-2.0-or-later (https://spdx.org/licenses/LGPL-2.0-or-later)
* PURPOSE: Test for SHChangeNotify
* COPYRIGHT: Copyright 2020 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
*/
// NOTE: This test program closes the Explorer cabinets before tests.
#include "shelltest.h"
#include <shlwapi.h>
#include <stdio.h>
#include "SHChangeNotify.h"
#define DONT_SEND 0x24242424
static HWND s_hwnd = NULL;
static const WCHAR s_szName[] = L"SHChangeNotify testcase";
static WCHAR s_szSubProgram[MAX_PATH];
typedef void (*ACTION)(void);
typedef struct TEST_ENTRY
{
INT line;
DWORD event;
LPCVOID item1;
LPCVOID item2;
LPCSTR pattern;
ACTION action;
LPCWSTR path1;
LPCWSTR path2;
} TEST_ENTRY;
static BOOL
DoCreateEmptyFile(LPCWSTR pszFileName)
{
FILE *fp = _wfopen(pszFileName, L"wb");
fclose(fp);
return fp != NULL;
}
static void
DoAction1(void)
{
ok_int(CreateDirectoryW(s_dir2, NULL), TRUE);
}
static void
DoAction2(void)
{
ok_int(RemoveDirectoryW(s_dir2), TRUE);
}
static void
DoAction3(void)
{
ok_int(MoveFileExW(s_dir2, s_dir3, MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING), TRUE);
}
static void
DoAction4(void)
{
ok_int(DoCreateEmptyFile(s_file1), TRUE);
}
static void
DoAction5(void)
{
ok_int(MoveFileExW(s_file1, s_file2, MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING), TRUE);
}
static void
DoAction6(void)
{
ok_int(DeleteFileW(s_file2), TRUE);
}
static void
DoAction7(void)
{
DeleteFileW(s_file1);
DeleteFileW(s_file2);
ok_int(RemoveDirectoryW(s_dir3), TRUE);
}
static void
DoAction8(void)
{
BOOL ret = RemoveDirectoryW(s_dir1);
ok(ret, "RemoveDirectoryW failed. GetLastError() == %ld\n", GetLastError());
}
static const TEST_ENTRY s_TestEntriesMode0[] =
{
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, NULL, DoAction1, NULL, NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "00001000", NULL, s_dir2, L""},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "00001000", DoAction2, s_dir2, L""},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "00010000", DoAction1, s_dir2, L""},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, s_dir3, "00000001", NULL, s_dir2, s_dir3},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, s_dir3, "00000001", DoAction3, s_dir2, s_dir3},
{__LINE__, SHCNE_CREATE, s_file1, NULL, "01000000", NULL, s_file1, L""},
{__LINE__, SHCNE_CREATE, s_file1, s_file2, "01000000", NULL, s_file1, s_file2},
{__LINE__, SHCNE_CREATE, s_file1, NULL, "01000000", DoAction4, s_file1, L""},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "10000000", NULL, s_file1, s_file2},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "10000000", DoAction5, s_file1, s_file2},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "10000000", NULL, s_file1, s_file2},
{__LINE__, SHCNE_UPDATEITEM, s_file1, NULL, "00000010", NULL, s_file1, L""},
{__LINE__, SHCNE_UPDATEITEM, s_file2, NULL, "00000010", NULL, s_file2, L""},
{__LINE__, SHCNE_DELETE, s_file1, NULL, "00100000", NULL, s_file1, L""},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "00100000", NULL, s_file2, L""},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "00100000", DoAction6, s_file2, L""},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "00100000", NULL, s_file2, L""},
{__LINE__, SHCNE_DELETE, s_file1, NULL, "00100000", NULL, s_file1, L""},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "00001000", NULL, s_dir2, L""},
{__LINE__, SHCNE_RMDIR, s_dir3, NULL, "00001000", DoAction7, s_dir3, L""},
{__LINE__, SHCNE_RMDIR, s_dir1, NULL, "00001000", NULL, s_dir1, L""},
{__LINE__, SHCNE_RMDIR, s_dir1, NULL, "00001000", DoAction8, s_dir1, L""},
};
#define s_TestEntriesMode1 s_TestEntriesMode0
#define s_TestEntriesMode2 s_TestEntriesMode0
static const TEST_ENTRY s_TestEntriesMode3[] =
{
{__LINE__, DONT_SEND, s_dir2, NULL, NULL, DoAction1, NULL, NULL},
{__LINE__, DONT_SEND, s_dir2, NULL, "00001000", DoAction2, s_dir2, L""},
{__LINE__, DONT_SEND, s_dir2, NULL, "00010000", DoAction1, s_dir2, L""},
{__LINE__, DONT_SEND, s_dir2, s_dir3, "00000001", DoAction3, s_dir2, s_dir3},
{__LINE__, DONT_SEND, s_file1, NULL, "01000000", DoAction4, s_file1, L""},
{__LINE__, DONT_SEND, s_file1, s_file2, "10000000", DoAction5, s_file1, s_file2},
{__LINE__, DONT_SEND, s_file2, NULL, "00100000", DoAction6, s_file2, L""},
{__LINE__, DONT_SEND, s_dir3, NULL, "00001000", DoAction7, s_dir3, L""},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_MKDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
};
static const TEST_ENTRY s_TestEntriesMode4[] =
{
{__LINE__, DONT_SEND, s_dir2, NULL, NULL, DoAction1, NULL, NULL},
{__LINE__, DONT_SEND, s_dir2, NULL, "00001000", DoAction2, s_dir2, L""},
{__LINE__, DONT_SEND, s_dir2, NULL, "00010000", DoAction1, s_dir2, L""},
{__LINE__, DONT_SEND, s_dir2, s_dir3, "00000001", DoAction3, s_dir2, s_dir3},
{__LINE__, DONT_SEND, s_file1, NULL, "01000000", DoAction4, s_file1, L""},
{__LINE__, DONT_SEND, s_file1, s_file2, "10000000", DoAction5, s_file1, s_file2},
{__LINE__, DONT_SEND, s_file2, NULL, "00100000", DoAction6, s_file2, L""},
{__LINE__, DONT_SEND, s_dir3, NULL, "00001000", DoAction7, s_dir3, L""},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_MKDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
};
static const TEST_ENTRY s_TestEntriesMode5[] =
{
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "00000000", DoAction1, NULL, NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "00000000", DoAction2, NULL, NULL},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "00000000", DoAction1, NULL, NULL},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, s_dir3, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, s_dir3, "00000000", DoAction3, NULL, NULL},
{__LINE__, SHCNE_CREATE, s_file1, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_CREATE, s_file1, s_file2, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_CREATE, s_file1, NULL, "00000000", DoAction4, NULL, NULL},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "00000000", DoAction5, NULL, NULL},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file1, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file1, s_file2, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file2, s_file1, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file1, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file2, s_file1, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file1, s_file2, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "00000000", DoAction6, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_DELETE, s_file1, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir1, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir2, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir3, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir1, s_dir2, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir1, s_dir3, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir2, s_dir1, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir2, s_dir3, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir3, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir3, NULL, "00000000", DoAction7, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir1, NULL, "00000000", NULL, NULL, NULL},
{__LINE__, SHCNE_INTERRUPT | SHCNE_RMDIR, s_dir1, NULL, "00000000", DoAction8, NULL, NULL},
};
LPCSTR PatternFromFlags(DWORD flags)
{
static char s_buf[TYPE_RENAMEFOLDER + 1 + 1];
DWORD i;
for (i = 0; i <= TYPE_RENAMEFOLDER; ++i)
{
s_buf[i] = (char)('0' + !!(flags & (1 << i)));
}
s_buf[i] = 0;
return s_buf;
}
static BOOL
DoGetPaths(LPWSTR pszPath1, LPWSTR pszPath2)
{
pszPath1[0] = pszPath2[0] = 0;
WCHAR szText[MAX_PATH * 2];
szText[0] = 0;
if (FILE *fp = fopen(TEMP_FILE, "rb"))
{
fread(szText, 1, sizeof(szText), fp);
fclose(fp);
}
LPWSTR pch = wcschr(szText, L'|');
if (pch == NULL)
return FALSE;
*pch = 0;
lstrcpynW(pszPath1, szText, MAX_PATH);
lstrcpynW(pszPath2, pch + 1, MAX_PATH);
return TRUE;
}
static void
DoTestEntry(const TEST_ENTRY *entry)
{
if (entry->action)
{
(*entry->action)();
}
if (entry->event != DONT_SEND)
{
SHChangeNotify(entry->event, SHCNF_PATHW | SHCNF_FLUSH, entry->item1, entry->item2);
}
else
{
SHChangeNotify(0, SHCNF_FLUSH, NULL, NULL);
}
DWORD flags = SendMessageW(s_hwnd, WM_GET_NOTIFY_FLAGS, 0, 0);
LPCSTR pattern = PatternFromFlags(flags);
if (entry->pattern)
{
ok(lstrcmpA(pattern, entry->pattern) == 0 ||
lstrcmpA(pattern, "00000100") == 0, // SHCNE_UPDATEDIR
"Line %d: pattern mismatch '%s'\n", entry->line, pattern);
}
SendMessageW(s_hwnd, WM_SET_PATHS, 0, 0);
Sleep(50);
WCHAR szPath1[MAX_PATH], szPath2[MAX_PATH];
szPath1[0] = szPath2[0] = 0;
BOOL bOK = DoGetPaths(szPath1, szPath2);
if (lstrcmpA(pattern, "00000100") == 0) // SHCNE_UPDATEDIR
{
if (entry->path1)
ok(bOK && lstrcmpiW(s_dir1, szPath1) == 0,
"Line %d: path1 mismatch '%S' (%d)\n", entry->line, szPath1, bOK);
if (entry->path2)
ok(bOK && lstrcmpiW(L"", szPath2) == 0,
"Line %d: path2 mismatch '%S' (%d)\n", entry->line, szPath2, bOK);
}
else
{
if (entry->path1)
ok(bOK && lstrcmpiW(entry->path1, szPath1) == 0,
"Line %d: path1 mismatch '%S' (%d)\n", entry->line, szPath1, bOK);
if (entry->path2)
ok(bOK && lstrcmpiW(entry->path2, szPath2) == 0,
"Line %d: path2 mismatch '%S' (%d)\n", entry->line, szPath2, bOK);
}
SendMessageW(s_hwnd, WM_CLEAR_FLAGS, 0, 0);
}
static BOOL
DoInit(void)
{
DoInitPaths();
CreateDirectoryW(s_dir1, NULL);
// close Explorer before tests
INT i, nCount = 50;
for (i = 0; i < nCount; ++i)
{
HWND hwnd = FindWindowW(L"CabinetWClass", NULL);
if (hwnd == NULL)
break;
PostMessage(hwnd, WM_CLOSE, 0, 0);
Sleep(100);
}
if (i == nCount)
skip("Unable to close Explorer cabinet\n");
return PathIsDirectoryW(s_dir1);
}
static void
DoEnd(HWND hwnd)
{
DeleteFileW(s_file1);
DeleteFileW(s_file2);
RemoveDirectoryW(s_dir3);
RemoveDirectoryW(s_dir2);
RemoveDirectoryW(s_dir1);
DeleteFileA(TEMP_FILE);
SendMessageW(s_hwnd, WM_COMMAND, IDOK, 0);
}
static BOOL
GetSubProgramPath(void)
{
GetModuleFileNameW(NULL, s_szSubProgram, _countof(s_szSubProgram));
PathRemoveFileSpecW(s_szSubProgram);
PathAppendW(s_szSubProgram, L"shell32_apitest_sub.exe");
if (!PathFileExistsW(s_szSubProgram))
{
PathRemoveFileSpecW(s_szSubProgram);
PathAppendW(s_szSubProgram, L"testdata\\shell32_apitest_sub.exe");
if (!PathFileExistsW(s_szSubProgram))
{
return FALSE;
}
}
return TRUE;
}
static void
JustDoIt(INT nMode)
{
trace("nMode: %d\n", nMode);
SHChangeNotify(0, SHCNF_FLUSH, NULL, NULL);
if (!DoInit())
{
skip("Unable to initialize.\n");
return;
}
WCHAR szParams[8];
wsprintfW(szParams, L"%u", nMode);
HINSTANCE hinst = ShellExecuteW(NULL, NULL, s_szSubProgram, szParams, NULL, SW_SHOWNORMAL);
if ((INT_PTR)hinst <= 32)
{
skip("Unable to run shell32_apitest_sub.exe.\n");
return;
}
for (int i = 0; i < 15; ++i)
{
s_hwnd = FindWindowW(s_szName, s_szName);
if (s_hwnd)
break;
Sleep(50);
}
if (!s_hwnd)
{
skip("Unable to find window.\n");
return;
}
switch (nMode)
{
case 0:
case 1:
case 2:
for (size_t i = 0; i < _countof(s_TestEntriesMode0); ++i)
{
DoTestEntry(&s_TestEntriesMode0[i]);
}
break;
case 3:
for (size_t i = 0; i < _countof(s_TestEntriesMode3); ++i)
{
DoTestEntry(&s_TestEntriesMode3[i]);
}
break;
case 4:
for (size_t i = 0; i < _countof(s_TestEntriesMode4); ++i)
{
DoTestEntry(&s_TestEntriesMode4[i]);
}
break;
case 5:
for (size_t i = 0; i < _countof(s_TestEntriesMode5); ++i)
{
DoTestEntry(&s_TestEntriesMode5[i]);
}
break;
}
DoEnd(s_hwnd);
for (int i = 0; i < 15; ++i)
{
s_hwnd = FindWindowW(s_szName, s_szName);
if (!s_hwnd)
break;
Sleep(50);
}
}
START_TEST(SHChangeNotify)
{
if (!GetSubProgramPath())
{
skip("shell32_apitest_sub.exe not found\n");
}
JustDoIt(0);
JustDoIt(1);
JustDoIt(2);
JustDoIt(3);
JustDoIt(4);
JustDoIt(5);
}