disk.h File Reference

#include "ntddk.h"
#include "scsi.h"
#include <wmidata.h>
#include "classpnp.h"
#include <wmistr.h>
#include "ntstrsafe.h"
#include <storswtr.h>

Include dependency graph for disk.h:

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

Go to the source code of this file.

Classes
struct	_DISK_GROUP_CONTEXT
struct	_DISK_DATA
struct	_DISK_MEDIA_TYPES_LIST
struct	DISKREREGREQUEST
struct	_DISK_VERIFY_WORKITEM_CONTEXT

Macros
#define	DEBUG_COMP_ID   DPFLTR_DISK_ID
#define	WPP_GUID_DISK   (945186BF, 3DD6, 4f3f, 9C8E, 9EDD3FC9D558)
#define	WPP_CONTROL_GUIDS   WPP_CONTROL_GUIDS_NORMAL_FLAGS(WPP_GUID_DISK)
#define	DISK_TAG_GENERAL   ' DcS'
#define	DISK_TAG_SMART   'aDcS'
#define	DISK_TAG_INFO_EXCEPTION   'ADcS'
#define	DISK_TAG_DISABLE_CACHE   'CDcS'
#define	DISK_TAG_CCONTEXT   'cDcS'
#define	DISK_TAG_DISK_GEOM   'GDcS'
#define	DISK_TAG_UPDATE_GEOM   'gDcS'
#define	DISK_TAG_SENSE_INFO   'IDcS'
#define	DISK_TAG_PNP_ID   'iDcS'
#define	DISK_TAG_MODE_DATA   'MDcS'
#define	DISK_CACHE_MBR_CHECK   'mDcS'
#define	DISK_TAG_NAME   'NDcS'
#define	DISK_TAG_READ_CAP   'PDcS'
#define	DISK_TAG_PART_LIST   'pDcS'
#define	DISK_TAG_SRB   'SDcS'
#define	DISK_TAG_START   'sDcS'
#define	DISK_TAG_UPDATE_CAP   'UDcS'
#define	DISK_TAG_WI_CONTEXT   'WDcS'
#define	HackDisableTaggedQueuing   (0x01)
#define	HackDisableSynchronousTransfers   (0x02)
#define	HackDisableSpinDown   (0x04)
#define	HackDisableWriteCache   (0x08)
#define	HackCauseNotReportableHack   (0x10)
#define	HackRequiresStartUnitCommand   (0x20)
#define	DiskDeviceParameterSubkey   L"Disk"
#define	DiskDeviceUserWriteCacheSetting   L"UserWriteCacheSetting"
#define	DiskDeviceCacheIsPowerProtected   L"CacheIsPowerProtected"
#define	FUNCTIONAL_EXTENSION_SIZE   sizeof(FUNCTIONAL_DEVICE_EXTENSION) + sizeof(DISK_DATA)
#define	MODE_DATA_SIZE   192
#define	VALUE_BUFFER_SIZE   2048
#define	SCSI_DISK_TIMEOUT   10
#define	PARTITION0_LIST_SIZE   4
#define	MAX_MEDIA_TYPES   4
#define	MAX_SECTORS_PER_VERIFY   0x100
#define	ONE_MILLI_SECOND   ((ULONGLONG)10 * 1000)
#define	DISK_DEFAULT_FAILURE_POLLING_PERIOD   1 * 60 * 60
#define	CHECK_IRQL()
#define	DiskIsValidSmartSelfTest(Subcommand)
#define	DiskReadDriveCapacity(Fdo)   ClassReadDriveCapacity(Fdo)
#define	DiskGetDetectInfo(FdoExtension, DetectInfo)   (STATUS_UNSUCCESSFUL)
#define	DiskHashGuid(Guid)   (((PULONG) &Guid)[0] ^ ((PULONG) &Guid)[0] ^ ((PULONG) &Guid)[0] ^ ((PULONG) &Guid)[0])

Typedefs
typedef struct _DISK_GROUP_CONTEXT	DISK_GROUP_CONTEXT
typedef struct _DISK_GROUP_CONTEXT *	PDISK_GROUP_CONTEXT
typedef enum _DISK_USER_WRITE_CACHE_SETTING	DISK_USER_WRITE_CACHE_SETTING
typedef enum _DISK_USER_WRITE_CACHE_SETTING *	PDISK_USER_WRITE_CACHE_SETTING
typedef struct _DISK_DATA	DISK_DATA
typedef struct _DISK_DATA *	PDISK_DATA
typedef struct _DISK_MEDIA_TYPES_LIST	DISK_MEDIA_TYPES_LIST
typedef struct _DISK_MEDIA_TYPES_LIST *	PDISK_MEDIA_TYPES_LIST
typedef struct DISKREREGREQUEST *	PDISKREREGREQUEST
typedef struct _DISK_VERIFY_WORKITEM_CONTEXT	DISK_VERIFY_WORKITEM_CONTEXT
typedef struct _DISK_VERIFY_WORKITEM_CONTEXT *	PDISK_VERIFY_WORKITEM_CONTEXT

Enumerations
enum	_DISK_USER_WRITE_CACHE_SETTING { DiskWriteCacheDisable = 0 , DiskWriteCacheEnable = 1 , DiskWriteCacheDefault = -1 }

Functions
VOID NTAPI	DiskUnload (IN PDRIVER_OBJECT DriverObject)
NTSTATUS NTAPI	DiskAddDevice (IN PDRIVER_OBJECT DriverObject, IN PDEVICE_OBJECT Pdo)
NTSTATUS NTAPI	DiskInitFdo (IN PDEVICE_OBJECT Fdo)
NTSTATUS NTAPI	DiskStartFdo (IN PDEVICE_OBJECT Fdo)
NTSTATUS NTAPI	DiskStopDevice (IN PDEVICE_OBJECT DeviceObject, IN UCHAR Type)
NTSTATUS NTAPI	DiskRemoveDevice (IN PDEVICE_OBJECT DeviceObject, IN UCHAR Type)
NTSTATUS NTAPI	DiskReadWriteVerification (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	DiskDeviceControl (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
VOID NTAPI	DiskFdoProcessError (PDEVICE_OBJECT DeviceObject, PSCSI_REQUEST_BLOCK Srb, NTSTATUS *Status, BOOLEAN *Retry)
NTSTATUS NTAPI	DiskShutdownFlush (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	DiskGetCacheInformation (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, IN PDISK_CACHE_INFORMATION CacheInfo)
NTSTATUS NTAPI	DiskSetCacheInformation (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, IN PDISK_CACHE_INFORMATION CacheInfo)
VOID	DiskLogCacheInformation (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, IN PDISK_CACHE_INFORMATION CacheInfo, IN NTSTATUS Status)
NTSTATUS	DiskIoctlGetCacheSetting (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS	DiskIoctlSetCacheSetting (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
VOID	DiskFlushDispatch (IN PDEVICE_OBJECT Fdo, IN PDISK_GROUP_CONTEXT FlushContext)
NTSTATUS	DiskModeSelect (IN PDEVICE_OBJECT DeviceObject, _In_reads_bytes_(Length) PCHAR ModeSelectBuffer, IN ULONG Length, IN BOOLEAN SavePage)
NTSTATUS	DiskPerformSmartCommand (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, IN ULONG SrbControlCode, IN UCHAR Command, IN UCHAR Feature, IN UCHAR SectorCount, IN UCHAR SectorNumber, IN OUT PSRB_IO_CONTROL SrbControl, OUT PULONG BufferSize)
NTSTATUS	DiskGetInfoExceptionInformation (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, OUT PMODE_INFO_EXCEPTIONS ReturnPageData)
NTSTATUS	DiskSetInfoExceptionInformation (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, IN PMODE_INFO_EXCEPTIONS PageData)
NTSTATUS	DiskGetModePage (_In_ PDEVICE_OBJECT Fdo, _In_ UCHAR PageMode, _In_ UCHAR PageControl, _In_ PMODE_PARAMETER_HEADER ModeData, _Inout_ PULONG ModeDataSize, _Out_ PVOID *PageData)
NTSTATUS	DiskEnableInfoExceptions (_In_ PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, _In_ BOOLEAN Enable)
NTSTATUS	DiskDetectFailurePrediction (PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, PFAILURE_PREDICTION_METHOD FailurePredictCapability, BOOLEAN ScsiAddressAvailable)
NTSTATUS	DiskCreateFdo (IN PDRIVER_OBJECT DriverObject, IN PDEVICE_OBJECT LowerDeviceObject, IN PULONG DeviceCount, IN BOOLEAN DasdAccessOnly)
VOID NTAPI	DiskSetSpecialHacks (IN PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, IN ULONG_PTR Data)
VOID	ResetBus (IN PDEVICE_OBJECT DeviceObject)
NTSTATUS	DiskGenerateDeviceName (IN ULONG DeviceNumber, OUT PCCHAR *RawName)
VOID	DiskCreateSymbolicLinks (IN PDEVICE_OBJECT DeviceObject)
VOID	DiskDeleteSymbolicLinks (IN PDEVICE_OBJECT DeviceObject)
NTSTATUS NTAPI	DiskFdoQueryWmiRegInfo (IN PDEVICE_OBJECT DeviceObject, OUT ULONG *RegFlags, OUT PUNICODE_STRING InstanceName)
NTSTATUS NTAPI	DiskFdoQueryWmiRegInfoEx (IN PDEVICE_OBJECT DeviceObject, OUT ULONG *RegFlags, OUT PUNICODE_STRING InstanceName, OUT PUNICODE_STRING MofName)
NTSTATUS NTAPI	DiskFdoQueryWmiDataBlock (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN ULONG GuidIndex, IN ULONG BufferAvail, OUT PUCHAR Buffer)
NTSTATUS NTAPI	DiskFdoSetWmiDataBlock (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN ULONG GuidIndex, IN ULONG BufferSize, IN PUCHAR Buffer)
NTSTATUS NTAPI	DiskFdoSetWmiDataItem (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN ULONG GuidIndex, IN ULONG DataItemId, IN ULONG BufferSize, IN PUCHAR Buffer)
NTSTATUS NTAPI	DiskFdoExecuteWmiMethod (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN ULONG GuidIndex, IN ULONG MethodId, IN ULONG InBufferSize, IN ULONG OutBufferSize, IN PUCHAR Buffer)
NTSTATUS NTAPI	DiskWmiFunctionControl (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN ULONG GuidIndex, IN CLASSENABLEDISABLEFUNCTION Function, IN BOOLEAN Enable)
NTSTATUS	DiskReadFailurePredictStatus (PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, PSTORAGE_FAILURE_PREDICT_STATUS DiskSmartStatus)
NTSTATUS	DiskReadFailurePredictData (PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, PSTORAGE_FAILURE_PREDICT_DATA DiskSmartData)
NTSTATUS	DiskEnableDisableFailurePrediction (PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, BOOLEAN Enable)
NTSTATUS	DiskEnableDisableFailurePredictPolling (PFUNCTIONAL_DEVICE_EXTENSION FdoExtension, BOOLEAN Enable, ULONG PollTimeInSeconds)
NTSTATUS	DiskInitializeReregistration (VOID)
NTSTATUS	DiskDetermineMediaTypes (IN PDEVICE_OBJECT Fdo, IN PIRP Irp, IN UCHAR MediumType, IN UCHAR DensityCode, IN BOOLEAN MediaPresent, IN BOOLEAN IsWritable)
NTSTATUS	DiskIoctlGetLengthInfo (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS	DiskIoctlGetDriveGeometry (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlGetDriveGeometryEx (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS	DiskIoctlGetCacheInformation (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlSetCacheInformation (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlGetMediaTypesEx (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlPredictFailure (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlEnableFailurePrediction (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlVerify (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlReassignBlocks (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlReassignBlocksEx (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlIsWritable (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlSetVerify (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlClearVerify (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlUpdateDriveSize (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlGetVolumeDiskExtents (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlSmartGetVersion (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlSmartReceiveDriveData (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
NTSTATUS	DiskIoctlSmartSendDriveCommand (IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp)
FORCEINLINE PCDB	GetSrbScsiData (_In_ PSTORAGE_REQUEST_BLOCK SrbEx, _In_opt_ PUCHAR CdbLength8, _In_opt_ PULONG CdbLength32, _In_opt_ PUCHAR ScsiStatus, _In_opt_ PVOID *SenseInfoBuffer, _In_opt_ PUCHAR SenseInfoBufferLength)
FORCEINLINE VOID	SetSrbScsiData (_In_ PSTORAGE_REQUEST_BLOCK SrbEx, _In_ UCHAR CdbLength8, _In_ ULONG CdbLength32, _In_ UCHAR ScsiStatus, _In_opt_ PVOID SenseInfoBuffer, _In_ UCHAR SenseInfoBufferLength)

Variables
CLASSPNP_SCAN_FOR_SPECIAL_INFO	DiskBadControllers []
const DISK_MEDIA_TYPES_LIST	DiskMediaTypes []
const DISK_MEDIA_TYPES_LIST	DiskMediaTypesExclude []
DRIVER_INITIALIZE	DriverEntry
IO_WORKITEM_ROUTINE	DisableWriteCache
IO_WORKITEM_ROUTINE	DiskIoctlVerifyThread
IO_COMPLETION_ROUTINE	DiskFlushComplete
GUIDREGINFO	DiskWmiFdoGuidList []

Macro Definition Documentation

CHECK_IRQL

#define CHECK_IRQL

(

)

Value:

    if (KeGetCurrentIrql() >= DISPATCH_LEVEL) {         \
        NT_ASSERT(KeGetCurrentIrql() < DISPATCH_LEVEL);    \
        return STATUS_INVALID_LEVEL;                    \
    }

Definition at line 458 of file disk.h.

DEBUG_COMP_ID

#define DEBUG_COMP_ID   DPFLTR_DISK_ID

Definition at line 35 of file disk.h.

DISK_CACHE_MBR_CHECK

#define DISK_CACHE_MBR_CHECK   'mDcS'

Definition at line 63 of file disk.h.

DISK_DEFAULT_FAILURE_POLLING_PERIOD

#define DISK_DEFAULT_FAILURE_POLLING_PERIOD   1 * 60 * 60

Definition at line 456 of file disk.h.

DISK_TAG_CCONTEXT

#define DISK_TAG_CCONTEXT   'cDcS'

Definition at line 57 of file disk.h.

DISK_TAG_DISABLE_CACHE

#define DISK_TAG_DISABLE_CACHE   'CDcS'

Definition at line 56 of file disk.h.

DISK_TAG_DISK_GEOM

#define DISK_TAG_DISK_GEOM   'GDcS'

Definition at line 58 of file disk.h.

DISK_TAG_GENERAL

#define DISK_TAG_GENERAL   ' DcS'

Definition at line 53 of file disk.h.

DISK_TAG_INFO_EXCEPTION

#define DISK_TAG_INFO_EXCEPTION   'ADcS'

Definition at line 55 of file disk.h.

DISK_TAG_MODE_DATA

#define DISK_TAG_MODE_DATA   'MDcS'

Definition at line 62 of file disk.h.

DISK_TAG_NAME

#define DISK_TAG_NAME   'NDcS'

Definition at line 64 of file disk.h.

DISK_TAG_PART_LIST

#define DISK_TAG_PART_LIST   'pDcS'

Definition at line 66 of file disk.h.

DISK_TAG_PNP_ID

#define DISK_TAG_PNP_ID   'iDcS'

Definition at line 61 of file disk.h.

DISK_TAG_READ_CAP

#define DISK_TAG_READ_CAP   'PDcS'

Definition at line 65 of file disk.h.

DISK_TAG_SENSE_INFO

#define DISK_TAG_SENSE_INFO   'IDcS'

Definition at line 60 of file disk.h.

DISK_TAG_SMART

#define DISK_TAG_SMART   'aDcS'

Definition at line 54 of file disk.h.

DISK_TAG_SRB

#define DISK_TAG_SRB   'SDcS'

Definition at line 67 of file disk.h.

DISK_TAG_START

#define DISK_TAG_START   'sDcS'

Definition at line 68 of file disk.h.

DISK_TAG_UPDATE_CAP

#define DISK_TAG_UPDATE_CAP   'UDcS'

Definition at line 69 of file disk.h.

DISK_TAG_UPDATE_GEOM

#define DISK_TAG_UPDATE_GEOM   'gDcS'

Definition at line 59 of file disk.h.

DISK_TAG_WI_CONTEXT

#define DISK_TAG_WI_CONTEXT   'WDcS'

Definition at line 70 of file disk.h.

DiskDeviceCacheIsPowerProtected

#define DiskDeviceCacheIsPowerProtected   L"CacheIsPowerProtected"

Definition at line 405 of file disk.h.

DiskDeviceParameterSubkey

#define DiskDeviceParameterSubkey   L"Disk"

Definition at line 403 of file disk.h.

DiskDeviceUserWriteCacheSetting

#define DiskDeviceUserWriteCacheSetting   L"UserWriteCacheSetting"

Definition at line 404 of file disk.h.

DiskGetDetectInfo

#define DiskGetDetectInfo	(		FdoExtension,
			DetectInfo
	)		(STATUS_UNSUCCESSFUL)

Definition at line 861 of file disk.h.

DiskHashGuid

#define DiskHashGuid

(

Guid

)

(((PULONG) &Guid)[0] ^ ((PULONG) &Guid)[0] ^ ((PULONG) &Guid)[0] ^ ((PULONG) &Guid)[0])

Definition at line 865 of file disk.h.

DiskIsValidSmartSelfTest

#define DiskIsValidSmartSelfTest

(

Subcommand

)

Value:

    ( ((Subcommand) == SMART_OFFLINE_ROUTINE_OFFLINE) || \
      ((Subcommand) == SMART_SHORT_SELFTEST_OFFLINE) || \
      ((Subcommand) == SMART_EXTENDED_SELFTEST_OFFLINE) )

Definition at line 619 of file disk.h.

DiskReadDriveCapacity

#define DiskReadDriveCapacity

(

Fdo

)

ClassReadDriveCapacity(Fdo)

Definition at line 818 of file disk.h.

FUNCTIONAL_EXTENSION_SIZE

#define FUNCTIONAL_EXTENSION_SIZE   sizeof(FUNCTIONAL_DEVICE_EXTENSION) + sizeof(DISK_DATA)

Definition at line 408 of file disk.h.

HackCauseNotReportableHack

#define HackCauseNotReportableHack   (0x10)

Definition at line 399 of file disk.h.

HackDisableSpinDown

#define HackDisableSpinDown   (0x04)

Definition at line 397 of file disk.h.

HackDisableSynchronousTransfers

#define HackDisableSynchronousTransfers   (0x02)

Definition at line 396 of file disk.h.

HackDisableTaggedQueuing

#define HackDisableTaggedQueuing   (0x01)

Definition at line 395 of file disk.h.

HackDisableWriteCache

#define HackDisableWriteCache   (0x08)

Definition at line 398 of file disk.h.

HackRequiresStartUnitCommand

#define HackRequiresStartUnitCommand   (0x20)

Definition at line 400 of file disk.h.

MAX_MEDIA_TYPES

#define MAX_MEDIA_TYPES   4

Definition at line 415 of file disk.h.

MAX_SECTORS_PER_VERIFY

#define MAX_SECTORS_PER_VERIFY   0x100

Definition at line 435 of file disk.h.

MODE_DATA_SIZE

#define MODE_DATA_SIZE   192

Definition at line 410 of file disk.h.

ONE_MILLI_SECOND

#define ONE_MILLI_SECOND   ((ULONGLONG)10 * 1000)

Definition at line 440 of file disk.h.

PARTITION0_LIST_SIZE

#define PARTITION0_LIST_SIZE   4

Definition at line 413 of file disk.h.

SCSI_DISK_TIMEOUT

#define SCSI_DISK_TIMEOUT   10

Definition at line 412 of file disk.h.

VALUE_BUFFER_SIZE

#define VALUE_BUFFER_SIZE   2048

Definition at line 411 of file disk.h.

WPP_CONTROL_GUIDS

#define WPP_CONTROL_GUIDS   WPP_CONTROL_GUIDS_NORMAL_FLAGS(WPP_GUID_DISK)

Definition at line 44 of file disk.h.

WPP_GUID_DISK

#define WPP_GUID_DISK   (945186BF, 3DD6, 4f3f, 9C8E, 9EDD3FC9D558)

Definition at line 42 of file disk.h.

Typedef Documentation

DISK_DATA

typedef struct _DISK_DATA DISK_DATA

DISK_GROUP_CONTEXT

typedef struct _DISK_GROUP_CONTEXT DISK_GROUP_CONTEXT

DISK_MEDIA_TYPES_LIST

typedef struct _DISK_MEDIA_TYPES_LIST DISK_MEDIA_TYPES_LIST

DISK_USER_WRITE_CACHE_SETTING

typedef enum _DISK_USER_WRITE_CACHE_SETTING DISK_USER_WRITE_CACHE_SETTING

DISK_VERIFY_WORKITEM_CONTEXT

typedef struct _DISK_VERIFY_WORKITEM_CONTEXT DISK_VERIFY_WORKITEM_CONTEXT

PDISK_DATA

typedef struct _DISK_DATA * PDISK_DATA

PDISK_GROUP_CONTEXT

typedef struct _DISK_GROUP_CONTEXT * PDISK_GROUP_CONTEXT

PDISK_MEDIA_TYPES_LIST

typedef struct _DISK_MEDIA_TYPES_LIST * PDISK_MEDIA_TYPES_LIST

PDISK_USER_WRITE_CACHE_SETTING

typedef enum _DISK_USER_WRITE_CACHE_SETTING * PDISK_USER_WRITE_CACHE_SETTING

PDISK_VERIFY_WORKITEM_CONTEXT

typedef struct _DISK_VERIFY_WORKITEM_CONTEXT * PDISK_VERIFY_WORKITEM_CONTEXT

PDISKREREGREQUEST

typedef struct DISKREREGREQUEST * PDISKREREGREQUEST

Enumeration Type Documentation

_DISK_USER_WRITE_CACHE_SETTING

enum _DISK_USER_WRITE_CACHE_SETTING

Enumerator
DiskWriteCacheDisable
DiskWriteCacheEnable
DiskWriteCacheDefault

Definition at line 177 of file disk.h.

{
    DiskWriteCacheDisable =  0,
    DiskWriteCacheEnable  =  1,
    DiskWriteCacheDefault = -1
 
} DISK_USER_WRITE_CACHE_SETTING, *PDISK_USER_WRITE_CACHE_SETTING;

Function Documentation

DiskAddDevice()

NTSTATUS NTAPI DiskAddDevice	(	IN PDRIVER_OBJECT	DriverObject,
		IN PDEVICE_OBJECT	Pdo
	)

Definition at line 53 of file pnp.c.

{
    ULONG rootPartitionMountable = FALSE;
 
    PCONFIGURATION_INFORMATION configurationInformation;
    ULONG diskCount;
 
    NTSTATUS status;
 
    PAGED_CODE();
 
    //
    // See if we should be allowing file systems to mount on partition zero.
    //
 
    TRY {
        HANDLE deviceKey = NULL;
 
        UNICODE_STRING diskKeyName;
        OBJECT_ATTRIBUTES objectAttributes = {0};
        HANDLE diskKey;
 
        RTL_QUERY_REGISTRY_TABLE queryTable[2] = { 0 };
 
        status = IoOpenDeviceRegistryKey(PhysicalDeviceObject,
                                         PLUGPLAY_REGKEY_DEVICE,
                                         KEY_READ,
                                         &deviceKey);
 
        if(!NT_SUCCESS(status)) {
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_PNP, "DiskAddDevice: Error %#08lx opening device key "
                           "for pdo %p\n",
                        status, PhysicalDeviceObject));
            LEAVE;
        }
 
        RtlInitUnicodeString(&diskKeyName, L"Disk");
        InitializeObjectAttributes(&objectAttributes,
                                   &diskKeyName,
                                   OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                                   deviceKey,
                                   NULL);
 
        status = ZwOpenKey(&diskKey, KEY_READ, &objectAttributes);
        ZwClose(deviceKey);
 
        if(!NT_SUCCESS(status)) {
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_PNP, "DiskAddDevice: Error %#08lx opening disk key "
                           "for pdo %p device key %p\n",
                        status, PhysicalDeviceObject, deviceKey));
            LEAVE;
        }
 
        queryTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_QUERY_REGISTRY_TYPECHECK;
        queryTable[0].Name = L"RootPartitionMountable";
        queryTable[0].EntryContext = &(rootPartitionMountable);
        queryTable[0].DefaultType = (REG_DWORD << RTL_QUERY_REGISTRY_TYPECHECK_SHIFT) | REG_NONE;
 
#ifdef _MSC_VER
#pragma prefast(suppress:6309, "We don't have QueryRoutine so Context doesn't make any sense")
#endif
        status = RtlQueryRegistryValues(RTL_REGISTRY_HANDLE,
                                        diskKey,
                                        queryTable,
                                        NULL,
                                        NULL);
 
        if(!NT_SUCCESS(status)) {
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_PNP, "DiskAddDevice: Error %#08lx reading value from "
                           "disk key %p for pdo %p\n",
                        status, diskKey, PhysicalDeviceObject));
        }
 
        ZwClose(diskKey);
 
    } FINALLY {
 
        //
        // Do nothing.
        //
 
        if(!NT_SUCCESS(status)) {
            TracePrint((TRACE_LEVEL_WARNING, TRACE_FLAG_PNP, "DiskAddDevice: Will %sallow file system to mount on "
                           "partition zero of disk %p\n",
                        (rootPartitionMountable ? "" : "not "),
                        PhysicalDeviceObject));
        }
    }
 
    //
    // Create device objects for disk
    //
 
    diskCount = 0;
 
    status = DiskCreateFdo(
                 DriverObject,
                 PhysicalDeviceObject,
                 &diskCount,
                 (BOOLEAN) !rootPartitionMountable
                 );
 
    //
    // Get the number of disks already initialized.
    //
 
    configurationInformation = IoGetConfigurationInformation();
 
    if (NT_SUCCESS(status)) {
 
        //
        // Increment system disk device count.
        //
 
        configurationInformation->DiskCount++;
 
    }
 
    return status;
 
} // end DiskAddDevice()

Referenced by DriverEntry().

DiskCreateFdo()

NTSTATUS DiskCreateFdo	(	IN PDRIVER_OBJECT	DriverObject,
		IN PDEVICE_OBJECT	LowerDeviceObject,
		IN PULONG	DeviceCount,
		IN BOOLEAN	DasdAccessOnly
	)

Definition at line 307 of file disk.c.

{
    PCCHAR deviceName = NULL;
    HANDLE handle = NULL;
    PDEVICE_OBJECT lowerDevice  = NULL;
    PDEVICE_OBJECT deviceObject = NULL;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension;
    NTSTATUS status;
 
    PAGED_CODE();
 
    *DeviceCount = 0;
 
    //
    // Set up an object directory to contain the objects for this
    // device and all its partitions.
    //
 
    do {
 
        WCHAR dirBuffer[64] = { 0 };
        UNICODE_STRING dirName;
        OBJECT_ATTRIBUTES objectAttribs;
 
        status = RtlStringCchPrintfW(dirBuffer, sizeof(dirBuffer) / sizeof(dirBuffer[0]) - 1, L"\\Device\\Harddisk%d", *DeviceCount);
        if (!NT_SUCCESS(status)) {
            TracePrint((TRACE_LEVEL_FATAL, TRACE_FLAG_PNP, "DiskCreateFdo: Format symbolic link failed with error: 0x%X\n", status));
            return status;
        }
 
        RtlInitUnicodeString(&dirName, dirBuffer);
 
        InitializeObjectAttributes(&objectAttribs,
                                   &dirName,
                                   OBJ_CASE_INSENSITIVE | OBJ_PERMANENT | OBJ_KERNEL_HANDLE,
                                   NULL,
                                   NULL);
 
        status = ZwCreateDirectoryObject(&handle,
                                         DIRECTORY_ALL_ACCESS,
                                         &objectAttribs);
 
        (*DeviceCount)++;
 
    } while((status == STATUS_OBJECT_NAME_COLLISION) ||
            (status == STATUS_OBJECT_NAME_EXISTS));
 
    if (!NT_SUCCESS(status)) {
 
        TracePrint((TRACE_LEVEL_FATAL, TRACE_FLAG_PNP, "DiskCreateFdo: Could not create directory - %lx\n", status));
 
        return(status);
    }
 
    //
    // When this loop exits the count is inflated by one - fix that.
    //
 
    (*DeviceCount)--;
 
    //
    // Claim the device.
    //
 
    lowerDevice = IoGetAttachedDeviceReference(PhysicalDeviceObject);
 
    status = ClassClaimDevice(lowerDevice, FALSE);
 
    if (!NT_SUCCESS(status)) {
        ZwMakeTemporaryObject(handle);
        ZwClose(handle);
        ObDereferenceObject(lowerDevice);
        return status;
    }
 
    //
    // Create a device object for this device. Each physical disk will
    // have at least one device object. The required device object
    // describes the entire device. Its directory path is
    // \Device\HarddiskN\Partition0, where N = device number.
    //
 
    status = DiskGenerateDeviceName(*DeviceCount, &deviceName);
 
    if(!NT_SUCCESS(status)) {
        TracePrint((TRACE_LEVEL_FATAL, TRACE_FLAG_PNP, "DiskCreateFdo - couldn't create name %lx\n", status));
 
        goto DiskCreateFdoExit;
 
    }
 
    status = ClassCreateDeviceObject(DriverObject,
                                     deviceName,
                                     PhysicalDeviceObject,
                                     TRUE,
                                     &deviceObject);
 
    if (!NT_SUCCESS(status)) {
        TracePrint((TRACE_LEVEL_FATAL, TRACE_FLAG_PNP, "DiskCreateFdo: Can not create device object %s\n", deviceName));
        goto DiskCreateFdoExit;
    }
 
    FREE_POOL(deviceName);
 
    //
    // Indicate that IRPs should include MDLs for data transfers.
    //
 
    SET_FLAG(deviceObject->Flags, DO_DIRECT_IO);
 
    fdoExtension = deviceObject->DeviceExtension;
 
    if(DasdAccessOnly) {
 
        //
        // Inidicate that only RAW should be allowed to mount on the root
        // partition object.  This ensures that a file system can't doubly
        // mount on a super-floppy by mounting once on P0 and once on P1.
        //
 
#ifdef _MSC_VER
#pragma prefast(suppress:28175);
#endif
        SET_FLAG(deviceObject->Vpb->Flags, VPB_RAW_MOUNT);
    }
 
    //
    // Initialize lock count to zero. The lock count is used to
    // disable the ejection mechanism on devices that support
    // removable media. Only the lock count in the physical
    // device extension is used.
    //
 
    fdoExtension->LockCount = 0;
 
    //
    // Save system disk number.
    //
 
    fdoExtension->DeviceNumber = *DeviceCount;
 
    //
    // Set the alignment requirements for the device based on the
    // host adapter requirements
    //
 
    if (lowerDevice->AlignmentRequirement > deviceObject->AlignmentRequirement) {
        deviceObject->AlignmentRequirement = lowerDevice->AlignmentRequirement;
    }
 
    //
    // Finally, attach to the pdo
    //
 
    fdoExtension->LowerPdo = PhysicalDeviceObject;
 
    fdoExtension->CommonExtension.LowerDeviceObject =
        IoAttachDeviceToDeviceStack(deviceObject, PhysicalDeviceObject);
 
 
    if(fdoExtension->CommonExtension.LowerDeviceObject == NULL) {
 
        //
        // Uh - oh, we couldn't attach
        // cleanup and return
        //
 
        status = STATUS_UNSUCCESSFUL;
        goto DiskCreateFdoExit;
    }
 
    //
    // Clear the init flag.
    //
 
    CLEAR_FLAG(deviceObject->Flags, DO_DEVICE_INITIALIZING);
 
    //
    // Store a handle to the device object directory for this disk
    //
 
    fdoExtension->DeviceDirectory = handle;
 
    ObDereferenceObject(lowerDevice);
 
    return STATUS_SUCCESS;
 
DiskCreateFdoExit:
 
    if (deviceObject != NULL)
    {
        IoDeleteDevice(deviceObject);
    }
 
    FREE_POOL(deviceName);
 
    ObDereferenceObject(lowerDevice);
 
    ZwMakeTemporaryObject(handle);
    ZwClose(handle);
 
    return status;
}

Referenced by DiskAddDevice().

DiskCreateSymbolicLinks()

VOID DiskCreateSymbolicLinks

(

IN PDEVICE_OBJECT

DeviceObject

)

Definition at line 676 of file pnp.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = commonExtension->DriverData;
 
    WCHAR wideSourceName[64] = { 0 };
    UNICODE_STRING unicodeSourceName;
 
    NTSTATUS status;
 
    PAGED_CODE();
 
    //
    // Build the destination for the link first using the device name
    // stored in the device object
    //
 
    NT_ASSERT(commonExtension->DeviceName.Buffer);
 
    if(!diskData->LinkStatus.WellKnownNameCreated) {
        //
        // Put together the source name using the partition and device number
        // in the device extension and disk data segment
        //
 
        status = RtlStringCchPrintfW(wideSourceName, sizeof(wideSourceName) / sizeof(wideSourceName[0]) - 1,
                                     L"\\Device\\Harddisk%d\\Partition0",
                                     commonExtension->PartitionZeroExtension->DeviceNumber);
 
        if (NT_SUCCESS(status)) {
 
            RtlInitUnicodeString(&unicodeSourceName, wideSourceName);
 
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_PNP, "DiskCreateSymbolicLink: Linking %wZ to %wZ\n",
                       &unicodeSourceName,
                       &commonExtension->DeviceName));
 
            status = IoCreateSymbolicLink(&unicodeSourceName,
                                          &commonExtension->DeviceName);
 
            if(NT_SUCCESS(status)){
                diskData->LinkStatus.WellKnownNameCreated = TRUE;
            }
        }
    }
 
    if (!diskData->LinkStatus.PhysicalDriveLinkCreated) {
 
        //
        // Create a physical drive N link using the device number we saved
        // away during AddDevice.
        //
 
        status = RtlStringCchPrintfW(wideSourceName, sizeof(wideSourceName) / sizeof(wideSourceName[0]) - 1,
                                     L"\\DosDevices\\PhysicalDrive%d",
                                     commonExtension->PartitionZeroExtension->DeviceNumber);
        if (NT_SUCCESS(status)) {
 
            RtlInitUnicodeString(&unicodeSourceName, wideSourceName);
 
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_PNP, "DiskCreateSymbolicLink: Linking %wZ to %wZ\n",
                        &unicodeSourceName,
                        &(commonExtension->DeviceName)));
 
            status = IoCreateSymbolicLink(&unicodeSourceName,
                                          &(commonExtension->DeviceName));
 
            if(NT_SUCCESS(status)) {
                diskData->LinkStatus.PhysicalDriveLinkCreated = TRUE;
            }
        }
    }
 
 
    return;
}

Referenced by DiskInitFdo().

DiskDeleteSymbolicLinks()

VOID DiskDeleteSymbolicLinks

(

IN PDEVICE_OBJECT

DeviceObject

)

Definition at line 780 of file pnp.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = commonExtension->DriverData;
 
    WCHAR wideLinkName[64] = { 0 };
    UNICODE_STRING unicodeLinkName;
    NTSTATUS status;
 
    PAGED_CODE();
 
    if(diskData->LinkStatus.WellKnownNameCreated) {
 
        status = RtlStringCchPrintfW(wideLinkName, sizeof(wideLinkName) / sizeof(wideLinkName[0]) - 1,
                                    L"\\Device\\Harddisk%d\\Partition0",
                                    commonExtension->PartitionZeroExtension->DeviceNumber);
        if (NT_SUCCESS(status)) {
            RtlInitUnicodeString(&unicodeLinkName, wideLinkName);
            IoDeleteSymbolicLink(&unicodeLinkName);
        }
        diskData->LinkStatus.WellKnownNameCreated = FALSE;
    }
 
    if(diskData->LinkStatus.PhysicalDriveLinkCreated) {
 
        status = RtlStringCchPrintfW(wideLinkName, sizeof(wideLinkName) / sizeof(wideLinkName[0]) - 1,
                                    L"\\DosDevices\\PhysicalDrive%d",
                                    commonExtension->PartitionZeroExtension->DeviceNumber);
        if (NT_SUCCESS(status)) {
            RtlInitUnicodeString(&unicodeLinkName, wideLinkName);
            IoDeleteSymbolicLink(&unicodeLinkName);
        }
        diskData->LinkStatus.PhysicalDriveLinkCreated = FALSE;
    }
 
 
    return;
}

Referenced by DiskRemoveDevice().

DiskDetectFailurePrediction()

NTSTATUS DiskDetectFailurePrediction	(	PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		PFAILURE_PREDICTION_METHOD	FailurePredictCapability,
		BOOLEAN	ScsiAddressAvailable
	)

Definition at line 2230 of file diskwmi.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = (PCOMMON_DEVICE_EXTENSION)FdoExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    BOOLEAN supportFP;
    NTSTATUS status;
    STORAGE_PREDICT_FAILURE checkFailure;
    STORAGE_FAILURE_PREDICT_STATUS diskSmartStatus;
 
    PAGED_CODE();
 
    //
    // Assume no failure predict mechanisms
    //
    *FailurePredictCapability = FailurePredictionNone;
 
    //
    // See if this is an IDE drive that supports SMART. If so enable SMART
    // and then ensure that it suports the SMART READ STATUS command
    //
 
    if (ScsiAddressAvailable)
    {
        DiskGetIdentifyInfo(FdoExtension, &supportFP);
 
        if (supportFP)
        {
            status = DiskEnableSmart(FdoExtension);
            if (NT_SUCCESS(status))
            {
                *FailurePredictCapability = FailurePredictionSmart;
                diskData->FailurePredictionEnabled = TRUE;
 
                status = DiskReadFailurePredictStatus(FdoExtension,
                                                  &diskSmartStatus);
 
                TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: Device %p %s IDE SMART\n",
                           FdoExtension->DeviceObject,
                           NT_SUCCESS(status) ? "does" : "does not"));
 
                if (!NT_SUCCESS(status))
                {
                    *FailurePredictCapability = FailurePredictionNone;
                    diskData->FailurePredictionEnabled = FALSE;
                }
            }
            return(status);
        }
    }
    //
    // See if there is a a filter driver to intercept
    // IOCTL_STORAGE_PREDICT_FAILURE
    //
    status = DiskSendFailurePredictIoctl(FdoExtension,
                                         &checkFailure);
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: Device %p %s IOCTL_STORAGE_FAILURE_PREDICT\n",
                       FdoExtension->DeviceObject,
                       NT_SUCCESS(status) ? "does" : "does not"));
 
    if (NT_SUCCESS(status))
    {
        *FailurePredictCapability = FailurePredictionIoctl;
        diskData->FailurePredictionEnabled = TRUE;
        if (checkFailure.PredictFailure)
        {
            checkFailure.PredictFailure = 512;
            ClassNotifyFailurePredicted(FdoExtension,
                                        (PUCHAR)&checkFailure,
                                        sizeof(checkFailure),
                                        (BOOLEAN)(FdoExtension->FailurePredicted == FALSE),
                                        0x11,
                                        diskData->ScsiAddress.PathId,
                                        diskData->ScsiAddress.TargetId,
                                        diskData->ScsiAddress.Lun);
 
            FdoExtension->FailurePredicted = TRUE;
        }
        return(status);
    }
 
    //
    // Finally we assume it will not be a scsi smart drive. but
    // we'll also send off an asynchronous mode sense so that if
    // it is SMART we'll reregister the device object
    //
 
    *FailurePredictCapability = FailurePredictionNone;
 
    DiskInfoExceptionCheck(FdoExtension);
 
    return(STATUS_SUCCESS);
}

Referenced by DiskInitFdo().

DiskDetermineMediaTypes()

NTSTATUS DiskDetermineMediaTypes	(	IN PDEVICE_OBJECT	Fdo,
		IN PIRP	Irp,
		IN UCHAR	MediumType,
		IN UCHAR	DensityCode,
		IN BOOLEAN	MediaPresent,
		IN BOOLEAN	IsWritable
	)

Definition at line 655 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = Fdo->DeviceExtension;
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation(Irp);
 
    PGET_MEDIA_TYPES  mediaTypes = Irp->AssociatedIrp.SystemBuffer;
    PDEVICE_MEDIA_INFO mediaInfo = &mediaTypes->MediaInfo[0];
    BOOLEAN deviceMatched = FALSE;
 
    PAGED_CODE();
 
    //
    // this should be checked prior to calling into this routine
    // as we use the buffer as mediaTypes
    //
 
    NT_ASSERT(irpStack->Parameters.DeviceIoControl.OutputBufferLength >=
           sizeof(GET_MEDIA_TYPES));
 
    //
    // Determine if this device is removable or fixed.
    //
 
    if (!TEST_FLAG(Fdo->Characteristics, FILE_REMOVABLE_MEDIA)) {
 
        //
        // Fixed disk.
        //
 
        mediaTypes->DeviceType = FILE_DEVICE_DISK;
        mediaTypes->MediaInfoCount = 1;
 
        mediaInfo->DeviceSpecific.DiskInfo.Cylinders.QuadPart   = fdoExtension->DiskGeometry.Cylinders.QuadPart;
        mediaInfo->DeviceSpecific.DiskInfo.MediaType            = (STORAGE_MEDIA_TYPE)FixedMedia;
        mediaInfo->DeviceSpecific.DiskInfo.TracksPerCylinder    = fdoExtension->DiskGeometry.TracksPerCylinder;
        mediaInfo->DeviceSpecific.DiskInfo.SectorsPerTrack      = fdoExtension->DiskGeometry.SectorsPerTrack;
        mediaInfo->DeviceSpecific.DiskInfo.BytesPerSector       = fdoExtension->DiskGeometry.BytesPerSector;
        mediaInfo->DeviceSpecific.DiskInfo.NumberMediaSides     = 1;
        mediaInfo->DeviceSpecific.DiskInfo.MediaCharacteristics = (MEDIA_CURRENTLY_MOUNTED | MEDIA_READ_WRITE);
 
        if (!IsWritable) {
 
            SET_FLAG(mediaInfo->DeviceSpecific.DiskInfo.MediaCharacteristics,
                     MEDIA_WRITE_PROTECTED);
        }
 
    } else {
 
        PCCHAR vendorId = (PCCHAR) fdoExtension->DeviceDescriptor + fdoExtension->DeviceDescriptor->VendorIdOffset;
        PCCHAR productId = (PCCHAR) fdoExtension->DeviceDescriptor + fdoExtension->DeviceDescriptor->ProductIdOffset;
        PCCHAR productRevision = (PCCHAR) fdoExtension->DeviceDescriptor + fdoExtension->DeviceDescriptor->ProductRevisionOffset;
        DISK_MEDIA_TYPES_LIST const *mediaListEntry;
        ULONG  currentMedia;
        ULONG  i;
        ULONG  j;
        ULONG  sizeNeeded;
 
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL,
                   "DiskDetermineMediaTypes: Vendor %s, Product %s\n",
                   vendorId,
                   productId));
 
 
        //
        // If there's an entry with such vendorId & ProductId in the DiskMediaTypesExclude list,
        // this device shouldn't be looked up in the DiskMediaTypes list to determine a medium type.
        // The exclude table allows to narrow down the set of devices described by the DiskMediaTypes
        // list (e.g.: DiskMediaTypes says "all HP devices" and DiskMediaTypesExlclude says
        // "except for HP RDX")
        //
 
        for (i = 0; DiskMediaTypesExclude[i].VendorId != NULL; i++) {
            mediaListEntry = &DiskMediaTypesExclude[i];
 
            if (strncmp(mediaListEntry->VendorId,vendorId,strlen(mediaListEntry->VendorId))) {
                continue;
            }
 
            if ((mediaListEntry->ProductId != NULL) &&
                 strncmp(mediaListEntry->ProductId, productId, strlen(mediaListEntry->ProductId))) {
                continue;
            }
 
            goto SkipTable;
        }
 
        //
        // Run through the list until we find the entry with a NULL Vendor Id.
        //
 
        for (i = 0; DiskMediaTypes[i].VendorId != NULL; i++) {
 
            mediaListEntry = &DiskMediaTypes[i];
 
            if (strncmp(mediaListEntry->VendorId,vendorId,strlen(mediaListEntry->VendorId))) {
                continue;
            }
 
            if ((mediaListEntry->ProductId != NULL) &&
                 strncmp(mediaListEntry->ProductId, productId, strlen(mediaListEntry->ProductId))) {
                continue;
            }
 
            if ((mediaListEntry->Revision != NULL) &&
                 strncmp(mediaListEntry->Revision, productRevision, strlen(mediaListEntry->Revision))) {
                continue;
            }
 
            deviceMatched = TRUE;
 
            mediaTypes->DeviceType = FILE_DEVICE_DISK;
            mediaTypes->MediaInfoCount = mediaListEntry->NumberOfTypes;
 
            //
            // Ensure that buffer is large enough.
            //
 
            sizeNeeded = FIELD_OFFSET(GET_MEDIA_TYPES, MediaInfo[0]) +
                         (mediaListEntry->NumberOfTypes *
                          sizeof(DEVICE_MEDIA_INFO)
                          );
 
            if (irpStack->Parameters.DeviceIoControl.OutputBufferLength <
                sizeNeeded) {
 
                //
                // Buffer too small
                //
 
                Irp->IoStatus.Status = STATUS_BUFFER_TOO_SMALL;
                return STATUS_BUFFER_TOO_SMALL;
            }
 
            for (j = 0; j < mediaListEntry->NumberOfTypes; j++) {
 
                mediaInfo->DeviceSpecific.RemovableDiskInfo.Cylinders.QuadPart = fdoExtension->DiskGeometry.Cylinders.QuadPart;
                mediaInfo->DeviceSpecific.RemovableDiskInfo.TracksPerCylinder = fdoExtension->DiskGeometry.TracksPerCylinder;
                mediaInfo->DeviceSpecific.RemovableDiskInfo.SectorsPerTrack = fdoExtension->DiskGeometry.SectorsPerTrack;
                mediaInfo->DeviceSpecific.RemovableDiskInfo.BytesPerSector = fdoExtension->DiskGeometry.BytesPerSector;
                mediaInfo->DeviceSpecific.RemovableDiskInfo.NumberMediaSides = mediaListEntry->NumberOfSides;
 
                //
                // Set the type.
                //
 
                mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaType = mediaListEntry->MediaTypes[j];
 
                if (mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaType == MO_5_WO) {
                    mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics = MEDIA_WRITE_ONCE;
                } else {
                    mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics = MEDIA_READ_WRITE;
                }
 
                //
                // Status will either be success, if media is present, or no media.
                // It would be optimal to base from density code and medium type, but not all devices
                // have values for these fields.
                //
 
                if (MediaPresent) {
 
                    //
                    // The usage of MediumType and DensityCode is device specific, so this may need
                    // to be extended to further key off of product/vendor ids.
                    // Currently, the MO units are the only devices that return this information.
                    //
 
                    if (MediumType == 2) {
                        currentMedia = MO_5_WO;
                    } else if (MediumType == 3) {
                        currentMedia = MO_5_RW;
 
                        if (DensityCode == 0x87) {
 
                            //
                            // Indicate that the pinnacle 4.6 G media
                            // is present. Other density codes will default to normal
                            // RW MO media.
                            //
 
                            currentMedia = PINNACLE_APEX_5_RW;
                        }
                    } else {
                        currentMedia = 0;
                    }
 
                    if (currentMedia) {
                        if (mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaType == (STORAGE_MEDIA_TYPE)currentMedia) {
                            SET_FLAG(mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics, MEDIA_CURRENTLY_MOUNTED);
                        }
 
                    } else {
                        SET_FLAG(mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics, MEDIA_CURRENTLY_MOUNTED);
                    }
                }
 
                if (!IsWritable) {
                    SET_FLAG(mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics, MEDIA_WRITE_PROTECTED);
                }
 
                //
                // Advance to next entry.
                //
 
                mediaInfo++;
            }
        }
 
SkipTable:
 
        if (!deviceMatched) {
 
            TracePrint((TRACE_LEVEL_WARNING, TRACE_FLAG_IOCTL,
                       "DiskDetermineMediaTypes: Unknown device. Vendor: %s Product: %s Revision: %s\n",
                                   vendorId,
                                   productId,
                                   productRevision));
            //
            // Build an entry for unknown.
            //
 
            mediaTypes->DeviceType = FILE_DEVICE_DISK;
            mediaTypes->MediaInfoCount = 1;
 
            mediaInfo->DeviceSpecific.RemovableDiskInfo.Cylinders.QuadPart   = fdoExtension->DiskGeometry.Cylinders.QuadPart;
            mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaType            = (STORAGE_MEDIA_TYPE)RemovableMedia;
            mediaInfo->DeviceSpecific.RemovableDiskInfo.TracksPerCylinder    = fdoExtension->DiskGeometry.TracksPerCylinder;
            mediaInfo->DeviceSpecific.RemovableDiskInfo.SectorsPerTrack      = fdoExtension->DiskGeometry.SectorsPerTrack;
            mediaInfo->DeviceSpecific.RemovableDiskInfo.BytesPerSector       = fdoExtension->DiskGeometry.BytesPerSector;
            mediaInfo->DeviceSpecific.RemovableDiskInfo.NumberMediaSides     = 1;
            mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics = MEDIA_READ_WRITE;
 
            if (MediaPresent) {
 
                SET_FLAG(mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics, MEDIA_CURRENTLY_MOUNTED);
            }
 
            if (!IsWritable) {
 
                SET_FLAG(mediaInfo->DeviceSpecific.RemovableDiskInfo.MediaCharacteristics, MEDIA_WRITE_PROTECTED);
            }
        }
    }
 
    Irp->IoStatus.Information =
        FIELD_OFFSET(GET_MEDIA_TYPES, MediaInfo[0]) +
        (mediaTypes->MediaInfoCount * sizeof(DEVICE_MEDIA_INFO));
 
    return STATUS_SUCCESS;
}

Referenced by DiskIoctlGetMediaTypesEx().

DiskDeviceControl()

NTSTATUS NTAPI DiskDeviceControl	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

DiskEnableDisableFailurePrediction()

NTSTATUS DiskEnableDisableFailurePrediction	(	PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		BOOLEAN	Enable
	)

Definition at line 1126 of file diskwmi.c.

{
    NTSTATUS status;
    PCOMMON_DEVICE_EXTENSION commonExtension = &(FdoExtension->CommonExtension);
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
 
    PAGED_CODE();
 
    switch(diskData->FailurePredictionCapability)
    {
        case FailurePredictionSmart:
        {
            if (Enable)
            {
                status = DiskEnableSmart(FdoExtension);
            } else {
                status = DiskDisableSmart(FdoExtension);
            }
 
            if (NT_SUCCESS(status)) {
                diskData->FailurePredictionEnabled = Enable;
            }
 
            break;
        }
 
        case  FailurePredictionSense:
        case  FailurePredictionIoctl:
        {
            //
            // We assume that the drive is already setup properly for
            // failure prediction
            //
            status = STATUS_SUCCESS;
            break;
        }
 
        default:
        {
            status = STATUS_INVALID_DEVICE_REQUEST;
        }
    }
    return status;
}

Referenced by DiskEnableDisableFailurePredictPolling(), DiskFdoExecuteWmiMethod(), DiskIoctlEnableFailurePrediction(), and DiskWmiFunctionControl().

DiskEnableDisableFailurePredictPolling()

NTSTATUS DiskEnableDisableFailurePredictPolling	(	PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		BOOLEAN	Enable,
		ULONG	PollTimeInSeconds
	)

Definition at line 1193 of file diskwmi.c.

{
    NTSTATUS status;
    PCOMMON_DEVICE_EXTENSION commonExtension = (PCOMMON_DEVICE_EXTENSION)FdoExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
 
    PAGED_CODE();
 
    if (Enable)
    {
        status = DiskEnableDisableFailurePrediction(FdoExtension,
                                           Enable);
    } else {
        status = STATUS_SUCCESS;
    }
 
    if (NT_SUCCESS(status))
    {
        status = ClassSetFailurePredictionPoll(FdoExtension,
                        Enable ? diskData->FailurePredictionCapability :
                                 FailurePredictionNone,
                                     PollTimeInSeconds);
 
        //
        // Even if this failed we do not want to disable FP on the
        // hardware. FP is only ever disabled on the hardware by
        // specific command of the user.
        //
    }
 
    return status;
}

Referenced by DiskFdoExecuteWmiMethod(), DiskInitFdo(), and DiskWmiFunctionControl().

DiskEnableInfoExceptions()

NTSTATUS DiskEnableInfoExceptions	(	_In_ PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		_In_ BOOLEAN	Enable
	)

Definition at line 979 of file diskwmi.c.

{
    PDISK_DATA diskData = (PDISK_DATA)(FdoExtension->CommonExtension.DriverData);
    NTSTATUS status = STATUS_NOT_SUPPORTED;
    PMODE_PARAMETER_HEADER modeData;
    PMODE_INFO_EXCEPTIONS pageData;
    MODE_INFO_EXCEPTIONS changeablePageData;
    ULONG modeDataSize;
 
    PAGED_CODE();
 
    modeDataSize = MODE_DATA_SIZE;
 
    modeData = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                         modeDataSize,
                                         DISK_TAG_INFO_EXCEPTION);
 
    if (modeData == NULL) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_WMI, "DiskEnableInfoExceptions: Unable to allocate mode "
                       "data buffer\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    //
    // First see which data is actually changeable.
    //
    status = DiskGetModePage(FdoExtension->DeviceObject,
                             MODE_PAGE_FAULT_REPORTING,
                             1, // Page Control = 1 indicates we want changeable values.
                             modeData,
                             &modeDataSize,
                             (PVOID*)&pageData);
 
    if (!NT_SUCCESS(status) || pageData == NULL) {
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskEnableInfoExceptions: does NOT support SMART for device %p\n",
                  FdoExtension->DeviceObject));
        FREE_POOL(modeData);
        return STATUS_NOT_SUPPORTED;
    }
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskEnableInfoExceptions: DOES support SMART for device %p\n",
                FdoExtension->DeviceObject));
 
    //
    // At the very least, the DEXCPT bit must be changeable.
    // If it's not, bail out now.
    //
    if (pageData->Dexcpt == 0) {
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskEnableInfoExceptions: does NOT support DEXCPT bit for device %p\n",
                  FdoExtension->DeviceObject));
        FREE_POOL(modeData);
        return STATUS_NOT_SUPPORTED;
    }
 
    //
    // Cache away which values are changeable.
    //
    RtlCopyMemory(&changeablePageData, pageData, sizeof(MODE_INFO_EXCEPTIONS));
 
    //
    // Now get the current values.
    //
    status = DiskGetModePage(FdoExtension->DeviceObject,
                             MODE_PAGE_FAULT_REPORTING,
                             0, // Page Control = 0 indicates we want current values.
                             modeData,
                             &modeDataSize,
                             (PVOID*)&pageData);
 
    if (!NT_SUCCESS(status) || pageData == NULL) {
        //
        // At this point we know the device supports this mode page so
        // assert if something goes wrong here.
        //
        NT_ASSERT(NT_SUCCESS(status) && pageData);
        FREE_POOL(modeData);
        return STATUS_NOT_SUPPORTED;
    }
 
    //
    // If the device is currently configured to not report any informational
    // exceptions and we cannot change the value of that field, there's
    // nothing to be done.
    //
    if (pageData->ReportMethod == 0 && changeablePageData.ReportMethod == 0) {
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskEnableInfoExceptions: MRIE field is 0 and is not changeable for device %p\n",
                  FdoExtension->DeviceObject));
        FREE_POOL(modeData);
        return STATUS_NOT_SUPPORTED;
    }
 
    //
    // If the PERF bit is changeable, set it now.
    //
    if (changeablePageData.Perf) {
        pageData->Perf = diskData->AllowFPPerfHit ? 0 : 1;
    }
 
    //
    // If the MRIE field is changeable, set it to 4 so that informational
    // exceptions get reported with the "Recovered Error" sense key.
    //
    if (changeablePageData.ReportMethod) {
        pageData->ReportMethod = 4;
    }
 
    //
    // Finally, set the DEXCPT bit appropriately to enable/disable
    // informational exception reporting and send the Mode Select.
    //
    pageData->Dexcpt = !Enable;
 
    status = ClassModeSelect(FdoExtension->DeviceObject,
                                (PCHAR)modeData,
                                modeDataSize,
                                pageData->PSBit);
 
    //
    // Update the failure prediction state.  Note that for this particular
    // mode FailurePredictionNone is used when it's not enabled.
    //
    if (NT_SUCCESS(status)) {
        if (Enable) {
            diskData->FailurePredictionCapability = FailurePredictionSense;
            diskData->FailurePredictionEnabled = TRUE;
        } else {
            diskData->FailurePredictionCapability = FailurePredictionNone;
            diskData->FailurePredictionEnabled = FALSE;
        }
    }
 
    FREE_POOL(modeData);
 
    return status;
}

Referenced by DiskIoctlEnableFailurePrediction().

DiskFdoExecuteWmiMethod()

NTSTATUS NTAPI DiskFdoExecuteWmiMethod	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp,
		IN ULONG	GuidIndex,
		IN ULONG	MethodId,
		IN ULONG	InBufferSize,
		IN ULONG	OutBufferSize,
		IN PUCHAR	Buffer
	)

Definition at line 3045 of file diskwmi.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    ULONG sizeNeeded = 0;
    NTSTATUS status;
 
    PAGED_CODE();
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskExecuteWmiMethod, DeviceObject %p, Irp %p, Guid Id %d, MethodId %d\n"
             "      InBufferSize %#x, OutBufferSize %#x, Buffer %p\n",
             DeviceObject, Irp,
             GuidIndex, MethodId, InBufferSize, OutBufferSize, Buffer));
 
    switch(GuidIndex)
    {
        case SmartPerformFunction:
        {
 
            NT_ASSERT((diskData->FailurePredictionCapability ==
                                                  FailurePredictionSmart) ||
                   (diskData->FailurePredictionCapability ==
                                                  FailurePredictionIoctl) ||
                   (diskData->FailurePredictionCapability ==
                                                  FailurePredictionSense));
 
 
            switch(MethodId)
            {
                //
                // void AllowPerformanceHit([in] boolean Allow)
                //
                case AllowDisallowPerformanceHit:
                {
                    BOOLEAN allowPerfHit;
 
                    sizeNeeded = 0;
                    if (InBufferSize >= sizeof(BOOLEAN))
                    {
                        status = STATUS_SUCCESS;
 
                        allowPerfHit = *((PBOOLEAN)Buffer);
                        if (diskData->AllowFPPerfHit != allowPerfHit)
                        {
                            diskData->AllowFPPerfHit = allowPerfHit;
                            if (diskData->FailurePredictionCapability ==
                                FailurePredictionSense)
                            {
                                MODE_INFO_EXCEPTIONS modeInfo;
 
                                status = DiskGetInfoExceptionInformation(fdoExtension,
                                                                         &modeInfo);
                                if (NT_SUCCESS(status))
                                {
                                    modeInfo.Perf = allowPerfHit ? 0 : 1;
                                    status = DiskSetInfoExceptionInformation(fdoExtension,
                                                                             &modeInfo);
                                }
                            }
                            else
                            {
                                status = STATUS_INVALID_DEVICE_REQUEST;
                            }
                        }
 
                        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskFdoWmiExecuteMethod: AllowPerformanceHit %x for device %p --> %lx\n",
                                    allowPerfHit,
                                    fdoExtension->DeviceObject,
                                    status));
                    } else {
                        status = STATUS_INVALID_PARAMETER;
                    }
                    break;
                }
 
                //
                // void EnableDisableHardwareFailurePrediction([in] boolean Enable)
                //
                case EnableDisableHardwareFailurePrediction:
                {
                    BOOLEAN enable;
 
                    sizeNeeded = 0;
                    if (InBufferSize >= sizeof(BOOLEAN))
                    {
                        status = STATUS_SUCCESS;
                        enable = *((PBOOLEAN)Buffer);
                        if (!enable)
                        {
                            //
                            // If we are disabling we need to also disable
                            // polling
                            //
                            DiskEnableDisableFailurePredictPolling(
                                                               fdoExtension,
                                                               enable,
                                                               0);
                        }
 
                        status = DiskEnableDisableFailurePrediction(
                                                           fdoExtension,
                                                           enable);
 
                        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskFdoWmiExecuteMethod: EnableDisableHardwareFailurePrediction: %x for device %p --> %lx\n",
                                    enable,
                                    fdoExtension->DeviceObject,
                                    status));
                    } else {
                        status = STATUS_INVALID_PARAMETER;
                    }
                    break;
                }
 
                //
                // void EnableDisableFailurePredictionPolling(
                //                               [in] uint32 Period,
                //                               [in] boolean Enable)
                //
                case EnableDisableFailurePredictionPolling:
                {
                    BOOLEAN enable;
                    ULONG period;
 
                    sizeNeeded = 0;
                    if (InBufferSize >= (sizeof(ULONG) + sizeof(BOOLEAN)))
                    {
                        period = *((PULONG)Buffer);
                        Buffer += sizeof(ULONG);
                        enable = *((PBOOLEAN)Buffer);
 
                           status = DiskEnableDisableFailurePredictPolling(
                                                               fdoExtension,
                                                               enable,
                                                               period);
 
                           TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskFdoWmiExecuteMethod: EnableDisableFailurePredictionPolling: %x %x for device %p --> %lx\n",
                                    enable,
                                    period,
                                    fdoExtension->DeviceObject,
                                    status));
                    } else {
                        status = STATUS_INVALID_PARAMETER;
                    }
                    break;
                }
 
                //
                // void GetFailurePredictionCapability([out] uint32 Capability)
                //
                case GetFailurePredictionCapability:
                {
                    sizeNeeded = sizeof(ULONG);
                    if (OutBufferSize >= sizeNeeded)
                    {
                        status = STATUS_SUCCESS;
                        *((PFAILURE_PREDICTION_METHOD)Buffer) = diskData->FailurePredictionCapability;
                        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskFdoWmiExecuteMethod: GetFailurePredictionCapability: %x for device %p --> %lx\n",
                                    *((PFAILURE_PREDICTION_METHOD)Buffer),
                                    fdoExtension->DeviceObject,
                                    status));
                    } else {
                        status = STATUS_BUFFER_TOO_SMALL;
                    }
                    break;
                }
 
                //
                // void EnableOfflineDiags([out] boolean Success);
                //
                case EnableOfflineDiags:
                {
                    sizeNeeded = sizeof(BOOLEAN);
                    if (OutBufferSize >= sizeNeeded)
                    {
                        if (diskData->FailurePredictionCapability ==
                                  FailurePredictionSmart)
                        {
                            //
                            // Initiate or resume offline diagnostics.
                            // This may cause a loss of performance
                            // to the disk, but mayincrease the amount
                            // of disk checking.
                            //
                            status = DiskExecuteSmartDiagnostics(fdoExtension,
                                                                0);
 
                        } else {
                            status = STATUS_INVALID_DEVICE_REQUEST;
                        }
 
                        *((PBOOLEAN)Buffer) = NT_SUCCESS(status);
 
                        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskFdoWmiExecuteMethod: EnableOfflineDiags for device %p --> %lx\n",
                                    fdoExtension->DeviceObject,
                                    status));
                    } else {
                        status = STATUS_BUFFER_TOO_SMALL;
                    }
                    break;
                }
 
                //
                //    void ReadLogSectors([in] uint8 LogAddress,
                //        [in] uint8 SectorCount,
                //        [out] uint32 Length,
                //        [out, WmiSizeIs("Length")] uint8 LogSectors[]
                //       );
                //
                case ReadLogSectors:
                {
                    if (diskData->FailurePredictionCapability ==
                                  FailurePredictionSmart)
                    {
                        if (InBufferSize >= sizeof(READ_LOG_SECTORS_IN))
                        {
                            PREAD_LOG_SECTORS_IN inParams;
                            PREAD_LOG_SECTORS_OUT outParams;
                            ULONG readSize;
 
                            inParams = (PREAD_LOG_SECTORS_IN)Buffer;
                            readSize = inParams->SectorCount * SMART_LOG_SECTOR_SIZE;
                            sizeNeeded = FIELD_OFFSET(READ_LOG_SECTORS_OUT,
                                                  LogSectors) + readSize;
 
                            if (OutBufferSize >= sizeNeeded)
                            {
                                outParams = (PREAD_LOG_SECTORS_OUT)Buffer;
                                status = DiskReadSmartLog(fdoExtension,
                                                        inParams->SectorCount,
                                                        inParams->LogAddress,
                                                        outParams->LogSectors);
 
                                if (NT_SUCCESS(status))
                                {
                                    outParams->Length = readSize;
                                } else {
                                    //
                                    // SMART command failure is
                                    // indicated by successful
                                    // execution, but no data returned
                                    //
                                    outParams->Length = 0;
                                    status = STATUS_SUCCESS;
                                }
                            } else {
                                status = STATUS_BUFFER_TOO_SMALL;
                            }
 
                        } else {
                            status = STATUS_INVALID_PARAMETER;
                        }
                    } else {
                        status = STATUS_INVALID_DEVICE_REQUEST;
                    }
                    break;
                }
 
                //    void WriteLogSectors([in] uint8 LogAddress,
                //        [in] uint8 SectorCount,
                //        [in] uint32 Length,
                //        [in, WmiSizeIs("Length")] uint8 LogSectors[],
                //        [out] boolean Success
                //       );
                case WriteLogSectors:
                {
                    if (diskData->FailurePredictionCapability ==
                                  FailurePredictionSmart)
                    {
                        if ((LONG)InBufferSize >= FIELD_OFFSET(WRITE_LOG_SECTORS_IN,
                                                        LogSectors))
                        {
                            PWRITE_LOG_SECTORS_IN inParams;
                            PWRITE_LOG_SECTORS_OUT outParams;
                            ULONG writeSize;
 
                            inParams = (PWRITE_LOG_SECTORS_IN)Buffer;
                            writeSize = inParams->SectorCount * SMART_LOG_SECTOR_SIZE;
                            if (InBufferSize >= (FIELD_OFFSET(WRITE_LOG_SECTORS_IN,
                                                             LogSectors) +
                                                 writeSize))
                            {
                                sizeNeeded = sizeof(WRITE_LOG_SECTORS_OUT);
 
                                if (OutBufferSize >= sizeNeeded)
                                {
                                    outParams = (PWRITE_LOG_SECTORS_OUT)Buffer;
                                    status = DiskWriteSmartLog(fdoExtension,
                                                        inParams->SectorCount,
                                                        inParams->LogAddress,
                                                        inParams->LogSectors);
 
                                    if (NT_SUCCESS(status))
                                    {
                                        outParams->Success = TRUE;
                                    } else {
                                        outParams->Success = FALSE;
                                        status = STATUS_SUCCESS;
                                    }
                                } else {
                                    status = STATUS_BUFFER_TOO_SMALL;
                                }
                            } else {
                                status = STATUS_INVALID_PARAMETER;
                            }
                        } else {
                            status = STATUS_INVALID_PARAMETER;
                        }
                    } else {
                        status = STATUS_INVALID_DEVICE_REQUEST;
                    }
                    break;
                }
 
                //    void ExecuteSelfTest([in] uint8 Subcommand,
                //         [out,
                //          Values{"0", "1", "2"},
                //          ValueMap{"Successful Completion",
                //                   "Captive Mode Required",
                //                   "Unsuccessful Completion"}
                //         ]
                //         uint32 ReturnCode);
                case ExecuteSelfTest:
                {
                    if (diskData->FailurePredictionCapability ==
                              FailurePredictionSmart)
                    {
                        if (InBufferSize >= sizeof(EXECUTE_SELF_TEST_IN))
                        {
                            sizeNeeded = sizeof(EXECUTE_SELF_TEST_OUT);
                            if (OutBufferSize >= sizeNeeded)
                            {
                                PEXECUTE_SELF_TEST_IN inParam;
                                PEXECUTE_SELF_TEST_OUT outParam;
 
                                inParam = (PEXECUTE_SELF_TEST_IN)Buffer;
                                outParam = (PEXECUTE_SELF_TEST_OUT)Buffer;
 
                                if (DiskIsValidSmartSelfTest(inParam->Subcommand))
                                {
                                   status = DiskExecuteSmartDiagnostics(fdoExtension,
                                                            inParam->Subcommand);
                                   if (NT_SUCCESS(status))
                                   {
                                       //
                                       // Return self test executed
                                       // without a problem
                                       //
                                       outParam->ReturnCode = 0;
                                   } else {
                                       //
                                       // Return Self test execution
                                       // failed status
                                       //
                                       outParam->ReturnCode = 2;
                                       status = STATUS_SUCCESS;
                                   }
                                } else {
                                    //
                                    // If self test subcommand requires
                                    // captive mode then return that
                                    // status
                                    //
                                    outParam->ReturnCode = 1;
                                    status = STATUS_SUCCESS;
                                }
 
                            } else {
                                status = STATUS_BUFFER_TOO_SMALL;
                            }
 
                        } else {
                            status = STATUS_INVALID_PARAMETER;
                        }
                    } else {
                        status = STATUS_INVALID_DEVICE_REQUEST;
                    }
 
                    break;
                }
 
                default :
                {
                    sizeNeeded = 0;
                    status = STATUS_WMI_ITEMID_NOT_FOUND;
                    break;
                }
            }
 
            break;
        }
 
        case DiskGeometryGuid:
        case SmartStatusGuid:
        case SmartDataGuid:
        case SmartEventGuid:
        case SmartThresholdsGuid:
        case ScsiInfoExceptionsGuid:
        {
            sizeNeeded = 0;
            status = STATUS_INVALID_DEVICE_REQUEST;
            break;
        }
 
        default:
        {
            sizeNeeded = 0;
            status = STATUS_WMI_GUID_NOT_FOUND;
        }
    }
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskExecuteMethod Device %p, Irp %p returns %lx\n",
             DeviceObject, Irp, status));
 
    status = ClassWmiCompleteRequest(DeviceObject,
                                     Irp,
                                     status,
                                     sizeNeeded,
                                     IO_NO_INCREMENT);
 
    return status;
}

Referenced by DriverEntry().

DiskFdoProcessError()

VOID NTAPI DiskFdoProcessError	(	PDEVICE_OBJECT	DeviceObject,
		PSCSI_REQUEST_BLOCK	Srb,
		NTSTATUS *	Status,
		BOOLEAN *	Retry
	)

Definition at line 2276 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = Fdo->DeviceExtension;
    PSTORAGE_REQUEST_BLOCK srbEx;
    PCDB cdb = NULL;
    UCHAR scsiStatus = 0;
    UCHAR senseBufferLength = 0;
    PVOID senseBuffer = NULL;
    CDB noOp = {0};
 
    //
    // Get relevant fields from SRB
    //
    if (Srb->Function == SRB_FUNCTION_STORAGE_REQUEST_BLOCK) {
 
        srbEx = (PSTORAGE_REQUEST_BLOCK)Srb;
 
        //
        // Look for SCSI SRB specific fields
        //
        if ((srbEx->SrbFunction == SRB_FUNCTION_EXECUTE_SCSI) &&
            (srbEx->NumSrbExData > 0)) {
            cdb = GetSrbScsiData(srbEx, NULL, NULL, &scsiStatus, &senseBuffer, &senseBufferLength);
 
            //
            // cdb and sense buffer should not be NULL
            //
            NT_ASSERT(cdb != NULL);
            NT_ASSERT(senseBuffer != NULL);
 
        }
 
        if (cdb == NULL) {
 
            //
            // Use a cdb that is all 0s
            //
            cdb = &noOp;
        }
 
    } else {
 
        cdb = (PCDB)(Srb->Cdb);
        scsiStatus = Srb->ScsiStatus;
        senseBufferLength = Srb->SenseInfoBufferLength;
        senseBuffer = Srb->SenseInfoBuffer;
    }
 
    if (*Status == STATUS_DATA_OVERRUN &&
        (cdb != NULL) &&
        (IS_SCSIOP_READWRITE(cdb->CDB10.OperationCode))) {
 
            *Retry = TRUE;
 
            //
            // Update the error count for the device.
            //
 
            fdoExtension->ErrorCount++;
 
    } else if (SRB_STATUS(Srb->SrbStatus) == SRB_STATUS_ERROR &&
               scsiStatus == SCSISTAT_BUSY) {
 
        //
        // a disk drive should never be busy this long. Reset the scsi bus
        // maybe this will clear the condition.
        //
 
        ResetBus(Fdo);
 
        //
        // Update the error count for the device.
        //
 
        fdoExtension->ErrorCount++;
 
    } else {
 
        BOOLEAN invalidatePartitionTable = FALSE;
 
        //
        // See if this might indicate that something on the drive has changed.
        //
 
        if ((Srb->SrbStatus & SRB_STATUS_AUTOSENSE_VALID) &&
            (senseBuffer != NULL) && (cdb != NULL)) {
 
            BOOLEAN validSense = FALSE;
            UCHAR senseKey = 0;
            UCHAR asc = 0;
            UCHAR ascq = 0;
 
            validSense = ScsiGetSenseKeyAndCodes(senseBuffer,
                                                 senseBufferLength,
                                                 SCSI_SENSE_OPTIONS_FIXED_FORMAT_IF_UNKNOWN_FORMAT_INDICATED,
                                                 &senseKey,
                                                 &asc,
                                                 &ascq);
 
            if (validSense) {
 
                switch (senseKey) {
 
                    case SCSI_SENSE_ILLEGAL_REQUEST: {
 
                        switch (asc) {
 
                            case SCSI_ADSENSE_INVALID_CDB:
                            {
                                //
                                // Look to see if this is an Io request with the ForceUnitAccess flag set
                                //
                                if (((cdb->CDB10.OperationCode == SCSIOP_WRITE)   ||
                                    (cdb->CDB10.OperationCode == SCSIOP_WRITE16)) &&
                                    (cdb->CDB10.ForceUnitAccess))
                                {
                                    PDISK_DATA diskData = (PDISK_DATA)fdoExtension->CommonExtension.DriverData;
 
                                    if (diskData->WriteCacheOverride == DiskWriteCacheEnable)
                                    {
                                        PIO_ERROR_LOG_PACKET logEntry = NULL;
 
                                        //
                                        // The user has explicitly requested that write caching be turned on.
                                        // Warn the user that writes with FUA enabled are not working and that
                                        // they should disable write cache.
                                        //
 
                                        logEntry = IoAllocateErrorLogEntry(fdoExtension->DeviceObject,
                                                                           sizeof(IO_ERROR_LOG_PACKET) + (4 * sizeof(ULONG)));
 
                                        if (logEntry != NULL)
                                        {
                                            logEntry->FinalStatus       = *Status;
                                            logEntry->ErrorCode         = IO_WARNING_WRITE_FUA_PROBLEM;
                                            logEntry->SequenceNumber    = 0;
                                            logEntry->MajorFunctionCode = IRP_MJ_SCSI;
                                            logEntry->IoControlCode     = 0;
                                            logEntry->RetryCount        = 0;
                                            logEntry->UniqueErrorValue  = 0;
                                            logEntry->DumpDataSize      = 4 * sizeof(ULONG);
 
                                            logEntry->DumpData[0] = diskData->ScsiAddress.PortNumber;
                                            logEntry->DumpData[1] = diskData->ScsiAddress.PathId;
                                            logEntry->DumpData[2] = diskData->ScsiAddress.TargetId;
                                            logEntry->DumpData[3] = diskData->ScsiAddress.Lun;
 
                                            //
                                            // Write the error log packet.
                                            //
 
                                            IoWriteErrorLogEntry(logEntry);
                                        }
                                    }
                                    else
                                    {
                                        //
                                        // Turn off write caching on this device. This is so that future
                                        // critical requests need not be sent down with  ForceUnitAccess
                                        //
                                        PIO_WORKITEM workItem = IoAllocateWorkItem(Fdo);
 
                                        if (workItem)
                                        {
                                            IoQueueWorkItem(workItem, DisableWriteCache, CriticalWorkQueue, workItem);
                                        }
                                    }
 
                                    SET_FLAG(fdoExtension->ScanForSpecialFlags, CLASS_SPECIAL_FUA_NOT_SUPPORTED);
                                    ADJUST_FUA_FLAG(fdoExtension);
 
 
                                    cdb->CDB10.ForceUnitAccess = FALSE;
                                    *Retry = TRUE;
 
                                } else if ((cdb->CDB6FORMAT.OperationCode == SCSIOP_MODE_SENSE) &&
                                           (cdb->MODE_SENSE.PageCode == MODE_SENSE_RETURN_ALL)) {
 
                                    //
                                    // Mode sense for all pages failed. This command could fail with
                                    // SCSI_SENSE_ILLEGAL_REQUEST / SCSI_ADSENSE_INVALID_CDB if the data
                                    // to be returned is more than 256 bytes. In which case, try to get
                                    // only MODE_PAGE_CACHING since we only need the block descriptor.
                                    //
                                    // Simply change the page code and retry the request
                                    //
 
                                    cdb->MODE_SENSE.PageCode = MODE_PAGE_CACHING;
                                    *Retry = TRUE;
                                }
 
                                break;
                            }
                        } // end switch(asc)
                        break;
                    }
 
                    case SCSI_SENSE_NOT_READY: {
 
                        switch (asc) {
                        case SCSI_ADSENSE_LUN_NOT_READY: {
                            switch (ascq) {
                            case SCSI_SENSEQ_BECOMING_READY:
                            case SCSI_SENSEQ_MANUAL_INTERVENTION_REQUIRED:
                            case SCSI_SENSEQ_CAUSE_NOT_REPORTABLE: {
                                invalidatePartitionTable = TRUE;
                                break;
                            }
                            } // end switch(ascq)
                            break;
                        }
 
                        case SCSI_ADSENSE_NO_MEDIA_IN_DEVICE: {
                            invalidatePartitionTable = TRUE;
                            break;
                        }
                        } // end switch(asc)
                        break;
                    }
 
                    case SCSI_SENSE_MEDIUM_ERROR: {
                        invalidatePartitionTable = TRUE;
                        break;
                    }
 
                    case SCSI_SENSE_HARDWARE_ERROR: {
                        invalidatePartitionTable = TRUE;
                        break;
                    }
 
                    case SCSI_SENSE_UNIT_ATTENTION:
                    {
                        invalidatePartitionTable = TRUE;
                        break;
                    }
 
                    case SCSI_SENSE_RECOVERED_ERROR: {
                        invalidatePartitionTable = TRUE;
                        break;
                    }
 
                } // end switch(senseKey)
            } // end if (validSense)
        } else {
 
            //
            // On any exceptional scsi condition which might indicate that the
            // device was changed we will flush out the state of the partition
            // table.
            //
 
            switch (SRB_STATUS(Srb->SrbStatus)) {
                case SRB_STATUS_INVALID_LUN:
                case SRB_STATUS_INVALID_TARGET_ID:
                case SRB_STATUS_NO_DEVICE:
                case SRB_STATUS_NO_HBA:
                case SRB_STATUS_INVALID_PATH_ID:
                case SRB_STATUS_COMMAND_TIMEOUT:
                case SRB_STATUS_TIMEOUT:
                case SRB_STATUS_SELECTION_TIMEOUT:
                case SRB_STATUS_REQUEST_FLUSHED:
                case SRB_STATUS_UNEXPECTED_BUS_FREE:
                case SRB_STATUS_PARITY_ERROR:
                {
                    invalidatePartitionTable = TRUE;
                    break;
                }
 
                case SRB_STATUS_ERROR:
                {
                    if (scsiStatus == SCSISTAT_RESERVATION_CONFLICT)
                    {
                        invalidatePartitionTable = TRUE;
                    }
 
                    break;
                }
            } // end switch(Srb->SrbStatus)
        }
 
        if (invalidatePartitionTable && TEST_FLAG(Fdo->Characteristics, FILE_REMOVABLE_MEDIA)) {
 
            //
            // Inform the upper layers that the volume
            // on this disk is in need of verification
            //
 
            SET_FLAG(Fdo->Flags, DO_VERIFY_VOLUME);
        }
    }
 
    return;
}

Referenced by DriverEntry().

DiskFdoQueryWmiDataBlock()

NTSTATUS NTAPI DiskFdoQueryWmiDataBlock	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp,
		IN ULONG	GuidIndex,
		IN ULONG	BufferAvail,
		OUT PUCHAR	Buffer
	)

Definition at line 2638 of file diskwmi.c.

{
    NTSTATUS status;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    ULONG sizeNeeded;
 
    PAGED_CODE();
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskQueryWmiDataBlock, Device %p, Irp %p, GuiIndex %d\n"
             "      BufferAvail %lx Buffer %p\n",
             DeviceObject, Irp,
             GuidIndex, BufferAvail, Buffer));
 
    switch (GuidIndex)
    {
        case DiskGeometryGuid:
        {
            sizeNeeded = sizeof(DISK_GEOMETRY);
            if (BufferAvail >= sizeNeeded)
            {
                if (DeviceObject->Characteristics & FILE_REMOVABLE_MEDIA)
                {
                    //
                    // Issue ReadCapacity to update device extension
                    // with information for current media.
                    status = DiskReadDriveCapacity(commonExtension->PartitionZeroExtension->DeviceObject);
 
                    //
                    // Note whether the drive is ready.
                    diskData->ReadyStatus = status;
 
                    if (!NT_SUCCESS(status))
                    {
                        break;
                    }
                }
 
                //
                // Copy drive geometry information from device extension.
                RtlMoveMemory(Buffer,
                              &(fdoExtension->DiskGeometry),
                              sizeof(DISK_GEOMETRY));
 
                status = STATUS_SUCCESS;
            } else {
                status = STATUS_BUFFER_TOO_SMALL;
            }
            break;
        }
 
        case SmartStatusGuid:
        {
            PSTORAGE_FAILURE_PREDICT_STATUS diskSmartStatus;
 
            NT_ASSERT(diskData->FailurePredictionCapability != FailurePredictionNone);
 
            sizeNeeded = sizeof(STORAGE_FAILURE_PREDICT_STATUS);
            if (BufferAvail >= sizeNeeded)
            {
                STORAGE_PREDICT_FAILURE checkFailure;
 
                diskSmartStatus = (PSTORAGE_FAILURE_PREDICT_STATUS)Buffer;
 
                status = DiskSendFailurePredictIoctl(fdoExtension,
                                                     &checkFailure);
 
                if (NT_SUCCESS(status))
                {
                    if (diskData->FailurePredictionCapability ==
                                                      FailurePredictionSense)
                    {
                        diskSmartStatus->Reason =  *((PULONG)checkFailure.VendorSpecific);
                    } else {
                        diskSmartStatus->Reason =  0; // unknown
                    }
 
                    diskSmartStatus->PredictFailure = (checkFailure.PredictFailure != 0);
                }
            } else {
                status = STATUS_BUFFER_TOO_SMALL;
            }
            break;
        }
 
        case SmartDataGuid:
        {
            PSTORAGE_FAILURE_PREDICT_DATA diskSmartData;
 
            NT_ASSERT((diskData->FailurePredictionCapability ==
                                                  FailurePredictionSmart) ||
                   (diskData->FailurePredictionCapability ==
                                                  FailurePredictionIoctl));
 
            sizeNeeded = sizeof(STORAGE_FAILURE_PREDICT_DATA);
            if (BufferAvail >= sizeNeeded)
            {
                PSTORAGE_PREDICT_FAILURE checkFailure = (PSTORAGE_PREDICT_FAILURE)Buffer;
 
                diskSmartData = (PSTORAGE_FAILURE_PREDICT_DATA)Buffer;
 
                status = DiskSendFailurePredictIoctl(fdoExtension,
                                                     checkFailure);
 
                if (NT_SUCCESS(status))
                {
                    diskSmartData->Length = 512;
                }
            } else {
                status = STATUS_BUFFER_TOO_SMALL;
            }
 
            break;
        }
 
        case SmartThresholdsGuid:
        {
            PSTORAGE_FAILURE_PREDICT_THRESHOLDS diskSmartThresholds;
 
            NT_ASSERT((diskData->FailurePredictionCapability ==
                                                  FailurePredictionSmart));
 
            sizeNeeded = sizeof(STORAGE_FAILURE_PREDICT_THRESHOLDS);
            if (BufferAvail >= sizeNeeded)
            {
                diskSmartThresholds = (PSTORAGE_FAILURE_PREDICT_THRESHOLDS)Buffer;
                status = DiskReadFailurePredictThresholds(fdoExtension,
                                                          diskSmartThresholds);
            } else {
                status = STATUS_BUFFER_TOO_SMALL;
            }
 
            break;
        }
 
        case SmartPerformFunction:
        {
            sizeNeeded = 0;
            status = STATUS_SUCCESS;
            break;
        }
 
        case ScsiInfoExceptionsGuid:
        {
            PSTORAGE_SCSI_INFO_EXCEPTIONS infoExceptions;
            MODE_INFO_EXCEPTIONS modeInfo;
 
            NT_ASSERT((diskData->FailurePredictionCapability ==
                                                  FailurePredictionSense));
 
            sizeNeeded = sizeof(STORAGE_SCSI_INFO_EXCEPTIONS);
            if (BufferAvail >= sizeNeeded)
            {
                infoExceptions = (PSTORAGE_SCSI_INFO_EXCEPTIONS)Buffer;
                status = DiskGetInfoExceptionInformation(fdoExtension,
                                                         &modeInfo);
                if (NT_SUCCESS(status))
                {
                    infoExceptions->PageSavable = modeInfo.PSBit;
                    infoExceptions->Flags = modeInfo.Flags;
                    infoExceptions->MRIE = modeInfo.ReportMethod;
                    infoExceptions->Padding = 0;
                    REVERSE_BYTES(&infoExceptions->IntervalTimer,
                                  &modeInfo.IntervalTimer);
                    REVERSE_BYTES(&infoExceptions->ReportCount,
                                  &modeInfo.ReportCount)
                }
            } else {
                status = STATUS_BUFFER_TOO_SMALL;
            }
 
            break;
        }
 
        default:
        {
            sizeNeeded = 0;
            status = STATUS_WMI_GUID_NOT_FOUND;
        }
    }
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskQueryWmiDataBlock Device %p, Irp %p returns %lx\n",
             DeviceObject, Irp, status));
 
    status = ClassWmiCompleteRequest(DeviceObject,
                                     Irp,
                                     status,
                                     sizeNeeded,
                                     IO_NO_INCREMENT);
 
    return status;
}

Referenced by DriverEntry().

DiskFdoQueryWmiRegInfo()

NTSTATUS NTAPI DiskFdoQueryWmiRegInfo	(	IN PDEVICE_OBJECT	DeviceObject,
		OUT ULONG *	RegFlags,
		OUT PUNICODE_STRING	InstanceName
	)

Definition at line 2479 of file diskwmi.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
 
    PAGED_CODE();
    UNREFERENCED_PARAMETER(InstanceName);
 
    SET_FLAG(DiskWmiFdoGuidList[SmartThresholdsGuid].Flags,  WMIREG_FLAG_REMOVE_GUID);
    SET_FLAG(DiskWmiFdoGuidList[ScsiInfoExceptionsGuid].Flags,  WMIREG_FLAG_REMOVE_GUID);
 
    switch (diskData->FailurePredictionCapability)
    {
        case FailurePredictionSmart:
        {
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartThresholdsGuid].Flags,  WMIREG_FLAG_REMOVE_GUID);
            //
            // Fall Through
            //
        }
        case FailurePredictionIoctl:
        {
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartStatusGuid].Flags,      WMIREG_FLAG_REMOVE_GUID);
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartDataGuid].Flags,        WMIREG_FLAG_REMOVE_GUID);
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartEventGuid].Flags,       WMIREG_FLAG_REMOVE_GUID);
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartPerformFunction].Flags, WMIREG_FLAG_REMOVE_GUID);
 
            break;
        }
 
        case FailurePredictionSense:
        {
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartStatusGuid].Flags,      WMIREG_FLAG_REMOVE_GUID);
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartEventGuid].Flags,       WMIREG_FLAG_REMOVE_GUID);
            CLEAR_FLAG(DiskWmiFdoGuidList[SmartPerformFunction].Flags, WMIREG_FLAG_REMOVE_GUID);
            CLEAR_FLAG(DiskWmiFdoGuidList[ScsiInfoExceptionsGuid].Flags,  WMIREG_FLAG_REMOVE_GUID);
            SET_FLAG  (DiskWmiFdoGuidList[SmartDataGuid].Flags,        WMIREG_FLAG_REMOVE_GUID);
            break;
        }
 
 
        default:
        {
            SET_FLAG  (DiskWmiFdoGuidList[SmartStatusGuid].Flags,      WMIREG_FLAG_REMOVE_GUID);
            SET_FLAG  (DiskWmiFdoGuidList[SmartDataGuid].Flags,        WMIREG_FLAG_REMOVE_GUID);
            SET_FLAG  (DiskWmiFdoGuidList[SmartEventGuid].Flags,       WMIREG_FLAG_REMOVE_GUID);
            SET_FLAG  (DiskWmiFdoGuidList[SmartPerformFunction].Flags, WMIREG_FLAG_REMOVE_GUID);
            break;
        }
    }
 
    //
    // Use devnode for FDOs
    *RegFlags = WMIREG_FLAG_INSTANCE_PDO;
 
    return STATUS_SUCCESS;
}

Referenced by DiskFdoQueryWmiRegInfoEx(), and DriverEntry().

DiskFdoQueryWmiRegInfoEx()

NTSTATUS NTAPI DiskFdoQueryWmiRegInfoEx	(	IN PDEVICE_OBJECT	DeviceObject,
		OUT ULONG *	RegFlags,
		OUT PUNICODE_STRING	InstanceName,
		OUT PUNICODE_STRING	MofName
	)

Definition at line 2577 of file diskwmi.c.

{
    NTSTATUS status;
 
    UNREFERENCED_PARAMETER(MofName);
 
    status = DiskFdoQueryWmiRegInfo(DeviceObject,
                                    RegFlags,
                                    InstanceName);
 
    //
    // Leave MofName alone since disk doesn't have one
    //
    return(status);
}

Referenced by DriverEntry().

DiskFdoSetWmiDataBlock()

NTSTATUS NTAPI DiskFdoSetWmiDataBlock	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp,
		IN ULONG	GuidIndex,
		IN ULONG	BufferSize,
		IN PUCHAR	Buffer
	)

Definition at line 2869 of file diskwmi.c.

{
    NTSTATUS status;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
 
    PAGED_CODE();
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskSetWmiDataBlock, Device %p, Irp %p, GuiIndex %d\n"
             "      BufferSize %#x Buffer %p\n",
             DeviceObject, Irp,
             GuidIndex, BufferSize, Buffer));
 
    if (GuidIndex == ScsiInfoExceptionsGuid)
    {
        PSTORAGE_SCSI_INFO_EXCEPTIONS infoExceptions;
        MODE_INFO_EXCEPTIONS modeInfo = {0};
 
        if (BufferSize >= sizeof(STORAGE_SCSI_INFO_EXCEPTIONS))
        {
            infoExceptions = (PSTORAGE_SCSI_INFO_EXCEPTIONS)Buffer;
 
            modeInfo.PageCode = MODE_PAGE_FAULT_REPORTING;
            modeInfo.PageLength = sizeof(MODE_INFO_EXCEPTIONS) - 2;
 
            modeInfo.PSBit = 0;
            modeInfo.Flags = infoExceptions->Flags;
 
            modeInfo.ReportMethod = infoExceptions->MRIE;
 
            REVERSE_BYTES(&modeInfo.IntervalTimer[0],
                          &infoExceptions->IntervalTimer);
 
            REVERSE_BYTES(&modeInfo.ReportCount[0],
                          &infoExceptions->ReportCount);
 
            if (modeInfo.Perf == 1)
            {
                diskData->AllowFPPerfHit = FALSE;
            } else {
                diskData->AllowFPPerfHit = TRUE;
            }
 
            status = DiskSetInfoExceptionInformation(fdoExtension,
                                                     &modeInfo);
        } else {
            status = STATUS_INVALID_PARAMETER;
        }
 
    } else if (GuidIndex <= SmartThresholdsGuid)
    {
        status = STATUS_WMI_READ_ONLY;
    } else {
        status = STATUS_WMI_GUID_NOT_FOUND;
    }
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskSetWmiDataBlock Device %p, Irp %p returns %lx\n",
             DeviceObject, Irp, status));
 
    status = ClassWmiCompleteRequest(DeviceObject,
                                     Irp,
                                     status,
                                     0,
                                     IO_NO_INCREMENT);
 
    return status;
}

Referenced by DriverEntry().

DiskFdoSetWmiDataItem()

NTSTATUS NTAPI DiskFdoSetWmiDataItem	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp,
		IN ULONG	GuidIndex,
		IN ULONG	DataItemId,
		IN ULONG	BufferSize,
		IN PUCHAR	Buffer
	)

Definition at line 2975 of file diskwmi.c.

{
    NTSTATUS status;
 
    PAGED_CODE();
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskSetWmiDataItem, Device %p, Irp %p, GuiIndex %d, DataId %d\n"
             "      BufferSize %#x Buffer %p\n",
             DeviceObject, Irp,
             GuidIndex, DataItemId, BufferSize, Buffer));
 
    if (GuidIndex <= SmartThresholdsGuid)
    {
        status = STATUS_WMI_READ_ONLY;
    } else {
        status = STATUS_WMI_GUID_NOT_FOUND;
    }
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DiskSetWmiDataItem Device %p, Irp %p returns %lx\n",
             DeviceObject, Irp, status));
 
    status = ClassWmiCompleteRequest(DeviceObject,
                                     Irp,
                                     status,
                                     0,
                                     IO_NO_INCREMENT);
 
    return status;
}

Referenced by DriverEntry().

DiskFlushDispatch()

VOID DiskFlushDispatch	(	IN PDEVICE_OBJECT	Fdo,
		IN PDISK_GROUP_CONTEXT	FlushContext
	)

Definition at line 1534 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExt = Fdo->DeviceExtension;
    PSCSI_REQUEST_BLOCK srb = &FlushContext->Srb.Srb;
    PSTORAGE_REQUEST_BLOCK srbEx = &FlushContext->Srb.SrbEx;
    PIO_STACK_LOCATION  irpSp = NULL;
    PSTOR_ADDR_BTL8 storAddrBtl8;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16;
    NTSTATUS SyncCacheStatus = STATUS_SUCCESS;
 
    //
    // Fill in the srb fields appropriately
    //
    if (fdoExt->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        RtlZeroMemory(srbEx, sizeof(FlushContext->Srb.SrbExBuffer));
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = sizeof(FlushContext->Srb.SrbExBuffer);
        srbEx->RequestPriority = IoGetIoPriorityHint(FlushContext->CurrIrp);
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
        srbEx->TimeOutValue = fdoExt->TimeOutValue * 4;
        srbEx->RequestTag = SP_UNTAGGED;
        srbEx->RequestAttribute  = SRB_SIMPLE_TAG_REQUEST;
        srbEx->SrbFlags = fdoExt->SrbFlags;
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
    } else {
        RtlZeroMemory(srb, SCSI_REQUEST_BLOCK_SIZE);
 
        srb->Length       = SCSI_REQUEST_BLOCK_SIZE;
        srb->TimeOutValue = fdoExt->TimeOutValue * 4;
        srb->QueueTag     = SP_UNTAGGED;
        srb->QueueAction  = SRB_SIMPLE_TAG_REQUEST;
        srb->SrbFlags     = fdoExt->SrbFlags;
    }
 
    //
    // If write caching is enabled then send down a synchronize cache request
    //
    if (TEST_FLAG(fdoExt->DeviceFlags, DEV_WRITE_CACHE))
    {
 
        if (fdoExt->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
            srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
            srbEx->NumSrbExData = 1;
 
            //
            // Set up SCSI SRB extended data fields
            //
 
            srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
                sizeof(STOR_ADDR_BTL8);
            if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
                srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
                srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
                srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
                srbExDataCdb16->CdbLength = 10;
                srbExDataCdb16->Cdb[0] = SCSIOP_SYNCHRONIZE_CACHE;
            } else {
                // Should not happen
                NT_ASSERT(FALSE);
                return;
            }
 
        } else {
            srb->Function = SRB_FUNCTION_EXECUTE_SCSI;
            srb->CdbLength = 10;
            srb->Cdb[0] = SCSIOP_SYNCHRONIZE_CACHE;
        }
 
        TracePrint((TRACE_LEVEL_VERBOSE, TRACE_FLAG_SCSI, "DiskFlushDispatch: sending sync cache\n"));
 
        SyncCacheStatus = ClassSendSrbSynchronous(Fdo, srb, NULL, 0, TRUE);
    }
 
    //
    // Set up a FLUSH SRB
    //
    if (fdoExt->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx->SrbFunction = SRB_FUNCTION_FLUSH;
        srbEx->NumSrbExData = 0;
        srbEx->SrbExDataOffset[0] = 0;
        srbEx->OriginalRequest = FlushContext->CurrIrp;
        srbEx->SrbStatus = 0;
 
        //
        // Make sure that this srb does not get freed
        //
        SET_FLAG(srbEx->SrbFlags, SRB_CLASS_FLAGS_PERSISTANT);
 
   } else {
        srb->Function  = SRB_FUNCTION_FLUSH;
        srb->CdbLength = 0;
        srb->OriginalRequest = FlushContext->CurrIrp;
        srb->SrbStatus = 0;
        srb->ScsiStatus = 0;
 
        //
        // Make sure that this srb does not get freed
        //
        SET_FLAG(srb->SrbFlags, SRB_CLASS_FLAGS_PERSISTANT);
    }
 
    //
    // Make sure that this request does not get retried
    //
    irpSp = IoGetCurrentIrpStackLocation(FlushContext->CurrIrp);
 
    irpSp->Parameters.Others.Argument4 = (PVOID) 0;
 
    //
    // Fill in the irp fields appropriately
    //
    irpSp = IoGetNextIrpStackLocation(FlushContext->CurrIrp);
 
    irpSp->MajorFunction       = IRP_MJ_SCSI;
    irpSp->Parameters.Scsi.Srb = srb;
 
    IoSetCompletionRoutine(FlushContext->CurrIrp, DiskFlushComplete, (PVOID)(ULONG_PTR)SyncCacheStatus, TRUE, TRUE, TRUE);
 
    TracePrint((TRACE_LEVEL_VERBOSE, TRACE_FLAG_SCSI, "DiskFlushDispatch: sending srb flush on irp %p\n", FlushContext->CurrIrp));
 
    //
    // Send down the flush request
    //
    IoCallDriver(((PCOMMON_DEVICE_EXTENSION)fdoExt)->LowerDeviceObject, FlushContext->CurrIrp);
}

Referenced by DiskShutdownFlush().

DiskGenerateDeviceName()

NTSTATUS DiskGenerateDeviceName	(	IN ULONG	DeviceNumber,
		OUT PCCHAR *	RawName
	)

Definition at line 612 of file pnp.c.

{
    CHAR rawName[64] = { 0 };
    NTSTATUS status;
 
    PAGED_CODE();
 
        status = RtlStringCchPrintfA(rawName, sizeof(rawName) - 1, FDO_NAME_FORMAT, DeviceNumber,
                                    diskDeviceSequenceNumber++);
        if (!NT_SUCCESS(status)) {
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_PNP, "DiskGenerateDeviceName: Format FDO name failed with error: 0x%X\n", status));
            return status;
        }
 
    *RawName = ExAllocatePoolWithTag(PagedPool,
                                     strlen(rawName) + 1,
                                     DISK_TAG_NAME);
 
    if(*RawName == NULL) {
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    status = RtlStringCchCopyA(*RawName, strlen(rawName) + 1, rawName);
    if (!NT_SUCCESS(status)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_PNP, "DiskGenerateDeviceName: Device name copy failed with error: 0x%X\n", status));
        FREE_POOL(*RawName);
        return status;
    }
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_PNP, "DiskGenerateDeviceName: generated \"%s\"\n", rawName));
 
    return STATUS_SUCCESS;
}

Referenced by DiskCreateFdo().

DiskGetCacheInformation()

NTSTATUS NTAPI DiskGetCacheInformation	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		IN PDISK_CACHE_INFORMATION	CacheInfo
	)

Definition at line 3019 of file disk.c.

{
    PMODE_PARAMETER_HEADER modeData;
    PMODE_CACHING_PAGE pageData;
 
    ULONG length;
 
    PAGED_CODE();
 
 
 
    modeData = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                     MODE_DATA_SIZE,
                                     DISK_TAG_DISABLE_CACHE);
 
    if (modeData == NULL) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskGetSetCacheInformation: Unable to allocate mode "
                       "data buffer\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(modeData, MODE_DATA_SIZE);
 
    length = ClassModeSense(FdoExtension->DeviceObject,
                            (PCHAR) modeData,
                            MODE_DATA_SIZE,
                            MODE_PAGE_CACHING);
 
    if (length < sizeof(MODE_PARAMETER_HEADER)) {
 
        //
        // Retry the request in case of a check condition.
        //
 
        length = ClassModeSense(FdoExtension->DeviceObject,
                                (PCHAR) modeData,
                                MODE_DATA_SIZE,
                                MODE_PAGE_CACHING);
 
        if (length < sizeof(MODE_PARAMETER_HEADER)) {
 
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskGetCacheInformation: Mode Sense failed\n"));
 
            FREE_POOL(modeData);
            return STATUS_IO_DEVICE_ERROR;
        }
    }
 
    //
    // If the length is greater than length indicated by the mode data reset
    // the data to the mode data.
    //
 
    if (length > (ULONG) (modeData->ModeDataLength + 1)) {
        length = modeData->ModeDataLength + 1;
    }
 
    //
    // Check to see if the write cache is enabled.
    //
 
    pageData = ClassFindModePage((PCHAR) modeData,
                                 length,
                                 MODE_PAGE_CACHING,
                                 TRUE);
 
    //
    // Check if valid caching page exists.
    //
 
    if (pageData == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskGetCacheInformation: Unable to find caching mode page.\n"));
        FREE_POOL(modeData);
        return STATUS_NOT_SUPPORTED;
    }
 
    //
    // Copy the parameters over.
    //
 
    RtlZeroMemory(CacheInfo, sizeof(DISK_CACHE_INFORMATION));
 
    CacheInfo->ParametersSavable = pageData->PageSavable;
 
    CacheInfo->ReadCacheEnabled = !(pageData->ReadDisableCache);
    CacheInfo->WriteCacheEnabled = pageData->WriteCacheEnable;
 
 
    //
    // Translate the values in the mode page into the ones defined in
    // ntdddisk.h.
    //
 
    CacheInfo->ReadRetentionPriority =
        TRANSLATE_RETENTION_PRIORITY(pageData->ReadRetensionPriority);
    CacheInfo->WriteRetentionPriority =
        TRANSLATE_RETENTION_PRIORITY(pageData->WriteRetensionPriority);
 
    CacheInfo->DisablePrefetchTransferLength =
        ((pageData->DisablePrefetchTransfer[0] << 8) +
         pageData->DisablePrefetchTransfer[1]);
 
    CacheInfo->ScalarPrefetch.Minimum =
        ((pageData->MinimumPrefetch[0] << 8) + pageData->MinimumPrefetch[1]);
 
    CacheInfo->ScalarPrefetch.Maximum =
        ((pageData->MaximumPrefetch[0] << 8) + pageData->MaximumPrefetch[1]);
 
    if(pageData->MultiplicationFactor) {
        CacheInfo->PrefetchScalar = TRUE;
        CacheInfo->ScalarPrefetch.MaximumBlocks =
            ((pageData->MaximumPrefetchCeiling[0] << 8) +
             pageData->MaximumPrefetchCeiling[1]);
    }
 
 
    FREE_POOL(modeData);
    return STATUS_SUCCESS;
}

Referenced by DiskIoctlGetCacheInformation(), and DiskStartFdo().

DiskGetInfoExceptionInformation()

NTSTATUS DiskGetInfoExceptionInformation	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		OUT PMODE_INFO_EXCEPTIONS	ReturnPageData
	)

DiskGetModePage()

NTSTATUS DiskGetModePage	(	_In_ PDEVICE_OBJECT	Fdo,
		_In_ UCHAR	PageMode,
		_In_ UCHAR	PageControl,
		_In_ PMODE_PARAMETER_HEADER	ModeData,
		_Inout_ PULONG	ModeDataSize,
		_Out_ PVOID *	PageData
	)

Definition at line 903 of file diskwmi.c.

{
    ULONG size = 0;
    PVOID pageData = NULL;
 
    PAGED_CODE();
 
    if (ModeData == NULL ||
        ModeDataSize == NULL ||
        *ModeDataSize < sizeof(MODE_PARAMETER_HEADER) ||
        PageData == NULL) {
        return STATUS_INVALID_PARAMETER;
    }
 
    RtlZeroMemory (ModeData, *ModeDataSize);
 
    size = ClassModeSenseEx(Fdo,
                            (PCHAR) ModeData,
                            *ModeDataSize,
                            PageMode,
                            PageControl);
 
    if (size < sizeof(MODE_PARAMETER_HEADER)) {
 
        //
        // Retry the request in case of a check condition.
        //
        size = ClassModeSenseEx(Fdo,
                            (PCHAR) ModeData,
                            *ModeDataSize,
                            PageMode,
                            PageControl);
 
        if (size < sizeof(MODE_PARAMETER_HEADER)) {
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_WMI, "DiskGetModePage: Mode Sense for Page Mode %d with Page Control %d failed\n",
                PageMode, PageControl));
            *ModeDataSize = 0;
            return STATUS_IO_DEVICE_ERROR;
        }
    }
 
    //
    // If the length is greater than length indicated by the mode data reset
    // the data to the mode data.
    //
    if (size > (ULONG) (ModeData->ModeDataLength + 1)) {
        size = ModeData->ModeDataLength + 1;
    }
 
    *ModeDataSize = size;
 
    //
    // Find the mode page
    //
    pageData = ClassFindModePage((PCHAR) ModeData,
                                 size,
                                 PageMode,
                                 TRUE);
 
    if (pageData) {
        *PageData = pageData;
        return STATUS_SUCCESS;
    } else {
        *PageData = NULL;
        return STATUS_NOT_SUPPORTED;
    }
}

Referenced by DiskEnableInfoExceptions().

DiskInitFdo()

NTSTATUS NTAPI DiskInitFdo

(

IN PDEVICE_OBJECT

Fdo

)

Definition at line 202 of file pnp.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = Fdo->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA) fdoExtension->CommonExtension.DriverData;
 
    ULONG srbFlags = 0;
    ULONG timeOut = 0;
    ULONG bytesPerSector;
 
    PULONG dmSkew;
 
    NTSTATUS status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    //
    // Build the lookaside list for srb's for the physical disk. Should only
    // need a couple.  If this fails then we don't have an emergency SRB so
    // fail the call to initialize.
    //
 
    ClassInitializeSrbLookasideList((PCOMMON_DEVICE_EXTENSION) fdoExtension,
                                    PARTITION0_LIST_SIZE);
 
    if (fdoExtension->DeviceDescriptor->RemovableMedia)
    {
        SET_FLAG(Fdo->Characteristics, FILE_REMOVABLE_MEDIA);
    }
 
    //
    // Initialize the srb flags.
    //
 
    //
    // Because all requests share a common sense buffer, it is possible
    // for the buffer to be overwritten if the port driver completes
    // multiple failed requests that require a request sense before the
    // class driver's completion routine can consume the data in the buffer.
    // To prevent this, we allow the port driver to allocate a unique sense
    // buffer each time it needs one.  We are responsible for freeing this
    // buffer.  This also allows the adapter to be configured to support
    // additional sense data beyond the minimum 18 bytes.
    //
 
    SET_FLAG(fdoExtension->SrbFlags, SRB_FLAGS_PORT_DRIVER_ALLOCSENSE);
 
    if (fdoExtension->DeviceDescriptor->CommandQueueing &&
        fdoExtension->AdapterDescriptor->CommandQueueing) {
 
        SET_FLAG(fdoExtension->SrbFlags, SRB_FLAGS_QUEUE_ACTION_ENABLE);
 
    }
 
    //
    // Look for controllers that require special flags.
    //
 
    ClassScanForSpecial(fdoExtension, DiskBadControllers, DiskSetSpecialHacks);
 
    //
    // Clear buffer for drive geometry.
    //
 
    RtlZeroMemory(&(fdoExtension->DiskGeometry),
                  sizeof(DISK_GEOMETRY));
 
    //
    // Allocate request sense buffer.
    //
 
    fdoExtension->SenseData = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                                    SENSE_BUFFER_SIZE_EX,
                                                    DISK_TAG_START);
 
    if (fdoExtension->SenseData == NULL) {
 
        //
        // The buffer can not be allocated.
        //
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_PNP, "DiskInitFdo: Can not allocate request sense buffer\n"));
 
        status = STATUS_INSUFFICIENT_RESOURCES;
        return status;
    }
 
    //
    // Set the buffer size of SenseData
    //
 
    fdoExtension->SenseDataLength = SENSE_BUFFER_SIZE_EX;
 
    //
    // Physical device object will describe the entire
    // device, starting at byte offset 0.
    //
 
    fdoExtension->CommonExtension.StartingOffset.QuadPart = (LONGLONG)(0);
 
    //
    // Set timeout value in seconds.
    //
    if ( (fdoExtension->MiniportDescriptor != NULL) &&
         (fdoExtension->MiniportDescriptor->IoTimeoutValue > 0) ) {
        //
        // use the value set by Storport miniport driver
        //
        fdoExtension->TimeOutValue = fdoExtension->MiniportDescriptor->IoTimeoutValue;
    } else {
        //
        // get timeout value from registry
        //
        timeOut = ClassQueryTimeOutRegistryValue(Fdo);
 
        if (timeOut) {
            fdoExtension->TimeOutValue = timeOut;
        } else {
            fdoExtension->TimeOutValue = SCSI_DISK_TIMEOUT;
        }
    }
    //
    // If this is a removable drive, build an entry in devicemap\scsi
    // indicating it's physicaldriveN name, set up the appropriate
    // update partitions routine and set the flags correctly.
    // note: only do this after the timeout value is set, above.
    //
 
    if (TEST_FLAG(Fdo->Characteristics, FILE_REMOVABLE_MEDIA)) {
 
        ClassUpdateInformationInRegistry( Fdo,
                                          "PhysicalDrive",
                                          fdoExtension->DeviceNumber,
                                          NULL,
                                          0);
        //
        // Enable media change notification for removable disks
        //
        ClassInitializeMediaChangeDetection(fdoExtension,
                                            (PUCHAR)"Disk");
 
    } else {
 
        SET_FLAG(fdoExtension->DeviceFlags, DEV_SAFE_START_UNIT);
        SET_FLAG(fdoExtension->SrbFlags, SRB_FLAGS_NO_QUEUE_FREEZE);
 
    }
 
    //
    // The commands we send during the init could cause the flags to change
    // in case of any error.  Save the SRB flags locally and restore it at
    // the end of this function, so that the class driver can get it.
    //
 
    srbFlags = fdoExtension->SrbFlags;
 
 
    //
    // Read the drive capacity.  Don't use the disk version of the routine here
    // since we don't know the disk signature yet - the disk version will
    // attempt to determine the BIOS reported geometry.
    //
 
    (VOID)ClassReadDriveCapacity(Fdo);
 
    //
    // Set up sector size fields.
    //
    // Stack variables will be used to update
    // the partition device extensions.
    //
    // The device extension field SectorShift is
    // used to calculate sectors in I/O transfers.
    //
    // The DiskGeometry structure is used to service
    // IOCTls used by the format utility.
    //
 
    bytesPerSector = fdoExtension->DiskGeometry.BytesPerSector;
 
    //
    // Make sure sector size is not zero.
    //
 
    if (bytesPerSector == 0) {
 
        //
        // Default sector size for disk is 512.
        //
 
        bytesPerSector = fdoExtension->DiskGeometry.BytesPerSector = 512;
        fdoExtension->SectorShift = 9;
    }
 
    //
    // Determine is DM Driver is loaded on an IDE drive that is
    // under control of Atapi - this could be either a crashdump or
    // an Atapi device is sharing the controller with an IDE disk.
    //
 
    HalExamineMBR(fdoExtension->CommonExtension.DeviceObject,
                  fdoExtension->DiskGeometry.BytesPerSector,
                  (ULONG)0x54,
                  (PVOID *)&dmSkew);
 
    if (dmSkew) {
 
        //
        // Update the device extension, so that the call to IoReadPartitionTable
        // will get the correct information. Any I/O to this disk will have
        // to be skewed by *dmSkew sectors aka DMByteSkew.
        //
 
        fdoExtension->DMSkew     = *dmSkew;
        fdoExtension->DMActive   = TRUE;
        fdoExtension->DMByteSkew = fdoExtension->DMSkew * bytesPerSector;
 
        FREE_POOL(dmSkew);
    }
 
#if defined(_X86_) || defined(_AMD64_)
 
    //
    // Try to read the signature off the disk and determine the correct drive
    // geometry based on that.  This requires rereading the disk size to get
    // the cylinder count updated correctly.
    //
 
    if(!TEST_FLAG(Fdo->Characteristics, FILE_REMOVABLE_MEDIA)) {
 
        DiskReadSignature(Fdo);
        DiskReadDriveCapacity(Fdo);
 
        if (diskData->GeometrySource == DiskGeometryUnknown)
        {
            //
            // Neither the  BIOS  nor the port driver could provide us with a  reliable
            // geometry.  Before we use the default,  look to see if it was partitioned
            // under Windows NT4 [or earlier] and apply the one that was used back then
            //
 
            if (DiskIsNT4Geometry(fdoExtension))
            {
                diskData->RealGeometry = fdoExtension->DiskGeometry;
                diskData->RealGeometry.SectorsPerTrack   = 0x20;
                diskData->RealGeometry.TracksPerCylinder = 0x40;
                fdoExtension->DiskGeometry = diskData->RealGeometry;
 
                diskData->GeometrySource = DiskGeometryFromNT4;
            }
        }
    }
 
#endif
 
    DiskCreateSymbolicLinks(Fdo);
 
    //
    // Get the SCSI address if it's available for use with SMART ioctls.
    // SMART ioctls are used for failure prediction, so we need to get
    // the SCSI address before initializing failure prediction.
    //
 
    {
        PIRP irp;
        KEVENT event;
        IO_STATUS_BLOCK statusBlock = { 0 };
 
        KeInitializeEvent(&event, SynchronizationEvent, FALSE);
 
        irp = IoBuildDeviceIoControlRequest(IOCTL_SCSI_GET_ADDRESS,
                                            fdoExtension->CommonExtension.LowerDeviceObject,
                                            NULL,
                                            0L,
                                            &(diskData->ScsiAddress),
                                            sizeof(SCSI_ADDRESS),
                                            FALSE,
                                            &event,
                                            &statusBlock);
 
        status = STATUS_UNSUCCESSFUL;
 
        if(irp != NULL) {
 
            status = IoCallDriver(fdoExtension->CommonExtension.LowerDeviceObject, irp);
 
            if(status == STATUS_PENDING) {
                KeWaitForSingleObject(&event,
                                      Executive,
                                      KernelMode,
                                      FALSE,
                                      NULL);
                status = statusBlock.Status;
            }
        }
    }
 
    //
    // Determine the type of disk and enable failure prediction in the hardware
    // and enable failure prediction polling.
    //
 
    if (InitSafeBootMode == 0) // __REACTOS__
    {
        DiskDetectFailurePrediction(fdoExtension,
                                    &diskData->FailurePredictionCapability,
                                    NT_SUCCESS(status));
 
        if (diskData->FailurePredictionCapability != FailurePredictionNone)
        {
            //
            // Cool, we've got some sort of failure prediction, enable it
            // at the hardware and then enable polling for it
            //
 
            //
            // By default we allow performance to be degradeded if failure
            // prediction is enabled.
            //
 
            diskData->AllowFPPerfHit = TRUE;
 
            //
            // Enable polling only after Atapi and SBP2 add support for the new
            // SRB flag that indicates that the request should not reset the
            // drive spin down idle timer.
            //
 
            status = DiskEnableDisableFailurePredictPolling(fdoExtension,
                                          TRUE,
                                          DISK_DEFAULT_FAILURE_POLLING_PERIOD);
 
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_PNP, "DiskInitFdo: Failure Prediction Poll enabled as "
                           "%d for device %p, Status = %lx\n",
                     diskData->FailurePredictionCapability,
                     Fdo,
                     status));
        }
    } else {
 
        //
        // In safe boot mode we do not enable failure prediction, as perhaps
        // it is the reason why normal boot does not work
        //
 
        diskData->FailurePredictionCapability = FailurePredictionNone;
 
    }
 
    //
    // Initialize the verify mutex
    //
 
    KeInitializeMutex(&diskData->VerifyMutex, MAX_SECTORS_PER_VERIFY);
 
    //
    // Initialize the flush group context
    //
 
    RtlZeroMemory(&diskData->FlushContext, sizeof(DISK_GROUP_CONTEXT));
 
    InitializeListHead(&diskData->FlushContext.CurrList);
    InitializeListHead(&diskData->FlushContext.NextList);
 
    KeInitializeSpinLock(&diskData->FlushContext.Spinlock);
    KeInitializeEvent(&diskData->FlushContext.Event, SynchronizationEvent, FALSE);
 
 
    //
    // Restore the saved value
    //
    fdoExtension->SrbFlags = srbFlags;
 
    return STATUS_SUCCESS;
 
} // end DiskInitFdo()

Referenced by DriverEntry().

DiskInitializeReregistration()

NTSTATUS DiskInitializeReregistration

(

VOID

)

Definition at line 1560 of file diskwmi.c.

{
    PAGED_CODE();
 
    //
    // Initialize the spinlock used to manage the
    // list of disks reregistering their guids
    //
    KeInitializeSpinLock(&DiskReregSpinlock);
 
    return(STATUS_SUCCESS);
}

Referenced by DriverEntry().

DiskIoctlClearVerify()

NTSTATUS DiskIoctlClearVerify	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5313 of file disk.c.

{
    NTSTATUS status = STATUS_NOT_SUPPORTED;
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlClearVerify: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    //
    // If the caller is kernel mode, set the verify bit.
    //
 
    if (Irp->RequestorMode == KernelMode) {
 
        CLEAR_FLAG(DeviceObject->Flags, DO_VERIFY_VOLUME);
        status = STATUS_SUCCESS;
 
    }
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlEnableFailurePrediction()

NTSTATUS DiskIoctlEnableFailurePrediction	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 4419 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status = STATUS_SUCCESS;
    PSTORAGE_FAILURE_PREDICTION_CONFIG enablePrediction;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlEnableFailurePrediction: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(STORAGE_FAILURE_PREDICTION_CONFIG) ||
        irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(STORAGE_FAILURE_PREDICTION_CONFIG)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlEnableFailurePrediction: Buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    enablePrediction = (PSTORAGE_FAILURE_PREDICTION_CONFIG)Irp->AssociatedIrp.SystemBuffer;
 
    if (enablePrediction->Version != STORAGE_FAILURE_PREDICTION_CONFIG_V1 ||
        enablePrediction->Size < sizeof(STORAGE_FAILURE_PREDICTION_CONFIG)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlEnableFailurePrediction: Buffer version or size is incorrect.\n"));
        status = STATUS_INVALID_PARAMETER;
    }
 
    if (enablePrediction->Reserved != 0) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlEnableFailurePrediction: Reserved bytes are not zero!\n"));
        status = STATUS_INVALID_PARAMETER;
    }
 
    //
    // Default to success.  This might get overwritten on failure below.
    //
    status = STATUS_SUCCESS;
 
    //
    // If this is a "set" and the current state (enabled/disabled) is
    // different from the sender's desired state,
    //
    if (enablePrediction->Set && enablePrediction->Enabled != diskData->FailurePredictionEnabled) {
        if (diskData->FailurePredictionCapability == FailurePredictionSmart ||
            diskData->FailurePredictionCapability == FailurePredictionIoctl) {
            //
            // SMART or IOCTL based failure prediction is being used so call
            // the generic function that is normally called in the WMI path.
            //
            status = DiskEnableDisableFailurePrediction(fdoExtension, enablePrediction->Enabled);
        } else if (diskData->ScsiInfoExceptionsSupported) {
            //
            // If we know that the device supports the Informational Exceptions
            // mode page, try to enable/disable failure prediction that way.
            //
            status = DiskEnableInfoExceptions(fdoExtension, enablePrediction->Enabled);
        }
    }
 
    //
    // Return the current state regardless if this was a "set" or a "get".
    //
    enablePrediction->Enabled = diskData->FailurePredictionEnabled;
 
    if (NT_SUCCESS(status)) {
        Irp->IoStatus.Information = sizeof(STORAGE_FAILURE_PREDICTION_CONFIG);
    }
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlGetCacheInformation()

NTSTATUS DiskIoctlGetCacheInformation	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 3840 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    PDISK_CACHE_INFORMATION cacheInfo = Irp->AssociatedIrp.SystemBuffer;
    NTSTATUS status;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlGetCacheInformation: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(DISK_CACHE_INFORMATION)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetCacheInformation: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    status = DiskGetCacheInformation(fdoExtension, cacheInfo);
 
    if (NT_SUCCESS(status)) {
        Irp->IoStatus.Information = sizeof(DISK_CACHE_INFORMATION);
 
        //
        // Make sure write cache setting is reflected in device extension
        //
        if (cacheInfo->WriteCacheEnabled)
        {
            SET_FLAG(fdoExtension->DeviceFlags, DEV_WRITE_CACHE);
        }
        else
        {
            CLEAR_FLAG(fdoExtension->DeviceFlags, DEV_WRITE_CACHE);
        }
        ADJUST_FUA_FLAG(fdoExtension);
 
    }
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlGetCacheSetting()

NTSTATUS DiskIoctlGetCacheSetting	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 3347 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(Irp);
    NTSTATUS status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    if (irpSp->Parameters.DeviceIoControl.OutputBufferLength < sizeof(DISK_CACHE_SETTING))
    {
        status = STATUS_BUFFER_TOO_SMALL;
    }
    else
    {
        PDISK_CACHE_SETTING cacheSetting = (PDISK_CACHE_SETTING)Irp->AssociatedIrp.SystemBuffer;
 
        cacheSetting->Version = sizeof(DISK_CACHE_SETTING);
        cacheSetting->State   = DiskCacheNormal;
 
        //
        // Determine whether it is safe to turn on the cache
        //
        if (TEST_FLAG(fdoExtension->ScanForSpecialFlags, CLASS_SPECIAL_FUA_NOT_SUPPORTED))
        {
            cacheSetting->State = DiskCacheWriteThroughNotSupported;
        }
 
        //
        // Determine whether it is possible to modify the cache setting
        //
        if (TEST_FLAG(fdoExtension->ScanForSpecialFlags, CLASS_SPECIAL_MODIFY_CACHE_UNSUCCESSFUL))
        {
            cacheSetting->State = DiskCacheModifyUnsuccessful;
        }
 
        cacheSetting->IsPowerProtected = TEST_FLAG(fdoExtension->DeviceFlags, DEV_POWER_PROTECTED);
 
        Irp->IoStatus.Information = sizeof(DISK_CACHE_SETTING);
    }
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlGetDriveGeometry()

NTSTATUS DiskIoctlGetDriveGeometry	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 3582 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(DISK_GEOMETRY)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetDriveGeometry: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    if (TEST_FLAG(DeviceObject->Characteristics, FILE_REMOVABLE_MEDIA)) {
 
        //
        // Issue ReadCapacity to update device extension
        // with information for current media.
        //
 
        status = DiskReadDriveCapacity(commonExtension->PartitionZeroExtension->DeviceObject);
 
        //
        // Note whether the drive is ready.
        //
 
        diskData->ReadyStatus = status;
 
        if (!NT_SUCCESS(status)) {
            return status;
        }
    }
 
    //
    // Copy drive geometry information from device extension.
    //
 
    RtlMoveMemory(Irp->AssociatedIrp.SystemBuffer,
                  &(fdoExtension->DiskGeometry),
                  sizeof(DISK_GEOMETRY));
 
    if (((PDISK_GEOMETRY)Irp->AssociatedIrp.SystemBuffer)->BytesPerSector == 0) {
        ((PDISK_GEOMETRY)Irp->AssociatedIrp.SystemBuffer)->BytesPerSector = 512;
    }
    Irp->IoStatus.Information = sizeof(DISK_GEOMETRY);
    return STATUS_SUCCESS;
}

Referenced by DiskDeviceControl().

DiskIoctlGetDriveGeometryEx()

NTSTATUS DiskIoctlGetDriveGeometryEx	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

DiskIoctlGetLengthInfo()

NTSTATUS DiskIoctlGetLengthInfo	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

DiskIoctlGetMediaTypesEx()

NTSTATUS DiskIoctlGetMediaTypesEx	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 3994 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status;
 
    PMODE_PARAMETER_HEADER modeData;
    PMODE_PARAMETER_BLOCK blockDescriptor;
    PSCSI_REQUEST_BLOCK srb;
    PCDB cdb;
    ULONG modeLength;
    ULONG retries = 4;
    UCHAR densityCode = 0;
    BOOLEAN writable = TRUE;
    BOOLEAN mediaPresent = FALSE;
    ULONG srbSize;
    PSTORAGE_REQUEST_BLOCK srbEx = NULL;
    PSTOR_ADDR_BTL8 storAddrBtl8 = NULL;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16 = NULL;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlGetMediaTypesEx: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(GET_MEDIA_TYPES)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetMediaTypesEx: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbSize = CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE;
    } else {
        srbSize = SCSI_REQUEST_BLOCK_SIZE;
    }
 
    srb = ExAllocatePoolWithTag(NonPagedPoolNx,
                                srbSize,
                                DISK_TAG_SRB);
 
    if (srb == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetMediaTypesEx: Unable to allocate memory.\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(srb, srbSize);
 
    //
    // Send a TUR to determine if media is present.
    //
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx = (PSTORAGE_REQUEST_BLOCK)srb;
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = srbSize;
        srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
        srbEx->RequestPriority = IoGetIoPriorityHint(Irp);
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
        srbEx->NumSrbExData = 1;
 
        // Set timeout value.
        srbEx->TimeOutValue = fdoExtension->TimeOutValue;
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
        //
        // Set up SCSI SRB extended data fields
        //
 
        srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
            sizeof(STOR_ADDR_BTL8);
        if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
            srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
            srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
            srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
            srbExDataCdb16->CdbLength = 6;
 
            cdb = (PCDB)srbExDataCdb16->Cdb;
        } else {
            // Should not happen
            NT_ASSERT(FALSE);
 
            FREE_POOL(srb);
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetMediaTypesEx: Insufficient extended SRB size.\n"));
            return STATUS_INTERNAL_ERROR;
        }
 
    } else {
 
        srb->Length = SCSI_REQUEST_BLOCK_SIZE;
        srb->Function = SRB_FUNCTION_EXECUTE_SCSI;
        srb->CdbLength = 6;
        cdb = (PCDB)srb->Cdb;
 
        //
        // Set timeout value.
        //
 
        srb->TimeOutValue = fdoExtension->TimeOutValue;
 
    }
    cdb->CDB6GENERIC.OperationCode = SCSIOP_TEST_UNIT_READY;
 
    status = ClassSendSrbSynchronous(DeviceObject,
                                     srb,
                                     NULL,
                                     0,
                                     FALSE);
 
    if (NT_SUCCESS(status)) {
        mediaPresent = TRUE;
    }
 
    modeLength = MODE_DATA_SIZE;
    modeData = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                     modeLength,
                                     DISK_TAG_MODE_DATA);
 
    if (modeData == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetMediaTypesEx: Unable to allocate memory.\n"));
        FREE_POOL(srb);
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(modeData, modeLength);
 
    //
    // Build the MODE SENSE CDB using previous SRB.
    //
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx->SrbStatus = 0;
        srbExDataCdb16->ScsiStatus = 0;
        srbExDataCdb16->CdbLength = 6;
 
        //
        // Set timeout value from device extension.
        //
 
        srbEx->TimeOutValue = fdoExtension->TimeOutValue;
    } else {
        srb->SrbStatus = 0;
        srb->ScsiStatus = 0;
        srb->CdbLength = 6;
 
        //
        // Set timeout value from device extension.
        //
 
        srb->TimeOutValue = fdoExtension->TimeOutValue;
    }
 
    //
    // Page code of 0x3F will return all pages.
    // This command could fail if the data to be returned is
    // more than 256 bytes. In which case, we should get only
    // the caching page since we only need the block descriptor.
    // DiskFdoProcessError will change the page code to
    // MODE_PAGE_CACHING if there is an error.
    //
 
    cdb->MODE_SENSE.OperationCode    = SCSIOP_MODE_SENSE;
    cdb->MODE_SENSE.PageCode         = MODE_SENSE_RETURN_ALL;
    cdb->MODE_SENSE.AllocationLength = (UCHAR)modeLength;
 
Retry:
    status = ClassSendSrbSynchronous(DeviceObject,
                                     srb,
                                     modeData,
                                     modeLength,
                                     FALSE);
 
    if (status == STATUS_VERIFY_REQUIRED) {
 
        if (retries--) {
 
            //
            // Retry request.
            //
 
            goto Retry;
        }
    } else if (SRB_STATUS(srb->SrbStatus) == SRB_STATUS_DATA_OVERRUN) {
        status = STATUS_SUCCESS;
    }
 
    if (NT_SUCCESS(status) || (status == STATUS_NO_MEDIA_IN_DEVICE)) {
 
        //
        // Get the block descriptor.
        //
 
        if (modeData->BlockDescriptorLength != 0) {
 
            blockDescriptor = (PMODE_PARAMETER_BLOCK)((ULONG_PTR)modeData + sizeof(MODE_PARAMETER_HEADER));
            densityCode = blockDescriptor->DensityCode;
        }
 
        if (TEST_FLAG(modeData->DeviceSpecificParameter,
                      MODE_DSP_WRITE_PROTECT)) {
 
            writable = FALSE;
        }
 
        status = DiskDetermineMediaTypes(DeviceObject,
                                         Irp,
                                         modeData->MediumType,
                                         densityCode,
                                         mediaPresent,
                                         writable);
        //
        // If the buffer was too small, DetermineMediaTypes updated the status and information and the request will fail.
        //
 
    } else {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetMediaTypesEx: Mode sense for header/bd failed. %lx\n", status));
    }
 
    FREE_POOL(srb);
    FREE_POOL(modeData);
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlGetVolumeDiskExtents()

NTSTATUS DiskIoctlGetVolumeDiskExtents	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5461 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status = STATUS_INVALID_DEVICE_REQUEST;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlGetVolumeDiskExtents: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
 
    if (TEST_FLAG(DeviceObject->Characteristics, FILE_REMOVABLE_MEDIA)) {
 
        if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(VOLUME_DISK_EXTENTS)) {
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlGetVolumeDiskExtents: Output buffer too small.\n"));
            return STATUS_BUFFER_TOO_SMALL;
        }
 
        status = DiskReadDriveCapacity(commonExtension->PartitionZeroExtension->DeviceObject);
 
        //
        // Note whether the drive is ready.
        //
 
        diskData->ReadyStatus = status;
 
        if (NT_SUCCESS(status)) {
 
            PVOLUME_DISK_EXTENTS pVolExt = (PVOLUME_DISK_EXTENTS)Irp->AssociatedIrp.SystemBuffer;
 
            pVolExt->NumberOfDiskExtents = 1;
            pVolExt->Extents[0].DiskNumber     = commonExtension->PartitionZeroExtension->DeviceNumber;
            pVolExt->Extents[0].StartingOffset = commonExtension->StartingOffset;
            pVolExt->Extents[0].ExtentLength   = commonExtension->PartitionLength;
 
            Irp->IoStatus.Information = sizeof(VOLUME_DISK_EXTENTS);
        }
    }
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlIsWritable()

NTSTATUS DiskIoctlIsWritable	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5063 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    NTSTATUS status = STATUS_SUCCESS;
 
    PMODE_PARAMETER_HEADER modeData;
    PSCSI_REQUEST_BLOCK srb;
    PCDB cdb = NULL;
    ULONG modeLength;
    ULONG retries = 4;
    ULONG srbSize;
    PSTORAGE_REQUEST_BLOCK srbEx;
    PSTOR_ADDR_BTL8 storAddrBtl8;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlIsWritable: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbSize = CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE;
    } else {
        srbSize = SCSI_REQUEST_BLOCK_SIZE;
    }
    srb = ExAllocatePoolWithTag(NonPagedPoolNx,
                                srbSize,
                                DISK_TAG_SRB);
 
    if (srb == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlIsWritable: Unable to allocate memory.\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(srb, srbSize);
 
    //
    // Allocate memory for a mode header and then some
    // for port drivers that need to convert to MODE10
    // or always return the MODE_PARAMETER_BLOCK (even
    // when memory was not allocated for this purpose)
    //
 
    modeLength = MODE_DATA_SIZE;
    modeData = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                     modeLength,
                                     DISK_TAG_MODE_DATA);
 
    if (modeData == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlIsWritable: Unable to allocate memory.\n"));
        FREE_POOL(srb);
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(modeData, modeLength);
 
    //
    // Build the MODE SENSE CDB
    //
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx = (PSTORAGE_REQUEST_BLOCK)srb;
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = srbSize;
        srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
        srbEx->RequestPriority = IoGetIoPriorityHint(Irp);
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
        srbEx->NumSrbExData = 1;
 
        // Set timeout value.
        srbEx->TimeOutValue = fdoExtension->TimeOutValue;
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
        //
        // Set up SCSI SRB extended data fields
        //
 
        srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
            sizeof(STOR_ADDR_BTL8);
        if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
            srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
            srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
            srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
            srbExDataCdb16->CdbLength = 6;
 
            cdb = (PCDB)srbExDataCdb16->Cdb;
        } else {
            // Should not happen
            NT_ASSERT(FALSE);
 
            FREE_POOL(srb);
            FREE_POOL(modeData);
            return STATUS_INTERNAL_ERROR;
        }
 
    } else {
        srb->Length = SCSI_REQUEST_BLOCK_SIZE;
        srb->Function = SRB_FUNCTION_EXECUTE_SCSI;
        srb->CdbLength = 6;
 
        //
        // Set timeout value.
        //
 
        srb->TimeOutValue = fdoExtension->TimeOutValue;
 
        cdb = (PCDB)srb->Cdb;
    }
 
    //
    // Page code of 0x3F will return all pages.
    // This command could fail if the data to be returned is
    // more than 256 bytes. In which case, we should get only
    // the caching page since we only need the block descriptor.
    // DiskFdoProcessError will change the page code to
    // MODE_PAGE_CACHING if there is an error.
    //
 
    cdb->MODE_SENSE.OperationCode    = SCSIOP_MODE_SENSE;
    cdb->MODE_SENSE.PageCode         = MODE_SENSE_RETURN_ALL;
    cdb->MODE_SENSE.AllocationLength = (UCHAR)modeLength;
 
    while (retries != 0) {
 
        status = ClassSendSrbSynchronous(DeviceObject,
                                         srb,
                                         modeData,
                                         modeLength,
                                         FALSE);
 
        if (status != STATUS_VERIFY_REQUIRED) {
            if (SRB_STATUS(srb->SrbStatus) == SRB_STATUS_DATA_OVERRUN) {
                status = STATUS_SUCCESS;
            }
            break;
        }
        retries--;
    }
 
    if (NT_SUCCESS(status)) {
 
        if (TEST_FLAG(modeData->DeviceSpecificParameter, MODE_DSP_WRITE_PROTECT)) {
            status = STATUS_MEDIA_WRITE_PROTECTED;
        }
    }
 
    FREE_POOL(srb);
    FREE_POOL(modeData);
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlPredictFailure()

NTSTATUS DiskIoctlPredictFailure	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 4268 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status = STATUS_SUCCESS;
 
    PSTORAGE_PREDICT_FAILURE checkFailure;
    STORAGE_FAILURE_PREDICT_STATUS diskSmartStatus;
    IO_STATUS_BLOCK ioStatus = { 0 };
    KEVENT event;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlPredictFailure: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(STORAGE_PREDICT_FAILURE)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlPredictFailure: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    //
    // See if the disk is predicting failure
    //
 
    checkFailure = (PSTORAGE_PREDICT_FAILURE)Irp->AssociatedIrp.SystemBuffer;
 
    if (diskData->FailurePredictionCapability == FailurePredictionSense) {
        ULONG readBufferSize;
        PUCHAR readBuffer;
        PIRP readIrp;
        PDEVICE_OBJECT topOfStack;
 
        checkFailure->PredictFailure = 0;
 
        KeInitializeEvent(&event, SynchronizationEvent, FALSE);
 
        topOfStack = IoGetAttachedDeviceReference(DeviceObject);
 
        //
        // SCSI disks need to have a read sent down to provoke any
        // failures to be reported.
        //
        // Issue a normal read operation.  The error-handling code in
        // classpnp will take care of a failure prediction by logging the
        // correct event.
        //
 
        readBufferSize = fdoExtension->DiskGeometry.BytesPerSector;
        readBuffer = ExAllocatePoolWithTag(NonPagedPoolNx,
                                           readBufferSize,
                                           DISK_TAG_SMART);
 
        if (readBuffer != NULL) {
            LARGE_INTEGER offset;
 
            offset.QuadPart = 0;
            readIrp = IoBuildSynchronousFsdRequest(IRP_MJ_READ,
                                                   topOfStack,
                                                   readBuffer,
                                                   readBufferSize,
                                                   &offset,
                                                   &event,
                                                   &ioStatus);
 
            if (readIrp != NULL) {
 
                status = IoCallDriver(topOfStack, readIrp);
                if (status == STATUS_PENDING) {
                    KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL);
                    status = ioStatus.Status;
                }
 
 
            } else {
                status = STATUS_INSUFFICIENT_RESOURCES;
            }
 
            FREE_POOL(readBuffer);
        } else {
            status = STATUS_INSUFFICIENT_RESOURCES;
        }
 
        ObDereferenceObject(topOfStack);
    }
 
    if (status != STATUS_INSUFFICIENT_RESOURCES)
    {
        if ((diskData->FailurePredictionCapability == FailurePredictionSmart) ||
            (diskData->FailurePredictionCapability == FailurePredictionSense)) {
 
            status = DiskReadFailurePredictStatus(fdoExtension, &diskSmartStatus);
 
            if (NT_SUCCESS(status)) {
 
                status = DiskReadFailurePredictData(fdoExtension,
                                                    Irp->AssociatedIrp.SystemBuffer);
 
                if (diskSmartStatus.PredictFailure) {
                    checkFailure->PredictFailure = 1;
                } else {
                    checkFailure->PredictFailure = 0;
                }
 
                Irp->IoStatus.Information = sizeof(STORAGE_PREDICT_FAILURE);
            }
        } else {
            status = STATUS_INVALID_DEVICE_REQUEST;
        }
    }
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlReassignBlocks()

NTSTATUS DiskIoctlReassignBlocks	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 4648 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status;
    PREASSIGN_BLOCKS badBlocks = Irp->AssociatedIrp.SystemBuffer;
    PSCSI_REQUEST_BLOCK srb;
    PCDB cdb;
    ULONG bufferSize;
    ULONG blockNumber;
    ULONG blockCount;
    ULONG srbSize;
    PSTORAGE_REQUEST_BLOCK srbEx;
    PSTOR_ADDR_BTL8 storAddrBtl8;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(REASSIGN_BLOCKS)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Input buffer length mismatch.\n"));
        return STATUS_INFO_LENGTH_MISMATCH;
    }
 
    //
    // Make sure we have some data in the input buffer.
    //
 
    if (badBlocks->Count == 0) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Invalid block count\n"));
        return STATUS_INVALID_PARAMETER;
    }
 
    bufferSize = sizeof(REASSIGN_BLOCKS) + ((badBlocks->Count - 1) * sizeof(ULONG));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < bufferSize) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Input buffer length mismatch for bad blocks.\n"));
        return STATUS_INFO_LENGTH_MISMATCH;
    }
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbSize = CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE;
    } else {
        srbSize = SCSI_REQUEST_BLOCK_SIZE;
    }
    srb = ExAllocatePoolWithTag(NonPagedPoolNx,
                                srbSize,
                                DISK_TAG_SRB);
 
    if (srb == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Unable to allocate memory.\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(srb, srbSize);
 
    //
    // Build the data buffer to be transferred in the input buffer.
    // The format of the data to the device is:
    //
    //      2 bytes Reserved
    //      2 bytes Length
    //      x * 4 btyes Block Address
    //
    // All values are big endian.
    //
 
    badBlocks->Reserved = 0;
    blockCount = badBlocks->Count;
 
    //
    // Convert # of entries to # of bytes.
    //
 
    blockCount *= 4;
    badBlocks->Count = (USHORT) ((blockCount >> 8) & 0XFF);
    badBlocks->Count |= (USHORT) ((blockCount << 8) & 0XFF00);
 
    //
    // Convert back to number of entries.
    //
 
    blockCount /= 4;
 
    for (; blockCount > 0; blockCount--) {
 
        blockNumber = badBlocks->BlockNumber[blockCount-1];
        REVERSE_BYTES((PFOUR_BYTE) &badBlocks->BlockNumber[blockCount-1], (PFOUR_BYTE) &blockNumber);
    }
 
    //
    // Build a SCSI SRB containing a SCSIOP_REASSIGN_BLOCKS cdb
    //
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx = (PSTORAGE_REQUEST_BLOCK)srb;
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = srbSize;
        srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
        srbEx->RequestPriority = IoGetIoPriorityHint(Irp);
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
        srbEx->NumSrbExData = 1;
 
        // Set timeout value.
        srbEx->TimeOutValue = fdoExtension->TimeOutValue;
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
        //
        // Set up SCSI SRB extended data fields
        //
 
        srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
            sizeof(STOR_ADDR_BTL8);
        if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
            srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
            srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
            srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
            srbExDataCdb16->CdbLength = 6;
 
            cdb = (PCDB)srbExDataCdb16->Cdb;
        } else {
            // Should not happen
            NT_ASSERT(FALSE);
 
            FREE_POOL(srb);
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Insufficient extended SRB size.\n"));
            return STATUS_INTERNAL_ERROR;
        }
 
    } else {
        srb->Length = SCSI_REQUEST_BLOCK_SIZE;
        srb->Function = SRB_FUNCTION_EXECUTE_SCSI;
        srb->CdbLength = 6;
 
        //
        // Set timeout value.
        //
 
        srb->TimeOutValue = fdoExtension->TimeOutValue;
 
        cdb = (PCDB)srb->Cdb;
    }
 
    cdb->CDB6GENERIC.OperationCode = SCSIOP_REASSIGN_BLOCKS;
 
    status = ClassSendSrbSynchronous(DeviceObject,
                                     srb,
                                     badBlocks,
                                     bufferSize,
                                     TRUE);
 
    FREE_POOL(srb);
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlReassignBlocksEx()

NTSTATUS DiskIoctlReassignBlocksEx	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 4855 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status;
    PREASSIGN_BLOCKS_EX badBlocks = Irp->AssociatedIrp.SystemBuffer;
    PSCSI_REQUEST_BLOCK srb;
    PCDB cdb;
    LARGE_INTEGER blockNumber;
    ULONG bufferSize;
    ULONG blockCount;
    ULONG srbSize;
    PSTORAGE_REQUEST_BLOCK srbEx;
    PSTOR_ADDR_BTL8 storAddrBtl8;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocksEx: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(REASSIGN_BLOCKS_EX)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocksEx: Input buffer length mismatch.\n"));
        return STATUS_INFO_LENGTH_MISMATCH;
    }
 
    //
    // Make sure we have some data in the input buffer.
    //
 
    if (badBlocks->Count == 0) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocksEx: Invalid block count\n"));
        return STATUS_INVALID_PARAMETER;
    }
 
    bufferSize = sizeof(REASSIGN_BLOCKS_EX) + ((badBlocks->Count - 1) * sizeof(LARGE_INTEGER));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < bufferSize) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocksEx: Input buffer length mismatch for bad blocks.\n"));
        return STATUS_INFO_LENGTH_MISMATCH;
    }
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbSize = CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE;
    } else {
        srbSize = SCSI_REQUEST_BLOCK_SIZE;
    }
    srb = ExAllocatePoolWithTag(NonPagedPoolNx,
                                srbSize,
                                DISK_TAG_SRB);
 
    if (srb == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Unable to allocate memory.\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(srb, srbSize);
 
    //
    // Build the data buffer to be transferred in the input buffer.
    // The format of the data to the device is:
    //
    //      2 bytes Reserved
    //      2 bytes Length
    //      x * 8 btyes Block Address
    //
    // All values are big endian.
    //
 
    badBlocks->Reserved = 0;
    blockCount = badBlocks->Count;
 
    //
    // Convert # of entries to # of bytes.
    //
 
    blockCount *= 8;
    badBlocks->Count = (USHORT) ((blockCount >> 8) & 0XFF);
    badBlocks->Count |= (USHORT) ((blockCount << 8) & 0XFF00);
 
    //
    // Convert back to number of entries.
    //
 
    blockCount /= 8;
 
    for (; blockCount > 0; blockCount--) {
 
        blockNumber = badBlocks->BlockNumber[blockCount-1];
        REVERSE_BYTES_QUAD(&badBlocks->BlockNumber[blockCount-1], &blockNumber);
    }
 
    //
    // Build a SCSI SRB containing a SCSIOP_REASSIGN_BLOCKS cdb
    //
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx = (PSTORAGE_REQUEST_BLOCK)srb;
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = srbSize;
        srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
        srbEx->RequestPriority = IoGetIoPriorityHint(Irp);
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
        srbEx->NumSrbExData = 1;
 
        // Set timeout value.
        srbEx->TimeOutValue = fdoExtension->TimeOutValue;
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
        //
        // Set up SCSI SRB extended data fields
        //
 
        srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
            sizeof(STOR_ADDR_BTL8);
        if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
            srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
            srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
            srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
            srbExDataCdb16->CdbLength = 6;
 
            cdb = (PCDB)srbExDataCdb16->Cdb;
        } else {
            // Should not happen
            NT_ASSERT(FALSE);
 
            FREE_POOL(srb);
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlReassignBlocks: Insufficient extended SRB size.\n"));
            return STATUS_INTERNAL_ERROR;
        }
 
    } else {
        srb->Length = SCSI_REQUEST_BLOCK_SIZE;
        srb->Function = SRB_FUNCTION_EXECUTE_SCSI;
        srb->CdbLength = 6;
 
        //
        // Set timeout value.
        //
 
        srb->TimeOutValue = fdoExtension->TimeOutValue;
 
        cdb = (PCDB)srb->Cdb;
    }
 
    cdb->CDB6GENERIC.OperationCode = SCSIOP_REASSIGN_BLOCKS;
    cdb->CDB6GENERIC.CommandUniqueBits =  1; // LONGLBA
 
    status = ClassSendSrbSynchronous(DeviceObject,
                                     srb,
                                     badBlocks,
                                     bufferSize,
                                     TRUE);
 
    FREE_POOL(srb);
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlSetCacheInformation()

NTSTATUS DiskIoctlSetCacheInformation	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 3917 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    PDISK_CACHE_INFORMATION cacheInfo = Irp->AssociatedIrp.SystemBuffer;
    NTSTATUS status;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL is equal or above DISPATCH_LEVEL.
    //
 
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlSetCacheInformation: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(DISK_CACHE_INFORMATION)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSetCacheInformation: Input buffer length mismatch.\n"));
        return STATUS_INFO_LENGTH_MISMATCH;
    }
 
    status = DiskSetCacheInformation(fdoExtension, cacheInfo);
 
    //
    // Save away the user-defined override in our extension and the registry
    //
    if (cacheInfo->WriteCacheEnabled) {
        diskData->WriteCacheOverride = DiskWriteCacheEnable;
    } else {
        diskData->WriteCacheOverride = DiskWriteCacheDisable;
    }
 
    ClassSetDeviceParameter(fdoExtension, DiskDeviceParameterSubkey,
                            DiskDeviceUserWriteCacheSetting, diskData->WriteCacheOverride);
 
    DiskLogCacheInformation(fdoExtension, cacheInfo, status);
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlSetCacheSetting()

NTSTATUS DiskIoctlSetCacheSetting	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 3415 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(Irp);
    NTSTATUS status = STATUS_SUCCESS;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    if (irpSp->Parameters.DeviceIoControl.InputBufferLength < sizeof(DISK_CACHE_SETTING))
    {
        status = STATUS_INFO_LENGTH_MISMATCH;
    }
    else
    {
        PDISK_CACHE_SETTING cacheSetting = (PDISK_CACHE_SETTING)Irp->AssociatedIrp.SystemBuffer;
 
        if (cacheSetting->Version == sizeof(DISK_CACHE_SETTING))
        {
            ULONG isPowerProtected;
 
            //
            // Save away the user-defined override in our extension and the registry
            //
            if (cacheSetting->IsPowerProtected)
            {
                SET_FLAG(fdoExtension->DeviceFlags, DEV_POWER_PROTECTED);
                isPowerProtected = 1;
            }
            else
            {
                CLEAR_FLAG(fdoExtension->DeviceFlags, DEV_POWER_PROTECTED);
                isPowerProtected = 0;
            }
            ADJUST_FUA_FLAG(fdoExtension);
 
            ClassSetDeviceParameter(fdoExtension, DiskDeviceParameterSubkey, DiskDeviceCacheIsPowerProtected, isPowerProtected);
        }
        else
        {
            status = STATUS_INVALID_PARAMETER;
        }
    }
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlSetVerify()

NTSTATUS DiskIoctlSetVerify	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5264 of file disk.c.

{
    NTSTATUS status = STATUS_NOT_SUPPORTED;
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlSetVerify: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    //
    // If the caller is kernel mode, set the verify bit.
    //
 
    if (Irp->RequestorMode == KernelMode) {
 
        SET_FLAG(DeviceObject->Flags, DO_VERIFY_VOLUME);
        status = STATUS_SUCCESS;
 
    }
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlSmartGetVersion()

NTSTATUS DiskIoctlSmartGetVersion	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5540 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status;
 
    PGETVERSIONINPARAMS versionParams;
    PSRB_IO_CONTROL srbControl;
    IO_STATUS_BLOCK ioStatus = { 0 };
    PUCHAR buffer;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlSmartGetVersion: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(GETVERSIONINPARAMS)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartGetVersion: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    srbControl = ExAllocatePoolWithTag(NonPagedPoolNx,
                                       sizeof(SRB_IO_CONTROL) +
                                       sizeof(GETVERSIONINPARAMS),
                                       DISK_TAG_SMART);
 
    if (srbControl == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartGetVersion: Unable to allocate memory.\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(srbControl, sizeof(SRB_IO_CONTROL) + sizeof(GETVERSIONINPARAMS));
 
    //
    // fill in srbControl fields
    //
 
    srbControl->HeaderLength = sizeof(SRB_IO_CONTROL);
    RtlMoveMemory (srbControl->Signature, "SCSIDISK", 8);
    srbControl->Timeout = fdoExtension->TimeOutValue;
    srbControl->Length = sizeof(GETVERSIONINPARAMS);
    srbControl->ControlCode = IOCTL_SCSI_MINIPORT_SMART_VERSION;
 
    //
    // Point to the 'buffer' portion of the SRB_CONTROL
    //
 
    buffer = (PUCHAR)srbControl + srbControl->HeaderLength;
 
    //
    // Ensure correct target is set in the cmd parameters.
    //
 
    versionParams = (PGETVERSIONINPARAMS)buffer;
    versionParams->bIDEDeviceMap = diskData->ScsiAddress.TargetId;
 
    ClassSendDeviceIoControlSynchronous(
                                    IOCTL_SCSI_MINIPORT,
                                    commonExtension->LowerDeviceObject,
                                    srbControl,
                                    sizeof(SRB_IO_CONTROL) + sizeof(GETVERSIONINPARAMS),
                                    sizeof(SRB_IO_CONTROL) + sizeof(GETVERSIONINPARAMS),
                                    FALSE,
                                    &ioStatus);
 
    status = ioStatus.Status;
 
    //
    // If successful, copy the data received into the output buffer.
    // This should only fail in the event that the IDE driver is older
    // than this driver.
    //
 
    if (NT_SUCCESS(status)) {
 
        buffer = (PUCHAR)srbControl + srbControl->HeaderLength;
 
        RtlMoveMemory (Irp->AssociatedIrp.SystemBuffer, buffer, sizeof(GETVERSIONINPARAMS));
        Irp->IoStatus.Information = sizeof(GETVERSIONINPARAMS);
    }
 
    FREE_POOL(srbControl);
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlSmartReceiveDriveData()

NTSTATUS DiskIoctlSmartReceiveDriveData	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5663 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status = STATUS_INVALID_PARAMETER;
 
    PSENDCMDINPARAMS cmdInParameters = ((PSENDCMDINPARAMS)Irp->AssociatedIrp.SystemBuffer);
    PSRB_IO_CONTROL srbControl;
    IO_STATUS_BLOCK ioStatus = { 0 };
    ULONG controlCode = 0;
    PUCHAR buffer;
    PIRP irp2;
    KEVENT event;
    ULONG length = 0;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < (sizeof(SENDCMDINPARAMS) - 1)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: Input buffer length invalid.\n"));
        return STATUS_INVALID_PARAMETER;
    }
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < (sizeof(SENDCMDOUTPARAMS) + 512 - 1)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    //
    // Create notification event object to be used to signal the
    // request completion.
    //
 
    KeInitializeEvent(&event, NotificationEvent, FALSE);
 
    //
    // use controlCode as a sort of 'STATUS_SUCCESS' to see if it's
    // a valid request type
    //
 
    if (cmdInParameters->irDriveRegs.bCommandReg == ID_CMD) {
 
        length = IDENTIFY_BUFFER_SIZE + sizeof(SENDCMDOUTPARAMS);
        controlCode = IOCTL_SCSI_MINIPORT_IDENTIFY;
 
    } else if (cmdInParameters->irDriveRegs.bCommandReg == SMART_CMD) {
 
        switch (cmdInParameters->irDriveRegs.bFeaturesReg) {
 
            case READ_ATTRIBUTES:
                controlCode = IOCTL_SCSI_MINIPORT_READ_SMART_ATTRIBS;
                length = READ_ATTRIBUTE_BUFFER_SIZE + sizeof(SENDCMDOUTPARAMS);
                break;
 
            case READ_THRESHOLDS:
                controlCode = IOCTL_SCSI_MINIPORT_READ_SMART_THRESHOLDS;
                length = READ_THRESHOLD_BUFFER_SIZE + sizeof(SENDCMDOUTPARAMS);
                break;
 
            case SMART_READ_LOG: {
 
                if (diskData->FailurePredictionCapability != FailurePredictionSmart) {
                    TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: SMART failure prediction not supported.\n"));
                    return STATUS_INVALID_DEVICE_REQUEST;
                }
 
                //
                // Calculate additional length based on number of sectors to be read.
                // Then verify the output buffer is large enough.
                //
 
                length = cmdInParameters->irDriveRegs.bSectorCountReg * SMART_LOG_SECTOR_SIZE;
 
                //
                // Ensure at least 1 sector is going to be read
                //
                if (length == 0) {
                    return STATUS_INVALID_PARAMETER;
                }
 
                length += max(sizeof(SENDCMDOUTPARAMS), sizeof(SENDCMDINPARAMS));
 
                if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < length - 1) {
                    TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: Output buffer too small for SMART_READ_LOG.\n"));
                    return STATUS_BUFFER_TOO_SMALL;
                }
 
                controlCode = IOCTL_SCSI_MINIPORT_READ_SMART_LOG;
                break;
            }
        }
    }
 
    if (controlCode == 0) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: Invalid request.\n"));
        return STATUS_INVALID_PARAMETER;
    }
 
    srbControl = ExAllocatePoolWithTag(NonPagedPoolNx,
                                       sizeof(SRB_IO_CONTROL) + length,
                                       DISK_TAG_SMART);
 
    if (srbControl == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: Unable to allocate memory.\n"));
        return  STATUS_INSUFFICIENT_RESOURCES;
    }
 
    //
    // fill in srbControl fields
    //
 
    srbControl->HeaderLength = sizeof(SRB_IO_CONTROL);
    RtlMoveMemory (srbControl->Signature, "SCSIDISK", 8);
    srbControl->Timeout = fdoExtension->TimeOutValue;
    srbControl->Length = length;
    srbControl->ControlCode = controlCode;
 
    //
    // Point to the 'buffer' portion of the SRB_CONTROL
    //
 
    buffer = (PUCHAR)srbControl + srbControl->HeaderLength;
 
    //
    // Ensure correct target is set in the cmd parameters.
    //
 
    cmdInParameters->bDriveNumber = diskData->ScsiAddress.TargetId;
 
    //
    // Copy the IOCTL parameters to the srb control buffer area.
    //
 
    RtlMoveMemory(buffer,
                  Irp->AssociatedIrp.SystemBuffer,
                  sizeof(SENDCMDINPARAMS) - 1);
 
    irp2 = IoBuildDeviceIoControlRequest(IOCTL_SCSI_MINIPORT,
                                        commonExtension->LowerDeviceObject,
                                        srbControl,
                                        sizeof(SRB_IO_CONTROL) + sizeof(SENDCMDINPARAMS) - 1,
                                        srbControl,
                                        sizeof(SRB_IO_CONTROL) + length,
                                        FALSE,
                                        &event,
                                        &ioStatus);
 
    if (irp2 == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartReceiveDriveData: Unable to allocate IRP.\n"));
        FREE_POOL(srbControl);
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    //
    // Call the port driver with the request and wait for it to complete.
    //
 
    status = IoCallDriver(commonExtension->LowerDeviceObject, irp2);
 
    if (status == STATUS_PENDING) {
        KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL);
        status = ioStatus.Status;
    }
 
    //
    // Copy the data received into the output buffer. Since the status buffer
    // contains error information also, always perform this copy. IO will
    // either pass this back to the app, or zero it, in case of error.
    //
 
    buffer = (PUCHAR)srbControl + srbControl->HeaderLength;
 
    if (NT_SUCCESS(status)) {
 
        RtlMoveMemory ( Irp->AssociatedIrp.SystemBuffer, buffer, length - 1);
        Irp->IoStatus.Information = length - 1;
 
    } else {
 
        RtlMoveMemory ( Irp->AssociatedIrp.SystemBuffer, buffer, (sizeof(SENDCMDOUTPARAMS) - 1));
        Irp->IoStatus.Information = sizeof(SENDCMDOUTPARAMS) - 1;
 
    }
 
    FREE_POOL(srbControl);
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlSmartSendDriveCommand()

NTSTATUS DiskIoctlSmartSendDriveCommand	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5890 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    NTSTATUS status = STATUS_INVALID_PARAMETER;
 
    PSENDCMDINPARAMS cmdInParameters = ((PSENDCMDINPARAMS)Irp->AssociatedIrp.SystemBuffer);
    PSRB_IO_CONTROL srbControl;
    IO_STATUS_BLOCK ioStatus = { 0 };
    ULONG controlCode = 0;
    PUCHAR buffer;
    PIRP irp2;
    KEVENT event;
    ULONG length = 0;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < (sizeof(SENDCMDINPARAMS) - 1)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: Input buffer size invalid.\n"));
        return STATUS_INVALID_PARAMETER;
    }
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < (sizeof(SENDCMDOUTPARAMS) - 1)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    //
    // Create notification event object to be used to signal the
    // request completion.
    //
 
    KeInitializeEvent(&event, NotificationEvent, FALSE);
 
    if (cmdInParameters->irDriveRegs.bCommandReg == SMART_CMD) {
 
        switch (cmdInParameters->irDriveRegs.bFeaturesReg) {
 
            case SMART_WRITE_LOG: {
 
                if (diskData->FailurePredictionCapability != FailurePredictionSmart) {
 
                    TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: SMART failure prediction not supported.\n"));
                    return STATUS_INVALID_DEVICE_REQUEST;
                }
 
                //
                // Calculate additional length based on number of sectors to be written.
                // Then verify the input buffer is large enough.
                //
 
                length = cmdInParameters->irDriveRegs.bSectorCountReg * SMART_LOG_SECTOR_SIZE;
 
                //
                // Ensure at least 1 sector is going to be written
                //
 
                if (length == 0) {
                    return STATUS_INVALID_PARAMETER;
                }
 
                if (irpStack->Parameters.DeviceIoControl.InputBufferLength <
                        (sizeof(SENDCMDINPARAMS) - 1) + length) {
 
                    TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: Input buffer too small for SMART_WRITE_LOG.\n"));
                    return STATUS_BUFFER_TOO_SMALL;
                }
 
                controlCode = IOCTL_SCSI_MINIPORT_WRITE_SMART_LOG;
                break;
            }
 
            case ENABLE_SMART:
                controlCode = IOCTL_SCSI_MINIPORT_ENABLE_SMART;
                break;
 
            case DISABLE_SMART:
                controlCode = IOCTL_SCSI_MINIPORT_DISABLE_SMART;
                break;
 
            case RETURN_SMART_STATUS:
 
                //
                // Ensure bBuffer is at least 2 bytes (to hold the values of
                // cylinderLow and cylinderHigh).
                //
 
                if (irpStack->Parameters.DeviceIoControl.OutputBufferLength <
                    (sizeof(SENDCMDOUTPARAMS) - 1 + sizeof(IDEREGS))) {
 
                    return STATUS_BUFFER_TOO_SMALL;
                    break;
                }
 
                controlCode = IOCTL_SCSI_MINIPORT_RETURN_STATUS;
                length = sizeof(IDEREGS);
                break;
 
            case ENABLE_DISABLE_AUTOSAVE:
                controlCode = IOCTL_SCSI_MINIPORT_ENABLE_DISABLE_AUTOSAVE;
                break;
 
            case SAVE_ATTRIBUTE_VALUES:
                controlCode = IOCTL_SCSI_MINIPORT_SAVE_ATTRIBUTE_VALUES;
                break;
 
            case EXECUTE_OFFLINE_DIAGS:
                //
                // Validate that this is an ok self test command
                //
                if (DiskIsValidSmartSelfTest(cmdInParameters->irDriveRegs.bSectorNumberReg)) {
 
                    controlCode = IOCTL_SCSI_MINIPORT_EXECUTE_OFFLINE_DIAGS;
                }
                break;
 
            case ENABLE_DISABLE_AUTO_OFFLINE:
                controlCode = IOCTL_SCSI_MINIPORT_ENABLE_DISABLE_AUTO_OFFLINE;
                break;
        }
    }
 
    if (controlCode == 0) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: Invalid request.\n"));
        return STATUS_INVALID_PARAMETER;
    }
 
    length += max(sizeof(SENDCMDOUTPARAMS), sizeof(SENDCMDINPARAMS));
    srbControl = ExAllocatePoolWithTag(NonPagedPoolNx,
                                       sizeof(SRB_IO_CONTROL) + length,
                                       DISK_TAG_SMART);
 
    if (srbControl == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: Unable to allocate memory.\n"));
        return  STATUS_INSUFFICIENT_RESOURCES;
    }
 
    //
    // fill in srbControl fields
    //
 
    srbControl->HeaderLength = sizeof(SRB_IO_CONTROL);
    RtlMoveMemory (srbControl->Signature, "SCSIDISK", 8);
    srbControl->Timeout = fdoExtension->TimeOutValue;
    srbControl->Length = length;
    srbControl->ControlCode = controlCode;
 
    //
    // Point to the 'buffer' portion of the SRB_CONTROL
    //
 
    buffer = (PUCHAR)srbControl + srbControl->HeaderLength;
 
    //
    // Ensure correct target is set in the cmd parameters.
    //
 
    cmdInParameters->bDriveNumber = diskData->ScsiAddress.TargetId;
 
    //
    // Copy the IOCTL parameters to the srb control buffer area.
    //
 
    if (cmdInParameters->irDriveRegs.bFeaturesReg == SMART_WRITE_LOG) {
        RtlMoveMemory(buffer,
                     Irp->AssociatedIrp.SystemBuffer,
                     sizeof(SENDCMDINPARAMS) - 1 +
                        cmdInParameters->irDriveRegs.bSectorCountReg * SMART_LOG_SECTOR_SIZE);
    } else {
        RtlMoveMemory(buffer, Irp->AssociatedIrp.SystemBuffer, sizeof(SENDCMDINPARAMS) - 1);
    }
 
    irp2 = IoBuildDeviceIoControlRequest(IOCTL_SCSI_MINIPORT,
                                        commonExtension->LowerDeviceObject,
                                        srbControl,
                                        sizeof(SRB_IO_CONTROL) + length,
                                        srbControl,
                                        sizeof(SRB_IO_CONTROL) + length,
                                        FALSE,
                                        &event,
                                        &ioStatus);
 
    if (irp2 == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlSmartSendDriveCommand: Unable to allocate IRP.\n"));
        FREE_POOL(srbControl);
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    //
    // Call the port driver with the request and wait for it to complete.
    //
 
    status = IoCallDriver(commonExtension->LowerDeviceObject, irp2);
 
    if (status == STATUS_PENDING) {
        KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL);
        status = ioStatus.Status;
    }
 
    //
    // Copy the data received into the output buffer. Since the status buffer
    // contains error information also, always perform this copy. IO will
    // either pass this back to the app, or zero it, in case of error.
    //
 
    buffer = (PUCHAR)srbControl + srbControl->HeaderLength;
 
    //
    // Update the return buffer size based on the sub-command.
    //
 
    if (cmdInParameters->irDriveRegs.bFeaturesReg == RETURN_SMART_STATUS) {
        length = sizeof(SENDCMDOUTPARAMS) - 1 + sizeof(IDEREGS);
    } else {
        length = sizeof(SENDCMDOUTPARAMS) - 1;
    }
 
    RtlMoveMemory ( Irp->AssociatedIrp.SystemBuffer, buffer, length);
    Irp->IoStatus.Information = length;
 
    FREE_POOL(srbControl);
 
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlUpdateDriveSize()

NTSTATUS DiskIoctlUpdateDriveSize	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 5362 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    TARGET_DEVICE_CUSTOM_NOTIFICATION Notification = {0};
    NTSTATUS status;
 
    //
    // This function must be called at less than dispatch level.
    // Fail if IRQL >= DISPATCH_LEVEL.
    //
    PAGED_CODE();
    CHECK_IRQL();
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlUpdateDriveSize: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(DISK_GEOMETRY)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlUpdateDriveSize: Output buffer too small.\n"));
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    status = DiskReadDriveCapacity(DeviceObject);
 
    //
    // Note whether the drive is ready.
    //
 
    diskData->ReadyStatus = status;
 
    if (NT_SUCCESS(status)) {
 
        //
        // Copy drive geometry information from the device extension.
        //
 
        RtlMoveMemory(Irp->AssociatedIrp.SystemBuffer,
                      &(fdoExtension->DiskGeometry),
                      sizeof(DISK_GEOMETRY));
 
        if (((PDISK_GEOMETRY)Irp->AssociatedIrp.SystemBuffer)->BytesPerSector == 0) {
            ((PDISK_GEOMETRY)Irp->AssociatedIrp.SystemBuffer)->BytesPerSector = 512;
        }
        Irp->IoStatus.Information = sizeof(DISK_GEOMETRY);
        status = STATUS_SUCCESS;
 
        //
        // Notify everyone that the disk layout may have changed
        //
 
        Notification.Event   = GUID_IO_DISK_LAYOUT_CHANGE;
        Notification.Version = 1;
        Notification.Size    = (USHORT)FIELD_OFFSET(TARGET_DEVICE_CUSTOM_NOTIFICATION, CustomDataBuffer);
        Notification.FileObject = NULL;
        Notification.NameBufferOffset = -1;
 
        IoReportTargetDeviceChangeAsynchronous(fdoExtension->LowerPdo,
                                               &Notification,
                                               NULL,
                                               NULL);
    }
    return status;
}

Referenced by DiskDeviceControl().

DiskIoctlVerify()

NTSTATUS DiskIoctlVerify	(	IN PDEVICE_OBJECT	DeviceObject,
		IN OUT PIRP	Irp
	)

Definition at line 4526 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation (Irp);
    PVERIFY_INFORMATION verifyInfo = Irp->AssociatedIrp.SystemBuffer;
    PDISK_VERIFY_WORKITEM_CONTEXT Context = NULL;
    PSCSI_REQUEST_BLOCK srb;
    LARGE_INTEGER byteOffset;
    ULONG srbSize;
 
    //
    // Validate the request.
    //
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_IOCTL, "DiskIoctlVerify: DeviceObject %p Irp %p\n", DeviceObject, Irp));
 
    if (irpStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(VERIFY_INFORMATION)) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlVerify: Input buffer length mismatch.\n"));
        return STATUS_INFO_LENGTH_MISMATCH;
    }
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbSize = CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE;
    } else {
        srbSize = SCSI_REQUEST_BLOCK_SIZE;
    }
    srb = ExAllocatePoolWithTag(NonPagedPoolNx,
                                srbSize,
                                DISK_TAG_SRB);
 
    if (srb == NULL) {
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlVerify: Unable to allocate memory.\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(srb, srbSize);
 
    //
    // Add disk offset to starting sector.
    //
 
    byteOffset.QuadPart = commonExtension->StartingOffset.QuadPart +
                          verifyInfo->StartingOffset.QuadPart;
 
    //
    // Perform a bounds check on the sector range
    //
 
    if ((verifyInfo->StartingOffset.QuadPart > commonExtension->PartitionLength.QuadPart) ||
        (verifyInfo->StartingOffset.QuadPart < 0)) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlVerify: Verify request to invalid sector.\n"));
        FREE_POOL(srb)
        return STATUS_NONEXISTENT_SECTOR;
    } else {
 
        ULONGLONG bytesRemaining = commonExtension->PartitionLength.QuadPart - verifyInfo->StartingOffset.QuadPart;
 
        if ((ULONGLONG)verifyInfo->Length > bytesRemaining) {
 
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskIoctlVerify: Verify request to invalid sector.\n"));
            FREE_POOL(srb)
            return STATUS_NONEXISTENT_SECTOR;
        }
    }
 
    Context = ExAllocatePoolWithTag(NonPagedPoolNx,
                                    sizeof(DISK_VERIFY_WORKITEM_CONTEXT),
                                    DISK_TAG_WI_CONTEXT);
    if (Context) {
 
        Context->Irp = Irp;
        Context->Srb = srb;
        Context->WorkItem = IoAllocateWorkItem(DeviceObject);
 
        if (Context->WorkItem) {
 
            //
            // Queue the work item and return.
            //
 
            IoMarkIrpPending(Irp);
 
            IoQueueWorkItem(Context->WorkItem,
                            DiskIoctlVerifyThread,
                            DelayedWorkQueue,
                            Context);
 
            return STATUS_PENDING;
        }
        FREE_POOL(Context);
    }
    FREE_POOL(srb)
    return STATUS_INSUFFICIENT_RESOURCES;
}

Referenced by DiskDeviceControl().

DiskLogCacheInformation()

VOID DiskLogCacheInformation	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		IN PDISK_CACHE_INFORMATION	CacheInfo,
		IN NTSTATUS	Status
	)

Definition at line 2856 of file disk.c.

{
    PIO_ERROR_LOG_PACKET logEntry = NULL;
 
    PAGED_CODE();
 
    logEntry = IoAllocateErrorLogEntry(FdoExtension->DeviceObject, sizeof(IO_ERROR_LOG_PACKET) + (4 * sizeof(ULONG)));
 
    if (logEntry != NULL)
    {
        PDISK_DATA diskData = FdoExtension->CommonExtension.DriverData;
        BOOLEAN bIsEnabled  = TEST_FLAG(FdoExtension->DeviceFlags, DEV_WRITE_CACHE);
 
        logEntry->FinalStatus       = Status;
        logEntry->ErrorCode         = (bIsEnabled) ? IO_WRITE_CACHE_ENABLED : IO_WRITE_CACHE_DISABLED;
        logEntry->SequenceNumber    = 0;
        logEntry->MajorFunctionCode = IRP_MJ_SCSI;
        logEntry->IoControlCode     = 0;
        logEntry->RetryCount        = 0;
        logEntry->UniqueErrorValue  = 0x1;
        logEntry->DumpDataSize      = 4 * sizeof(ULONG);
 
        logEntry->DumpData[0] = diskData->ScsiAddress.PathId;
        logEntry->DumpData[1] = diskData->ScsiAddress.TargetId;
        logEntry->DumpData[2] = diskData->ScsiAddress.Lun;
        logEntry->DumpData[3] = CacheInfo->WriteCacheEnabled;
 
        //
        // Write the error log packet.
        //
 
        IoWriteErrorLogEntry(logEntry);
    }
}

Referenced by DiskIoctlSetCacheInformation().

DiskModeSelect()

NTSTATUS DiskModeSelect	(	IN PDEVICE_OBJECT	DeviceObject,
		_In_reads_bytes_(Length) PCHAR	ModeSelectBuffer,
		IN ULONG	Length,
		IN BOOLEAN	SavePage
	)

Definition at line 1802 of file disk.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = Fdo->DeviceExtension;
    PCDB cdb;
    SCSI_REQUEST_BLOCK srb = {0};
    ULONG retries = 1;
    ULONG length2;
    NTSTATUS status;
    PULONG buffer;
    PMODE_PARAMETER_BLOCK blockDescriptor;
    UCHAR srbExBuffer[CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE] = {0};
    PSTORAGE_REQUEST_BLOCK srbEx = (PSTORAGE_REQUEST_BLOCK)srbExBuffer;
    PSTOR_ADDR_BTL8 storAddrBtl8;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16;
    PSCSI_REQUEST_BLOCK srbPtr;
 
    PAGED_CODE();
 
    //
    // Check whether block length is available
    //
 
    if (fdoExtension->DiskGeometry.BytesPerSector == 0) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskModeSelect: Block length is not available. Unable to send mode select\n"));
        NT_ASSERT(fdoExtension->DiskGeometry.BytesPerSector != 0);
        return STATUS_INVALID_PARAMETER;
    }
 
 
 
    length2 = Length + sizeof(MODE_PARAMETER_HEADER) + sizeof(MODE_PARAMETER_BLOCK);
 
    //
    // Allocate buffer for mode select header, block descriptor, and mode page.
    //
 
    buffer = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                   length2,
                                   DISK_TAG_MODE_DATA);
 
    if (buffer == NULL) {
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(buffer, length2);
 
    //
    // Set length in header to size of mode page.
    //
 
    ((PMODE_PARAMETER_HEADER)buffer)->BlockDescriptorLength = sizeof(MODE_PARAMETER_BLOCK);
 
    blockDescriptor = (PMODE_PARAMETER_BLOCK)(buffer + 1);
 
    //
    // Set block length from the cached disk geometry
    //
 
    blockDescriptor->BlockLength[2] = (UCHAR) (fdoExtension->DiskGeometry.BytesPerSector >> 16);
    blockDescriptor->BlockLength[1] = (UCHAR) (fdoExtension->DiskGeometry.BytesPerSector >> 8);
    blockDescriptor->BlockLength[0] = (UCHAR) (fdoExtension->DiskGeometry.BytesPerSector);
 
    //
    // Copy mode page to buffer.
    //
 
    RtlCopyMemory(buffer + 3, ModeSelectBuffer, Length);
 
    //
    // Build the MODE SELECT CDB.
    //
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = sizeof(srbExBuffer);
        srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
        srbEx->RequestPriority = IoPriorityNormal;
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
        srbEx->NumSrbExData = 1;
 
        // Set timeout value from device extension.
        srbEx->TimeOutValue = fdoExtension->TimeOutValue * 2;
 
       //
       // Set up address fields
       //
 
       storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
       storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
       storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
       //
       // Set up SCSI SRB extended data fields
       //
 
       srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
           sizeof(STOR_ADDR_BTL8);
       if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
           srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
           srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
           srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
           srbExDataCdb16->CdbLength = 6;
 
           cdb = (PCDB)srbExDataCdb16->Cdb;
       } else {
           // Should not happen
           NT_ASSERT(FALSE);
 
           FREE_POOL(buffer);
           TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskModeSelect: Insufficient extended SRB size\n"));
           return STATUS_INTERNAL_ERROR;
       }
 
       srbPtr = (PSCSI_REQUEST_BLOCK)srbEx;
 
    } else {
 
        srb.CdbLength = 6;
        cdb = (PCDB)srb.Cdb;
 
        //
        // Set timeout value from device extension.
        //
 
        srb.TimeOutValue = fdoExtension->TimeOutValue * 2;
 
        srbPtr = &srb;
    }
 
    cdb->MODE_SELECT.OperationCode = SCSIOP_MODE_SELECT;
    cdb->MODE_SELECT.SPBit = SavePage;
    cdb->MODE_SELECT.PFBit = 1;
    cdb->MODE_SELECT.ParameterListLength = (UCHAR)(length2);
 
Retry:
 
    status = ClassSendSrbSynchronous(Fdo,
                                     srbPtr,
                                     buffer,
                                     length2,
                                     TRUE);
 
    if (status == STATUS_VERIFY_REQUIRED) {
 
        //
        // Routine ClassSendSrbSynchronous does not retry requests returned with
        // this status.
        //
 
        if (retries--) {
 
            //
            // Retry request.
            //
 
            goto Retry;
        }
 
    } else if (SRB_STATUS(srbPtr->SrbStatus) == SRB_STATUS_DATA_OVERRUN) {
        status = STATUS_SUCCESS;
    }
 
    FREE_POOL(buffer);
 
    return status;
} // end DiskModeSelect()

Referenced by DiskSetCacheInformation(), and DiskSetInfoExceptionInformation().

DiskPerformSmartCommand()

NTSTATUS DiskPerformSmartCommand	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		IN ULONG	SrbControlCode,
		IN UCHAR	Command,
		IN UCHAR	Feature,
		IN UCHAR	SectorCount,
		IN UCHAR	SectorNumber,
		IN OUT PSRB_IO_CONTROL	SrbControl,
		OUT PULONG	BufferSize
	)

Definition at line 476 of file diskwmi.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = (PCOMMON_DEVICE_EXTENSION)FdoExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    PUCHAR buffer;
    PSENDCMDINPARAMS cmdInParameters;
    NTSTATUS status;
    ULONG availableBufferSize;
    KEVENT event;
    PIRP irp;
    IO_STATUS_BLOCK ioStatus = { 0 };
    SCSI_REQUEST_BLOCK srb = {0};
    LARGE_INTEGER startingOffset;
    ULONG length;
    PIO_STACK_LOCATION irpStack;
    UCHAR srbExBuffer[CLASS_SRBEX_NO_SRBEX_DATA_BUFFER_SIZE] = {0};
    PSTORAGE_REQUEST_BLOCK srbEx = (PSTORAGE_REQUEST_BLOCK)srbExBuffer;
    PSTOR_ADDR_BTL8 storAddrBtl8;
 
    PAGED_CODE();
 
    //
    // Point to the 'buffer' portion of the SRB_CONTROL and compute how
    // much room we have left in the srb control. Abort if the buffer
    // isn't at least the size of SRB_IO_CONTROL.
    //
 
    buffer = (PUCHAR)SrbControl + sizeof(SRB_IO_CONTROL);
 
    cmdInParameters = (PSENDCMDINPARAMS)buffer;
 
    if (*BufferSize >= sizeof(SRB_IO_CONTROL)) {
        availableBufferSize = *BufferSize - sizeof(SRB_IO_CONTROL);
    } else {
        return STATUS_BUFFER_TOO_SMALL;
    }
 
#if DBG
 
    //
    // Ensure control codes and buffer lengths passed are correct
    //
    {
        ULONG controlCode = 0;
        ULONG lengthNeeded = sizeof(SENDCMDINPARAMS);
 
        if (Command == SMART_CMD)
        {
            switch (Feature)
            {
                case ENABLE_SMART:
                {
                    controlCode = IOCTL_SCSI_MINIPORT_ENABLE_SMART;
                    break;
                }
 
                case DISABLE_SMART:
                {
                    controlCode = IOCTL_SCSI_MINIPORT_DISABLE_SMART;
                    break;
                }
 
                case RETURN_SMART_STATUS:
                {
                    controlCode = IOCTL_SCSI_MINIPORT_RETURN_STATUS;
                    lengthNeeded = max( lengthNeeded, sizeof(SENDCMDOUTPARAMS) - 1 + sizeof(IDEREGS) );
                    break;
                }
 
                case ENABLE_DISABLE_AUTOSAVE:
                {
                    controlCode = IOCTL_SCSI_MINIPORT_ENABLE_DISABLE_AUTOSAVE;
                    break;
                }
 
                case SAVE_ATTRIBUTE_VALUES:
                {
                    controlCode = IOCTL_SCSI_MINIPORT_SAVE_ATTRIBUTE_VALUES;
                    break;
                }
 
 
                case EXECUTE_OFFLINE_DIAGS:
                {
                    controlCode = IOCTL_SCSI_MINIPORT_EXECUTE_OFFLINE_DIAGS;
                    break;
                }
 
                case READ_ATTRIBUTES:
                {
                    controlCode  = IOCTL_SCSI_MINIPORT_READ_SMART_ATTRIBS;
                    lengthNeeded = max( lengthNeeded, sizeof(SENDCMDOUTPARAMS) - 1 + READ_ATTRIBUTE_BUFFER_SIZE );
                    break;
                }
 
                case READ_THRESHOLDS:
                {
                    controlCode  = IOCTL_SCSI_MINIPORT_READ_SMART_THRESHOLDS;
                    lengthNeeded = max( lengthNeeded, sizeof(SENDCMDOUTPARAMS) - 1 + READ_THRESHOLD_BUFFER_SIZE );
                    break;
                }
 
                case SMART_READ_LOG:
                {
                    controlCode  = IOCTL_SCSI_MINIPORT_READ_SMART_LOG;
                    lengthNeeded = max( lengthNeeded, sizeof(SENDCMDOUTPARAMS) - 1 + (SectorCount * SMART_LOG_SECTOR_SIZE) );
                    break;
                }
 
                case SMART_WRITE_LOG:
                {
                    controlCode  = IOCTL_SCSI_MINIPORT_WRITE_SMART_LOG;
                    lengthNeeded = lengthNeeded - 1 + (SectorCount * SMART_LOG_SECTOR_SIZE);
                    break;
                }
 
            }
 
        } else if (Command == ID_CMD) {
 
            controlCode  = IOCTL_SCSI_MINIPORT_IDENTIFY;
            lengthNeeded = max( lengthNeeded, sizeof(SENDCMDOUTPARAMS) - 1 + IDENTIFY_BUFFER_SIZE );
 
        } else {
 
            NT_ASSERT(FALSE);
        }
 
        NT_ASSERT(controlCode == SrbControlCode);
        NT_ASSERT(availableBufferSize >= lengthNeeded);
    }
 
#endif
 
    //
    // Build SrbControl and input to SMART command
    //
    SrbControl->HeaderLength = sizeof(SRB_IO_CONTROL);
    RtlMoveMemory (SrbControl->Signature, "SCSIDISK", 8);
    SrbControl->Timeout      = FdoExtension->TimeOutValue;
    SrbControl->Length       = availableBufferSize;
    SrbControl->ControlCode  = SrbControlCode;
 
    cmdInParameters->cBufferSize  = sizeof(SENDCMDINPARAMS);
    cmdInParameters->bDriveNumber = diskData->ScsiAddress.TargetId;
    cmdInParameters->irDriveRegs.bFeaturesReg     = Feature;
    cmdInParameters->irDriveRegs.bSectorCountReg  = SectorCount;
    cmdInParameters->irDriveRegs.bSectorNumberReg = SectorNumber;
    cmdInParameters->irDriveRegs.bCylLowReg       = SMART_CYL_LOW;
    cmdInParameters->irDriveRegs.bCylHighReg      = SMART_CYL_HI;
    cmdInParameters->irDriveRegs.bCommandReg      = Command;
 
    //
    // Create and send irp
    //
    KeInitializeEvent(&event, NotificationEvent, FALSE);
 
    startingOffset.QuadPart = (LONGLONG) 1;
 
    length = SrbControl->HeaderLength + SrbControl->Length;
 
    irp = IoBuildSynchronousFsdRequest(
                IRP_MJ_SCSI,
                commonExtension->LowerDeviceObject,
                SrbControl,
                length,
                &startingOffset,
                &event,
                &ioStatus);
 
    if (irp == NULL) {
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    irpStack = IoGetNextIrpStackLocation(irp);
 
    //
    // Set major and minor codes.
    //
 
    irpStack->MajorFunction = IRP_MJ_SCSI;
    irpStack->MinorFunction = 1;
 
    //
    // Fill in SRB fields.
    //
 
    if (FdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        irpStack->Parameters.Others.Argument1 = srbEx;
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = sizeof(srbExBuffer);
        srbEx->SrbFunction = SRB_FUNCTION_IO_CONTROL;
        srbEx->RequestPriority = IoGetIoPriorityHint(irp);
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
 
        srbEx->SrbFlags = FdoExtension->SrbFlags;
        SET_FLAG(srbEx->SrbFlags, SRB_FLAGS_DATA_IN);
        SET_FLAG(srbEx->SrbFlags, SRB_FLAGS_NO_QUEUE_FREEZE);
        SET_FLAG(srbEx->SrbFlags, SRB_FLAGS_NO_KEEP_AWAKE);
 
        srbEx->RequestAttribute = SRB_SIMPLE_TAG_REQUEST;
        srbEx->RequestTag = SP_UNTAGGED;
 
        srbEx->OriginalRequest = irp;
 
        //
        // Set timeout to requested value.
        //
 
        srbEx->TimeOutValue = SrbControl->Timeout;
 
        //
        // Set the data buffer.
        //
 
        srbEx->DataBuffer = SrbControl;
        srbEx->DataTransferLength = length;
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
        storAddrBtl8->Path = diskData->ScsiAddress.PathId;
        storAddrBtl8->Target = diskData->ScsiAddress.TargetId;
        storAddrBtl8->Lun = srb.Lun = diskData->ScsiAddress.Lun;
 
    } else {
        irpStack->Parameters.Others.Argument1 = &srb;
 
        srb.PathId = diskData->ScsiAddress.PathId;
        srb.TargetId = diskData->ScsiAddress.TargetId;
        srb.Lun = diskData->ScsiAddress.Lun;
 
        srb.Function = SRB_FUNCTION_IO_CONTROL;
        srb.Length = sizeof(SCSI_REQUEST_BLOCK);
 
        srb.SrbFlags = FdoExtension->SrbFlags;
        SET_FLAG(srb.SrbFlags, SRB_FLAGS_DATA_IN);
        SET_FLAG(srb.SrbFlags, SRB_FLAGS_NO_QUEUE_FREEZE);
        SET_FLAG(srb.SrbFlags, SRB_FLAGS_NO_KEEP_AWAKE);
 
        srb.QueueAction = SRB_SIMPLE_TAG_REQUEST;
        srb.QueueTag = SP_UNTAGGED;
 
        srb.OriginalRequest = irp;
 
        //
        // Set timeout to requested value.
        //
 
        srb.TimeOutValue = SrbControl->Timeout;
 
        //
        // Set the data buffer.
        //
 
        srb.DataBuffer = SrbControl;
        srb.DataTransferLength = length;
    }
 
    //
    // Flush the data buffer for output. This will insure that the data is
    // written back to memory.  Since the data-in flag is the the port driver
    // will flush the data again for input which will ensure the data is not
    // in the cache.
    //
 
    KeFlushIoBuffers(irp->MdlAddress, FALSE, TRUE);
 
    //
    // Call port driver to handle this request.
    //
 
    status = IoCallDriver(commonExtension->LowerDeviceObject, irp);
 
    if (status == STATUS_PENDING) {
        KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL);
        status = ioStatus.Status;
    }
 
    return status;
}

Referenced by DiskDisableSmart(), DiskDisableSmartAttributeAutosave(), DiskEnableSmart(), DiskEnableSmartAttributeAutosave(), DiskExecuteSmartDiagnostics(), DiskReadSmartLog(), and DiskWriteSmartLog().

DiskReadFailurePredictData()

NTSTATUS DiskReadFailurePredictData	(	PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		PSTORAGE_FAILURE_PREDICT_DATA	DiskSmartData
	)

Definition at line 1327 of file diskwmi.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = (PCOMMON_DEVICE_EXTENSION)FdoExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    NTSTATUS status;
 
    PAGED_CODE();
 
    switch(diskData->FailurePredictionCapability)
    {
        case FailurePredictionSmart:
        {
            PUCHAR outBuffer;
            ULONG outBufferSize;
            PSENDCMDOUTPARAMS cmdOutParameters;
 
            outBufferSize = sizeof(SRB_IO_CONTROL) + max( sizeof(SENDCMDINPARAMS), sizeof(SENDCMDOUTPARAMS) - 1 + READ_ATTRIBUTE_BUFFER_SIZE );
 
            outBuffer = ExAllocatePoolWithTag(NonPagedPoolNx,
                                              outBufferSize,
                                              DISK_TAG_SMART);
 
            if (outBuffer != NULL)
            {
                status = DiskReadSmartData(FdoExtension,
                                           (PSRB_IO_CONTROL)outBuffer,
                                           &outBufferSize);
 
                if (NT_SUCCESS(status))
                {
                    cmdOutParameters = (PSENDCMDOUTPARAMS)(outBuffer +
                                                    sizeof(SRB_IO_CONTROL));
 
                    DiskSmartData->Length = READ_ATTRIBUTE_BUFFER_SIZE;
                    RtlCopyMemory(DiskSmartData->VendorSpecific,
                                  cmdOutParameters->bBuffer,
                                  min(READ_ATTRIBUTE_BUFFER_SIZE, sizeof(DiskSmartData->VendorSpecific)));
                }
                FREE_POOL(outBuffer);
            } else {
                status = STATUS_INSUFFICIENT_RESOURCES;
            }
 
            break;
        }
 
        case FailurePredictionSense:
        {
            DiskSmartData->Length = sizeof(ULONG);
            *((PULONG)DiskSmartData->VendorSpecific) = FdoExtension->FailureReason;
 
            status = STATUS_SUCCESS;
            break;
        }
 
        case FailurePredictionIoctl:
        case FailurePredictionNone:
        default:
        {
            status = STATUS_INVALID_DEVICE_REQUEST;
            break;
        }
    }
 
    return status;
}

Referenced by DiskIoctlPredictFailure().

DiskReadFailurePredictStatus()

NTSTATUS DiskReadFailurePredictStatus	(	PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		PSTORAGE_FAILURE_PREDICT_STATUS	DiskSmartStatus
	)

Definition at line 1251 of file diskwmi.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = (PCOMMON_DEVICE_EXTENSION)FdoExtension;
    PDISK_DATA diskData = (PDISK_DATA)(commonExtension->DriverData);
    NTSTATUS status;
 
    PAGED_CODE();
 
    DiskSmartStatus->PredictFailure = FALSE;
 
    switch(diskData->FailurePredictionCapability)
    {
        case FailurePredictionSmart:
        {
            UCHAR outBuffer[sizeof(SRB_IO_CONTROL) + max( sizeof(SENDCMDINPARAMS), sizeof(SENDCMDOUTPARAMS) - 1 + sizeof(IDEREGS) )] = {0};
            ULONG outBufferSize = sizeof(outBuffer);
            PSENDCMDOUTPARAMS cmdOutParameters;
 
            status = DiskReadSmartStatus(FdoExtension,
                                     (PSRB_IO_CONTROL)outBuffer,
                                     &outBufferSize);
 
            if (NT_SUCCESS(status))
            {
                cmdOutParameters = (PSENDCMDOUTPARAMS)(outBuffer +
                                               sizeof(SRB_IO_CONTROL));
 
                DiskSmartStatus->Reason = 0; // Unknown;
                DiskSmartStatus->PredictFailure = ((cmdOutParameters->bBuffer[3] == 0xf4) &&
                                                   (cmdOutParameters->bBuffer[4] == 0x2c));
            }
            break;
        }
 
        case FailurePredictionSense:
        {
            DiskSmartStatus->Reason = FdoExtension->FailureReason;
            DiskSmartStatus->PredictFailure = FdoExtension->FailurePredicted;
            status = STATUS_SUCCESS;
            break;
        }
 
        case FailurePredictionIoctl:
        case FailurePredictionNone:
        default:
        {
            status = STATUS_INVALID_DEVICE_REQUEST;
            break;
        }
    }
 
    return status;
}

Referenced by DiskDetectFailurePrediction(), and DiskIoctlPredictFailure().

DiskReadWriteVerification()

NTSTATUS NTAPI DiskReadWriteVerification	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 546 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(Irp);
    ULONG residualBytes;
    ULONG residualOffset;
    NTSTATUS status = STATUS_SUCCESS;
 
    //
    // Make sure that the request is within the bounds of the partition,
    // the number of bytes to transfer and the byte offset are a
    // multiple of the sector size.
    //
 
    residualBytes = irpSp->Parameters.Read.Length & (commonExtension->PartitionZeroExtension->DiskGeometry.BytesPerSector - 1);
    residualOffset = irpSp->Parameters.Read.ByteOffset.LowPart & (commonExtension->PartitionZeroExtension->DiskGeometry.BytesPerSector - 1);
 
    if ((irpSp->Parameters.Read.ByteOffset.QuadPart > commonExtension->PartitionLength.QuadPart) ||
        (irpSp->Parameters.Read.ByteOffset.QuadPart < 0) ||
        (residualBytes != 0) ||
        (residualOffset != 0))
    {
        NT_ASSERT(residualOffset == 0);
        status = STATUS_INVALID_PARAMETER;
    }
    else
    {
        ULONGLONG bytesRemaining = commonExtension->PartitionLength.QuadPart - irpSp->Parameters.Read.ByteOffset.QuadPart;
 
        if ((ULONGLONG)irpSp->Parameters.Read.Length > bytesRemaining)
        {
            status = STATUS_INVALID_PARAMETER;
        }
    }
 
    if (!NT_SUCCESS(status))
    {
        //
        // This error may be caused by the fact that the drive is not ready.
        //
 
        status = ((PDISK_DATA) commonExtension->DriverData)->ReadyStatus;
 
        if (!NT_SUCCESS(status)) {
 
            //
            // Flag this as a user error so that a popup is generated.
            //
 
            TracePrint((TRACE_LEVEL_WARNING, TRACE_FLAG_RW, "DiskReadWriteVerification: ReadyStatus is %lx\n", status));
 
            if (IoIsErrorUserInduced(status) && Irp->Tail.Overlay.Thread != NULL) {
                IoSetHardErrorOrVerifyDevice(Irp, DeviceObject);
            }
 
            //
            // status will keep the current error
            //
 
        } else if ((residualBytes == 0) && (residualOffset == 0)) {
 
            //
            // This failed because we think the physical disk is too small.
            // Send it down to the drive and let the hardware decide for
            // itself.
            //
 
            status = STATUS_SUCCESS;
 
        } else {
 
            //
            // Note fastfat depends on this parameter to determine when to
            // remount due to a sector size change.
            //
 
            status = STATUS_INVALID_PARAMETER;
        }
    }
 
    Irp->IoStatus.Status = status;
 
    return status;
 
} // end DiskReadWrite()

Referenced by DriverEntry().

DiskRemoveDevice()

NTSTATUS NTAPI DiskRemoveDevice	(	IN PDEVICE_OBJECT	DeviceObject,
		IN UCHAR	Type
	)

Definition at line 843 of file pnp.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
 
    PAGED_CODE();
 
    //
    // Handle query and cancel
    //
 
    if((Type == IRP_MN_QUERY_REMOVE_DEVICE) ||
       (Type == IRP_MN_CANCEL_REMOVE_DEVICE)) {
        return STATUS_SUCCESS;
    }
 
    //
    // Delete our object directory.
    //
 
    if(fdoExtension->DeviceDirectory != NULL) {
        ZwMakeTemporaryObject(fdoExtension->DeviceDirectory);
        ZwClose(fdoExtension->DeviceDirectory);
        fdoExtension->DeviceDirectory = NULL;
    }
 
    if(Type == IRP_MN_REMOVE_DEVICE) {
 
        FREE_POOL(fdoExtension->SenseData);
 
        IoGetConfigurationInformation()->DiskCount--;
 
    }
 
    DiskDeleteSymbolicLinks(DeviceObject);
 
    if (Type == IRP_MN_REMOVE_DEVICE)
    {
        ClassDeleteSrbLookasideList(commonExtension);
    }
 
    return STATUS_SUCCESS;
}

Referenced by DriverEntry().

DiskSetCacheInformation()

NTSTATUS NTAPI DiskSetCacheInformation	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		IN PDISK_CACHE_INFORMATION	CacheInfo
	)

Definition at line 3167 of file disk.c.

{
    PMODE_PARAMETER_HEADER modeData;
    ULONG length;
    PMODE_CACHING_PAGE pageData;
    ULONG i;
    NTSTATUS status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    modeData = ExAllocatePoolWithTag(NonPagedPoolNxCacheAligned,
                                     MODE_DATA_SIZE,
                                     DISK_TAG_DISABLE_CACHE);
 
    if (modeData == NULL) {
 
        TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskSetCacheInformation: Unable to allocate mode "
                       "data buffer\n"));
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    RtlZeroMemory(modeData, MODE_DATA_SIZE);
 
    length = ClassModeSense(FdoExtension->DeviceObject,
                            (PCHAR) modeData,
                            MODE_DATA_SIZE,
                            MODE_PAGE_CACHING);
 
    if (length < sizeof(MODE_PARAMETER_HEADER)) {
 
        //
        // Retry the request in case of a check condition.
        //
 
        length = ClassModeSense(FdoExtension->DeviceObject,
                                (PCHAR) modeData,
                                MODE_DATA_SIZE,
                                MODE_PAGE_CACHING);
 
        if (length < sizeof(MODE_PARAMETER_HEADER)) {
 
            TracePrint((TRACE_LEVEL_ERROR, TRACE_FLAG_IOCTL, "DiskSetCacheInformation: Mode Sense failed\n"));
 
            FREE_POOL(modeData);
            return STATUS_IO_DEVICE_ERROR;
        }
    }
 
    //
    // If the length is greater than length indicated by the mode data reset
    // the data to the mode data.
    //
 
    if (length > (ULONG) (modeData->ModeDataLength + 1)) {
        length = modeData->ModeDataLength + 1;
    }
 
    //
    // Check to see if the write cache is enabled.
    //
 
    pageData = ClassFindModePage((PCHAR) modeData,
                                 length,
                                 MODE_PAGE_CACHING,
                                 TRUE);
 
    //
    // Check if valid caching page exists.
    //
 
    if (pageData == NULL) {
        FREE_POOL(modeData);
        return STATUS_NOT_SUPPORTED;
    }
 
    //
    // Don't touch any of the normal parameters - not all drives actually
    // use the correct size of caching mode page.  Just change the things
    // which the user could have modified.
    //
 
    pageData->PageSavable = FALSE;
 
    pageData->ReadDisableCache = !(CacheInfo->ReadCacheEnabled);
    pageData->MultiplicationFactor = CacheInfo->PrefetchScalar;
    pageData->WriteCacheEnable = CacheInfo->WriteCacheEnabled;
 
    pageData->WriteRetensionPriority = (UCHAR)
        TRANSLATE_RETENTION_PRIORITY(CacheInfo->WriteRetentionPriority);
    pageData->ReadRetensionPriority = (UCHAR)
        TRANSLATE_RETENTION_PRIORITY(CacheInfo->ReadRetentionPriority);
 
    pageData->DisablePrefetchTransfer[0] =
        (UCHAR) (CacheInfo->DisablePrefetchTransferLength >> 8);
    pageData->DisablePrefetchTransfer[1] =
        (UCHAR) (CacheInfo->DisablePrefetchTransferLength & 0x00ff);
 
    pageData->MinimumPrefetch[0] =
        (UCHAR) (CacheInfo->ScalarPrefetch.Minimum >> 8);
    pageData->MinimumPrefetch[1] =
        (UCHAR) (CacheInfo->ScalarPrefetch.Minimum & 0x00ff);
 
    pageData->MaximumPrefetch[0] =
        (UCHAR) (CacheInfo->ScalarPrefetch.Maximum >> 8);
    pageData->MaximumPrefetch[1] =
        (UCHAR) (CacheInfo->ScalarPrefetch.Maximum & 0x00ff);
 
    if(pageData->MultiplicationFactor) {
 
        pageData->MaximumPrefetchCeiling[0] =
            (UCHAR) (CacheInfo->ScalarPrefetch.MaximumBlocks >> 8);
        pageData->MaximumPrefetchCeiling[1] =
            (UCHAR) (CacheInfo->ScalarPrefetch.MaximumBlocks & 0x00ff);
    }
 
    //
    // We will attempt (twice) to issue the mode select with the page.
    //
 
    for (i = 0; i < 2; i++) {
 
        status = DiskModeSelect(FdoExtension->DeviceObject,
                                (PCHAR) pageData,
                                (pageData->PageLength + 2),
                                CacheInfo->ParametersSavable);
 
        if (NT_SUCCESS(status)) {
 
            if (CacheInfo->WriteCacheEnabled)
            {
                SET_FLAG(FdoExtension->DeviceFlags, DEV_WRITE_CACHE);
            }
            else
            {
                CLEAR_FLAG(FdoExtension->DeviceFlags, DEV_WRITE_CACHE);
            }
            ADJUST_FUA_FLAG(FdoExtension);
 
            break;
        }
    }
 
    if (NT_SUCCESS(status))
    {
    } else {
 
        //
        // We were unable to modify the disk write cache setting
        //
 
        SET_FLAG(FdoExtension->ScanForSpecialFlags, CLASS_SPECIAL_MODIFY_CACHE_UNSUCCESSFUL);
    }
 
    FREE_POOL(modeData);
    return status;
}

Referenced by DiskIoctlSetCacheInformation(), and DiskStartFdo().

DiskSetInfoExceptionInformation()

NTSTATUS DiskSetInfoExceptionInformation	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		IN PMODE_INFO_EXCEPTIONS	PageData
	)

Definition at line 2985 of file disk.c.

{
    ULONG i;
    NTSTATUS status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    //
    // We will attempt (twice) to issue the mode select with the page.
    // Make the setting persistant so that we don't have to turn it back
    // on after a bus reset.
    //
 
    for (i = 0; i < 2; i++)
    {
        status = DiskModeSelect(FdoExtension->DeviceObject,
                                (PCHAR) PageData,
                                sizeof(MODE_INFO_EXCEPTIONS),
                                TRUE);
    }
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "DiskSetInfoExceptionInformation: %s for device %p\n",
                        NT_SUCCESS(status) ? "succeeded" : "failed",
                        FdoExtension->DeviceObject));
 
    return status;
}

Referenced by DiskFdoExecuteWmiMethod(), and DiskFdoSetWmiDataBlock().

DiskSetSpecialHacks()

VOID NTAPI DiskSetSpecialHacks	(	IN PFUNCTIONAL_DEVICE_EXTENSION	FdoExtension,
		IN ULONG_PTR	Data
	)

Definition at line 2602 of file disk.c.

{
    PDEVICE_OBJECT fdo = FdoExtension->DeviceObject;
 
    PAGED_CODE();
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_INIT, "Disk SetSpecialHacks, Setting Hacks %p\n", (void*) Data));
 
    //
    // Found a listed controller.  Determine what must be done.
    //
 
    if (TEST_FLAG(Data, HackDisableTaggedQueuing)) {
 
        //
        // Disable tagged queuing.
        //
 
        CLEAR_FLAG(FdoExtension->SrbFlags, SRB_FLAGS_QUEUE_ACTION_ENABLE);
    }
 
    if (TEST_FLAG(Data, HackDisableSynchronousTransfers)) {
 
        //
        // Disable synchronous data transfers.
        //
 
        SET_FLAG(FdoExtension->SrbFlags, SRB_FLAGS_DISABLE_SYNCH_TRANSFER);
 
    }
 
    if (TEST_FLAG(Data, HackDisableSpinDown)) {
 
        //
        // Disable spinning down of drives.
        //
 
        SET_FLAG(FdoExtension->ScanForSpecialFlags,
                 CLASS_SPECIAL_DISABLE_SPIN_DOWN);
 
    }
 
    if (TEST_FLAG(Data, HackDisableWriteCache)) {
 
        //
        // Disable the drive's write cache
        //
 
        SET_FLAG(FdoExtension->ScanForSpecialFlags,
                 CLASS_SPECIAL_DISABLE_WRITE_CACHE);
 
    }
 
    if (TEST_FLAG(Data, HackCauseNotReportableHack)) {
 
        SET_FLAG(FdoExtension->ScanForSpecialFlags,
                 CLASS_SPECIAL_CAUSE_NOT_REPORTABLE_HACK);
    }
 
    if (TEST_FLAG(fdo->Characteristics, FILE_REMOVABLE_MEDIA) &&
        TEST_FLAG(Data, HackRequiresStartUnitCommand)
        ) {
 
        //
        // this is a list of vendors who require the START_UNIT command
        //
 
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_INIT, "DiskScanForSpecial (%p) => This unit requires "
                    " START_UNITS\n", fdo));
        SET_FLAG(FdoExtension->DeviceFlags, DEV_SAFE_START_UNIT);
 
    }
 
    return;
}

Referenced by DiskInitFdo().

DiskShutdownFlush()

NTSTATUS NTAPI DiskShutdownFlush	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 1122 of file disk.c.

{
    PCOMMON_DEVICE_EXTENSION commonExtension = DeviceObject->DeviceExtension;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = commonExtension->PartitionZeroExtension;
    PDISK_DATA diskData = (PDISK_DATA) commonExtension->DriverData;
    PIO_STACK_LOCATION irpStack;
    NTSTATUS status = STATUS_SUCCESS;
    ULONG srbSize;
    PSCSI_REQUEST_BLOCK srb;
    PSTORAGE_REQUEST_BLOCK srbEx = NULL;
    PSTOR_ADDR_BTL8 storAddrBtl8 = NULL;
    PSRBEX_DATA_SCSI_CDB16 srbExDataCdb16 = NULL;
    PCDB cdb;
    KIRQL irql;
 
    //
    // Flush requests are combined and need to be handled in a special manner
    //
 
    irpStack = IoGetCurrentIrpStackLocation(Irp);
 
    if (irpStack->MajorFunction == IRP_MJ_FLUSH_BUFFERS) {
 
        if (TEST_FLAG(fdoExtension->DeviceFlags, DEV_POWER_PROTECTED)) {
 
            //
            // We've been assured that both the disk
            // and adapter caches are battery-backed
            //
 
            Irp->IoStatus.Status = STATUS_SUCCESS;
            ClassReleaseRemoveLock(DeviceObject, Irp);
            ClassCompleteRequest(DeviceObject, Irp, IO_NO_INCREMENT);
            return STATUS_SUCCESS;
        }
 
        KeAcquireSpinLock(&diskData->FlushContext.Spinlock, &irql);
 
        TracePrint((TRACE_LEVEL_VERBOSE, TRACE_FLAG_SCSI, "DiskShutdownFlush: IRP %p flags = 0x%x\n", Irp, irpStack->Flags));
 
        //
        // This request will most likely be completed asynchronously
        //
        IoMarkIrpPending(Irp);
 
        //
        // Look to see if a flush is in progress
        //
 
        if (diskData->FlushContext.CurrIrp != NULL) {
 
            //
            // There is an outstanding flush. Queue this
            // request to the group that is next in line
            //
 
            if (diskData->FlushContext.NextIrp != NULL) {
 
                #if DBG
                    diskData->FlushContext.DbgTagCount++;
                #endif
 
                InsertTailList(&diskData->FlushContext.NextList, &Irp->Tail.Overlay.ListEntry);
 
                KeReleaseSpinLock(&diskData->FlushContext.Spinlock, irql);
 
                //
                // This request will be completed by its representative
                //
 
            } else {
 
                #if DBG
                    if (diskData->FlushContext.DbgTagCount < 64) {
 
                        diskData->FlushContext.DbgRefCount[diskData->FlushContext.DbgTagCount]++;
                    }
 
                    diskData->FlushContext.DbgSavCount += diskData->FlushContext.DbgTagCount;
                    diskData->FlushContext.DbgTagCount  = 0;
                #endif
 
                diskData->FlushContext.NextIrp = Irp;
                NT_ASSERT(IsListEmpty(&diskData->FlushContext.NextList));
 
 
                KeReleaseSpinLock(&diskData->FlushContext.Spinlock, irql);
 
 
                    TracePrint((TRACE_LEVEL_VERBOSE, TRACE_FLAG_SCSI, "DiskShutdownFlush: waiting for event\n"));
 
                    //
                    // Wait for the outstanding flush to complete
                    //
                    KeWaitForSingleObject(&diskData->FlushContext.Event, Executive, KernelMode, FALSE, NULL);
 
                    TracePrint((TRACE_LEVEL_VERBOSE, TRACE_FLAG_SCSI, "DiskShutdownFlush: event signal\n"));
 
                    //
                    // Make this group the outstanding one and free up the next slot
                    //
 
                    KeAcquireSpinLock(&diskData->FlushContext.Spinlock, &irql);
 
                    NT_ASSERT(IsListEmpty(&diskData->FlushContext.CurrList));
 
                    while (!IsListEmpty(&diskData->FlushContext.NextList)) {
 
                        PLIST_ENTRY listEntry = RemoveHeadList(&diskData->FlushContext.NextList);
                        InsertTailList(&diskData->FlushContext.CurrList, listEntry);
                    }
 
#ifndef __REACTOS__
                    // ReactOS hits this assert, because CurrIrp can already be freed at this point
                    // and it's possible that NextIrp has the same pointer value
                    NT_ASSERT(diskData->FlushContext.CurrIrp != diskData->FlushContext.NextIrp);
#endif
                    diskData->FlushContext.CurrIrp = diskData->FlushContext.NextIrp;
                    diskData->FlushContext.NextIrp = NULL;
 
                    KeReleaseSpinLock(&diskData->FlushContext.Spinlock, irql);
 
                    //
                    // Send this request down to the device
                    //
                    DiskFlushDispatch(DeviceObject, &diskData->FlushContext);
            }
 
        } else {
 
            diskData->FlushContext.CurrIrp = Irp;
            NT_ASSERT(IsListEmpty(&diskData->FlushContext.CurrList));
 
            NT_ASSERT(diskData->FlushContext.NextIrp == NULL);
            NT_ASSERT(IsListEmpty(&diskData->FlushContext.NextList));
 
 
            KeReleaseSpinLock(&diskData->FlushContext.Spinlock, irql);
 
                DiskFlushDispatch(DeviceObject, &diskData->FlushContext);
        }
 
    } else {
 
        //
        // Allocate SCSI request block.
        //
 
        if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
            srbSize = CLASS_SRBEX_SCSI_CDB16_BUFFER_SIZE;
        } else {
            srbSize = sizeof(SCSI_REQUEST_BLOCK);
        }
 
        srb = ExAllocatePoolWithTag(NonPagedPoolNx,
                                    srbSize,
                                    DISK_TAG_SRB);
        if (srb == NULL) {
 
            //
            // Set the status and complete the request.
            //
 
            Irp->IoStatus.Status = STATUS_INSUFFICIENT_RESOURCES;
            ClassReleaseRemoveLock(DeviceObject, Irp);
            ClassCompleteRequest(DeviceObject, Irp, IO_NO_INCREMENT);
            return(STATUS_INSUFFICIENT_RESOURCES);
        }
 
        RtlZeroMemory(srb, srbSize);
        if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
 
            srbEx = (PSTORAGE_REQUEST_BLOCK)srb;
 
            //
            // Set up STORAGE_REQUEST_BLOCK fields
            //
 
            srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
            srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
            srbEx->Signature = SRB_SIGNATURE;
            srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
            srbEx->SrbLength = srbSize;
            srbEx->SrbFunction = SRB_FUNCTION_EXECUTE_SCSI;
            srbEx->RequestPriority = IoGetIoPriorityHint(Irp);
            srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
            srbEx->NumSrbExData = 1;
 
            // Set timeout value and mark the request as not being a tagged request.
            srbEx->TimeOutValue = fdoExtension->TimeOutValue * 4;
            srbEx->RequestTag = SP_UNTAGGED;
            srbEx->RequestAttribute = SRB_SIMPLE_TAG_REQUEST;
            srbEx->SrbFlags = fdoExtension->SrbFlags;
 
            //
            // Set up address fields
            //
 
            storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
            storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
            storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
            //
            // Set up SCSI SRB extended data fields
            //
 
            srbEx->SrbExDataOffset[0] = sizeof(STORAGE_REQUEST_BLOCK) +
                sizeof(STOR_ADDR_BTL8);
            if ((srbEx->SrbExDataOffset[0] + sizeof(SRBEX_DATA_SCSI_CDB16)) <= srbEx->SrbLength) {
                srbExDataCdb16 = (PSRBEX_DATA_SCSI_CDB16)((PUCHAR)srbEx + srbEx->SrbExDataOffset[0]);
                srbExDataCdb16->Type = SrbExDataTypeScsiCdb16;
                srbExDataCdb16->Length = SRBEX_DATA_SCSI_CDB16_LENGTH;
 
                cdb = (PCDB)srbExDataCdb16->Cdb;
            } else {
                // Should not happen
                NT_ASSERT(FALSE);
 
                //
                // Set the status and complete the request.
                //
 
                Irp->IoStatus.Status = STATUS_INTERNAL_ERROR;
                ClassReleaseRemoveLock(DeviceObject, Irp);
                ClassCompleteRequest(DeviceObject, Irp, IO_NO_INCREMENT);
                return(STATUS_INTERNAL_ERROR);
            }
 
        } else {
 
            //
            // Write length to SRB.
            //
 
            srb->Length = SCSI_REQUEST_BLOCK_SIZE;
 
            //
            // Set timeout value and mark the request as not being a tagged request.
            //
 
            srb->TimeOutValue = fdoExtension->TimeOutValue * 4;
            srb->QueueTag = SP_UNTAGGED;
            srb->QueueAction = SRB_SIMPLE_TAG_REQUEST;
            srb->SrbFlags = fdoExtension->SrbFlags;
            srb->Function = SRB_FUNCTION_EXECUTE_SCSI;
 
            cdb = (PCDB)srb->Cdb;
        }
 
        //
        // If the write cache is enabled then send a synchronize cache request.
        //
 
        if (TEST_FLAG(fdoExtension->DeviceFlags, DEV_WRITE_CACHE)) {
 
            if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
                srbExDataCdb16->CdbLength = 10;
            } else {
                srb->CdbLength = 10;
            }
 
            cdb->CDB10.OperationCode = SCSIOP_SYNCHRONIZE_CACHE;
 
            status = ClassSendSrbSynchronous(DeviceObject,
                                             srb,
                                             NULL,
                                             0,
                                             TRUE);
 
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_GENERAL, "DiskShutdownFlush: Synchonize cache sent. Status = %lx\n", status));
        }
 
        //
        // Unlock the device if it contains removable media
        //
 
        if (TEST_FLAG(DeviceObject->Characteristics, FILE_REMOVABLE_MEDIA))
        {
 
            if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
 
                //
                // Reinitialize status fields to 0 in case there was a previous request
                //
 
                srbEx->SrbStatus = 0;
                srbExDataCdb16->ScsiStatus = 0;
 
                srbExDataCdb16->CdbLength = 6;
 
                //
                // Set timeout value
                //
 
                srbEx->TimeOutValue = fdoExtension->TimeOutValue;
 
            } else {
 
                //
                // Reinitialize status fields to 0 in case there was a previous request
                //
 
                srb->SrbStatus = 0;
                srb->ScsiStatus = 0;
 
                srb->CdbLength = 6;
 
                //
                // Set timeout value.
                //
 
                srb->TimeOutValue = fdoExtension->TimeOutValue;
            }
 
            cdb->MEDIA_REMOVAL.OperationCode = SCSIOP_MEDIUM_REMOVAL;
            cdb->MEDIA_REMOVAL.Prevent = FALSE;
 
            status = ClassSendSrbSynchronous(DeviceObject,
                                             srb,
                                             NULL,
                                             0,
                                             TRUE);
 
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_GENERAL, "DiskShutdownFlush: Unlock device request sent. Status = %lx\n", status));
        }
 
        //
        // Set up a SHUTDOWN SRB
        //
 
        if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
            srbEx->NumSrbExData = 0;
            srbEx->SrbExDataOffset[0] = 0;
            srbEx->SrbFunction = SRB_FUNCTION_SHUTDOWN;
            srbEx->OriginalRequest = Irp;
            srbEx->SrbLength = CLASS_SRBEX_NO_SRBEX_DATA_BUFFER_SIZE;
            srbEx->SrbStatus = 0;
        } else {
            srb->CdbLength = 0;
            srb->Function = SRB_FUNCTION_SHUTDOWN;
            srb->SrbStatus = 0;
            srb->OriginalRequest = Irp;
        }
 
        //
        // Set the retry count to zero.
        //
 
        irpStack->Parameters.Others.Argument4 = (PVOID) 0;
 
        //
        // Set up IoCompletion routine address.
        //
 
        IoSetCompletionRoutine(Irp, ClassIoComplete, srb, TRUE, TRUE, TRUE);
 
        //
        // Get next stack location and
        // set major function code.
        //
 
        irpStack = IoGetNextIrpStackLocation(Irp);
 
        irpStack->MajorFunction = IRP_MJ_SCSI;
 
        //
        // Set up SRB for execute scsi request.
        // Save SRB address in next stack for port driver.
        //
 
        irpStack->Parameters.Scsi.Srb = srb;
 
        //
        // Call the port driver to process the request.
        //
 
        IoMarkIrpPending(Irp);
        IoCallDriver(commonExtension->LowerDeviceObject, Irp);
    }
 
    return STATUS_PENDING;
}

Referenced by DriverEntry().

DiskStartFdo()

NTSTATUS NTAPI DiskStartFdo

(

IN PDEVICE_OBJECT

Fdo

)

Definition at line 911 of file pnp.c.

{
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = Fdo->DeviceExtension;
    PCOMMON_DEVICE_EXTENSION commonExtension = &(fdoExtension->CommonExtension);
    PDISK_DATA diskData = commonExtension->DriverData;
    STORAGE_HOTPLUG_INFO hotplugInfo = { 0 };
    DISK_CACHE_INFORMATION cacheInfo = { 0 };
    ULONG isPowerProtected = 0;
    NTSTATUS status;
 
    PAGED_CODE();
 
    //
    // Get the hotplug information, so we can turn off write cache if needed
    //
    // NOTE: Capabilities info is not good enough to determine hotplugedness
    //       as  we cannot determine device relations  information and other
    //       dependencies. Get the hotplug info instead
    //
 
    {
        PIRP irp;
        KEVENT event;
        IO_STATUS_BLOCK statusBlock = { 0 };
 
        KeInitializeEvent(&event, SynchronizationEvent, FALSE);
 
        irp = IoBuildDeviceIoControlRequest(IOCTL_STORAGE_GET_HOTPLUG_INFO,
                                            Fdo,
                                            NULL,
                                            0L,
                                            &hotplugInfo,
                                            sizeof(STORAGE_HOTPLUG_INFO),
                                            FALSE,
                                            &event,
                                            &statusBlock);
 
        if (irp != NULL) {
 
            // send to self -- classpnp handles this
            status = IoCallDriver(Fdo, irp);
            if (status == STATUS_PENDING) {
                KeWaitForSingleObject(&event,
                                      Executive,
                                      KernelMode,
                                      FALSE,
                                      NULL);
 
                status = statusBlock.Status;
            }
            NT_ASSERT(NT_SUCCESS(status));
        }
    }
 
    //
    // Clear the DEV_WRITE_CACHE flag now  and set
    // it below only if we read that from the disk
    //
    CLEAR_FLAG(fdoExtension->DeviceFlags, DEV_WRITE_CACHE);
    ADJUST_FUA_FLAG(fdoExtension);
 
    diskData->WriteCacheOverride = DiskWriteCacheDefault;
 
    //
    // Look into the registry to  see if the user
    // has chosen to override the default setting
    //
    ClassGetDeviceParameter(fdoExtension,
                            DiskDeviceParameterSubkey,
                            DiskDeviceUserWriteCacheSetting,
                            (PULONG)&diskData->WriteCacheOverride);
 
    if (diskData->WriteCacheOverride == DiskWriteCacheDefault)
    {
        //
        // The user has not overridden the default settings
        //
        if (TEST_FLAG(fdoExtension->ScanForSpecialFlags, CLASS_SPECIAL_DISABLE_WRITE_CACHE))
        {
            //
            // This flag indicates that we have faulty firmware and this
            // may cause the filesystem to refuse to mount on this media
            //
            TracePrint((TRACE_LEVEL_WARNING, TRACE_FLAG_PNP, "DiskStartFdo: Turning off write cache for %p due to a firmware issue\n", Fdo));
 
            diskData->WriteCacheOverride = DiskWriteCacheDisable;
        }
        else if (hotplugInfo.DeviceHotplug && !hotplugInfo.WriteCacheEnableOverride)
        {
            //
            // This flag indicates that the device is hotpluggable making it unsafe to enable caching
            //
            TracePrint((TRACE_LEVEL_WARNING, TRACE_FLAG_PNP, "DiskStartFdo: Turning off write cache for %p due to hotpluggable device\n", Fdo));
 
            diskData->WriteCacheOverride = DiskWriteCacheDisable;
        }
        else if (hotplugInfo.MediaHotplug)
        {
            //
            // This flag indicates that the media in the device cannot be reliably locked
            //
            TracePrint((TRACE_LEVEL_WARNING, TRACE_FLAG_PNP, "DiskStartFdo: Turning off write cache for %p due to unlockable media\n", Fdo));
 
            diskData->WriteCacheOverride = DiskWriteCacheDisable;
        }
        else
        {
            //
            // Even though the device does  not seem to have any obvious problems
            // we leave it to the user to modify the previous write cache setting
            //
        }
    }
 
    //
    // Query the disk to see if write cache is enabled
    // and  set the DEV_WRITE_CACHE flag appropriately
    //
 
    status = DiskGetCacheInformation(fdoExtension, &cacheInfo);
 
    if (NT_SUCCESS(status))
    {
        if (cacheInfo.WriteCacheEnabled == TRUE)
        {
            SET_FLAG(fdoExtension->DeviceFlags, DEV_WRITE_CACHE);
            ADJUST_FUA_FLAG(fdoExtension);
 
            if (diskData->WriteCacheOverride == DiskWriteCacheDisable)
            {
                //
                // Write cache is currently enabled on this
                // device, but we would like to turn it off
                //
                cacheInfo.WriteCacheEnabled = FALSE;
 
                DiskSetCacheInformation(fdoExtension, &cacheInfo);
            }
        }
        else
        {
            if (diskData->WriteCacheOverride == DiskWriteCacheEnable)
            {
                //
                // Write cache is currently disabled on this
                // device, but we  would  like to turn it on
                //
                cacheInfo.WriteCacheEnabled = TRUE;
 
                DiskSetCacheInformation(fdoExtension, &cacheInfo);
            }
        }
    }
 
    //
    // Query the registry to see if this disk is power-protected or not
    //
 
    CLEAR_FLAG(fdoExtension->DeviceFlags, DEV_POWER_PROTECTED);
 
    ClassGetDeviceParameter(fdoExtension, DiskDeviceParameterSubkey, DiskDeviceCacheIsPowerProtected, &isPowerProtected);
 
    if (isPowerProtected == 1)
    {
        SET_FLAG(fdoExtension->DeviceFlags, DEV_POWER_PROTECTED);
    }
 
    ADJUST_FUA_FLAG(fdoExtension);
 
    return STATUS_SUCCESS;
 
} // end DiskStartFdo()

Referenced by DriverEntry().

DiskStopDevice()

NTSTATUS NTAPI DiskStopDevice	(	IN PDEVICE_OBJECT	DeviceObject,
		IN UCHAR	Type
	)

Definition at line 600 of file pnp.c.

{
    UNREFERENCED_PARAMETER(DeviceObject);
    UNREFERENCED_PARAMETER(Type);
    return STATUS_SUCCESS;
}

Referenced by DriverEntry().

DiskUnload()

VOID NTAPI DiskUnload

(

IN PDRIVER_OBJECT

DriverObject

)

Definition at line 281 of file disk.c.

{
    PAGED_CODE();
 
#if defined(_X86_) || defined(_AMD64_)
    DiskCleanupDetectInfo(DriverObject);
#else
    // NB: Need to use UNREFERENCED_PARAMETER to prevent build error
    //     DriverObject is not referenced below in WPP_CLEANUP.
    //     WPP_CLEANUP is currently an "NOOP" marco.
    UNREFERENCED_PARAMETER(DriverObject);
#endif
 
 
    //
    // Cleans up tracing
    //
    WPP_CLEANUP(DriverObject);
 
    return;
}

Referenced by DriverEntry().

DiskWmiFunctionControl()

NTSTATUS NTAPI DiskWmiFunctionControl	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp,
		IN ULONG	GuidIndex,
		IN CLASSENABLEDISABLEFUNCTION	Function,
		IN BOOLEAN	Enable
	)

Definition at line 2366 of file diskwmi.c.

{
    NTSTATUS status = STATUS_SUCCESS;
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = DeviceObject->DeviceExtension;
 
    PAGED_CODE();
 
    if ((Function == DataBlockCollection) && Enable)
    {
        if ((GuidIndex == SmartStatusGuid) ||
            (GuidIndex == SmartDataGuid) ||
            (GuidIndex == SmartThresholdsGuid) ||
            (GuidIndex == SmartPerformFunction))
        {
            status = DiskEnableDisableFailurePrediction(fdoExtension,
                                                        TRUE);
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DeviceObject %p, Irp %p Enable -> %lx\n",
                       DeviceObject,
                       Irp,
                       status));
 
        } else {
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DeviceObject %p, Irp %p, GuidIndex %d %s for Collection\n",
                      DeviceObject, Irp,
                      GuidIndex,
                      Enable ? "Enabled" : "Disabled"));
        }
    } else if (Function == EventGeneration) {
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DeviceObject %p, Irp %p, GuidIndex %d %s for Event Generation\n",
                  DeviceObject, Irp,
                  GuidIndex,
                  Enable ? "Enabled" : "Disabled"));
 
 
        if ((GuidIndex == SmartEventGuid) && Enable)
        {
            status = DiskEnableDisableFailurePredictPolling(fdoExtension,
                                                   Enable,
                                                   0);
            TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DeviceObject %p, Irp %p %s -> %lx\n",
                       DeviceObject,
                       Irp,
                       Enable ? "DiskEnableSmartPolling" : "DiskDisableSmartPolling",
                       status));
        }
 
#if DBG
    } else {
        TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_WMI, "Disk: DeviceObject %p, Irp %p, GuidIndex %d %s for function %d\n",
                  DeviceObject, Irp,
                  GuidIndex,
                  Enable ? "Enabled" : "Disabled",
                  Function));
#endif
    }
 
    status = ClassWmiCompleteRequest(DeviceObject,
                                     Irp,
                                     status,
                                     0,
                                     IO_NO_INCREMENT);
    return status;
}

Referenced by DriverEntry().

GetSrbScsiData()

FORCEINLINE PCDB GetSrbScsiData	(	_In_ PSTORAGE_REQUEST_BLOCK	SrbEx,
		_In_opt_ PUCHAR	CdbLength8,
		_In_opt_ PULONG	CdbLength32,
		_In_opt_ PUCHAR	ScsiStatus,
		_In_opt_ PVOID *	SenseInfoBuffer,
		_In_opt_ PUCHAR	SenseInfoBufferLength
	)

Definition at line 994 of file disk.h.

{
    PCDB Cdb = NULL;
    ULONG i;
    PSRBEX_DATA SrbExData = NULL;
    BOOLEAN FoundEntry = FALSE;
 
    if ((SrbEx->Function == SRB_FUNCTION_STORAGE_REQUEST_BLOCK) &&
        (SrbEx->SrbFunction == SRB_FUNCTION_EXECUTE_SCSI)) {
        NT_ASSERT(SrbEx->NumSrbExData > 0);
 
        for (i = 0; i < SrbEx->NumSrbExData; i++) {
 
            // Skip any invalid offsets
            if ((SrbEx->SrbExDataOffset[i] < sizeof(STORAGE_REQUEST_BLOCK)) ||
                (SrbEx->SrbExDataOffset[i] > SrbEx->SrbLength)){
                // Catch any invalid offsets
                NT_ASSERT(FALSE);
                continue;
            }
 
            SrbExData = (PSRBEX_DATA)((PUCHAR)SrbEx + SrbEx->SrbExDataOffset[i]);
 
            switch (SrbExData->Type) {
 
                case SrbExDataTypeScsiCdb16:
                    if (SrbEx->SrbExDataOffset[i] + sizeof(SRBEX_DATA_SCSI_CDB16) <= SrbEx->SrbLength) {
                        FoundEntry = TRUE;
                        if (CdbLength8) {
                            *CdbLength8 = ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->CdbLength;
                        }
 
                        if (((PSRBEX_DATA_SCSI_CDB16) SrbExData)->CdbLength > 0) {
                            Cdb = (PCDB)((PSRBEX_DATA_SCSI_CDB16) SrbExData)->Cdb;
                        }
 
                        if (ScsiStatus) {
                            *ScsiStatus =
                                ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->ScsiStatus;
                        }
 
                        if (SenseInfoBuffer) {
                            *SenseInfoBuffer =
                                ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->SenseInfoBuffer;
                        }
 
                        if (SenseInfoBufferLength) {
                            *SenseInfoBufferLength =
                                ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->SenseInfoBufferLength;
                        }
 
                    } else {
                        // Catch invalid offset
                        NT_ASSERT(FALSE);
                    }
                    break;
 
                case SrbExDataTypeScsiCdb32:
                    if (SrbEx->SrbExDataOffset[i] + sizeof(SRBEX_DATA_SCSI_CDB32) <= SrbEx->SrbLength) {
                        FoundEntry = TRUE;
                        if (CdbLength8) {
                            *CdbLength8 = ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->CdbLength;
                        }
 
                        if (((PSRBEX_DATA_SCSI_CDB32) SrbExData)->CdbLength > 0) {
                            Cdb = (PCDB)((PSRBEX_DATA_SCSI_CDB32) SrbExData)->Cdb;
                        }
 
                        if (ScsiStatus) {
                            *ScsiStatus =
                                ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->ScsiStatus;
                        }
 
                        if (SenseInfoBuffer) {
                            *SenseInfoBuffer =
                                ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->SenseInfoBuffer;
                        }
 
                        if (SenseInfoBufferLength) {
                            *SenseInfoBufferLength =
                                ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->SenseInfoBufferLength;
                        }
 
                    } else {
                        // Catch invalid offset
                        NT_ASSERT(FALSE);
                    }
                    break;
 
                case SrbExDataTypeScsiCdbVar:
                    if (SrbEx->SrbExDataOffset[i] + sizeof(SRBEX_DATA_SCSI_CDB_VAR) <= SrbEx->SrbLength) {
                        FoundEntry = TRUE;
                        if (CdbLength32) {
                            *CdbLength32 = ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->CdbLength;
                        }
 
                        if (((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->CdbLength > 0) {
                            Cdb = (PCDB)((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->Cdb;
                        }
 
                        if (ScsiStatus) {
                            *ScsiStatus =
                                ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->ScsiStatus;
                        }
 
                        if (SenseInfoBuffer) {
                            *SenseInfoBuffer =
                                ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->SenseInfoBuffer;
                        }
 
                        if (SenseInfoBufferLength) {
                            *SenseInfoBufferLength =
                                ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->SenseInfoBufferLength;
                        }
 
                    } else {
                        // Catch invalid offset
                        NT_ASSERT(FALSE);
                    }
                    break;
            }
 
            if (FoundEntry) {
                break;
            }
        }
 
    } else {
 
        if (CdbLength8) {
            *CdbLength8 = 0;
        }
 
        if (CdbLength32) {
            *CdbLength32 = 0;
        }
 
        if (ScsiStatus) {
            *ScsiStatus = 0;
        }
 
        if (SenseInfoBuffer) {
            *SenseInfoBuffer = NULL;
        }
 
        if (SenseInfoBufferLength) {
            *SenseInfoBufferLength = 0;
        }
    }
 
    return Cdb;
}

Referenced by DiskFdoProcessError(), and DiskInfoExceptionComplete().

ResetBus()

VOID ResetBus

(

IN PDEVICE_OBJECT

DeviceObject

)

Definition at line 2705 of file disk.c.

{
    PIO_STACK_LOCATION irpStack;
    PIRP irp;
 
    PFUNCTIONAL_DEVICE_EXTENSION fdoExtension = Fdo->DeviceExtension;
    PSCSI_REQUEST_BLOCK srb;
    PCOMPLETION_CONTEXT context;
    PSTORAGE_REQUEST_BLOCK srbEx = NULL;
    PSTOR_ADDR_BTL8 storAddrBtl8 = NULL;
 
    TracePrint((TRACE_LEVEL_INFORMATION, TRACE_FLAG_GENERAL, "Disk ResetBus: Sending reset bus request to port driver.\n"));
 
    //
    // Allocate Srb from nonpaged pool.
    //
 
    context = ExAllocatePoolWithTag(NonPagedPoolNx,
                                    sizeof(COMPLETION_CONTEXT),
                                    DISK_TAG_CCONTEXT);
 
    if(context == NULL) {
        return;
    }
 
    //
    // Save the device object in the context for use by the completion
    // routine.
    //
 
    context->DeviceObject = Fdo;
    srb = &context->Srb.Srb;
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx = &context->Srb.SrbEx;
 
        //
        // Zero out srb
        //
 
        RtlZeroMemory(srbEx, sizeof(context->Srb.SrbExBuffer));
 
        //
        // Set up STORAGE_REQUEST_BLOCK fields
        //
 
        srbEx->Length = FIELD_OFFSET(STORAGE_REQUEST_BLOCK, Signature);
        srbEx->Function = SRB_FUNCTION_STORAGE_REQUEST_BLOCK;
        srbEx->Signature = SRB_SIGNATURE;
        srbEx->Version = STORAGE_REQUEST_BLOCK_VERSION_1;
        srbEx->SrbLength = sizeof(context->Srb.SrbExBuffer);
        srbEx->SrbFunction = SRB_FUNCTION_RESET_BUS;
        srbEx->AddressOffset = sizeof(STORAGE_REQUEST_BLOCK);
 
        //
        // Set up address fields
        //
 
        storAddrBtl8 = (PSTOR_ADDR_BTL8) ((PUCHAR)srbEx + srbEx->AddressOffset);
        storAddrBtl8->Type = STOR_ADDRESS_TYPE_BTL8;
        storAddrBtl8->AddressLength = STOR_ADDR_BTL8_ADDRESS_LENGTH;
 
    } else {
 
        //
        // Zero out srb.
        //
 
        RtlZeroMemory(srb, SCSI_REQUEST_BLOCK_SIZE);
 
        //
        // Write length to SRB.
        //
 
        srb->Length = SCSI_REQUEST_BLOCK_SIZE;
 
        srb->Function = SRB_FUNCTION_RESET_BUS;
 
    }
 
    //
    // Build the asynchronous request to be sent to the port driver.
    // Since this routine is called from a DPC the IRP should always be
    // available.
    //
 
    irp = IoAllocateIrp(Fdo->StackSize, FALSE);
 
    if (irp == NULL) {
        FREE_POOL(context);
        return;
    }
 
    ClassAcquireRemoveLock(Fdo, irp);
 
    IoSetCompletionRoutine(irp,
                           (PIO_COMPLETION_ROUTINE)ClassAsynchronousCompletion,
                           context,
                           TRUE,
                           TRUE,
                           TRUE);
 
    irpStack = IoGetNextIrpStackLocation(irp);
 
    irpStack->MajorFunction = IRP_MJ_SCSI;
 
    if (fdoExtension->AdapterDescriptor->SrbType == SRB_TYPE_STORAGE_REQUEST_BLOCK) {
        srbEx->RequestPriority = IoGetIoPriorityHint(irp);
        srbEx->OriginalRequest = irp;
    } else {
        srb->OriginalRequest = irp;
    }
 
    //
    // Store the SRB address in next stack for port driver.
    //
 
    irpStack->Parameters.Scsi.Srb = srb;
 
    //
    // Call the port driver with the IRP.
    //
 
    IoCallDriver(fdoExtension->CommonExtension.LowerDeviceObject, irp);
 
    return;
 
} // end ResetBus()

Referenced by DiskFdoProcessError().

SetSrbScsiData()

FORCEINLINE VOID SetSrbScsiData	(	_In_ PSTORAGE_REQUEST_BLOCK	SrbEx,
		_In_ UCHAR	CdbLength8,
		_In_ ULONG	CdbLength32,
		_In_ UCHAR	ScsiStatus,
		_In_opt_ PVOID	SenseInfoBuffer,
		_In_ UCHAR	SenseInfoBufferLength
	)

Definition at line 1184 of file disk.h.

{
    ULONG i;
    PSRBEX_DATA SrbExData = NULL;
    BOOLEAN FoundEntry = FALSE;
 
    if ((SrbEx->Function == SRB_FUNCTION_STORAGE_REQUEST_BLOCK) &&
        (SrbEx->SrbFunction == SRB_FUNCTION_EXECUTE_SCSI)) {
        NT_ASSERT(SrbEx->NumSrbExData > 0);
 
        for (i = 0; i < SrbEx->NumSrbExData; i++) {
 
            // Skip any invalid offsets
            if ((SrbEx->SrbExDataOffset[i] < sizeof(STORAGE_REQUEST_BLOCK)) ||
                (SrbEx->SrbExDataOffset[i] > SrbEx->SrbLength)){
                // Catch any invalid offsets
                NT_ASSERT(FALSE);
                continue;
            }
 
            SrbExData = (PSRBEX_DATA)((PUCHAR)SrbEx + SrbEx->SrbExDataOffset[i]);
 
            switch (SrbExData->Type) {
 
                case SrbExDataTypeScsiCdb16:
                    if (SrbEx->SrbExDataOffset[i] + sizeof(SRBEX_DATA_SCSI_CDB16) <= SrbEx->SrbLength) {
                        FoundEntry = TRUE;
                        ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->CdbLength = CdbLength8;
                        ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->ScsiStatus = ScsiStatus;
                        ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->SenseInfoBuffer = SenseInfoBuffer;
                        ((PSRBEX_DATA_SCSI_CDB16) SrbExData)->SenseInfoBufferLength = SenseInfoBufferLength;
                    } else {
                        // Catch invalid offset
                        NT_ASSERT(FALSE);
                    }
                    break;
 
                case SrbExDataTypeScsiCdb32:
                    if (SrbEx->SrbExDataOffset[i] + sizeof(SRBEX_DATA_SCSI_CDB32) <= SrbEx->SrbLength) {
                        FoundEntry = TRUE;
                        ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->CdbLength = CdbLength8;
                        ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->ScsiStatus = ScsiStatus;
                        ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->SenseInfoBuffer = SenseInfoBuffer;
                        ((PSRBEX_DATA_SCSI_CDB32) SrbExData)->SenseInfoBufferLength = SenseInfoBufferLength;
                    } else {
                        // Catch invalid offset
                        NT_ASSERT(FALSE);
                    }
                    break;
 
                case SrbExDataTypeScsiCdbVar:
                    if (SrbEx->SrbExDataOffset[i] + sizeof(SRBEX_DATA_SCSI_CDB_VAR) <= SrbEx->SrbLength) {
                        FoundEntry = TRUE;
                        ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->CdbLength = CdbLength32;
                        ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->ScsiStatus = ScsiStatus;
                        ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->SenseInfoBuffer = SenseInfoBuffer;
                        ((PSRBEX_DATA_SCSI_CDB_VAR) SrbExData)->SenseInfoBufferLength = SenseInfoBufferLength;
                    } else {
                        // Catch invalid offset
                        NT_ASSERT(FALSE);
                    }
                    break;
            }
 
            if (FoundEntry) {
                break;
            }
        }
 
    }
 
    return;
}

Referenced by DiskInfoExceptionComplete().

Variable Documentation

DisableWriteCache

IO_WORKITEM_ROUTINE DisableWriteCache

Definition at line 589 of file disk.h.

Referenced by DiskFdoProcessError().

DiskBadControllers

CLASSPNP_SCAN_FOR_SPECIAL_INFO DiskBadControllers[]

extern

Definition at line 42 of file data.c.

Referenced by DiskInitFdo().

DiskFlushComplete

IO_COMPLETION_ROUTINE DiskFlushComplete

Definition at line 599 of file disk.h.

Referenced by DiskFlushDispatch().

DiskIoctlVerifyThread

IO_WORKITEM_ROUTINE DiskIoctlVerifyThread

Definition at line 591 of file disk.h.

Referenced by DiskIoctlVerify().

DiskMediaTypes

const DISK_MEDIA_TYPES_LIST DiskMediaTypes[]

extern

Definition at line 82 of file data.c.

Referenced by DiskDetermineMediaTypes().

DiskMediaTypesExclude

const DISK_MEDIA_TYPES_LIST DiskMediaTypesExclude[]

extern

Definition at line 74 of file data.c.

Referenced by DiskDetermineMediaTypes().

DiskWmiFdoGuidList

GUIDREGINFO DiskWmiFdoGuidList[]

extern

Definition at line 92 of file diskwmi.c.

Referenced by DiskFdoQueryWmiRegInfo(), and DriverEntry().

DriverEntry

DRIVER_INITIALIZE DriverEntry

Definition at line 485 of file disk.h.