481 lines
13 KiB
C
481 lines
13 KiB
C
/*++
|
||
|
||
Copyright (c) 1997 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
wmi.c
|
||
|
||
Abstract:
|
||
|
||
This module contains the code that handles the wmi IRPs for the
|
||
serial driver.
|
||
|
||
Environment:
|
||
|
||
Kernel mode
|
||
|
||
Revision History :
|
||
--*/
|
||
|
||
#include "precomp.h"
|
||
#include <wmistr.h>
|
||
|
||
#ifdef ALLOC_PRAGMA
|
||
#pragma alloc_text(PAGESRP0, SerialSystemControlDispatch)
|
||
#pragma alloc_text(PAGESRP0, SerialTossWMIRequest)
|
||
#pragma alloc_text(PAGESRP0, SerialSetWmiDataItem)
|
||
#pragma alloc_text(PAGESRP0, SerialSetWmiDataBlock)
|
||
#pragma alloc_text(PAGESRP0, SerialQueryWmiDataBlock)
|
||
#pragma alloc_text(PAGESRP0, SerialQueryWmiRegInfo)
|
||
#endif
|
||
|
||
|
||
NTSTATUS
|
||
SerialSystemControlDispatch(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
|
||
{
|
||
SYSCTL_IRP_DISPOSITION disposition;
|
||
NTSTATUS status;
|
||
PSERIAL_DEVICE_EXTENSION pDevExt
|
||
= (PSERIAL_DEVICE_EXTENSION)DeviceObject->DeviceExtension;
|
||
|
||
PAGED_CODE();
|
||
|
||
status = WmiSystemControl( &pDevExt->WmiLibInfo,
|
||
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:
|
||
{
|
||
//
|
||
// This irp is either not a WMI irp or is a WMI irp targetted
|
||
// at a device lower in the stack.
|
||
IoSkipCurrentIrpStackLocation(Irp);
|
||
status = IoCallDriver(pDevExt->LowerDeviceObject, Irp);
|
||
break;
|
||
}
|
||
|
||
default:
|
||
{
|
||
//
|
||
// We really should never get here, but if we do just forward....
|
||
ASSERT(FALSE);
|
||
IoSkipCurrentIrpStackLocation(Irp);
|
||
status = IoCallDriver(pDevExt->LowerDeviceObject, Irp);
|
||
break;
|
||
}
|
||
}
|
||
|
||
return(status);
|
||
|
||
}
|
||
|
||
|
||
#define WMI_SERIAL_PORT_NAME_INFORMATION 0
|
||
#define WMI_SERIAL_PORT_COMM_INFORMATION 1
|
||
#define WMI_SERIAL_PORT_HW_INFORMATION 2
|
||
#define WMI_SERIAL_PORT_PERF_INFORMATION 3
|
||
#define WMI_SERIAL_PORT_PROPERTIES 4
|
||
|
||
GUID SerialPortNameGuid = SERIAL_PORT_WMI_NAME_GUID;
|
||
GUID SerialPortCommGuid = SERIAL_PORT_WMI_COMM_GUID;
|
||
GUID SerialPortHWGuid = SERIAL_PORT_WMI_HW_GUID;
|
||
GUID SerailPortPerfGuid = SERIAL_PORT_WMI_PERF_GUID;
|
||
GUID SerialPortPropertiesGuid = SERIAL_PORT_WMI_PROPERTIES_GUID;
|
||
|
||
WMIGUIDREGINFO SerialWmiGuidList[SERIAL_WMI_GUID_LIST_SIZE] =
|
||
{
|
||
{ &SerialPortNameGuid, 1, 0 },
|
||
{ &SerialPortCommGuid, 1, 0 },
|
||
{ &SerialPortHWGuid, 1, 0 },
|
||
{ &SerailPortPerfGuid, 1, 0 },
|
||
{ &SerialPortPropertiesGuid, 1, 0}
|
||
};
|
||
|
||
//
|
||
// WMI System Call back functions
|
||
//
|
||
|
||
|
||
|
||
NTSTATUS
|
||
SerialTossWMIRequest(IN PDEVICE_OBJECT PDevObj, IN PIRP PIrp,
|
||
IN ULONG GuidIndex)
|
||
{
|
||
PSERIAL_DEVICE_EXTENSION pDevExt;
|
||
NTSTATUS status;
|
||
|
||
PAGED_CODE();
|
||
|
||
pDevExt = (PSERIAL_DEVICE_EXTENSION)PDevObj->DeviceExtension;
|
||
|
||
switch (GuidIndex) {
|
||
|
||
case WMI_SERIAL_PORT_NAME_INFORMATION:
|
||
case WMI_SERIAL_PORT_COMM_INFORMATION:
|
||
case WMI_SERIAL_PORT_HW_INFORMATION:
|
||
case WMI_SERIAL_PORT_PERF_INFORMATION:
|
||
case WMI_SERIAL_PORT_PROPERTIES:
|
||
status = STATUS_INVALID_DEVICE_REQUEST;
|
||
break;
|
||
|
||
default:
|
||
status = STATUS_WMI_GUID_NOT_FOUND;
|
||
break;
|
||
}
|
||
|
||
status = WmiCompleteRequest(PDevObj, PIrp,
|
||
status, 0, IO_NO_INCREMENT);
|
||
|
||
return status;
|
||
}
|
||
|
||
|
||
NTSTATUS
|
||
SerialSetWmiDataItem(IN PDEVICE_OBJECT PDevObj, IN PIRP PIrp,
|
||
IN ULONG GuidIndex, IN ULONG InstanceIndex,
|
||
IN ULONG DataItemId,
|
||
IN ULONG BufferSize, IN PUCHAR PBuffer)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine is a callback into the driver to set for the contents of
|
||
a data block. When the driver has finished filling the data block it
|
||
must call ClassWmiCompleteRequest to complete the irp. The driver can
|
||
return STATUS_PENDING if the irp cannot be completed immediately.
|
||
|
||
Arguments:
|
||
|
||
PDevObj is the device whose data block is being queried
|
||
|
||
PIrp is the Irp that makes this request
|
||
|
||
GuidIndex is the index into the list of guids provided when the
|
||
device registered
|
||
|
||
InstanceIndex is the index that denotes which instance of the data block
|
||
is being queried.
|
||
|
||
DataItemId has the id of the data item being set
|
||
|
||
BufferSize has the size of the data item passed
|
||
|
||
PBuffer has the new values for the data item
|
||
|
||
|
||
Return Value:
|
||
|
||
status
|
||
|
||
--*/
|
||
{
|
||
PAGED_CODE();
|
||
|
||
//
|
||
// Toss this request -- we don't support anything for it
|
||
//
|
||
|
||
return SerialTossWMIRequest(PDevObj, PIrp, GuidIndex);
|
||
}
|
||
|
||
|
||
NTSTATUS
|
||
SerialSetWmiDataBlock(IN PDEVICE_OBJECT PDevObj, IN PIRP PIrp,
|
||
IN ULONG GuidIndex, IN ULONG InstanceIndex,
|
||
IN ULONG BufferSize,
|
||
IN PUCHAR PBuffer)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine is a callback into the driver to set the contents of
|
||
a data block. When the driver has finished filling the data block it
|
||
must call ClassWmiCompleteRequest to complete the irp. The driver can
|
||
return STATUS_PENDING if the irp cannot be completed immediately.
|
||
|
||
Arguments:
|
||
|
||
PDevObj is the device whose data block is being queried
|
||
|
||
PIrp is the Irp that makes this request
|
||
|
||
GuidIndex is the index into the list of guids provided when the
|
||
device registered
|
||
|
||
InstanceIndex is the index that denotes which instance of the data block
|
||
is being queried.
|
||
|
||
BufferSize has the size of the data block passed
|
||
|
||
PBuffer has the new values for the data block
|
||
|
||
|
||
Return Value:
|
||
|
||
status
|
||
|
||
--*/
|
||
{
|
||
PAGED_CODE();
|
||
|
||
//
|
||
// Toss this request -- we don't support anything for it
|
||
//
|
||
|
||
return SerialTossWMIRequest(PDevObj, PIrp, GuidIndex);
|
||
}
|
||
|
||
|
||
NTSTATUS
|
||
SerialQueryWmiDataBlock(IN PDEVICE_OBJECT PDevObj, IN PIRP PIrp,
|
||
IN ULONG GuidIndex,
|
||
IN ULONG InstanceIndex,
|
||
IN ULONG InstanceCount,
|
||
IN OUT PULONG InstanceLengthArray,
|
||
IN ULONG OutBufferSize,
|
||
OUT PUCHAR PBuffer)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine is a callback into the driver to query for the contents of
|
||
a data block. When the driver has finished filling the data block it
|
||
must call ClassWmiCompleteRequest to complete the irp. The driver can
|
||
return STATUS_PENDING if the irp cannot be completed immediately.
|
||
|
||
Arguments:
|
||
|
||
PDevObj is the device whose data block is being queried
|
||
|
||
PIrp is the Irp that makes this request
|
||
|
||
GuidIndex is the index into the list of guids provided when the
|
||
device registered
|
||
|
||
InstanceIndex is the index that denotes which instance of the data block
|
||
is being queried.
|
||
|
||
InstanceCount is the number of instnaces expected to be returned for
|
||
the data block.
|
||
|
||
InstanceLengthArray is a pointer to an array of ULONG that returns the
|
||
lengths of each instance of the data block. If this is NULL then
|
||
there was not enough space in the output buffer to fufill the request
|
||
so the irp should be completed with the buffer needed.
|
||
|
||
BufferAvail on has the maximum size available to write the data
|
||
block.
|
||
|
||
PBuffer on return is filled with the returned data block
|
||
|
||
|
||
Return Value:
|
||
|
||
status
|
||
|
||
--*/
|
||
{
|
||
NTSTATUS status;
|
||
ULONG size = 0;
|
||
PSERIAL_DEVICE_EXTENSION pDevExt
|
||
= (PSERIAL_DEVICE_EXTENSION)PDevObj->DeviceExtension;
|
||
|
||
PAGED_CODE();
|
||
|
||
switch (GuidIndex) {
|
||
case WMI_SERIAL_PORT_NAME_INFORMATION:
|
||
size = pDevExt->WmiIdentifier.Length;
|
||
|
||
if (OutBufferSize < (size + sizeof(USHORT))) {
|
||
size += sizeof(USHORT);
|
||
status = STATUS_BUFFER_TOO_SMALL;
|
||
break;
|
||
}
|
||
|
||
if (pDevExt->WmiIdentifier.Buffer == NULL) {
|
||
status = STATUS_INSUFFICIENT_RESOURCES;
|
||
break;
|
||
}
|
||
|
||
//
|
||
// First, copy the string over containing our identifier
|
||
//
|
||
|
||
*(USHORT *)PBuffer = (USHORT)size;
|
||
(UCHAR *)PBuffer += sizeof(USHORT);
|
||
|
||
RtlCopyMemory(PBuffer, pDevExt->WmiIdentifier.Buffer, size);
|
||
|
||
//
|
||
// Increment total size to include the WORD containing our len
|
||
//
|
||
|
||
size += sizeof(USHORT);
|
||
*InstanceLengthArray = size;
|
||
|
||
status = STATUS_SUCCESS;
|
||
|
||
break;
|
||
|
||
case WMI_SERIAL_PORT_COMM_INFORMATION:
|
||
size = sizeof(SERIAL_WMI_COMM_DATA);
|
||
|
||
if (OutBufferSize < size) {
|
||
status = STATUS_BUFFER_TOO_SMALL;
|
||
break;
|
||
}
|
||
|
||
*InstanceLengthArray = size;
|
||
*(PSERIAL_WMI_COMM_DATA)PBuffer = pDevExt->WmiCommData;
|
||
|
||
status = STATUS_SUCCESS;
|
||
|
||
break;
|
||
|
||
case WMI_SERIAL_PORT_HW_INFORMATION:
|
||
size = sizeof(SERIAL_WMI_HW_DATA);
|
||
|
||
if (OutBufferSize < size) {
|
||
status = STATUS_BUFFER_TOO_SMALL;
|
||
break;
|
||
}
|
||
|
||
*InstanceLengthArray = size;
|
||
*(PSERIAL_WMI_HW_DATA)PBuffer = pDevExt->WmiHwData;
|
||
|
||
status = STATUS_SUCCESS;
|
||
|
||
break;
|
||
|
||
case WMI_SERIAL_PORT_PERF_INFORMATION:
|
||
size = sizeof(SERIAL_WMI_PERF_DATA);
|
||
|
||
if (OutBufferSize < size) {
|
||
status = STATUS_BUFFER_TOO_SMALL;
|
||
break;
|
||
}
|
||
|
||
*InstanceLengthArray = size;
|
||
*(PSERIAL_WMI_PERF_DATA)PBuffer = pDevExt->WmiPerfData;
|
||
|
||
status = STATUS_SUCCESS;
|
||
|
||
break;
|
||
|
||
case WMI_SERIAL_PORT_PROPERTIES:
|
||
size = sizeof(SERIAL_COMMPROP) + sizeof(ULONG);
|
||
|
||
if (OutBufferSize < size) {
|
||
status = STATUS_BUFFER_TOO_SMALL;
|
||
break;
|
||
}
|
||
|
||
*InstanceLengthArray = size;
|
||
SerialGetProperties(
|
||
pDevExt,
|
||
(PSERIAL_COMMPROP)PBuffer
|
||
);
|
||
|
||
*((PULONG)(((PSERIAL_COMMPROP)PBuffer)->ProvChar)) = 0;
|
||
|
||
status = STATUS_SUCCESS;
|
||
|
||
break;
|
||
|
||
default:
|
||
status = STATUS_WMI_GUID_NOT_FOUND;
|
||
break;
|
||
}
|
||
|
||
status = WmiCompleteRequest( PDevObj, PIrp,
|
||
status, size, IO_NO_INCREMENT);
|
||
|
||
return status;
|
||
}
|
||
|
||
|
||
NTSTATUS
|
||
SerialQueryWmiRegInfo(IN PDEVICE_OBJECT PDevObj, OUT PULONG PRegFlags,
|
||
OUT PUNICODE_STRING PInstanceName,
|
||
OUT PUNICODE_STRING *PRegistryPath,
|
||
OUT PUNICODE_STRING MofResourceName,
|
||
OUT PDEVICE_OBJECT *Pdo)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine is a callback into the driver to retrieve information about
|
||
the guids being registered.
|
||
|
||
Implementations of this routine may be in paged memory
|
||
|
||
Arguments:
|
||
|
||
DeviceObject is the device whose registration information is needed
|
||
|
||
*RegFlags returns with a set of flags that describe all of the guids being
|
||
registered for this device. If the device wants enable and disable
|
||
collection callbacks before receiving queries for the registered
|
||
guids then it should return the WMIREG_FLAG_EXPENSIVE flag. Also the
|
||
returned flags may specify WMIREG_FLAG_INSTANCE_PDO in which case
|
||
the instance name is determined from the PDO associated with the
|
||
device object. Note that the PDO must have an associated devnode. If
|
||
WMIREG_FLAG_INSTANCE_PDO is not set then Name must return a unique
|
||
name for the device. These flags are ORed into the flags specified
|
||
by the GUIDREGINFO for each guid.
|
||
|
||
InstanceName returns with the instance name for the guids if
|
||
WMIREG_FLAG_INSTANCE_PDO is not set in the returned *RegFlags. The
|
||
caller will call ExFreePool with the buffer returned.
|
||
|
||
*RegistryPath returns with the registry path of the driver. This is
|
||
required
|
||
|
||
*MofResourceName returns with the name of the MOF resource attached to
|
||
the binary file. If the driver does not have a mof resource attached
|
||
then this can be returned as NULL.
|
||
|
||
*Pdo returns with the device object for the PDO associated with this
|
||
device if the WMIREG_FLAG_INSTANCE_PDO flag is retured in
|
||
*RegFlags.
|
||
|
||
Return Value:
|
||
|
||
status
|
||
|
||
--*/
|
||
{
|
||
PSERIAL_DEVICE_EXTENSION pDevExt
|
||
= (PSERIAL_DEVICE_EXTENSION)PDevObj->DeviceExtension;
|
||
|
||
PAGED_CODE();
|
||
|
||
*PRegFlags = WMIREG_FLAG_INSTANCE_PDO;
|
||
*PRegistryPath = &SerialGlobals.RegistryPath;
|
||
*Pdo = pDevExt->Pdo;
|
||
|
||
return STATUS_SUCCESS;
|
||
}
|
||
|