From aaaac6ffecff2b025a78513560b349c506958ee7 Mon Sep 17 00:00:00 2001
From: Mikhail Paulyshka <me@mixaill.tk>
Date: Mon, 10 Jul 2017 12:59:02 +0200
Subject: [PATCH] msvcp140: Implement _To_byte.

Signed-off-by: Mikhail Paulyshka <me@mixaill.tk>
Signed-off-by: Piotr Caban <piotr@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
---
 dlls/msvcp140/msvcp140.spec    |  2 +-
 dlls/msvcp140/tests/msvcp140.c | 49 ++++++++++++++++++++++++++++++++++
 dlls/msvcp90/locale.c          |  6 +++++
 3 files changed, 56 insertions(+), 1 deletion(-)

diff --git a/dlls/msvcp140/msvcp140.spec b/dlls/msvcp140/msvcp140.spec
index 284f6747367..0216450fe7a 100644
--- a/dlls/msvcp140/msvcp140.spec
+++ b/dlls/msvcp140/msvcp140.spec
@@ -3731,7 +3731,7 @@
 @ cdecl _Thrd_sleep(ptr) _Thrd_sleep
 @ cdecl _Thrd_start(ptr ptr ptr) _Thrd_start
 @ cdecl _Thrd_yield() _Thrd_yield
-@ stub _To_byte
+@ cdecl _To_byte(wstr ptr) _To_byte
 @ cdecl _To_wide(str ptr)
 @ cdecl _Tolower(long ptr) _Tolower
 @ cdecl _Toupper(long ptr) _Toupper
diff --git a/dlls/msvcp140/tests/msvcp140.c b/dlls/msvcp140/tests/msvcp140.c
index f3dd4a0ea01..7cb75f518e2 100644
--- a/dlls/msvcp140/tests/msvcp140.c
+++ b/dlls/msvcp140/tests/msvcp140.c
@@ -152,6 +152,7 @@ static int (__cdecl *p__Schedule_chore)(_Threadpool_chore*);
 static int (__cdecl *p__Reschedule_chore)(const _Threadpool_chore*);
 static void (__cdecl *p__Release_chore)(_Threadpool_chore*);
 
+static int (__cdecl *p_To_byte)(const WCHAR *src, char *dst);
 static int (__cdecl *p_To_wide)(const char *src, WCHAR *dst);
 
 static HMODULE msvcp;
@@ -215,6 +216,7 @@ static BOOL init(void)
         SET(p__Release_chore, "?_Release_chore@details@Concurrency@@YAXPAU_Threadpool_chore@12@@Z");
     }
 
+    SET(p_To_byte, "_To_byte");
     SET(p_To_wide, "_To_wide");
 
     init_thiscall_thunk();
@@ -469,6 +471,52 @@ static void test_chore(void)
     p__Release_chore(&chore);
 }
 
+static void test_to_byte(void)
+{
+    static const WCHAR test_1[] = {'T', 'E', 'S', 'T', 0};
+    static const WCHAR test_2[] = {0x9580, 0x9581, 0x9582, 0x9583, 0}; /* some CJK characters */
+    static const WCHAR *tests[] = {test_1, test_2};
+
+    char dst[MAX_PATH + 4] = "ABC\0XXXXXXX";
+    char compare[MAX_PATH + 4] = "ABC\0XXXXXXX";
+    int ret,expected;
+    unsigned int i, j;
+    WCHAR longstr[MAX_PATH + 3];
+
+    ret = p_To_byte(NULL, NULL);
+    ok(!ret, "Got unexpected result %d\n", ret);
+    ret = p_To_byte(tests[0], NULL);
+    ok(!ret, "Got unexpected result %d\n", ret);
+    ret = p_To_byte(NULL, dst);
+    ok(!ret, "Got unexpected result %d\n", ret);
+
+    ok(!memcmp(dst, compare, sizeof(compare)), "Destination was modified: %s\n", dst);
+
+    for (i = 0; i < sizeof(tests) / sizeof(*tests); ++i)
+    {
+        ret = p_To_byte(tests[i], dst);
+        expected = WideCharToMultiByte(CP_ACP, 0, tests[i], -1, compare, sizeof(compare) / sizeof(*compare),
+                NULL, NULL);
+        ok(ret == expected,  "Got unexpected result %d, expected %d, test case %u\n", ret, expected, i);
+        ok(!memcmp(dst, compare, sizeof(compare)), "Got unexpected output %s, test case %u\n", dst, i);
+    }
+
+    /* Output length is limited to MAX_PATH.*/
+    for (i = MAX_PATH - 2; i < MAX_PATH + 2; ++i)
+    {
+        for (j = 0; j < i; j++)
+            longstr[j] = 'A';
+        longstr[i] = 0;
+        memset(dst, 0xff, sizeof(dst));
+        memset(compare, 0xff, sizeof(compare));
+
+        ret = p_To_byte(longstr, dst);
+        expected = WideCharToMultiByte(CP_ACP, 0, longstr, -1, compare, MAX_PATH, NULL, NULL);
+        ok(ret == expected,  "Got unexpected result %d, expected %d, length %u\n", ret, expected, i);
+        ok(!memcmp(dst, compare, sizeof(compare)), "Got unexpected output %s, length %u\n", dst, i);
+    }
+}
+
 static void test_to_wide(void)
 {
      /* öäü߀Ÿ.A.B in cp1252, the two . are an undefined value and delete.
@@ -524,6 +572,7 @@ START_TEST(msvcp140)
     test__ContextCallback();
     test__TaskEventLogger();
     test_chore();
+    test_to_byte();
     test_to_wide();
     FreeLibrary(msvcp);
 }
diff --git a/dlls/msvcp90/locale.c b/dlls/msvcp90/locale.c
index f80c9b63dcd..70c36772814 100644
--- a/dlls/msvcp90/locale.c
+++ b/dlls/msvcp90/locale.c
@@ -11224,6 +11224,12 @@ size_t __cdecl wcsrtombs(char *dst, const wchar_t **pstr, size_t n, mbstate_t *s
 }
 #endif
 
+int __cdecl _To_byte(const wchar_t *src, char *dst)
+{
+    TRACE("(%s %p)\n", debugstr_w(src), dst);
+    return WideCharToMultiByte(CP_ACP, 0, src, -1, dst, MAX_PATH, NULL, NULL);
+}
+
 int __cdecl _To_wide(const char *src, wchar_t *dst)
 {
     TRACE("(%s %p)\n", debugstr_a(src), dst);