[mirror_edk2.git] / StdLib / LibC / Uefi / Console.c

/** @file\r
  File abstractions of the console.\r
\r
  Manipulates EFI_FILE_PROTOCOL abstractions for stdin, stdout, stderr.\r
\r
  Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
  This program and the accompanying materials are licensed and made available under\r
  the terms and conditions of the BSD License that accompanies this distribution.\r
  The full text of the license may be found at\r
  http://opensource.org/licenses/bsd-license.php.\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
#include  <Uefi.h>\r
#include  <Library/BaseLib.h>\r
#include  <Library/BaseMemoryLib.h>\r
#include  <Library/MemoryAllocationLib.h>\r
#include  <Library/UefiBootServicesTableLib.h>\r
\r
#include  <LibConfig.h>\r
#include  <sys/EfiCdefs.h>\r
\r
#include  <stdio.h>\r
#include  <sys/fcntl.h>\r
#include  <MainData.h>\r
#include  <Efi/Console.h>\r
\r
static const CHAR16 * stdioNames[NUM_SPECIAL]   = {\r
  L"stdin:", L"stdout:", L"stderr:"\r
};\r
static const int stdioFlags[NUM_SPECIAL] = {\r
  O_RDONLY,             // stdin\r
  O_WRONLY,             // stdout\r
  O_WRONLY              // stderr\r
};\r
\r
static const int  stdioMode[NUM_SPECIAL] = {\r
  0444,   // stdin\r
  0222,   // stdout\r
  0222    // stderr\r
};\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConClose(\r
  IN EFI_FILE_PROTOCOL         *This\r
  )\r
{\r
  ConInstance    *Stream;\r
\r
  Stream = BASE_CR(This, ConInstance, Abstraction);\r
  // Quick check to see if Stream looks reasonable\r
  if(Stream->Cookie != 0x62416F49) {    // Cookie == 'IoAb'\r
    return RETURN_INVALID_PARAMETER;    // Looks like a bad This pointer\r
  }\r
  // Nothing to Flush.\r
  // Mark the Stream as closed.\r
  Stream->Dev = NULL;\r
\r
  return RETURN_SUCCESS;\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConDelete(\r
  IN EFI_FILE_PROTOCOL         *This\r
  )\r
{\r
  return RETURN_UNSUPPORTED;\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConRead(\r
  IN EFI_FILE_PROTOCOL         *This,\r
  IN OUT UINTN                 *BufferSize,\r
  OUT VOID                     *Buffer\r
  )\r
{\r
  EFI_SIMPLE_TEXT_INPUT_PROTOCOL   *Proto;\r
  ConInstance    *Stream;\r
  CHAR16         *OutPtr;\r
  EFI_INPUT_KEY   Key;\r
  UINTN           NumChar;\r
  UINTN           Edex;\r
  EFI_STATUS      Status;\r
  UINTN           i;\r
\r
  Stream = BASE_CR(This, ConInstance, Abstraction);\r
  // Quick check to see if Stream looks reasonable\r
  if(Stream->Cookie != 0x62416F49) {    // Cookie == 'IoAb'\r
    return RETURN_INVALID_PARAMETER;    // Looks like a bad This pointer\r
  }\r
  if(Stream->Dev == NULL) {\r
    // Can't read from an unopened Stream\r
    return RETURN_DEVICE_ERROR;\r
  }\r
  if(Stream != &gMD->StdIo[0]) {\r
    // Read only valid for stdin\r
    return RETURN_UNSUPPORTED;\r
  }\r
  // It looks like things are OK for trying to read\r
  // We will accumulate *BufferSize characters or until we encounter\r
  // an "activation" character.  Currently any control character.\r
  Proto = (EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)Stream->Dev;\r
  OutPtr = Buffer;\r
  NumChar = (*BufferSize - 1) / sizeof(CHAR16);\r
  i = 0;\r
  do {\r
    Status = gBS->WaitForEvent( 1, &Proto->WaitForKey, &Edex);\r
    if(Status != EFI_SUCCESS) {\r
      break;\r
    }\r
    Status = Proto->ReadKeyStroke(Proto, &Key);\r
    if(Status != EFI_SUCCESS) {\r
      break;\r
    }\r
    if(Key.ScanCode == SCAN_NULL) {\r
      *OutPtr++ = Key.UnicodeChar;\r
      ++i;\r
    }\r
    if(Key.UnicodeChar < 0x0020) {    // If a control character, or a scan code\r
      break;\r
    }\r
  } while(i < NumChar);\r
  *BufferSize = i * sizeof(CHAR16);\r
  return Status;\r
}\r
\r
/* Write a NULL terminated WCS to the EFI console.\r
\r
  @param[in,out]  BufferSize  Number of bytes in Buffer.  Set to zero if\r
                              the string couldn't be displayed.\r
  @parem[in]      Buffer      The WCS string to be displayed\r
\r
*/\r
static\r
EFI_STATUS\r
EFIAPI\r
ConWrite(\r
  IN EFI_FILE_PROTOCOL         *This,\r
  IN OUT UINTN                 *BufferSize,\r
  IN VOID                      *Buffer\r
  )\r
{\r
  EFI_STATUS      Status;\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL   *Proto;\r
  ConInstance    *Stream;\r
  CHAR16         *MyBuf;\r
  UINTN           NumChar;\r
\r
  Stream = BASE_CR(This, ConInstance, Abstraction);\r
  // Quick check to see if Stream looks reasonable\r
  if(Stream->Cookie != 0x62416F49) {    // Cookie == 'IoAb'\r
    return RETURN_INVALID_PARAMETER;    // Looks like a bad This pointer\r
  }\r
  if(Stream->Dev == NULL) {\r
    // Can't write to an unopened Stream\r
    return RETURN_DEVICE_ERROR;\r
  }\r
  if(Stream == &gMD->StdIo[0]) {\r
    // Write is not valid for stdin\r
    return RETURN_UNSUPPORTED;\r
  }\r
  // Everything is OK to do the write.\r
  Proto = (EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)Stream->Dev;\r
  MyBuf = (CHAR16 *)Buffer;\r
  NumChar = *BufferSize;\r
\r
  // Send MyBuf to the console\r
  Status = Proto->OutputString( Proto, MyBuf);\r
  // Depending on status, update BufferSize and return\r
  if(Status != EFI_SUCCESS) {\r
    *BufferSize = 0;    // We don't really know how many characters made it out\r
  }\r
  else {\r
    *BufferSize = NumChar;\r
    Stream->NumWritten += NumChar;\r
  }\r
  return Status;\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConGetPosition(\r
  IN EFI_FILE_PROTOCOL         *This,\r
  OUT UINT64                   *Position\r
  )\r
{\r
  ConInstance                       *Stream;\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL   *Proto;\r
  XYoffset                           CursorPos;\r
\r
  Stream = BASE_CR(This, ConInstance, Abstraction);\r
  // Quick check to see if Stream looks reasonable\r
  if(Stream->Cookie != 0x62416F49) {    // Cookie == 'IoAb'\r
    return RETURN_INVALID_PARAMETER;    // Looks like a bad This pointer\r
  }\r
  if(Stream == &gMD->StdIo[0]) {\r
    // This is stdin\r
    *Position = Stream->NumRead;\r
  }\r
  else {\r
    Proto = (EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)Stream->Dev;\r
    CursorPos.XYpos.Column = (UINT32)Proto->Mode->CursorColumn;\r
    CursorPos.XYpos.Row    = (UINT32)Proto->Mode->CursorRow;\r
    *Position = CursorPos.Offset;\r
  }\r
  return RETURN_SUCCESS;\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConSetPosition(\r
  IN EFI_FILE_PROTOCOL         *This,\r
  IN UINT64                     Position\r
  )\r
{\r
  ConInstance                       *Stream;\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL   *Proto;\r
  XYoffset                           CursorPos;\r
\r
  Stream = BASE_CR(This, ConInstance, Abstraction);\r
  // Quick check to see if Stream looks reasonable\r
  if(Stream->Cookie != 0x62416F49) {    // Cookie == 'IoAb'\r
    return RETURN_INVALID_PARAMETER;    // Looks like a bad This pointer\r
  }\r
  if(Stream->Dev == NULL) {\r
    // Can't write to an unopened Stream\r
    return RETURN_DEVICE_ERROR;\r
  }\r
  if(Stream == &gMD->StdIo[0]) {\r
    // Seek is not valid for stdin\r
    return RETURN_UNSUPPORTED;\r
  }\r
  // Everything is OK to do the final verification and "seek".\r
  Proto = (EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)Stream->Dev;\r
  CursorPos.Offset = Position;\r
\r
  return Proto->SetCursorPosition(Proto,\r
                                  (INTN)CursorPos.XYpos.Column,\r
                                  (INTN)CursorPos.XYpos.Row);\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConGetInfo(\r
  IN EFI_FILE_PROTOCOL         *This,\r
  IN EFI_GUID                  *InformationType,\r
  IN OUT UINTN                 *BufferSize,\r
  OUT VOID                     *Buffer\r
  )\r
{\r
  EFI_FILE_INFO       *InfoBuf;\r
  ConInstance         *Stream;\r
\r
  Stream = BASE_CR(This, ConInstance, Abstraction);\r
  // Quick check to see if Stream looks reasonable\r
  if((Stream->Cookie != 0x62416F49) ||    // Cookie == 'IoAb'\r
     (Buffer == NULL))\r
  {\r
    return RETURN_INVALID_PARAMETER;\r
  }\r
  if(*BufferSize < sizeof(EFI_FILE_INFO)) {\r
    *BufferSize = sizeof(EFI_FILE_INFO);\r
    return RETURN_BUFFER_TOO_SMALL;\r
  }\r
  // All of our parameters are correct, so fill in the information.\r
  (void) ZeroMem(Buffer, sizeof(EFI_FILE_INFO));\r
  InfoBuf = (EFI_FILE_INFO *)Buffer;\r
  InfoBuf->Size = sizeof(EFI_FILE_INFO);\r
  InfoBuf->FileSize = 1;\r
  InfoBuf->PhysicalSize = 1;\r
  *BufferSize = sizeof(EFI_FILE_INFO);\r
\r
  return RETURN_SUCCESS;\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConSetInfo(\r
  IN EFI_FILE_PROTOCOL         *This,\r
  IN EFI_GUID                  *InformationType,\r
  IN UINTN                      BufferSize,\r
  IN VOID                      *Buffer\r
  )\r
{\r
  return RETURN_UNSUPPORTED;\r
}\r
\r
static\r
EFI_STATUS\r
EFIAPI\r
ConFlush(\r
  IN EFI_FILE_PROTOCOL         *This\r
  )\r
{\r
  return RETURN_SUCCESS;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
ConOpen(\r
  IN  EFI_FILE_PROTOCOL        *This,\r
  OUT EFI_FILE_PROTOCOL       **NewHandle,\r
  IN  CHAR16                   *FileName,\r
  IN  UINT64                    OpenMode,   // Ignored\r
  IN  UINT64                    Attributes  // Ignored\r
  )\r
{\r
  ConInstance                      *Stream;\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *Proto;\r
  UINTN                             Columns;\r
  UINTN                             Rows;\r
  int                               i;\r
\r
  if((NewHandle == NULL)  ||\r
     (FileName  == NULL))\r
  {\r
    return RETURN_INVALID_PARAMETER;\r
  }\r
  // Process FileName\r
  for(i = 0; i < NUM_SPECIAL; ++i) {\r
    if(StrCmp( stdioNames[i], FileName) == 0) {\r
      break;\r
    }\r
  }\r
  if(i >= NUM_SPECIAL) {\r
    return RETURN_NO_MAPPING;\r
  }\r
  // Get pointer to instance.\r
  Stream = &gMD->StdIo[i];\r
  if(Stream->Dev == NULL) {\r
    // If this stream has been closed, then\r
    // Initialize instance.\r
    Stream->Cookie      = 0x62416F49;\r
    Stream->NumRead     = 0;\r
    Stream->NumWritten  = 0;\r
    switch(i) {\r
      case 0:\r
        Stream->Dev = gST->ConIn;\r
        break;\r
      case 1:\r
        Stream->Dev = gST->ConOut;\r
        break;\r
      case 2:\r
        if(gST->StdErr == NULL) {\r
          Stream->Dev = gST->ConOut;\r
        }\r
        else {\r
          Stream->Dev = gST->StdErr;\r
        }\r
        break;\r
      default:\r
        return RETURN_VOLUME_CORRUPTED;     // This is a "should never happen" case.\r
    }\r
    Stream->Abstraction.Revision      = 0x00010000;\r
    Stream->Abstraction.Open          = &ConOpen;\r
    Stream->Abstraction.Close         = &ConClose;\r
    Stream->Abstraction.Delete        = &ConDelete;\r
    Stream->Abstraction.Read          = &ConRead;\r
    Stream->Abstraction.Write         = &ConWrite;\r
    Stream->Abstraction.GetPosition   = &ConGetPosition;\r
    Stream->Abstraction.SetPosition   = &ConSetPosition;\r
    Stream->Abstraction.GetInfo       = &ConGetInfo;\r
    Stream->Abstraction.SetInfo       = &ConSetInfo;\r
    Stream->Abstraction.Flush         = &ConFlush;\r
    // Get additional information if this is an Output stream\r
    if(i != 0) {\r
      Proto = Stream->Dev;\r
      Stream->ConOutMode = Proto->Mode->Mode;\r
      if( Proto->QueryMode(Proto, Stream->ConOutMode, &Columns, &Rows) != RETURN_SUCCESS) {\r
        Stream->Dev = NULL;   // Mark this stream as closed\r
        return RETURN_INVALID_PARAMETER;\r
      }\r
      Stream->MaxConXY.XYpos.Column = (UINT32)Columns;\r
      Stream->MaxConXY.XYpos.Row    = (UINT32)Rows;\r
    }\r
  }\r
  // Save NewHandle and return.\r
  *NewHandle = &Stream->Abstraction;\r
\r
  return RETURN_SUCCESS;\r
}\r
Commit	Line	Data
2aa62f2b	1	/** @file\r
	2	File abstractions of the console.\r
	3	\r
	4	Manipulates EFI_FILE_PROTOCOL abstractions for stdin, stdout, stderr.\r
	5	\r
	6	Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
	7	This program and the accompanying materials are licensed and made available under\r
	8	the terms and conditions of the BSD License that accompanies this distribution.\r
	9	The full text of the license may be found at\r
	10	http://opensource.org/licenses/bsd-license.php.\r
	11	\r
	12	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	13	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	14	\r
	15	**/\r
	16	#include <Uefi.h>\r
	17	#include <Library/BaseLib.h>\r
	18	#include <Library/BaseMemoryLib.h>\r
	19	#include <Library/MemoryAllocationLib.h>\r
	20	#include <Library/UefiBootServicesTableLib.h>\r
	21	\r
	22	#include <LibConfig.h>\r
	23	#include <sys/EfiCdefs.h>\r
	24	\r
	25	#include <stdio.h>\r
	26	#include <sys/fcntl.h>\r
	27	#include <MainData.h>\r
	28	#include <Efi/Console.h>\r
	29	\r
	30	static const CHAR16 * stdioNames[NUM_SPECIAL] = {\r
	31	L"stdin:", L"stdout:", L"stderr:"\r
	32	};\r
	33	static const int stdioFlags[NUM_SPECIAL] = {\r
	34	O_RDONLY, // stdin\r
	35	O_WRONLY, // stdout\r
	36	O_WRONLY // stderr\r
	37	};\r
	38	\r
	39	static const int stdioMode[NUM_SPECIAL] = {\r
	40	0444, // stdin\r
	41	0222, // stdout\r
	42	0222 // stderr\r
	43	};\r
	44	\r
	45	static\r
	46	EFI_STATUS\r
	47	EFIAPI\r
	48	ConClose(\r
	49	IN EFI_FILE_PROTOCOL *This\r
	50	)\r
	51	{\r
	52	ConInstance *Stream;\r
	53	\r
	54	Stream = BASE_CR(This, ConInstance, Abstraction);\r
	55	// Quick check to see if Stream looks reasonable\r
	56	if(Stream->Cookie != 0x62416F49) { // Cookie == 'IoAb'\r
	57	return RETURN_INVALID_PARAMETER; // Looks like a bad This pointer\r
	58	}\r
	59	// Nothing to Flush.\r
	60	// Mark the Stream as closed.\r
	61	Stream->Dev = NULL;\r
	62	\r
	63	return RETURN_SUCCESS;\r
	64	}\r
65	\r
66	static\r
67	EFI_STATUS\r
68	EFIAPI\r
69	ConDelete(\r
70	IN EFI_FILE_PROTOCOL *This\r
71	)\r
72	{\r
73	return RETURN_UNSUPPORTED;\r
74	}\r
75	\r
76	static\r
77	EFI_STATUS\r
78	EFIAPI\r
79	ConRead(\r
80	IN EFI_FILE_PROTOCOL *This,\r
81	IN OUT UINTN *BufferSize,\r
82	OUT VOID *Buffer\r
83	)\r
84	{\r
85	EFI_SIMPLE_TEXT_INPUT_PROTOCOL *Proto;\r
86	ConInstance *Stream;\r
87	CHAR16 *OutPtr;\r
88	EFI_INPUT_KEY Key;\r
89	UINTN NumChar;\r
90	UINTN Edex;\r
91	EFI_STATUS Status;\r
92	UINTN i;\r
93	\r
94	Stream = BASE_CR(This, ConInstance, Abstraction);\r
95	// Quick check to see if Stream looks reasonable\r
96	if(Stream->Cookie != 0x62416F49) { // Cookie == 'IoAb'\r
97	return RETURN_INVALID_PARAMETER; // Looks like a bad This pointer\r
98	}\r
99	if(Stream->Dev == NULL) {\r
100	// Can't read from an unopened Stream\r
101	return RETURN_DEVICE_ERROR;\r
102	}\r
103	if(Stream != &gMD->StdIo[0]) {\r
104	// Read only valid for stdin\r
105	return RETURN_UNSUPPORTED;\r
106	}\r
107	// It looks like things are OK for trying to read\r
108	// We will accumulate *BufferSize characters or until we encounter\r
109	// an "activation" character. Currently any control character.\r
110	Proto = (EFI_SIMPLE_TEXT_INPUT_PROTOCOL *)Stream->Dev;\r
111	OutPtr = Buffer;\r
112	NumChar = (*BufferSize - 1) / sizeof(CHAR16);\r
113	i = 0;\r
114	do {\r
115	Status = gBS->WaitForEvent( 1, &Proto->WaitForKey, &Edex);\r
116	if(Status != EFI_SUCCESS) {\r
117	break;\r
118	}\r
119	Status = Proto->ReadKeyStroke(Proto, &Key);\r
120	if(Status != EFI_SUCCESS) {\r
121	break;\r
122	}\r
123	if(Key.ScanCode == SCAN_NULL) {\r
124	*OutPtr++ = Key.UnicodeChar;\r
125	++i;\r
126	}\r
127	if(Key.UnicodeChar < 0x0020) { // If a control character, or a scan code\r
128	break;\r
129	}\r
130	} while(i < NumChar);\r
131	BufferSize = i sizeof(CHAR16);\r
132	return Status;\r
133	}\r
134	\r
135	/* Write a NULL terminated WCS to the EFI console.\r
136	\r
137	@param[in,out] BufferSize Number of bytes in Buffer. Set to zero if\r
138	the string couldn't be displayed.\r
139	@parem[in] Buffer The WCS string to be displayed\r
140	\r
141	*/\r
142	static\r
143	EFI_STATUS\r
144	EFIAPI\r
145	ConWrite(\r
146	IN EFI_FILE_PROTOCOL *This,\r
147	IN OUT UINTN *BufferSize,\r
148	IN VOID *Buffer\r
149	)\r
150	{\r
151	EFI_STATUS Status;\r
152	EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *Proto;\r
153	ConInstance *Stream;\r
154	CHAR16 *MyBuf;\r
155	UINTN NumChar;\r
156	\r
157	Stream = BASE_CR(This, ConInstance, Abstraction);\r
158	// Quick check to see if Stream looks reasonable\r
159	if(Stream->Cookie != 0x62416F49) { // Cookie == 'IoAb'\r
160	return RETURN_INVALID_PARAMETER; // Looks like a bad This pointer\r
161	}\r
162	if(Stream->Dev == NULL) {\r
163	// Can't write to an unopened Stream\r
164	return RETURN_DEVICE_ERROR;\r
165	}\r
166	if(Stream == &gMD->StdIo[0]) {\r
167	// Write is not valid for stdin\r
168	return RETURN_UNSUPPORTED;\r
169	}\r
170	// Everything is OK to do the write.\r
171	Proto = (EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)Stream->Dev;\r
172	MyBuf = (CHAR16 *)Buffer;\r
173	NumChar = *BufferSize;\r
174	\r
175	// Send MyBuf to the console\r
176	Status = Proto->OutputString( Proto, MyBuf);\r
177	// Depending on status, update BufferSize and return\r
178	if(Status != EFI_SUCCESS) {\r
179	*BufferSize = 0; // We don't really know how many characters made it out\r
180	}\r
181	else {\r
182	*BufferSize = NumChar;\r
183	Stream->NumWritten += NumChar;\r
184	}\r
185	return Status;\r
186	}\r
187	\r
188	static\r
189	EFI_STATUS\r
190	EFIAPI\r
191	ConGetPosition(\r
192	IN EFI_FILE_PROTOCOL *This,\r
193	OUT UINT64 *Position\r
194	)\r
195	{\r
196	ConInstance *Stream;\r
197	EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *Proto;\r
198	XYoffset CursorPos;\r
199	\r
200	Stream = BASE_CR(This, ConInstance, Abstraction);\r
201	// Quick check to see if Stream looks reasonable\r
202	if(Stream->Cookie != 0x62416F49) { // Cookie == 'IoAb'\r
203	return RETURN_INVALID_PARAMETER; // Looks like a bad This pointer\r
204	}\r
205	if(Stream == &gMD->StdIo[0]) {\r
206	// This is stdin\r
207	*Position = Stream->NumRead;\r
208	}\r
209	else {\r
210	Proto = (EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)Stream->Dev;\r
211	CursorPos.XYpos.Column = (UINT32)Proto->Mode->CursorColumn;\r
212	CursorPos.XYpos.Row = (UINT32)Proto->Mode->CursorRow;\r
213	*Position = CursorPos.Offset;\r
214	}\r
215	return RETURN_SUCCESS;\r
216	}\r
217	\r
218	static\r
219	EFI_STATUS\r
220	EFIAPI\r
221	ConSetPosition(\r
222	IN EFI_FILE_PROTOCOL *This,\r
223	IN UINT64 Position\r
224	)\r
225	{\r
226	ConInstance *Stream;\r
227	EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *Proto;\r
228	XYoffset CursorPos;\r
229	\r
230	Stream = BASE_CR(This, ConInstance, Abstraction);\r
231	// Quick check to see if Stream looks reasonable\r
232	if(Stream->Cookie != 0x62416F49) { // Cookie == 'IoAb'\r
233	return RETURN_INVALID_PARAMETER; // Looks like a bad This pointer\r
234	}\r
235	if(Stream->Dev == NULL) {\r
236	// Can't write to an unopened Stream\r
237	return RETURN_DEVICE_ERROR;\r
238	}\r
239	if(Stream == &gMD->StdIo[0]) {\r
240	// Seek is not valid for stdin\r
241	return RETURN_UNSUPPORTED;\r
242	}\r
243	// Everything is OK to do the final verification and "seek".\r
244	Proto = (EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *)Stream->Dev;\r
245	CursorPos.Offset = Position;\r
246	\r
247	return Proto->SetCursorPosition(Proto,\r
248	(INTN)CursorPos.XYpos.Column,\r
249	(INTN)CursorPos.XYpos.Row);\r
250	}\r
251	\r
252	static\r
253	EFI_STATUS\r
254	EFIAPI\r
255	ConGetInfo(\r
256	IN EFI_FILE_PROTOCOL *This,\r
257	IN EFI_GUID *InformationType,\r
258	IN OUT UINTN *BufferSize,\r
259	OUT VOID *Buffer\r
260	)\r
261	{\r
262	EFI_FILE_INFO *InfoBuf;\r
263	ConInstance *Stream;\r
264	\r
265	Stream = BASE_CR(This, ConInstance, Abstraction);\r
266	// Quick check to see if Stream looks reasonable\r
267	if((Stream->Cookie != 0x62416F49) \|\| // Cookie == 'IoAb'\r
268	(Buffer == NULL))\r
269	{\r
270	return RETURN_INVALID_PARAMETER;\r
271	}\r
272	if(*BufferSize < sizeof(EFI_FILE_INFO)) {\r
273	*BufferSize = sizeof(EFI_FILE_INFO);\r
274	return RETURN_BUFFER_TOO_SMALL;\r
275	}\r
276	// All of our parameters are correct, so fill in the information.\r
277	(void) ZeroMem(Buffer, sizeof(EFI_FILE_INFO));\r
278	InfoBuf = (EFI_FILE_INFO *)Buffer;\r
279	InfoBuf->Size = sizeof(EFI_FILE_INFO);\r
280	InfoBuf->FileSize = 1;\r
281	InfoBuf->PhysicalSize = 1;\r
282	*BufferSize = sizeof(EFI_FILE_INFO);\r
283	\r
284	return RETURN_SUCCESS;\r
285	}\r
286	\r
287	static\r
288	EFI_STATUS\r
289	EFIAPI\r
290	ConSetInfo(\r
291	IN EFI_FILE_PROTOCOL *This,\r
292	IN EFI_GUID *InformationType,\r
293	IN UINTN BufferSize,\r
294	IN VOID *Buffer\r
295	)\r
296	{\r
297	return RETURN_UNSUPPORTED;\r
298	}\r
299	\r
300	static\r
301	EFI_STATUS\r
302	EFIAPI\r
303	ConFlush(\r
304	IN EFI_FILE_PROTOCOL *This\r
305	)\r
306	{\r
307	return RETURN_SUCCESS;\r
308	}\r
309	\r
310	EFI_STATUS\r
311	EFIAPI\r
312	ConOpen(\r
313	IN EFI_FILE_PROTOCOL *This,\r
314	OUT EFI_FILE_PROTOCOL **NewHandle,\r
315	IN CHAR16 *FileName,\r
316	IN UINT64 OpenMode, // Ignored\r
317	IN UINT64 Attributes // Ignored\r
318	)\r
319	{\r
320	ConInstance *Stream;\r
321	EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *Proto;\r
322	UINTN Columns;\r
323	UINTN Rows;\r
324	int i;\r
325	\r
326	if((NewHandle == NULL) \|\|\r
327	(FileName == NULL))\r
328	{\r
329	return RETURN_INVALID_PARAMETER;\r
330	}\r
331	// Process FileName\r
332	for(i = 0; i < NUM_SPECIAL; ++i) {\r
333	if(StrCmp( stdioNames[i], FileName) == 0) {\r
334	break;\r
335	}\r
336	}\r
337	if(i >= NUM_SPECIAL) {\r
338	return RETURN_NO_MAPPING;\r
339	}\r
340	// Get pointer to instance.\r
341	Stream = &gMD->StdIo[i];\r
342	if(Stream->Dev == NULL) {\r
343	// If this stream has been closed, then\r
344	// Initialize instance.\r
345	Stream->Cookie = 0x62416F49;\r
346	Stream->NumRead = 0;\r
347	Stream->NumWritten = 0;\r
348	switch(i) {\r
349	case 0:\r
350	Stream->Dev = gST->ConIn;\r
351	break;\r
352	case 1:\r
353	Stream->Dev = gST->ConOut;\r
354	break;\r
355	case 2:\r
356	if(gST->StdErr == NULL) {\r
357	Stream->Dev = gST->ConOut;\r
358	}\r
359	else {\r
360	Stream->Dev = gST->StdErr;\r
361	}\r
362	break;\r
363	default:\r
364	return RETURN_VOLUME_CORRUPTED; // This is a "should never happen" case.\r
365	}\r
366	Stream->Abstraction.Revision = 0x00010000;\r
367	Stream->Abstraction.Open = &ConOpen;\r
368	Stream->Abstraction.Close = &ConClose;\r
369	Stream->Abstraction.Delete = &ConDelete;\r
370	Stream->Abstraction.Read = &ConRead;\r
371	Stream->Abstraction.Write = &ConWrite;\r
372	Stream->Abstraction.GetPosition = &ConGetPosition;\r
373	Stream->Abstraction.SetPosition = &ConSetPosition;\r
374	Stream->Abstraction.GetInfo = &ConGetInfo;\r
375	Stream->Abstraction.SetInfo = &ConSetInfo;\r
376	Stream->Abstraction.Flush = &ConFlush;\r
377	// Get additional information if this is an Output stream\r
378	if(i != 0) {\r
379	Proto = Stream->Dev;\r
380	Stream->ConOutMode = Proto->Mode->Mode;\r
381	if( Proto->QueryMode(Proto, Stream->ConOutMode, &Columns, &Rows) != RETURN_SUCCESS) {\r
382	Stream->Dev = NULL; // Mark this stream as closed\r
383	return RETURN_INVALID_PARAMETER;\r
384	}\r
385	Stream->MaxConXY.XYpos.Column = (UINT32)Columns;\r
386	Stream->MaxConXY.XYpos.Row = (UINT32)Rows;\r
387	}\r
388	}\r
389	// Save NewHandle and return.\r
390	*NewHandle = &Stream->Abstraction;\r
391	\r
392	return RETURN_SUCCESS;\r
393	}\r