Home | History | Annotate | Download | only in Edit
      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 }
    163