1 files changed, 88 insertions, 0 deletions
diff --git a/src/pal/tests/palsuite/locale_info/WideCharToMultiByte/test2/test2.cpp b/src/pal/tests/palsuite/locale_info/WideCharToMultiByte/test2/test2.cpp
new file mode 100644
index 0000000000..f5d40ae903
--- /dev/null
+++ b/src/pal/tests/palsuite/locale_info/WideCharToMultiByte/test2/test2.cpp
@@ -0,0 +1,88 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+/*============================================================================
+**
+** Source: test2.c
+**
+** Purpose: Tests that WideCharToMultiByte respects the length of the wide 
+**          character string.
+**
+**
+**==========================================================================*/
+
+#include <palsuite.h>
+
+
+int __cdecl main(int argc, char *argv[])
+{    
+    char mbStr[128];
+    WCHAR wideStr[128];
+    int ret;
+    int i;
+    int k;
+    BOOL bRet=TRUE;
+
+    /* These codepages are currently supported by the PAL */
+    int codePages[] ={
+        CP_ACP,
+        CP_UTF8
+    };
+
+    if (PAL_Initialize(argc, argv))
+    {
+        return FAIL;
+    }
+
+    /* Go through all of the code pages */
+    for(i=0; i<(sizeof(codePages)/sizeof(int)); i++)
+    {
+
+        /* Filling the arrays */
+        for (k=0; k<128; k++)
+        {
+            wideStr[k] = 'a';
+            mbStr[i] = 0;
+        }
+
+        wideStr[127] = 0;
+
+        /* Passing a buffer that is too small */
+        ret = WideCharToMultiByte(codePages[i], 0, wideStr, 10, 
+                                  mbStr, 0, NULL, NULL);
+        if (ret != 10)
+        {
+            Trace("WideCharToMultiByte did not return correct string length!\n"
+                  "Got %d, expected %d for %d with error %u.\n", ret, 10, 
+                  codePages[i], GetLastError());
+            bRet = FALSE;
+        }
+
+        /* Passing a sufficiently large buffer */
+        mbStr[10] = 'b';
+        ret = WideCharToMultiByte(codePages[i], 0, wideStr, 10, 
+                                  mbStr, 128, NULL, NULL);
+        if (ret != 10)
+        {
+            Trace("WideCharToMultiByte did not return correct string length!\n"
+                  "Got %d, expected %d for code page %d with error %u.\n", 
+                  ret, 10, codePages[i], GetLastError());
+            bRet = FALSE;
+        }
+
+        /* Verifying overflow of the destination string did not occur */
+        if (mbStr[10] != 'b')
+        {
+            Trace("WideCharToMultiByte overflowed the destination buffer for "
+                  "code page %d.\n", codePages[i]);
+            bRet = FALSE;
+        }
+
+    }
+
+    int result = bRet ? PASS : FAIL;
+    PAL_TerminateEx(result);
+    return result;
+}
+