rostests/tests/diskspeed/diskspeed.c

   1 /*
   2  * Copyright (C) 2004 ReactOS Team
   3  *
   4  * COPYRIGHT:       See COPYING in the top level directory
   5  * PROJECT:         ReactOS diskspeed.exe
   6  * FILE:            apps/tests/diskspeed/diskspeed.c
   7  * PURPOSE:         Determines disk transfer rates
   8  * PROGRAMMER:
   9  */
  10
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #define WIN32_NO_STATUS
  14 #include <windows.h>
  15 #define NTOS_MODE_USER
  16 #include <ndk/ntndk.h>
  17
  18 #define _NTSCSI_USER_MODE_
  19 #include <ntddscsi.h>
  20 #include <scsi.h>
  21
  22 BOOL GetInquiryData(HANDLE hDevice, PINQUIRYDATA InquiryData)
  23 {
  24   BOOL Result;
  25   DWORD dwReturned;
  26   SCSI_ADDRESS ScsiAddress;
  27   PSCSI_ADAPTER_BUS_INFO AdapterInfo;
  28   PSCSI_INQUIRY_DATA InquiryBuffer;
  29   BYTE Buffer[4096];
  30   int i;
  31
  32   Result = DeviceIoControl(hDevice,
  33                            IOCTL_SCSI_GET_ADDRESS,
  34                            NULL,
  35                            0,
  36                            &ScsiAddress,
  37                            sizeof(ScsiAddress),
  38                            &dwReturned,
  39                            FALSE);
  40   if (Result == FALSE)
  41     {
  42       return FALSE;
  43     }
  44   Result = DeviceIoControl(hDevice,
  45                            IOCTL_SCSI_GET_INQUIRY_DATA,
  46                            NULL,
  47                            0,
  48                            Buffer,
  49                            sizeof(Buffer),
  50                            &dwReturned,
  51                            FALSE);
  52   if (Result)
  53     {
  54       AdapterInfo = (PSCSI_ADAPTER_BUS_INFO)Buffer;
  55       for (i = 0; i < AdapterInfo->NumberOfBuses; i++)
  56         {
  57           InquiryBuffer = (PSCSI_INQUIRY_DATA) (Buffer + AdapterInfo->BusData[i].InquiryDataOffset);
  58           if (AdapterInfo->BusData[i].InquiryDataOffset)
  59             {
  60                while (1)
  61                  {
  62                    if (InquiryBuffer->PathId == ScsiAddress.PathId &&
  63                        InquiryBuffer->TargetId == ScsiAddress.TargetId &&
  64                        InquiryBuffer->Lun == ScsiAddress.Lun)
  65                      {
  66                        memcpy(InquiryData, InquiryBuffer->InquiryData, sizeof(INQUIRYDATA));
  67                        return TRUE;
  68                      }
  69                    if (InquiryBuffer->NextInquiryDataOffset == 0)
  70                      {
  71                        break;
  72                      }
  73                    InquiryBuffer = (PSCSI_INQUIRY_DATA) (Buffer + InquiryBuffer->NextInquiryDataOffset);
  74                  }
  75             }
  76         }
  77     }
  78   return FALSE;
  79 }
  80
  81
  82
  83 int main(void)
  84 {
  85     HANDLE hDevice;
  86     OVERLAPPED ov;
  87
  88     PBYTE Buffer = NULL ;
  89     DWORD Start;
  90     DWORD dwReturned;
  91     DWORD dwReadTotal;
  92     DWORD Size;
  93     BOOL Result;
  94     ULONG Drive;
  95     CHAR Name[20];
  96
  97     INQUIRYDATA InquiryData;
  98
  99
 100     Drive = 0;
 101     while (1)
 102       {
 103         sprintf(Name, "\\\\.\\PHYSICALDRIVE%ld", Drive);
 104         hDevice = CreateFile(Name,
 105                              GENERIC_READ,
 106                              FILE_SHARE_READ,
 107                              NULL,
 108                              OPEN_EXISTING,
 109                              0,
 110                              NULL);
 111         if (hDevice == INVALID_HANDLE_VALUE)
 112           {
 113             if (Drive > 0)
 114               {
 115                 VirtualFree(Buffer, 512 * 1024, MEM_RELEASE);
 116               }
 117             else
 118               {
 119                 printf("Cannot open '%s'\n", Name);
 120               }
 121             break;
 122           }
 123         if (Drive == 0)
 124           {
 125             printf("Transfer Size (kB)           1     2     4     8    16    32    64   128   256\n");
 126             printf("Transfer Rate (MB/s)\n");
 127             printf("-------------------------------------------------------------------------------\n");
 128
 129             Buffer = VirtualAlloc(NULL, 512 * 1024, MEM_COMMIT, PAGE_READWRITE);
 130           }
 131         Result = GetInquiryData(hDevice, &InquiryData);
 132         if (Result)
 133           {
 134             printf("%.24s ", InquiryData.VendorId);
 135           }
 136         else
 137           {
 138             printf("Disk %ld                   ", Drive + 1);
 139           }
 140         Size = 1024;
 141         memset(&ov, 0, sizeof(OVERLAPPED));
 142         while (Size <= 256 * 1024)
 143           {
 144             memset(Buffer, 0, Size);
 145             dwReadTotal = 0;
 146
 147             Start = GetTickCount() + 2000;
 148             while (Start > GetTickCount())
 149               {
 150                 Result = ReadFile(hDevice, Buffer, Size, &dwReturned, &ov);
 151                 if (Result)
 152                   {
 153                     dwReadTotal += dwReturned;
 154                     ov.Offset += dwReturned;
 155                   }
 156               }
 157             dwReadTotal /= 2048;
 158             printf("%3ld.%ld ", dwReadTotal / 1024, (dwReadTotal % 1024) * 10 / 1024);
 159             Size *= 2;
 160           }
 161         printf("\n");
 162         CloseHandle(hDevice);
 163         Drive++;
 164       }
 165     printf("\n");
 166
 167
 168     return 0;
 169 }