git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@3 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			371 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			371 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*++
 | |
| 
 | |
| Copyright (c) 2006, Intel Corporation                                                         
 | |
| All rights reserved. This program and the accompanying materials                          
 | |
| are licensed and made available under the terms and conditions of the BSD License         
 | |
| which accompanies this distribution.  The full text of the license may be found at        
 | |
| http://opensource.org/licenses/bsd-license.php                                            
 | |
|                                                                                           
 | |
| THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     
 | |
| WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             
 | |
| 
 | |
| Module Name:
 | |
| 
 | |
|   BdsConsole.c
 | |
| 
 | |
| Abstract:
 | |
| 
 | |
|   BDS Lib functions which contain all the code to connect console device
 | |
| 
 | |
| --*/
 | |
| 
 | |
| EFI_STATUS
 | |
| BdsLibUpdateConsoleVariable (
 | |
|   IN  CHAR16                    *ConVarName,
 | |
|   IN  EFI_DEVICE_PATH_PROTOCOL  *CustomizedConDevicePath,
 | |
|   IN  EFI_DEVICE_PATH_PROTOCOL  *ExclusiveDevicePath
 | |
|   )
 | |
| /*++
 | |
| 
 | |
| Routine Description:
 | |
| 
 | |
|   This function update console variable based on ConVarName, it can 
 | |
|   add or remove one specific console device path from the variable
 | |
| 
 | |
| Arguments:
 | |
| 
 | |
|   ConVarName   - Console related variable name, ConIn, ConOut, ErrOut.
 | |
| 
 | |
|   CustomizedConDevicePath - The console device path which will be added to
 | |
|                             the console variable ConVarName, this parameter
 | |
|                             can not be multi-instance.
 | |
| 
 | |
|   ExclusiveDevicePath     - The console device path which will be removed
 | |
|                             from the console variable ConVarName, this
 | |
|                             parameter can not be multi-instance.
 | |
| 
 | |
| Returns:
 | |
| 
 | |
|   EFI_UNSUPPORTED         - Add or remove the same device path.
 | |
|   
 | |
|   EFI_SUCCESS             - Success add or remove the device path from 
 | |
|                             the console variable.
 | |
| 
 | |
| --*/
 | |
| {
 | |
|   EFI_STATUS                Status;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *VarConsole;
 | |
|   UINTN                     DevicePathSize;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *Instance;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *NewDevicePath;
 | |
| 
 | |
|   VarConsole      = NULL;
 | |
|   DevicePathSize  = 0;
 | |
|   NewDevicePath   = NULL;
 | |
|   Status          = EFI_UNSUPPORTED;
 | |
| 
 | |
|   //
 | |
|   // Notes: check the device path point, here should check
 | |
|   // with compare memory
 | |
|   //
 | |
|   if (CustomizedConDevicePath == ExclusiveDevicePath) {
 | |
|     return EFI_UNSUPPORTED;
 | |
|   }
 | |
|   //
 | |
|   // Delete the ExclusiveDevicePath from current default console
 | |
|   //
 | |
|   VarConsole = BdsLibGetVariableAndSize (
 | |
|                 ConVarName,
 | |
|                 &gEfiGlobalVariableGuid,
 | |
|                 &DevicePathSize
 | |
|                 );
 | |
| 
 | |
|   if (ExclusiveDevicePath != NULL && VarConsole != NULL) {
 | |
|     if (BdsLibMatchDevicePaths (VarConsole, ExclusiveDevicePath)) {
 | |
| 
 | |
|       Instance = GetNextDevicePathInstance (&VarConsole, &DevicePathSize);
 | |
| 
 | |
|       while (VarConsole != NULL) {
 | |
|         if (CompareMem (
 | |
|               Instance,
 | |
|               ExclusiveDevicePath,
 | |
|               DevicePathSize - sizeof (EFI_DEVICE_PATH_PROTOCOL)
 | |
|               ) == 0) {
 | |
|           //
 | |
|           // Remove the match part
 | |
|           //
 | |
|           NewDevicePath = AppendDevicePathInstance (NewDevicePath, VarConsole);
 | |
|           break;
 | |
|         } else {
 | |
|           //
 | |
|           // Continue the next instance
 | |
|           //
 | |
|           NewDevicePath = AppendDevicePathInstance (NewDevicePath, Instance);
 | |
|         }
 | |
| 
 | |
|         Instance = GetNextDevicePathInstance (&VarConsole, &DevicePathSize);
 | |
|       }
 | |
|       //
 | |
|       // Reset the console variable with new device path
 | |
|       //
 | |
|       gRT->SetVariable (
 | |
|             ConVarName,
 | |
|             &gEfiGlobalVariableGuid,
 | |
|             EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,
 | |
|             GetDevicePathSize (NewDevicePath),
 | |
|             NewDevicePath
 | |
|             );
 | |
|     }
 | |
|   }
 | |
|   //
 | |
|   // Try to append customized device path
 | |
|   //
 | |
|   VarConsole = BdsLibGetVariableAndSize (
 | |
|                 ConVarName,
 | |
|                 &gEfiGlobalVariableGuid,
 | |
|                 &DevicePathSize
 | |
|                 );
 | |
| 
 | |
|   if (CustomizedConDevicePath != NULL) {
 | |
|     if (!BdsLibMatchDevicePaths (VarConsole, CustomizedConDevicePath)) {
 | |
|       //
 | |
|       // In the first check, the default console variable will be null,
 | |
|       // just append current customized device path
 | |
|       //
 | |
|       VarConsole = AppendDevicePathInstance (VarConsole, CustomizedConDevicePath);
 | |
| 
 | |
|       //
 | |
|       // Update the variable of the default console
 | |
|       //
 | |
|       gRT->SetVariable (
 | |
|             ConVarName,
 | |
|             &gEfiGlobalVariableGuid,
 | |
|             EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,
 | |
|             GetDevicePathSize (VarConsole),
 | |
|             VarConsole
 | |
|             );
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   return EFI_SUCCESS;
 | |
| 
 | |
| }
 | |
| 
 | |
| EFI_STATUS
 | |
| BdsLibConnectConsoleVariable (
 | |
|   IN  CHAR16                 *ConVarName
 | |
|   )
 | |
| /*++
 | |
| 
 | |
| Routine Description:
 | |
| 
 | |
|   Connect the console device base on the variable ConVarName, if
 | |
|   device path of the ConVarName is multi-instance device path, if
 | |
|   anyone of the instances is connected success, then this function
 | |
|   will return success.
 | |
| 
 | |
| Arguments:
 | |
| 
 | |
|   ConVarName   - Console related variable name, ConIn, ConOut, ErrOut.
 | |
| 
 | |
| Returns:
 | |
| 
 | |
|   EFI_NOT_FOUND           - There is not any console devices connected success
 | |
|   
 | |
|   EFI_SUCCESS             - Success connect any one instance of the console
 | |
|                             device path base on the variable ConVarName.
 | |
| 
 | |
| --*/
 | |
| {
 | |
|   EFI_STATUS                Status;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *StartDevicePath;
 | |
|   UINTN                     VariableSize;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *Instance;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *Next;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *CopyOfDevicePath;
 | |
|   UINTN                     Size;
 | |
|   BOOLEAN                   DeviceExist;
 | |
| 
 | |
|   Status      = EFI_SUCCESS;
 | |
|   DeviceExist = FALSE;
 | |
| 
 | |
|   //
 | |
|   // Check if the console variable exist
 | |
|   //
 | |
|   StartDevicePath = BdsLibGetVariableAndSize (
 | |
|                       ConVarName,
 | |
|                       &gEfiGlobalVariableGuid,
 | |
|                       &VariableSize
 | |
|                       );
 | |
|   if (StartDevicePath == NULL) {
 | |
|     return EFI_UNSUPPORTED;
 | |
|   }
 | |
| 
 | |
|   CopyOfDevicePath = DuplicateDevicePath (StartDevicePath);
 | |
|   do {
 | |
|     //
 | |
|     // Check every instance of the console variable
 | |
|     //
 | |
|     Instance  = GetNextDevicePathInstance (&CopyOfDevicePath, &Size);
 | |
|     Next      = Instance;
 | |
|     while (!IsDevicePathEndType (Next)) {
 | |
|       Next = NextDevicePathNode (Next);
 | |
|     }
 | |
| 
 | |
|     SetDevicePathEndNode (Next);
 | |
| 
 | |
|     //
 | |
|     // Connect the instance device path
 | |
|     //
 | |
|     Status = BdsLibConnectDevicePath (Instance);
 | |
|     if (EFI_ERROR (Status)) {
 | |
|       //
 | |
|       // Delete the instance from the console varialbe
 | |
|       //
 | |
|       BdsLibUpdateConsoleVariable (ConVarName, NULL, Instance);
 | |
|     } else {
 | |
|       DeviceExist = TRUE;
 | |
|     }
 | |
| 
 | |
|   } while (CopyOfDevicePath != NULL);
 | |
| 
 | |
|   gBS->FreePool (StartDevicePath);
 | |
| 
 | |
|   if (DeviceExist == FALSE) {
 | |
|     return EFI_NOT_FOUND;
 | |
|   }
 | |
| 
 | |
|   return EFI_SUCCESS;
 | |
| }
 | |
| 
 | |
| VOID
 | |
| BdsLibConnectAllConsoles (
 | |
|   VOID
 | |
|   )
 | |
| /*++
 | |
| 
 | |
| Routine Description:
 | |
| 
 | |
|   This function will search every simpletxt devive in current system,
 | |
|   and make every simpletxt device as pertantial console device.
 | |
| 
 | |
| Arguments:
 | |
| 
 | |
|   None
 | |
| 
 | |
| Returns:
 | |
| 
 | |
|   None
 | |
| 
 | |
| --*/
 | |
| {
 | |
|   EFI_STATUS                Status;
 | |
|   UINTN                     Index;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *ConDevicePath;
 | |
|   UINTN                     HandleCount;
 | |
|   EFI_HANDLE                *HandleBuffer;
 | |
| 
 | |
|   Index         = 0;
 | |
|   HandleCount   = 0;
 | |
|   HandleBuffer  = NULL;
 | |
|   ConDevicePath = NULL;
 | |
| 
 | |
|   //
 | |
|   // Update all the console varables
 | |
|   //
 | |
|   Status = gBS->LocateHandleBuffer (
 | |
|                   ByProtocol,
 | |
|                   &gEfiSimpleTextInProtocolGuid,
 | |
|                   NULL,
 | |
|                   &HandleCount,
 | |
|                   &HandleBuffer
 | |
|                   );
 | |
|   for (Index = 0; Index < HandleCount; Index++) {
 | |
|     Status = gBS->HandleProtocol (
 | |
|                     HandleBuffer[Index],
 | |
|                     &gEfiDevicePathProtocolGuid,
 | |
|                     (VOID **) &ConDevicePath
 | |
|                     );
 | |
|     BdsLibUpdateConsoleVariable (L"ConIn", ConDevicePath, NULL);
 | |
|   }
 | |
| 
 | |
|   Status = gBS->LocateHandleBuffer (
 | |
|                   ByProtocol,
 | |
|                   &gEfiSimpleTextOutProtocolGuid,
 | |
|                   NULL,
 | |
|                   &HandleCount,
 | |
|                   &HandleBuffer
 | |
|                   );
 | |
|   for (Index = 0; Index < HandleCount; Index++) {
 | |
|     Status = gBS->HandleProtocol (
 | |
|                     HandleBuffer[Index],
 | |
|                     &gEfiDevicePathProtocolGuid,
 | |
|                     (VOID **) &ConDevicePath
 | |
|                     );
 | |
|     BdsLibUpdateConsoleVariable (L"ConOut", ConDevicePath, NULL);
 | |
|     BdsLibUpdateConsoleVariable (L"ErrOut", ConDevicePath, NULL);
 | |
|   }
 | |
|   //
 | |
|   // Connect all console variables
 | |
|   //
 | |
|   BdsLibConnectAllDefaultConsoles ();
 | |
| 
 | |
| }
 | |
| 
 | |
| EFI_STATUS
 | |
| BdsLibConnectAllDefaultConsoles (
 | |
|   VOID
 | |
|   )
 | |
| /*++
 | |
| 
 | |
| Routine Description:
 | |
| 
 | |
|   This function will connect console device base on the console 
 | |
|   device variable ConIn, ConOut and ErrOut.
 | |
| 
 | |
| Arguments:
 | |
| 
 | |
|   None
 | |
| 
 | |
| Returns:
 | |
| 
 | |
|   EFI_SUCCESS      - At least one of the ConIn and ConOut device have
 | |
|                      been connected success.
 | |
|                      
 | |
|   EFI_STATUS       - Return the status of BdsLibConnectConsoleVariable ().
 | |
| 
 | |
| --*/
 | |
| {
 | |
|   EFI_STATUS                Status;
 | |
|   EFI_DEVICE_PATH_PROTOCOL  *VarErrout;
 | |
|   UINTN                     DevicePathSize;
 | |
| 
 | |
|   //
 | |
|   // Connect all default console variables
 | |
|   //
 | |
|   Status = BdsLibConnectConsoleVariable (L"ConIn");
 | |
|   if (EFI_ERROR (Status)) {
 | |
|     return Status;
 | |
|   }
 | |
| 
 | |
|   Status = BdsLibConnectConsoleVariable (L"ConOut");
 | |
|   if (EFI_ERROR (Status)) {
 | |
|     return Status;
 | |
|   }
 | |
|   //
 | |
|   // Special treat the err out device, becaues the null
 | |
|   // err out var is legal.
 | |
|   //
 | |
|   VarErrout = BdsLibGetVariableAndSize (
 | |
|                 L"ErrOut",
 | |
|                 &gEfiGlobalVariableGuid,
 | |
|                 &DevicePathSize
 | |
|                 );
 | |
|   if (VarErrout != NULL) {
 | |
|     BdsLibConnectConsoleVariable (L"ErrOut");
 | |
|   }
 | |
| 
 | |
|   return EFI_SUCCESS;
 | |
| 
 | |
| }
 |