ArmPkg: Update OemGetChassisType function to return MISC_CHASSIS_TYPE

[mirror_edk2.git] / ArmPkg / Universal / Smbios / SmbiosMiscDxe / Type03 / MiscChassisManufacturerFunction.c

/** @file\r
  This driver parses the mMiscSubclassDataTable structure and reports\r
  any generated data to smbios.\r
\r
  Based on files under Nt32Pkg/MiscSubClassPlatformDxe/\r
\r
  Copyright (c) 2021, NUVIA Inc. All rights reserved.<BR>\r
  Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>\r
  Copyright (c) 2015, Hisilicon Limited. All rights reserved.<BR>\r
  Copyright (c) 2015, Linaro Limited. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/HiiLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/OemMiscLib.h>\r
#include <Library/PrintLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
\r
#include "SmbiosMisc.h"\r
\r
/**\r
  This function makes boot time changes to the contents of the\r
  MiscChassisManufacturer (Type 3) record.\r
\r
  @param  RecordData                 Pointer to SMBIOS table with default values.\r
  @param  Smbios                     SMBIOS protocol.\r
\r
  @retval EFI_SUCCESS                The SMBIOS table was successfully added.\r
  @retval EFI_INVALID_PARAMETER      Invalid parameter was found.\r
  @retval EFI_OUT_OF_RESOURCES       Failed to allocate required memory.\r
\r
**/\r
SMBIOS_MISC_TABLE_FUNCTION(MiscChassisManufacturer)\r
{\r
  CHAR8                           *OptionalStrStart;\r
  CHAR8                           *StrStart;\r
  UINTN                           RecordLength;\r
  UINTN                           ManuStrLen;\r
  UINTN                           VerStrLen;\r
  UINTN                           AssertTagStrLen;\r
  UINTN                           SerialNumStrLen;\r
  UINTN                           ChaNumStrLen;\r
  EFI_STRING                      Manufacturer;\r
  EFI_STRING                      Version;\r
  EFI_STRING                      SerialNumber;\r
  EFI_STRING                      AssertTag;\r
  EFI_STRING                      ChassisSkuNumber;\r
  EFI_STRING_ID                   TokenToGet;\r
  SMBIOS_TABLE_TYPE3              *SmbiosRecord;\r
  SMBIOS_TABLE_TYPE3              *InputData;\r
  EFI_STATUS                      Status;\r
\r
  UINT8                           ContainedElementCount;\r
  CONTAINED_ELEMENT               ContainedElements;\r
  UINT8                           ExtendLength;\r
\r
  ExtendLength = 0;\r
\r
  //\r
  // First check for invalid parameters.\r
  //\r
  if (RecordData == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  InputData = (SMBIOS_TABLE_TYPE3 *)RecordData;\r
\r
  OemUpdateSmbiosInfo (\r
    mSmbiosMiscHiiHandle,\r
    STRING_TOKEN (STR_MISC_CHASSIS_ASSET_TAG),\r
    AssetTagType03\r
    );\r
  OemUpdateSmbiosInfo (\r
    mSmbiosMiscHiiHandle,\r
    STRING_TOKEN (STR_MISC_CHASSIS_SERIAL_NUMBER),\r
    SerialNumberType03\r
    );\r
  OemUpdateSmbiosInfo (\r
    mSmbiosMiscHiiHandle,\r
    STRING_TOKEN (STR_MISC_CHASSIS_VERSION),\r
    VersionType03\r
    );\r
  OemUpdateSmbiosInfo (\r
    mSmbiosMiscHiiHandle,\r
    STRING_TOKEN (STR_MISC_CHASSIS_MANUFACTURER),\r
    ManufacturerType03\r
    );\r
  OemUpdateSmbiosInfo (\r
    mSmbiosMiscHiiHandle,\r
    STRING_TOKEN (STR_MISC_CHASSIS_SKU_NUMBER),\r
    SkuNumberType03\r
    );\r
\r
  TokenToGet = STRING_TOKEN (STR_MISC_CHASSIS_MANUFACTURER);\r
  Manufacturer = HiiGetPackageString (&gEfiCallerIdGuid, TokenToGet, NULL);\r
  ManuStrLen = StrLen (Manufacturer);\r
\r
  TokenToGet = STRING_TOKEN (STR_MISC_CHASSIS_VERSION);\r
  Version = HiiGetPackageString (&gEfiCallerIdGuid, TokenToGet, NULL);\r
  VerStrLen = StrLen (Version);\r
\r
  TokenToGet = STRING_TOKEN (STR_MISC_CHASSIS_SERIAL_NUMBER);\r
  SerialNumber = HiiGetPackageString (&gEfiCallerIdGuid, TokenToGet, NULL);\r
  SerialNumStrLen = StrLen (SerialNumber);\r
\r
  TokenToGet = STRING_TOKEN (STR_MISC_CHASSIS_ASSET_TAG);\r
  AssertTag = HiiGetPackageString (&gEfiCallerIdGuid, TokenToGet, NULL);\r
  AssertTagStrLen = StrLen (AssertTag);\r
\r
  TokenToGet = STRING_TOKEN (STR_MISC_CHASSIS_SKU_NUMBER);\r
  ChassisSkuNumber = HiiGetPackageString (&gEfiCallerIdGuid, TokenToGet, NULL);\r
  ChaNumStrLen = StrLen (ChassisSkuNumber);\r
\r
  ContainedElementCount = InputData->ContainedElementCount;\r
\r
  if (ContainedElementCount > 1) {\r
    ExtendLength = (ContainedElementCount - 1) * sizeof (CONTAINED_ELEMENT);\r
  }\r
\r
  //\r
  // Two zeros following the last string.\r
  //\r
  RecordLength = sizeof (SMBIOS_TABLE_TYPE3) +\r
                 ExtendLength    + 1 +\r
                 ManuStrLen      + 1 +\r
                 VerStrLen       + 1 +\r
                 SerialNumStrLen + 1 +\r
                 AssertTagStrLen + 1 +\r
                 ChaNumStrLen    + 1 + 1;\r
  SmbiosRecord = AllocateZeroPool (RecordLength);\r
  if (SmbiosRecord == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Exit;\r
  }\r
\r
  (VOID)CopyMem (SmbiosRecord, InputData, sizeof (SMBIOS_TABLE_TYPE3));\r
\r
  SmbiosRecord->Hdr.Length = sizeof (SMBIOS_TABLE_TYPE3) + ExtendLength + 1;\r
\r
  SmbiosRecord->Type = OemGetChassisType ();\r
\r
  //ContainedElements\r
  ASSERT (ContainedElementCount < 2);\r
  (VOID)CopyMem (SmbiosRecord + 1, &ContainedElements, ExtendLength);\r
\r
  //ChassisSkuNumber\r
  *((UINT8 *)SmbiosRecord + sizeof (SMBIOS_TABLE_TYPE3) + ExtendLength) = 5;\r
\r
  OptionalStrStart = (CHAR8 *)((UINT8 *)SmbiosRecord + sizeof (SMBIOS_TABLE_TYPE3) +\r
                                        ExtendLength + 1);\r
  UnicodeStrToAsciiStrS (Manufacturer, OptionalStrStart, ManuStrLen + 1);\r
  StrStart = OptionalStrStart + ManuStrLen + 1;\r
  UnicodeStrToAsciiStrS (Version, StrStart, VerStrLen + 1);\r
  StrStart += VerStrLen + 1;\r
  UnicodeStrToAsciiStrS (SerialNumber, StrStart, SerialNumStrLen + 1);\r
  StrStart += SerialNumStrLen + 1;\r
  UnicodeStrToAsciiStrS (AssertTag, StrStart, AssertTagStrLen + 1);\r
  StrStart += AssertTagStrLen + 1;\r
  UnicodeStrToAsciiStrS (ChassisSkuNumber, StrStart, ChaNumStrLen + 1);\r
  //\r
  // Now we have got the full smbios record, call smbios protocol to add this record.\r
  //\r
  Status = SmbiosMiscAddRecord ((UINT8*)SmbiosRecord, NULL);\r
  if (EFI_ERROR (Status)) {\r
    DEBUG ((DEBUG_ERROR, "[%a]:[%dL] Smbios Type03 Table Log Failed! %r \n",\r
            __FUNCTION__, __LINE__, Status));\r
  }\r
\r
  FreePool (SmbiosRecord);\r
\r
Exit:\r
  if (Manufacturer != NULL) {\r
    FreePool (Manufacturer);\r
  }\r
\r
  if (Version != NULL) {\r
    FreePool (Version);\r
  }\r
\r
  if (SerialNumber != NULL) {\r
    FreePool (SerialNumber);\r
  }\r
\r
  if (AssertTag != NULL) {\r
    FreePool (AssertTag);\r
  }\r
\r
  if (ChassisSkuNumber != NULL) {\r
    FreePool (ChassisSkuNumber);\r
  }\r
\r
  return 0;\r
}\r