mirror of
https://github.com/reactos/reactos.git
synced 2025-03-09 17:54:55 +00:00
194 lines
4.1 KiB
C++
194 lines
4.1 KiB
C++
/*++
|
|
|
|
Copyright (c) Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
FxObjectUm.cpp
|
|
|
|
Abstract:
|
|
|
|
User mode implementations of FxObject APIs
|
|
|
|
Author:
|
|
|
|
|
|
Environment:
|
|
|
|
user mode only
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "fxobjectpch.hpp"
|
|
|
|
extern "C" {
|
|
|
|
#if defined(EVENT_TRACING)
|
|
#include "FxObjectUm.tmh"
|
|
#endif
|
|
|
|
}
|
|
|
|
extern "C" {
|
|
|
|
#define INITGUID
|
|
#include <guiddef.h>
|
|
|
|
#include <WdfFileObject_private.h>
|
|
|
|
//
|
|
// Function declarations for the WdfObjectQuery DDIs
|
|
//
|
|
_IRQL_requires_max_(PASSIVE_LEVEL)
|
|
WDFAPI
|
|
NTSTATUS
|
|
WDFEXPORT(WdfFileObjectIncrementProcessKeepAliveCount)(
|
|
_In_
|
|
PWDF_DRIVER_GLOBALS DriverGlobals,
|
|
_In_
|
|
WDFFILEOBJECT FileObject
|
|
);
|
|
|
|
_IRQL_requires_max_(PASSIVE_LEVEL)
|
|
WDFAPI
|
|
NTSTATUS
|
|
WDFEXPORT(WdfFileObjectDecrementProcessKeepAliveCount)(
|
|
_In_
|
|
PWDF_DRIVER_GLOBALS DriverGlobals,
|
|
_In_
|
|
WDFFILEOBJECT FileObject
|
|
);
|
|
|
|
}
|
|
|
|
|
|
extern "C" {
|
|
|
|
_Must_inspect_result_
|
|
NTSTATUS
|
|
FxObject::_ObjectQuery(
|
|
_In_ FxObject* Object,
|
|
_In_ CONST GUID* Guid,
|
|
_In_ ULONG QueryBufferLength,
|
|
_Out_writes_bytes_(QueryBufferLength)
|
|
PVOID QueryBuffer
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Query the object handle for specific information
|
|
|
|
This allows dynamic extensions to DDI's.
|
|
|
|
Currently, it is used to allow test hooks for verification
|
|
which are not available in a production release.
|
|
|
|
Arguments:
|
|
|
|
Object - Object to query
|
|
|
|
Guid - GUID to represent the information/DDI to query for
|
|
|
|
QueryBufferLength - Length of QueryBuffer to return data in
|
|
|
|
QueryBuffer - Pointer to QueryBuffer
|
|
|
|
Returns:
|
|
|
|
NTSTATUS
|
|
|
|
--*/
|
|
|
|
{
|
|
PFX_DRIVER_GLOBALS pFxDriverGlobals = Object->GetDriverGlobals();
|
|
|
|
//
|
|
// Design Note: This interface does not look strongly typed
|
|
// but it is. The GUID defines a specific strongly typed
|
|
// contract for QueryBuffer and QueryBufferLength.
|
|
//
|
|
|
|
#if DBG
|
|
|
|
//
|
|
// These operations are only available on checked builds for deep unit
|
|
// testing, code coverage analysis, and model verification.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//
|
|
|
|
// Add code based on the GUID
|
|
|
|
// IsEqualGUID(guid1, guid2), DEFINE_GUID, INITGUID, inc\wnet\guiddef.h
|
|
#endif
|
|
|
|
if (IsEqualGUID(*Guid, GUID_WDFP_FILEOBJECT_INTERFACE)) {
|
|
|
|
//
|
|
// Check the query buffer size before performing the cast
|
|
//
|
|
const ULONG RequiredBufferLength = sizeof(WDFP_FILEOBJECT_INTERFACE);
|
|
|
|
if (QueryBufferLength < RequiredBufferLength) {
|
|
DoTraceLevelMessage(
|
|
pFxDriverGlobals, TRACE_LEVEL_ERROR, TRACINGDEVICE,
|
|
"Insufficient query buffer size for file object query "
|
|
"Required size %d, %!STATUS!",
|
|
RequiredBufferLength,
|
|
STATUS_BUFFER_TOO_SMALL);
|
|
FxVerifierDbgBreakPoint(pFxDriverGlobals);
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
}
|
|
|
|
if (nullptr == QueryBuffer) {
|
|
DoTraceLevelMessage(
|
|
pFxDriverGlobals, TRACE_LEVEL_ERROR, TRACINGDEVICE,
|
|
"NULL query buffer for file object query, %!STATUS!",
|
|
STATUS_BUFFER_TOO_SMALL);
|
|
FxVerifierDbgBreakPoint(pFxDriverGlobals);
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
PWDFP_FILEOBJECT_INTERFACE FileObjectInterface =
|
|
reinterpret_cast<PWDFP_FILEOBJECT_INTERFACE>(QueryBuffer);
|
|
|
|
//
|
|
// Check the struct version (require an exact match for a private DDI)
|
|
//
|
|
if (FileObjectInterface->Size != RequiredBufferLength) {
|
|
DoTraceLevelMessage(
|
|
pFxDriverGlobals, TRACE_LEVEL_ERROR, TRACINGDEVICE,
|
|
"Wrong struct version provided for file object query, "
|
|
"%!STATUS!",
|
|
STATUS_INVALID_PARAMETER);
|
|
FxVerifierDbgBreakPoint(pFxDriverGlobals);
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
FileObjectInterface->WdfpFileObjectIncrementProcessKeepAliveCount =
|
|
WDFEXPORT(WdfFileObjectIncrementProcessKeepAliveCount);
|
|
FileObjectInterface->WdfpFileObjectDecrementProcessKeepAliveCount =
|
|
WDFEXPORT(WdfFileObjectDecrementProcessKeepAliveCount);
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
return STATUS_NOT_FOUND;
|
|
}
|
|
|
|
} // extern "C"
|