3 Copyright (c) 2004 - 2007, Intel Corporation
4 All rights reserved. This program and the accompanying materials
5 are licensed and made available under the terms and conditions of the BSD License
6 which accompanies this distribution. The full text of the license may be found at
7 http://opensource.org/licenses/bsd-license.php
9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
25 #include <Guid/SmBios.h>
26 #include <Protocol/Bis.h>
27 #include <Protocol/PxeBaseCode.h>
28 #include <Protocol/PxeBaseCodeCallBack.h>
29 #include <Protocol/NetworkInterfaceIdentifier.h>
30 #include <Protocol/SimpleNetwork.h>
31 #include <Protocol/LoadFile.h>
32 #include <Protocol/DevicePath.h>
33 #include <Protocol/Tcp.h>
35 #include <Library/DebugLib.h>
36 #include <Library/BaseMemoryLib.h>
37 #include <Library/MemoryAllocationLib.h>
38 #include <Library/UefiDriverEntryPoint.h>
39 #include <Library/UefiBootServicesTableLib.h>
40 #include <Library/BaseLib.h>
41 #include <Library/UefiLib.h>
43 #define CALLBACK_INTERVAL 100 // ten times a second
44 #define FILTER_BITS (EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP | \
45 EFI_PXE_BASE_CODE_IP_FILTER_BROADCAST | \
46 EFI_PXE_BASE_CODE_IP_FILTER_PROMISCUOUS | \
47 EFI_PXE_BASE_CODE_IP_FILTER_PROMISCUOUS_MULTICAST \
50 #define WAIT_TX_TIMEOUT 1000
52 #define SUPPORT_IPV6 0
54 #define PXE_BASECODE_DEVICE_SIGNATURE EFI_SIGNATURE_32 ('p', 'x', 'e', 'd')
57 // Determine the classes of IPv4 address
59 #define IS_CLASSA_IPADDR(x) ((((EFI_IP_ADDRESS*)x)->v4.Addr[0] & 0x80) == 0x00)
60 #define IS_CLASSB_IPADDR(x) ((((EFI_IP_ADDRESS*)x)->v4.Addr[0] & 0xc0) == 0x80)
61 #define IS_CLASSC_IPADDR(x) ((((EFI_IP_ADDRESS*)x)->v4.Addr[0] & 0xe0) == 0xc0)
62 #define IS_INADDR_UNICAST(x) ((IS_CLASSA_IPADDR(x) || IS_CLASSB_IPADDR(x) || IS_CLASSC_IPADDR(x)) && (((EFI_IP_ADDRESS*)x)->Addr[0] != 0) )
65 // Definitions for internet group management protocol version 2 message
67 // Per RFC 2236, November 1997
73 UINT8 MaxRespTime
; // in tenths of a second
74 UINT16 Checksum
; // ones complement of ones complement sum of
75 // 16 bit words of message
76 UINT32 GroupAddress
; // for general query, all systems group,
77 // for group specific, the group
80 #define IGMP_TYPE_QUERY 0x11
81 #define IGMP_TYPE_REPORT 0x16
82 #define IGMP_TYPE_V1REPORT 0x12
83 #define IGMP_TYPE_LEAVE_GROUP 0x17
85 #define IGMP_DEFAULT_MAX_RESPONSE_TIME 10 // 10 second default
88 #define MAX_MCAST_GROUPS 8 // most we allow ourselves to join at once
94 BOOLEAN ShowErrorMessages
;
96 EFI_PXE_BASE_CODE_PROTOCOL EfiBc
;
97 EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL
*CallbackProtocolPtr
;
100 EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL
*NiiPtr
;
101 EFI_SIMPLE_NETWORK_PROTOCOL
*SimpleNetwork
;
102 UINT8
*TransmitBufferPtr
;
103 UINT8
*ReceiveBufferPtr
;
104 EFI_PXE_BASE_CODE_FUNCTION Function
;
106 UINTN OldestArpEntry
;
107 UINTN MCastGroupCount
;
108 EFI_EVENT Igmpv1TimeoutEvent
;
109 BOOLEAN UseIgmpv1Reporting
;
110 EFI_EVENT IgmpGroupEvent
[MAX_MCAST_GROUPS
];
118 UINT32 MCastGroup
[MAX_MCAST_GROUPS
];
121 BOOLEAN GoodStationIp
;
124 VOID
*DhcpPacketBuffer
;
126 VOID
*BootServerReceiveBuffer
;
127 EFI_IP_ADDRESS ServerIp
;
133 VOID
*ReceiveBuffers
;
134 VOID
*TransmitBuffer
;
135 UINTN NumOffersReceived
;
139 // arrays for different types of offers
141 UINT8 ServerCount
[4];
142 UINT8 OfferCount
[4][MAX_OFFERS
];
145 UINT8 BinlProxies
[MAX_OFFERS
];
148 UINT8
*TftpAckBuffer
;
149 UINT8
*TftpErrorBuffer
;
150 IGMPV2_MESSAGE IgmpMessage
;
151 BOOLEAN BigBlkNumFlag
;
154 } PXE_BASECODE_DEVICE
;
159 #define DHCP_ONLY_IX 0
164 #define PXE_RND_PORT_LOW 2070
169 #define LOADFILE_DEVICE_SIGNATURE EFI_SIGNATURE_32 ('p', 'x', 'e', 'l')
174 EFI_LOAD_FILE_PROTOCOL LoadFile
;
175 PXE_BASECODE_DEVICE
*Private
;
178 #define EFI_BASE_CODE_DEV_FROM_THIS(a) CR (a, PXE_BASECODE_DEVICE, efi_bc, PXE_BASECODE_DEVICE_SIGNATURE);
180 #define EFI_BASE_CODE_DEV_FROM_TCP(a) CR (a, PXE_BASECODE_DEVICE, Tcp, PXE_BASECODE_DEVICE_SIGNATURE);
182 #define EFI_LOAD_FILE_DEV_FROM_THIS(a) CR (a, LOADFILE_DEVICE, LoadFile, LOADFILE_DEVICE_SIGNATURE)
186 PXE_BASECODE_DEVICE
*Private
,
187 BIS_APPLICATION_HANDLE
*BisAppHandle
,
188 EFI_BIS_DATA
**BisDataSigInfo
194 EFI_BIS_PROTOCOL
*Bis
,
195 BIS_APPLICATION_HANDLE BisAppHandle
,
196 EFI_BIS_DATA
*BisDataSigInfo
202 PXE_BASECODE_DEVICE
*Private
,
204 UINTN FileBufferLength
,
205 VOID
*CredentialBuffer
,
206 UINTN CredentialBufferLength
212 PXE_BASECODE_DEVICE
*Private
219 extern EFI_COMPONENT_NAME_PROTOCOL gPxeBcComponentName
;
220 extern EFI_COMPONENT_NAME2_PROTOCOL gPxeBcComponentName2
;
223 // //////////////////////////////////////////////////////////
229 Initialize the base code drivers and install the driver binding
231 Standard EFI Image Entry
233 @retval EFI_SUCCESS This driver was successfully bound
239 IN EFI_HANDLE ImageHandle
,
240 IN EFI_SYSTEM_TABLE
*SystemTable
247 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
255 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
262 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
263 IN BOOLEAN SortOffers
270 IN EFI_PXE_BASE_CODE_PROTOCOL
* This
,
274 IN EFI_PXE_BASE_CODE_DISCOVER_INFO
* Info OPTIONAL
281 IN EFI_PXE_BASE_CODE_PROTOCOL
* This
,
282 IN EFI_PXE_BASE_CODE_TFTP_OPCODE Operation
,
283 IN OUT VOID
*BufferPtr
,
284 IN BOOLEAN Overwrite
,
285 IN OUT UINT64
*BufferSize
,
286 IN UINTN
*BlockSize OPTIONAL
,
287 IN EFI_IP_ADDRESS
* ServerIp
,
289 IN EFI_PXE_BASE_CODE_MTFTP_INFO
* Info OPTIONAL
,
290 IN BOOLEAN DontUseBuffer
297 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
299 IN EFI_IP_ADDRESS
*DestIp
,
300 IN EFI_PXE_BASE_CODE_UDP_PORT
*DestPort
,
301 IN EFI_IP_ADDRESS
*GatewayIp
, OPTIONAL
302 IN EFI_IP_ADDRESS
*SrcIp
, OPTIONAL
303 IN OUT EFI_PXE_BASE_CODE_UDP_PORT
*SrcPort
, OPTIONAL
304 IN UINTN
*HeaderSize
, OPTIONAL
305 IN VOID
*HeaderPtr
, OPTIONAL
306 IN UINTN
*BufferSize
,
314 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
316 IN OUT EFI_IP_ADDRESS
*DestIp
, OPTIONAL
317 IN OUT EFI_PXE_BASE_CODE_UDP_PORT
*DestPort
, OPTIONAL
318 IN OUT EFI_IP_ADDRESS
*SrcIp
, OPTIONAL
319 IN OUT EFI_PXE_BASE_CODE_UDP_PORT
*SrcPort
, OPTIONAL
320 IN UINTN
*HeaderSize
, OPTIONAL
321 IN VOID
*HeaderPtr
, OPTIONAL
322 IN OUT UINTN
*BufferSize
,
330 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
332 IN UINT16
*UrgentPointer
,
333 IN UINT32
*SequenceNumber
,
334 IN UINT32
*AckNumber
,
335 IN UINT16
*HlenResCode
,
337 IN EFI_IP_ADDRESS
*DestIp
,
338 IN EFI_PXE_BASE_CODE_TCP_PORT
*DestPort
,
339 IN EFI_IP_ADDRESS
*GatewayIp
, OPTIONAL
340 IN EFI_IP_ADDRESS
*SrcIp
, OPTIONAL
341 IN OUT EFI_PXE_BASE_CODE_TCP_PORT
*SrcPort
, OPTIONAL
342 IN UINTN
*HeaderSize
, OPTIONAL
343 IN VOID
*HeaderPtr
, OPTIONAL
344 IN UINTN
*BufferSize
,
352 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
354 IN OUT EFI_IP_ADDRESS
*DestIp
, OPTIONAL
355 IN OUT EFI_PXE_BASE_CODE_TCP_PORT
*DestPort
, OPTIONAL
356 IN OUT EFI_IP_ADDRESS
*SrcIp
, OPTIONAL
357 IN OUT EFI_PXE_BASE_CODE_TCP_PORT
*SrcPort
, OPTIONAL
358 IN UINTN
*HeaderSize
, OPTIONAL
359 IN VOID
*HeaderPtr
, OPTIONAL
360 IN OUT UINTN
*BufferSize
,
368 IN EFI_PXE_BASE_CODE_PROTOCOL
* This
,
369 IN EFI_IP_ADDRESS
* IpAddr
,
370 IN EFI_MAC_ADDRESS
* MacAddr OPTIONAL
377 IN EFI_PXE_BASE_CODE_PROTOCOL
*This
,
378 IN EFI_PXE_BASE_CODE_IP_FILTER
*NewFilter
385 IN EFI_PXE_BASE_CODE_PROTOCOL
* This
,
386 IN BOOLEAN
*NewAutoArp
, OPTIONAL
387 IN BOOLEAN
*NewSendGUID
, OPTIONAL
388 IN UINT8
*NewTTL
, OPTIONAL
389 IN UINT8
*NewToS
, OPTIONAL
390 IN BOOLEAN
*NewMakeCallback OPTIONAL
397 IN EFI_PXE_BASE_CODE_PROTOCOL
* This
,
398 IN EFI_IP_ADDRESS
* NewStationIp
, OPTIONAL
399 IN EFI_IP_ADDRESS
* NewSubnetMask OPTIONAL
406 IN EFI_PXE_BASE_CODE_PROTOCOL
* This
,
407 BOOLEAN
*NewDhcpDiscoverValid
, OPTIONAL
408 BOOLEAN
*NewDhcpAckReceived
, OPTIONAL
409 BOOLEAN
*NewProxyOfferReceived
, OPTIONAL
410 BOOLEAN
*NewPxeDiscoverValid
, OPTIONAL
411 BOOLEAN
*NewPxeReplyReceived
, OPTIONAL
412 BOOLEAN
*NewPxeBisReplyReceived
, OPTIONAL
413 IN EFI_PXE_BASE_CODE_PACKET
* NewDhcpDiscover
, OPTIONAL
414 IN EFI_PXE_BASE_CODE_PACKET
* NewDhcpAck
, OPTIONAL
415 IN EFI_PXE_BASE_CODE_PACKET
* NewProxyOffer
, OPTIONAL
416 IN EFI_PXE_BASE_CODE_PACKET
* NewPxeDiscover
, OPTIONAL
417 IN EFI_PXE_BASE_CODE_PACKET
* NewPxeReply
, OPTIONAL
418 IN EFI_PXE_BASE_CODE_PACKET
* NewPxeBisReply OPTIONAL
425 IN EFI_LOAD_FILE_PROTOCOL
*This
,
426 IN EFI_DEVICE_PATH_PROTOCOL
*FilePath
,
427 IN BOOLEAN BootPolicy
,
428 IN OUT UINTN
*BufferSize
,
434 PxeBcLibGetSmbiosSystemGuidAndSerialNumber (
435 IN EFI_GUID
*SystemGuid
,
436 OUT CHAR8
**SystemSerialNumber
442 // Define SMBIOS tables.
446 UINT8 AnchorString
[4];
447 UINT8 EntryPointStructureChecksum
;
448 UINT8 EntryPointLength
;
451 UINT16 MaxStructureSize
;
452 UINT8 EntryPointRevision
;
453 UINT8 FormattedArea
[5];
454 UINT8 IntermediateAnchorString
[5];
455 UINT8 IntermediateChecksum
;
458 UINT16 NumberOfSmbiosStructures
;
459 UINT8 SmbiosBcdRevision
;
460 } SMBIOS_STRUCTURE_TABLE
;
463 // Please note that SMBIOS structures can be odd byte aligned since the
464 // unformated section of each record is a set of arbitrary size strings.
472 typedef UINT8 SMBIOS_STRING
;
476 SMBIOS_STRING Vendor
;
477 SMBIOS_STRING BiosVersion
;
478 UINT8 BiosSegment
[2];
479 SMBIOS_STRING BiosReleaseDate
;
481 UINT8 BiosCharacteristics
[8];
486 SMBIOS_STRING Manufacturer
;
487 SMBIOS_STRING ProductName
;
488 SMBIOS_STRING Version
;
489 SMBIOS_STRING SerialNumber
;
492 // always byte copy this data to prevent alignment faults!
501 SMBIOS_STRING Manufacturer
;
502 SMBIOS_STRING ProductName
;
503 SMBIOS_STRING Version
;
504 SMBIOS_STRING SerialNumber
;
509 SMBIOS_STRING Manufacturer
;
511 SMBIOS_STRING Version
;
512 SMBIOS_STRING SerialNumber
;
513 SMBIOS_STRING AssetTag
;
515 UINT8 PowerSupplyState
;
517 UINT8 SecurityStatus
;
525 UINT8 ProcessorFamily
;
526 SMBIOS_STRING ProcessorManufacture
;
527 UINT8 ProcessorId
[8];
528 SMBIOS_STRING ProcessorVersion
;
530 UINT8 ExternalClock
[2];
532 UINT8 CurrentSpeed
[2];
534 UINT8 ProcessorUpgrade
;
535 UINT8 L1CacheHandle
[2];
536 UINT8 L2CacheHandle
[2];
537 UINT8 L3CacheHandle
[2];
548 } SMBIOS_STRUCTURE_POINTER
;