285 lines
8.4 KiB
C
285 lines
8.4 KiB
C
/*++
|
||
|
||
Copyright (C) Microsoft Corporation, 1998 - 1998
|
||
|
||
Module Name:
|
||
|
||
wmi.c
|
||
|
||
Abstract:
|
||
|
||
This module contains the code that handles the wmi IRPs for
|
||
parallel driver PDOs and PODOs.
|
||
|
||
Environment:
|
||
|
||
Kernel mode
|
||
|
||
Revision History :
|
||
--*/
|
||
|
||
#include "pch.h"
|
||
#include <wmistr.h>
|
||
#include "wmipdo.h"
|
||
|
||
#ifdef ALLOC_PRAGMA
|
||
#pragma alloc_text(PAGEPARWMI0, ParWmiPdoInitWmi)
|
||
#pragma alloc_text(PAGEPARWMI0, ParWmiPdoSystemControlDispatch)
|
||
#pragma alloc_text(PAGEPARWMI0, ParWmiPdoQueryWmiRegInfo)
|
||
#pragma alloc_text(PAGEPARWMI0, ParWmiPdoQueryWmiDataBlock)
|
||
#endif
|
||
|
||
#define PAR_WMI_PDO_GUID_COUNT 1
|
||
#define PAR_WMI_BYTES_TRANSFERRED_GUID_INDEX 0
|
||
|
||
GUID ParWmiPdoBytesTransferredGuid = PARALLEL_WMI_BYTES_TRANSFERRED_GUID;
|
||
|
||
WMIGUIDREGINFO ParWmiPdoGuidList[ PAR_WMI_PDO_GUID_COUNT ] =
|
||
{
|
||
{ &ParWmiPdoBytesTransferredGuid, 1, 0 }
|
||
};
|
||
|
||
|
||
NTSTATUS
|
||
ParWmiPdoInitWmi(PDEVICE_OBJECT DeviceObject)
|
||
{
|
||
PDEVICE_EXTENSION devExt = DeviceObject->DeviceExtension;
|
||
PWMILIB_CONTEXT wmiContext = &devExt->WmiLibContext;
|
||
|
||
PAGED_CODE();
|
||
|
||
wmiContext->GuidCount = sizeof(ParWmiPdoGuidList) / sizeof(WMIGUIDREGINFO);
|
||
wmiContext->GuidList = ParWmiPdoGuidList;
|
||
|
||
wmiContext->QueryWmiRegInfo = ParWmiPdoQueryWmiRegInfo;
|
||
wmiContext->QueryWmiDataBlock = ParWmiPdoQueryWmiDataBlock;
|
||
wmiContext->SetWmiDataBlock = NULL;
|
||
wmiContext->SetWmiDataItem = NULL;
|
||
wmiContext->ExecuteWmiMethod = NULL;
|
||
wmiContext->WmiFunctionControl = NULL;
|
||
|
||
return ParWMIRegistrationControl( DeviceObject, WMIREG_ACTION_REGISTER );
|
||
}
|
||
|
||
NTSTATUS
|
||
ParWmiPdoSystemControlDispatch(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
|
||
{
|
||
SYSCTL_IRP_DISPOSITION disposition;
|
||
NTSTATUS status;
|
||
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)DeviceObject->DeviceExtension;
|
||
|
||
PAGED_CODE();
|
||
|
||
status = WmiSystemControl( &pDevExt->WmiLibContext, DeviceObject, Irp, &disposition);
|
||
switch(disposition) {
|
||
case IrpProcessed:
|
||
|
||
//
|
||
// This irp has been processed and may be completed or pending.
|
||
//
|
||
break;
|
||
|
||
case IrpNotCompleted:
|
||
|
||
//
|
||
// This irp has not been completed, but has been fully processed.
|
||
// we will complete it now
|
||
//
|
||
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
break;
|
||
|
||
case IrpForward:
|
||
case IrpNotWmi:
|
||
default:
|
||
|
||
//
|
||
// This irp is either not a WMI irp or is a WMI irp targetted
|
||
// at a device lower in the stack.
|
||
//
|
||
// If this was an FDO we would pass the IRP down the stack, but
|
||
// this is a PDO (or PODO) so there no one below us in the
|
||
// stack.
|
||
//
|
||
Irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST;
|
||
status = STATUS_INVALID_DEVICE_REQUEST;
|
||
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
break;
|
||
|
||
}
|
||
|
||
return status;
|
||
|
||
}
|
||
|
||
NTSTATUS
|
||
ParWmiPdoQueryWmiRegInfo(
|
||
IN PDEVICE_OBJECT PDevObj,
|
||
OUT PULONG PRegFlags,
|
||
OUT PUNICODE_STRING PInstanceName,
|
||
OUT PUNICODE_STRING *PRegistryPath,
|
||
OUT PUNICODE_STRING MofResourceName,
|
||
OUT PDEVICE_OBJECT *Pdo
|
||
)
|
||
{
|
||
PDEVICE_EXTENSION devExt = (PDEVICE_EXTENSION)PDevObj->DeviceExtension;
|
||
|
||
PAGED_CODE();
|
||
|
||
if( ParIsPodo(PDevObj) ) {
|
||
// This is a PODO
|
||
PWSTR buffer = ExAllocatePool(PagedPool, devExt->SymbolicLinkName.MaximumLength);
|
||
ParDumpV( ("wmipdo::ParWmiPdoQueryWmiRegInfo: PODO - %wZ\n", &devExt->SymbolicLinkName) );
|
||
if( !buffer ) {
|
||
PInstanceName->Length = 0;
|
||
PInstanceName->MaximumLength = 0;
|
||
PInstanceName->Buffer = NULL;
|
||
} else {
|
||
PInstanceName->Length = 0;
|
||
PInstanceName->MaximumLength = devExt->SymbolicLinkName.MaximumLength;
|
||
PInstanceName->Buffer = buffer;
|
||
*PRegFlags = WMIREG_FLAG_INSTANCE_BASENAME;
|
||
RtlCopyUnicodeString( PInstanceName, &devExt->SymbolicLinkName );
|
||
}
|
||
*PRegistryPath = &RegistryPath;
|
||
} else {
|
||
// this is a PDO
|
||
ParDumpV( ("wmipdo::ParWmiPdoQueryWmiRegInfo: PDO - %x\n", PDevObj) );
|
||
*PRegFlags = WMIREG_FLAG_INSTANCE_PDO;
|
||
*PRegistryPath = &RegistryPath;
|
||
*Pdo = PDevObj;
|
||
}
|
||
return STATUS_SUCCESS;
|
||
}
|
||
|
||
NTSTATUS
|
||
ParWmiPdoQueryWmiDataBlock(
|
||
IN PDEVICE_OBJECT DeviceObject,
|
||
IN PIRP Irp,
|
||
IN ULONG GuidIndex,
|
||
IN ULONG InstanceIndex,
|
||
IN ULONG InstanceCount,
|
||
IN OUT PULONG InstanceLengthArray,
|
||
IN ULONG OutBufferSize,
|
||
OUT PUCHAR Buffer
|
||
)
|
||
{
|
||
NTSTATUS status;
|
||
ULONG size = sizeof(PARALLEL_WMI_LOG_INFO);
|
||
PDEVICE_EXTENSION devExt = DeviceObject->DeviceExtension;
|
||
|
||
PAGED_CODE();
|
||
|
||
//
|
||
// Only ever registers 1 instance per guid
|
||
//
|
||
ASSERT(InstanceIndex == 0 && InstanceCount == 1);
|
||
|
||
switch (GuidIndex) {
|
||
case PAR_WMI_BYTES_TRANSFERRED_GUID_INDEX:
|
||
|
||
//
|
||
// Request is for Bytes Transferred
|
||
//
|
||
if (OutBufferSize < size) {
|
||
status = STATUS_BUFFER_TOO_SMALL;
|
||
break;
|
||
}
|
||
*( (PPARALLEL_WMI_LOG_INFO)Buffer ) = devExt->log;
|
||
*InstanceLengthArray = size;
|
||
status = STATUS_SUCCESS;
|
||
break;
|
||
|
||
default:
|
||
//
|
||
// Index value larger than our largest supported - invalid request
|
||
//
|
||
status = STATUS_WMI_GUID_NOT_FOUND;
|
||
break;
|
||
}
|
||
|
||
status = WmiCompleteRequest( DeviceObject, Irp, status, size, IO_NO_INCREMENT );
|
||
|
||
return status;
|
||
}
|
||
|
||
NTSTATUS
|
||
ParWMIRegistrationControl(
|
||
IN PDEVICE_OBJECT DeviceObject,
|
||
IN ULONG Action
|
||
)
|
||
/*+
|
||
|
||
Wrapper function for IoWMIRegistrationControl that uses Device
|
||
Extension variable WmiRegistrationCount to prevent device from
|
||
registering more than once or from unregistering if not
|
||
registered.
|
||
|
||
-*/
|
||
{
|
||
PDEVICE_EXTENSION devExt = DeviceObject->DeviceExtension;
|
||
NTSTATUS status;
|
||
LONG count;
|
||
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: enter - %wZ\n", &devExt->SymbolicLinkName) );
|
||
|
||
switch( Action ) {
|
||
|
||
case WMIREG_ACTION_REGISTER :
|
||
|
||
//
|
||
// Verify that we don't register more than once
|
||
//
|
||
count = InterlockedIncrement(&devExt->WmiRegistrationCount);
|
||
if( count == 1 ) {
|
||
status = IoWMIRegistrationControl(DeviceObject, Action);
|
||
if( !NT_SUCCESS(status) ) {
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: REGISTER - FAIL\n") );
|
||
// registration failed - back out the increment
|
||
InterlockedDecrement(&devExt->WmiRegistrationCount);
|
||
} else {
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: REGISTER - SUCCEED\n") );
|
||
}
|
||
} else {
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: REGISTER - ABORT - already REGISTERed\n") );
|
||
// back out the increment
|
||
InterlockedDecrement(&devExt->WmiRegistrationCount);
|
||
status = STATUS_UNSUCCESSFUL;
|
||
// ASSERTMSG( "Already registered for WMI registration, fail registration", FALSE );
|
||
}
|
||
break;
|
||
|
||
case WMIREG_ACTION_DEREGISTER :
|
||
|
||
//
|
||
// verify that we don't unregister if we are not registered
|
||
//
|
||
count = InterlockedDecrement(&devExt->WmiRegistrationCount);
|
||
if( count == 0 ) {
|
||
status = IoWMIRegistrationControl(DeviceObject, Action);
|
||
if( !NT_SUCCESS(status) ) {
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: DEREGISTER - FAIL\n") );
|
||
// unregistration failed?
|
||
// ASSERTMSG( "WMI unregistration failed?", FALSE );
|
||
InterlockedIncrement(&devExt->WmiRegistrationCount);
|
||
} else {
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: DEREGISTER - SUCCEED\n") );
|
||
}
|
||
} else {
|
||
ParDumpV( ("wmipdo::ParWMIRegistrationControl: DEREGISTER - ABORT - not registered\n") );
|
||
// unregistration failed - back out the decrement
|
||
InterlockedIncrement(&devExt->WmiRegistrationCount);
|
||
status = STATUS_UNSUCCESSFUL;
|
||
// ASSERTMSG( "Not registered for WMI, fail unregister", FALSE );
|
||
}
|
||
break;
|
||
|
||
default:
|
||
|
||
// unrecognized action
|
||
status = STATUS_UNSUCCESSFUL;
|
||
ASSERTMSG("wmipdo::ParWMIRegistrationControl: Unrecognized WMI registration Action \n",FALSE);
|
||
}
|
||
|
||
return status;
|
||
}
|