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
ShellCommandRunEdit(IN EFI_HANDLE ImageHandle,IN EFI_SYSTEM_TABLE * SystemTable)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