[mirror_edk2.git] / MdeModulePkg / Universal / Network / UefiPxeBcDxe / PxeBcImpl.h

/** @file\r
\r
Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution.  The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#ifndef __EFI_PXEBC_IMPL_H__\r
#define __EFI_PXEBC_IMPL_H__\r
\r
\r
typedef struct _PXEBC_PRIVATE_DATA  PXEBC_PRIVATE_DATA;\r
\r
#include <Uefi.h>\r
\r
#include <Guid/SmBios.h>\r
#include <IndustryStandard/SmBios.h>\r
#include <IndustryStandard/Dhcp.h>\r
#include <Protocol/Dhcp4.h>\r
#include <Protocol/PxeBaseCode.h>\r
#include <Protocol/Mtftp4.h>\r
#include <Protocol/Udp4.h>\r
#include <Protocol/LoadFile.h>\r
#include <Protocol/NetworkInterfaceIdentifier.h>\r
#include <Protocol/PxeBaseCodeCallBack.h>\r
#include <Protocol/Arp.h>\r
#include <Protocol/Ip4.h>\r
#include <Protocol/Ip4Config2.h>\r
\r
#include <Library/DebugLib.h>\r
#include <Library/DevicePathLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiDriverEntryPoint.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/UefiLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/NetLib.h>\r
#include <Library/DpcLib.h>\r
#include <Library/PcdLib.h>\r
\r
#include "PxeBcDriver.h"\r
#include "PxeBcDhcp.h"\r
#include "PxeBcMtftp.h"\r
#include "PxeBcSupport.h"\r
\r
#define PXEBC_PRIVATE_DATA_SIGNATURE  SIGNATURE_32 ('P', 'X', 'E', 'P')\r
#define PXEBC_MTFTP_TIMEOUT                4\r
#define PXEBC_MTFTP_RETRIES                6\r
#define PXEBC_DEFAULT_UDP_OVERHEAD_SIZE    8\r
#define PXEBC_DEFAULT_TFTP_OVERHEAD_SIZE   4\r
#define PXEBC_DEFAULT_PACKET_SIZE          1480\r
#define PXEBC_DEFAULT_LIFETIME             50000  // 50ms, unit is microsecond\r
\r
struct _PXEBC_PRIVATE_DATA {\r
  UINT32                                    Signature;\r
  EFI_HANDLE                                Controller;\r
  EFI_HANDLE                                Image;\r
  EFI_HANDLE                                ArpChild;\r
  EFI_HANDLE                                Dhcp4Child;\r
  EFI_HANDLE                                Ip4Child;\r
  EFI_HANDLE                                Mtftp4Child;\r
  EFI_HANDLE                                Udp4ReadChild;\r
  EFI_HANDLE                                Udp4WriteChild;\r
\r
  EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL *Nii;\r
\r
  EFI_PXE_BASE_CODE_PROTOCOL                PxeBc;\r
  EFI_LOAD_FILE_PROTOCOL                    LoadFile;\r
  EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL       LoadFileCallback;\r
  EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL       *PxeBcCallback;\r
  EFI_ARP_PROTOCOL                          *Arp;\r
  EFI_DHCP4_PROTOCOL                        *Dhcp4;\r
  EFI_IP4_PROTOCOL                          *Ip4;\r
  EFI_IP4_CONFIG2_PROTOCOL                  *Ip4Config2;\r
  EFI_IP4_CONFIG_DATA                       Ip4ConfigData;\r
  EFI_MTFTP4_PROTOCOL                       *Mtftp4;\r
  EFI_UDP4_PROTOCOL                         *Udp4Read;\r
  EFI_UDP4_PROTOCOL                         *Udp4Write;\r
  UINT16                                    CurrentUdpSrcPort;\r
  EFI_UDP4_CONFIG_DATA                      Udp4CfgData;\r
\r
\r
  EFI_PXE_BASE_CODE_MODE                    Mode;\r
  EFI_PXE_BASE_CODE_FUNCTION                Function;\r
\r
  CHAR8                                     *BootFileName;\r
\r
  EFI_IP_ADDRESS                            StationIp;\r
  EFI_IP_ADDRESS                            SubnetMask;\r
  EFI_IP_ADDRESS                            GatewayIp;\r
  EFI_IP_ADDRESS                            ServerIp;\r
  BOOLEAN                                   AddressIsOk;\r
  UINT32                                    Ip4MaxPacketSize;\r
  UINTN                                     BlockSize;\r
  UINTN                                     FileSize;\r
\r
  UINT8                                     OptionBuffer[PXEBC_DHCP4_MAX_OPTION_SIZE];\r
  EFI_DHCP4_PACKET                          SeedPacket;\r
  EFI_MAC_ADDRESS                           Mac;\r
  UINT8                                     MacLen;\r
\r
  BOOLEAN                                   SortOffers;\r
  BOOLEAN                                   GotProxyOffer;\r
  UINT32                                    NumOffers;\r
  UINT32                                    SelectedOffer;\r
  UINT32                                    ProxyOfferType;\r
\r
  //\r
  // Cached packets as complements of pxe mode data\r
  //\r
  PXEBC_CACHED_DHCP4_PACKET                 ProxyOffer;\r
  PXEBC_CACHED_DHCP4_PACKET                 Dhcp4Ack;\r
  PXEBC_CACHED_DHCP4_PACKET                 PxeReply;\r
  PXEBC_CACHED_DHCP4_PACKET                 Dhcp4Offers[PXEBC_MAX_OFFER_NUM];\r
\r
  //\r
  // Arrays for different types of offers:\r
  //   ServerCount records the count of the servers we got the offers,\r
  //   OfferIndex records the index of the offer sent by the server indexed by ServerCount.\r
  //\r
  UINT32                                    ServerCount[DHCP4_PACKET_TYPE_MAX];\r
  UINT32                                    OfferIndex[DHCP4_PACKET_TYPE_MAX][PXEBC_MAX_OFFER_NUM];\r
  UINT32                                    BootpIndex;\r
  UINT32                                    ProxyIndex[DHCP4_PACKET_TYPE_MAX];\r
  UINT32                                    BinlIndex[PXEBC_MAX_OFFER_NUM];\r
\r
  EFI_EVENT                                 GetArpCacheEvent;\r
  //\r
  // token and event used to get ICMP error data from IP\r
  //\r
  EFI_IP4_COMPLETION_TOKEN                  IcmpErrorRcvToken;\r
};\r
\r
#define PXEBC_PRIVATE_DATA_FROM_PXEBC(a)          CR (a, PXEBC_PRIVATE_DATA, PxeBc, PXEBC_PRIVATE_DATA_SIGNATURE)\r
\r
#define PXEBC_PRIVATE_DATA_FROM_LOADFILE(a)       CR (a, PXEBC_PRIVATE_DATA, LoadFile, PXEBC_PRIVATE_DATA_SIGNATURE)\r
\r
#define PXEBC_PRIVATE_DATA_FROM_PXEBCCALLBACK(a)  CR (a, PXEBC_PRIVATE_DATA, PxeBcCallback, PXEBC_PRIVATE_DATA_SIGNATURE)\r
\r
extern EFI_PXE_BASE_CODE_PROTOCOL mPxeBcProtocolTemplate;\r
extern EFI_LOAD_FILE_PROTOCOL     mLoadFileProtocolTemplate;\r
\r
/**\r
  Causes the driver to load a specified file.\r
\r
  @param  This                  Protocol instance pointer.\r
  @param  FilePath              The device specific path of the file to load.\r
  @param  BootPolicy            If TRUE, indicates that the request originates from the\r
                                boot manager is attempting to load FilePath as a boot\r
                                selection. If FALSE, then FilePath must match as exact file\r
                                to be loaded.\r
  @param  BufferSize            On input the size of Buffer in bytes. On output with a return\r
                                code of EFI_SUCCESS, the amount of data transferred to\r
                                Buffer. On output with a return code of EFI_BUFFER_TOO_SMALL,\r
                                the size of Buffer required to retrieve the requested file.\r
  @param  Buffer                The memory buffer to transfer the file to. IF Buffer is NULL,\r
                                then no the size of the requested file is returned in\r
                                BufferSize.\r
\r
  @retval EFI_SUCCESS                 The file was loaded.\r
  @retval EFI_UNSUPPORTED             The device does not support the provided BootPolicy\r
  @retval EFI_INVALID_PARAMETER       FilePath is not a valid device path, or\r
                                      BufferSize is NULL.\r
  @retval EFI_NO_MEDIA                No medium was present to load the file.\r
  @retval EFI_DEVICE_ERROR            The file was not loaded due to a device error.\r
  @retval EFI_NO_RESPONSE             The remote system did not respond.\r
  @retval EFI_NOT_FOUND               The file was not found.\r
  @retval EFI_ABORTED                 The file load process was manually cancelled.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
EfiPxeLoadFile (\r
  IN EFI_LOAD_FILE_PROTOCOL           * This,\r
  IN EFI_DEVICE_PATH_PROTOCOL         * FilePath,\r
  IN BOOLEAN                          BootPolicy,\r
  IN OUT UINTN                        *BufferSize,\r
  IN VOID                             *Buffer OPTIONAL\r
  );\r
\r
#endif\r
Commit	Line	Data
30368135	1	/** @file\r
30368135	2	\r
ac6c3d90	3	Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR>\r
e5eed7d3	4	This program and the accompanying materials\r
30368135	5	are licensed and made available under the terms and conditions of the BSD License\r
	6	which accompanies this distribution. The full text of the license may be found at\r
	7	http://opensource.org/licenses/bsd-license.php\r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	11	\r
30368135	12	**/\r
	13	\r
	14	#ifndef __EFI_PXEBC_IMPL_H__\r
	15	#define __EFI_PXEBC_IMPL_H__\r
	16	\r
	17	\r
	18	typedef struct _PXEBC_PRIVATE_DATA PXEBC_PRIVATE_DATA;\r
	19	\r
2517435c	20	#include <Uefi.h>\r
30368135	21	\r
	22	#include <Guid/SmBios.h>\r
	23	#include <IndustryStandard/SmBios.h>\r
ac6c3d90	24	#include <IndustryStandard/Dhcp.h>\r
30368135	25	#include <Protocol/Dhcp4.h>\r
	26	#include <Protocol/PxeBaseCode.h>\r
	27	#include <Protocol/Mtftp4.h>\r
	28	#include <Protocol/Udp4.h>\r
	29	#include <Protocol/LoadFile.h>\r
	30	#include <Protocol/NetworkInterfaceIdentifier.h>\r
	31	#include <Protocol/PxeBaseCodeCallBack.h>\r
	32	#include <Protocol/Arp.h>\r
	33	#include <Protocol/Ip4.h>\r
14e84fd8	34	#include <Protocol/Ip4Config2.h>\r
30368135	35	\r
30368135	36	#include <Library/DebugLib.h>\r
6758032d	37	#include <Library/DevicePathLib.h>\r
30368135	38	#include <Library/BaseMemoryLib.h>\r
20610fa8	39	#include <Library/MemoryAllocationLib.h>\r
30368135	40	#include <Library/UefiDriverEntryPoint.h>\r
	41	#include <Library/UefiBootServicesTableLib.h>\r
	42	#include <Library/UefiLib.h>\r
	43	#include <Library/BaseLib.h>\r
	44	#include <Library/NetLib.h>\r
d8d26fb2	45	#include <Library/DpcLib.h>\r
cfbc1a75	46	#include <Library/PcdLib.h>\r
30368135	47	\r
30368135	48	#include "PxeBcDriver.h"\r
30368135	49	#include "PxeBcDhcp.h"\r
	50	#include "PxeBcMtftp.h"\r
	51	#include "PxeBcSupport.h"\r
	52	\r
f3f2e05d	53	#define PXEBC_PRIVATE_DATA_SIGNATURE SIGNATURE_32 ('P', 'X', 'E', 'P')\r
fa6d3ee4	54	#define PXEBC_MTFTP_TIMEOUT 4\r
fa6d3ee4	55	#define PXEBC_MTFTP_RETRIES 6\r
4b738c76	56	#define PXEBC_DEFAULT_UDP_OVERHEAD_SIZE 8\r
fa6d3ee4	57	#define PXEBC_DEFAULT_TFTP_OVERHEAD_SIZE 4\r
7bc01e86	58	#define PXEBC_DEFAULT_PACKET_SIZE 1480\r
2e4c2a04	59	#define PXEBC_DEFAULT_LIFETIME 50000 // 50ms, unit is microsecond\r
30368135	60	\r
	61	struct _PXEBC_PRIVATE_DATA {\r
	62	UINT32 Signature;\r
	63	EFI_HANDLE Controller;\r
	64	EFI_HANDLE Image;\r
	65	EFI_HANDLE ArpChild;\r
	66	EFI_HANDLE Dhcp4Child;\r
	67	EFI_HANDLE Ip4Child;\r
	68	EFI_HANDLE Mtftp4Child;\r
	69	EFI_HANDLE Udp4ReadChild;\r
	70	EFI_HANDLE Udp4WriteChild;\r
	71	\r
	72	EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL *Nii;\r
	73	\r
	74	EFI_PXE_BASE_CODE_PROTOCOL PxeBc;\r
	75	EFI_LOAD_FILE_PROTOCOL LoadFile;\r
	76	EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL LoadFileCallback;\r
	77	EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL *PxeBcCallback;\r
	78	EFI_ARP_PROTOCOL *Arp;\r
	79	EFI_DHCP4_PROTOCOL *Dhcp4;\r
	80	EFI_IP4_PROTOCOL *Ip4;\r
14e84fd8	81	EFI_IP4_CONFIG2_PROTOCOL *Ip4Config2;\r
30368135	82	EFI_IP4_CONFIG_DATA Ip4ConfigData;\r
	83	EFI_MTFTP4_PROTOCOL *Mtftp4;\r
	84	EFI_UDP4_PROTOCOL *Udp4Read;\r
	85	EFI_UDP4_PROTOCOL *Udp4Write;\r
	86	UINT16 CurrentUdpSrcPort;\r
	87	EFI_UDP4_CONFIG_DATA Udp4CfgData;\r
	88	\r
	89	\r
	90	EFI_PXE_BASE_CODE_MODE Mode;\r
	91	EFI_PXE_BASE_CODE_FUNCTION Function;\r
	92	\r
	93	CHAR8 *BootFileName;\r
	94	\r
	95	EFI_IP_ADDRESS StationIp;\r
	96	EFI_IP_ADDRESS SubnetMask;\r
	97	EFI_IP_ADDRESS GatewayIp;\r
	98	EFI_IP_ADDRESS ServerIp;\r
	99	BOOLEAN AddressIsOk;\r
fa6d3ee4	100	UINT32 Ip4MaxPacketSize;\r
fa6d3ee4	101	UINTN BlockSize;\r
30368135	102	UINTN FileSize;\r
	103	\r
	104	UINT8 OptionBuffer[PXEBC_DHCP4_MAX_OPTION_SIZE];\r
	105	EFI_DHCP4_PACKET SeedPacket;\r
	106	EFI_MAC_ADDRESS Mac;\r
	107	UINT8 MacLen;\r
	108	\r
	109	BOOLEAN SortOffers;\r
	110	BOOLEAN GotProxyOffer;\r
	111	UINT32 NumOffers;\r
	112	UINT32 SelectedOffer;\r
	113	UINT32 ProxyOfferType;\r
	114	\r
	115	//\r
	116	// Cached packets as complements of pxe mode data\r
	117	//\r
	118	PXEBC_CACHED_DHCP4_PACKET ProxyOffer;\r
	119	PXEBC_CACHED_DHCP4_PACKET Dhcp4Ack;\r
	120	PXEBC_CACHED_DHCP4_PACKET PxeReply;\r
	121	PXEBC_CACHED_DHCP4_PACKET Dhcp4Offers[PXEBC_MAX_OFFER_NUM];\r
	122	\r
	123	//\r
	124	// Arrays for different types of offers:\r
	125	// ServerCount records the count of the servers we got the offers,\r
	126	// OfferIndex records the index of the offer sent by the server indexed by ServerCount.\r
	127	//\r
	128	UINT32 ServerCount[DHCP4_PACKET_TYPE_MAX];\r
	129	UINT32 OfferIndex[DHCP4_PACKET_TYPE_MAX][PXEBC_MAX_OFFER_NUM];\r
	130	UINT32 BootpIndex;\r
	131	UINT32 ProxyIndex[DHCP4_PACKET_TYPE_MAX];\r
	132	UINT32 BinlIndex[PXEBC_MAX_OFFER_NUM];\r
	133	\r
	134	EFI_EVENT GetArpCacheEvent;\r
	135	//\r
	136	// token and event used to get ICMP error data from IP\r
	137	//\r
	138	EFI_IP4_COMPLETION_TOKEN IcmpErrorRcvToken;\r
	139	};\r
	140	\r
	141	#define PXEBC_PRIVATE_DATA_FROM_PXEBC(a) CR (a, PXEBC_PRIVATE_DATA, PxeBc, PXEBC_PRIVATE_DATA_SIGNATURE)\r
	142	\r
	143	#define PXEBC_PRIVATE_DATA_FROM_LOADFILE(a) CR (a, PXEBC_PRIVATE_DATA, LoadFile, PXEBC_PRIVATE_DATA_SIGNATURE)\r
	144	\r
	145	#define PXEBC_PRIVATE_DATA_FROM_PXEBCCALLBACK(a) CR (a, PXEBC_PRIVATE_DATA, PxeBcCallback, PXEBC_PRIVATE_DATA_SIGNATURE)\r
	146	\r
	147	extern EFI_PXE_BASE_CODE_PROTOCOL mPxeBcProtocolTemplate;\r
	148	extern EFI_LOAD_FILE_PROTOCOL mLoadFileProtocolTemplate;\r
	149	\r
f737cfb9	150	/**\r
	151	Causes the driver to load a specified file.\r
	152	\r
	153	@param This Protocol instance pointer.\r
	154	@param FilePath The device specific path of the file to load.\r
7bc01e86	155	@param BootPolicy If TRUE, indicates that the request originates from the\r
f737cfb9	156	boot manager is attempting to load FilePath as a boot\r
	157	selection. If FALSE, then FilePath must match as exact file\r
	158	to be loaded.\r
	159	@param BufferSize On input the size of Buffer in bytes. On output with a return\r
7bc01e86	160	code of EFI_SUCCESS, the amount of data transferred to\r
f737cfb9	161	Buffer. On output with a return code of EFI_BUFFER_TOO_SMALL,\r
	162	the size of Buffer required to retrieve the requested file.\r
	163	@param Buffer The memory buffer to transfer the file to. IF Buffer is NULL,\r
7bc01e86	164	then no the size of the requested file is returned in\r
f737cfb9	165	BufferSize.\r
	166	\r
	167	@retval EFI_SUCCESS The file was loaded.\r
	168	@retval EFI_UNSUPPORTED The device does not support the provided BootPolicy\r
7bc01e86	169	@retval EFI_INVALID_PARAMETER FilePath is not a valid device path, or\r
f737cfb9	170	BufferSize is NULL.\r
	171	@retval EFI_NO_MEDIA No medium was present to load the file.\r
	172	@retval EFI_DEVICE_ERROR The file was not loaded due to a device error.\r
	173	@retval EFI_NO_RESPONSE The remote system did not respond.\r
	174	@retval EFI_NOT_FOUND The file was not found.\r
	175	@retval EFI_ABORTED The file load process was manually cancelled.\r
	176	\r
	177	**/\r
	178	EFI_STATUS\r
	179	EFIAPI\r
	180	EfiPxeLoadFile (\r
	181	IN EFI_LOAD_FILE_PROTOCOL * This,\r
	182	IN EFI_DEVICE_PATH_PROTOCOL * FilePath,\r
	183	IN BOOLEAN BootPolicy,\r
	184	IN OUT UINTN *BufferSize,\r
	185	IN VOID *Buffer OPTIONAL\r
	186	);\r
	187	\r
30368135	188	#endif\r