From fa397a0e33376ac14ac2d8a3ca0ec1a8f2f39bb7 Mon Sep 17 00:00:00 2001
From: Hartmut Birr <osexpert@googlemail.com>
Date: Sun, 7 Mar 2004 20:12:13 +0000
Subject: [PATCH] - Added a disk speed test utility.

svn path=/trunk/; revision=8574
---
 reactos/apps/tests/diskspeed/.cvsignore  |   6 +
 reactos/apps/tests/diskspeed/diskspeed.c | 165 +++++++++++++++++++++++
 reactos/apps/tests/diskspeed/makefile    |  23 ++++
 3 files changed, 194 insertions(+)
 create mode 100644 reactos/apps/tests/diskspeed/.cvsignore
 create mode 100644 reactos/apps/tests/diskspeed/diskspeed.c
 create mode 100644 reactos/apps/tests/diskspeed/makefile

diff --git a/reactos/apps/tests/diskspeed/.cvsignore b/reactos/apps/tests/diskspeed/.cvsignore
new file mode 100644
index 00000000000..d63774a7353
--- /dev/null
+++ b/reactos/apps/tests/diskspeed/.cvsignore
@@ -0,0 +1,6 @@
+*.o
+*.d
+*.exe
+*.coff
+*.sym
+*.map
diff --git a/reactos/apps/tests/diskspeed/diskspeed.c b/reactos/apps/tests/diskspeed/diskspeed.c
new file mode 100644
index 00000000000..d19f8c0f344
--- /dev/null
+++ b/reactos/apps/tests/diskspeed/diskspeed.c
@@ -0,0 +1,165 @@
+/*
+ * Copyright (C) 2004 ReactOS Team
+ *
+ * COPYRIGHT:       See COPYING in the top level directory
+ * PROJECT:         ReactOS diskspeed.exe
+ * FILE:            apps/tests/diskspeed/diskspeed.c
+ * PURPOSE:         Determines disk transfer rates
+ * PROGRAMMER:      Hartmut Birr 
+ */
+
+#include <windows.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <ddk/ntddk.h>
+#include <ddk/ntddscsi.h>
+#include <ddk/scsi.h>
+
+BOOL GetInquiryData(HANDLE hDevice, PINQUIRYDATA InquiryData)
+{
+  BOOL Result;
+  DWORD dwReturned;
+  SCSI_ADDRESS ScsiAddress;
+  PSCSI_ADAPTER_BUS_INFO AdapterInfo;
+  PSCSI_INQUIRY_DATA InquiryBuffer;
+  BYTE Buffer[4096];
+  int i;
+
+  Result = DeviceIoControl(hDevice,
+	                   IOCTL_SCSI_GET_ADDRESS,
+			   NULL,
+			   0,
+			   &ScsiAddress,
+			   sizeof(ScsiAddress),
+			   &dwReturned,
+			   FALSE);
+  if (Result == FALSE)
+    {
+      return FALSE;
+    }
+  Result = DeviceIoControl(hDevice,
+	                   IOCTL_SCSI_GET_INQUIRY_DATA,
+			   NULL,
+			   0,
+			   Buffer,
+			   sizeof(Buffer),
+			   &dwReturned,
+			   FALSE);
+  if (Result)
+    {
+      AdapterInfo = (PSCSI_ADAPTER_BUS_INFO)Buffer;
+      for (i = 0; i < AdapterInfo->NumberOfBuses; i++)
+	{
+	  InquiryBuffer = (PSCSI_INQUIRY_DATA) (Buffer + AdapterInfo->BusData[i].InquiryDataOffset);
+	  if (AdapterInfo->BusData[i].InquiryDataOffset)
+	    {
+	       while (1)
+	         {
+		   if (InquiryBuffer->PathId == ScsiAddress.PathId &&
+		       InquiryBuffer->TargetId == ScsiAddress.TargetId &&
+		       InquiryBuffer->Lun == ScsiAddress.Lun)
+		     {
+		       memcpy(InquiryData, InquiryBuffer->InquiryData, sizeof(INQUIRYDATA));
+		       return TRUE;
+		     }
+		   if (InquiryBuffer->NextInquiryDataOffset == 0)
+		     {
+		       break;
+		     }
+		   InquiryBuffer = (PSCSI_INQUIRY_DATA) (Buffer + InquiryBuffer->NextInquiryDataOffset);
+		 }
+	    }
+	}
+    }
+  return FALSE;
+}
+
+
+
+int main(void)
+{
+    HANDLE hDevice;
+    OVERLAPPED ov;
+
+    PBYTE Buffer;
+    DWORD Start;
+    DWORD dwReturned;
+    DWORD dwReadTotal;
+    DWORD Size;
+    BOOL Result;
+    ULONG Drive;
+    CHAR Name[20];
+
+    INQUIRYDATA InquiryData;
+
+
+    Drive = 0; 
+    while (1)
+      {
+        sprintf(Name, "\\\\.\\PHYSICALDRIVE%ld", Drive);
+	hDevice = CreateFile(Name,
+	                     GENERIC_READ,
+			     FILE_SHARE_READ,
+			     NULL,
+			     OPEN_EXISTING,
+			     0,
+			     NULL);
+	if (hDevice == INVALID_HANDLE_VALUE)
+	  {
+	    if (Drive > 0)
+	      {
+	        VirtualFree(Buffer, 512 * 1024, MEM_RELEASE);
+	      }
+	    else
+	      {
+	        printf("Cannot open '%s'\n", Name);
+	      }
+	    break;
+	  }
+        if (Drive == 0)
+	  {
+            printf("Transfer Size (kB)           1     2     4     8    16    32    64   128   256\n");
+            printf("Transfer Rate (MB/s)\n");
+            printf("-------------------------------------------------------------------------------\n");
+            
+	    Buffer = VirtualAlloc(NULL, 512 * 1024, MEM_COMMIT, PAGE_READWRITE);
+	  }
+        Result = GetInquiryData(hDevice, &InquiryData);
+        if (Result)
+	  {
+            printf("%.24s ", InquiryData.VendorId);
+	  }
+	else
+	  {
+	    printf("Disk %ld                   ", Drive + 1);
+	  }
+        Size = 1024;
+        memset(&ov, 0, sizeof(OVERLAPPED));
+	while (Size <= 256 * 1024)
+	  {
+	    memset(Buffer, 0, Size);
+	    dwReadTotal = 0;
+
+            Start = GetTickCount() + 2000;
+	    while (Start > GetTickCount())
+	      {
+	        Result = ReadFile(hDevice, Buffer, Size, &dwReturned, &ov);
+		if (Result)
+		  {
+		    dwReadTotal += dwReturned;
+		    ov.Offset += dwReturned;
+		  }
+	      }
+	    dwReadTotal /= 2048;
+            printf("%3ld.%ld ", dwReadTotal / 1024, (dwReadTotal % 1024) * 10 / 1024);
+	    Size *= 2;
+	  }
+        printf("\n");
+	CloseHandle(hDevice);
+	Drive++;
+      }
+    printf("\n");
+
+
+    return 0;
+}
diff --git a/reactos/apps/tests/diskspeed/makefile b/reactos/apps/tests/diskspeed/makefile
new file mode 100644
index 00000000000..ead30091079
--- /dev/null
+++ b/reactos/apps/tests/diskspeed/makefile
@@ -0,0 +1,23 @@
+# $Id: makefile,v 1.1 2004/03/07 20:12:13 hbirr Exp $
+
+PATH_TO_TOP = ../../..
+
+TARGET_NORC = yes
+
+TARGET_TYPE = program
+
+TARGET_APPTYPE = console
+
+TARGET_NAME = diskspeed
+
+TARGET_SDKLIBS = kernel32.a
+
+TARGET_OBJECTS = $(TARGET_NAME).o
+
+TARGET_CFLAGS = -Wall -Werror
+
+include $(PATH_TO_TOP)/rules.mak
+
+include $(TOOLS_PATH)/helper.mk
+
+# EOF