kmtest.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	KMT_STRINGIZE(x)   #x
#define	location(file, line)   do { ErrorFileAndLine = file ":" KMT_STRINGIZE(line); } while (0)
#define	error_value(Error, value)   do { location(__FILE__, __LINE__); Error = value; } while (0)
#define	error(Error)   error_value(Error, GetLastError())
#define	error_goto(Error, label)   do { error(Error); goto label; } while (0)
#define	error_value_goto(Error, value, label)   do { error_value(Error, value); goto label; } while (0)

Functions
DWORD	KmtServiceInit (VOID)
DWORD	KmtServiceCleanup (BOOLEAN IgnoreErrors)
DWORD	KmtCreateService (IN PCWSTR ServiceName, IN PCWSTR ServicePath, IN PCWSTR DisplayName OPTIONAL, OUT SC_HANDLE *ServiceHandle)
DWORD	KmtStartService (IN PCWSTR ServiceName OPTIONAL, IN OUT SC_HANDLE *ServiceHandle)
DWORD	KmtCreateAndStartService (IN PCWSTR ServiceName, IN PCWSTR ServicePath, IN PCWSTR DisplayName OPTIONAL, OUT SC_HANDLE *ServiceHandle, IN BOOLEAN RestartIfRunning)
DWORD	KmtStopService (IN PCWSTR ServiceName OPTIONAL, IN OUT SC_HANDLE *ServiceHandle)
DWORD	KmtDeleteService (IN PCWSTR ServiceName OPTIONAL, IN OUT SC_HANDLE *ServiceHandle)
DWORD	KmtCloseService (IN OUT SC_HANDLE *ServiceHandle)

Variables
PCSTR	ErrorFileAndLine

Macro Definition Documentation

error

#define error

(

Error

)

error_value(Error, GetLastError())

Definition at line 20 of file kmtest.h.

error_goto

#define error_goto	(		Error,
			label
	)		do { error(Error); goto label; } while (0)

Definition at line 21 of file kmtest.h.

error_value

#define error_value	(		Error,
			value
	)		do { location(__FILE__, __LINE__); Error = value; } while (0)

Definition at line 19 of file kmtest.h.

error_value_goto

#define error_value_goto	(		Error,
			value,
			label
	)		do { error_value(Error, value); goto label; } while (0)

Definition at line 22 of file kmtest.h.

KMT_STRINGIZE

#define KMT_STRINGIZE

(

x

)

#x

Definition at line 15 of file kmtest.h.

location

#define location	(		file,
			line
	)		do { ErrorFileAndLine = file ":" KMT_STRINGIZE(line); } while (0)

Definition at line 18 of file kmtest.h.

Function Documentation

KmtCloseService()

DWORD KmtCloseService

(

IN OUT SC_HANDLE *

ServiceHandle

)

Definition at line 392 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
 
    assert(ServiceHandle);
 
    if (*ServiceHandle && !CloseServiceHandle(*ServiceHandle))
        error_goto(Error, cleanup);
 
    *ServiceHandle = NULL;
 
cleanup:
    return Error;
}

Referenced by KmtFltCloseService(), and main().

KmtCreateAndStartService()

DWORD KmtCreateAndStartService	(	IN PCWSTR	ServiceName,
		IN PCWSTR	ServicePath,
		IN PCWSTR DisplayName	OPTIONAL,
		OUT SC_HANDLE *	ServiceHandle,
		IN BOOLEAN	RestartIfRunning
	)

Definition at line 262 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
 
    assert(ServiceHandle);
 
    Error = KmtCreateService(ServiceName, ServicePath, DisplayName, ServiceHandle);
 
    if (Error && Error != ERROR_SERVICE_EXISTS)
        goto cleanup;
 
    Error = KmtStartService(ServiceName, ServiceHandle);
 
    if (Error != ERROR_SERVICE_ALREADY_RUNNING)
        goto cleanup;
 
    Error = ERROR_SUCCESS;
 
    if (!RestartIfRunning)
        goto cleanup;
 
    Error = KmtStopService(ServiceName, ServiceHandle);
    if (Error)
        goto cleanup;
 
    Error = KmtStartService(ServiceName, ServiceHandle);
    if (Error)
        goto cleanup;
 
cleanup:
    assert(Error || *ServiceHandle);
    return Error;
}

Referenced by KmtLoadDriver(), and main().

KmtCreateService()

DWORD KmtCreateService	(	IN PCWSTR	ServiceName,
		IN PCWSTR	ServicePath,
		IN PCWSTR DisplayName	OPTIONAL,
		OUT SC_HANDLE *	ServiceHandle
	)

Definition at line 92 of file service.c.

{
    return KmtpCreateService(ServiceName,
                             ServicePath,
                             DisplayName,
                             SERVICE_KERNEL_DRIVER,
                             ServiceHandle);
}

Referenced by KmtCreateAndStartService(), and main().

KmtDeleteService()

DWORD KmtDeleteService	(	IN PCWSTR ServiceName	OPTIONAL,
		IN OUT SC_HANDLE *	ServiceHandle
	)

Definition at line 356 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
 
    assert(ServiceHandle);
    assert(ServiceName || *ServiceHandle);
 
    if (!*ServiceHandle)
        *ServiceHandle = OpenService(ScmHandle, ServiceName, SERVICE_ACCESS);
 
    if (!*ServiceHandle)
        error_goto(Error, cleanup);
 
    if (!DeleteService(*ServiceHandle))
        error_goto(Error, cleanup);
 
    if (*ServiceHandle)
        CloseServiceHandle(*ServiceHandle);
 
cleanup:
    return Error;
}

Referenced by KmtFltDeleteService(), KmtUnloadDriver(), and main().

KmtServiceCleanup()

DWORD KmtServiceCleanup

(

BOOLEAN

IgnoreErrors

)

Definition at line 64 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
 
    if (ScmHandle && !CloseServiceHandle(ScmHandle) && !IgnoreErrors)
        error(Error);
 
    return Error;
}

Referenced by main().

KmtServiceInit()

DWORD KmtServiceInit

(

VOID

)

Definition at line 40 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
 
    assert(!ScmHandle);
 
    ScmHandle = OpenSCManager(NULL, NULL, SC_MANAGER_CREATE_SERVICE);
    if (!ScmHandle)
        error(Error);
 
    return Error;
}

Referenced by main().

KmtStartService()

DWORD KmtStartService	(	IN PCWSTR ServiceName	OPTIONAL,
		IN OUT SC_HANDLE *	ServiceHandle
	)

Definition at line 217 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
 
    assert(ServiceHandle);
    assert(ServiceName || *ServiceHandle);
 
    if (!*ServiceHandle)
        *ServiceHandle = OpenService(ScmHandle, ServiceName, SERVICE_ACCESS);
 
    if (!*ServiceHandle)
        error_goto(Error, cleanup);
 
    if (!StartService(*ServiceHandle, 0, NULL))
        error_goto(Error, cleanup);
 
    Error = KmtEnsureServiceState(ServiceName, *ServiceHandle, SERVICE_RUNNING);
    if (Error)
        goto cleanup;
 
cleanup:
    return Error;
}

Referenced by KmtCreateAndStartService().

KmtStopService()

DWORD KmtStopService	(	IN PCWSTR ServiceName	OPTIONAL,
		IN OUT SC_HANDLE *	ServiceHandle
	)

Definition at line 315 of file service.c.

{
    DWORD Error = ERROR_SUCCESS;
    SERVICE_STATUS ServiceStatus;
 
    assert(ServiceHandle);
    assert(ServiceName || *ServiceHandle);
 
    if (!*ServiceHandle)
        *ServiceHandle = OpenService(ScmHandle, ServiceName, SERVICE_ACCESS);
 
    if (!*ServiceHandle)
        error_goto(Error, cleanup);
 
    if (!ControlService(*ServiceHandle, SERVICE_CONTROL_STOP, &ServiceStatus))
        error_goto(Error, cleanup);
 
    Error = KmtEnsureServiceState(ServiceName, *ServiceHandle, SERVICE_STOPPED);
    if (Error)
        goto cleanup;
 
cleanup:
    return Error;
}

Referenced by KmtCreateAndStartService(), KmtUnloadDriver(), KmtUnloadDriverKeepService(), and main().

Variable Documentation

ErrorFileAndLine

PCSTR ErrorFileAndLine

extern

Definition at line 34 of file kmtest.c.

Referenced by OutputError().