Daniel Lehman : kernel32/tests: Test if FormatMessage modifies va_list on error.

[Alexandre Julliard]

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

Author: Daniel Lehman <dlehman25(a)gmail.com>
Date:   Sat Jun  4 08:03:02 2022 -0700

kernel32/tests: Test if FormatMessage modifies va_list on error.

Signed-off-by: Daniel Lehman <dlehman25(a)gmail.com>

---

 dlls/kernel32/tests/format_msg.c | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/dlls/kernel32/tests/format_msg.c b/dlls/kernel32/tests/format_msg.c
index 2d317d1bb9e..e2a4f05ffdb 100644
--- a/dlls/kernel32/tests/format_msg.c
+++ b/dlls/kernel32/tests/format_msg.c
@@ -1005,6 +1005,25 @@ static void test_message_wrap(void)
     ok(!strcmp("short\r\nlong?line", out),"failed out=[%s]\n",out);
 }
 
+static void test_message_arg_eaten( const WCHAR *src, ... )
+{
+    DWORD ret;
+    va_list list;
+    WCHAR *arg, out[1];
+
+    out[0] = 0xcccc;
+    va_start(list, src);
+    SetLastError(0xdeadbeef);
+    ret = FormatMessageW(FORMAT_MESSAGE_FROM_STRING, src, 0, 0, out, ARRAY_SIZE(out), &list);
+    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
+       "Expected GetLastError() to return ERROR_INSUFFICIENT_BUFFER, got %lu\n", GetLastError());
+    ok(ret == 0, "Expected FormatMessageW to return 0, got %lu\n", ret);
+    ok(out[0] == 0, "Expected null, got %ls\n", out);
+    arg = va_arg( list, WCHAR * );
+    ok(!wcscmp( L"unused", arg ), "Expected 'unused', got %s\n", wine_dbgstr_w(arg));
+    va_end(list);
+}
+
 static void test_message_insufficient_buffer(void)
 {
     static const char init_buf[] = {'x', 'x', 'x', 'x', 'x'};
@@ -1153,6 +1172,9 @@ static void test_message_insufficient_buffer_wide(void)
         HeapFree( GetProcessHeap(), 0, tmp );
         HeapFree( GetProcessHeap(), 0, buf );
     }
+
+    /* va_arg is eaten even in case of insufficient buffer */
+    test_message_arg_eaten( L"%1!s! %2!s!", L"eaten", L"unused" );
 }
 
 static void test_message_null_buffer(void)