116 lines
3.2 KiB
C++
116 lines
3.2 KiB
C++
/**
|
|
* 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;
|
|
}
|