CloverBootloader/rEFIt_UEFI/cpp_unit_test/LoadOptions_test.cpp

#include <Platform.h>
#include "../cpp_foundation/XString.h"
#include "../cpp_foundation/XStringArray.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 = UINTN(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;
}