[jgardou] 65546: [KMTEST] - Add some tests for tcpip.sys TDI capabilities (for now, TDI_PROVIDER_INFO)
Author: jgardou Date: Tue Dec 2 18:06:48 2014 New Revision: 65546
URL: http://svn.reactos.org/svn/reactos?rev=65546&view=rev Log: [KMTEST] - Add some tests for tcpip.sys TDI capabilities (for now, TDI_PROVIDER_INFO)
Added: trunk/rostests/kmtests/tcpip/ trunk/rostests/kmtests/tcpip/CMakeLists.txt (with props) trunk/rostests/kmtests/tcpip/TcpIp_drv.c (with props) trunk/rostests/kmtests/tcpip/TcpIp_user.c (with props) trunk/rostests/kmtests/tcpip/tcpip.h (with props) trunk/rostests/kmtests/tcpip/tdi.c (with props) Modified: trunk/rostests/kmtests/CMakeLists.txt trunk/rostests/kmtests/kmtest/testlist.c
Modified: trunk/rostests/kmtests/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/CMakeLists.txt?rev... ============================================================================== --- trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] (original) +++ trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -7,6 +7,7 @@ add_subdirectory(example) add_subdirectory(kernel32) add_subdirectory(ntos_io) +add_subdirectory(tcpip)
list(APPEND COMMON_SOURCE example/GuardedMemory.c @@ -108,6 +109,7 @@ example/Example_user.c kernel32/FindFile_user.c ntos_io/IoDeviceObject_user.c + tcpip/TcpIp_user.c ${COMMON_SOURCE}
kmtest/kmtest.rc) @@ -129,7 +131,8 @@ kmtest_drv example_drv iodeviceobject_drv - iohelper_drv) + iohelper_drv + tcpip_test_drv)
add_custom_target(kmtest_all) add_dependencies(kmtest_all kmtest_drivers kmtest)
Modified: trunk/rostests/kmtests/kmtest/testlist.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/kmtest/testlist.c?... ============================================================================== --- trunk/rostests/kmtests/kmtest/testlist.c [iso-8859-1] (original) +++ trunk/rostests/kmtests/kmtest/testlist.c [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -17,6 +17,8 @@ KMT_TESTFUNC Test_RtlRegistry; KMT_TESTFUNC Test_RtlSplayTree; KMT_TESTFUNC Test_RtlUnicodeString; +KMT_TESTFUNC Test_TcpIpIoctl; +KMT_TESTFUNC Test_TcpIpTdi;
/* tests with a leading '-' will not be listed */ const KMT_TEST TestList[] = @@ -31,5 +33,6 @@ { "RtlRegistry", Test_RtlRegistry }, { "RtlSplayTree", Test_RtlSplayTree }, { "RtlUnicodeString", Test_RtlUnicodeString }, + { "TcpIpTdi", Test_TcpIpTdi }, { NULL, NULL }, };
Added: trunk/rostests/kmtests/tcpip/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/tcpip/CMakeLists.t... ============================================================================== --- trunk/rostests/kmtests/tcpip/CMakeLists.txt (added) +++ trunk/rostests/kmtests/tcpip/CMakeLists.txt [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -0,0 +1,15 @@ + +include_directories(../include) + +list(APPEND TCPIP_TEST_DRV_SOURCE + ../kmtest_drv/kmtest_standalone.c + tdi.c + TcpIp_drv.c) + +add_library(tcpip_drv SHARED ${TCPIP_TEST_DRV_SOURCE}) +set_module_type(tcpip_drv kernelmodedriver) +target_link_libraries(tcpip_drv kmtest_printf ${PSEH_LIB}) +add_importlibs(tcpip_drv ntoskrnl hal) +add_target_compile_definitions(tcpip_drv KMT_STANDALONE_DRIVER) +#add_pch(example_drv ../include/kmt_test.h) +add_cd_file(TARGET tcpip_drv DESTINATION reactos/bin FOR all)
Propchange: trunk/rostests/kmtests/tcpip/CMakeLists.txt ------------------------------------------------------------------------------ svn:eol-style = native
Added: trunk/rostests/kmtests/tcpip/TcpIp_drv.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/tcpip/TcpIp_drv.c?... ============================================================================== --- trunk/rostests/kmtests/tcpip/TcpIp_drv.c (added) +++ trunk/rostests/kmtests/tcpip/TcpIp_drv.c [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -0,0 +1,55 @@ +/* + * PROJECT: ReactOS kernel-mode tests + * LICENSE: GPLv2+ - See COPYING in the top level directory + * PURPOSE: Kernel-Mode Test Suite for tcpip.sys + * PROGRAMMER: Jérôme Gardou jerome.gardou@reactos.org + */ + +#include <kmt_test.h> +#include "tcpip.h" + +extern KMT_MESSAGE_HANDLER TestTdi; + +static struct +{ + ULONG ControlCode; + PKMT_MESSAGE_HANDLER Handler; +} MessageHandlers[] = +{ + { IOCTL_TEST_TDI, TestTdi }, +}; + +NTSTATUS +TestEntry( + _In_ PDRIVER_OBJECT DriverObject, + _In_ PCUNICODE_STRING RegistryPath, + _Out_ PCWSTR *DeviceName, + _Inout_ INT *Flags) +{ + ULONG i; + + PAGED_CODE(); + + UNREFERENCED_PARAMETER(DriverObject); + UNREFERENCED_PARAMETER(RegistryPath); + UNREFERENCED_PARAMETER(Flags); + + *DeviceName = L"TcpIp"; + + for (i = 0; i < (sizeof(MessageHandlers) / sizeof(MessageHandlers[0])); i++) + KmtRegisterMessageHandler(MessageHandlers[i].ControlCode, NULL, MessageHandlers[i].Handler); + + trace("TcpIp test driver loaded.\n"); + + return STATUS_SUCCESS; +} + +VOID +TestUnload( + _In_ PDRIVER_OBJECT DriverObject) +{ + PAGED_CODE(); + + UNREFERENCED_PARAMETER(DriverObject); +} +
Propchange: trunk/rostests/kmtests/tcpip/TcpIp_drv.c ------------------------------------------------------------------------------ charset = UTF-8
Propchange: trunk/rostests/kmtests/tcpip/TcpIp_drv.c ------------------------------------------------------------------------------ svn:eol-style = native
Propchange: trunk/rostests/kmtests/tcpip/TcpIp_drv.c ------------------------------------------------------------------------------ svn:mime-type = text/plain
Added: trunk/rostests/kmtests/tcpip/TcpIp_user.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/tcpip/TcpIp_user.c... ============================================================================== --- trunk/rostests/kmtests/tcpip/TcpIp_user.c (added) +++ trunk/rostests/kmtests/tcpip/TcpIp_user.c [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -0,0 +1,37 @@ +/* + * PROJECT: ReactOS kernel-mode tests + * LICENSE: GPLv2+ - See COPYING in the top level directory + * PURPOSE: User mode part of the TcpIp.sys test suite + * PROGRAMMER: Jérôme Gardou jerome.gardou@reactos.org + */ + +#include <kmt_test.h> + +#include "tcpip.h" + +static +void +LoadTcpIpTestDriver(void) +{ + /* Start the special-purpose driver */ + KmtLoadDriver(L"TcpIp", FALSE); + KmtOpenDriver(); +} + +static +void +UnloadTcpIpTestDriver(void) +{ + /* Stop the driver. */ + KmtCloseDriver(); + KmtUnloadDriver(); +} + +START_TEST(TcpIpTdi) +{ + LoadTcpIpTestDriver(); + + ok(KmtSendToDriver(IOCTL_TEST_TDI) == ERROR_SUCCESS, "\n"); + + UnloadTcpIpTestDriver(); +}
Propchange: trunk/rostests/kmtests/tcpip/TcpIp_user.c ------------------------------------------------------------------------------ charset = UTF-8
Propchange: trunk/rostests/kmtests/tcpip/TcpIp_user.c ------------------------------------------------------------------------------ svn:eol-style = native
Propchange: trunk/rostests/kmtests/tcpip/TcpIp_user.c ------------------------------------------------------------------------------ svn:mime-type = text/plain
Added: trunk/rostests/kmtests/tcpip/tcpip.h URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/tcpip/tcpip.h?rev=... ============================================================================== --- trunk/rostests/kmtests/tcpip/tcpip.h (added) +++ trunk/rostests/kmtests/tcpip/tcpip.h [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -0,0 +1,2 @@ + +#define IOCTL_TEST_TDI 1
Propchange: trunk/rostests/kmtests/tcpip/tcpip.h ------------------------------------------------------------------------------ charset = UTF-8
Propchange: trunk/rostests/kmtests/tcpip/tcpip.h ------------------------------------------------------------------------------ svn:eol-style = native
Propchange: trunk/rostests/kmtests/tcpip/tcpip.h ------------------------------------------------------------------------------ svn:mime-type = text/plain
Added: trunk/rostests/kmtests/tcpip/tdi.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/tcpip/tdi.c?rev=65... ============================================================================== --- trunk/rostests/kmtests/tcpip/tdi.c (added) +++ trunk/rostests/kmtests/tcpip/tdi.c [iso-8859-1] Tue Dec 2 18:06:48 2014 @@ -0,0 +1,291 @@ +/* + * PROJECT: ReactOS kernel-mode tests + * LICENSE: LGPLv2+ - See COPYING.LIB in the top level directory + * PURPOSE: Kernel-Mode Test Suite for TCPIP.sys + * PROGRAMMER: Jérôme Gardou jerome.gardou@reactos.org + */ + +#include <kmt_test.h> +#include <tdikrnl.h> + +static +NTAPI +NTSTATUS +IrpCompletionRoutine( + _In_ PDEVICE_OBJECT DeviceObject, + _In_ PIRP Irp, + _In_ PVOID Context) +{ + UNREFERENCED_PARAMETER(DeviceObject); + UNREFERENCED_PARAMETER(Irp); + + KeSetEvent((PKEVENT)Context, IO_NETWORK_INCREMENT, FALSE); + + return STATUS_MORE_PROCESSING_REQUIRED; +} + +static +VOID +TestProviderInfo(void) +{ + struct + { + UNICODE_STRING DeviceName; + NTSTATUS CreateStatus, IrpStatus; + TDI_PROVIDER_INFO ExpectedInfo; + } TestData[] = + { + { + RTL_CONSTANT_STRING(L"\Device\Tcp"), + STATUS_SUCCESS, STATUS_SUCCESS, + { + 0x0002, // Version + 0x3FFFFFFF, // MaxSendSize + 0, // MaxConnectionUserData + 65515, // MaxDatagramSize + TDI_SERVICE_CONNECTION_MODE | + TDI_SERVICE_ORDERLY_RELEASE | + TDI_SERVICE_CONNECTIONLESS_MODE | + TDI_SERVICE_ERROR_FREE_DELIVERY | + TDI_SERVICE_BROADCAST_SUPPORTED | + TDI_SERVICE_DELAYED_ACCEPTANCE | + TDI_SERVICE_EXPEDITED_DATA | + TDI_SERVICE_NO_ZERO_LENGTH | + TDI_SERVICE_DGRAM_CONNECTION | + TDI_SERVICE_FORCE_ACCESS_CHECK | + TDI_SERVICE_DIRECT_ACCEPT | + TDI_SERVICE_ADDRESS_SECURITY | + TDI_SERVICE_NO_PUSH, // ServiceFlags + 1, // MinimumLookaheadData + 65535, // MaximumLookaheadData + 0, // NumberOfResources + {{0}} // StartTime + } + }, + { + RTL_CONSTANT_STRING(L"\Device\Udp"), + STATUS_SUCCESS, STATUS_SUCCESS, + { + 0x0002, // Version + 0x3FFFFFFF, // MaxSendSize + 0, // MaxConnectionUserData + 65507, // MaxDatagramSize + TDI_SERVICE_CONNECTION_MODE | + TDI_SERVICE_ORDERLY_RELEASE | + TDI_SERVICE_CONNECTIONLESS_MODE | + TDI_SERVICE_ERROR_FREE_DELIVERY | + TDI_SERVICE_BROADCAST_SUPPORTED | + TDI_SERVICE_DELAYED_ACCEPTANCE | + TDI_SERVICE_EXPEDITED_DATA | + TDI_SERVICE_NO_ZERO_LENGTH | + TDI_SERVICE_DGRAM_CONNECTION | + TDI_SERVICE_FORCE_ACCESS_CHECK | + TDI_SERVICE_DIRECT_ACCEPT | + TDI_SERVICE_ADDRESS_SECURITY, // ServiceFlags + 1, // MinimumLookaheadData + 65535, // MaximumLookaheadData + 0, // NumberOfResources + {{0}} // StartTime + } + }, + { + RTL_CONSTANT_STRING(L"\Device\Ip"), + STATUS_SUCCESS, STATUS_NOT_IMPLEMENTED, + }, + { + RTL_CONSTANT_STRING(L"\Device\RawIp"), + STATUS_SUCCESS, STATUS_SUCCESS, + { + 0x0002, // Version + 0x3FFFFFFF, // MaxSendSize + 0, // MaxConnectionUserData + 65515, // MaxDatagramSize + TDI_SERVICE_CONNECTION_MODE | + TDI_SERVICE_ORDERLY_RELEASE | + TDI_SERVICE_CONNECTIONLESS_MODE | + TDI_SERVICE_ERROR_FREE_DELIVERY | + TDI_SERVICE_BROADCAST_SUPPORTED | + TDI_SERVICE_DELAYED_ACCEPTANCE | + TDI_SERVICE_EXPEDITED_DATA | + TDI_SERVICE_NO_ZERO_LENGTH | + TDI_SERVICE_DGRAM_CONNECTION | + TDI_SERVICE_FORCE_ACCESS_CHECK | + TDI_SERVICE_DIRECT_ACCEPT | + TDI_SERVICE_ADDRESS_SECURITY, // ServiceFlags + 1, // MinimumLookaheadData + 65535, // MaximumLookaheadData + 0, // NumberOfResources + {{0}} // StartTime + } + }, + { + RTL_CONSTANT_STRING(L"\Device\IPMULTICAST"), + STATUS_OBJECT_NAME_NOT_FOUND, + }, + }; + ULONG i; + + for (i = 0; i < (sizeof(TestData) / sizeof(TestData[0])); i++) + { + IO_STATUS_BLOCK StatusBlock; + NTSTATUS Status; + FILE_OBJECT* FileObject; + DEVICE_OBJECT* DeviceObject; + PIRP Irp; + KEVENT Event; + PMDL Mdl; + TDI_PROVIDER_INFO* ProviderInfo; + HANDLE FileHandle; + OBJECT_ATTRIBUTES ObjectAttributes; + + trace("Testing device %wZ\n", &TestData[i].DeviceName); + + InitializeObjectAttributes( + &ObjectAttributes, + &TestData[i].DeviceName, + OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE, + NULL, + NULL); + + Status = ZwCreateFile( + &FileHandle, + FILE_READ_DATA | FILE_WRITE_DATA, + &ObjectAttributes, + &StatusBlock, + NULL, + FILE_ATTRIBUTE_NORMAL, + FILE_SHARE_READ | FILE_SHARE_WRITE, + FILE_OPEN, + 0, + NULL, + 0); + ok_eq_hex(Status, TestData[i].CreateStatus); + if (!NT_SUCCESS(Status)) + continue; + + Status = ObReferenceObjectByHandle( + FileHandle, + GENERIC_READ, + *IoFileObjectType, + KernelMode, + (PVOID*)&FileObject, + NULL); + ok_eq_hex(Status, STATUS_SUCCESS); + if (!NT_SUCCESS(Status)) + return; + + DeviceObject = IoGetRelatedDeviceObject(FileObject); + ok(DeviceObject != NULL, "Device object is NULL!\n"); + if (!DeviceObject) + { + ObDereferenceObject(FileObject); + return; + } + + ProviderInfo = ExAllocatePoolWithTag(NonPagedPool, sizeof(*ProviderInfo), 'tseT'); + ok(ProviderInfo != NULL, "Ran out of memory.\n"); + if (!ProviderInfo) + { + ObDereferenceObject(FileObject); + return; + } + + Mdl = IoAllocateMdl(ProviderInfo, sizeof(*ProviderInfo), FALSE, FALSE, NULL); + ok(Mdl != NULL, "Could not allocate the MDL!\n"); + if (!Mdl) + { + ExFreePoolWithTag(ProviderInfo, 'tseT'); + ObDereferenceObject(FileObject); + return; + } + + MmBuildMdlForNonPagedPool(Mdl); + + /* Build the IRP */ + KeInitializeEvent(&Event, NotificationEvent, FALSE); + Irp = IoAllocateIrp(DeviceObject->StackSize, FALSE); + ok(Irp != NULL, "TdiBuildInternalDeviceControlIrp returned NULL!\n"); + if (!Irp) + { + IoFreeMdl(Mdl); + ExFreePoolWithTag(ProviderInfo, 'tseT'); + ObDereferenceObject(FileObject); + return; + } + + TdiBuildQueryInformation( + Irp, + DeviceObject, + FileObject, + NULL, + NULL, + TDI_QUERY_PROVIDER_INFO, + Mdl); + + IoSetCompletionRoutine(Irp, IrpCompletionRoutine, &Event, TRUE, TRUE, TRUE); + + Status = IoCallDriver(DeviceObject, Irp); + if (Status == STATUS_PENDING) + { + KeWaitForSingleObject( + &Event, + Executive, + KernelMode, + FALSE, + NULL); + Status = StatusBlock.Status; + } + ok_eq_hex(Status, TestData[i].IrpStatus); + + IoFreeIrp(Irp); + IoFreeMdl(Mdl); + ObDereferenceObject(FileObject); + + if (!NT_SUCCESS(Status)) + { + ExFreePoolWithTag(ProviderInfo, 'tseT'); + continue; + } + + ok_eq_hex(ProviderInfo->Version, TestData[i].ExpectedInfo.Version); + ok_eq_ulong(ProviderInfo->MaxSendSize, TestData[i].ExpectedInfo.MaxSendSize); + ok_eq_ulong(ProviderInfo->MaxConnectionUserData, TestData[i].ExpectedInfo.MaxConnectionUserData); + ok_eq_ulong(ProviderInfo->MaxDatagramSize, TestData[i].ExpectedInfo.MaxDatagramSize); + ok_eq_hex(ProviderInfo->ServiceFlags, TestData[i].ExpectedInfo.ServiceFlags); + ok_eq_ulong(ProviderInfo->MinimumLookaheadData, TestData[i].ExpectedInfo.MinimumLookaheadData); + ok_eq_ulong(ProviderInfo->MaximumLookaheadData, TestData[i].ExpectedInfo.MaximumLookaheadData); + ok_eq_ulong(ProviderInfo->NumberOfResources, TestData[i].ExpectedInfo.NumberOfResources); + + ExFreePoolWithTag(ProviderInfo, 'tseT'); + } +} + +static KSTART_ROUTINE RunTest; +static +VOID +NTAPI +RunTest( + _In_ PVOID Context) +{ + UNREFERENCED_PARAMETER(Context); + + TestProviderInfo(); +} + +KMT_MESSAGE_HANDLER TestTdi; +NTSTATUS +TestTdi( + _In_ PDEVICE_OBJECT DeviceObject, + _In_ ULONG ControlCode, + _In_opt_ PVOID Buffer, + _In_ SIZE_T InLength, + _Inout_ PSIZE_T OutLength +) +{ + PKTHREAD Thread; + + Thread = KmtStartThread(RunTest, NULL); + KmtFinishThread(Thread, NULL); + + return STATUS_SUCCESS; +}
Propchange: trunk/rostests/kmtests/tcpip/tdi.c ------------------------------------------------------------------------------ charset = UTF-8
Propchange: trunk/rostests/kmtests/tcpip/tdi.c ------------------------------------------------------------------------------ svn:eol-style = native
Propchange: trunk/rostests/kmtests/tcpip/tdi.c ------------------------------------------------------------------------------ svn:mime-type = text/plain
-
jgardou＠svn.reactos.org