Added ::from_unicode to convert unicode strings to ansi string in kernel mode.

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>

4 files changed, 34 insertions, 1 deletions

diff --git a/CRT/eastl_compat.cpp b/CRT/eastl_compat.cpp
index 43068b3..cc2df45 100644
--- a/CRT/eastl_compat.cpp
+++ b/CRT/eastl_compat.cpp
@@ -10,6 +10,10 @@
 #define NANOPRINTF_IMPLEMENTATION 1
 #include "nanoprintf.h"
 
+#ifndef NATIVE
+#include <wdm.h>
+#endif
+
 /*
  * eastl::to_string(...) does not work yet event if with a provided Vsnprintf/Vsnprintf8
  * The issue seems to be caused by a broken va_list/va_copy.
@@ -118,3 +122,23 @@ eastl::string to_string(double value)
     else
         return "";
 }
+
+#ifndef NATIVE
+eastl::string from_unicode(wchar_t * wstr, unsigned short wlen, unsigned short wmax)
+{
+    ANSI_STRING ansi;
+    UNICODE_STRING unicode;
+
+    unicode.Buffer = wstr;
+    unicode.Length = wlen;
+    unicode.MaximumLength = (wmax > 0 ? wmax : wlen);
+
+    if (NT_SUCCESS(RtlUnicodeStringToAnsiString(&ansi, &unicode, TRUE))) {
+        eastl::string result(ansi.Buffer, ansi.Length);
+        RtlFreeAnsiString(&ansi);
+        return result;
+    }
+
+    return "";
+}
+#endif
diff --git a/CRT/eastl_compat.hpp b/CRT/eastl_compat.hpp
index 773f7fa..442dc03 100644
--- a/CRT/eastl_compat.hpp
+++ b/CRT/eastl_compat.hpp
@@ -12,4 +12,8 @@ eastl::string to_string(unsigned long long int value);
 eastl::string to_string(float value);
 eastl::string to_string(double value);
 
+#ifndef NATIVE
+eastl::string from_unicode(wchar_t * wstr, unsigned short wlen, unsigned short wmax = 0);
+#endif
+
 #endif
diff --git a/Makefile.native.inc b/Makefile.native.inc
index e782d61..b24425a 100644
--- a/Makefile.native.inc
+++ b/Makefile.native.inc
@@ -22,7 +22,7 @@ AR = /usr/bin/ar
 CFLAGS := -Wall -Wextra -Wno-sign-compare -Wno-strict-aliasing -Wno-c++20-compat \
 	-m64 -fPIC -fvisibility=hidden \
 	-ffunction-sections -fdata-sections \
-	-I$(DPP_ROOT)/CRT
+	-I$(DPP_ROOT)/CRT -DNATIVE=1
 ifneq ($(WERROR),)
 CFLAGS += -Werror
 endif
diff --git a/examples/dpp-example-cplusplus.cpp b/examples/dpp-example-cplusplus.cpp
index 50ff6b0..dd4fa8f 100644
--- a/examples/dpp-example-cplusplus.cpp
+++ b/examples/dpp-example-cplusplus.cpp
@@ -22,6 +22,11 @@ public:
         const auto & eastl_to_string = eastl::to_string(0xDEADC0DE);
         DbgPrint("Using eastl::to_string should return a warning: %s\n", eastl_to_string.c_str());
 
+        wchar_t test_unicode_str[] = L"test_unicode_string";
+        unsigned short test_unicode_strlen = sizeof(test_unicode_str);
+        const auto & eastl_from_unicode = ::from_unicode(test_unicode_str, test_unicode_strlen);
+        DbgPrint("unicode2ansi: \"%s\"\n", eastl_from_unicode.c_str());
+
         const auto & number_ud = ::to_string(1337u);
         DbgPrint("Value 1337u to String: %s\n", number_ud.c_str());
         const auto & number_d = ::to_string(1337);