Add basic unicode string support

xtoskrnl/CMakeLists.txt

@@ -106,6 +106,7 @@ list(APPEND XTOSKRNL_SOURCE
     ${XTOSKRNL_SOURCE_DIR}/rtl/sha1.cc
     ${XTOSKRNL_SOURCE_DIR}/rtl/string.cc
     ${XTOSKRNL_SOURCE_DIR}/rtl/time.cc
+    ${XTOSKRNL_SOURCE_DIR}/rtl/unicode.cc
     ${XTOSKRNL_SOURCE_DIR}/rtl/widestr.cc)
 
 # Set module definition SPEC file

xtoskrnl/includes/rtl.hh

@@ -27,6 +27,7 @@
 #include <rtl/slist.hh>
 #include <rtl/string.hh>
 #include <rtl/time.hh>
+#include <rtl/unicode.hh>
 #include <rtl/widestr.hh>
 
 #endif /* __XTOSKRNL_RTL_HH */

xtoskrnl/includes/rtl/unicode.hh

@@ -0,0 +1,29 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/includes/rtl/unicode.hh
+ * DESCRIPTION:     Unicode support
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#ifndef __XTOSKRNL_RTL_UNICODE_HH
+#define __XTOSKRNL_RTL_UNICODE_HH
+
+#include <xtos.hh>
+
+
+/* Runtime Library */
+namespace RTL
+{
+    class Unicode
+    {
+        public:
+            STATIC XTAPI VOID CopyString(IN OUT PUNICODE_STRING Destination,
+                                         IN PCUNICODE_STRING Source);
+            STATIC XTAPI XTSTATUS InitializeString(OUT PUNICODE_STRING Destination,
+                                                   IN PCWSTR Source,
+                                                   IN BOOLEAN Truncate = FALSE);
+    };
+}
+
+#endif /* __XTOSKRNL_RTL_UNICODE_HH */

xtoskrnl/rtl/unicode.cc

@@ -0,0 +1,115 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/rtl/unicode.cc
+ * DESCRIPTION:     Unicode support
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#include <xtos.hh>
+
+
+/**
+ * Copies a source Unicode string to a destination Unicode string.
+ *
+ * @param Destination
+ *        Supplies a pointer to the destination string.
+ *
+ * @param Source
+ *        Supplies an optional pointer to the source string. If this parameter is not provided,
+ *        the destination string is effectively emptied.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+RTL::Unicode::CopyString(IN OUT PUNICODE_STRING Destination,
+                         IN PCUNICODE_STRING Source)
+{
+    ULONG Length;
+
+    /* Check if the source string pointer is valid */
+    if(!Source)
+    {
+        /* Source is missing, clear the destination string and return */
+        Destination->Length = 0;
+        return;
+    }
+
+    /* Calculate the length to copy */
+    Length = MIN(Source->Length, Destination->MaximumLength);
+    Destination->Length = Length;
+
+    /* Copy the source string data into the destination buffer */
+    RTL::Memory::CopyMemory(Destination->Buffer, Source->Buffer, Length);
+
+    /* Check if there is enough space left in the buffer to NULL-terminate */
+    if(Destination->Length < Destination->MaximumLength)
+    {
+        /* Append a NULL terminator */
+        Destination->Buffer[Length / sizeof(WCHAR)] = (WCHAR)0;
+    }
+}
+
+/**
+ * Initializes a Unicode string.
+ *
+ * @param Destination
+ *        Supplies a pointer to the UNICODE_STRING structure to be initialized.
+ *
+ * @param Source
+ *        Supplies an optional pointer to a NULL-terminated wide character string.
+ *
+ * @param Truncate
+ *        Specifies whether to truncate the source string to the maximum allowed length.
+ *
+ * @return This routine returns a status code indicating the success or failure of the operation.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+RTL::Unicode::InitializeString(OUT PUNICODE_STRING Destination,
+                               IN PCWSTR Source,
+                               IN BOOLEAN Truncate)
+{
+    ULONG Size;
+
+    /* Check if the source string pointer is valid */
+    if(!Source)
+    {
+        /* Source is missing, initialize the destination as empty string */
+        Destination->Length = 0;
+        Destination->MaximumLength = 0;
+
+        /* Return success */
+        return STATUS_SUCCESS;
+    }
+
+    /* Calculate the size of the string in bytes */
+    Size = RTL::WideString::WideStringLength(Source, 0) * sizeof(WCHAR);
+
+    /* Check if the size exceeds the maximum allowed */
+    if(Size > ((MAXUSHORT & ~1) - sizeof(WCHAR)))
+    {
+        /* Check if truncation is not explicitly forced */
+        if(!Truncate)
+        {
+            /* Return error code */
+            return STATUS_NAME_TOO_LONG;
+        }
+
+        /* Force truncation to the maximum safe length */
+        Size = (MAXUSHORT & ~1) - sizeof(WCHAR);
+    }
+
+    /* Initialize the destination string fields */
+    Destination->Buffer = (PWCHAR)Source;
+    Destination->Length = (USHORT)Size;
+    Destination->MaximumLength = (USHORT)Size + sizeof(WCHAR);
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}

xtoskrnl/xtoskrnl.spec

@@ -65,6 +65,7 @@
 @ stdcall RtlConvertToLargeIntegerUnsigned32(long)
 @ stdcall RtlCopyMemory(ptr ptr long)
 @ stdcall RtlCopyString(ptr ptr long)
+@ stdcall RtlCopyUnicodeString(ptr ptr)
 @ stdcall RtlCopyWideString(ptr ptr long)
 @ stdcall RtlDivideLargeInteger(long long long ptr)
 @ stdcall RtlFindClearBits(ptr long long)
@@ -75,6 +76,8 @@
 @ stdcall RtlFindWideStringInsensitive(wstr wstr)
 @ stdcall RtlInitializeBitMap(ptr ptr long)
 @ stdcall RtlInitializeListHead(ptr)
+@ stdcall RtlInitializeUnicodeString(ptr wstr)
+@ stdcall RtlInitializeUnicodeStringEx(ptr wstr)
 @ stdcall RtlInsertHeadList(ptr ptr)
 @ stdcall RtlInsertTailList(ptr ptr)
 @ stdcall RtlListEmpty(ptr)