From 04d0e17fdf1faf00472f06412b68f091ffb6f6fd Mon Sep 17 00:00:00 2001
From: Pierre Schweitzer <pierre@reactos.org>
Date: Thu, 2 Mar 2017 19:33:25 +0000
Subject: [PATCH] [KERNEL32_APITEST] Add tests for DeviceIoControl. These are
 mainly used to test our storage stack

CORE-12858

svn path=/trunk/; revision=74025
---
 rostests/apitests/kernel32/CMakeLists.txt    |  1 +
 rostests/apitests/kernel32/DeviceIoControl.c | 68 ++++++++++++++++++++
 rostests/apitests/kernel32/testlist.c        |  2 +
 3 files changed, 71 insertions(+)
 create mode 100644 rostests/apitests/kernel32/DeviceIoControl.c

diff --git a/rostests/apitests/kernel32/CMakeLists.txt b/rostests/apitests/kernel32/CMakeLists.txt
index 1cfd0bdf951..a6fe595aca4 100644
--- a/rostests/apitests/kernel32/CMakeLists.txt
+++ b/rostests/apitests/kernel32/CMakeLists.txt
@@ -3,6 +3,7 @@ add_subdirectory(redirptest)
 
 list(APPEND SOURCE
     DefaultActCtx.c
+    DeviceIoControl.c
     dosdev.c
     FindActCtxSectionStringW.c
     FindFiles.c
diff --git a/rostests/apitests/kernel32/DeviceIoControl.c b/rostests/apitests/kernel32/DeviceIoControl.c
new file mode 100644
index 00000000000..e421c4598e9
--- /dev/null
+++ b/rostests/apitests/kernel32/DeviceIoControl.c
@@ -0,0 +1,68 @@
+/*
+ * PROJECT:         ReactOS api tests
+ * LICENSE:         GPLv2+ - See COPYING in the top level directory
+ * PURPOSE:         Test for DeviceIoControl
+ * PROGRAMMER:      Pierre Schweitzer <pierre@reactos.org>
+ */
+
+#include <apitest.h>
+#include <strsafe.h>
+#include <winioctl.h>
+
+START_TEST(DeviceIoControl)
+{
+    UINT Ret;
+    WCHAR Letter;
+    HANDLE Device;
+    DISK_GEOMETRY DG;
+    DWORD Size, Error;
+    WCHAR Path[MAX_PATH];
+    DISK_GEOMETRY_EX DGE;
+
+    Path[0] = 'C';
+    Path[1] = ':';
+    Path[2] = '\\';
+    Ret = GetSystemDirectoryW(Path, MAX_PATH);
+    ok(Ret > 0, "GetSystemDirectory failed\n");
+
+    Letter = Path[0];
+    ok(Path[1] == ':', "Not a drive letter: %c\n", Path[1]);
+    ok(Path[2] == '\\', "Not a drive letter: %c\n", Path[2]);
+
+    Ret = StringCchPrintfW(Path, MAX_PATH, L"%\\\\?\\c:", Letter);
+    ok(Ret == S_OK, "StringCchPrintfW failed: %d\n", Ret);
+
+    Device = CreateFileW(Path, 0, FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+    ok(Device != INVALID_HANDLE_VALUE, "CreateFileW for %S failed: %ld\n", Path, GetLastError());
+
+    Size = 0;
+    Ret = DeviceIoControl(Device, IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0, &DG, sizeof(DG) - 1, &Size, NULL);
+    ok(Ret == 0, "DeviceIoControl succeed\n");
+    Error = GetLastError();
+    ok(Error == ERROR_INSUFFICIENT_BUFFER, "Expecting ERROR_INSUFFICIENT_BUFFER, got %ld\n", Error);
+    ok(Size == 0, "Invalid output size: %ld\n", Size);
+
+    Size = 0;
+    Ret = DeviceIoControl(Device, IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0, &DG, sizeof(DG), &Size, NULL);
+    ok(Ret != 0, "DeviceIoControl failed: %ld\n", GetLastError());
+    ok(Size == sizeof(DG), "Invalid output size: %ld\n", Size);
+
+    Size = 0;
+    Ret = DeviceIoControl(Device, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, NULL, 0, &DGE, FIELD_OFFSET(DISK_GEOMETRY_EX, Data) - 1, &Size, NULL);
+    ok(Ret == 0, "DeviceIoControl succeed\n");
+    Error = GetLastError();
+    ok(Error == ERROR_INSUFFICIENT_BUFFER, "Expecting ERROR_INSUFFICIENT_BUFFER, got %ld\n", Error);
+    ok(Size == 0, "Invalid output size: %ld\n", Size);
+
+    Size = 0;
+    Ret = DeviceIoControl(Device, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, NULL, 0, &DGE, FIELD_OFFSET(DISK_GEOMETRY_EX, Data), &Size, NULL);
+    ok(Ret != 0, "DeviceIoControl failed: %ld\n", GetLastError());
+    ok(Size == FIELD_OFFSET(DISK_GEOMETRY_EX, Data), "Invalid output size: %ld\n", Size);
+
+    Size = 0;
+    Ret = DeviceIoControl(Device, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, NULL, 0, &DGE, sizeof(DGE), &Size, NULL);
+    ok(Ret != 0, "DeviceIoControl failed: %ld\n", GetLastError());
+    ok(Size == sizeof(DGE), "Invalid output size: %ld\n", Size);
+
+    CloseHandle(Device);
+}
diff --git a/rostests/apitests/kernel32/testlist.c b/rostests/apitests/kernel32/testlist.c
index b78b16f28ca..002aa406350 100644
--- a/rostests/apitests/kernel32/testlist.c
+++ b/rostests/apitests/kernel32/testlist.c
@@ -4,6 +4,7 @@
 #include <apitest.h>
 
 extern void func_DefaultActCtx(void);
+extern void func_DeviceIoControl(void);
 extern void func_dosdev(void);
 extern void func_FindActCtxSectionStringW(void);
 extern void func_FindFiles(void);
@@ -27,6 +28,7 @@ extern void func_WideCharToMultiByte(void);
 const struct test winetest_testlist[] =
 {
     { "DefaultActCtx",               func_DefaultActCtx },
+    { "DeviceIoControl",             func_DeviceIoControl },
     { "dosdev",                      func_dosdev },
     { "FindActCtxSectionStringW",    func_FindActCtxSectionStringW },
     { "FindFiles",                   func_FindFiles },