Improvements in string and wide string support
Some checks failed
Builds / ExectOS (amd64) (push) Failing after 19s
Builds / ExectOS (i686) (push) Failing after 15s

* Implement RtlCompareString(), RtlCompareStringInsensitive(), RtlCompareWideString() and RtlCompareWideStringInsensitive()
* Rename some routines t omatch naming conventions
* Switch to CHAR in string operations
This commit is contained in:
Rafal Kupiec 2023-12-06 22:47:54 +01:00
parent 6733146b71
commit 1c94f9ff02
Signed by: belliash
GPG Key ID: 4E829243E0CFE6B4
7 changed files with 294 additions and 87 deletions

View File

@ -54,6 +54,36 @@ RtlCompareMemory(IN PCVOID LeftBuffer,
IN PCVOID RightBuffer, IN PCVOID RightBuffer,
IN SIZE_T Length); IN SIZE_T Length);
XTCDECL
SIZE_T
RtlCompareString(IN CONST PCHAR String1,
IN CONST PCHAR String2,
IN SIZE_T Length);
XTCDECL
SIZE_T
RtlCompareStringInsensitive(IN CONST PCHAR String1,
IN CONST PCHAR String2,
IN SIZE_T Length);
XTCDECL
SIZE_T
RtlCompareWideString(IN CONST PWCHAR String1,
IN CONST PWCHAR String2,
IN SIZE_T Length);
XTCDECL
SIZE_T
RtlCompareWideStringInsensitive(IN CONST PWCHAR String1,
IN CONST PWCHAR String2,
IN SIZE_T Length);
XTCDECL
PWCHAR
RtlConcatenateWideString(OUT PWCHAR Destination,
IN PWCHAR Source,
IN SIZE_T Count);
XTAPI XTAPI
VOID VOID
RtlCopyMemory(OUT PVOID Destination, RtlCopyMemory(OUT PVOID Destination,
@ -86,38 +116,26 @@ RtlSetMemory(OUT PVOID Destination,
XTCDECL XTCDECL
SIZE_T SIZE_T
RtlStringLength(IN CONST PUCHAR String, RtlStringLength(IN CONST PCHAR String,
IN SIZE_T MaxLength); IN SIZE_T MaxLength);
XTCDECL XTCDECL
INT SIZE_T
RtlStringToWideString(OUT PWCHAR Destination, RtlStringToWideString(OUT PWCHAR Destination,
IN CONST PUCHAR *Source, IN CONST PCHAR *Source,
IN SIZE_T Length); IN SIZE_T Length);
XTCDECL
INT
RtlWideStringCompare(IN CONST PWCHAR String1,
IN CONST PWCHAR String2,
IN CONST SIZE_T Length);
XTCDECL XTCDECL
PWCHAR PWCHAR
RtlWideStringConcatenate(OUT PWCHAR Destination, RtlTokenizeWideString(IN PWCHAR String,
IN PWCHAR Source, IN CONST PWCHAR Delimiter,
IN SIZE_T Count); IN OUT PWCHAR *SavePtr);
XTCDECL XTCDECL
SIZE_T SIZE_T
RtlWideStringLength(IN CONST PWCHAR String, RtlWideStringLength(IN CONST PWCHAR String,
IN SIZE_T MaxLength); IN SIZE_T MaxLength);
XTCDECL
PWCHAR
RtlWideStringTokenize(IN PWCHAR String,
IN CONST PWCHAR Delimiter,
IN OUT PWCHAR *SavePtr);
XTAPI XTAPI
VOID VOID
RtlWriteRegisterByte(IN VOLATILE PVOID Register, RtlWriteRegisterByte(IN VOLATILE PVOID Register,

View File

@ -146,19 +146,19 @@ BlComPortInitialize()
CommandLine = (PWCHAR)LoadedImage->LoadOptions; CommandLine = (PWCHAR)LoadedImage->LoadOptions;
/* Find command in command line */ /* Find command in command line */
Argument = RtlWideStringTokenize(CommandLine, L" ", &LastArg); Argument = RtlTokenizeWideString(CommandLine, L" ", &LastArg);
/* Iterate over all arguments passed to boot loader */ /* Iterate over all arguments passed to boot loader */
while(Argument != NULL) while(Argument != NULL)
{ {
/* Check if this is DEBUG parameter */ /* Check if this is DEBUG parameter */
if(RtlWideStringCompare(Argument, L"DEBUG=", 6) == 0) if(RtlCompareWideString(Argument, L"DEBUG=", 6) == 0)
{ {
/* Skip to the argument value */ /* Skip to the argument value */
Argument += 6; Argument += 6;
/* Make sure COM port is being used */ /* Make sure COM port is being used */
if(RtlWideStringCompare(Argument, L"COM", 3)) if(RtlCompareWideString(Argument, L"COM", 3))
{ {
/* Invalid debug port specified */ /* Invalid debug port specified */
BlEfiPrint(L"ERROR: Invalid debug port specified, falling back to defaults\n"); BlEfiPrint(L"ERROR: Invalid debug port specified, falling back to defaults\n");
@ -176,7 +176,7 @@ BlComPortInitialize()
} }
/* Check if custom COM port address supplied */ /* Check if custom COM port address supplied */
if(PortNumber == 0 && RtlWideStringCompare(Argument, L":0x", 3) == 0) if(PortNumber == 0 && RtlCompareWideString(Argument, L":0x", 3) == 0)
{ {
/* COM port address provided */ /* COM port address provided */
Argument += 3; Argument += 3;
@ -221,7 +221,7 @@ BlComPortInitialize()
} }
/* Take next argument */ /* Take next argument */
Argument = RtlWideStringTokenize(NULL, L" ", &LastArg); Argument = RtlTokenizeWideString(NULL, L" ", &LastArg);
} }
} }
} }
@ -349,7 +349,7 @@ BlEfiPrint(IN PUINT16 Format,
BlStringPrint(BlConsolePutChar, Format, Arguments); BlStringPrint(BlConsolePutChar, Format, Arguments);
/* Print to serial console only if not running under OVMF */ /* Print to serial console only if not running under OVMF */
if(RtlWideStringCompare(EfiSystemTable->FirmwareVendor, L"EDK II", 6) != 0) if(RtlCompareWideString(EfiSystemTable->FirmwareVendor, L"EDK II", 6) != 0)
{ {
/* Check if debugging enabled and if EFI serial port is fully initialized */ /* Check if debugging enabled and if EFI serial port is fully initialized */
if(DEBUG && (EfiSerialPort.Flags & COMPORT_FLAG_INIT)) if(DEBUG && (EfiSerialPort.Flags & COMPORT_FLAG_INIT))

View File

@ -132,7 +132,7 @@ XtBootSystem(IN PXT_BOOT_PROTOCOL_PARAMETERS Parameters)
} }
/* System path has to point to the boot directory */ /* System path has to point to the boot directory */
RtlWideStringConcatenate(Parameters->SystemPath, L"\\Boot", 0); RtlConcatenateWideString(Parameters->SystemPath, L"\\Boot", 0);
/* Open XTOS system boot directory */ /* Open XTOS system boot directory */
Status = FsHandle->Open(FsHandle, &BootDir, Parameters->SystemPath, EFI_FILE_MODE_READ, 0); Status = FsHandle->Open(FsHandle, &BootDir, Parameters->SystemPath, EFI_FILE_MODE_READ, 0);

View File

@ -116,7 +116,7 @@ BlLoadEfiModules()
/* Set correct path to the module file */ /* Set correct path to the module file */
RtlCopyMemory(ModulePath, ModulesDirPath, sizeof(ModulePath) / sizeof(WCHAR)); RtlCopyMemory(ModulePath, ModulesDirPath, sizeof(ModulePath) / sizeof(WCHAR));
RtlWideStringConcatenate(ModulePath, ModuleName, 0); RtlConcatenateWideString(ModulePath, ModuleName, 0);
/* Find valid device path */ /* Find valid device path */
Status = BlFindVolumeDevicePath(DevicePath, ModulePath, &VolumeDevicePath); Status = BlFindVolumeDevicePath(DevicePath, ModulePath, &VolumeDevicePath);

View File

@ -9,6 +9,125 @@
#include <xtos.h> #include <xtos.h>
/**
* Compares at most specified number of characters of two C strings.
*
* @param String1
* String to be compared.
*
* @param String2
* String to be compared.
*
* @param Length
* Maximum number of characters to compare. If no limit set, it compares whole strings.
*
* @return Integral value indicating the relationship between the strings.
*
* @since XT 1.0
*/
XTCDECL
SIZE_T
RtlCompareString(IN CONST PCHAR String1,
IN CONST PCHAR String2,
IN SIZE_T Length)
{
SIZE_T Index;
/* Iterate through the strings */
for(Index = 0; ; Index++) {
/* Check if length limit reached */
if(Index != 0 && Index == Length)
{
/* Skip checking next characters */
break;
}
/* Check if string characters are equal */
if(String1[Index] != String2[Index])
{
/* Different characters found */
return String1[Index] < String2[Index] ? -1 : 1;
}
/* Check if end of string reached */
if(!String1[Index] || !String2[Index])
{
/* Equal strings until the end of one of them */
return 0;
}
}
/* Strings are equal */
return 0;
}
/**
* Compares at most specified number of characters of two C strings, while ignoring differences in case.
*
* @param String1
* String to be compared.
*
* @param String2
* String to be compared.
*
* @param Length
* Maximum number of characters to compare. If no limit set, it compares whole strings.
*
* @return Integral value indicating the relationship between the strings.
*
* @since XT 1.0
*/
XTCDECL
SIZE_T
RtlCompareStringInsensitive(IN CONST PCHAR String1,
IN CONST PCHAR String2,
IN SIZE_T Length)
{
CHAR Character1;
CHAR Character2;
ULONG Index = 0;
/* Iterate through the strings */
while(String1[Index] != '\0' && String2[Index] != '\0')
{
/* Check if length limit reached */
if(Index != 0 && Index == Length)
{
/* Skip checking next characters */
break;
}
/* Get the characters */
Character1 = String1[Index];
Character2 = String2[Index];
/* Lowercase string1 character if needed */
if(String1[Index] >= 'A' && String1[Index] <= 'Z')
{
Character1 = String1[Index] - 'A' + 'a';
}
/* Lowercase string2 character if needed */
if(String2[Index] >= 'A' && String2[Index] <= 'Z')
{
Character2 = String2[Index] - 'A' + 'a';
}
/* Compare the characters */
if(Character1 != Character2)
{
/* Strings are not equal */
return Character1 > Character2 ? 1 : -1;
}
/* Get next character */
Index++;
}
/* Strings are equal */
return 0;
}
/** /**
* Calculates the length of a given string. * Calculates the length of a given string.
* *
@ -24,7 +143,7 @@
*/ */
XTCDECL XTCDECL
SIZE_T SIZE_T
RtlStringLength(IN CONST PUCHAR String, RtlStringLength(IN CONST PCHAR String,
IN SIZE_T MaxLength) IN SIZE_T MaxLength)
{ {
SIZE_T Length; SIZE_T Length;
@ -47,7 +166,7 @@ RtlStringLength(IN CONST PUCHAR String,
} }
} }
/* Return wide string length */ /* Return string length */
return Length; return Length;
} }
@ -68,12 +187,12 @@ RtlStringLength(IN CONST PUCHAR String,
* @since XT 1.0 * @since XT 1.0
*/ */
XTCDECL XTCDECL
INT SIZE_T
RtlStringToWideString(OUT PWCHAR Destination, RtlStringToWideString(OUT PWCHAR Destination,
IN CONST PUCHAR *Source, IN CONST PCHAR *Source,
IN SIZE_T Length) IN SIZE_T Length)
{ {
PUCHAR LocalSource = *Source; PCHAR LocalSource = *Source;
SIZE_T Count = Length; SIZE_T Count = Length;
if(Destination == NULL) if(Destination == NULL)
@ -83,7 +202,7 @@ RtlStringToWideString(OUT PWCHAR Destination,
while(Count) while(Count)
{ {
if((*Destination = (UCHAR)*LocalSource) == 0) if((*Destination = *LocalSource) == 0)
{ {
LocalSource = NULL; LocalSource = NULL;
break; break;

View File

@ -19,17 +19,17 @@
* Wide string to be compared. * Wide string to be compared.
* *
* @param Length * @param Length
* Maximum number of characters to compare. If no limit set, it compares whole strings. * Maximum number of characters to compare. If no limit set, it compares whole wide strings.
* *
* @return Integral value indicating the relationship between the wide strings. * @return Integral value indicating the relationship between the wide strings.
* *
* @since XT 1.0 * @since XT 1.0
*/ */
XTCDECL XTCDECL
INT SIZE_T
RtlWideStringCompare(IN CONST PWCHAR String1, RtlCompareWideString(IN CONST PWCHAR String1,
IN CONST PWCHAR String2, IN CONST PWCHAR String2,
IN CONST SIZE_T Length) IN SIZE_T Length)
{ {
SIZE_T Index; SIZE_T Index;
@ -42,22 +42,89 @@ RtlWideStringCompare(IN CONST PWCHAR String1,
break; break;
} }
/* Check if string characters are equal */ /* Check if wide string characters are equal */
if(String1[Index] != String2[Index]) if(String1[Index] != String2[Index])
{ {
/* Different characters found */ /* Different characters found */
return String1[Index] < String2[Index] ? -1 : 1; return String1[Index] < String2[Index] ? -1 : 1;
} }
/* Check if end of string reached */ /* Check if end of wide string reached */
if(!String1[Index] || !String2[Index]) if(!String1[Index] || !String2[Index])
{ {
/* Equal strings until the end of one of them */ /* Equal wide strings until the end of one of them */
return 0; return 0;
} }
} }
/* Strings are equal */ /* Wide strings are equal */
return 0;
}
/**
* Compares at most specified number of characters of two C wide strings, while ignoring differences in case.
*
* @param String1
* Wide string to be compared.
*
* @param String2
* Wide string to be compared.
*
* @param Length
* Maximum number of characters to compare. If no limit set, it compares whole wide strings.
*
* @return Integral value indicating the relationship between the wide strings.
*
* @since XT 1.0
*/
XTCDECL
SIZE_T
RtlCompareWideStringInsensitive(IN CONST PWCHAR String1,
IN CONST PWCHAR String2,
IN SIZE_T Length)
{
WCHAR Character1;
WCHAR Character2;
ULONG Index = 0;
/* Iterate through the wide strings */
while(String1[Index] != '\0' && String2[Index] != '\0')
{
/* Check if length limit reached */
if(Index != 0 && Index == Length)
{
/* Skip checking next characters */
break;
}
/* Get the characters */
Character1 = String1[Index];
Character2 = String2[Index];
/* Lowercase wide string1 character if needed */
if(String1[Index] >= 'A' && String1[Index] <= 'Z')
{
Character1 = String1[Index] - 'A' + 'a';
}
/* Lowercase wide string2 character if needed */
if(String2[Index] >= 'A' && String2[Index] <= 'Z')
{
Character2 = String2[Index] - 'A' + 'a';
}
/* Compare the characters */
if(Character1 != Character2)
{
/* Wide strings are not equal */
return Character1 > Character2 ? 1 : -1;
}
/* Get next character */
Index++;
}
/* Wide strings are equal */
return 0; return 0;
} }
@ -79,7 +146,7 @@ RtlWideStringCompare(IN CONST PWCHAR String1,
*/ */
XTCDECL XTCDECL
PWCHAR PWCHAR
RtlWideStringConcatenate(OUT PWCHAR Destination, RtlConcatenateWideString(OUT PWCHAR Destination,
IN PWCHAR Source, IN PWCHAR Source,
IN SIZE_T Count) IN SIZE_T Count)
{ {
@ -120,48 +187,6 @@ RtlWideStringConcatenate(OUT PWCHAR Destination,
return DestString; return DestString;
} }
/**
* Calculates the length of a given wide string.
*
* @param String
* Pointer to the null-terminated wide string to be examined.
*
* @param MaxLength
* Maximum number of wide characters to examine. If no limit set, it examines whole string.
*
* @return The length of the null-terminated wide string.
*
* @since: XT 1.0
*/
XTCDECL
SIZE_T
RtlWideStringLength(IN CONST PWCHAR String,
IN SIZE_T MaxLength)
{
SIZE_T Length;
/* Check if NULL pointer passed */
if(String == NULL)
{
return 0;
}
/* Iterate through the wide string */
for(Length = 0; ; Length++)
{
/* Check if NULL found or max length limit reached */
if((Length != 0 && Length == MaxLength) || !String[Length])
{
/* Finish examination */
break;
}
}
/* Return wide string length */
return Length;
}
/** /**
* Finds the next token in a null-terminated wide string. * Finds the next token in a null-terminated wide string.
* *
@ -180,7 +205,7 @@ RtlWideStringLength(IN CONST PWCHAR String,
*/ */
XTCDECL XTCDECL
PWCHAR PWCHAR
RtlWideStringTokenize(IN PWCHAR String, RtlTokenizeWideString(IN PWCHAR String,
IN CONST PWCHAR Delimiter, IN CONST PWCHAR Delimiter,
IN OUT PWCHAR *SavePtr) IN OUT PWCHAR *SavePtr)
{ {
@ -231,3 +256,45 @@ RtlWideStringTokenize(IN PWCHAR String,
while(SpanChar != L'\0'); while(SpanChar != L'\0');
} }
} }
/**
* Calculates the length of a given wide string.
*
* @param String
* Pointer to the null-terminated wide string to be examined.
*
* @param MaxLength
* Maximum number of wide characters to examine. If no limit set, it examines whole string.
*
* @return The length of the null-terminated wide string.
*
* @since: XT 1.0
*/
XTCDECL
SIZE_T
RtlWideStringLength(IN CONST PWCHAR String,
IN SIZE_T MaxLength)
{
SIZE_T Length;
/* Check if NULL pointer passed */
if(String == NULL)
{
return 0;
}
/* Iterate through the wide string */
for(Length = 0; ; Length++)
{
/* Check if NULL found or max length limit reached */
if((Length != 0 && Length == MaxLength) || !String[Length])
{
/* Finish examination */
break;
}
}
/* Return wide string length */
return Length;
}

View File

@ -32,12 +32,15 @@
@ stdcall RtlReadRegisterByte(ptr) @ stdcall RtlReadRegisterByte(ptr)
@ stdcall RtlReadRegisterLong(ptr) @ stdcall RtlReadRegisterLong(ptr)
@ stdcall RtlReadRegisterShort(ptr) @ stdcall RtlReadRegisterShort(ptr)
@ cdecl RtlCompareString(str str long)
@ cdecl RtlCompareStringInsensitive(str str long)
@ cdecl RtlStringLength(str long) @ cdecl RtlStringLength(str long)
@ cdecl RtlStringToWideString(wstr str long) @ cdecl RtlStringToWideString(wstr str long)
@ cdecl RtlWideStringCompare(wstr wstr long) @ cdecl RtlCompareWideString(wstr wstr long)
@ cdecl RtlWideStringConcatenate(wstr wstr long) @ cdecl RtlCompareWideStringInsensitive(wstr wstr long)
@ cdecl RtlConcatenateWideString(wstr wstr long)
@ cdecl RtlTokenizeWideString(wstr wstr wstr)
@ cdecl RtlWideStringLength(wstr long) @ cdecl RtlWideStringLength(wstr long)
@ cdecl RtlWideStringTokenize(wstr wstr wstr)
@ stdcall RtlWriteRegisterByte(ptr long) @ stdcall RtlWriteRegisterByte(ptr long)
@ stdcall RtlWriteRegisterLong(ptr long) @ stdcall RtlWriteRegisterLong(ptr long)
@ stdcall RtlWriteRegisterShort(ptr long) @ stdcall RtlWriteRegisterShort(ptr long)