git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10282 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			318 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			318 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /** @file
 | |
|   Support routines for PxeBc.
 | |
| 
 | |
| Copyright (c) 2007 - 2010, 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;
 | |
|   }
 | |
|   ASSERT (SmbiosTable != NULL);
 | |
| 
 | |
|   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.
 | |
| 
 | |
| **/
 | |
| VOID
 | |
| EFIAPI
 | |
| 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 = PXEBC_DEFAULT_LIFETIME;
 | |
|   Udp4CfgData.TypeOfService  = DEFAULT_ToS;
 | |
|   Udp4CfgData.TimeToLive     = DEFAULT_TTL;
 | |
|   Udp4CfgData.AllowDuplicatePort = TRUE;
 | |
| 
 | |
|   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;
 | |
|     Length--;
 | |
|     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;
 | |
| }
 | |
| 
 |