Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-29 02:25:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

[SHELL32_APITEST] Add SHChangeNotify testcase (#2378)

Browse source 
Add a testcase for shell32!SHChangeNotify function to investigate the shell notification mechanism. CORE-13950
This commit is contained in:
Katayama Hirofumi MZ 2020-02-25 00:45:44 +09:00 committed by GitHub
parent 5a81a5bc14
commit fe39760206
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 477 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
modules/rostests/apitests/shell32/CMakeLists.txt
View file

@ -22,6 +22,7 @@ list(APPEND SOURCE
IShellFolderViewCB.cpp
OpenAs_RunDLL.cpp
PathResolve.cpp
SHChangeNotify.cpp
SHCreateDataObject.cpp
SHCreateFileExtractIconW.cpp
SHParseDisplayName.cpp

474
modules/rostests/apitests/shell32/SHChangeNotify.cpp Normal file
View file

@ -0,0 +1,474 @@
/*
* 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)
*/
#include "shelltest.h"
#include <shlwapi.h>
#include <stdio.h>
#define WM_SHELL_NOTIFY (WM_USER + 100)
#define ID_TEST 1000
static WCHAR s_dir1[MAX_PATH]; // "%TEMP%\\WatchDir1"
static WCHAR s_dir2[MAX_PATH]; // "%TEMP%\\WatchDir1\\Dir2"
static WCHAR s_dir3[MAX_PATH]; // "%TEMP%\\WatchDir1\\Dir3"
static WCHAR s_file1[MAX_PATH]; // "%TEMP%\\WatchDir1\\File1.txt"
static WCHAR s_file2[MAX_PATH]; // "%TEMP%\\WatchDir1\\File2.txt"
static HWND s_hwnd = NULL;
static WCHAR s_szName[] = L"SHChangeNotify testcase";
static LPITEMIDLIST s_pidl = NULL;
static UINT s_uRegID = 0;
static SHChangeNotifyEntry s_entry;
static CHAR s_path1[MAX_PATH], s_path2[MAX_PATH];
typedef enum TYPE
{
TYPE_RENAMEITEM,
TYPE_CREATE,
TYPE_DELETE,
TYPE_MKDIR,
TYPE_RMDIR,
TYPE_UPDATEDIR,
TYPE_UPDATEITEM,
TYPE_RENAMEFOLDER,
TYPE_FREESPACE
} TYPE;
static BYTE s_counters[TYPE_FREESPACE + 1];
static LPCSTR
DoGetPattern(void)
{
size_t i;
static char buf[TYPE_FREESPACE + 1 + 1];
for (i = 0; i < TYPE_FREESPACE + 1; ++i)
{
buf[i] = (char)('0' + s_counters[i]);
}
buf[i] = 0;
return buf;
}
typedef void (*ACTION)(void);
typedef struct TEST_ENTRY
{
INT line;
LONG event;
LPCVOID item1;
LPCVOID item2;
LPCSTR pattern;
ACTION action;
} 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)
{
ok_int(RemoveDirectoryW(s_dir1), TRUE);
}
static const TEST_ENTRY s_TestEntries[] = {
{__LINE__, SHCNE_MKDIR, s_dir1, NULL, "000100000", NULL},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "000100000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "000010000", NULL},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "000100000", DoAction1},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "000010000", DoAction2},
{__LINE__, SHCNE_MKDIR, s_dir2, NULL, "000100000", DoAction1},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, s_dir3, "000000010", NULL},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, NULL, "000000010", NULL},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, s_dir3, "000000010", DoAction3},
{__LINE__, SHCNE_RENAMEFOLDER, s_dir2, NULL, "000000010", NULL},
{__LINE__, SHCNE_CREATE, s_file1, NULL, "010000000", NULL},
{__LINE__, SHCNE_CREATE, s_file1, s_file2, "010000000", NULL},
{__LINE__, SHCNE_CREATE, s_file1, NULL, "010000000", DoAction4},
{__LINE__, SHCNE_RENAMEITEM, s_file1, NULL, "100000000", NULL},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "100000000", NULL},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "100000000", DoAction5},
{__LINE__, SHCNE_RENAMEITEM, s_file1, s_file2, "100000000", NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file1, NULL, "000000100", NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file2, NULL, "000000100", NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file1, s_file2, "000000100", NULL},
{__LINE__, SHCNE_UPDATEITEM, s_file2, s_file1, "000000100", NULL},
{__LINE__, SHCNE_DELETE, s_file1, NULL, "001000000", NULL},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "001000000", NULL},
{__LINE__, SHCNE_DELETE, s_file2, s_file1, "001000000", NULL},
{__LINE__, SHCNE_DELETE, s_file1, s_file2, "001000000", NULL},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "001000000", DoAction6},
{__LINE__, SHCNE_DELETE, s_file2, NULL, "001000000", NULL},
{__LINE__, SHCNE_DELETE, s_file1, NULL, "001000000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_file1, NULL, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_file2, NULL, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_file1, s_file2, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_file2, s_file1, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_dir1, NULL, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_dir2, NULL, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_dir1, s_dir2, "000001000", NULL},
{__LINE__, SHCNE_UPDATEDIR, s_dir2, s_dir1, "000001000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir1, NULL, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, NULL, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir3, NULL, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir1, s_dir2, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir1, s_dir3, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, s_dir1, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir2, s_dir3, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir3, NULL, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir3, NULL, "000010000", DoAction7},
{__LINE__, SHCNE_RMDIR, s_dir1, NULL, "000010000", NULL},
{__LINE__, SHCNE_RMDIR, s_dir1, NULL, "000010000", DoAction8},
};
static const size_t s_nTestEntries = _countof(s_TestEntries);
static size_t s_iTest = 0;
static void
DoTestEntry1(const TEST_ENTRY *entry)
{
if (entry->action)
{
(*entry->action)();
}
SHChangeNotify(entry->event, SHCNF_PATHW | SHCNF_FLUSH, entry->item1, entry->item2);
SendMessageW(s_hwnd, WM_COMMAND, ID_TEST + s_iTest, 0);
ZeroMemory(&s_counters, sizeof(s_counters));
}
static void
DoTestEntry2(const TEST_ENTRY *entry)
{
LPCSTR pattern = DoGetPattern();
ok(lstrcmpA(pattern, entry->pattern) == 0, "Line %d: pattern mismatch '%s'\n", entry->line, pattern);
}
static BOOL
DoInit(HWND hwnd)
{
WCHAR szTemp[MAX_PATH], szPath[MAX_PATH];
GetTempPathW(_countof(szTemp), szTemp);
GetLongPathNameW(szTemp, szPath, _countof(szPath));
lstrcpyW(s_dir1, szPath);
PathAppendW(s_dir1, L"WatchDir1");
CreateDirectoryW(s_dir1, NULL);
lstrcpyW(s_dir2, s_dir1);
PathAppendW(s_dir2, L"Dir2");
lstrcpyW(s_dir3, s_dir1);
PathAppendW(s_dir3, L"Dir3");
lstrcpyW(s_file1, s_dir1);
PathAppendW(s_file1, L"File1.txt");
lstrcpyW(s_file2, s_dir1);
PathAppendW(s_file2, L"File2.txt");
s_pidl = ILCreateFromPathW(s_dir1);
s_entry.pidl = s_pidl;
s_entry.fRecursive = TRUE;
LONG fEvents = SHCNE_ALLEVENTS;
s_uRegID = SHChangeNotifyRegister(hwnd, SHCNRF_ShellLevel, fEvents, WM_SHELL_NOTIFY,
1, &s_entry);
return s_uRegID != 0;
}
static DWORD WINAPI ThreadFunc(LPVOID)
{
for (size_t i = 0; i < s_nTestEntries; ++i)
{
s_iTest = i;
DoTestEntry1(&s_TestEntries[i]);
}
SendMessageW(s_hwnd, WM_COMMAND, IDOK, 0);
return 0;
}
static BOOL
OnCreate(HWND hwnd)
{
s_hwnd = hwnd;
BOOL bOK = DoInit(hwnd);
if (!bOK)
{
skip("SHChangeNotifyRegister failed\n");
return FALSE;
}
DWORD tid;
HANDLE hThread = CreateThread(NULL, 0, ThreadFunc, NULL, 0, &tid);
if (hThread == NULL)
{
skip("CreateThread failed\n");
return FALSE;
}
CloseHandle(hThread);
return TRUE;
}
static void
OnCommand(HWND hwnd, UINT id)
{
switch (id)
{
case IDOK:
case IDCANCEL:
DestroyWindow(hwnd);
break;
default:
if (ID_TEST <= id && id < ID_TEST + 1000)
{
DoTestEntry2(&s_TestEntries[s_iTest]);
}
break;
}
}
static void
OnDestroy(HWND hwnd)
{
SHChangeNotifyDeregister(s_uRegID);
CoTaskMemFree(s_pidl);
DeleteFileW(s_file1);
DeleteFileW(s_file2);
RemoveDirectoryW(s_dir3);
RemoveDirectoryW(s_dir2);
RemoveDirectoryW(s_dir1);
PostQuitMessage(0);
s_hwnd = NULL;
}
static void
DoShellNotify(HWND hwnd, PIDLIST_ABSOLUTE pidl1, PIDLIST_ABSOLUTE pidl2, LONG lEvent)
{
if (pidl1)
SHGetPathFromIDListA(pidl1, s_path1);
else
s_path1[0] = 0;
if (pidl2)
SHGetPathFromIDListA(pidl2, s_path2);
else
s_path2[0] = 0;
switch (lEvent)
{
case SHCNE_RENAMEITEM:
trace("SHCNE_RENAMEITEM('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_RENAMEITEM] = 1;
break;
case SHCNE_CREATE:
trace("SHCNE_CREATE('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_CREATE] = 1;
break;
case SHCNE_DELETE:
trace("SHCNE_DELETE('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_DELETE] = 1;
break;
case SHCNE_MKDIR:
trace("SHCNE_MKDIR('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_MKDIR] = 1;
break;
case SHCNE_RMDIR:
trace("SHCNE_RMDIR('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_RMDIR] = 1;
break;
case SHCNE_MEDIAINSERTED:
trace("SHCNE_MEDIAINSERTED('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_MEDIAREMOVED:
trace("SHCNE_MEDIAREMOVED('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_DRIVEREMOVED:
trace("SHCNE_DRIVEREMOVED('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_DRIVEADD:
trace("SHCNE_DRIVEADD('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_NETSHARE:
trace("SHCNE_NETSHARE('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_NETUNSHARE:
trace("SHCNE_NETUNSHARE('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_ATTRIBUTES:
trace("SHCNE_ATTRIBUTES('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_UPDATEDIR:
trace("SHCNE_UPDATEDIR('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_UPDATEDIR] = 1;
break;
case SHCNE_UPDATEITEM:
trace("SHCNE_UPDATEITEM('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_UPDATEITEM] = 1;
break;
case SHCNE_SERVERDISCONNECT:
trace("SHCNE_SERVERDISCONNECT('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_UPDATEIMAGE:
trace("SHCNE_UPDATEIMAGE('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_DRIVEADDGUI:
trace("SHCNE_DRIVEADDGUI('%s', '%s')\n", s_path1, s_path2);
break;
case SHCNE_RENAMEFOLDER:
trace("SHCNE_RENAMEFOLDER('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_RENAMEFOLDER] = 1;
break;
case SHCNE_FREESPACE:
trace("SHCNE_FREESPACE('%s', '%s')\n", s_path1, s_path2);
s_counters[TYPE_FREESPACE] = 1;
break;
case SHCNE_EXTENDED_EVENT:
trace("SHCNE_EXTENDED_EVENT('%p', '%p')\n", pidl1, pidl2);
break;
case SHCNE_ASSOCCHANGED:
trace("SHCNE_ASSOCCHANGED('%s', '%s')\n", s_path1, s_path2);
break;
default:
trace("(lEvent:%08lX)('%s', '%s')\n", lEvent, s_path1, s_path2);
break;
}
}
static INT_PTR
OnShellNotify(HWND hwnd, WPARAM wParam, LPARAM lParam)
{
LONG lEvent;
PIDLIST_ABSOLUTE *pidlAbsolute;
HANDLE hLock = SHChangeNotification_Lock((HANDLE)wParam, (DWORD)lParam, &pidlAbsolute, &lEvent);
if (hLock)
{
DoShellNotify(hwnd, pidlAbsolute[0], pidlAbsolute[1], lEvent);
SHChangeNotification_Unlock(hLock);
}
else
{
pidlAbsolute = (PIDLIST_ABSOLUTE *)wParam;
DoShellNotify(hwnd, pidlAbsolute[0], pidlAbsolute[1], lParam);
}
return TRUE;
}
static LRESULT CALLBACK
WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
case WM_CREATE:
return (OnCreate(hwnd) ? 0 : -1);
case WM_COMMAND:
OnCommand(hwnd, LOWORD(wParam));
break;
case WM_SHELL_NOTIFY:
return OnShellNotify(hwnd, wParam, lParam);
case WM_DESTROY:
OnDestroy(hwnd);
break;
default:
return DefWindowProcW(hwnd, uMsg, wParam, lParam);
}
return 0;
}
START_TEST(SHChangeNotify)
{
WNDCLASSW wc;
ZeroMemory(&wc, sizeof(wc));
wc.lpfnWndProc = WindowProc;
wc.hInstance = GetModuleHandleW(NULL);
wc.hIcon = LoadIcon(NULL, IDI_APPLICATION);
wc.hCursor = LoadCursor(NULL, IDC_ARROW);
wc.hbrBackground = (HBRUSH)(COLOR_3DFACE + 1);
wc.lpszClassName = s_szName;
if (!RegisterClassW(&wc))
{
skip("RegisterClassW failed\n");
return;
}
HWND hwnd = CreateWindowW(s_szName, s_szName, WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT, CW_USEDEFAULT, 100, 100,
NULL, NULL, GetModuleHandleW(NULL), NULL);
if (!hwnd)
{
skip("CreateWindowW failed\n");
return;
}
ShowWindow(hwnd, SW_SHOWNORMAL);
UpdateWindow(hwnd);
MSG msg;
while (GetMessageW(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessageW(&msg);
}
}

2
modules/rostests/apitests/shell32/testlist.c
View file

@ -17,6 +17,7 @@ extern void func_IShellFolderViewCB(void);
extern void func_menu(void);
extern void func_OpenAs_RunDLL(void);
extern void func_PathResolve(void);
extern void func_SHChangeNotify(void);
extern void func_SHCreateDataObject(void);
extern void func_SHCreateFileExtractIconW(void);
extern void func_ShellExecCmdLine(void);
@ -42,6 +43,7 @@ const struct test winetest_testlist[] =
{ "menu", func_menu },
{ "OpenAs_RunDLL", func_OpenAs_RunDLL },
{ "PathResolve", func_PathResolve },
{ "SHChangeNotify", func_SHChangeNotify },
{ "SHCreateDataObject", func_SHCreateDataObject },
{ "SHCreateFileExtractIconW", func_SHCreateFileExtractIconW },
{ "ShellExecCmdLine", func_ShellExecCmdLine },