dca9d1724ba0bdd65984e43e34ee05349b936c46
[reactos.git] / rosapps / 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: Hartmut Birr
9 */
10
11 #include <stdio.h>
12 #include <stdlib.h>
13
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 }