git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7452 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			316 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			316 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  Support routines for PxeBc.
 | 
						|
  
 | 
						|
Copyright (c) 2007 - 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 "PxeBcImpl.h"
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function returns SMBIOS string given the string number.
 | 
						|
  
 | 
						|
  @param  Smbios              Pointer to SMBIOS structure
 | 
						|
  @param  StringNumber        String number to return. 0 is used to skip all
 | 
						|
                              strings and  point to the next SMBIOS structure.
 | 
						|
 | 
						|
  @return Pointer to string, or pointer to next SMBIOS strcuture if StringNumber == 0
 | 
						|
 | 
						|
**/
 | 
						|
CHAR8 *
 | 
						|
GetSmbiosString (
 | 
						|
  IN  SMBIOS_STRUCTURE_POINTER  *Smbios,
 | 
						|
  IN  UINT16                    StringNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  UINT16  Index;
 | 
						|
  CHAR8   *String;
 | 
						|
 | 
						|
  //
 | 
						|
  // Skip over formatted section
 | 
						|
  //
 | 
						|
  String = (CHAR8 *) (Smbios->Raw + Smbios->Hdr->Length);
 | 
						|
 | 
						|
  //
 | 
						|
  // Look through unformated section
 | 
						|
  //
 | 
						|
  for (Index = 1; Index <= StringNumber || StringNumber == 0; Index++) {
 | 
						|
    if (StringNumber == Index) {
 | 
						|
      return String;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Skip string
 | 
						|
    //
 | 
						|
    for (; *String != 0; String++)
 | 
						|
      ;
 | 
						|
    String++;
 | 
						|
 | 
						|
    if (*String == 0) {
 | 
						|
      //
 | 
						|
      // If double NULL then we are done.
 | 
						|
      //  Return pointer to next structure in Smbios.
 | 
						|
      //  if you pass in a 0 you will always get here
 | 
						|
      //
 | 
						|
      Smbios->Raw = (UINT8 *)++String;
 | 
						|
      return NULL;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return NULL;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function gets system guid and serial number from the smbios table.
 | 
						|
 | 
						|
  @param  SystemGuid          The pointer of returned system guid.
 | 
						|
  @param  SystemSerialNumber  The pointer of returned system serial number.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS         Successfully get the system guid and system serial
 | 
						|
                              number.
 | 
						|
  @retval EFI_NOT_FOUND       Not find the SMBIOS table.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
GetSmbiosSystemGuidAndSerialNumber (
 | 
						|
  IN  EFI_GUID  *SystemGuid,
 | 
						|
  OUT CHAR8     **SystemSerialNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                Status;
 | 
						|
  SMBIOS_TABLE_ENTRY_POINT  *SmbiosTable;
 | 
						|
  SMBIOS_STRUCTURE_POINTER  Smbios;
 | 
						|
  SMBIOS_STRUCTURE_POINTER  SmbiosEnd;
 | 
						|
  UINT16                    Index;
 | 
						|
 | 
						|
  Status = EfiGetSystemConfigurationTable (&gEfiSmbiosTableGuid, (VOID **) &SmbiosTable);
 | 
						|
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return EFI_NOT_FOUND;
 | 
						|
  }
 | 
						|
 | 
						|
  Smbios.Hdr    = (SMBIOS_STRUCTURE *) (UINTN) SmbiosTable->TableAddress;
 | 
						|
  SmbiosEnd.Raw = (UINT8 *) (UINTN) (SmbiosTable->TableAddress + SmbiosTable->TableLength);
 | 
						|
 | 
						|
  for (Index = 0; Index < SmbiosTable->TableLength; Index++) {
 | 
						|
    if (Smbios.Hdr->Type == 1) {
 | 
						|
      if (Smbios.Hdr->Length < 0x19) {
 | 
						|
        //
 | 
						|
        // Older version did not support Guid and Serial number
 | 
						|
        //
 | 
						|
        continue;
 | 
						|
      }
 | 
						|
      //
 | 
						|
      // SMBIOS tables are byte packed so we need to do a byte copy to
 | 
						|
      // prevend alignment faults on Itanium-based platform.
 | 
						|
      //
 | 
						|
      CopyMem (SystemGuid, &Smbios.Type1->Uuid, sizeof (EFI_GUID));
 | 
						|
      *SystemSerialNumber = GetSmbiosString (&Smbios, Smbios.Type1->SerialNumber);
 | 
						|
 | 
						|
      return EFI_SUCCESS;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Make Smbios point to the next record
 | 
						|
    //
 | 
						|
    GetSmbiosString (&Smbios, 0);
 | 
						|
 | 
						|
    if (Smbios.Raw >= SmbiosEnd.Raw) {
 | 
						|
      //
 | 
						|
      // SMBIOS 2.1 incorrectly stated the length of SmbiosTable as 0x1e.
 | 
						|
      // given this we must double check against the length of the structure.
 | 
						|
      //
 | 
						|
      return EFI_SUCCESS;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  The common notify function associated with various PxeBc events. 
 | 
						|
 | 
						|
  @param  Event     The event signaled.
 | 
						|
  @param  Context   The context.
 | 
						|
 | 
						|
  @return None
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
PxeBcCommonNotify (
 | 
						|
  IN EFI_EVENT           Event,
 | 
						|
  IN VOID                *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  *((BOOLEAN *) Context) = TRUE;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function initialize(or configure) the Udp4Write instance.
 | 
						|
  
 | 
						|
  @param  Udp4       Pointer to the EFI_UDP4_PROTOCOL instance.
 | 
						|
  @param  StationIp  Pointer to the station ip address.
 | 
						|
  @param  SubnetMask Pointer to the subnetmask of the station ip address.
 | 
						|
  @param  Gateway    Pointer to the gateway ip address.
 | 
						|
  @param  SrcPort    Pointer to the srouce port of the station.
 | 
						|
  
 | 
						|
  @retval EFI_SUCCESS           The configuration settings were set, changed, or reset successfully.
 | 
						|
  @retval EFI_NO_MAPPING        When using a default address, configuration (DHCP, BOOTP,
 | 
						|
                                RARP, etc.) is not finished yet.
 | 
						|
  @retval EFI_INVALID_PARAMETER One or more following conditions are TRUE:
 | 
						|
  @retval EFI_ALREADY_STARTED   The EFI UDPv4 Protocol instance is already started/configured
 | 
						|
                                and must be stopped/reset before it can be reconfigured.
 | 
						|
  @retval EFI_ACCESS_DENIED     UdpConfigData. AllowDuplicatePort is FALSE
 | 
						|
                                and UdpConfigData.StationPort is already used by
 | 
						|
                                other instance.
 | 
						|
  @retval EFI_OUT_OF_RESOURCES  The EFI UDPv4 Protocol driver cannot allocate memory for this
 | 
						|
                                EFI UDPv4 Protocol instance.
 | 
						|
  @retval EFI_DEVICE_ERROR      An unexpected network or system error occurred and this instance
 | 
						|
                                was not opened.
 | 
						|
  @retval Others                Please examine the function Udp4->Routes(Udp4, FALSE, &mZeroIp4Addr, &mZeroIp4Addr, Gateway) returns.
 | 
						|
  
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
PxeBcConfigureUdpWriteInstance (
 | 
						|
  IN EFI_UDP4_PROTOCOL  *Udp4,
 | 
						|
  IN EFI_IPv4_ADDRESS   *StationIp,
 | 
						|
  IN EFI_IPv4_ADDRESS   *SubnetMask,
 | 
						|
  IN EFI_IPv4_ADDRESS   *Gateway,
 | 
						|
  IN OUT UINT16         *SrcPort
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_UDP4_CONFIG_DATA  Udp4CfgData;
 | 
						|
  EFI_STATUS            Status;
 | 
						|
 | 
						|
  ZeroMem (&Udp4CfgData, sizeof (Udp4CfgData));
 | 
						|
 | 
						|
  Udp4CfgData.ReceiveTimeout = 1000;
 | 
						|
  Udp4CfgData.TypeOfService  = DEFAULT_ToS;
 | 
						|
  Udp4CfgData.TimeToLive     = DEFAULT_TTL;
 | 
						|
 | 
						|
  CopyMem (&Udp4CfgData.StationAddress, StationIp, sizeof (*StationIp));
 | 
						|
  CopyMem (&Udp4CfgData.SubnetMask, SubnetMask, sizeof (*SubnetMask));
 | 
						|
 | 
						|
  Udp4CfgData.StationPort    = *SrcPort;
 | 
						|
 | 
						|
  //
 | 
						|
  // Reset the instance.
 | 
						|
  //
 | 
						|
  Udp4->Configure (Udp4, NULL);
 | 
						|
 | 
						|
  Status = Udp4->Configure (Udp4, &Udp4CfgData);
 | 
						|
  if (!EFI_ERROR (Status) && (Gateway->Addr[0] != 0)) {
 | 
						|
    //
 | 
						|
    // basic configuration OK, need to add the default route entry
 | 
						|
    //
 | 
						|
    Status = Udp4->Routes (Udp4, FALSE, &mZeroIp4Addr, &mZeroIp4Addr, Gateway);
 | 
						|
    if (EFI_ERROR (Status)) {
 | 
						|
      //
 | 
						|
      // roll back
 | 
						|
      //
 | 
						|
      Udp4->Configure (Udp4, NULL);
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (!EFI_ERROR (Status) && (*SrcPort == 0)) {
 | 
						|
    Udp4->GetModeData (Udp4, &Udp4CfgData, NULL, NULL, NULL);
 | 
						|
    *SrcPort = Udp4CfgData.StationPort;
 | 
						|
  }
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Convert number to ASCII value.
 | 
						|
 | 
						|
  @param  Number              Numeric value to convert to decimal ASCII value.
 | 
						|
  @param  Buffer              Buffer to place ASCII version of the Number.
 | 
						|
  @param  Length              Length of Buffer.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
CvtNum (
 | 
						|
  IN UINTN  Number,
 | 
						|
  IN UINT8  *Buffer,
 | 
						|
  IN UINTN   Length
 | 
						|
  )
 | 
						|
{
 | 
						|
  UINTN Remainder;
 | 
						|
 | 
						|
  while (Length-- > 0) {
 | 
						|
    Remainder = Number % 10;
 | 
						|
    Number /= 10;
 | 
						|
    Buffer[Length] = (UINT8) ('0' + Remainder);
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Convert unsigned int number to decimal number.
 | 
						|
 | 
						|
  @param  Number   The unsigned int number will be converted.
 | 
						|
  @param  Buffer   Pointer to the buffer to store the decimal number after transform.
 | 
						|
 | 
						|
  @return the length of the number after transform.
 | 
						|
 | 
						|
**/
 | 
						|
UINTN
 | 
						|
UtoA10 (
 | 
						|
  IN UINTN Number,
 | 
						|
  IN CHAR8 *Buffer
 | 
						|
  )
 | 
						|
{
 | 
						|
  UINTN Index;
 | 
						|
  CHAR8 TempStr[64];
 | 
						|
 | 
						|
  Index           = 63;
 | 
						|
  TempStr[Index]  = 0;
 | 
						|
 | 
						|
  do {
 | 
						|
    Index--;
 | 
						|
    TempStr[Index]  = (CHAR8) ('0' + (Number % 10));
 | 
						|
    Number          = Number / 10;
 | 
						|
  } while (Number != 0);
 | 
						|
 | 
						|
  AsciiStrCpy (Buffer, &TempStr[Index]);
 | 
						|
 | 
						|
  return AsciiStrLen (Buffer);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Convert ASCII numeric string to a UINTN value.
 | 
						|
 | 
						|
  @param  Buffer  Pointer to the 8-byte unsigned int value.
 | 
						|
 | 
						|
  @return UINTN value of the ASCII string.
 | 
						|
 | 
						|
**/
 | 
						|
UINT64
 | 
						|
AtoU64 (
 | 
						|
  IN UINT8 *Buffer
 | 
						|
  )
 | 
						|
{
 | 
						|
  UINT64  Value;
 | 
						|
  UINT8   Character;
 | 
						|
 | 
						|
  Value = 0;
 | 
						|
  while ((Character = *Buffer++) != '\0') {
 | 
						|
    Value = MultU64x32 (Value, 10) + (Character - '0');
 | 
						|
  }
 | 
						|
 | 
						|
  return Value;
 | 
						|
}
 | 
						|
 |