mirror of
https://github.com/reactos/reactos.git
synced 2025-04-20 20:36:35 +00:00
[KMTESTS]
- Add a kernel32 test (I know right) to check how FindFirstFile* modifies wildcards ROSTESTS-104 #resolve svn path=/trunk/; revision=58813
This commit is contained in:
Thomas Faber
parent
a3ee37624f
commit
3e16da2604
6 changed files with 250 additions and 0 deletions
|
|
@ -5,6 +5,7 @@ include_directories(
|
|||
# subdirectories containing special-purpose drivers
|
||||
#
|
||||
add_subdirectory(example)
|
||||
add_subdirectory(kernel32)
|
||||
add_subdirectory(ntos_io)
|
||||
|
||||
list(APPEND COMMON_SOURCE
|
||||
|
|
@ -87,6 +88,7 @@ list(APPEND KMTEST_SOURCE
|
|||
kmtest/testlist.c
|
||||
|
||||
example/Example_user.c
|
||||
kernel32/FindFile_user.c
|
||||
ntos_io/IoDeviceObject_user.c
|
||||
${COMMON_SOURCE}
|
||||
|
||||
|
|
|
|||
16
rostests/kmtests/kernel32/CMakeLists.txt
Normal file
16
rostests/kmtests/kernel32/CMakeLists.txt
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
include_directories(
|
||||
../include)
|
||||
|
||||
list(APPEND FINDFILE_DRV_SOURCE
|
||||
../kmtest_drv/kmtest_standalone.c
|
||||
FindFile_drv.c)
|
||||
|
||||
add_library(findfile_drv SHARED ${FINDFILE_DRV_SOURCE})
|
||||
|
||||
set_module_type(findfile_drv kernelmodedriver)
|
||||
target_link_libraries(findfile_drv kmtest_printf ${PSEH_LIB})
|
||||
add_importlibs(findfile_drv ntoskrnl hal)
|
||||
add_target_compile_definitions(findfile_drv KMT_STANDALONE_DRIVER)
|
||||
#add_pch(findfile_drv ../include/kmt_test.h)
|
||||
|
||||
add_cd_file(TARGET findfile_drv DESTINATION reactos/bin FOR all)
|
||||
13
rostests/kmtests/kernel32/FindFile.h
Normal file
13
rostests/kmtests/kernel32/FindFile.h
Normal file
|
|
@ -0,0 +1,13 @@
|
|||
/*
|
||||
* PROJECT: ReactOS kernel-mode tests
|
||||
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
||||
* PURPOSE: FindFirstFile wildcard substitution test declarations
|
||||
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
|
||||
*/
|
||||
|
||||
#ifndef _KMTEST_FINDFILE_H_
|
||||
#define _KMTEST_FINDFILE_H_
|
||||
|
||||
#define IOCTL_EXPECT 1
|
||||
|
||||
#endif /* !defined _KMTEST_FINDFILE_H_ */
|
||||
123
rostests/kmtests/kernel32/FindFile_drv.c
Normal file
123
rostests/kmtests/kernel32/FindFile_drv.c
Normal file
|
|
@ -0,0 +1,123 @@
|
|||
/*
|
||||
* PROJECT: ReactOS kernel-mode tests
|
||||
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
||||
* PURPOSE: Test driver for FindFirstFile's wildcard substitution
|
||||
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
|
||||
*/
|
||||
|
||||
#include <kmt_test.h>
|
||||
|
||||
#define NDEBUG
|
||||
#include <debug.h>
|
||||
|
||||
#include "FindFile.h"
|
||||
|
||||
static KMT_MESSAGE_HANDLER TestMessageHandler;
|
||||
static KMT_IRP_HANDLER TestIrpHandler;
|
||||
|
||||
static UNICODE_STRING ExpectedExpression = RTL_CONSTANT_STRING(L"<not set>");
|
||||
static WCHAR ExpressionBuffer[MAX_PATH];
|
||||
|
||||
NTSTATUS
|
||||
TestEntry(
|
||||
IN PDRIVER_OBJECT DriverObject,
|
||||
IN PCUNICODE_STRING RegistryPath,
|
||||
OUT PCWSTR *DeviceName,
|
||||
IN OUT INT *Flags)
|
||||
{
|
||||
NTSTATUS Status = STATUS_SUCCESS;
|
||||
|
||||
PAGED_CODE();
|
||||
|
||||
UNREFERENCED_PARAMETER(RegistryPath);
|
||||
|
||||
*DeviceName = L"FindFile";
|
||||
*Flags = TESTENTRY_NO_EXCLUSIVE_DEVICE;
|
||||
|
||||
KmtRegisterIrpHandler(IRP_MJ_DIRECTORY_CONTROL, NULL, TestIrpHandler);
|
||||
KmtRegisterMessageHandler(0, NULL, TestMessageHandler);
|
||||
|
||||
return Status;
|
||||
}
|
||||
|
||||
VOID
|
||||
TestUnload(
|
||||
IN PDRIVER_OBJECT DriverObject)
|
||||
{
|
||||
PAGED_CODE();
|
||||
}
|
||||
|
||||
static
|
||||
NTSTATUS
|
||||
TestMessageHandler(
|
||||
IN PDEVICE_OBJECT DeviceObject,
|
||||
IN ULONG ControlCode,
|
||||
IN PVOID Buffer OPTIONAL,
|
||||
IN SIZE_T InLength,
|
||||
IN OUT PSIZE_T OutLength)
|
||||
{
|
||||
NTSTATUS Status = STATUS_SUCCESS;
|
||||
|
||||
PAGED_CODE();
|
||||
|
||||
switch (ControlCode)
|
||||
{
|
||||
case IOCTL_EXPECT:
|
||||
{
|
||||
C_ASSERT(sizeof(ExpressionBuffer) <= UNICODE_STRING_MAX_BYTES);
|
||||
DPRINT("IOCTL_EXPECT, InLength = %lu\n", InLength);
|
||||
if (InLength > sizeof(ExpressionBuffer))
|
||||
return STATUS_BUFFER_OVERFLOW;
|
||||
|
||||
if (InLength % sizeof(WCHAR) != 0)
|
||||
return STATUS_INVALID_PARAMETER;
|
||||
|
||||
RtlInitEmptyUnicodeString(&ExpectedExpression, ExpressionBuffer, sizeof(ExpressionBuffer));
|
||||
RtlCopyMemory(ExpressionBuffer, Buffer, InLength);
|
||||
ExpectedExpression.Length = (USHORT)InLength;
|
||||
DPRINT("IOCTL_EXPECT: %wZ\n", &ExpectedExpression);
|
||||
|
||||
break;
|
||||
}
|
||||
default:
|
||||
return STATUS_NOT_SUPPORTED;
|
||||
}
|
||||
|
||||
return Status;
|
||||
}
|
||||
|
||||
static
|
||||
NTSTATUS
|
||||
TestIrpHandler(
|
||||
IN PDEVICE_OBJECT DeviceObject,
|
||||
IN PIRP Irp,
|
||||
IN PIO_STACK_LOCATION IoStackLocation)
|
||||
{
|
||||
NTSTATUS Status = STATUS_NOT_SUPPORTED;
|
||||
|
||||
PAGED_CODE();
|
||||
|
||||
DPRINT("IRP %x/%x\n", IoStackLocation->MajorFunction, IoStackLocation->MinorFunction);
|
||||
ASSERT(IoStackLocation->MajorFunction == IRP_MJ_DIRECTORY_CONTROL);
|
||||
|
||||
ok(IoStackLocation->MinorFunction == IRP_MN_QUERY_DIRECTORY, "Minor function: %u\n", IoStackLocation->MinorFunction);
|
||||
if (IoStackLocation->MinorFunction == IRP_MN_QUERY_DIRECTORY)
|
||||
{
|
||||
ok(IoStackLocation->Parameters.QueryDirectory.FileInformationClass == FileBothDirectoryInformation,
|
||||
"FileInformationClass: %d\n", IoStackLocation->Parameters.QueryDirectory.FileInformationClass);
|
||||
if (IoStackLocation->Parameters.QueryDirectory.FileInformationClass == FileBothDirectoryInformation)
|
||||
{
|
||||
ok(RtlEqualUnicodeString(IoStackLocation->Parameters.QueryDirectory.FileName, &ExpectedExpression, FALSE),
|
||||
"Expression is '%wZ', expected '%wZ'\n", IoStackLocation->Parameters.QueryDirectory.FileName, &ExpectedExpression);
|
||||
RtlZeroMemory(Irp->UserBuffer, IoStackLocation->Parameters.QueryDirectory.Length);
|
||||
Status = STATUS_SUCCESS;
|
||||
}
|
||||
}
|
||||
|
||||
Irp->IoStatus.Status = Status;
|
||||
Irp->IoStatus.Information = 0;
|
||||
|
||||
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||||
|
||||
return Status;
|
||||
}
|
||||
94
rostests/kmtests/kernel32/FindFile_user.c
Normal file
94
rostests/kmtests/kernel32/FindFile_user.c
Normal file
|
|
@ -0,0 +1,94 @@
|
|||
/*
|
||||
* PROJECT: ReactOS kernel-mode tests
|
||||
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
||||
* PURPOSE: Test for FindFirstFile's wildcard substitution
|
||||
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
|
||||
*/
|
||||
|
||||
#include <kmt_test.h>
|
||||
|
||||
#include "FindFile.h"
|
||||
|
||||
START_TEST(FindFile)
|
||||
{
|
||||
HANDLE FindHandle;
|
||||
WIN32_FIND_DATAW FindData;
|
||||
struct
|
||||
{
|
||||
PCWSTR Expression;
|
||||
PCWSTR ExpectedExpression;
|
||||
} Tests[] =
|
||||
{
|
||||
{ L"Hello", L"Hello" },
|
||||
|
||||
{ L"*", L"*" },
|
||||
{ L"a*", L"a*" },
|
||||
{ L"*a", L"*a" },
|
||||
{ L"a*a", L"a*a" },
|
||||
{ L"**", L"**" },
|
||||
{ L"*a*", L"*a*" },
|
||||
{ L"a*a*a", L"a*a*a" },
|
||||
|
||||
{ L"*.*", L"*" },
|
||||
{ L"a*.*", L"a<\"*" },
|
||||
{ L"*.*a", L"<\"*a" },
|
||||
{ L"a*.*a", L"a<\"*a" },
|
||||
{ L"*.**.*", L"<\"*<\"*" },
|
||||
{ L"*.*a*.*", L"<\"*a<\"*" },
|
||||
{ L"a*.*a*.*a", L"a<\"*a<\"*a" },
|
||||
|
||||
{ L".*", L"\"*" },
|
||||
{ L"a.*", L"a\"*" },
|
||||
{ L".*a", L"\"*a" },
|
||||
{ L"a.*a", L"a\"*a" },
|
||||
{ L".*.*", L"\"<\"*" },
|
||||
{ L".*a.*", L"\"*a\"*" },
|
||||
{ L"a.*a.*a", L"a\"*a\"*a" },
|
||||
|
||||
{ L"*.", L"<" },
|
||||
{ L"a*.", L"a<" },
|
||||
{ L"*.a", L"<.a" },
|
||||
{ L"a*.a", L"a<.a" },
|
||||
{ L"*.*.", L"*" },
|
||||
{ L"*.a*.", L"<.a<" },
|
||||
{ L"a*.a*.a", L"a<.a<.a" },
|
||||
|
||||
{ L"?", L">" },
|
||||
{ L"a?", L"a>" },
|
||||
{ L"?a", L">a" },
|
||||
{ L"a?a", L"a>a" },
|
||||
{ L"??", L">>" },
|
||||
{ L"?a?", L">a>" },
|
||||
{ L"a?a?a", L"a>a>a" },
|
||||
|
||||
{ L"f*.", L"f<" },
|
||||
{ L"f.*", L"f\"*" },
|
||||
{ L"f*.*", L"f<\"*" },
|
||||
{ L"f*.f*", L"f<.f*" },
|
||||
{ L"f*f.*", L"f*f\"*" },
|
||||
{ L"f*.*f", L"f<\"*f" },
|
||||
|
||||
/* TODO: add more. Have fun */
|
||||
};
|
||||
const INT TestCount = sizeof(Tests) / sizeof(Tests[0]);
|
||||
INT i;
|
||||
WCHAR ExpressionBuffer[MAX_PATH];
|
||||
|
||||
KmtLoadDriver(L"FindFile", FALSE);
|
||||
KmtOpenDriver();
|
||||
|
||||
for (i = 0; i < TestCount; i++)
|
||||
{
|
||||
trace("[%d] '%ls', '%ls'\n", i, Tests[i].Expression, Tests[i].ExpectedExpression);
|
||||
KmtSendWStringToDriver(IOCTL_EXPECT, Tests[i].ExpectedExpression);
|
||||
wcscpy(ExpressionBuffer, L"\\\\.\\Global\\GLOBALROOT\\Device\\Kmtest-FindFile\\");
|
||||
wcscat(ExpressionBuffer, Tests[i].Expression);
|
||||
FindHandle = FindFirstFileW(ExpressionBuffer, &FindData);
|
||||
ok(FindHandle != NULL && FindHandle != INVALID_HANDLE_VALUE, "Handle: %p, error=%lu\n", (PVOID)FindHandle, GetLastError());
|
||||
if (FindHandle != INVALID_HANDLE_VALUE)
|
||||
FindClose(FindHandle);
|
||||
}
|
||||
|
||||
KmtCloseDriver();
|
||||
KmtUnloadDriver();
|
||||
}
|
||||
|
|
@ -8,6 +8,7 @@
|
|||
#include <kmt_test.h>
|
||||
|
||||
KMT_TESTFUNC Test_Example;
|
||||
KMT_TESTFUNC Test_FindFile;
|
||||
KMT_TESTFUNC Test_IoDeviceObject;
|
||||
KMT_TESTFUNC Test_RtlAvlTree;
|
||||
KMT_TESTFUNC Test_RtlException;
|
||||
|
|
@ -19,6 +20,7 @@ KMT_TESTFUNC Test_RtlUnicodeString;
|
|||
const KMT_TEST TestList[] =
|
||||
{
|
||||
{ "Example", Test_Example },
|
||||
{ "FindFile", Test_FindFile },
|
||||
{ "IoDeviceObject", Test_IoDeviceObject },
|
||||
{ "RtlAvlTree", Test_RtlAvlTree },
|
||||
{ "RtlException", Test_RtlException },
|
||||
|
|
|
|||
Loading…
Reference in a new issue