/** @file\r
Provides interface to shell console logger.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
EFI_STATUS Status;\r
ASSERT(ConsoleInfo != NULL);\r
\r
- (*ConsoleInfo) = AllocatePool(sizeof(CONSOLE_LOGGER_PRIVATE_DATA));\r
+ (*ConsoleInfo) = AllocateZeroPool(sizeof(CONSOLE_LOGGER_PRIVATE_DATA));\r
if ((*ConsoleInfo) == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
\r
- (*ConsoleInfo)->Signature = CONSOLE_LOGGER_PRIVATE_DATA_SIGNATURE;\r
- (*ConsoleInfo)->OldConOut = NULL;\r
- (*ConsoleInfo)->OldConHandle = NULL;\r
- (*ConsoleInfo)->Buffer = NULL;\r
- (*ConsoleInfo)->BufferSize = 0;\r
- (*ConsoleInfo)->OriginalStartRow = 0;\r
- (*ConsoleInfo)->CurrentStartRow = 0;\r
- (*ConsoleInfo)->RowsPerScreen = 0;\r
- (*ConsoleInfo)->ColsPerScreen = 0;\r
- (*ConsoleInfo)->Attributes = NULL;\r
- (*ConsoleInfo)->AttribSize = 0;\r
- (*ConsoleInfo)->ScreenCount = ScreensToSave;\r
- (*ConsoleInfo)->HistoryMode.MaxMode = 1;\r
- (*ConsoleInfo)->HistoryMode.Mode = 0;\r
- (*ConsoleInfo)->HistoryMode.Attribute = 0;\r
- (*ConsoleInfo)->HistoryMode.CursorColumn = 0;\r
- (*ConsoleInfo)->HistoryMode.CursorRow = 0;\r
- (*ConsoleInfo)->HistoryMode.CursorVisible = FALSE;\r
- (*ConsoleInfo)->OurConOut.Reset = ConsoleLoggerReset;\r
- (*ConsoleInfo)->OurConOut.OutputString = ConsoleLoggerOutputString;\r
- (*ConsoleInfo)->OurConOut.TestString = ConsoleLoggerTestString;\r
- (*ConsoleInfo)->OurConOut.QueryMode = ConsoleLoggerQueryMode;\r
- (*ConsoleInfo)->OurConOut.SetMode = ConsoleLoggerSetMode;\r
- (*ConsoleInfo)->OurConOut.SetAttribute = ConsoleLoggerSetAttribute;\r
- (*ConsoleInfo)->OurConOut.ClearScreen = ConsoleLoggerClearScreen;\r
+ (*ConsoleInfo)->Signature = CONSOLE_LOGGER_PRIVATE_DATA_SIGNATURE;\r
+ (*ConsoleInfo)->OldConOut = gST->ConOut;\r
+ (*ConsoleInfo)->OldConHandle = gST->ConsoleOutHandle;\r
+ (*ConsoleInfo)->Buffer = NULL;\r
+ (*ConsoleInfo)->BufferSize = 0;\r
+ (*ConsoleInfo)->OriginalStartRow = 0;\r
+ (*ConsoleInfo)->CurrentStartRow = 0;\r
+ (*ConsoleInfo)->RowsPerScreen = 0;\r
+ (*ConsoleInfo)->ColsPerScreen = 0;\r
+ (*ConsoleInfo)->Attributes = NULL;\r
+ (*ConsoleInfo)->AttribSize = 0;\r
+ (*ConsoleInfo)->ScreenCount = ScreensToSave;\r
+ (*ConsoleInfo)->HistoryMode.MaxMode = 1;\r
+ (*ConsoleInfo)->HistoryMode.Mode = 0;\r
+ (*ConsoleInfo)->HistoryMode.Attribute = 0;\r
+ (*ConsoleInfo)->HistoryMode.CursorColumn = 0;\r
+ (*ConsoleInfo)->HistoryMode.CursorRow = 0;\r
+ (*ConsoleInfo)->HistoryMode.CursorVisible = FALSE;\r
+ (*ConsoleInfo)->OurConOut.Reset = ConsoleLoggerReset;\r
+ (*ConsoleInfo)->OurConOut.OutputString = ConsoleLoggerOutputString;\r
+ (*ConsoleInfo)->OurConOut.TestString = ConsoleLoggerTestString;\r
+ (*ConsoleInfo)->OurConOut.QueryMode = ConsoleLoggerQueryMode;\r
+ (*ConsoleInfo)->OurConOut.SetMode = ConsoleLoggerSetMode;\r
+ (*ConsoleInfo)->OurConOut.SetAttribute = ConsoleLoggerSetAttribute;\r
+ (*ConsoleInfo)->OurConOut.ClearScreen = ConsoleLoggerClearScreen;\r
(*ConsoleInfo)->OurConOut.SetCursorPosition = ConsoleLoggerSetCursorPosition;\r
- (*ConsoleInfo)->OurConOut.EnableCursor = ConsoleLoggerEnableCursor;\r
- (*ConsoleInfo)->OurConOut.Mode = NULL;\r
- (*ConsoleInfo)->Enabled = TRUE;\r
+ (*ConsoleInfo)->OurConOut.EnableCursor = ConsoleLoggerEnableCursor;\r
+ (*ConsoleInfo)->OurConOut.Mode = gST->ConOut->Mode;\r
+ (*ConsoleInfo)->Enabled = TRUE;\r
\r
Status = ConsoleLoggerResetBuffers(*ConsoleInfo);\r
if (EFI_ERROR(Status)) {\r
return (Status);\r
}\r
\r
- (*ConsoleInfo)->OldConOut = gST->ConOut;\r
- (*ConsoleInfo)->OldConHandle = gST->ConsoleOutHandle;\r
-\r
gST->ConsoleOutHandle = gImageHandle;\r
- gST->ConOut = &(*ConsoleInfo)->OurConOut;\r
+ gST->ConOut = &(*ConsoleInfo)->OurConOut;\r
\r
return (Status);\r
}\r
Return the system to the state it was before InstallConsoleLogger\r
was installed.\r
\r
- @param[in,out] ConsoleInfo The object from the install function.\r
+ @param[in] ConsoleInfo The object from the install function.\r
\r
@retval EFI_SUCCESS The operation was successful\r
@return other The operation failed. This was from UninstallProtocolInterface.\r
}\r
if (*Resp == ShellPromptResponseContinue) {\r
FreePool(Resp);\r
- ShellInfoObject.ConsoleInfo->RowCounter = 0;\r
+ ShellInfoObject.ConsoleInfo->RowCounter = 0;\r
+// ShellInfoObject.ConsoleInfo->OurConOut.Mode->CursorRow = 0;\r
+// ShellInfoObject.ConsoleInfo->OurConOut.Mode->CursorColumn = 0;\r
+\r
return (EFI_SUCCESS);\r
} else if (*Resp == ShellPromptResponseQuit) {\r
FreePool(Resp);\r
EFI_STATUS\r
EFIAPI\r
ConsoleLoggerPrintWithPageBreak(\r
- IN CHAR16 *String,\r
+ IN CONST CHAR16 *String,\r
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo\r
)\r
{\r
CONST CHAR16 *Walker;\r
CONST CHAR16 *LineStart;\r
+ CHAR16 *StringCopy;\r
CHAR16 TempChar;\r
\r
- for ( Walker = String\r
- , LineStart = String\r
+ StringCopy = NULL;\r
+ StringCopy = StrnCatGrow(&StringCopy, NULL, String, 0);\r
+ if (StringCopy == NULL) {\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
+\r
+ for ( Walker = StringCopy\r
+ , LineStart = StringCopy\r
; Walker != NULL && *Walker != CHAR_NULL\r
; Walker++\r
){\r
//\r
// We got an error which means 'break' and halt the printing\r
//\r
+ SHELL_FREE_NON_NULL(StringCopy);\r
return (EFI_DEVICE_ERROR);\r
}\r
}\r
ConsoleLoggerOutputStringSplit (LineStart, ConsoleInfo);\r
}\r
\r
+ SHELL_FREE_NON_NULL(StringCopy);\r
return (EFI_SUCCESS);\r
}\r
\r
{\r
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;\r
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);\r
+ if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleOut) {\r
+ return (EFI_UNSUPPORTED);\r
+ }\r
ASSERT(ShellInfoObject.ConsoleInfo == ConsoleInfo);\r
if (!ShellInfoObject.ConsoleInfo->Enabled) {\r
return (EFI_DEVICE_ERROR);\r
EFI_STATUS\r
EFIAPI\r
ConsoleLoggerSetMode (\r
- IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
- IN UINTN ModeNumber\r
+ IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,\r
+ IN UINTN ModeNumber\r
)\r
{\r
EFI_STATUS Status;\r
// Check that the buffers are still correct for logging\r
//\r
if (!EFI_ERROR (Status)) {\r
+ ConsoleInfo->OurConOut.Mode = gST->ConOut->Mode;\r
ConsoleLoggerResetBuffers(ConsoleInfo);\r
}\r
\r
INT32 *Attributes;\r
UINTN Row;\r
UINTN Column;\r
+ CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;\r
\r
+ if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleOut) {\r
+ return (EFI_UNSUPPORTED);\r
+ }\r
\r
- CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;\r
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);\r
\r
//\r
)\r
{\r
EFI_STATUS Status;\r
-\r
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;\r
+\r
+ if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleOut) {\r
+ return (EFI_UNSUPPORTED);\r
+ }\r
+\r
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);\r
//\r
// Forward the request to the original ConOut\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
\r
- ConsoleInfo->OurConOut.Mode = gST->ConOut->Mode;\r
- ConsoleInfo->OldConOut = gST->ConOut;\r
CopyMem (&ConsoleInfo->HistoryMode, ConsoleInfo->OldConOut->Mode, sizeof (EFI_SIMPLE_TEXT_OUTPUT_MODE));\r
\r
return (EFI_SUCCESS);\r
/** @file\r
Function definitions for shell simple text in and out on top of file handles.\r
\r
- Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2010 - 2011, 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
ReadKeyStroke function for the fake simple text input.\r
\r
@param[in] This A pointer to the SimpleTextIn structure.\r
- @param[out] Key A pointer to the Key structure to fill.\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 EFI_INPUT_KEY *Key\r
+ IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL *This,\r
+ IN OUT EFI_INPUT_KEY *Key\r
)\r
{\r
UINTN Size;\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
+ @param[in] SimpleTextOut The pointer to the SimpleTextOUT to close.\r
\r
@retval EFI_SUCCESS The object was closed.\r
**/\r
EFI_STATUS\r
EFIAPI\r
CloseSimpleTextOutOnFile(\r
- OUT EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *SimpleTextOut\r
+ IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *SimpleTextOut\r
)\r
{\r
EFI_STATUS Status;\r
/** @file\r
Function definitions for shell simple text in and out on top of file handles.\r
\r
- Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2010 - 2011, 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
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
+ @param[in] SimpleTextOut The pointer to the SimpleTextOUT to close.\r
\r
@retval EFI_SUCCESS The object was closed.\r
**/\r
EFI_STATUS\r
EFIAPI\r
CloseSimpleTextOutOnFile(\r
- OUT EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *SimpleTextOut\r
+ IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *SimpleTextOut\r
);\r
\r
#endif //_SHELL_CONSOLE_WRAPPERS_HEADER_\r
EFI_FILE_PROTOCOL wrappers for other items (Like Environment Variables,\r
StdIn, StdOut, StdErr, etc...).\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
/**\r
File style interface for StdIn (Write).\r
\r
- @param[in] This Ignored.\r
- @param[in] BufferSize Ignored.\r
- @param[in] Buffer Ignored.\r
+ @param[in] This Ignored.\r
+ @param[in,out] BufferSize Ignored.\r
+ @param[in] Buffer Ignored.\r
\r
@retval EFI_UNSUPPORTED\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileInterfaceStdInWrite(\r
- IN EFI_FILE_PROTOCOL *This,\r
- IN OUT UINTN *BufferSize,\r
- IN VOID *Buffer\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN OUT UINTN *BufferSize,\r
+ IN VOID *Buffer\r
)\r
{\r
return (EFI_UNSUPPORTED);\r
@param[in,out] BufferSize Ignored.\r
@param[out] Buffer Ignored.\r
\r
- @retval EFI_UNSUPPORTED\r
+ @retval EFI_UNSUPPORTED Always.\r
**/\r
EFI_STATUS\r
EFIAPI\r
\r
@param[in] This Ignored.\r
@param[in,out] BufferSize Ignored.\r
- @param[in] Buffer Ignored.\r
+ @param[out] Buffer Ignored.\r
\r
- @retval EFI_SUCCESS\r
+ @retval EFI_SUCCESS Always.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileInterfaceNulRead(\r
- IN EFI_FILE_PROTOCOL *This,\r
- IN OUT UINTN *BufferSize,\r
- OUT VOID *Buffer\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN OUT UINTN *BufferSize,\r
+ OUT VOID *Buffer\r
)\r
{\r
return (EFI_SUCCESS);\r
//\r
// Ascii\r
//\r
- AsciiBuffer = AllocatePool(*BufferSize);\r
+ AsciiBuffer = AllocateZeroPool(*BufferSize);\r
AsciiSPrint(AsciiBuffer, *BufferSize, "%S", Buffer);\r
if ((UINTN)(((EFI_FILE_PROTOCOL_MEM*)This)->Position + AsciiStrSize(AsciiBuffer)) > (UINTN)(((EFI_FILE_PROTOCOL_MEM*)This)->BufferSize)) {\r
((EFI_FILE_PROTOCOL_MEM*)This)->Buffer = ReallocatePool((UINTN)(((EFI_FILE_PROTOCOL_MEM*)This)->BufferSize), (UINTN)(((EFI_FILE_PROTOCOL_MEM*)This)->BufferSize) + AsciiStrSize(AsciiBuffer) + 10, ((EFI_FILE_PROTOCOL_MEM*)This)->Buffer);\r
EFI_FILE_PROTOCOL *Orig;\r
} EFI_FILE_PROTOCOL_FILE;\r
\r
+/**\r
+ Set a files current position\r
+\r
+ @param This Protocol instance pointer.\r
+ @param Position Byte position from the start of the file.\r
+ \r
+ @retval EFI_SUCCESS Data was written.\r
+ @retval EFI_UNSUPPORTED Seek request for non-zero is not valid on open.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileSetPosition(\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN UINT64 Position\r
+ )\r
+{\r
+ return ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->SetPosition(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, Position);\r
+}\r
+\r
+/**\r
+ Get a file's current position\r
+\r
+ @param This Protocol instance pointer.\r
+ @param Position Byte position from the start of the file.\r
+ \r
+ @retval EFI_SUCCESS Data was written.\r
+ @retval EFI_UNSUPPORTED Seek request for non-zero is not valid on open..\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileGetPosition(\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ OUT UINT64 *Position\r
+ )\r
+{\r
+ return ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->GetPosition(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, Position);\r
+}\r
+\r
+/**\r
+ Get information about a file.\r
+\r
+ @param This Protocol instance pointer.\r
+ @param InformationType Type of information to return in Buffer.\r
+ @param BufferSize On input size of buffer, on output amount of data in buffer.\r
+ @param Buffer The buffer to return data.\r
+\r
+ @retval EFI_SUCCESS Data was returned.\r
+ @retval EFI_UNSUPPORT InformationType is not supported.\r
+ @retval EFI_NO_MEDIA The device has no media.\r
+ @retval EFI_DEVICE_ERROR The device reported an error.\r
+ @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.\r
+ @retval EFI_WRITE_PROTECTED The device is write protected.\r
+ @retval EFI_ACCESS_DENIED The file was open for read only.\r
+ @retval EFI_BUFFER_TOO_SMALL Buffer was too small; required size returned in BufferSize.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileGetInfo(\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN EFI_GUID *InformationType,\r
+ IN OUT UINTN *BufferSize,\r
+ OUT VOID *Buffer\r
+ )\r
+{\r
+ return ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->GetInfo(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, InformationType, BufferSize, Buffer);\r
+}\r
+\r
+/**\r
+ Set information about a file\r
+\r
+ @param File Protocol instance pointer.\r
+ @param InformationType Type of information in Buffer.\r
+ @param BufferSize Size of buffer.\r
+ @param Buffer The data to write.\r
+\r
+ @retval EFI_SUCCESS Data was returned.\r
+ @retval EFI_UNSUPPORT InformationType is not supported.\r
+ @retval EFI_NO_MEDIA The device has no media.\r
+ @retval EFI_DEVICE_ERROR The device reported an error.\r
+ @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.\r
+ @retval EFI_WRITE_PROTECTED The device is write protected.\r
+ @retval EFI_ACCESS_DENIED The file was open for read only.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileSetInfo(\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN EFI_GUID *InformationType,\r
+ IN UINTN BufferSize,\r
+ IN VOID *Buffer\r
+ )\r
+{\r
+ return ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->SetInfo(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, InformationType, BufferSize, Buffer);\r
+}\r
+\r
+/**\r
+ Flush data back for the file handle.\r
+\r
+ @param This Protocol instance pointer.\r
+\r
+ @retval EFI_SUCCESS Data was written.\r
+ @retval EFI_UNSUPPORT Writes to Open directory are not supported.\r
+ @retval EFI_NO_MEDIA The device has no media.\r
+ @retval EFI_DEVICE_ERROR The device reported an error.\r
+ @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.\r
+ @retval EFI_WRITE_PROTECTED The device is write protected.\r
+ @retval EFI_ACCESS_DENIED The file was open for read only.\r
+ @retval EFI_VOLUME_FULL The volume is full.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileFlush(\r
+ IN EFI_FILE_PROTOCOL *This\r
+ )\r
+{\r
+ return ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Flush(((EFI_FILE_PROTOCOL_FILE*)This)->Orig);\r
+}\r
+\r
+/**\r
+ Read data from the file.\r
+\r
+ @param This Protocol instance pointer.\r
+ @param BufferSize On input size of buffer, on output amount of data in buffer.\r
+ @param Buffer The buffer in which data is read.\r
+\r
+ @retval EFI_SUCCESS Data was read.\r
+ @retval EFI_NO_MEDIA The device has no media.\r
+ @retval EFI_DEVICE_ERROR The device reported an error.\r
+ @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.\r
+ @retval EFI_BUFFER_TO_SMALL BufferSize is too small. BufferSize contains required size.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileRead(\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN OUT UINTN *BufferSize,\r
+ OUT VOID *Buffer\r
+ )\r
+{\r
+ CHAR8 *AsciiBuffer;\r
+ UINTN Size;\r
+ EFI_STATUS Status;\r
+ if (((EFI_FILE_PROTOCOL_FILE*)This)->Unicode) {\r
+ //\r
+ // Unicode\r
+ //\r
+ return (((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Read(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, BufferSize, Buffer));\r
+ } else {\r
+ //\r
+ // Ascii\r
+ //\r
+ AsciiBuffer = AllocateZeroPool((Size = *BufferSize));\r
+ Status = (((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Read(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, &Size, AsciiBuffer));\r
+ UnicodeSPrint(Buffer, *BufferSize, L"%a", AsciiBuffer);\r
+ FreePool(AsciiBuffer);\r
+ return (Status);\r
+ }\r
+}\r
+\r
+/**\r
+ Opens a new file relative to the source file's location.\r
+\r
+ @param[in] This The protocol instance pointer.\r
+ @param[out] NewHandle Returns File Handle for FileName.\r
+ @param[in] FileName Null terminated string. "\", ".", and ".." are supported.\r
+ @param[in] OpenMode Open mode for file.\r
+ @param[in] Attributes Only used for EFI_FILE_MODE_CREATE.\r
+\r
+ @retval EFI_SUCCESS The device was opened.\r
+ @retval EFI_NOT_FOUND The specified file could not be found on the device.\r
+ @retval EFI_NO_MEDIA The device has no media.\r
+ @retval EFI_MEDIA_CHANGED The media has changed.\r
+ @retval EFI_DEVICE_ERROR The device reported an error.\r
+ @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.\r
+ @retval EFI_ACCESS_DENIED The service denied access to the file.\r
+ @retval EFI_OUT_OF_RESOURCES The volume was not opened due to lack of resources.\r
+ @retval EFI_VOLUME_FULL The volume is full.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileOpen (\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ OUT EFI_FILE_PROTOCOL **NewHandle,\r
+ IN CHAR16 *FileName,\r
+ IN UINT64 OpenMode,\r
+ IN UINT64 Attributes\r
+ )\r
+{\r
+ return ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Open(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, NewHandle, FileName, OpenMode, Attributes);\r
+}\r
+\r
+/**\r
+ Close and delete the file handle.\r
+\r
+ @param This Protocol instance pointer.\r
+ \r
+ @retval EFI_SUCCESS The device was opened.\r
+ @retval EFI_WARN_DELETE_FAILURE The handle was closed but the file was not deleted.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+FileInterfaceFileDelete(\r
+ IN EFI_FILE_PROTOCOL *This\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+ Status = ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Delete(((EFI_FILE_PROTOCOL_FILE*)This)->Orig);\r
+ FreePool(This);\r
+ return (Status);\r
+}\r
+\r
/**\r
File style interface for File (Close).\r
\r
IN EFI_FILE_PROTOCOL *This\r
)\r
{\r
- ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Close(((EFI_FILE_PROTOCOL_FILE*)This)->Orig);\r
+ EFI_STATUS Status;\r
+ Status = ((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Close(((EFI_FILE_PROTOCOL_FILE*)This)->Orig);\r
FreePool(This);\r
- return (EFI_SUCCESS);\r
+ return (Status);\r
}\r
\r
/**\r
If the file was opened with ASCII mode the data will be processed through \r
AsciiSPrint before writing.\r
\r
- @param[in] This The pointer to the EFI_FILE_PROTOCOL object.\r
- @param[in] BufferSize Size in bytes of Buffer.\r
- @param[in] Buffer The pointer to the buffer to write.\r
+ @param[in] This The pointer to the EFI_FILE_PROTOCOL object.\r
+ @param[in,out] BufferSize Size in bytes of Buffer.\r
+ @param[in] Buffer The pointer to the buffer to write.\r
\r
@retval EFI_SUCCESS The data was written.\r
**/\r
EFI_STATUS\r
EFIAPI\r
FileInterfaceFileWrite(\r
- IN EFI_FILE_PROTOCOL *This,\r
- IN OUT UINTN *BufferSize,\r
- IN VOID *Buffer\r
+ IN EFI_FILE_PROTOCOL *This,\r
+ IN OUT UINTN *BufferSize,\r
+ IN VOID *Buffer\r
)\r
{\r
CHAR8 *AsciiBuffer;\r
//\r
// Ascii\r
//\r
- AsciiBuffer = AllocatePool(*BufferSize);\r
+ AsciiBuffer = AllocateZeroPool(*BufferSize);\r
AsciiSPrint(AsciiBuffer, *BufferSize, "%S", Buffer);\r
Size = AsciiStrSize(AsciiBuffer) - 1; // (we dont need the null terminator)\r
Status = (((EFI_FILE_PROTOCOL_FILE*)This)->Orig->Write(((EFI_FILE_PROTOCOL_FILE*)This)->Orig, &Size, AsciiBuffer));\r
{\r
EFI_FILE_PROTOCOL_FILE *NewOne;\r
\r
- NewOne = AllocatePool(sizeof(EFI_FILE_PROTOCOL_FILE));\r
+ NewOne = AllocateZeroPool(sizeof(EFI_FILE_PROTOCOL_FILE));\r
if (NewOne == NULL) {\r
return (NULL);\r
}\r
CopyMem(NewOne, Template, sizeof(EFI_FILE_PROTOCOL_FILE));\r
- NewOne->Orig = (EFI_FILE_PROTOCOL *)Template;\r
- NewOne->Unicode = Unicode;\r
- NewOne->Close = FileInterfaceFileClose;\r
- NewOne->Write = FileInterfaceFileWrite;\r
+ NewOne->Orig = (EFI_FILE_PROTOCOL *)Template;\r
+ NewOne->Unicode = Unicode;\r
+ NewOne->Open = FileInterfaceFileOpen;\r
+ NewOne->Close = FileInterfaceFileClose;\r
+ NewOne->Delete = FileInterfaceFileDelete;\r
+ NewOne->Read = FileInterfaceFileRead;\r
+ NewOne->Write = FileInterfaceFileWrite;\r
+ NewOne->GetPosition = FileInterfaceFileGetPosition;\r
+ NewOne->SetPosition = FileInterfaceFileSetPosition;\r
+ NewOne->GetInfo = FileInterfaceFileGetInfo;\r
+ NewOne->SetInfo = FileInterfaceFileSetInfo;\r
+ NewOne->Flush = FileInterfaceFileFlush;\r
\r
return ((EFI_FILE_PROTOCOL *)NewOne);\r
}\r
/** @file\r
This is THE shell (application)\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
STATIC CONST CHAR16 mExecutableExtensions[] = L".NSH;.EFI";\r
STATIC CONST CHAR16 mStartupScript[] = L"startup.nsh";\r
\r
+/**\r
+ Function to start monitoring for CTRL-S using SimpleTextInputEx. This \r
+ feature's enabled state was not known when the shell initially launched.\r
+\r
+ @retval EFI_SUCCESS The feature is enabled.\r
+ @retval EFI_OUT_OF_RESOURCES There is not enough mnemory available.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+InternalEfiShellStartCtrlSMonitor(\r
+ VOID\r
+ )\r
+{\r
+ EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL *SimpleEx;\r
+ EFI_KEY_DATA KeyData;\r
+ EFI_STATUS Status;\r
+\r
+ Status = gBS->OpenProtocol(\r
+ gST->ConsoleInHandle,\r
+ &gEfiSimpleTextInputExProtocolGuid,\r
+ (VOID**)&SimpleEx,\r
+ gImageHandle,\r
+ NULL,\r
+ EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
+ if (EFI_ERROR(Status)) {\r
+ ShellPrintHiiEx(\r
+ -1, \r
+ -1, \r
+ NULL,\r
+ STRING_TOKEN (STR_SHELL_NO_IN_EX),\r
+ ShellInfoObject.HiiHandle);\r
+ return (EFI_SUCCESS);\r
+ }\r
+\r
+ KeyData.KeyState.KeyToggleState = 0;\r
+ KeyData.Key.ScanCode = 0;\r
+ KeyData.KeyState.KeyShiftState = EFI_SHIFT_STATE_VALID|EFI_LEFT_CONTROL_PRESSED;\r
+ KeyData.Key.UnicodeChar = L's';\r
+\r
+ Status = SimpleEx->RegisterKeyNotify(\r
+ SimpleEx,\r
+ &KeyData,\r
+ NotificationFunction,\r
+ &ShellInfoObject.CtrlSNotifyHandle1);\r
+ \r
+ KeyData.KeyState.KeyShiftState = EFI_SHIFT_STATE_VALID|EFI_RIGHT_CONTROL_PRESSED;\r
+ if (!EFI_ERROR(Status)) {\r
+ Status = SimpleEx->RegisterKeyNotify(\r
+ SimpleEx,\r
+ &KeyData,\r
+ NotificationFunction,\r
+ &ShellInfoObject.CtrlSNotifyHandle2);\r
+ }\r
+ KeyData.KeyState.KeyShiftState = EFI_SHIFT_STATE_VALID|EFI_LEFT_CONTROL_PRESSED;\r
+ KeyData.Key.UnicodeChar = 19;\r
+\r
+ if (!EFI_ERROR(Status)) {\r
+ Status = SimpleEx->RegisterKeyNotify(\r
+ SimpleEx,\r
+ &KeyData,\r
+ NotificationFunction,\r
+ &ShellInfoObject.CtrlSNotifyHandle2);\r
+ } \r
+ KeyData.KeyState.KeyShiftState = EFI_SHIFT_STATE_VALID|EFI_RIGHT_CONTROL_PRESSED;\r
+ if (!EFI_ERROR(Status)) {\r
+ Status = SimpleEx->RegisterKeyNotify(\r
+ SimpleEx,\r
+ &KeyData,\r
+ NotificationFunction,\r
+ &ShellInfoObject.CtrlSNotifyHandle2);\r
+ }\r
+ return (Status);\r
+}\r
+\r
+\r
+\r
/**\r
The entry point for the application.\r
\r
Status = InernalEfiShellStartMonitor();\r
}\r
\r
+ if (!EFI_ERROR(Status) && !ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleIn) {\r
+ //\r
+ // Set up the event for CTRL-S monitoring...\r
+ //\r
+ Status = InternalEfiShellStartCtrlSMonitor();\r
+ }\r
+\r
if (!EFI_ERROR(Status) && PcdGet8(PcdShellSupportLevel) >= 1) {\r
//\r
// process the startup script or launch the called app.\r
UINTN Count;\r
UINTN LoopVar;\r
CHAR16 *ProblemParam;\r
+ UINT64 Intermediate;\r
\r
Package = NULL;\r
ProblemParam = NULL;\r
Size = 0;\r
TempConst = ShellCommandLineGetRawValue(Package, Count++);\r
if (TempConst != NULL && StrLen(TempConst)) {\r
- ShellInfoObject.ShellInitSettings.FileName = AllocatePool(StrSize(TempConst));\r
+ ShellInfoObject.ShellInitSettings.FileName = AllocateZeroPool(StrSize(TempConst));\r
if (ShellInfoObject.ShellInitSettings.FileName == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoVersion = ShellCommandLineGetFlag(Package, L"-noversion");\r
ShellInfoObject.ShellInitSettings.BitUnion.Bits.Delay = ShellCommandLineGetFlag(Package, L"-delay");\r
\r
- if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.Delay) {\r
+ ShellInfoObject.ShellInitSettings.Delay = 5;\r
+\r
+ if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoInterrupt) {\r
+ ShellInfoObject.ShellInitSettings.Delay = 0;\r
+ } else if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.Delay) {\r
TempConst = ShellCommandLineGetValue(Package, L"-delay");\r
- if (TempConst != NULL) {\r
- ShellInfoObject.ShellInitSettings.Delay = StrDecimalToUintn (TempConst);\r
- } else {\r
- ShellInfoObject.ShellInitSettings.Delay = 5;\r
+ if (*TempConst == L':') {\r
+ TempConst++;\r
+ }\r
+ if (TempConst != NULL && !EFI_ERROR(ShellConvertStringToUint64(TempConst, &Intermediate, FALSE, FALSE))) {\r
+ ShellInfoObject.ShellInitSettings.Delay = (UINTN)Intermediate;\r
}\r
- } else {\r
- ShellInfoObject.ShellInitSettings.Delay = 5;\r
}\r
-\r
ShellCommandLineFreeVarList(Package);\r
\r
return (Status);\r
EFI_DEVICE_PATH_PROTOCOL *NewPath;\r
EFI_DEVICE_PATH_PROTOCOL *NamePath;\r
CHAR16 *FileStringPath;\r
+ CHAR16 *TempSpot;\r
UINTN NewSize;\r
+ CONST CHAR16 *MapName;\r
\r
Key.UnicodeChar = CHAR_NULL;\r
Key.ScanCode = 0;\r
//\r
// print out our warning and see if they press a key\r
//\r
- for ( Status = EFI_UNSUPPORTED, Delay = (ShellInfoObject.ShellInitSettings.Delay * 10)\r
- ; Delay > 0 && EFI_ERROR(Status)\r
+ for ( Status = EFI_UNSUPPORTED, Delay = ShellInfoObject.ShellInitSettings.Delay * 10\r
+ ; Delay != 0 && EFI_ERROR(Status)\r
; Delay--\r
){\r
ShellPrintHiiEx(0, gST->ConOut->Mode->CursorRow, NULL, STRING_TOKEN (STR_SHELL_STARTUP_QUESTION), ShellInfoObject.HiiHandle, Delay/10);\r
gBS->Stall (100000);\r
- Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
+ if (!ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleIn) {\r
+ Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
+ }\r
}\r
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SHELL_CRLF), ShellInfoObject.HiiHandle);\r
\r
return (EFI_SUCCESS);\r
}\r
\r
- NamePath = FileDevicePath (NULL, mStartupScript);\r
//\r
- // Try the first location\r
+ // Try the first location (must be file system)\r
//\r
- NewPath = AppendDevicePathNode (ImagePath, NamePath);\r
- Status = InternalOpenFileDevicePath(NewPath, &FileHandle, EFI_FILE_MODE_READ, 0);\r
+ MapName = ShellInfoObject.NewEfiShellProtocol->GetMapFromDevicePath(&ImagePath);\r
+ if (MapName != NULL) {\r
+ FileStringPath = NULL;\r
+ NewSize = 0;\r
+ FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, MapName, 0);\r
+ TempSpot = StrStr(FileStringPath, L";");\r
+ if (TempSpot != NULL) {\r
+ *TempSpot = CHAR_NULL;\r
+ }\r
+ FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, ((FILEPATH_DEVICE_PATH*)FilePath)->PathName, 0);\r
+ ChopLastSlash(FileStringPath);\r
+ FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, mStartupScript, 0);\r
+ Status = ShellInfoObject.NewEfiShellProtocol->OpenFileByName(FileStringPath, &FileHandle, EFI_FILE_MODE_READ);\r
+ FreePool(FileStringPath);\r
+ }\r
if (EFI_ERROR(Status)) {\r
+ NamePath = FileDevicePath (NULL, mStartupScript);\r
+ NewPath = AppendDevicePathNode (ImagePath, NamePath);\r
+ FreePool(NamePath);\r
+\r
//\r
- // Try the second location\r
+ // Try the location\r
//\r
- FreePool(NewPath);\r
- NewPath = AppendDevicePathNode (FilePath , NamePath);\r
Status = InternalOpenFileDevicePath(NewPath, &FileHandle, EFI_FILE_MODE_READ, 0);\r
+ FreePool(NewPath);\r
}\r
-\r
//\r
// If we got a file, run it\r
//\r
- if (!EFI_ERROR(Status)) {\r
+ if (!EFI_ERROR(Status) && FileHandle != NULL) {\r
Status = RunScriptFileHandle (FileHandle, mStartupScript);\r
ShellInfoObject.NewEfiShellProtocol->CloseFile(FileHandle);\r
} else {\r
}\r
}\r
\r
- FreePool(NamePath);\r
- FreePool(NewPath);\r
\r
return (Status);\r
}\r
return (EFI_SUCCESS);\r
}\r
FreePool(*CommandString);\r
- *CommandString = AllocatePool(StrSize(NewString));\r
+ *CommandString = AllocateZeroPool(StrSize(NewString));\r
if (*CommandString == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
UINTN PostAliasSize;\r
CHAR16 *PostVariableCmdLine;\r
CHAR16 *CommandWithPath;\r
- EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
+ CONST EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
CONST CHAR16 *TempLocation;\r
CONST CHAR16 *TempLocation2;\r
SHELL_FILE_HANDLE OriginalStdIn;\r
\r
Status = UpdateStdInStdOutStdErr(ShellInfoObject.NewShellParametersProtocol, PostVariableCmdLine, &OriginalStdIn, &OriginalStdOut, &OriginalStdErr, &OriginalSystemTableInfo);\r
if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SHELL_INVALID_REDIR), ShellInfoObject.HiiHandle);\r
+ if (Status == EFI_NOT_FOUND) {\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SHELL_REDUNDA_REDIR), ShellInfoObject.HiiHandle);\r
+ } else {\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SHELL_INVALID_REDIR), ShellInfoObject.HiiHandle);\r
+ }\r
} else {\r
//\r
// remove the < and/or > from the command line now\r
SHELL_FREE_NON_NULL(CommandName);\r
SHELL_FREE_NON_NULL(CommandWithPath);\r
SHELL_FREE_NON_NULL(PostVariableCmdLine);\r
- SHELL_FREE_NON_NULL(DevPath);\r
\r
return (Status);\r
}\r
//\r
// Now enumerate through the commands and run each one.\r
//\r
- CommandLine = AllocatePool(PcdGet16(PcdShellPrintBufferSize));\r
+ CommandLine = AllocateZeroPool(PcdGet16(PcdShellPrintBufferSize));\r
if (CommandLine == NULL) {\r
DeleteScriptFileStruct(NewScriptFile);\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
- CommandLine2 = AllocatePool(PcdGet16(PcdShellPrintBufferSize));\r
+ CommandLine2 = AllocateZeroPool(PcdGet16(PcdShellPrintBufferSize));\r
if (CommandLine2 == NULL) {\r
FreePool(CommandLine);\r
DeleteScriptFileStruct(NewScriptFile);\r
}\r
}\r
\r
- ShellCommandSetEchoState(PreScriptEchoState);\r
\r
FreePool(CommandLine);\r
FreePool(CommandLine2);\r
ShellCommandSetNewScript (NULL);\r
+\r
+ //\r
+ // Only if this was the last script reset the state.\r
+ //\r
+ if (ShellCommandGetCurrentScriptFile()==NULL) {\r
+ ShellCommandSetEchoState(PreScriptEchoState);\r
+ }\r
return (EFI_SUCCESS);\r
}\r
\r
/** @file\r
function definitions for internal to shell functions.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
EFI_HANDLE CtrlCNotifyHandle2; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
EFI_HANDLE CtrlCNotifyHandle3; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
EFI_HANDLE CtrlCNotifyHandle4; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
+ EFI_HANDLE CtrlSNotifyHandle1; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
+ EFI_HANDLE CtrlSNotifyHandle2; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
+ EFI_HANDLE CtrlSNotifyHandle3; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
+ EFI_HANDLE CtrlSNotifyHandle4; ///< The NotifyHandle returned from SimpleTextInputEx.RegisterKeyNotify.\r
} SHELL_INFO;\r
\r
extern SHELL_INFO ShellInfoObject;\r
/** @file\r
function declarations for shell environment functions.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
&Size,\r
Buffer);\r
if (Status == EFI_BUFFER_TOO_SMALL) {\r
- Buffer = AllocatePool(Size);\r
+ Buffer = AllocateZeroPool(Size);\r
ASSERT(Buffer != NULL);\r
Status = gRT->GetVariable((CHAR16*)EnvVarName,\r
&gShellVariableGuid,\r
}\r
\r
NameSize = (UINTN)MaxVarSize;\r
- VariableName = AllocatePool(NameSize);\r
+ VariableName = AllocateZeroPool(NameSize);\r
if (VariableName == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
ValSize = 0;\r
Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
if (Status == EFI_BUFFER_TOO_SMALL){\r
- VarList->Val = AllocatePool(ValSize);\r
+ VarList->Val = AllocateZeroPool(ValSize);\r
if (VarList->Val == NULL) {\r
SHELL_FREE_NON_NULL(VarList);\r
Status = EFI_OUT_OF_RESOURCES;\r
}\r
}\r
if (!EFI_ERROR(Status) && VarList != NULL) {\r
- VarList->Key = AllocatePool(StrSize(VariableName));\r
+ VarList->Key = AllocateZeroPool(StrSize(VariableName));\r
if (VarList->Key == NULL) {\r
SHELL_FREE_NON_NULL(VarList->Val);\r
SHELL_FREE_NON_NULL(VarList);\r
break;\r
}\r
ASSERT(StrStr(CurrentString, L"=") != NULL);\r
- Node = AllocatePool(sizeof(ENV_VAR_LIST));\r
+ Node = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
ASSERT(Node != NULL);\r
Node->Key = AllocateZeroPool((StrStr(CurrentString, L"=") - CurrentString + 1) * sizeof(CHAR16));\r
ASSERT(Node->Key != NULL);\r
/** @file\r
Provides interface to shell MAN file parser.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
\r
Status = EFI_SUCCESS;\r
\r
- TitleString = AllocatePool((7*sizeof(CHAR16)) + StrSize(Command));\r
+ TitleString = AllocateZeroPool((7*sizeof(CHAR16)) + StrSize(Command));\r
if (TitleString == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
; CurrentLocation++);\r
\r
TitleEnd = StrStr(CurrentLocation, L"\"");\r
- ASSERT(TitleEnd != NULL);\r
- if (BriefDesc != NULL) {\r
- *BriefSize = StrSize(TitleEnd);\r
- *BriefDesc = AllocateZeroPool(*BriefSize);\r
- if (*BriefDesc == NULL) {\r
- Status = EFI_OUT_OF_RESOURCES;\r
- } else {\r
- StrnCpy(*BriefDesc, CurrentLocation, TitleEnd-CurrentLocation);\r
+ if (TitleEnd == NULL) {\r
+ Status = EFI_DEVICE_ERROR;\r
+ } else {\r
+ if (BriefDesc != NULL) {\r
+ *BriefSize = StrSize(TitleEnd);\r
+ *BriefDesc = AllocateZeroPool(*BriefSize);\r
+ if (*BriefDesc == NULL) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ } else {\r
+ StrnCpy(*BriefDesc, CurrentLocation, TitleEnd-CurrentLocation);\r
+ }\r
}\r
- }\r
\r
- for (CurrentLocation = TitleEnd\r
- ; *CurrentLocation != L'\n'\r
- ; CurrentLocation++);\r
- for (\r
- ; *CurrentLocation == L' ' || *CurrentLocation == L'\n' || *CurrentLocation == L'\r'\r
- ; CurrentLocation++);\r
- *Buffer = CurrentLocation;\r
+ for (CurrentLocation = TitleEnd\r
+ ; *CurrentLocation != L'\n'\r
+ ; CurrentLocation++);\r
+ for (\r
+ ; *CurrentLocation == L' ' || *CurrentLocation == L'\n' || *CurrentLocation == L'\r'\r
+ ; CurrentLocation++);\r
+ *Buffer = CurrentLocation;\r
+ }\r
}\r
\r
FreePool(TitleString);\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
\r
- TitleString = AllocatePool((4*sizeof(CHAR16)) + StrSize(Command));\r
+ TitleString = AllocateZeroPool((4*sizeof(CHAR16)) + StrSize(Command));\r
if (TitleString == NULL) {\r
FreePool(ReadLine);\r
return (EFI_OUT_OF_RESOURCES);\r
Member functions of EFI_SHELL_PARAMETERS_PROTOCOL and functions for creation,\r
manipulation, and initialization of EFI_SHELL_PARAMETERS_PROTOCOL.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
for (NextDelim = *TempParameter ; NextDelim != NULL && *NextDelim != CHAR_NULL ; NextDelim++) {\r
if (*NextDelim == L'^' && *(NextDelim+1) == L'^') {\r
CopyMem(NextDelim, NextDelim+1, StrSize(NextDelim) - sizeof(NextDelim[0]));\r
- } else if (*NextDelim == L'^') {\r
+ }/* else if (*NextDelim == L'^') {\r
*NextDelim = L' ';\r
- }\r
+ }*/\r
}\r
while ((*TempParameter)[StrLen(*TempParameter)-1] == L' ') {\r
(*TempParameter)[StrLen(*TempParameter)-1] = CHAR_NULL;\r
}\r
\r
/**\r
- function to populate Argc and Argv.\r
+ Function to populate Argc and Argv.\r
\r
This function parses the CommandLine and divides it into standard C style Argc/Argv\r
parameters for inclusion in EFI_SHELL_PARAMETERS_PROTOCOL. this supports space\r
return (Status);\r
}\r
\r
+EFI_STATUS\r
+EFIAPI\r
+IsUnicodeFile(\r
+ IN CONST CHAR16 *FileName\r
+ )\r
+{\r
+ SHELL_FILE_HANDLE Handle;\r
+ EFI_STATUS Status;\r
+ UINT64 OriginalFilePosition;\r
+ UINTN CharSize;\r
+ CHAR16 CharBuffer;\r
+\r
+ Status = gEfiShellProtocol->OpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ);\r
+ if (EFI_ERROR(Status)) {\r
+ return (Status);\r
+ }\r
+ gEfiShellProtocol->GetFilePosition(Handle, &OriginalFilePosition);\r
+ gEfiShellProtocol->SetFilePosition(Handle, 0);\r
+ CharSize = sizeof(CHAR16);\r
+ Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);\r
+ if (EFI_ERROR(Status) || CharBuffer != gUnicodeFileTag) {\r
+ Status = EFI_BUFFER_TOO_SMALL;\r
+ }\r
+ gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);\r
+ gEfiShellProtocol->CloseFile(Handle);\r
+ return (Status); \r
+}\r
+\r
/**\r
Funcion will replace the current StdIn and StdOut in the ShellParameters protocol\r
structure by parsing NewCommandLine. The current values are returned to the\r
}\r
\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 2>>v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 12, L' ');\r
StdErrVarName = CommandLineWalker += 6;\r
ErrAppend = TRUE;\r
+ if (StrStr(CommandLineWalker, L" 2>>v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 1>>v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 12, L' ');\r
StdOutVarName = CommandLineWalker += 6;\r
OutAppend = TRUE;\r
+ if (StrStr(CommandLineWalker, L" 1>>v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
} else if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" >>v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
StdOutVarName = CommandLineWalker += 5;\r
OutAppend = TRUE;\r
+ if (StrStr(CommandLineWalker, L" >>v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
} else if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" >v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
StdOutVarName = CommandLineWalker += 4;\r
OutAppend = FALSE;\r
+ if (StrStr(CommandLineWalker, L" >v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 1>>a ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 12, L' ');\r
StdOutFileName = CommandLineWalker += 6;\r
OutAppend = TRUE;\r
OutUnicode = FALSE;\r
+ if (StrStr(CommandLineWalker, L" 1>>a ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 1>> ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdOutFileName = CommandLineWalker += 5;\r
OutAppend = TRUE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 1>> ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
} \r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" >> ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdOutFileName = CommandLineWalker += 4;\r
OutAppend = TRUE;\r
}\r
+ if (StrStr(CommandLineWalker, L" >> ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" >>a ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
OutAppend = TRUE;\r
OutUnicode = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" >>a ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
} \r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 1>a ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
OutAppend = FALSE;\r
OutUnicode = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 1>a ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
} \r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" >a ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
OutAppend = FALSE;\r
OutUnicode = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" >a ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 2>> ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdErrFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdErrFileName = CommandLineWalker += 5;\r
ErrAppend = TRUE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 2>> ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 2>v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdErrVarName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdErrVarName = CommandLineWalker += 5;\r
ErrAppend = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 2>v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 1>v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdOutVarName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdOutVarName = CommandLineWalker += 5;\r
OutAppend = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 1>v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 2>a ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 10, L' ');\r
if (StdErrFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
ErrAppend = FALSE;\r
ErrUnicode = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 2>a ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 2> ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
if (StdErrFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdErrFileName = CommandLineWalker += 4;\r
ErrAppend = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 2> ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" 1> ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdOutFileName = CommandLineWalker += 4;\r
OutAppend = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" 1> ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" > ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 6, L' ');\r
if (StdOutFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdOutFileName = CommandLineWalker += 3;\r
OutAppend = FALSE;\r
}\r
+ if (StrStr(CommandLineWalker, L" > ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
+ }\r
}\r
\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" < ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 6, L' ');\r
if (StdInFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdInFileName = CommandLineWalker += 3;\r
- OutAppend = FALSE;\r
+ }\r
+ if (StrStr(CommandLineWalker, L" < ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
}\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" <a ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
if (StdInFileName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
- StdInFileName = CommandLineWalker += 4;\r
- OutAppend = FALSE;\r
+ StdInFileName = CommandLineWalker += 4;\r
+ InUnicode = FALSE;\r
+ }\r
+ if (StrStr(CommandLineWalker, L" <a ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
}\r
}\r
if (!EFI_ERROR(Status) && (CommandLineWalker = StrStr(CommandLineCopy, L" <v ")) != NULL) {\r
+ SetMem16(CommandLineWalker, 8, L' ');\r
if (StdInVarName != NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
StdInVarName = CommandLineWalker += 4;\r
- OutAppend = FALSE;\r
+ }\r
+ if (StrStr(CommandLineWalker, L" <v ") != NULL) {\r
+ Status = EFI_NOT_FOUND;\r
}\r
}\r
\r
//\r
// Verify not the same and not duplicating something from a split\r
//\r
- if ((StdErrFileName != NULL && StdOutFileName!= NULL && StringNoCaseCompare(&StdErrFileName, &StdOutFileName) == 0)\r
+ if (\r
+ //\r
+ // Check that no 2 filenames are the same\r
+ //\r
+ (StdErrFileName != NULL && StdOutFileName!= NULL && StringNoCaseCompare(&StdErrFileName, &StdOutFileName) == 0)\r
||(StdErrFileName != NULL && StdInFileName != NULL && StringNoCaseCompare(&StdErrFileName, &StdInFileName ) == 0)\r
||(StdOutFileName != NULL && StdInFileName != NULL && StringNoCaseCompare(&StdOutFileName, &StdInFileName ) == 0)\r
+ //\r
+ // Check that no 2 variable names are the same\r
+ //\r
||(StdErrVarName != NULL && StdInVarName != NULL && StringNoCaseCompare(&StdErrVarName , &StdInVarName ) == 0)\r
||(StdOutVarName != NULL && StdInVarName != NULL && StringNoCaseCompare(&StdOutVarName , &StdInVarName ) == 0)\r
||(StdErrVarName != NULL && StdOutVarName != NULL && StringNoCaseCompare(&StdErrVarName , &StdOutVarName ) == 0)\r
+ //\r
+ // When a split (using | operator) is in place some are not allowed\r
+ //\r
||(Split != NULL && Split->SplitStdIn != NULL && (StdInVarName != NULL || StdInFileName != NULL))\r
||(Split != NULL && Split->SplitStdOut != NULL && (StdOutVarName != NULL || StdOutFileName != NULL))\r
+ //\r
+ // Check that nothing is trying to be output to 2 locations.\r
+ //\r
||(StdErrFileName != NULL && StdErrVarName != NULL)\r
||(StdOutFileName != NULL && StdOutVarName != NULL)\r
||(StdInFileName != NULL && StdInVarName != NULL)\r
+ //\r
+ // There should not be extra > or <\r
+ //\r
+ ||(StrStr(CommandLineCopy, L"<") != NULL)\r
+ ||(StrStr(CommandLineCopy, L">") != NULL)\r
+ //\r
+ // Check for no volatile environment variables\r
+ //\r
||(StdErrVarName != NULL && !IsVolatileEnv(StdErrVarName))\r
||(StdOutVarName != NULL && !IsVolatileEnv(StdOutVarName))\r
+ //\r
+ // Cant redirect during a reconnect operation.\r
+ //\r
||(StrStr(NewCommandLine, L"connect -r") != NULL \r
&& (StdOutVarName != NULL || StdOutFileName != NULL || StdErrFileName != NULL || StdErrVarName != NULL))\r
+ //\r
+ // Check that filetypes (Unicode/Ascii) do not change during an append\r
+ //\r
+ ||(StdOutFileName != NULL && OutUnicode && OutAppend && (!EFI_ERROR(ShellFileExists(StdOutFileName)) && EFI_ERROR(IsUnicodeFile(StdOutFileName))))\r
+ ||(StdErrFileName != NULL && ErrUnicode && ErrAppend && (!EFI_ERROR(ShellFileExists(StdErrFileName)) && EFI_ERROR(IsUnicodeFile(StdErrFileName))))\r
+ ||(StdOutFileName != NULL && !OutUnicode && OutAppend && (!EFI_ERROR(ShellFileExists(StdOutFileName)) && !EFI_ERROR(IsUnicodeFile(StdOutFileName))))\r
+ ||(StdErrFileName != NULL && !ErrUnicode && ErrAppend && (!EFI_ERROR(ShellFileExists(StdErrFileName)) && !EFI_ERROR(IsUnicodeFile(StdErrFileName))))\r
){\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
ShellInfoObject.NewEfiShellProtocol->DeleteFileByName(StdErrFileName);\r
}\r
Status = ShellOpenFileByName(StdErrFileName, &TempHandle, EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ|EFI_FILE_MODE_CREATE,0);\r
- ASSERT(TempHandle != NULL);\r
if (!ErrAppend && ErrUnicode && !EFI_ERROR(Status)) {\r
//\r
- // Write out the UnicodeFileTag\r
+ // Write out the gUnicodeFileTag\r
//\r
Size = sizeof(CHAR16);\r
- TagBuffer[0] = UnicodeFileTag;\r
+ TagBuffer[0] = gUnicodeFileTag;\r
TagBuffer[1] = CHAR_NULL;\r
ShellInfoObject.NewEfiShellProtocol->WriteFile(TempHandle, &Size, TagBuffer);\r
}\r
if (TempHandle == NULL) {\r
Status = EFI_INVALID_PARAMETER;\r
} else {\r
- if (!OutAppend && OutUnicode && !EFI_ERROR(Status)) {\r
+ if (StrStr(StdOutFileName, L"NUL")==StdOutFileName) {\r
+ //no-op\r
+ } else if (!OutAppend && OutUnicode && !EFI_ERROR(Status)) {\r
//\r
- // Write out the UnicodeFileTag\r
+ // Write out the gUnicodeFileTag\r
//\r
Size = sizeof(CHAR16);\r
- TagBuffer[0] = UnicodeFileTag;\r
+ TagBuffer[0] = gUnicodeFileTag;\r
TagBuffer[1] = CHAR_NULL;\r
ShellInfoObject.NewEfiShellProtocol->WriteFile(TempHandle, &Size, TagBuffer);\r
} else if (OutAppend) {\r
}\r
TempHandle = CreateFileInterfaceEnv(StdOutVarName);\r
ASSERT(TempHandle != NULL);\r
- if (!OutUnicode) {\r
- TempHandle = CreateFileInterfaceFile(TempHandle, FALSE);\r
- ASSERT(TempHandle != NULL);\r
- }\r
ShellParameters->StdOut = TempHandle;\r
gST->ConOut = CreateSimpleTextOutOnFile(TempHandle, &gST->ConsoleOutHandle);\r
}\r
}\r
TempHandle = CreateFileInterfaceEnv(StdErrVarName);\r
ASSERT(TempHandle != NULL);\r
- if (!ErrUnicode) {\r
- TempHandle = CreateFileInterfaceFile(TempHandle, FALSE);\r
- ASSERT(TempHandle != NULL);\r
- }\r
ShellParameters->StdErr = TempHandle;\r
gST->StdErr = CreateSimpleTextOutOnFile(TempHandle, &gST->StandardErrorHandle);\r
}\r
Member functions of EFI_SHELL_PROTOCOL and functions for creation,\r
manipulation, and initialization of EFI_SHELL_PROTOCOL.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
return (FileHandleClose(ConvertShellHandleToEfiFileProtocol(FileHandle)));\r
}\r
\r
+/**\r
+ Internal worker to determine whether there is a BlockIo somewhere\r
+ upon the device path specified.\r
+\r
+ @param[in] DevicePath The device path to test.\r
+\r
+ @retval TRUE gEfiBlockIoProtocolGuid was installed on a handle with this device path\r
+ @retval FALSE gEfiBlockIoProtocolGuid was not found.\r
+**/\r
+BOOLEAN\r
+EFIAPI\r
+InternalShellProtocolIsBlockIoPresent(\r
+ IN CONST EFI_DEVICE_PATH_PROTOCOL *DevicePath\r
+ )\r
+{\r
+ EFI_DEVICE_PATH_PROTOCOL *DevicePathCopy;\r
+ EFI_STATUS Status;\r
+ EFI_HANDLE Handle;\r
+\r
+ Handle = NULL;\r
+\r
+ DevicePathCopy = (EFI_DEVICE_PATH_PROTOCOL*)DevicePath;\r
+ Status = gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid, &DevicePathCopy, &Handle);\r
+\r
+ if ((Handle != NULL) && (!EFI_ERROR(Status))) {\r
+ return (TRUE);\r
+ }\r
+ return (FALSE);\r
+}\r
+\r
/**\r
Internal worker to determine whether there is a file system somewhere\r
upon the device path specified.\r
// make sure this is a valid to add device path\r
//\r
///@todo add BlockIo to this test...\r
- if (!InternalShellProtocolIsSimpleFileSystemPresent(DevicePath)) {\r
+ if (!InternalShellProtocolIsSimpleFileSystemPresent(DevicePath)\r
+ && !InternalShellProtocolIsBlockIoPresent(DevicePath)) {\r
return (EFI_INVALID_PARAMETER);\r
}\r
\r
NewPath = AllocateZeroPool(Size);\r
ASSERT(NewPath != NULL);\r
StrCpy(NewPath, Cwd);\r
- if ((Path[0] == (CHAR16)L'\\') &&\r
- (NewPath[StrLen(NewPath)-1] == (CHAR16)L'\\')\r
- ) {\r
- ((CHAR16*)NewPath)[StrLen(NewPath)-1] = CHAR_NULL;\r
+ if (*Path == L'\\') {\r
+ Path++;\r
+ while (ChopLastSlash(NewPath)) ;\r
}\r
StrCat(NewPath, Path);\r
DevicePathForReturn = EfiShellGetDevicePathFromFilePath(NewPath);\r
CHAR8 *Lang;\r
CHAR8 *TempChar;\r
\r
+ UINTN ParentControllerCount;\r
+ EFI_HANDLE *ParentControllerBuffer;\r
+ UINTN ParentDriverCount;\r
+ EFI_HANDLE *ParentDriverBuffer;\r
+\r
if (BestDeviceName == NULL ||\r
DeviceHandle == NULL\r
){\r
continue;\r
}\r
if (Language == NULL) {\r
- Lang = AllocatePool(AsciiStrSize(CompName2->SupportedLanguages));\r
+ Lang = AllocateZeroPool(AsciiStrSize(CompName2->SupportedLanguages));\r
if (Lang == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
*TempChar = CHAR_NULL;\r
}\r
} else {\r
- Lang = AllocatePool(AsciiStrSize(Language));\r
+ Lang = AllocateZeroPool(AsciiStrSize(Language));\r
if (Lang == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
if (HandleList != NULL) {\r
FreePool(HandleList);\r
}\r
+\r
+ //\r
+ // Now check the parent controller using this as the child.\r
+ //\r
+ if (DeviceNameToReturn == NULL){\r
+ PARSE_HANDLE_DATABASE_PARENTS(DeviceHandle, &ParentControllerCount, &ParentControllerBuffer);\r
+ for (LoopVar = 0 ; LoopVar < ParentControllerCount ; LoopVar++) {\r
+ PARSE_HANDLE_DATABASE_UEFI_DRIVERS(ParentControllerBuffer[LoopVar], &ParentDriverCount, &ParentDriverBuffer);\r
+ for (HandleCount = 0 ; HandleCount < ParentDriverCount ; HandleCount++) {\r
+ //\r
+ // try using that driver's component name with controller and our driver as the child.\r
+ //\r
+ Status = gBS->OpenProtocol(\r
+ ParentDriverBuffer[HandleCount],\r
+ &gEfiComponentName2ProtocolGuid,\r
+ (VOID**)&CompName2,\r
+ gImageHandle,\r
+ NULL,\r
+ EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
+ if (EFI_ERROR(Status)) {\r
+ Status = gBS->OpenProtocol(\r
+ ParentDriverBuffer[HandleCount],\r
+ &gEfiComponentNameProtocolGuid,\r
+ (VOID**)&CompName2,\r
+ gImageHandle,\r
+ NULL,\r
+ EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
+ }\r
+\r
+ if (EFI_ERROR(Status)) {\r
+ continue;\r
+ }\r
+ if (Language == NULL) {\r
+ Lang = AllocateZeroPool(AsciiStrSize(CompName2->SupportedLanguages));\r
+ if (Lang == NULL) {\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
+ AsciiStrCpy(Lang, CompName2->SupportedLanguages);\r
+ TempChar = AsciiStrStr(Lang, ";");\r
+ if (TempChar != NULL){\r
+ *TempChar = CHAR_NULL;\r
+ }\r
+ } else {\r
+ Lang = AllocateZeroPool(AsciiStrSize(Language));\r
+ if (Lang == NULL) {\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
+ AsciiStrCpy(Lang, Language);\r
+ }\r
+ Status = CompName2->GetControllerName(CompName2, ParentControllerBuffer[LoopVar], DeviceHandle, Lang, &DeviceNameToReturn);\r
+ FreePool(Lang);\r
+ Lang = NULL;\r
+ if (!EFI_ERROR(Status) && DeviceNameToReturn != NULL) {\r
+ break;\r
+ }\r
+\r
+\r
+\r
+ }\r
+ SHELL_FREE_NON_NULL(ParentDriverBuffer);\r
+ if (!EFI_ERROR(Status) && DeviceNameToReturn != NULL) {\r
+ break;\r
+ }\r
+ }\r
+ SHELL_FREE_NON_NULL(ParentControllerBuffer);\r
+ }\r
+ //\r
+ // dont return on fail since we will try device path if that bit is on\r
+ //\r
if (DeviceNameToReturn != NULL){\r
ASSERT(BestDeviceName != NULL);\r
StrnCatGrow(BestDeviceName, NULL, DeviceNameToReturn, 0);\r
return (EFI_SUCCESS);\r
}\r
- //\r
- // dont return on fail since we will try device path if that bit is on\r
- //\r
}\r
if ((Flags & EFI_DEVICE_NAME_USE_DEVICE_PATH) != 0) {\r
Status = gBS->LocateProtocol(\r
{\r
EFI_SHELL_FILE_INFO *NewNode;\r
\r
- NewNode = AllocatePool(sizeof(EFI_SHELL_FILE_INFO));\r
+ NewNode = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));\r
if (NewNode == NULL) {\r
return (NULL);\r
}\r
NewNode->FullName = AllocateZeroPool(StrSize(Node->FullName));\r
\r
NewNode->FileName = AllocateZeroPool(StrSize(Node->FileName));\r
- NewNode->Info = AllocatePool((UINTN)Node->Info->Size);\r
+ NewNode->Info = AllocateZeroPool((UINTN)Node->Info->Size);\r
if ( NewNode->FullName == NULL\r
|| NewNode->FileName == NULL\r
|| NewNode->Info == NULL\r
} else {\r
NewShellNode->Handle = NULL;\r
if (*FileList == NULL) {\r
- *FileList = AllocatePool(sizeof(EFI_SHELL_FILE_INFO));\r
+ *FileList = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));\r
InitializeListHead(&((*FileList)->Link));\r
}\r
\r
Status = EFI_OUT_OF_RESOURCES;\r
}\r
if (*FileList == NULL) {\r
- *FileList = AllocatePool(sizeof(EFI_SHELL_FILE_INFO));\r
+ *FileList = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));\r
InitializeListHead(&((*FileList)->Link));\r
}\r
\r
RootDevicePath = NULL;\r
RootFileHandle = NULL;\r
MapName = NULL;\r
- PatternCopy = AllocatePool(StrSize(FilePattern));\r
+ PatternCopy = AllocateZeroPool(StrSize(FilePattern));\r
if (PatternCopy == NULL) {\r
return (EFI_OUT_OF_RESOURCES);\r
}\r
CHAR16 *Path2;\r
UINTN Path2Size;\r
CONST CHAR16 *CurDir;\r
+ BOOLEAN Found;\r
\r
ShellCommandCleanPath(Path);\r
\r
Path2Size = 0;\r
Path2 = NULL;\r
\r
- ASSERT(FileList != NULL);\r
- ASSERT(*FileList != NULL);\r
+ if (FileList == NULL || *FileList == NULL) {\r
+ return (EFI_INVALID_PARAMETER);\r
+ }\r
\r
if (*Path == L'.' && *(Path+1) == L'\\') {\r
- Path++;\r
+ Path+=2;\r
}\r
\r
//\r
StrnCatGrow(&Path2, &Path2Size, CurDir, 0);\r
if (*Path == L'\\') {\r
Path++;\r
+ while (ChopLastSlash(Path2)) ;\r
}\r
ASSERT((Path2 == NULL && Path2Size == 0) || (Path2 != NULL));\r
StrnCatGrow(&Path2, &Path2Size, Path, 0);\r
return (Status);\r
}\r
\r
+ Found = FALSE;\r
//\r
// We had no errors so open all the files (that are not already opened...)\r
//\r
){\r
if (ShellFileListItem->Status == 0 && ShellFileListItem->Handle == NULL) {\r
ShellFileListItem->Status = EfiShellOpenFileByName (ShellFileListItem->FullName, &ShellFileListItem->Handle, OpenMode);\r
+ Found = TRUE;\r
}\r
}\r
\r
+ if (!Found) {\r
+ return (EFI_NOT_FOUND);\r
+ }\r
return(EFI_SUCCESS);\r
}\r
\r
\r
Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlCNotifyHandle1);\r
Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlCNotifyHandle2);\r
+ Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlCNotifyHandle3);\r
+ Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlCNotifyHandle4);\r
+ Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlSNotifyHandle1);\r
+ Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlSNotifyHandle2);\r
+ Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlSNotifyHandle3);\r
+ Status = SimpleEx->UnregisterKeyNotify(SimpleEx, ShellInfoObject.CtrlSNotifyHandle4);\r
\r
return (Status);\r
}\r
IN EFI_KEY_DATA *KeyData\r
)\r
{\r
- if (ShellInfoObject.NewEfiShellProtocol->ExecutionBreak == NULL) {\r
- return (EFI_UNSUPPORTED);\r
+ EFI_INPUT_KEY Key;\r
+ UINTN EventIndex;\r
+// ShellPrintEx(-1,-1,L" <Notify> ");\r
+ if ((KeyData->Key.UnicodeChar == L'c' || KeyData->Key.UnicodeChar == 3) &&\r
+ (KeyData->KeyState.KeyShiftState == (EFI_SHIFT_STATE_VALID|EFI_LEFT_CONTROL_PRESSED) || KeyData->KeyState.KeyShiftState == (EFI_SHIFT_STATE_VALID|EFI_RIGHT_CONTROL_PRESSED))\r
+ ){ \r
+ if (ShellInfoObject.NewEfiShellProtocol->ExecutionBreak == NULL) {\r
+ return (EFI_UNSUPPORTED);\r
+ }\r
+ return (gBS->SignalEvent(ShellInfoObject.NewEfiShellProtocol->ExecutionBreak));\r
+ } else if ((KeyData->Key.UnicodeChar == L's') &&\r
+ (KeyData->KeyState.KeyShiftState == (EFI_SHIFT_STATE_VALID|EFI_LEFT_CONTROL_PRESSED) || KeyData->KeyState.KeyShiftState == (EFI_SHIFT_STATE_VALID|EFI_RIGHT_CONTROL_PRESSED))\r
+ ){ \r
+ //\r
+ // just get some key\r
+ //\r
+ gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);\r
+ gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
}\r
- return (gBS->SignalEvent(ShellInfoObject.NewEfiShellProtocol->ExecutionBreak));\r
+ return (EFI_SUCCESS);\r
}\r
\r
/**\r
Member functions of EFI_SHELL_PROTOCOL and functions for creation,\r
manipulation, and initialization of EFI_SHELL_PROTOCOL.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
InernalEfiShellStartMonitor(\r
VOID\r
);\r
+\r
+/**\r
+ Notification function for keystrokes.\r
+\r
+ @param[in] KeyData The key that was pressed.\r
+\r
+ @retval EFI_SUCCESS The operation was successful.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+NotificationFunction(\r
+ IN EFI_KEY_DATA *KeyData\r
+ );\r
#endif //_SHELL_PROTOCOL_HEADER_\r
\r
projects / mirror_edk2.git / commitdiff