BaseTools: Fix corner-cases of --hash feature

[mirror_edk2.git] / Vlv2TbltDevicePkg / SmBiosMiscDxe / MiscSystemOptionStringFunction.c

/*++\r
\r
Copyright (c) 2009 - 2014, Intel Corporation. All rights reserved.<BR>\r
                                                                                   \r\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
                                                                                   \r\r
\r
\r
Module Name:\r
\r
  MiscSystemOptionStringFunction.c\r
\r
Abstract:\r
\r
  BIOS system option string boot time changes.\r
  SMBIOS type 12.\r
\r
--*/\r
\r
\r
#include "CommonHeader.h"\r
\r
#include "MiscSubclassDriver.h"\r
\r
\r
/**\r
  This function makes boot time changes to the contents of the\r
  MiscSystemOptionString (Type 12).\r
\r
  @param  RecordData                 Pointer to copy of RecordData from the Data Table.\r
\r
  @retval EFI_SUCCESS                All parameters were valid.\r
  @retval EFI_UNSUPPORTED            Unexpected RecordType value.\r
  @retval EFI_INVALID_PARAMETER      Invalid parameter was found.\r
\r
**/\r
MISC_SMBIOS_TABLE_FUNCTION(SystemOptionString)\r
{\r
  CHAR8                             *OptionalStrStart;\r
  UINTN                             OptStrLen;\r
  EFI_STRING                        OptionString;\r
  EFI_STATUS                        Status;\r
  STRING_REF                        TokenToGet;\r
  EFI_SMBIOS_HANDLE                 SmbiosHandle;\r
  SMBIOS_TABLE_TYPE12               *SmbiosRecord;\r
\r
  //\r
  // First check for invalid parameters.\r
  //\r
  if (RecordData == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  TokenToGet = STRING_TOKEN (STR_MISC_SYSTEM_OPTION_EN_US);\r
  OptionString = SmbiosMiscGetString (TokenToGet);\r
  OptStrLen = StrLen(OptionString);\r
  if (OptStrLen > SMBIOS_STRING_MAX_LENGTH) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  //\r
  // Two zeros following the last string.\r
  //\r
  SmbiosRecord = AllocatePool(sizeof (SMBIOS_TABLE_TYPE12) + OptStrLen + 1 + 1);\r
  ZeroMem(SmbiosRecord, sizeof (SMBIOS_TABLE_TYPE12) + OptStrLen + 1 + 1);\r
\r
  SmbiosRecord->Hdr.Type = EFI_SMBIOS_TYPE_SYSTEM_CONFIGURATION_OPTIONS;\r
  SmbiosRecord->Hdr.Length = sizeof (SMBIOS_TABLE_TYPE12);\r
\r
  //\r
  // Make handle chosen by smbios protocol.add automatically.\r
  //\r
  SmbiosRecord->Hdr.Handle = 0;\r
\r
  SmbiosRecord->StringCount = 1;\r
  OptionalStrStart = (CHAR8*) (SmbiosRecord + 1);\r
  UnicodeStrToAsciiStr(OptionString, OptionalStrStart);\r
\r
  //\r
  // Now we have got the full smbios record, call smbios protocol to add this record.\r
  //\r
  SmbiosHandle = SMBIOS_HANDLE_PI_RESERVED;\r
  Status = Smbios-> Add(\r
                      Smbios,\r
                      NULL,\r
                      &SmbiosHandle,\r
                      (EFI_SMBIOS_TABLE_HEADER *) SmbiosRecord\r
                      );\r
\r
  FreePool(SmbiosRecord);\r
  return Status;\r
}\r