Refine code to make it more safely.

[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / Edit / Misc.c

/** @file\r
  Implementation of various string and line routines.\r
\r
  Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved. <BR>\r
  This program and the accompanying materials\r
  are licensed and made available under the terms and conditions of the BSD License\r
  which accompanies this distribution.  The full text of the license may be found at\r
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "TextEditor.h"\r
#include "Misc.h"\r
\r
/**\r
  Duplicate a EFI_EDITOR_LINE structure.\r
\r
  @param Src                    The line structure to copy from.\r
\r
  @retval NULL                  A memory allocation failed.\r
  @return                       a pointer to the newly allcoated line.\r
**/\r
EFI_EDITOR_LINE *\r
EFIAPI\r
LineDup (\r
  IN  EFI_EDITOR_LINE *Src\r
  )\r
{\r
  EFI_EDITOR_LINE *Dest;\r
\r
  //\r
  // allocate for the line structure\r
  //\r
  Dest = AllocateZeroPool (sizeof (EFI_EDITOR_LINE));\r
  if (Dest == NULL) {\r
    return NULL;\r
  }\r
  //\r
  // allocate and set the line buffer\r
  //\r
  Dest->Buffer = CatSPrint (NULL, L"%s", Src->Buffer);\r
  if (Dest->Buffer == NULL) {\r
    FreePool (Dest);\r
    return NULL;\r
  }\r
\r
  //\r
  // set the other structure members\r
  //\r
  Dest->Signature = LINE_LIST_SIGNATURE;\r
  Dest->Size      = Src->Size;\r
  Dest->TotalSize = Dest->Size;\r
  Dest->Type      = Src->Type;\r
  Dest->Link      = Src->Link;\r
\r
  return Dest;\r
}\r
\r
/**\r
  Free a EFI_EDITOR_LINE structure.\r
\r
  @param Src                    The line structure to free.\r
**/\r
VOID\r
EFIAPI\r
LineFree (\r
  IN  EFI_EDITOR_LINE *Src\r
  )\r
{\r
  if (Src == NULL) {\r
    return ;\r
  }\r
  //\r
  // free the line buffer and then the line structure itself\r
  //\r
  SHELL_FREE_NON_NULL (Src->Buffer);\r
  SHELL_FREE_NON_NULL (Src);\r
\r
}\r
\r
\r
\r
\r
\r
\r
\r
\r
\r
\r