#include <ntifs.h>
#include <ntddscsi.h>
#include <ntdddisk.h>
#include <stdio.h>

void PrintIdeInfo (int drive, ULONG diskdata [256]);

char *ConvertToString (ULONG diskdata [256], int firstIndex, int lastIndex);

BOOLEAN getDiskID( PDEVICE_OBJECT DeviceObject,ULONG drive );

#define IDENTIFY_BUFFER_SIZE 512

#define  IDE_ATAPI_IDENTIFY  0xA1  //  Returns ID sector for ATAPI.
#define  IDE_ATA_IDENTIFY    0xEC  //  Returns ID sector for ATA.

#define  DFP_RECEIVE_DRIVE_DATA   0x0007c088

//輸出地址
char IdOutCmd [sizeof (SENDCMDOUTPARAMS) + IDENTIFY_BUFFER_SIZE - 1];

BOOLEAN getDiskID( PDEVICE_OBJECT DeviceObject,ULONG drive )
/*
* DeviceObject:物理PDO
* drive:磁盤設備號
*/
{
    PUCHAR buffer;
    PSRB_IO_CONTROL srbControl;
    ULONG         controlCode = 0;
    PIRP irp2;
    KEVENT event;
    IO_STATUS_BLOCK ioStatus;   
    NTSTATUS status;
    ULONG length;
 PSENDCMDINPARAMS cmdInParameters;
 
 SENDCMDINPARAMS  scip;

 memset (&scip, 0, sizeof(scip));
    memset (IdOutCmd, 0, sizeof(IdOutCmd));
 scip.cBufferSize = IDENTIFY_BUFFER_SIZE;
 scip.irDriveRegs.bFeaturesReg = 0;
 scip.irDriveRegs.bSectorCountReg = 1;
 scip.irDriveRegs.bSectorNumberReg = 1;
 scip.irDriveRegs.bCylLowReg = 0;
 scip.irDriveRegs.bCylHighReg = 0;

   // Compute the drive number.
 scip.irDriveRegs.bDriveHeadReg = 0xA0 | (((UCHAR)drive & 1) << 4);

   // The command can either be IDE identify or ATAPI identify.
 scip.irDriveRegs.bCommandReg = IDE_ATA_IDENTIFY;
 scip.bDriveNumber = (UCHAR)drive;
 scip.cBufferSize = IDENTIFY_BUFFER_SIZE;

    //
    // 發DeviceIo，IdOutCmd接收結果
    //
    irp2 = IoBuildDeviceIoControlRequest(DFP_RECEIVE_DRIVE_DATA,
                            DeviceObject,
                            &scip,
                            sizeof(SENDCMDINPARAMS) - 1,
                            IdOutCmd,
                            sizeof(SENDCMDOUTPARAMS) + IDENTIFY_BUFFER_SIZE - 1,
                            FALSE,
                            &event,
                            &ioStatus);

    if (irp2 == NULL) {
        DbgPrint("Build DeviceIoControl Error \n");
  return FALSE;
    }

    //
    // 向設備發Irp
    //

    status = IoCallDriver(DeviceObject, irp2);

    if (status == STATUS_PENDING) {
        KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL);
        status = ioStatus.Status;
    }

    //
    // 發送成功打印信息
    //
   
    if (NT_SUCCESS(status)) {
          ULONG diskdata [256];
          int ijk = 0;
          USHORT *pIdSector = (USHORT *)
                             ((PSENDCMDOUTPARAMS) IdOutCmd) -> bBuffer;
    DbgPrint("successs in read disk id \n");   

          for (ijk = 0; ijk < 256; ijk++)
            diskdata [ijk] = pIdSector [ijk];
       
     //打印函數
        PrintIdeInfo (drive, diskdata);
  //這要調用完成例程而不是釋放IRP，具體請看DDK文檔
  IoCompleteRequest( irp2,IO_NO_INCREMENT );
        return TRUE;    
         
    } else {
       DbgPrint("get disk id error\n");
   
    }
   
    return FALSE;
}

void PrintIdeInfo (int drive, ULONG diskdata [256])
{
 char string1 [1024];

 strcpy (string1, ConvertToString (diskdata, 10, 19));

 DbgPrint("Disk NO. %d DiskID %s \n",drive,ConvertToString (diskdata, 10, 19));
}

char *ConvertToString (ULONG diskdata [256], int firstIndex, int lastIndex)
{
/*
* 功能如下：高低字節對調,結尾補'\0'
* string[10-19]遇空結尾
*/
   static char string [1024];
   int index = 0;
   int position = 0;

      //  each integer has two characters stored in it backwards
   for (index = firstIndex; index <= lastIndex; index++)
   {
         //  高字節
      string [position] = (char) (diskdata [index] / 256);
      position++;

         //  低字節
      string [position] = (char) (diskdata [index] % 256);
      position++;
   }

      //  結尾
   string [position] = '\0';
     
   for (index = position - 1; index > 0 && ' ' == string [index]; index--)
      string [index] = '\0';

   return string;
}