wmi.c File Reference

#include "atapi.h"

Include dependency graph for wmi.c:

Go to the source code of this file.

Functions
static NTSTATUS NTAPI	AtaQueryWmiRegInfo (_Inout_ PDEVICE_OBJECT DeviceObject, _Inout_ PULONG RegFlags, _Inout_ PUNICODE_STRING InstanceName, _Outptr_result_maybenull_ PUNICODE_STRING *RegistryPath, _Inout_ PUNICODE_STRING MofResourceName, _Outptr_result_maybenull_ PDEVICE_OBJECT *Pdo)
static NTSTATUS NTAPI	AtaQueryWmiDataBlock (_Inout_ PDEVICE_OBJECT DeviceObject, _Inout_ PIRP Irp, _In_ ULONG GuidIndex, _In_ ULONG InstanceIndex, _In_ ULONG InstanceCount, _Out_writes_opt_(InstanceCount) PULONG InstanceLengthArray, _In_ ULONG BufferAvail, _Out_writes_bytes_opt_(BufferAvail) PUCHAR Buffer)
static NTSTATUS	AtaPdoWmi (_In_ PATAPORT_DEVICE_EXTENSION DevExt, _Inout_ PIRP Irp)
static NTSTATUS	AtaFdoWmi (_In_ PATAPORT_CHANNEL_EXTENSION ChanExt, _Inout_ PIRP Irp)
NTSTATUS NTAPI	AtaDispatchWmi (_In_ PDEVICE_OBJECT DeviceObject, _Inout_ PIRP Irp)
NTSTATUS	AtaPdoWmiRegistration (_In_ PATAPORT_DEVICE_EXTENSION DevExt, _In_ BOOLEAN Register)

Variables
static WMIGUIDREGINFO	AtapWmiGuidList []

Function Documentation

AtaDispatchWmi()

NTSTATUS NTAPI AtaDispatchWmi	(	_In_ PDEVICE_OBJECT	DeviceObject,
		_Inout_ PIRP	Irp
	)

Definition at line 175 of file wmi.c.

{
    PAGED_CODE();
 
    if (IS_FDO(DeviceObject->DeviceExtension))
        return AtaFdoWmi(DeviceObject->DeviceExtension, Irp);
    else
        return AtaPdoWmi(DeviceObject->DeviceExtension, Irp);
}

Referenced by DriverEntry().

AtaFdoWmi()

static NTSTATUS AtaFdoWmi ( _In_ PATAPORT_CHANNEL_EXTENSION  ChanExt, _Inout_ PIRP  Irp  )

static

Definition at line 162 of file wmi.c.

{
    PAGED_CODE();
 
    IoSkipCurrentIrpStackLocation(Irp);
    return IoCallDriver(ChanExt->Common.LowerDeviceObject, Irp);
}

Referenced by AtaDispatchWmi().

AtaPdoWmi()

static NTSTATUS AtaPdoWmi ( _In_ PATAPORT_DEVICE_EXTENSION  DevExt, _Inout_ PIRP  Irp  )

static

Definition at line 110 of file wmi.c.

{
    NTSTATUS Status;
    SYSCTL_IRP_DISPOSITION Disposition;
 
    PAGED_CODE();
 
    TRACE("%s(%p, %p)\n", __FUNCTION__, DevExt, Irp);
 
    Status = IoAcquireRemoveLock(&DevExt->Common.RemoveLock, Irp);
    if (!NT_SUCCESS(Status))
    {
        Irp->IoStatus.Status = Status;
        IoCompleteRequest(Irp, IO_NO_INCREMENT);
 
        return Status;
    }
 
    Status = WmiSystemControl(&DevExt->WmiLibInfo,
                              DevExt->Common.Self,
                              Irp,
                              &Disposition);
    switch (Disposition)
    {
        case IrpProcessed:
            break;
 
        case IrpNotCompleted:
            IoCompleteRequest(Irp, IO_NO_INCREMENT);
            break;
 
        default:
            ASSERT(FALSE);
            __fallthrough;
        case IrpForward:
        case IrpNotWmi:
            Status = STATUS_NOT_SUPPORTED;
            Irp->IoStatus.Status = Status;
            IoCompleteRequest(Irp, IO_NO_INCREMENT);
            break;
    }
 
    IoReleaseRemoveLock(&DevExt->Common.RemoveLock, Irp);
 
    return Status;
}

Referenced by AtaDispatchWmi().

AtaPdoWmiRegistration()

NTSTATUS AtaPdoWmiRegistration	(	_In_ PATAPORT_DEVICE_EXTENSION	DevExt,
		_In_ BOOLEAN	Register
	)

Definition at line 189 of file wmi.c.

{
    ULONG Action;
 
    PAGED_CODE();
 
    if (Register)
    {
        DevExt->WmiLibInfo.GuidCount = RTL_NUMBER_OF(AtapWmiGuidList);
        DevExt->WmiLibInfo.GuidList = AtapWmiGuidList;
 
        DevExt->WmiLibInfo.QueryWmiRegInfo = AtaQueryWmiRegInfo;
        DevExt->WmiLibInfo.QueryWmiDataBlock = AtaQueryWmiDataBlock;
 
        Action = WMIREG_ACTION_REGISTER;
    }
    else
    {
        Action = WMIREG_ACTION_DEREGISTER;
    }
 
    return IoWMIRegistrationControl(DevExt->Common.Self, Action);
}

Referenced by AtaPdoRemoveDevice(), and AtaPdoStartDevice().

AtaQueryWmiDataBlock()

static NTSTATUS NTAPI AtaQueryWmiDataBlock ( _Inout_ PDEVICE_OBJECT  DeviceObject, _Inout_ PIRP  Irp, _In_ ULONG  GuidIndex, _In_ ULONG  InstanceIndex, _In_ ULONG  InstanceCount, _Out_writes_opt_(InstanceCount) PULONG  InstanceLengthArray, _In_ ULONG  BufferAvail, _Out_writes_bytes_opt_(BufferAvail) PUCHAR  Buffer  )

static

Definition at line 51 of file wmi.c.

{
    PATAPORT_DEVICE_EXTENSION DevExt;
    NTSTATUS Status;
    ATA_SCSI_ADDRESS AtaScsiAddress;
    PMSIde_PortDeviceInfo DeviceInfo;
 
    PAGED_CODE();
 
    if (GuidIndex > RTL_NUMBER_OF(AtapWmiGuidList))
    {
        Status = STATUS_WMI_GUID_NOT_FOUND;
        goto Complete;
    }
 
    /* Only ever register 1 instance per GUID */
    if (InstanceIndex != 0 || InstanceCount != 1)
    {
        Status = STATUS_WMI_INSTANCE_NOT_FOUND;
        goto Complete;
    }
 
    if (!InstanceLengthArray || BufferAvail < sizeof(*DeviceInfo))
    {
        Status = STATUS_BUFFER_TOO_SMALL;
        goto Complete;
    }
 
    DevExt = DeviceObject->DeviceExtension;
    AtaScsiAddress = DevExt->Device.AtaScsiAddress;
 
    DeviceInfo = (PMSIde_PortDeviceInfo)Buffer;
    DeviceInfo->Bus = AtaScsiAddress.PathId;
    DeviceInfo->Target = AtaScsiAddress.TargetId;
    DeviceInfo->Lun = AtaScsiAddress.Lun;
 
    *InstanceLengthArray = sizeof(*DeviceInfo);
 
    Status = STATUS_SUCCESS;
 
Complete:
    return WmiCompleteRequest(DeviceObject,
                              Irp,
                              Status,
                              sizeof(*DeviceInfo),
                              IO_NO_INCREMENT);
}

Referenced by AtaPdoWmiRegistration().

AtaQueryWmiRegInfo()

static NTSTATUS NTAPI AtaQueryWmiRegInfo ( _Inout_ PDEVICE_OBJECT  DeviceObject, _Inout_ PULONG  RegFlags, _Inout_ PUNICODE_STRING  InstanceName, _Outptr_result_maybenull_ PUNICODE_STRING *  RegistryPath, _Inout_ PUNICODE_STRING  MofResourceName, _Outptr_result_maybenull_ PDEVICE_OBJECT *  Pdo  )

static

Definition at line 25 of file wmi.c.

{
    PATAPORT_DEVICE_EXTENSION DevExt = DeviceObject->DeviceExtension;
 
    UNREFERENCED_PARAMETER(InstanceName);
    UNREFERENCED_PARAMETER(MofResourceName);
 
    PAGED_CODE();
 
    *RegFlags = WMIREG_FLAG_INSTANCE_PDO;
    *RegistryPath = &AtapDriverRegistryPath;
    *Pdo = DevExt->Common.Self;
 
    return STATUS_SUCCESS;
}

Referenced by AtaPdoWmiRegistration().

Variable Documentation

AtapWmiGuidList

WMIGUIDREGINFO AtapWmiGuidList[]

static

Initial value:

=
{
    {&MSIde_PortDeviceInfo_GUID, 1, 0}
}

Definition at line 14 of file wmi.c.

Referenced by AtaPdoWmiRegistration(), and AtaQueryWmiDataBlock().