From a73624a3c4fda928b9377bb31df3b8df95515a39 Mon Sep 17 00:00:00 2001
From: Christophe Grenier <grenier@cgsecurity.org>
Date: Sat, 18 Jul 2015 14:06:38 +0200
Subject: [PATCH] Windows version - fix includes so
 IOCTL_STORAGE_QUERY_PROPERTY is defined

---
 configure.ac  |  6 ++++++
 src/hdwin32.c | 14 ++++++--------
 2 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/configure.ac b/configure.ac
index 11165d22..ca3f6d56 100644
--- a/configure.ac
+++ b/configure.ac
@@ -431,6 +431,12 @@ AC_CHECK_HEADERS([winioctl.h],,,
 #include <windef.h>
 #endif
 ]])
+AC_CHECK_HEADERS([w32api/winioctl.h],,,
+[[
+#ifdef HAVE_WINDEF_H
+#include <windef.h>
+#endif
+]])
 
 AC_CHECK_HEADERS([ext2fs/ext2_fs.h],,,
 [[
diff --git a/src/hdwin32.c b/src/hdwin32.c
index 8bc76f5c..523014c7 100644
--- a/src/hdwin32.c
+++ b/src/hdwin32.c
@@ -22,7 +22,7 @@
 #ifdef HAVE_CONFIG_H
 #include <config.h>
 #endif
- 
+
 #if defined(__CYGWIN__) || defined(__MINGW32__)
 #include <stdio.h>
 #include "types.h"
@@ -38,18 +38,17 @@
 #include <winbase.h>
 #endif
 #include <ctype.h>	/* isspace */
-#ifdef HAVE_W32API_DDK_NTDDDISK_H
-#include <w32api/ddk/ntdddisk.h>
+#ifdef HAVE_WINIOCTL_H
+#include <winioctl.h>
 #endif
-#ifdef HAVE_DDK_NTDDSTOR_H
-#include <ddk/ntddstor.h>
+#ifdef HAVE_W32API_WINIOCTL_H
+#include <w32api/winioctl.h>
 #endif
 #include "log.h"
 #include "hdwin32.h"
 
 void file_win32_disk_get_model(HANDLE handle, disk_t *dev, const int verbose)
 {
-#ifdef IOCTL_STORAGE_QUERY_PROPERTY
   DWORD               cbBytesReturned = 0;
   STORAGE_PROPERTY_QUERY query;
   char buffer [10240];
@@ -64,7 +63,7 @@ void file_win32_disk_get_model(HANDLE handle, disk_t *dev, const int verbose)
 	&buffer,
 	sizeof (buffer)-1,
 	&cbBytesReturned, NULL) )
-  {         
+  {
     const STORAGE_DEVICE_DESCRIPTOR * descrip = (const STORAGE_DEVICE_DESCRIPTOR *) & buffer;
     const unsigned int offsetVendor=descrip->VendorIdOffset;
     const unsigned int offsetProduct=descrip->ProductIdOffset;
@@ -111,6 +110,5 @@ void file_win32_disk_get_model(HANDLE handle, disk_t *dev, const int verbose)
       dev->model=NULL;
     }
   }
-#endif
 }
 #endif