Hans Leidekker : kernel32: Add support for %ws and %wc format specifiers to FormatMessageA/W.

[Alexandre Julliard]

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

Author: Hans Leidekker <hans(a)codeweavers.com>
Date:   Tue May 19 09:20:06 2009 +0200

kernel32: Add support for %ws and %wc format specifiers to FormatMessageA/W.

---

 dlls/kernel32/format_msg.c |   23 +++++++++++++++++++----
 1 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/dlls/kernel32/format_msg.c b/dlls/kernel32/format_msg.c
index c251514..1fca102 100644
--- a/dlls/kernel32/format_msg.c
+++ b/dlls/kernel32/format_msg.c
@@ -257,11 +257,20 @@ DWORD WINAPI FormatMessageA(
                                 argliststart=(*(DWORD**)args)+insertnr-1;
 
                                 /* FIXME: precision and width components are not handled correctly */
-                            if ( (strcmp(fmtstr, "%ls") == 0) || (strcmp(fmtstr,"%S") == 0) ) {
-                                sz = WideCharToMultiByte( CP_ACP, 0, *(WCHAR**)argliststart, -1, NULL, 0, NULL, NULL);
+                            if (strcmp(fmtstr, "%ls") == 0 || strcmp(fmtstr,"%S") == 0 || strcmp(fmtstr,"%ws") == 0)
+                            {
+                                sz = WideCharToMultiByte(CP_ACP, 0, *(WCHAR**)argliststart, -1, NULL, 0, NULL, NULL);
                                 b = HeapAlloc(GetProcessHeap(), 0, sz);
-                                WideCharToMultiByte( CP_ACP, 0, *(WCHAR**)argliststart, -1, b, sz, NULL, NULL);
-                            } else {
+                                WideCharToMultiByte(CP_ACP, 0, *(WCHAR**)argliststart, -1, b, sz, NULL, NULL);
+                            }
+                            else if (strcmp(fmtstr, "%wc") == 0)
+                            {
+                                sz = WideCharToMultiByte(CP_ACP, 0, (WCHAR *)argliststart, 1, NULL, 0, NULL, NULL);
+                                b = HeapAlloc(GetProcessHeap(), 0, sz);
+                                WideCharToMultiByte(CP_ACP, 0, (WCHAR *)argliststart, 1, b, sz, NULL, NULL);
+                            }
+                            else
+                            {
                                 b = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sz = 1000);
                                 /* CMF - This makes a BIG assumption about va_list */
                                 TRACE("A BIG assumption\n");
@@ -367,6 +376,7 @@ DWORD WINAPI FormatMessageW(
     DWORD width = dwFlags & FORMAT_MESSAGE_MAX_WIDTH_MASK;
     BOOL eos = FALSE;
     WCHAR ch;
+    static const WCHAR fmt_wc[] = {'%','w','c',0};
 
     TRACE("(0x%x,%p,%d,0x%x,%p,%d,%p)\n",
           dwFlags,lpSource,dwMessageId,dwLanguageId,lpBuffer,nSize,args);
@@ -483,6 +493,11 @@ DWORD WINAPI FormatMessageW(
 
                             /* CMF - This makes a BIG assumption about va_list */
                             vsprintfW(sprintfbuf, fmtstr, (va_list) xarr);
+                        }
+                        else if (strcmpW(fmtstr, fmt_wc) == 0) {
+                            sprintfbuf = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR) * 2);
+                            sprintfbuf[0] = *(WCHAR *)argliststart;
+                            sprintfbuf[1] = 0;
                         } else {
                             sprintfbuf=HeapAlloc(GetProcessHeap(),0,100);