ShellPkg: Fix the bug that handling Ctrl-C improperly

[mirror_edk2.git] / ShellPkg / Application / Shell / ConsoleWrappers.c

/** @file\r
  Function definitions for shell simple text in and out on top of file handles.\r
\r
  (C) Copyright 2013 Hewlett-Packard Development Company, L.P.<BR>\r
  Copyright (c) 2010 - 2015, 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 "Shell.h"\r
\r
extern BOOLEAN AsciiRedirection;\r
\r
typedef struct {\r
  EFI_SIMPLE_TEXT_INPUT_PROTOCOL  SimpleTextIn;\r
  SHELL_FILE_HANDLE               FileHandle;\r
  EFI_HANDLE                      TheHandle;\r
  UINT64                          RemainingBytesOfInputFile;\r
} SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL;\r
\r
typedef struct {\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL SimpleTextOut;\r
  SHELL_FILE_HANDLE               FileHandle;\r
  EFI_HANDLE                      TheHandle;\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *OriginalSimpleTextOut;\r
} SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL;\r
\r
/**\r
  Event notification function for EFI_SIMPLE_TEXT_INPUT_PROTOCOL.WaitForKey event\r
  Signal the event if there is key available\r
\r
  @param  Event                    Indicates the event that invoke this function.\r
  @param  Context                  Indicates the calling context.\r
\r
**/\r
VOID\r
EFIAPI\r
ConInWaitForKey (\r
  IN  EFI_EVENT       Event,\r
  IN  VOID            *Context\r
  )\r
{\r
  gBS->SignalEvent (Event);\r
}\r
\r
/**\r
  Reset function for the fake simple text input.\r
\r
  @param[in] This     A pointer to the SimpleTextIn structure.\r
  @param[in] ExtendedVerification TRUE for extra validation, FALSE otherwise.\r
\r
  @retval   EFI_SUCCESS The reset was successful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextInReset(\r
  IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL *This,\r
  IN BOOLEAN                        ExtendedVerification\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  ReadKeyStroke function for the fake simple text input.\r
\r
  @param[in] This      A pointer to the SimpleTextIn structure.\r
  @param[in, out] Key  A pointer to the Key structure to fill.\r
\r
  @retval   EFI_SUCCESS The read was successful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextInReadKeyStroke(\r
  IN      EFI_SIMPLE_TEXT_INPUT_PROTOCOL *This,\r
  IN OUT  EFI_INPUT_KEY                  *Key\r
  )\r
{\r
  UINTN Size;\r
  UINTN CharSize;\r
\r
  //\r
  // Verify the parameters\r
  //\r
  if (Key == NULL || This == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  //\r
  // Check if we have any characters left in the stream.\r
  //\r
  if (((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)This)->RemainingBytesOfInputFile == 0) {\r
    return (EFI_NOT_READY);\r
  }\r
\r
  Size = sizeof(CHAR16);\r
\r
  if(!AsciiRedirection) {\r
    CharSize = sizeof(CHAR16);\r
  } else {\r
    CharSize = sizeof(CHAR8);\r
  } \r
  //\r
  // Decrement the amount of free space by Size or set to zero (for odd length files)\r
  //\r
  if (((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)This)->RemainingBytesOfInputFile > CharSize) {\r
    ((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)This)->RemainingBytesOfInputFile -= CharSize;\r
  } else {\r
    ((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)This)->RemainingBytesOfInputFile = 0;\r
  }\r
\r
  Key->ScanCode = 0;\r
  return (ShellInfoObject.NewEfiShellProtocol->ReadFile(\r
    ((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)This)->FileHandle,\r
    &Size,\r
    &Key->UnicodeChar));\r
}\r
\r
/**\r
  Function to create a EFI_SIMPLE_TEXT_INPUT_PROTOCOL on top of a \r
  SHELL_FILE_HANDLE to support redirecting input from a file.\r
\r
  @param[in]  FileHandleToUse The pointer to the SHELL_FILE_HANDLE to use.\r
  @param[in]  HandleLocation  The pointer of a location to copy handle with protocol to.\r
\r
  @retval NULL                There was insufficient memory available.\r
  @return                     A pointer to the allocated protocol structure;\r
**/\r
EFI_SIMPLE_TEXT_INPUT_PROTOCOL*\r
CreateSimpleTextInOnFile(\r
  IN SHELL_FILE_HANDLE  FileHandleToUse,\r
  IN EFI_HANDLE         *HandleLocation\r
  )\r
{\r
  SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL  *ProtocolToReturn;\r
  EFI_STATUS                            Status;\r
  UINT64                                CurrentPosition;\r
  UINT64                                FileSize;\r
\r
  if (HandleLocation == NULL || FileHandleToUse == NULL) {\r
    return (NULL);\r
  }\r
\r
  ProtocolToReturn = AllocateZeroPool(sizeof(SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL));\r
  if (ProtocolToReturn == NULL) {\r
    return (NULL);\r
  }\r
\r
  ShellGetFileSize    (FileHandleToUse, &FileSize);\r
  ShellGetFilePosition(FileHandleToUse, &CurrentPosition);\r
\r
  //\r
  // Initialize the protocol members\r
  //\r
  ProtocolToReturn->RemainingBytesOfInputFile  = FileSize - CurrentPosition;\r
  ProtocolToReturn->FileHandle                 = FileHandleToUse;\r
  ProtocolToReturn->SimpleTextIn.Reset         = FileBasedSimpleTextInReset;\r
  ProtocolToReturn->SimpleTextIn.ReadKeyStroke = FileBasedSimpleTextInReadKeyStroke;\r
  \r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_WAIT,\r
                  TPL_NOTIFY,\r
                  ConInWaitForKey,\r
                  &ProtocolToReturn->SimpleTextIn,\r
                  &ProtocolToReturn->SimpleTextIn.WaitForKey\r
                  );\r
\r
  if (EFI_ERROR(Status)) {\r
    FreePool(ProtocolToReturn);\r
    return (NULL);\r
  }\r
  ///@todo possibly also install SimpleTextInputEx on the handle at this point.\r
  Status = gBS->InstallProtocolInterface(\r
    &(ProtocolToReturn->TheHandle), \r
    &gEfiSimpleTextInProtocolGuid, \r
    EFI_NATIVE_INTERFACE, \r
    &(ProtocolToReturn->SimpleTextIn));\r
  if (!EFI_ERROR(Status)) {\r
    *HandleLocation = ProtocolToReturn->TheHandle;\r
    return ((EFI_SIMPLE_TEXT_INPUT_PROTOCOL*)ProtocolToReturn);\r
  } else {\r
    FreePool(ProtocolToReturn);\r
    return (NULL);\r
  }\r
}\r
\r
/**\r
  Function to close a EFI_SIMPLE_TEXT_INPUT_PROTOCOL on top of a \r
  SHELL_FILE_HANDLE to support redirecting input from a file.\r
\r
  @param[in]  SimpleTextIn    The pointer to the SimpleTextIn to close.\r
\r
  @retval EFI_SUCCESS         The object was closed.\r
**/\r
EFI_STATUS\r
CloseSimpleTextInOnFile(\r
  IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL  *SimpleTextIn\r
  )\r
{\r
  EFI_STATUS Status;\r
  EFI_STATUS Status1;\r
\r
  if (SimpleTextIn == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  Status = gBS->CloseEvent(((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)SimpleTextIn)->SimpleTextIn.WaitForKey);\r
\r
  Status1 = gBS->UninstallProtocolInterface(\r
    ((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL*)SimpleTextIn)->TheHandle, \r
    &gEfiSimpleTextInProtocolGuid, \r
    &(((SHELL_EFI_SIMPLE_TEXT_INPUT_PROTOCOL*)SimpleTextIn)->SimpleTextIn));\r
\r
  FreePool(SimpleTextIn);\r
  if (!EFI_ERROR(Status)) {\r
    return (Status1);\r
  } else {\r
    return (Status);\r
  }\r
}\r
\r
/**\r
  Reset the text output device hardware and optionaly run diagnostics.\r
\r
  @param  This                pointer to EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL\r
  @param ExtendedVerification Indicates that a more extensive test may be performed\r
\r
  @retval EFI_SUCCESS         The text output device was reset.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutReset (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
  IN  BOOLEAN                         ExtendedVerification\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Verifies that all characters in a Unicode string can be output to the\r
  target device.\r
\r
  @param[in] This     Protocol instance pointer.\r
  @param[in] WString  The NULL-terminated Unicode string to be examined.\r
\r
  @retval EFI_SUCCESS The device(s) are capable of rendering the output string.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutTestString (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
  IN  CHAR16                          *WString\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Returns information for an available text mode that the output device(s)\r
  supports.\r
\r
  @param[in] This               Protocol instance pointer.\r
  @param[in] ModeNumber         The mode number to return information on.\r
  @param[out] Columns           Upon return, the number of columns in the selected geometry\r
  @param[out] Rows              Upon return, the number of rows in the selected geometry\r
\r
  @retval EFI_UNSUPPORTED       The mode number was not valid.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutQueryMode (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
  IN  UINTN                           ModeNumber,\r
  OUT UINTN                           *Columns,\r
  OUT UINTN                           *Rows\r
  )\r
{\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *PassThruProtocol;\r
  \r
  PassThruProtocol = ((SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)This)->OriginalSimpleTextOut;\r
  \r
  // Pass the QueryMode call thru to the original SimpleTextOutProtocol\r
  return (PassThruProtocol->QueryMode(\r
    PassThruProtocol,\r
    ModeNumber,\r
    Columns,\r
    Rows));\r
}\r
\r
/**\r
  Sets the output device(s) to a specified mode.\r
\r
  @param[in] This               Protocol instance pointer.\r
  @param[in] ModeNumber         The mode number to set.\r
\r
  @retval EFI_UNSUPPORTED       The mode number was not valid.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutSetMode (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *This,\r
  IN  UINTN                         ModeNumber\r
  )\r
{\r
  return (EFI_UNSUPPORTED);\r
}\r
\r
/**\r
  Sets the background and foreground colors for the OutputString () and\r
  ClearScreen () functions.\r
\r
  @param[in] This               Protocol instance pointer.\r
  @param[in] Attribute          The attribute to set. Bits 0..3 are the foreground color, and\r
                                bits 4..6 are the background color. All other bits are undefined\r
                                and must be zero. The valid Attributes are defined in this file.\r
\r
  @retval EFI_SUCCESS           The attribute was set.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutSetAttribute (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
  IN  UINTN                           Attribute\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Clears the output device(s) display to the currently selected background\r
  color.\r
\r
  @param[in] This               Protocol instance pointer.\r
\r
  @retval EFI_UNSUPPORTED       The output device is not in a valid text mode.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutClearScreen (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *This\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Sets the current coordinates of the cursor position\r
\r
  @param[in] This               Protocol instance pointer.\r
  @param[in] Column             Column to put the cursor in.  Must be between zero and Column returned from QueryMode\r
  @param[in] Row                Row to put the cursor in.  Must be between zero and Row returned from QueryMode\r
\r
  @retval EFI_SUCCESS           The operation completed successfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutSetCursorPosition (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *This,\r
  IN  UINTN                         Column,\r
  IN  UINTN                         Row\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Makes the cursor visible or invisible\r
\r
  @param[in] This       Protocol instance pointer.\r
  @param[in] Visible    If TRUE, the cursor is set to be visible. If FALSE, the cursor is\r
                        set to be invisible.\r
\r
  @retval EFI_SUCCESS           The operation completed successfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutEnableCursor (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *This,\r
  IN  BOOLEAN                       Visible\r
  )\r
{\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Write a Unicode string to the output device.\r
\r
  @param[in] This                 Protocol instance pointer.\r
  @param[in] WString              The NULL-terminated Unicode string to be displayed on the output\r
                                  device(s). All output devices must also support the Unicode\r
                                  drawing defined in this file.\r
  @retval EFI_SUCCESS             The string was output to the device.\r
  @retval EFI_DEVICE_ERROR        The device reported an error while attempting to output\r
                                  the text.\r
  @retval EFI_UNSUPPORTED         The output device's mode is not currently in a\r
                                  defined text mode.\r
  @retval EFI_WARN_UNKNOWN_GLYPH  This warning code indicates that some of the\r
                                  characters in the Unicode string could not be\r
                                  rendered and were skipped.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileBasedSimpleTextOutOutputString (\r
  IN  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
  IN  CHAR16                          *WString\r
  )\r
{\r
  UINTN Size;\r
  Size = StrLen(WString) * sizeof(CHAR16);\r
  return (ShellInfoObject.NewEfiShellProtocol->WriteFile(\r
    ((SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)This)->FileHandle,\r
    &Size,\r
    WString));\r
}\r
\r
/**\r
  Function to create a EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL on top of a \r
  SHELL_FILE_HANDLE to support redirecting output from a file.\r
\r
  @param[in]  FileHandleToUse  The pointer to the SHELL_FILE_HANDLE to use.\r
  @param[in]  HandleLocation   The pointer of a location to copy handle with protocol to.\r
  @param[in]  OriginalProtocol The pointer to the original output protocol for pass thru of functions.\r
\r
  @retval NULL                There was insufficient memory available.\r
  @return                     A pointer to the allocated protocol structure;\r
**/\r
EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL*\r
CreateSimpleTextOutOnFile(\r
  IN SHELL_FILE_HANDLE               FileHandleToUse,\r
  IN EFI_HANDLE                      *HandleLocation,\r
  IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *OriginalProtocol\r
  )\r
{\r
  SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *ProtocolToReturn;\r
  EFI_STATUS                            Status;\r
\r
  if (HandleLocation == NULL || FileHandleToUse == NULL) {\r
    return (NULL);\r
  }\r
\r
  ProtocolToReturn = AllocateZeroPool(sizeof(SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL));\r
  if (ProtocolToReturn == NULL) {\r
    return (NULL);\r
  }\r
  ProtocolToReturn->FileHandle                      = FileHandleToUse;\r
  ProtocolToReturn->OriginalSimpleTextOut           = OriginalProtocol;\r
  ProtocolToReturn->SimpleTextOut.Reset             = FileBasedSimpleTextOutReset;\r
  ProtocolToReturn->SimpleTextOut.TestString        = FileBasedSimpleTextOutTestString;\r
  ProtocolToReturn->SimpleTextOut.QueryMode         = FileBasedSimpleTextOutQueryMode;\r
  ProtocolToReturn->SimpleTextOut.SetMode           = FileBasedSimpleTextOutSetMode;\r
  ProtocolToReturn->SimpleTextOut.SetAttribute      = FileBasedSimpleTextOutSetAttribute;\r
  ProtocolToReturn->SimpleTextOut.ClearScreen       = FileBasedSimpleTextOutClearScreen;\r
  ProtocolToReturn->SimpleTextOut.SetCursorPosition = FileBasedSimpleTextOutSetCursorPosition;\r
  ProtocolToReturn->SimpleTextOut.EnableCursor      = FileBasedSimpleTextOutEnableCursor;\r
  ProtocolToReturn->SimpleTextOut.OutputString      = FileBasedSimpleTextOutOutputString;\r
  ProtocolToReturn->SimpleTextOut.Mode              = AllocateZeroPool(sizeof(EFI_SIMPLE_TEXT_OUTPUT_MODE));\r
  if (ProtocolToReturn->SimpleTextOut.Mode == NULL) {\r
    FreePool(ProtocolToReturn);\r
    return (NULL);\r
  }\r
  ProtocolToReturn->SimpleTextOut.Mode->MaxMode       = OriginalProtocol->Mode->MaxMode;\r
  ProtocolToReturn->SimpleTextOut.Mode->Mode          = OriginalProtocol->Mode->Mode;\r
  ProtocolToReturn->SimpleTextOut.Mode->Attribute     = OriginalProtocol->Mode->Attribute;\r
  ProtocolToReturn->SimpleTextOut.Mode->CursorColumn  = OriginalProtocol->Mode->CursorColumn;\r
  ProtocolToReturn->SimpleTextOut.Mode->CursorRow     = OriginalProtocol->Mode->CursorRow;\r
  ProtocolToReturn->SimpleTextOut.Mode->CursorVisible = OriginalProtocol->Mode->CursorVisible;\r
\r
  Status = gBS->InstallProtocolInterface(\r
    &(ProtocolToReturn->TheHandle), \r
    &gEfiSimpleTextOutProtocolGuid, \r
    EFI_NATIVE_INTERFACE, \r
    &(ProtocolToReturn->SimpleTextOut));\r
  if (!EFI_ERROR(Status)) {\r
    *HandleLocation = ProtocolToReturn->TheHandle;\r
    return ((EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL*)ProtocolToReturn);\r
  } else {\r
    SHELL_FREE_NON_NULL(ProtocolToReturn->SimpleTextOut.Mode);\r
    SHELL_FREE_NON_NULL(ProtocolToReturn);\r
    return (NULL);\r
  }\r
}\r
\r
/**\r
  Function to close a EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL on top of a \r
  SHELL_FILE_HANDLE to support redirecting output from a file.\r
\r
  @param[in] SimpleTextOut    The pointer to the SimpleTextOUT to close.\r
\r
  @retval EFI_SUCCESS         The object was closed.\r
**/\r
EFI_STATUS\r
CloseSimpleTextOutOnFile(\r
  IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *SimpleTextOut\r
  )\r
{\r
  EFI_STATUS  Status;\r
  if (SimpleTextOut == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
  Status = gBS->UninstallProtocolInterface(\r
    ((SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL*)SimpleTextOut)->TheHandle, \r
    &gEfiSimpleTextOutProtocolGuid, \r
    &(((SHELL_EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL*)SimpleTextOut)->SimpleTextOut));\r
  FreePool(SimpleTextOut->Mode);\r
  FreePool(SimpleTextOut);\r
  return (Status);\r
}\r