Alex Henrie : ntdll/tests: Add SystemFirmwareTableInformation tests.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 0bcee5eaf82d6fdf0a80e6d2d6fb9dc1f1c1a1ff
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=0bcee5eaf82d6fdf0a80e6d2d6fb9dc1f1c1a1ff

Author: Alex Henrie <alexhenrie24 at gmail.com>
Date:   Thu Jul 12 00:57:49 2018 +0200

ntdll/tests: Add SystemFirmwareTableInformation tests.

Signed-off-by: Alex Henrie <alexhenrie24 at gmail.com>
Signed-off-by: Huw Davies <huw at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/ntdll/tests/info.c | 60 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/dlls/ntdll/tests/info.c b/dlls/ntdll/tests/info.c
index 7d60b94..3af8fdd 100644
--- a/dlls/ntdll/tests/info.c
+++ b/dlls/ntdll/tests/info.c
@@ -54,6 +54,11 @@ static DWORD one_before_last_pid = 0;
     } \
   } while(0)
 
+/* Firmware table providers */
+#define ACPI 0x41435049
+#define FIRM 0x4649524D
+#define RSMB 0x52534D42
+
 static BOOL InitFunctionPtrs(void)
 {
     /* All needed functions are NT based, so using GetModuleHandle is a good check */
@@ -826,6 +831,57 @@ static void test_query_logicalprocex(void)
     }
 }
 
+static void test_query_firmware(void)
+{
+    static const ULONG min_sfti_len = FIELD_OFFSET(SYSTEM_FIRMWARE_TABLE_INFORMATION, TableBuffer);
+    ULONG len1, len2;
+    NTSTATUS status;
+    SYSTEM_FIRMWARE_TABLE_INFORMATION *sfti;
+
+    sfti = HeapAlloc(GetProcessHeap(), 0, min_sfti_len);
+    ok(!!sfti, "Failed to allocate memory\n");
+
+    sfti->ProviderSignature = 0;
+    sfti->Action = 0;
+    sfti->TableID = 0;
+
+    status = pNtQuerySystemInformation(SystemFirmwareTableInformation, sfti, min_sfti_len - 1, &len1);
+todo_wine
+    ok(status == STATUS_INFO_LENGTH_MISMATCH || broken(status == STATUS_INVALID_INFO_CLASS) /* xp */,
+       "Expected STATUS_INFO_LENGTH_MISMATCH, got %08x\n", status);
+    if (len1 == 0) /* xp, 2003 */
+    {
+        skip("SystemFirmwareTableInformation is not available\n");
+        HeapFree(GetProcessHeap(), 0, sfti);
+        return;
+    }
+    ok(len1 == min_sfti_len, "Expected length %u, got %u\n", min_sfti_len, len1);
+
+    status = pNtQuerySystemInformation(SystemFirmwareTableInformation, sfti, min_sfti_len, &len1);
+    ok(status == STATUS_NOT_IMPLEMENTED, "Expected STATUS_NOT_IMPLEMENTED, got %08x\n", status);
+    ok(len1 == 0, "Expected length 0, got %u\n", len1);
+
+    sfti->ProviderSignature = RSMB;
+    sfti->Action = SystemFirmwareTable_Get;
+
+    status = pNtQuerySystemInformation(SystemFirmwareTableInformation, sfti, min_sfti_len, &len1);
+    ok(status == STATUS_BUFFER_TOO_SMALL, "Expected STATUS_BUFFER_TOO_SMALL, got %08x\n", status);
+    ok(len1 >= min_sfti_len, "Expected length >= %u, got %u\n", min_sfti_len, len1);
+    ok(sfti->TableBufferLength == len1 - min_sfti_len,
+       "Expected length %u, got %u\n", len1 - min_sfti_len, sfti->TableBufferLength);
+
+    sfti = HeapReAlloc(GetProcessHeap(), 0, sfti, len1);
+    ok(!!sfti, "Failed to allocate memory\n");
+
+    status = pNtQuerySystemInformation(SystemFirmwareTableInformation, sfti, len1, &len2);
+    ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %08x\n", status);
+    ok(len2 == len1, "Expected length %u, got %u\n", len1, len2);
+    ok(sfti->TableBufferLength == len1 - min_sfti_len,
+       "Expected length %u, got %u\n", len1 - min_sfti_len, sfti->TableBufferLength);
+
+    HeapFree(GetProcessHeap(), 0, sfti);
+}
+
 static void test_query_processor_power_info(void)
 {
     NTSTATUS status;
@@ -2277,6 +2333,10 @@ START_TEST(info)
     trace("Starting test_process_debug_flags()\n");
     test_query_process_debug_flags(argc, argv);
 
+    /* 0x4C SystemFirmwareTableInformation */
+    trace("Starting test_query_firmware()\n");
+    test_query_firmware();
+
     /* belongs to its own file */
     trace("Starting test_readvirtualmemory()\n");
     test_readvirtualmemory();