git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@5020 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			387 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			387 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  BDS Lib functions which contain all the code to connect console device
 | 
						|
 | 
						|
Copyright (c) 2004 - 2008, Intel Corporation. <BR>
 | 
						|
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.
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
#include "InternalBdsLib.h"
 | 
						|
 | 
						|
BOOLEAN
 | 
						|
IsNvNeed (
 | 
						|
  IN CHAR16 *ConVarName
 | 
						|
  )
 | 
						|
{
 | 
						|
  CHAR16 *Ptr;
 | 
						|
 | 
						|
  Ptr = ConVarName;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the variable includes "Dev" at last, we consider
 | 
						|
  // it does not support NV attribute.
 | 
						|
  //
 | 
						|
  while (*Ptr) {
 | 
						|
    Ptr++;
 | 
						|
  }
 | 
						|
 | 
						|
  if ((*(Ptr-3) == 'D') && (*(Ptr-2) == 'e') && (*(Ptr-1) == 'v')) {
 | 
						|
    return FALSE;
 | 
						|
  } else {
 | 
						|
    return TRUE;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function update console variable based on ConVarName, it can
 | 
						|
  add or remove one specific console device path from the variable
 | 
						|
 | 
						|
  @param  ConVarName               Console related variable name, ConIn, ConOut,
 | 
						|
                                   ErrOut.
 | 
						|
  @param  CustomizedConDevicePath  The console device path which will be added to
 | 
						|
                                   the console variable ConVarName, this parameter
 | 
						|
                                   can not be multi-instance.
 | 
						|
  @param  ExclusiveDevicePath      The console device path which will be removed
 | 
						|
                                   from the console variable ConVarName, this
 | 
						|
                                   parameter can not be multi-instance.
 | 
						|
 | 
						|
  @retval EFI_UNSUPPORTED          Add or remove the same device path.
 | 
						|
  @retval EFI_SUCCESS              Success add or remove the device path from  the
 | 
						|
                                   console variable.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
BdsLibUpdateConsoleVariable (
 | 
						|
  IN  CHAR16                    *ConVarName,
 | 
						|
  IN  EFI_DEVICE_PATH_PROTOCOL  *CustomizedConDevicePath,
 | 
						|
  IN  EFI_DEVICE_PATH_PROTOCOL  *ExclusiveDevicePath
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_DEVICE_PATH_PROTOCOL  *VarConsole;
 | 
						|
  UINTN                     DevicePathSize;
 | 
						|
  EFI_DEVICE_PATH_PROTOCOL  *NewDevicePath;
 | 
						|
  EFI_DEVICE_PATH_PROTOCOL  *TempNewDevicePath;
 | 
						|
  UINT32                    Attributes;
 | 
						|
 | 
						|
  VarConsole      = NULL;
 | 
						|
  DevicePathSize  = 0;
 | 
						|
 | 
						|
  //
 | 
						|
  // 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
 | 
						|
                );
 | 
						|
 | 
						|
  //
 | 
						|
  // Initialize NewDevicePath
 | 
						|
  //
 | 
						|
  NewDevicePath  = VarConsole;
 | 
						|
 | 
						|
  //
 | 
						|
  // If ExclusiveDevicePath is even the part of the instance in VarConsole, delete it.
 | 
						|
  // In the end, NewDevicePath is the final device path.
 | 
						|
  //
 | 
						|
  if (ExclusiveDevicePath != NULL && VarConsole != NULL) {
 | 
						|
      NewDevicePath = BdsLibDelPartMatchInstance (VarConsole, ExclusiveDevicePath);
 | 
						|
  }
 | 
						|
  //
 | 
						|
  // Try to append customized device path to NewDevicePath.
 | 
						|
  //
 | 
						|
  if (CustomizedConDevicePath != NULL) {
 | 
						|
    if (!BdsLibMatchDevicePaths (NewDevicePath, CustomizedConDevicePath)) {
 | 
						|
      //
 | 
						|
      // Check if there is part of CustomizedConDevicePath in NewDevicePath, delete it.
 | 
						|
      //
 | 
						|
      NewDevicePath = BdsLibDelPartMatchInstance (NewDevicePath, CustomizedConDevicePath);
 | 
						|
      //
 | 
						|
      // In the first check, the default console variable will be _ModuleEntryPoint,
 | 
						|
      // just append current customized device path
 | 
						|
      //
 | 
						|
      TempNewDevicePath = NewDevicePath;
 | 
						|
      NewDevicePath = AppendDevicePathInstance (NewDevicePath, CustomizedConDevicePath);
 | 
						|
      SafeFreePool(TempNewDevicePath);
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // The attribute for ConInDev, ConOutDev and ErrOutDev does not include NV.
 | 
						|
  //
 | 
						|
  if (IsNvNeed(ConVarName)) {
 | 
						|
    //
 | 
						|
    // ConVarName has NV attribute.
 | 
						|
    //
 | 
						|
    Attributes = EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE;
 | 
						|
  } else {
 | 
						|
    //
 | 
						|
    // ConVarName does not have NV attribute.
 | 
						|
    //
 | 
						|
    Attributes = EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Finally, Update the variable of the default console by NewDevicePath
 | 
						|
  //
 | 
						|
  gRT->SetVariable (
 | 
						|
        ConVarName,
 | 
						|
        &gEfiGlobalVariableGuid,
 | 
						|
        Attributes,
 | 
						|
        GetDevicePathSize (NewDevicePath),
 | 
						|
        NewDevicePath
 | 
						|
        );
 | 
						|
 | 
						|
  if (VarConsole == NewDevicePath) {
 | 
						|
    SafeFreePool(VarConsole);
 | 
						|
  } else {
 | 
						|
    SafeFreePool(VarConsole);
 | 
						|
    SafeFreePool(NewDevicePath);
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  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.
 | 
						|
 | 
						|
  @param  ConVarName               Console related variable name, ConIn, ConOut,
 | 
						|
                                   ErrOut.
 | 
						|
 | 
						|
  @retval EFI_NOT_FOUND            There is not any console devices connected
 | 
						|
                                   success
 | 
						|
  @retval EFI_SUCCESS              Success connect any one instance of the console
 | 
						|
                                   device path base on the variable ConVarName.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
BdsLibConnectConsoleVariable (
 | 
						|
  IN  CHAR16                 *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 = StartDevicePath;
 | 
						|
  do {
 | 
						|
    //
 | 
						|
    // Check every instance of the console variable
 | 
						|
    //
 | 
						|
    Instance  = GetNextDevicePathInstance (&CopyOfDevicePath, &Size);
 | 
						|
    Next      = Instance;
 | 
						|
    while (!IsDevicePathEndType (Next)) {
 | 
						|
      Next = NextDevicePathNode (Next);
 | 
						|
    }
 | 
						|
 | 
						|
    SetDevicePathEndNode (Next);
 | 
						|
    //
 | 
						|
    // Check USB1.1 console
 | 
						|
    //
 | 
						|
    if ((DevicePathType (Instance) == MESSAGING_DEVICE_PATH) &&
 | 
						|
       ((DevicePathSubType (Instance) == MSG_USB_CLASS_DP)
 | 
						|
       || (DevicePathSubType (Instance) == MSG_USB_WWID_DP)
 | 
						|
       )) {
 | 
						|
      //
 | 
						|
      // Check the Usb console in Usb2.0 bus firstly, then Usb1.1 bus
 | 
						|
      //
 | 
						|
      Status = BdsLibConnectUsbDevByShortFormDP (PCI_CLASSC_PI_EHCI, Instance);
 | 
						|
      if (!EFI_ERROR (Status)) {
 | 
						|
        DeviceExist = TRUE;
 | 
						|
      }
 | 
						|
 | 
						|
      Status = BdsLibConnectUsbDevByShortFormDP (PCI_CLASSC_PI_UHCI, Instance);
 | 
						|
      if (!EFI_ERROR (Status)) {
 | 
						|
        DeviceExist = TRUE;
 | 
						|
      }
 | 
						|
    } else {
 | 
						|
      //
 | 
						|
      // 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;
 | 
						|
      }
 | 
						|
    }
 | 
						|
    SafeFreePool(Instance);
 | 
						|
  } while (CopyOfDevicePath != NULL);
 | 
						|
 | 
						|
  gBS->FreePool (StartDevicePath);
 | 
						|
 | 
						|
  if (DeviceExist == FALSE) {
 | 
						|
    return EFI_NOT_FOUND;
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function will search every simpletxt devive in current system,
 | 
						|
  and make every simpletxt device as pertantial console device.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
BdsLibConnectAllConsoles (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  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
 | 
						|
  //
 | 
						|
  gBS->LocateHandleBuffer (
 | 
						|
          ByProtocol,
 | 
						|
          &gEfiSimpleTextInProtocolGuid,
 | 
						|
          NULL,
 | 
						|
          &HandleCount,
 | 
						|
          &HandleBuffer
 | 
						|
          );
 | 
						|
  
 | 
						|
  for (Index = 0; Index < HandleCount; Index++) {
 | 
						|
    gBS->HandleProtocol (
 | 
						|
            HandleBuffer[Index],
 | 
						|
            &gEfiDevicePathProtocolGuid,
 | 
						|
            (VOID **) &ConDevicePath
 | 
						|
            );
 | 
						|
    BdsLibUpdateConsoleVariable (L"ConIn", ConDevicePath, NULL);
 | 
						|
  }
 | 
						|
 | 
						|
  SafeFreePool(HandleBuffer);
 | 
						|
 | 
						|
  gBS->LocateHandleBuffer (
 | 
						|
          ByProtocol,
 | 
						|
          &gEfiSimpleTextOutProtocolGuid,
 | 
						|
          NULL,
 | 
						|
          &HandleCount,
 | 
						|
          &HandleBuffer
 | 
						|
          );
 | 
						|
  for (Index = 0; Index < HandleCount; Index++) {
 | 
						|
    gBS->HandleProtocol (
 | 
						|
            HandleBuffer[Index],
 | 
						|
            &gEfiDevicePathProtocolGuid,
 | 
						|
            (VOID **) &ConDevicePath
 | 
						|
            );
 | 
						|
    BdsLibUpdateConsoleVariable (L"ConOut", ConDevicePath, NULL);
 | 
						|
    BdsLibUpdateConsoleVariable (L"ErrOut", ConDevicePath, NULL);
 | 
						|
  }
 | 
						|
 | 
						|
  SafeFreePool(HandleBuffer);
 | 
						|
 | 
						|
  //
 | 
						|
  // Connect all console variables
 | 
						|
  //
 | 
						|
  BdsLibConnectAllDefaultConsoles ();
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function will connect console device base on the console
 | 
						|
  device variable ConIn, ConOut and ErrOut.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS              At least one of the ConIn and ConOut device have
 | 
						|
                                   been connected success.
 | 
						|
  @retval EFI_STATUS               Return the status of
 | 
						|
                                   BdsLibConnectConsoleVariable ().
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
BdsLibConnectAllDefaultConsoles (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                Status;
 | 
						|
 | 
						|
  //
 | 
						|
  // Connect all default console variables
 | 
						|
  //
 | 
						|
 | 
						|
  //
 | 
						|
  // It seems impossible not to have any ConOut device on platform,
 | 
						|
  // so we check the status here.
 | 
						|
  //
 | 
						|
  Status = BdsLibConnectConsoleVariable (L"ConOut");
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Insert the performance probe for Console Out
 | 
						|
  //
 | 
						|
  PERF_START (NULL, "ConOut", "BDS", 1);
 | 
						|
  PERF_END (NULL, "ConOut", "BDS", 0);
 | 
						|
 | 
						|
  //
 | 
						|
  // Because possibly the platform is legacy free, in such case,
 | 
						|
  // ConIn devices (Serial Port and PS2 Keyboard ) does not exist,
 | 
						|
  // so we need not check the status.
 | 
						|
  //
 | 
						|
  BdsLibConnectConsoleVariable (L"ConIn");
 | 
						|
 | 
						|
  //
 | 
						|
  // The _ModuleEntryPoint err out var is legal.
 | 
						|
  //
 | 
						|
  BdsLibConnectConsoleVariable (L"ErrOut");
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
 | 
						|
}
 |