mirror of
https://github.com/reactos/reactos.git
synced 2024-08-27 13:48:42 +00:00
![Thomas Faber](/assets/img/avatar_default.png)
- Add KmtSendWStringToDriver to easily send unicode strings to special-purpose drivers - Add TESTENTRY_NO_EXCLUSIVE_DEVICE flag for non-exclusive devices in special-purpose drivers ROSTESTS-104 svn path=/trunk/; revision=58812
240 lines
5.1 KiB
C
240 lines
5.1 KiB
C
/*
|
|
* PROJECT: ReactOS kernel-mode tests
|
|
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
|
* PURPOSE: Kernel-Mode Test Suite Driver
|
|
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
|
|
*/
|
|
|
|
#include <kmt_test.h>
|
|
|
|
#include "kmtest.h"
|
|
#include <kmt_public.h>
|
|
|
|
#include <assert.h>
|
|
|
|
extern HANDLE KmtestHandle;
|
|
|
|
/**
|
|
* @name KmtRunKernelTest
|
|
*
|
|
* Run the specified kernel-mode test part
|
|
*
|
|
* @param TestName
|
|
* Name of the test to run
|
|
*
|
|
* @return Win32 error code as returned by DeviceIoControl
|
|
*/
|
|
DWORD
|
|
KmtRunKernelTest(
|
|
IN PCSTR TestName)
|
|
{
|
|
DWORD Error = ERROR_SUCCESS;
|
|
DWORD BytesRead;
|
|
|
|
if (!DeviceIoControl(KmtestHandle, IOCTL_KMTEST_RUN_TEST, (PVOID)TestName, (DWORD)strlen(TestName), NULL, 0, &BytesRead, NULL))
|
|
error(Error);
|
|
|
|
return Error;
|
|
}
|
|
|
|
static WCHAR TestServiceName[MAX_PATH];
|
|
static SC_HANDLE TestServiceHandle;
|
|
static HANDLE TestDeviceHandle;
|
|
|
|
/**
|
|
* @name KmtLoadDriver
|
|
*
|
|
* Load the specified special-purpose driver (create/start the service)
|
|
*
|
|
* @param ServiceName
|
|
* Name of the driver service (Kmtest- prefix will be added automatically)
|
|
* @param RestartIfRunning
|
|
* TRUE to stop and restart the service if it is already running
|
|
*/
|
|
VOID
|
|
KmtLoadDriver(
|
|
IN PCWSTR ServiceName,
|
|
IN BOOLEAN RestartIfRunning)
|
|
{
|
|
DWORD Error = ERROR_SUCCESS;
|
|
WCHAR ServicePath[MAX_PATH];
|
|
|
|
StringCbCopy(ServicePath, sizeof ServicePath, ServiceName);
|
|
StringCbCat(ServicePath, sizeof ServicePath, L"_drv.sys");
|
|
|
|
StringCbCopy(TestServiceName, sizeof TestServiceName, L"Kmtest-");
|
|
StringCbCat(TestServiceName, sizeof TestServiceName, ServiceName);
|
|
|
|
Error = KmtCreateAndStartService(TestServiceName, ServicePath, NULL, &TestServiceHandle, RestartIfRunning);
|
|
|
|
if (Error)
|
|
{
|
|
// TODO
|
|
__debugbreak();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @name KmtUnloadDriver
|
|
*
|
|
* Unload special-purpose driver (stop the service)
|
|
*/
|
|
VOID
|
|
KmtUnloadDriver(VOID)
|
|
{
|
|
DWORD Error = ERROR_SUCCESS;
|
|
|
|
Error = KmtStopService(TestServiceName, &TestServiceHandle);
|
|
|
|
if (Error)
|
|
{
|
|
// TODO
|
|
__debugbreak();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @name KmtOpenDriver
|
|
*
|
|
* Open special-purpose driver (acquire a device handle)
|
|
*/
|
|
VOID
|
|
KmtOpenDriver(VOID)
|
|
{
|
|
DWORD Error = ERROR_SUCCESS;
|
|
WCHAR DevicePath[MAX_PATH];
|
|
|
|
StringCbCopy(DevicePath, sizeof DevicePath, L"\\\\.\\Global\\GLOBALROOT\\Device\\");
|
|
StringCbCat(DevicePath, sizeof DevicePath, TestServiceName);
|
|
|
|
TestDeviceHandle = CreateFile(DevicePath, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
|
|
if (TestDeviceHandle == INVALID_HANDLE_VALUE)
|
|
error(Error);
|
|
|
|
if (Error)
|
|
{
|
|
// TODO
|
|
__debugbreak();
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* @name KmtCloseDriver
|
|
*
|
|
* Close special-purpose driver (close device handle)
|
|
*/
|
|
VOID
|
|
KmtCloseDriver(VOID)
|
|
{
|
|
DWORD Error = ERROR_SUCCESS;
|
|
|
|
if (TestDeviceHandle && !CloseHandle(TestDeviceHandle))
|
|
error(Error);
|
|
|
|
if (Error)
|
|
{
|
|
// TODO
|
|
__debugbreak();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @name KmtSendToDriver
|
|
*
|
|
* Unload special-purpose driver (stop the service)
|
|
*
|
|
* @param ControlCode
|
|
*
|
|
* @return Win32 error code as returned by DeviceIoControl
|
|
*/
|
|
DWORD
|
|
KmtSendToDriver(
|
|
IN DWORD ControlCode)
|
|
{
|
|
DWORD BytesRead;
|
|
|
|
assert(ControlCode < 0x400);
|
|
|
|
if (!DeviceIoControl(TestDeviceHandle, KMT_MAKE_CODE(ControlCode), NULL, 0, NULL, 0, &BytesRead, NULL))
|
|
return GetLastError();
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
* @name KmtSendStringToDriver
|
|
*
|
|
* Unload special-purpose driver (stop the service)
|
|
*
|
|
* @param ControlCode
|
|
* @param String
|
|
*
|
|
* @return Win32 error code as returned by DeviceIoControl
|
|
*/
|
|
DWORD
|
|
KmtSendStringToDriver(
|
|
IN DWORD ControlCode,
|
|
IN PCSTR String)
|
|
{
|
|
DWORD BytesRead;
|
|
|
|
assert(ControlCode < 0x400);
|
|
|
|
if (!DeviceIoControl(TestDeviceHandle, KMT_MAKE_CODE(ControlCode), (PVOID)String, (DWORD)strlen(String), NULL, 0, &BytesRead, NULL))
|
|
return GetLastError();
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
* @name KmtSendStringToDriver
|
|
*
|
|
* Unload special-purpose driver (stop the service)
|
|
*
|
|
* @param ControlCode
|
|
* @param String
|
|
*
|
|
* @return Win32 error code as returned by DeviceIoControl
|
|
*/
|
|
DWORD
|
|
KmtSendWStringToDriver(
|
|
IN DWORD ControlCode,
|
|
IN PCWSTR String)
|
|
{
|
|
DWORD BytesRead;
|
|
|
|
assert(ControlCode < 0x400);
|
|
|
|
if (!DeviceIoControl(TestDeviceHandle, KMT_MAKE_CODE(ControlCode), (PVOID)String, (DWORD)wcslen(String) * sizeof(WCHAR), NULL, 0, &BytesRead, NULL))
|
|
return GetLastError();
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
* @name KmtSendBufferToDriver
|
|
*
|
|
* @param ControlCode
|
|
* @param Buffer
|
|
* @param InLength
|
|
* @param OutLength
|
|
*
|
|
* @return Win32 error code as returned by DeviceIoControl
|
|
*/
|
|
DWORD
|
|
KmtSendBufferToDriver(
|
|
IN DWORD ControlCode,
|
|
IN OUT PVOID Buffer OPTIONAL,
|
|
IN DWORD InLength,
|
|
IN OUT PDWORD OutLength)
|
|
{
|
|
assert(OutLength);
|
|
assert(Buffer || (!InLength && !*OutLength));
|
|
assert(ControlCode < 0x400);
|
|
|
|
if (!DeviceIoControl(TestDeviceHandle, KMT_MAKE_CODE(ControlCode), Buffer, InLength, Buffer, *OutLength, OutLength, NULL))
|
|
return GetLastError();
|
|
|
|
return ERROR_SUCCESS;
|
|
} |