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

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