1 /** @file
2 Add external EblCmd Lib
3
4 Copyright (c) 2007, Intel Corporation. All rights reserved.<BR>
5 Portions copyright (c) 2008 - 2009, Apple Inc. All rights reserved.<BR>
6
7 This program and the accompanying materials
8 are licensed and made available under the terms and conditions of the BSD License
9 which accompanies this distribution. The full text of the license may be found at
10 http://opensource.org/licenses/bsd-license.php
11
12 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
13 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
14
15
16 **/
17
18 #include <Uefi.h>
19 #include <Library/UefiLib.h>
20 #include <Library/UefiBootServicesTableLib.h>
21 #include <Library/EblAddExternalCommandLib.h>
22 #include <Protocol/EblAddCommand.h>
23
24 STATIC BOOLEAN gInstalledCommand = FALSE;
25 STATIC EFI_EVENT mEblCommandRegistration = NULL;
26
27 STATIC const EBL_COMMAND_TABLE *mAddExternalCmdLibTemplate = NULL;
28 STATIC UINTN mAddExternalCmdLibTemplateSize = 0;
29 EBL_ADD_COMMAND_PROTOCOL *gEblExternalCommand = NULL;
30
31
32 /**
33 Return a keypress or optionally timeout if a timeout value was passed in.
34 An optional callback function is called every second when waiting for a
35 timeout.
36
37 @param Key EFI Key information returned
38 @param TimeoutInSec Number of seconds to wait to timeout
39 @param CallBack Callback called every second during the timeout wait
40
41 @return EFI_SUCCESS Key was returned
42 @return EFI_TIMEOUT If the TimoutInSec expired
43
44 **/
45 EFI_STATUS
46 EFIAPI
EblGetCharKey(IN OUT EFI_INPUT_KEY * Key,IN UINTN TimeoutInSec,IN EBL_GET_CHAR_CALL_BACK CallBack OPTIONAL)47 EblGetCharKey (
48 IN OUT EFI_INPUT_KEY *Key,
49 IN UINTN TimeoutInSec,
50 IN EBL_GET_CHAR_CALL_BACK CallBack OPTIONAL
51 )
52 {
53 if (gEblExternalCommand != NULL) {
54 return gEblExternalCommand->EblGetCharKey (Key, TimeoutInSec, CallBack);
55 }
56 return EFI_TIMEOUT;
57 }
58
59
60 /**
61 This routine is used prevent command output data from scrolling off the end
62 of the screen. The global gPageBreak is used to turn on or off this feature.
63 If the CurrentRow is near the end of the screen pause and print out a prompt
64 If the use hits Q to quit return TRUE else for any other key return FALSE.
65 PrefixNewline is used to figure out if a newline is needed before the prompt
66 string. This depends on the last print done before calling this function.
67 CurrentRow is updated by one on a call or set back to zero if a prompt is
68 needed.
69
70 @param CurrentRow Used to figure out if its the end of the page and updated
71 @param PrefixNewline Did previous print issue a newline
72
73 @return TRUE if Q was hit to quit, FALSE in all other cases.
74
75 **/
76 BOOLEAN
77 EFIAPI
EblAnyKeyToContinueQtoQuit(IN UINTN * CurrentRow,IN BOOLEAN PrefixNewline)78 EblAnyKeyToContinueQtoQuit (
79 IN UINTN *CurrentRow,
80 IN BOOLEAN PrefixNewline
81 )
82 {
83 if (gEblExternalCommand != NULL) {
84 return gEblExternalCommand->EblAnyKeyToContinueQtoQuit (CurrentRow, PrefixNewline);
85 }
86 return FALSE;
87 }
88
89
90
91 /**
92 Update mFvbEntry. Add new entry, or update existing entry if Fvb protocol is
93 reinstalled.
94
95 @param Event The Event that is being processed
96 @param Context Event Context
97
98 **/
99 VOID
100 EFIAPI
EblAddCommandNotificationEvent(IN EFI_EVENT Event,IN VOID * Context)101 EblAddCommandNotificationEvent (
102 IN EFI_EVENT Event,
103 IN VOID *Context
104 )
105 {
106 EFI_STATUS Status;
107
108 if (!gInstalledCommand) {
109 Status = gBS->LocateProtocol (&gEfiEblAddCommandProtocolGuid, NULL, (VOID **)&gEblExternalCommand);
110 if (!EFI_ERROR (Status)) {
111 gEblExternalCommand->AddCommands (mAddExternalCmdLibTemplate, mAddExternalCmdLibTemplateSize);
112 gInstalledCommand = TRUE;
113 }
114 }
115 }
116
117
118
119 /**
120 The user Entry Point for the driver. The user code starts with this function
121 as the real entry point for the image goes into a library that calls this
122 function.
123
124 @param[in] ImageHandle The firmware allocated handle for the EFI image.
125 @param[in] SystemTable A pointer to the EFI System Table.
126
127 @retval EFI_SUCCESS The entry point is executed successfully.
128 @retval other Some error occurs when executing this entry point.
129
130 **/
131 EFI_STATUS
132 EFIAPI
EblAddExternalCommands(IN const EBL_COMMAND_TABLE * EntryArray,IN UINTN ArrayCount)133 EblAddExternalCommands (
134 IN const EBL_COMMAND_TABLE *EntryArray,
135 IN UINTN ArrayCount
136 )
137 {
138 if (mAddExternalCmdLibTemplate != NULL) {
139 return EFI_ALREADY_STARTED;
140 }
141
142 mAddExternalCmdLibTemplate = EntryArray;
143 mAddExternalCmdLibTemplateSize = ArrayCount;
144
145 EfiCreateProtocolNotifyEvent (
146 &gEfiEblAddCommandProtocolGuid,
147 TPL_CALLBACK,
148 EblAddCommandNotificationEvent,
149 NULL,
150 &mEblCommandRegistration
151 );
152
153 return EFI_SUCCESS;
154 }
155
156