ShellPkg/Library/UefiShellDebug1CommandsLib/Edit/Edit.c

   1 /** @file
   2   Main file for Edit shell Debug1 function.
   3
   4   (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
   5   Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved. <BR>
   6   This program and the accompanying materials
   7   are licensed and made available under the terms and conditions of the BSD License
   8   which accompanies this distribution.  The full text of the license may be found at
   9   http://opensource.org/licenses/bsd-license.php
  10
  11   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  12   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  13
  14 **/
  15
  16 #include "UefiShellDebug1CommandsLib.h"
  17 #include "TextEditor.h"
  18
  19 /**
  20   Function for 'edit' command.
  21
  22   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
  23   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
  24 **/
  25 SHELL_STATUS
  26 EFIAPI
  27 ShellCommandRunEdit (
  28   IN EFI_HANDLE        ImageHandle,
  29   IN EFI_SYSTEM_TABLE  *SystemTable
  30   )
  31 {
  32   EFI_STATUS          Status;
  33   CHAR16              *Buffer;
  34   CHAR16              *ProblemParam;
  35   SHELL_STATUS        ShellStatus;
  36   LIST_ENTRY          *Package;
  37   CONST CHAR16        *Cwd;
  38   CHAR16              *Nfs;
  39   CHAR16              *Spot;
  40   CONST CHAR16        *TempParam;
  41 //  SHELL_FILE_HANDLE   TempHandle;
  42
  43   Buffer      = NULL;
  44   ShellStatus = SHELL_SUCCESS;
  45   Nfs         = NULL;
  46
  47   //
  48   // initialize the shell lib (we must be in non-auto-init...)
  49   //
  50   Status = ShellInitialize();
  51   ASSERT_EFI_ERROR(Status);
  52
  53   Status = CommandInit();
  54   ASSERT_EFI_ERROR(Status);
  55
  56   //
  57   // parse the command line
  58   //
  59   Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);
  60   if (EFI_ERROR(Status)) {
  61     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
  62       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"edit", ProblemParam);
  63       FreePool(ProblemParam);
  64       ShellStatus = SHELL_INVALID_PARAMETER;
  65     } else {
  66       ASSERT(FALSE);
  67     }
  68   } else {
  69     if (ShellCommandLineGetCount(Package) > 2) {
  70       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"edit");
  71       ShellStatus = SHELL_INVALID_PARAMETER;
  72     } else {
  73       Cwd = gEfiShellProtocol->GetCurDir(NULL);
  74       if (Cwd == NULL) {
  75         Cwd = ShellGetEnvironmentVariable(L"path");
  76         if (Cwd != NULL) {
  77           Nfs = StrnCatGrow(&Nfs, NULL, Cwd+3, 0);
  78           if (Nfs != NULL) {
  79             Spot = StrStr(Nfs, L";");
  80             if (Spot != NULL) {
  81               *Spot = CHAR_NULL;
  82             }
  83             Spot = StrStr(Nfs, L"\\");
  84             if (Spot != NULL) {
  85               Spot[1] = CHAR_NULL;
  86             }
  87             gEfiShellProtocol->SetCurDir(NULL, Nfs);
  88             FreePool(Nfs);
  89           }
  90         }
  91       }
  92
  93       Status = MainEditorInit ();
  94
  95       if (EFI_ERROR (Status)) {
  96         gST->ConOut->ClearScreen (gST->ConOut);
  97         gST->ConOut->EnableCursor (gST->ConOut, TRUE);
  98         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_EDIT_MAIN_INIT_FAILED), gShellDebug1HiiHandle);
  99       } else {
 100         MainEditorBackup ();
 101
 102         //
 103         // if editor launched with file named
 104         //
 105         if (ShellCommandLineGetCount(Package) == 2) {
 106           TempParam = ShellCommandLineGetRawValue(Package, 1);
 107           ASSERT(TempParam != NULL);
 108           FileBufferSetFileName (TempParam);
 109 //          if (EFI_ERROR(ShellFileExists(MainEditor.FileBuffer->FileName))) {
 110 //            Status = ShellOpenFileByName(MainEditor.FileBuffer->FileName, &TempHandle, EFI_FILE_MODE_CREATE|EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE, 0);
 111 //            if (!EFI_ERROR(Status)) {
 112 //              ShellCloseFile(&TempHandle);
 113 //            }
 114 //          }
 115         }
 116
 117         Status = FileBufferRead (MainEditor.FileBuffer->FileName, FALSE);
 118         if (!EFI_ERROR (Status)) {
 119           MainEditorRefresh ();
 120
 121           Status = MainEditorKeyInput ();
 122         }
 123
 124         if (Status != EFI_OUT_OF_RESOURCES) {
 125           //
 126           // back up the status string
 127           //
 128           Buffer = CatSPrint (NULL, L"%s", StatusBarGetString());
 129         }
 130
 131         MainEditorCleanup ();
 132
 133         //
 134         // print editor exit code on screen
 135         //
 136         if (Status == EFI_SUCCESS) {
 137         } else if (Status == EFI_OUT_OF_RESOURCES) {
 138           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"edit");
 139         } else {
 140           if (Buffer != NULL) {
 141             if (StrCmp (Buffer, L"") != 0) {
 142               //
 143               // print out the status string
 144               //
 145               ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_EDIT_MAIN_BUFFER), gShellDebug1HiiHandle, Buffer);
 146             } else {
 147               ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_EDIT_MAIN_UNKNOWN_EDITOR_ERR), gShellDebug1HiiHandle);
 148             }
 149           } else {
 150             ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_EDIT_MAIN_UNKNOWN_EDITOR_ERR), gShellDebug1HiiHandle);
 151           }
 152         }
 153
 154         if (Status != EFI_OUT_OF_RESOURCES) {
 155           SHELL_FREE_NON_NULL (Buffer);
 156         }
 157       }
 158     }
 159     ShellCommandLineFreeVarList (Package);
 160   }
 161   return ShellStatus;
 162 }