mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-11-27 12:15:19 +01:00
237 lines
7.7 KiB
C++
Executable File
237 lines
7.7 KiB
C++
Executable File
#include <Platform.h>
|
|
#include "../cpp_foundation/XString.h"
|
|
#include "../cpp_foundation/XStringArray.h"
|
|
#include "../cpp_foundation/XStringW.h"
|
|
#include "global_test.h"
|
|
|
|
|
|
|
|
CHAR16 *Old1_AddLoadOption(IN CONST CHAR16 *LoadOptions, IN CONST CHAR16 *LoadOption)
|
|
{
|
|
// If either option strings are null nothing to do
|
|
if (LoadOptions == NULL)
|
|
{
|
|
if (LoadOption == NULL) return NULL;
|
|
// Duplicate original options as nothing to add
|
|
return EfiStrDuplicate(LoadOption);
|
|
}
|
|
// If there is no option or it is already present duplicate original
|
|
// with XStringW it will be replaced by if (LoadOptions.ExistIn(LoadOption))
|
|
else if ((LoadOption == NULL) || StrStr(LoadOptions, LoadOption))
|
|
return EfiStrDuplicate(LoadOptions);
|
|
// Otherwise add option
|
|
// return PoolPrint(L"%s %s", LoadOptions, LoadOption); //LoadOptions + LoadOption
|
|
return SWPrintf("%ls %ls", LoadOptions, LoadOption).forgetDataWithoutFreeing(); //LoadOptions + LoadOption
|
|
}
|
|
|
|
CHAR16 *Old1_RemoveLoadOption(IN CONST CHAR16 *LoadOptions, IN CONST CHAR16 *LoadOption)
|
|
{
|
|
CONST CHAR16 *Placement;
|
|
CHAR16 *NewLoadOptions;
|
|
UINTN Length, Offset, OptionLength;
|
|
|
|
//DBG("LoadOptions: '%ls', remove LoadOption: '%ls'\n", LoadOptions, LoadOption);
|
|
// If there are no options then nothing to do
|
|
if (LoadOptions == NULL) return NULL;
|
|
// If there is no option to remove then duplicate original
|
|
if (LoadOption == NULL) return EfiStrDuplicate(LoadOptions);
|
|
// If not present duplicate original
|
|
Placement = StrStr(LoadOptions, LoadOption);
|
|
if (Placement == NULL) return EfiStrDuplicate(LoadOptions);
|
|
|
|
// Get placement of option in original options
|
|
Offset = (Placement - LoadOptions);
|
|
Length = StrLen(LoadOptions);
|
|
OptionLength = StrLen(LoadOption);
|
|
|
|
// If this is just part of some larger option (contains non-space at the beginning or end)
|
|
if ((Offset > 0 && LoadOptions[Offset - 1] != L' ') ||
|
|
((Offset + OptionLength) < Length && LoadOptions[Offset + OptionLength] != L' ')) {
|
|
return EfiStrDuplicate(LoadOptions);
|
|
}
|
|
|
|
// Consume preceeding spaces
|
|
while (Offset > 0 && LoadOptions[Offset - 1] == L' ') {
|
|
OptionLength++;
|
|
Offset--;
|
|
}
|
|
|
|
// Consume following spaces
|
|
while (LoadOptions[Offset + OptionLength] == L' ') {
|
|
OptionLength++;
|
|
}
|
|
|
|
// If it's the whole string return NULL
|
|
if (OptionLength == Length) return NULL;
|
|
|
|
if (Offset == 0) {
|
|
// Simple case - we just need substring after OptionLength position
|
|
NewLoadOptions = EfiStrDuplicate(LoadOptions + OptionLength);
|
|
} else {
|
|
// The rest of LoadOptions is Length - OptionLength, but we may need additional space and ending 0
|
|
NewLoadOptions = (__typeof__(NewLoadOptions))AllocateZeroPool((Length - OptionLength + 2) * sizeof(CHAR16));
|
|
// Copy preceeding substring
|
|
CopyMem(NewLoadOptions, (void*)LoadOptions, Offset * sizeof(CHAR16));
|
|
if ((Offset + OptionLength) < Length) {
|
|
// Copy following substring, but include one space also
|
|
OptionLength--;
|
|
CopyMem(NewLoadOptions + Offset, (void*)(LoadOptions + Offset + OptionLength), (Length - OptionLength - Offset) * sizeof(CHAR16));
|
|
}
|
|
}
|
|
return NewLoadOptions;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static XString AddLoadOption(IN CONST XString& LoadOptions, IN CONST XString& LoadOption)
|
|
{
|
|
// If either option strings are null nothing to do
|
|
if (LoadOptions.isEmpty())
|
|
{
|
|
// return LoadOption as nothing to add
|
|
return LoadOption;
|
|
}
|
|
// If there is no option or it is already present duplicate original
|
|
else {
|
|
if ( LoadOptions.ExistIn(LoadOption) ) return LoadOptions;
|
|
// Otherwise add option
|
|
return SPrintf("%s %s", LoadOptions.c_str(), LoadOption.c_str()); //LoadOptions + LoadOption
|
|
}
|
|
}
|
|
|
|
static XString RemoveLoadOption(IN const XString& LoadOptions, IN const XString& LoadOption)
|
|
{
|
|
// CONST CHAR16 *Placement;
|
|
// CHAR16 *NewLoadOptions;
|
|
// UINTN Length, Offset, OptionLength;
|
|
|
|
//DBG("LoadOptions: '%ls', remove LoadOption: '%ls'\n", LoadOptions, LoadOption);
|
|
// If there are no options then nothing to do
|
|
if (LoadOptions.isEmpty()) return ""_XS;
|
|
// If there is no option to remove then duplicate original
|
|
if (LoadOption.isEmpty()) return LoadOptions;
|
|
// If not present duplicate original
|
|
xsize Offset = LoadOptions.IdxOf(LoadOption);
|
|
if ( Offset == MAX_XSIZE ) return LoadOptions;
|
|
|
|
// Get placement of option in original options
|
|
// Offset = (Placement - LoadOptions);
|
|
xsize Length = LoadOptions.length();
|
|
xsize OptionLength = LoadOption.length();
|
|
|
|
// If this is just part of some larger option (contains non-space at the beginning or end)
|
|
if ((Offset > 0 && LoadOptions[Offset - 1] != ' ') ||
|
|
((Offset + OptionLength) < Length && LoadOptions[Offset + OptionLength] != ' ')) {
|
|
return LoadOptions;
|
|
}
|
|
|
|
// Consume preceeding spaces
|
|
while (Offset > 0 && LoadOptions[Offset - 1] == ' ') {
|
|
OptionLength++;
|
|
Offset--;
|
|
}
|
|
|
|
// Consume following spaces
|
|
while (LoadOptions[Offset + OptionLength] == ' ') {
|
|
OptionLength++;
|
|
}
|
|
|
|
// If it's the whole string return NULL
|
|
if (OptionLength == Length) return ""_XS;
|
|
|
|
XString NewLoadOptions;
|
|
if (Offset == 0) {
|
|
// Simple case - we just need substring after OptionLength position
|
|
NewLoadOptions = LoadOptions.SubString(OptionLength, MAX_XSIZE);
|
|
} else {
|
|
// Copy preceeding substring
|
|
NewLoadOptions = LoadOptions.SubString(0, Offset);
|
|
// CopyMem(NewLoadOptions, LoadOptions, Offset * sizeof(CHAR16));
|
|
if ((Offset + OptionLength) < Length) {
|
|
// Copy following substring, but include one space also
|
|
OptionLength--;
|
|
NewLoadOptions += LoadOptions.SubString(Offset + OptionLength, MAX_XSIZE);
|
|
// CopyMem(NewLoadOptions + Offset, LoadOptions + Offset + OptionLength, (Length - OptionLength - Offset) * sizeof(CHAR16));
|
|
}
|
|
}
|
|
return NewLoadOptions;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int BootOptions_tests()
|
|
{
|
|
|
|
#ifdef JIEF_DEBUG
|
|
// DebugLog(2, "XStringW_tests -> Enter\n");
|
|
#endif
|
|
|
|
{
|
|
CHAR16* LoadOptions = NULL;
|
|
|
|
LoadOptions = Old1_AddLoadOption(LoadOptions, L"opt1");
|
|
LoadOptions = Old1_AddLoadOption(LoadOptions, L"opt2");
|
|
LoadOptions = Old1_AddLoadOption(LoadOptions, L"opt3");
|
|
|
|
if ( XString().takeValueFrom(LoadOptions) != "opt1 opt2 opt3"_XS ) return 1;
|
|
|
|
CHAR16* LoadOptions1 = Old1_RemoveLoadOption(LoadOptions, L"opt1");
|
|
if ( XString().takeValueFrom(LoadOptions1) != "opt2 opt3"_XS ) return 1;
|
|
CHAR16* LoadOptions2 = Old1_RemoveLoadOption(LoadOptions, L"opt2");
|
|
if ( XString().takeValueFrom(LoadOptions2) != "opt1 opt3"_XS ) return 1;
|
|
CHAR16* LoadOptions3 = Old1_RemoveLoadOption(LoadOptions, L"opt3");
|
|
if ( XString().takeValueFrom(LoadOptions3) != "opt1 opt2"_XS ) return 1;
|
|
}
|
|
{
|
|
XString LoadOptions;
|
|
|
|
LoadOptions = AddLoadOption(LoadOptions, "opt1"_XS);
|
|
LoadOptions = AddLoadOption(LoadOptions, "opt2"_XS);
|
|
LoadOptions = AddLoadOption(LoadOptions, "opt3"_XS);
|
|
|
|
if ( LoadOptions != "opt1 opt2 opt3"_XS ) return 1;
|
|
|
|
XString LoadOptions1 = RemoveLoadOption(LoadOptions, "opt1"_XS);
|
|
if ( LoadOptions1 != "opt2 opt3"_XS ) return 1;
|
|
XString LoadOptions2 = RemoveLoadOption(LoadOptions, "opt2"_XS);
|
|
if ( LoadOptions2 != "opt1 opt3"_XS ) return 1;
|
|
XString LoadOptions3 = RemoveLoadOption(LoadOptions, "opt3"_XS);
|
|
if ( LoadOptions3 != "opt1 opt2"_XS ) return 1;
|
|
}
|
|
{
|
|
XStringArray LoadOptions;
|
|
|
|
LoadOptions.AddID("opt1"_XS);
|
|
LoadOptions.AddID("opt2"_XS);
|
|
LoadOptions.AddID("opt3"_XS);
|
|
|
|
if ( LoadOptions.ConcatAll(" "_XS) != "opt1 opt2 opt3"_XS ) return 1;
|
|
|
|
XStringArray LoadOptions1 = LoadOptions;
|
|
LoadOptions1.Remove("opt1"_XS);
|
|
if ( LoadOptions1.ConcatAll(" "_XS) != "opt2 opt3"_XS ) return 1;
|
|
XStringArray LoadOptions2 = LoadOptions;
|
|
LoadOptions2.Remove("opt2"_XS);
|
|
if ( LoadOptions2.ConcatAll(" "_XS) != "opt1 opt3"_XS ) return 1;
|
|
XStringArray LoadOptions3 = LoadOptions;
|
|
LoadOptions3.Remove("opt3"_XS);
|
|
if ( LoadOptions3.ConcatAll(" "_XS) != "opt1 opt2"_XS ) return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|