]> git.proxmox.com Git - mirror_edk2.git/blame - EdkModulePkg/Universal/Network/PxeBc/Dxe/pxe_bc_dhcp.c
projects / mirror_edk2.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Add some definitions for efi event in Uefi/UefiSpec.h to follow spec.
[mirror_edk2.git] / EdkModulePkg / Universal / Network / PxeBc / Dxe / pxe_bc_dhcp.c
CommitLineData
878ddf1f 1/*++\r
2\r
26aa0c2f 3Copyright (c) 2006 - 2007, Intel Corporation\r
3aaddf92 4All rights reserved. This program and the accompanying materials\r
5are licensed and made available under the terms and conditions of the BSD License\r
6which accompanies this distribution. The full text of the license may be found at\r
7http://opensource.org/licenses/bsd-license.php\r
8\r
9THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
10WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
878ddf1f 11\r
12Module Name:\r
13 pxe_bc_dhcp.c\r
3aaddf92 14\r
878ddf1f 15Abstract:\r
16 DHCP and PXE discovery protocol implementations.\r
17\r
18--*/\r
19\r
20\r
4cbd855e 21#include "Bc.h"\r
878ddf1f 22\r
23STATIC EFI_PXE_BASE_CODE_UDP_PORT DhcpServerPort = DHCP_SERVER_PORT;\r
24STATIC EFI_PXE_BASE_CODE_UDP_PORT DHCPClientPort = DHCP_CLIENT_PORT;\r
25STATIC EFI_PXE_BASE_CODE_UDP_PORT PseudoDhcpServerPort = PXE_DISCOVERY_PORT;\r
26#define PSEUDO_DHCP_CLIENT_PORT PseudoDhcpServerPort\r
27STATIC EFI_IP_ADDRESS BroadcastIP = { { 0xffffffff } };\r
28STATIC EFI_IP_ADDRESS DefaultSubnetMask = { { 0xffffff00 } };\r
29\r
30typedef union {\r
31 DHCPV4_OP_STRUCT *OpPtr;\r
32 PXE_OP_SERVER_LIST *BootServersStr;\r
33 PXE_SERVER_LIST *BootServerList;\r
34 PXE_BOOT_MENU_ENTRY *BootMenuItem;\r
35 PXE_OP_DISCOVERY_CONTROL *DiscoveryControl;\r
36 PXE_OP_BOOT_MENU *BootMenu;\r
37 PXE_OP_BOOT_ITEM *BootItem;\r
38 DHCPV4_OP_VENDOR_OPTIONS *VendorOptions;\r
39 DHCPV4_OP_OVERLOAD *Overload;\r
40 DHCPV4_OP_CLASS *PxeClassStr;\r
41 DHCPV4_OP_SUBNET_MASK *SubnetMaskStr;\r
42 DHCPV4_OP_MESSAGE_TYPE *MessageType;\r
43 UINT8 *BytePtr;\r
44} UNION_PTR;\r
45\r
878ddf1f 46#pragma pack(1)\r
47//\r
48// option structure for DHCPREQUEST at end of DISCOVER options\r
49// and for DHCPDECLINE\r
50//\r
51STATIC const struct requestopendstr {\r
52 DHCPV4_OP_REQUESTED_IP OpReqIP;\r
53 DHCPV4_OP_SERVER_IP DhcServerIpPtr;\r
54 UINT8 End[1];\r
55}\r
56RequestOpEndStr = {\r
57 {\r
58 {\r
59 OP_DHCP_REQ_IP_ADD,\r
60 DHCPV4_OPTION_LENGTH(DHCPV4_OP_REQUESTED_IP)\r
61 }\r
62 },\r
63 {\r
64 {\r
65 OP_DHCP_SERVER_IP,\r
66 DHCPV4_OPTION_LENGTH(DHCPV4_OP_SERVER_IP)\r
67 }\r
68 },\r
69 {\r
70 OP_END\r
71 }\r
72};\r
73\r
74#define DHCP_REQ_OPTIONS (*(struct requestopendstr *) DHCPV4_OPTIONS_BUFFER.End)\r
75\r
76PXE_OP_BOOT_ITEM DefaultBootItem = {\r
77 {\r
78 VEND_PXE_BOOT_ITEM,\r
79 DHCPV4_OPTION_LENGTH(PXE_OP_BOOT_ITEM)\r
80 },\r
81 0, 0,\r
82};\r
83\r
84//\r
85// PXE discovery control default structure\r
86//\r
87STATIC PXE_OP_DISCOVERY_CONTROL DefaultDisCtl = {\r
88 { VEND_PXE_DISCOVERY_CONTROL, DHCPV4_OPTION_LENGTH(PXE_OP_DISCOVERY_CONTROL) },\r
89 0\r
90};\r
91\r
92//\r
93// PXE credentials option structure\r
94//\r
95typedef struct {\r
96 UINT8 c[4];\r
97} PXE_CREDENTIAL;\r
98\r
99typedef struct {\r
100 DHCPV4_OP_HEADER Header;\r
101 PXE_CREDENTIAL Credentials[1];\r
102} PXE_OP_CREDENTIAL_TYPES;\r
103\r
104//\r
105// option structure for PXE discover (without credentials)\r
106//\r
107typedef struct { // discoveropendstr {\r
108 DHCPV4_OP_HEADER Header; // vendor options\r
109 PXE_OP_BOOT_ITEM BootItem;\r
110 UINT8 End[1]; // if credentials option, it starts here\r
111} PXE_DISCOVER_OPTIONS;\r
112\r
113#define DISCOVERoptions (*(PXE_DISCOVER_OPTIONS *) DHCPV4_OPTIONS_BUFFER.End)\r
114#define DISCREDoptions (*(PXE_OP_CREDENTIAL_TYPES *) DISCOVERoptions.End)\r
115\r
116//\r
117// common option beginning for all our DHCP messages except\r
118// DHCPDECLINE and DHCPRELEASE\r
119//\r
120STATIC struct optionsstr {\r
121 UINT8 DhcpCookie[4];\r
122 DHCPV4_OP_MESSAGE_TYPE DhcpMessageType;\r
123 DHCPV4_OP_MAX_MESSAGE_SIZE DhcpMaxMessageSize;\r
124 DHCPV4_OP_REQUESTED_OPTIONS DhcpRequestedOptions;\r
125 DHCPV4_OP_PLATFORM_ID DhcpPlatformId;\r
126 DHCPV4_OP_NETWORK_INTERFACE DhcpNetworkInterface;\r
127 DHCPV4_OP_ARCHITECTURE_TYPE DhcpClientArchitecture;\r
128 DHCPV4_OP_CLASS_ID DhcpClassIdentifier;\r
129 UINT8 End[1];\r
130} DHCPOpStart;\r
131\r
132/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
133VOID\r
134OptionsStrucInit (\r
135 VOID\r
136 )\r
137{\r
138 DHCPOpStart.DhcpCookie[0] = 99;\r
139 DHCPOpStart.DhcpCookie[1] = 130;\r
140 DHCPOpStart.DhcpCookie[2] = 83;\r
141 DHCPOpStart.DhcpCookie[3] = 99;\r
142 DHCPOpStart.DhcpMessageType.Header.OpCode = OP_DHCP_MESSAGE_TYPE;\r
143 DHCPOpStart.DhcpMessageType.Header.Length = 1;\r
144 DHCPOpStart.DhcpMessageType.Type = DHCPDISCOVER;\r
145 DHCPOpStart.DhcpMaxMessageSize.Header.OpCode = OP_DHCP_MAX_MESSAGE_SZ;\r
146 DHCPOpStart.DhcpMaxMessageSize.Header.Length = 2;\r
147 DHCPOpStart.DhcpMaxMessageSize.MaxSize[0] = MAX_DHCP_MSG_SZ >> 8;\r
148 DHCPOpStart.DhcpMaxMessageSize.MaxSize[1] = MAX_DHCP_MSG_SZ & 0xff;\r
149 DHCPOpStart.DhcpRequestedOptions.Header.OpCode = OP_DHCP_PARM_REQ_LIST;\r
150 DHCPOpStart.DhcpRequestedOptions.Header.Length = sizeof (DHCPV4_REQUESTED_OPTIONS_DATA);\r
151 DHCPOpStart.DhcpRequestedOptions.Data._OP_SUBNET_MASK = OP_SUBNET_MASK; /* 1 */\r
152 DHCPOpStart.DhcpRequestedOptions.Data._OP_TIME_OFFSET = OP_TIME_OFFSET; /* 2 */\r
153 DHCPOpStart.DhcpRequestedOptions.Data._OP_ROUTER_LIST = OP_ROUTER_LIST; /* 3 */\r
154 DHCPOpStart.DhcpRequestedOptions.Data._OP_TIME_SERVERS = OP_TIME_SERVERS; /* 4 */\r
155 DHCPOpStart.DhcpRequestedOptions.Data._OP_NAME_SERVERS = OP_NAME_SERVERS; /* 5 */\r
156 DHCPOpStart.DhcpRequestedOptions.Data._OP_DNS_SERVERS = OP_DNS_SERVERS; /* 6 */\r
157 DHCPOpStart.DhcpRequestedOptions.Data._OP_HOST_NAME = OP_HOST_NAME; /* 12 */\r
158 DHCPOpStart.DhcpRequestedOptions.Data._OP_BOOT_FILE_SZ = OP_BOOT_FILE_SZ; /* 13 */\r
159 DHCPOpStart.DhcpRequestedOptions.Data._OP_DOMAIN_NAME = OP_DOMAIN_NAME; /* 15 */\r
160 DHCPOpStart.DhcpRequestedOptions.Data._OP_ROOT_PATH = OP_ROOT_PATH; /* 17 */\r
161 DHCPOpStart.DhcpRequestedOptions.Data._OP_EXTENSION_PATH = OP_EXTENSION_PATH; /* 18 */\r
162 DHCPOpStart.DhcpRequestedOptions.Data._OP_MAX_DATAGRAM_SZ = OP_MAX_DATAGRAM_SZ; /* 22 */\r
163 DHCPOpStart.DhcpRequestedOptions.Data._OP_DEFAULT_TTL = OP_DEFAULT_TTL; /* 23 */\r
164 DHCPOpStart.DhcpRequestedOptions.Data._OP_BROADCAST_ADD = OP_BROADCAST_ADD; /* 28 */\r
165 DHCPOpStart.DhcpRequestedOptions.Data._OP_NIS_DOMAIN_NAME = OP_NIS_DOMAIN_NAME; /* 40 */\r
166 DHCPOpStart.DhcpRequestedOptions.Data._OP_NIS_SERVERS = OP_NIS_SERVERS; /* 41 */\r
167 DHCPOpStart.DhcpRequestedOptions.Data._OP_NTP_SERVERS = OP_NTP_SERVERS; /* 42 */\r
168 DHCPOpStart.DhcpRequestedOptions.Data._OP_VENDOR_SPECIFIC = OP_VENDOR_SPECIFIC; /* 43 */\r
169 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_REQ_IP_ADD = OP_DHCP_REQ_IP_ADD; /* 50 */\r
170 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_LEASE_TIME = OP_DHCP_LEASE_TIME; /* 51 */\r
171 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_SERVER_IP = OP_DHCP_SERVER_IP; /* 54 */\r
172 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_RENEWAL_TIME = OP_DHCP_RENEWAL_TIME; /* 58 */\r
173 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_REBINDING_TIME = OP_DHCP_REBINDING_TIME; /* 59 */\r
174 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_CLASS_IDENTIFIER = OP_DHCP_CLASS_IDENTIFIER; /* 60 */\r
175 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_TFTP_SERVER_NAME = OP_DHCP_TFTP_SERVER_NAME; /* 66 */\r
176 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_BOOTFILE = OP_DHCP_BOOTFILE; /* 67 */\r
177 DHCPOpStart.DhcpRequestedOptions.Data._OP_DHCP_PLATFORM_ID = OP_DHCP_PLATFORM_ID; /* 97 */\r
178 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption128 = 128;\r
179 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption129 = 129;\r
180 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption130 = 130;\r
181 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption131 = 131;\r
182 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption132 = 132;\r
183 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption133 = 133, DHCPOpStart.DhcpRequestedOptions.Data.VendorOption134 = 134;\r
184 DHCPOpStart.DhcpRequestedOptions.Data.VendorOption135 = 135;\r
185 DHCPOpStart.DhcpPlatformId.Header.OpCode = OP_DHCP_PLATFORM_ID;\r
186 DHCPOpStart.DhcpPlatformId.Header.Length = DHCPV4_OPTION_LENGTH (DHCPV4_OP_PLATFORM_ID);\r
187 DHCPOpStart.DhcpNetworkInterface.Header.OpCode = OP_DHCP_NETWORK_ARCH;\r
188 DHCPOpStart.DhcpNetworkInterface.Header.Length = DHCPV4_OPTION_LENGTH (DHCPV4_OP_NETWORK_INTERFACE);\r
189 DHCPOpStart.DhcpNetworkInterface.Type = 0;\r
190 DHCPOpStart.DhcpNetworkInterface.MajorVersion = 0;\r
191 DHCPOpStart.DhcpNetworkInterface.MinorVersion = 0;\r
192 DHCPOpStart.DhcpClientArchitecture.Header.OpCode = OP_DHCP_SYSTEM_ARCH;\r
193 DHCPOpStart.DhcpClientArchitecture.Header.Length = DHCPV4_OPTION_LENGTH (DHCPV4_OP_ARCHITECTURE_TYPE);\r
194 DHCPOpStart.DhcpClientArchitecture.Type = HTONS (SYS_ARCH);\r
195 DHCPOpStart.DhcpClassIdentifier.Header.OpCode = OP_DHCP_CLASS_IDENTIFIER;\r
196 DHCPOpStart.DhcpClassIdentifier.Header.Length = sizeof (DHCPV4_CLASS_ID_DATA);\r
197 CopyMem (\r
198 DHCPOpStart.DhcpClassIdentifier.Data.ClassIdentifier,\r
199 "PXEClient:",\r
200 sizeof ("PXEClient:")\r
201 );\r
202 CopyMem (DHCPOpStart.DhcpClassIdentifier.Data.Lit2, "Arch:", sizeof ("Arch:"));\r
203 CopyMem (\r
204 DHCPOpStart.DhcpClassIdentifier.Data.ArchitectureType,\r
205 "xxxxx",\r
206 sizeof ("xxxxx")\r
207 );\r
208 CopyMem (DHCPOpStart.DhcpClassIdentifier.Data.Lit3, ":", sizeof (":"));\r
209 CopyMem (DHCPOpStart.DhcpClassIdentifier.Data.InterfaceName, "XXXX", sizeof ("XXXX"));\r
210 CopyMem (DHCPOpStart.DhcpClassIdentifier.Data.Lit4, ":", sizeof (":"));\r
211 CopyMem (DHCPOpStart.DhcpClassIdentifier.Data.UndiMajor, "yyy", sizeof ("yyy"));\r
212 CopyMem (DHCPOpStart.DhcpClassIdentifier.Data.UndiMinor, "xxx", sizeof ("xxx"));\r
213 DHCPOpStart.End[0] = OP_END;\r
1cc8ee78 214}\r
878ddf1f 215\r
216/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
217\r
218//\r
219// DHCPDECLINE option structure\r
220//\r
221struct opdeclinestr {\r
222 UINT8 DhcpCookie[4];\r
223 DHCPV4_OP_MESSAGE_TYPE DhcpMessageType;\r
224 struct requestopendstr OpDeclineEnd;\r
225};\r
226\r
227#define DHCPDECLINEoptions (*(struct opdeclinestr *) DHCPV4_TRANSMIT_BUFFER.options)\r
228\r
229//\r
230// DHCPRELEASE option structure\r
231//\r
232struct opreleasestr {\r
233 UINT8 DhcpCookie[4];\r
234 DHCPV4_OP_MESSAGE_TYPE DhcpMessageType;\r
235 DHCPV4_OP_SERVER_IP DhcServerIpPtr;\r
236 UINT8 End[1];\r
237};\r
238\r
239#define DHCPRELEASEoptions (*(struct opreleasestr *) DHCPV4_TRANSMIT_BUFFER.options)\r
240\r
241//\r
242// array of PXE vendor options in which we are interested\r
243// value 0 -> not of interest, else value is index into PXE OPTION array\r
244// option values from 1 to MAX_OUR_PXE_OPT\r
245//\r
246STATIC UINT8 ourPXEopts[MAX_OUR_PXE_OPT] = {\r
247 VEND_PXE_MTFTP_IP_IX, // multicast IP address of bootfile for MTFTP listen\r
248 VEND_PXE_MTFTP_CPORT_IX, // UDP Port to monitor for MTFTP responses - Intel order\r
249 VEND_PXE_MTFTP_SPORT_IX, // Server UDP Port for MTFTP open - Intel order\r
250 VEND_PXE_MTFTP_TMOUT_IX, // Listen timeout - secs\r
251 VEND_PXE_MTFTP_DELAY_IX, // Transmission timeout - secs\r
252 VEND_PXE_DISCOVERY_CONTROL_IX, // bit field\r
253 VEND_PXE_DISCOVERY_MCAST_ADDR_IX, // boot server discovery multicast address\r
254 VEND_PXE_BOOT_SERVERS_IX, // list of boot servers of form tp(2) cnt(1) ips[cnt]\r
255 VEND_PXE_BOOT_MENU_IX,\r
256 VEND_PXE_BOOT_PROMPT_IX,\r
257 VEND_PXE_MCAST_ADDRS_ALLOC_IX, // not used by client\r
258 VEND_PXE_CREDENTIAL_TYPES_IX,\r
259 VEND_13_IX, // not used by client\r
260 VEND_14_IX, // not used by client\r
261 VEND_15_IX, // not used by client\r
262 VEND_16_IX, // not used by client\r
263 VEND_17_IX, // not used by client\r
264 VEND_18_IX, // not used by client\r
265 VEND_19_IX, // not used by client\r
266 VEND_20_IX, // not used by client\r
267 VEND_21_IX, // not used by client\r
268 VEND_22_IX, // not used by client\r
269 VEND_23_IX, // not used by client\r
270 VEND_24_IX, // not used by client\r
271 VEND_25_IX, // not used by client\r
272 VEND_26_IX, // not used by client\r
273 VEND_27_IX, // not used by client\r
274 VEND_28_IX, // not used by client\r
275 VEND_29_IX, // not used by client\r
276 VEND_30_IX, // not used by client\r
277 VEND_31_IX, // not used by client\r
278 VEND_32_IX, // not used by client\r
279 VEND_33_IX, // not used by client\r
280 VEND_34_IX, // not used by client\r
281 VEND_35_IX, // not used by client\r
282 VEND_36_IX, // not used by client\r
283 VEND_37_IX, // not used by client\r
284 VEND_38_IX, // not used by client\r
285 VEND_39_IX, // not used by client\r
286 VEND_40_IX, // not used by client\r
287 VEND_41_IX, // not used by client\r
288 VEND_42_IX, // not used by client\r
289 VEND_43_IX, // not used by client\r
290 VEND_44_IX, // not used by client\r
291 VEND_45_IX, // not used by client\r
292 VEND_46_IX, // not used by client\r
293 VEND_47_IX, // not used by client\r
294 VEND_48_IX, // not used by client\r
295 VEND_49_IX, // not used by client\r
296 VEND_50_IX, // not used by client\r
297 VEND_51_IX, // not used by client\r
298 VEND_52_IX, // not used by client\r
299 VEND_53_IX, // not used by client\r
300 VEND_54_IX, // not used by client\r
301 VEND_55_IX, // not used by client\r
302 VEND_56_IX, // not used by client\r
303 VEND_57_IX, // not used by client\r
304 VEND_58_IX, // not used by client\r
305 VEND_59_IX, // not used by client\r
306 VEND_60_IX, // not used by client\r
307 VEND_61_IX, // not used by client\r
308 VEND_62_IX, // not used by client\r
309 VEND_63_IX, // not used by client\r
310 VEND_64_IX, // not used by client\r
311 VEND_65_IX, // not used by client\r
312 VEND_66_IX, // not used by client\r
313 VEND_67_IX, // not used by client\r
314 VEND_68_IX, // not used by client\r
315 VEND_69_IX, // not used by client\r
316 VEND_70_IX, // not used by client\r
317 VEND_PXE_BOOT_ITEM_IX\r
318};\r
319\r
320/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
321\r
322//\r
323// array of options in which we are interested\r
324// value 0 -> not of interest, else value is index into OPTION array\r
325// option values from 1 to MAX_OUR_OPT\r
326//\r
327STATIC UINT8 OurDhcpOptions[MAX_OUR_OPT] = {\r
328 OP_SUBNET_MASK_IX, // OP_SUBNET_MASK 1 // data is the subnet mask\r
329 OP_TIME_OFFSET_IX, // OP_TIME_OFFSET 2 // data is the time offset of subnet to UTC in seconds\r
330 OP_ROUTER_LIST_IX, // OP_ROUTER_LIST 3 // list of routers on subnet\r
331 OP_TIME_SERVERS_IX, // OP_TIME_SERVERS 4 // list of time servers available\r
332 OP_NAME_SERVERS_IX, // OP_NAME_SERVERS 5 // list of name servers available\r
333 OP_DNS_SERVERS_IX, // OP_DNS_SERVERS 6 // list of DNS servers available\r
334 OP_LOG_SERVERS_IX, // OP_LOG_SERVERS 7\r
335 OP_COOKIE_SERVERS_IX, // OP_COOKIE_SERVERS 8\r
336 OP_LPR_SREVERS_IX, // OP_LPR_SREVERS 9\r
337 OP_IMPRESS_SERVERS_IX, // OP_IMPRESS_SERVERS 10\r
338 OP_RES_LOC_SERVERS_IX, // OP_RES_LOC_SERVERS 11\r
339 OP_HOST_NAME_IX, // OP_HOST_NAME 12 // client name\r
340 OP_BOOT_FILE_SZ_IX, // OP_BOOT_FILE_SZ 13 // number of 512 blocks of boot file\r
341 OP_DUMP_FILE_IX, // OP_DUMP_FILE 14 // path name of dump file if client crashes\r
342 OP_DOMAIN_NAME_IX, // OP_DOMAIN_NAME 15 // domain name to use\r
343 OP_SWAP_SERVER_IX, // OP_SWAP_SERVER 16\r
344 OP_ROOT_PATH_IX, // OP_ROOT_PATH 17 // path name containing root disk\r
345 OP_EXTENSION_PATH_IX, // OP_EXTENSION_PATH 18 // name of TFTP downloadable file of form of OP\r
346 OP_IP_FORWARDING_IX, // OP_IP_FORWARDING 19 // enable/disable IP packet forwarding\r
347 OP_NON_LOCAL_SRC_RTE_IX, // OP_NON_LOCAL_SRC_RTE 20 // enable/disable non local source routing\r
348 OP_POLICY_FILTER_IX, // OP_POLICY_FILTER 21 // policy filters for non local source routing\r
349 OP_MAX_DATAGRAM_SZ_IX, // OP_MAX_DATAGRAM_SZ 22 // maximum datagram reassembly size\r
350 OP_DEFAULT_TTL_IX, // OP_DEFAULT_TTL 23 // default IP time to live\r
351 OP_MTU_AGING_TIMEOUT_IX, // OP_MTU_AGING_TIMEOUT 24\r
352 OP_MTU_SIZES_IX, // OP_MTU_SIZES 25\r
353 OP_MTU_TO_USE_IX, // OP_MTU_TO_USE 26\r
354 OP_ALL_SUBNETS_LOCAL_IX, // OP_ALL_SUBNETS_LOCAL 27\r
355 OP_BROADCAST_ADD_IX, // OP_BROADCAST_ADD 28 // broadcast address used on subnet\r
356 OP_PERFORM_MASK_DISCOVERY_IX, // OP_PERFORM_MASK_DISCOVERY 29 // perform mask discovery using ICMP\r
357 OP_RESPOND_TO_MASK_REQ_IX, // OP_RESPOND_TO_MASK_REQ 30 // respond to subnet mask requests using ICMP\r
358 OP_PERFORM_ROUTER_DISCOVERY_IX, // OP_PERFORM_ROUTER_DISCOVERY 31\r
359 OP_ROUTER_SOLICIT_ADDRESS_IX, // OP_ROUTER_SOLICIT_ADDRESS 32\r
360 OP_STATIC_ROUTER_LIST_IX, // OP_STATIC_ROUTER_LIST 33 // list of dest/route pairs\r
361 OP_USE_ARP_TRAILERS_IX, // OP_USE_ARP_TRAILERS 34\r
362 OP_ARP_CACHE_TIMEOUT_IX, // OP_ARP_CACHE_TIMEOUT 35\r
363 OP_ETHERNET_ENCAPSULATION_IX, // OP_ETHERNET_ENCAPSULATION 36 // 0 -> RFC 894, 1 -> IEEE 802.3 (RFC 1042)\r
364 OP_TCP_DEFAULT_TTL_IX, // OP_TCP_DEFAULT_TTL 37 // default time to live when sending TCP segments\r
365 OP_TCP_KEEP_ALIVE_INT_IX, // OP_TCP_KEEP_ALIVE_INT 38 // keep alive interval in seconds\r
366 OP_KEEP_ALIVE_GARBAGE_IX, // OP_KEEP_ALIVE_GARBAGE 39\r
367 OP_NIS_DOMAIN_NAME_IX, // OP_NIS_DOMAIN_NAME 40\r
368 OP_NIS_SERVERS_IX, // OP_NIS_SERVERS 41\r
369 OP_NTP_SERVERS_IX, // OP_NTP_SERVERS 42\r
370 OP_VENDOR_SPECIFIC_IX, // OP_VENDOR_SPECIFIC 43\r
371 OP_NBNS_SERVERS_IX, // OP_NBNS_SERVERS 44\r
372 OP_NBDD_SERVERS_IX, // OP_NBDD_SERVERS 45\r
373 OP_NETBIOS_NODE_TYPE_IX, // OP_NETBIOS_NODE_TYPE 46\r
374 OP_NETBIOS_SCOPE_IX, // OP_NETBIOS_SCOPE 47\r
375 OP_XWINDOW_SYSTEM_FONT_SERVERS_IX, // OP_XWINDOW_SYSTEM_FONT_SERVERS 48\r
376 OP_XWINDOW_SYSTEM_DISPLAY_MANAGERS_IX, // OP_XWINDOW_SYSTEM_DISPLAY_MANAGERS 49\r
377 OP_DHCP_REQ_IP_ADD_IX, // OP_DHCP_REQ_IP_ADD 50 // requested IP address - in DHCPDISCOVER\r
378 OP_DHCP_LEASE_TIME_IX, // OP_DHCP_LEASE_TIME 51 // lease time requested/granted\r
379 OP_DHCP_OPTION_OVERLOAD_IX, // OP_DHCP_OPTION_OVERLOAD 52 // file/server name/both used to hold options\r
380 OP_DHCP_MESSAGE_TYPE_IX, // OP_DHCP_MESSAGE_TYPE 53 // message type\r
381 OP_DHCP_SERVER_IP_IX, // OP_DHCP_SERVER_IP 54 // IP of server\r
382 OP_DHCP_PARM_REQ_LIST_IX, // OP_DHCP_PARM_REQ_LIST 55 // list of requested parameters\r
383 OP_DHCP_ERROR_MESSAGE_IX, // OP_DHCP_ERROR_MESSAGE 56 // in DHCPNAK or DECLINE messages\r
384 OP_DHCP_MAX_MESSAGE_SZ_IX, // OP_DHCP_MAX_MESSAGE_SZ 57 // maximum DHCP message size client will accept\r
385 OP_DHCP_RENEWAL_TIME_IX, // OP_DHCP_RENEWAL_TIME 58 // time in seconds before transitioning to RENEWING state\r
386 OP_DHCP_REBINDING_TIME_IX, // OP_DHCP_REBINDING_TIME 59 // time in seconds before transitioning to REBINDING state\r
387 OP_DHCP_CLASS_IDENTIFIER_IX, // OP_DHCP_CLASS_IDENTIFIER 60\r
388 OP_DHCP_CLIENT_IDENTIFIER_IX, // OP_DHCP_CLIENT_IDENTIFIER 61\r
389 OP_RESERVED62_IX, // OP_RESERVED62\r
390 OP_RESERVED63_IX, // OP_RESERVED63\r
391 OP_NISPLUS_DOMAIN_NAME_IX, // OP_NISPLUS_DOMAIN_NAME 64\r
392 OP_NISPLUS_SERVERS_IX, // OP_NISPLUS_SERVERS 65\r
393 OP_DHCP_TFTP_SERVER_NAME_IX, // OP_DHCP_TFTP_SERVER_NAME 66\r
394 OP_DHCP_BOOTFILE_IX // OP_DHCP_BOOTFILE 67\r
395};\r
396\r
397#define RxBuf ((DHCP_RECEIVE_BUFFER *) (Private->ReceiveBuffers))\r
398\r
399#pragma pack()\r
400\r
401/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 402STATIC\r
878ddf1f 403CHAR8 *\r
404PxeBcLibGetSmbiosString (\r
405 IN SMBIOS_STRUCTURE_POINTER *Smbios,\r
406 IN UINT16 StringNumber\r
407 )\r
408/*++\r
409Routine description:\r
410 Return SMBIOS string given the string number.\r
411\r
412Arguments:\r
413 Smbios - Pointer to SMBIOS structure\r
3aaddf92 414 StringNumber - String number to return. 0 is used to skip all strings and\r
878ddf1f 415 point to the next SMBIOS structure.\r
416\r
417Returns:\r
418 Pointer to string, or pointer to next SMBIOS strcuture if StringNumber == 0\r
419--*/\r
420{\r
421 UINT16 Index;\r
422 CHAR8 *String;\r
423\r
424 //\r
425 // Skip over formatted section\r
426 //\r
427 String = (CHAR8 *) (Smbios->Raw + Smbios->Hdr->Length);\r
428\r
429 //\r
430 // Look through unformated section\r
431 //\r
432 for (Index = 1; Index <= StringNumber || StringNumber == 0; Index++) {\r
433 if (StringNumber == Index) {\r
434 return String;\r
435 }\r
436 //\r
437 // Skip string\r
438 //\r
439 for (; *String != 0; String++)\r
440 ;\r
441 String++;\r
442\r
443 if (*String == 0) {\r
444 //\r
445 // If double NULL then we are done.\r
446 // Return pointer to next structure in Smbios.\r
447 // if you pass in a 0 you will always get here\r
448 //\r
449 Smbios->Raw = (UINT8 *)++String;\r
450 return NULL;\r
451 }\r
452 }\r
453\r
454 return NULL;\r
455}\r
456\r
457EFI_STATUS\r
458PxeBcLibGetSmbiosSystemGuidAndSerialNumber (\r
459 IN EFI_GUID *SystemGuid,\r
460 OUT CHAR8 **SystemSerialNumber\r
461 )\r
462/*++\r
463\r
464Routine Description:\r
465 This function gets system guid and serial number from the smbios table\r
466\r
467Arguments:\r
468 SystemGuid - The pointer of returned system guid\r
469 SystemSerialNumber - The pointer of returned system serial number\r
470\r
471Returns:\r
472 EFI_SUCCESS - Successfully get the system guid and system serial number\r
473 EFI_NOT_FOUND - Not find the SMBIOS table\r
474--*/\r
475{\r
476 EFI_STATUS Status;\r
477 SMBIOS_STRUCTURE_TABLE *SmbiosTable;\r
478 SMBIOS_STRUCTURE_POINTER Smbios;\r
479 SMBIOS_STRUCTURE_POINTER SmbiosEnd;\r
480 UINT16 Index;\r
481\r
482 Status = EfiGetSystemConfigurationTable (&gEfiSmbiosTableGuid, (VOID **) &SmbiosTable);\r
483\r
484 if (EFI_ERROR (Status)) {\r
485 return EFI_NOT_FOUND;\r
486 }\r
487\r
488 Smbios.Hdr = (SMBIOS_HEADER *) (UINTN) SmbiosTable->TableAddress;\r
489 SmbiosEnd.Raw = (UINT8 *) (UINTN) (SmbiosTable->TableAddress + SmbiosTable->TableLength);\r
490\r
491 for (Index = 0; Index < SmbiosTable->TableLength; Index++) {\r
492 if (Smbios.Hdr->Type == 1) {\r
493 if (Smbios.Hdr->Length < 0x19) {\r
494 //\r
495 // Older version did not support Guid and Serial number\r
496 //\r
497 continue;\r
498 }\r
499 //\r
500 // SMBIOS tables are byte packed so we need to do a byte copy to\r
501 // prevend alignment faults on Itanium-based platform.\r
502 //\r
503 CopyMem (SystemGuid, &Smbios.Type1->Uuid, sizeof (EFI_GUID));\r
504 *SystemSerialNumber = PxeBcLibGetSmbiosString (&Smbios, Smbios.Type1->SerialNumber);\r
505\r
506 return EFI_SUCCESS;\r
507 }\r
508 //\r
509 // Make Smbios point to the next record\r
510 //\r
511 PxeBcLibGetSmbiosString (&Smbios, 0);\r
512\r
513 if (Smbios.Raw >= SmbiosEnd.Raw) {\r
514 //\r
515 // SMBIOS 2.1 incorrectly stated the length of SmbiosTable as 0x1e.\r
516 // given this we must double check against the lenght of\r
517 // the structure.\r
518 //\r
519 return EFI_SUCCESS;\r
520 }\r
521 }\r
522\r
523 return EFI_SUCCESS;\r
524}\r
525\r
526/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
527\r
528//\r
529// add router list to list\r
530//\r
531STATIC\r
532VOID\r
533Ip4AddRouterList (\r
534 PXE_BASECODE_DEVICE *Private,\r
535 DHCPV4_OP_IP_LIST *IpListPtr\r
536 )\r
537{\r
538 EFI_IP_ADDRESS TmpIp;\r
539 INTN Index;\r
540 INTN num;\r
541\r
542 if (IpListPtr == NULL) {\r
543 return ;\r
544 }\r
545\r
546 for (Index = 0, num = IpListPtr->Header.Length >> 2; Index < num; ++Index) {\r
547 CopyMem (&TmpIp, &IpListPtr->IpList[Index], 4);\r
548 Ip4AddRouter (Private, &TmpIp);\r
549 }\r
550}\r
551\r
552/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
553\r
554//\r
555// send ARP for our IP - fail if someone has it\r
556//\r
557STATIC\r
558BOOLEAN\r
559SetStationIP (\r
560 PXE_BASECODE_DEVICE *Private\r
561 )\r
562{\r
563 EFI_MAC_ADDRESS DestMac;\r
564 EFI_STATUS EfiStatus;\r
565\r
566 ZeroMem (&DestMac, sizeof DestMac);\r
567\r
568 if (GetHwAddr(Private, (EFI_IP_ADDRESS *)&DHCP_REQ_OPTIONS.OpReqIP.Ip, (EFI_MAC_ADDRESS *)&DestMac)\r
569 || DoArp(Private, (EFI_IP_ADDRESS *)&DHCP_REQ_OPTIONS.OpReqIP.Ip, (EFI_MAC_ADDRESS *)&DestMac) == EFI_SUCCESS) {\r
570 return FALSE; // somebody else has this IP\r
571 }\r
572\r
573 CopyMem (\r
574 (EFI_IPv4_ADDRESS *) &Private->EfiBc.Mode->StationIp,\r
575 &DHCP_REQ_OPTIONS.OpReqIP.Ip,\r
576 sizeof (EFI_IPv4_ADDRESS)\r
577 );\r
578\r
579 Private->GoodStationIp = TRUE;\r
580\r
581 if (!Private->UseIgmpv1Reporting) {\r
582 return TRUE;\r
583 }\r
584\r
585 if (Private->Igmpv1TimeoutEvent != NULL) {\r
586 return TRUE;\r
587 }\r
588\r
589 EfiStatus = gBS->CreateEvent (\r
93b0fbc8 590 EVT_TIMER,\r
591 TPL_CALLBACK,\r
878ddf1f 592 NULL,\r
593 NULL,\r
594 &Private->Igmpv1TimeoutEvent\r
595 );\r
596\r
597 if (EFI_ERROR (EfiStatus)) {\r
598 Private->Igmpv1TimeoutEvent = NULL;\r
599 return TRUE;\r
600 }\r
601\r
602 EfiStatus = gBS->SetTimer (\r
603 Private->Igmpv1TimeoutEvent,\r
604 TimerRelative,\r
605 (UINT64) V1ROUTER_PRESENT_TIMEOUT * 10000000\r
606 ); /* 400 seconds */\r
607\r
608 if (EFI_ERROR (EfiStatus)) {\r
609 gBS->CloseEvent (Private->Igmpv1TimeoutEvent);\r
610 Private->Igmpv1TimeoutEvent = NULL;\r
611 }\r
612\r
613 return TRUE;\r
614}\r
615\r
616/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
617STATIC\r
618VOID\r
619AddRouters (\r
620 PXE_BASECODE_DEVICE *Private,\r
621 DHCP_RECEIVE_BUFFER *RxBufPtr\r
622 )\r
623{\r
878ddf1f 624 Ip4AddRouterList (\r
625 Private,\r
626 (DHCPV4_OP_IP_LIST *) RxBufPtr->OpAdds.PktOptAdds[OP_ROUTER_LIST_IX - 1]\r
627 );\r
628}\r
629\r
630/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
631STATIC\r
632EFI_STATUS\r
633DoUdpWrite (\r
634 PXE_BASECODE_DEVICE *Private,\r
635 EFI_IP_ADDRESS *ServerIpPtr,\r
636 EFI_PXE_BASE_CODE_UDP_PORT *ServerPortPtr,\r
637 EFI_IP_ADDRESS *ClientIpPtr,\r
638 EFI_PXE_BASE_CODE_UDP_PORT *ClientPortPtr\r
639 )\r
640{\r
641 UINTN Len;\r
642\r
643 Len = sizeof DHCPV4_TRANSMIT_BUFFER;\r
644\r
645 return UdpWrite (\r
646 Private,\r
647 EFI_PXE_BASE_CODE_UDP_OPFLAGS_MAY_FRAGMENT,\r
648 ServerIpPtr,\r
649 ServerPortPtr,\r
650 0,\r
651 ClientIpPtr,\r
652 ClientPortPtr,\r
653 0,\r
654 0,\r
655 &Len,\r
656 Private->TransmitBuffer\r
657 );\r
658}\r
659\r
660/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
661\r
662//\r
663// initialize the DHCP structure\r
664//\r
665typedef struct {\r
666 UINT8 x[4];\r
667} C4Str;\r
668\r
669STATIC\r
670VOID\r
671InitDhcpv4TxBuf (\r
672 PXE_BASECODE_DEVICE *Private\r
673 )\r
674{\r
675 UINTN HwAddrLen;\r
676 UINT8 *String;\r
677 CHAR8 *SystemSerialNumber;\r
678 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
679\r
680 PxebcMode = Private->EfiBc.Mode;\r
681\r
682 ZeroMem (&DHCPV4_TRANSMIT_BUFFER, sizeof (DHCPV4_STRUCT));\r
683 DHCPV4_TRANSMIT_BUFFER.op = BOOTP_REQUEST;\r
684 DHCPV4_TRANSMIT_BUFFER.htype = Private->SimpleNetwork->Mode->IfType;\r
685 DHCPV4_TRANSMIT_BUFFER.flags = HTONS (DHCP_BROADCAST_FLAG);\r
686 CopyMem (&DHCPV4_OPTIONS_BUFFER, (VOID *) &DHCPOpStart, sizeof (DHCPOpStart));\r
687\r
688 //\r
689 // default to hardware address\r
690 //\r
691 HwAddrLen = Private->SimpleNetwork->Mode->HwAddressSize;\r
692\r
693 if (HwAddrLen > sizeof DHCPV4_TRANSMIT_BUFFER.chaddr) {\r
694 HwAddrLen = sizeof DHCPV4_TRANSMIT_BUFFER.chaddr;\r
695 }\r
696\r
697 String = (UINT8 *) &Private->SimpleNetwork->Mode->CurrentAddress;\r
698\r
699 if (PxeBcLibGetSmbiosSystemGuidAndSerialNumber (\r
700 (EFI_GUID *) DHCPV4_OPTIONS_BUFFER.DhcpPlatformId.Guid,\r
701 &SystemSerialNumber\r
702 ) == EFI_SUCCESS) {\r
703 if (PxebcMode->SendGUID) {\r
704 HwAddrLen = sizeof (EFI_GUID);\r
705 String = (UINT8 *) DHCPV4_OPTIONS_BUFFER.DhcpPlatformId.Guid;\r
706 }\r
707 } else {\r
708 //\r
709 // GUID not yet set - send all 0xff's to show programable (via SetVariable)\r
710 // SetMem(DHCPV4_OPTIONS_BUFFER.DhcpPlatformId.Guid, sizeof(EFI_GUID), 0xff);\r
711 // GUID not yet set - send all 0's to show not programable\r
712 //\r
713 ZeroMem (DHCPV4_OPTIONS_BUFFER.DhcpPlatformId.Guid, sizeof (EFI_GUID));\r
714 }\r
715\r
716 DHCPV4_TRANSMIT_BUFFER.hlen = (UINT8) HwAddrLen;\r
717 CopyMem (DHCPV4_TRANSMIT_BUFFER.chaddr, String, HwAddrLen);\r
718\r
719 CvtNum (\r
720 SYS_ARCH,\r
721 (UINT8 *) DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.ArchitectureType,\r
722 sizeof DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.ArchitectureType\r
723 );\r
724\r
725 DHCPV4_OPTIONS_BUFFER.DhcpNetworkInterface.Type = Private->NiiPtr->Type;\r
726 DHCPV4_OPTIONS_BUFFER.DhcpNetworkInterface.MajorVersion = Private->NiiPtr->MajorVer;\r
727 DHCPV4_OPTIONS_BUFFER.DhcpNetworkInterface.MinorVersion = Private->NiiPtr->MinorVer;\r
728\r
729 *(C4Str *) DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.InterfaceName = *(C4Str *) Private->NiiPtr->StringId;\r
730\r
731 CvtNum (\r
732 DHCPV4_OPTIONS_BUFFER.DhcpNetworkInterface.MajorVersion,\r
733 (UINT8 *) DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.UndiMajor,\r
734 sizeof DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.UndiMajor\r
735 );\r
736\r
737 CvtNum (\r
738 DHCPV4_OPTIONS_BUFFER.DhcpNetworkInterface.MinorVersion,\r
739 (UINT8 *) DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.UndiMinor,\r
740 sizeof DHCPV4_OPTIONS_BUFFER.DhcpClassIdentifier.Data.UndiMinor\r
741 );\r
742}\r
743\r
744/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
745STATIC\r
746UINT32\r
747DecodePxeOptions (\r
748 DHCP_RECEIVE_BUFFER *RxBufPtr,\r
749 UINT8 *ptr,\r
750 INTN Len\r
751 )\r
752{\r
753 UINT8 Op;\r
754 UINT8 *EndPtr;\r
755 INTN Index;\r
756 UNION_PTR LocalPtr;\r
757 UINT32 status;\r
758\r
759 status = 0;\r
760\r
761 for (EndPtr = ptr + Len; ptr < EndPtr; ptr += Len + 2) {\r
762 Op = ptr[0];\r
763 Len = ptr[1];\r
764\r
765 switch (Op) {\r
766 case OP_PAD:\r
767 Len = -1;\r
768 break;\r
769\r
770 case OP_END:\r
771 return status;\r
772\r
773 default:\r
774 LocalPtr.BytePtr = ptr;\r
775 if (Op <= MAX_OUR_PXE_OPT) {\r
776 Index = ourPXEopts[Op - 1];\r
777 if (Index) {\r
778 RxBufPtr->OpAdds.PxeOptAdds[Index - 1] = LocalPtr.OpPtr;\r
779 status |= 1 << Index;\r
780 if (Index == VEND_PXE_BOOT_ITEM && LocalPtr.BootItem->Header.Length == 3) {\r
781 RxBufPtr->OpAdds.Status |= USE_THREE_BYTE;\r
782 }\r
783 }\r
784 }\r
785 break;\r
786 }\r
787 }\r
788\r
789 return status;\r
790}\r
791\r
792/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
793STATIC\r
794VOID\r
795DecodeOptions (\r
796 DHCP_RECEIVE_BUFFER *RxBufPtr,\r
797 UINT8 *ptr,\r
798 INTN Len\r
799 )\r
800{\r
801 UINT8 Op;\r
802 UINT8 *EndPtr;\r
803 INTN Index;\r
804 UNION_PTR LocalPtr;\r
805\r
806 for (EndPtr = ptr + Len; ptr < EndPtr; ptr += Len + 2) {\r
807 Op = ptr[0];\r
808 Len = ptr[1];\r
809\r
810 switch (Op) {\r
811 case OP_PAD:\r
812 Len = -1;\r
813 break;\r
814\r
815 case OP_END:\r
816 return ;\r
817\r
818 default:\r
819 LocalPtr.BytePtr = ptr;\r
820 if (Op <= MAX_OUR_OPT) {\r
821 Index = OurDhcpOptions[Op - 1];\r
822 if (Index) {\r
823 RxBufPtr->OpAdds.PktOptAdds[Index - 1] = LocalPtr.OpPtr;\r
824 if (Index == OP_VENDOR_SPECIFIC_IX) {\r
825 UINT32 status;\r
826 status = DecodePxeOptions (\r
827 RxBufPtr,\r
828 (UINT8 *) LocalPtr.VendorOptions->VendorOptions,\r
829 LocalPtr.VendorOptions->Header.Length\r
830 );\r
831 if (status) {\r
832 RxBufPtr->OpAdds.Status |= PXE_TYPE;\r
833 //\r
834 // check for all the MTFTP info options present - any missing is a nogo\r
835 //\r
836 if ((status & WfM11a_OPTS) == WfM11a_OPTS) {\r
837 RxBufPtr->OpAdds.Status |= WfM11a_TYPE;\r
838 }\r
839\r
840 if (status & DISCOVER_OPTS) {\r
841 RxBufPtr->OpAdds.Status |= DISCOVER_TYPE;\r
842 }\r
843\r
844 if (status & CREDENTIALS_OPT) {\r
845 RxBufPtr->OpAdds.Status |= CREDENTIALS_TYPE;\r
846 }\r
847 }\r
848 }\r
849 }\r
850 }\r
851 break;\r
852 }\r
853 }\r
854}\r
855\r
856/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 857STATIC\r
878ddf1f 858VOID\r
859Parse (\r
860 DHCP_RECEIVE_BUFFER *RxBufPtr,\r
861 INTN Len\r
862 )\r
863{\r
864 UNION_PTR LocalPtr;\r
865\r
866 //\r
867 // initialize\r
868 //\r
869 SetMem (&RxBufPtr->OpAdds, sizeof RxBufPtr->OpAdds, 0);\r
870\r
871 DecodeOptions (\r
872 RxBufPtr,\r
873 RxBufPtr->u.Dhcpv4.options + 4,\r
874 Len - (sizeof RxBufPtr->u.Dhcpv4 - sizeof RxBufPtr->u.Dhcpv4.options + 4)\r
875 );\r
876\r
877 LocalPtr.OpPtr = RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_OPTION_OVERLOAD_IX - 1];\r
878\r
879 if ((LocalPtr.OpPtr) && (LocalPtr.Overload->Overload & OVLD_SRVR_NAME)) {\r
880 DecodeOptions (RxBufPtr, RxBufPtr->u.Dhcpv4.sname, sizeof RxBufPtr->u.Dhcpv4.sname);\r
881 }\r
882\r
883 if (LocalPtr.OpPtr && (LocalPtr.Overload->Overload & OVLD_FILE)) {\r
884 DecodeOptions (RxBufPtr, RxBufPtr->u.Dhcpv4.file, sizeof RxBufPtr->u.Dhcpv4.file);\r
885 } else if (!RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1] && RxBufPtr->u.Dhcpv4.file[0]) {\r
886 RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1] = (DHCPV4_OP_STRUCT *) (RxBufPtr->u.Dhcpv4.file - sizeof (DHCPV4_OP_HEADER));\r
887\r
888 RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1]->Header.Length = (UINT8) AsciiStrLen ((CHAR8 *)RxBufPtr->u.Dhcpv4.file);\r
889 }\r
890\r
891 LocalPtr.OpPtr = RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_CLASS_IDENTIFIER_IX - 1];\r
892\r
893 if ((LocalPtr.OpPtr) &&\r
894 LocalPtr.PxeClassStr->Header.Length >= 9 &&\r
895 !CompareMem (LocalPtr.PxeClassStr->Class, "PXEClient", 9)\r
896 ) {\r
897 RxBufPtr->OpAdds.Status |= PXE_TYPE;\r
898 }\r
899}\r
900\r
901/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
902STATIC\r
903VOID\r
904CopyParseRxBuf (\r
905 PXE_BASECODE_DEVICE *Private,\r
906 INTN RxBufIndex,\r
907 INTN PacketIndex\r
908 )\r
909{\r
910 DHCP_RECEIVE_BUFFER *RxBufPtr;\r
911\r
912 RxBufPtr = &((DHCP_RECEIVE_BUFFER *) Private->DhcpPacketBuffer)[PacketIndex];\r
913\r
914 CopyMem (\r
915 &RxBufPtr->u.Dhcpv4,\r
916 &RxBuf[RxBufIndex].u.Dhcpv4,\r
917 sizeof (RxBuf[RxBufIndex].u.Dhcpv4)\r
918 );\r
919\r
920 Parse (RxBufPtr, sizeof RxBufPtr->u.ReceiveBuffer);\r
921}\r
922\r
923/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
924STATIC\r
925VOID\r
926CopyProxyRxBuf (\r
927 PXE_BASECODE_DEVICE *Private,\r
928 INTN RxBufIndex\r
929 )\r
930{\r
931 Private->EfiBc.Mode->ProxyOfferReceived = TRUE;\r
932 CopyParseRxBuf (Private, RxBufIndex, PXE_OFFER_INDEX);\r
933}\r
934\r
935/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
936STATIC\r
937VOID\r
938CopyParse (\r
939 PXE_BASECODE_DEVICE *Private,\r
940 EFI_PXE_BASE_CODE_PACKET *PacketPtr,\r
941 EFI_PXE_BASE_CODE_PACKET *NewPacketPtr,\r
942 INTN Index\r
943 )\r
944{\r
945 DHCP_RECEIVE_BUFFER *DhcpRxBuf;\r
946\r
947 DhcpRxBuf = &((DHCP_RECEIVE_BUFFER *) Private->DhcpPacketBuffer)[Index];\r
948\r
949 CopyMem (\r
950 (EFI_PXE_BASE_CODE_PACKET *) &DhcpRxBuf->u.Dhcpv4,\r
951 NewPacketPtr,\r
952 sizeof (*NewPacketPtr)\r
953 );\r
954\r
955 CopyMem (&*PacketPtr, &*NewPacketPtr, sizeof (*NewPacketPtr));\r
956\r
957 Parse (DhcpRxBuf, sizeof DhcpRxBuf->u.ReceiveBuffer);\r
958}\r
959\r
960/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 961STATIC\r
878ddf1f 962BOOLEAN\r
963AckEdit (\r
964 DHCP_RECEIVE_BUFFER *DhcpRxBuf\r
965 )\r
966{\r
967 UNION_PTR LocalPtr;\r
968\r
969 LocalPtr.OpPtr = DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_MESSAGE_TYPE_IX - 1];\r
970\r
971 //\r
972 // check that an ACK\r
973 // if a DHCP type, must be DHCPOFFER and must have server id\r
974 //\r
975 return (BOOLEAN)\r
976 (\r
977 (LocalPtr.OpPtr) &&\r
978 (LocalPtr.MessageType->Type == DHCPACK) &&\r
979 DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1]\r
980 );\r
981}\r
982\r
983/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
984\r
985//\r
986// if a discover type packet, make sure all required fields are present\r
987//\r
1cc8ee78 988STATIC\r
878ddf1f 989BOOLEAN\r
990DHCPOfferAckEdit (\r
991 DHCP_RECEIVE_BUFFER *DhcpRxBuf\r
992 )\r
993{\r
994 PXE_OP_SERVER_LIST *BootServerOpPtr;\r
995 UNION_PTR LocalPtr;\r
996\r
997 if ((DhcpRxBuf->OpAdds.Status & DISCOVER_TYPE) == 0) {\r
998 return TRUE;\r
999 }\r
1000\r
1001 LocalPtr.OpPtr = DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_DISCOVERY_CONTROL_IX - 1];\r
1002\r
1003 if (LocalPtr.OpPtr == NULL) {\r
1004 LocalPtr.OpPtr = (DHCPV4_OP_STRUCT *) &DefaultDisCtl;\r
1005 DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_DISCOVERY_CONTROL_IX - 1] = (DHCPV4_OP_STRUCT *) &DefaultDisCtl;\r
1006 }\r
1007 //\r
1008 // make sure all required fields are here\r
1009 // if mucticast enabled, need multicast address\r
1010 //\r
1011 if (!(LocalPtr.DiscoveryControl->ControlBits & DISABLE_MCAST) &&\r
1012 (!DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_DISCOVERY_MCAST_ADDR_IX - 1] || !IS_MULTICAST (((DHCPV4_OP_STRUCT *) DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_DISCOVERY_MCAST_ADDR_IX - 1])->Data))\r
1013 ) {\r
1014 return FALSE;\r
1015 //\r
1016 // missing required field\r
1017 //\r
1018 }\r
1019 //\r
1020 // if a list, it better be good\r
1021 //\r
1022 BootServerOpPtr = (PXE_OP_SERVER_LIST *) DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_BOOT_SERVERS_IX - 1];\r
1023\r
1024 if (BootServerOpPtr != NULL) {\r
1025 PXE_SERVER_LIST *BootServerListPtr;\r
1026 INTN ServerListLen;\r
1027 INTN ServerEntryLen;\r
1028\r
1029 BootServerListPtr = BootServerOpPtr->ServerList;\r
1030 ServerListLen = BootServerOpPtr->Header.Length;\r
1031\r
1032 do {\r
1033 EFI_IPv4_ADDRESS *IpListPtr;\r
1034 INTN IpCnt;\r
1035\r
1036 IpCnt = BootServerListPtr->u.Ipv4List.IpCount;\r
1037\r
1038 ServerEntryLen = sizeof (PXEV4_SERVER_LIST) + 2 + (IpCnt - 1) * sizeof (EFI_IPv4_ADDRESS);\r
1039\r
1040 if (ServerListLen < ServerEntryLen) {\r
1041 //\r
1042 // missing required field\r
1043 //\r
1044 return FALSE;\r
1045 }\r
1046\r
1047 IpListPtr = BootServerListPtr->u.Ipv4List.IpList;\r
1048\r
1049 while (IpCnt--) {\r
1050 if (IS_MULTICAST (IpListPtr)) {\r
1051 //\r
1052 // missing required field\r
1053 //\r
1054 return FALSE;\r
1055 } else {\r
1056 ++IpListPtr;\r
1057 }\r
1058 }\r
1059\r
1060 BootServerListPtr = (PXE_SERVER_LIST *) IpListPtr;\r
1061 } while (ServerListLen -= ServerEntryLen);\r
1062 }\r
1063 //\r
1064 // else there must be a list if use list enabled or multicast and\r
1065 // broadcast disabled\r
1066 //\r
1067 else if ((LocalPtr.DiscoveryControl->ControlBits & USE_ACCEPT_LIST) ||\r
1068 ((LocalPtr.DiscoveryControl->ControlBits & (DISABLE_MCAST | DISABLE_BCAST)) == (DISABLE_MCAST | DISABLE_BCAST))\r
1069 ) {\r
1070 //\r
1071 // missing required field\r
1072 //\r
1073 return FALSE;\r
1074 }\r
1075 //\r
1076 // if not USE_BOOTFILE or no bootfile given, must have menu stuff\r
1077 //\r
1078 if (!(LocalPtr.DiscoveryControl->ControlBits & USE_BOOTFILE) ||\r
1079 !DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1]\r
1080 ) {\r
1081 INTN MenuLth;\r
1082\r
1083 LocalPtr.OpPtr = DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_BOOT_MENU_IX - 1];\r
1084\r
1085 if (LocalPtr.OpPtr == NULL || !DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_BOOT_PROMPT_IX - 1]) {\r
1086 //\r
1087 // missing required field\r
1088 //\r
1089 return FALSE;\r
1090 }\r
1091 //\r
1092 // make sure menu valid\r
1093 //\r
1094 MenuLth = LocalPtr.BootMenu->Header.Length;\r
1095 LocalPtr.BootMenuItem = LocalPtr.BootMenu->MenuItem;\r
1096\r
1097 do {\r
1098 INTN MenuItemLen;\r
1099\r
1100 MenuItemLen = LocalPtr.BootMenuItem->DataLen;\r
1101\r
1102 if (MenuItemLen == 0) {\r
1103 //\r
1104 // missing required field\r
1105 //\r
1106 return FALSE;\r
1107 }\r
1108\r
1109 MenuItemLen += sizeof (*LocalPtr.BootMenuItem) - sizeof (LocalPtr.BootMenuItem->Data);\r
1110\r
1111 MenuLth -= MenuItemLen;\r
1112 LocalPtr.BytePtr += MenuItemLen;\r
1113 } while (MenuLth > 0);\r
1114\r
1115 if (MenuLth != 0) {\r
1116 //\r
1117 // missing required field\r
1118 //\r
1119 return FALSE;\r
1120 }\r
1121 }\r
1122\r
1123 if (!DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_BOOT_ITEM_IX - 1]) {\r
1124 DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_BOOT_ITEM_IX - 1] = (DHCPV4_OP_STRUCT *) &DefaultBootItem;\r
1125 }\r
1126\r
1127 return TRUE;\r
1128}\r
1129\r
1130/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 1131STATIC\r
878ddf1f 1132BOOLEAN\r
1133DHCPAckEdit (\r
1134 DHCP_RECEIVE_BUFFER *RxBufPtr\r
1135 )\r
1136{\r
1137 return (BOOLEAN) (DHCPOfferAckEdit (RxBufPtr) ? AckEdit (RxBufPtr) : FALSE);\r
1138}\r
1139\r
1140/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1141\r
1142//\r
1143// get an offer/ack\r
1144//\r
1cc8ee78 1145STATIC\r
878ddf1f 1146EFI_STATUS\r
1147GetOfferAck (\r
1148 PXE_BASECODE_DEVICE *Private,\r
1149 BOOLEAN (*ExtraEdit)(DHCP_RECEIVE_BUFFER *DhcpRxBuf),\r
1150 UINT16 OpFlags, // for Udp read\r
1151 EFI_IP_ADDRESS *ServerIpPtr,\r
1152 EFI_PXE_BASE_CODE_UDP_PORT *ServerPortPtr,\r
1153 EFI_IP_ADDRESS *ClientIpPtr,\r
1154 EFI_PXE_BASE_CODE_UDP_PORT *ClientPortPtr,\r
1155 DHCP_RECEIVE_BUFFER *DhcpRxBuf,\r
1156 EFI_EVENT TimeoutEvent\r
1157 )\r
1158/*++\r
1159Routine description:\r
1160 Wait for an OFFER/ACK packet.\r
1161\r
1162Parameters:\r
1163 Private := Pointer to PxeBc interface\r
1164 ExtraEdit := Pointer to extra option checking function\r
1165 OpFlags := UdpRead() option flags\r
3aaddf92 1166 ServerIpPtr :=\r
1167 ServerPortPtr :=\r
1168 ClientIpPtr :=\r
1169 ClientPortPtr :=\r
1170 DhcpRxBuf :=\r
878ddf1f 1171 TimeoutEvent :=\r
1172\r
1173Returns:\r
1174--*/\r
1175{\r
1176 EFI_IP_ADDRESS ServerIp;\r
1177 EFI_STATUS StatCode;\r
1178 INTN RxBufLen;\r
1179\r
1180 for (;;) {\r
1181 //\r
1182 // Wait until we get a UDP packet.\r
1183 //\r
1184 ZeroMem (&ServerIp, sizeof (EFI_IP_ADDRESS));\r
1185 RxBufLen = sizeof RxBuf[0].u.ReceiveBuffer;\r
1186\r
1187 if ((StatCode = UdpRead (\r
1188 Private,\r
1189 OpFlags,\r
1190 ClientIpPtr,\r
1191 ClientPortPtr,\r
1192 ServerIpPtr,\r
1193 ServerPortPtr,\r
1194 0,\r
1195 0,\r
1196 (UINTN *) &RxBufLen,\r
1197 &DhcpRxBuf->u.Dhcpv4,\r
1198 TimeoutEvent\r
1199 )) != EFI_SUCCESS) {\r
1200 if (StatCode == EFI_TIMEOUT) {\r
1201 StatCode = EFI_NO_RESPONSE;\r
1202 }\r
1203\r
1204 break;\r
1205 }\r
1206 //\r
1207 // got a packet - see if a good offer\r
1208 //\r
1209 if (DhcpRxBuf->u.Dhcpv4.op != BOOTP_REPLY) {\r
1210 continue;\r
1211 }\r
1212\r
1213 if (DhcpRxBuf->u.Dhcpv4.xid != DHCPV4_TRANSMIT_BUFFER.xid) {\r
1214 continue;\r
1215 }\r
1216\r
1217 if (*(UINT32 *) DHCPV4_TRANSMIT_BUFFER.options != * (UINT32 *) DhcpRxBuf->u.Dhcpv4.options) {\r
1218 continue;\r
1219 }\r
1220\r
1221 if (*(UINT8 *) &DhcpRxBuf->u.Dhcpv4.yiaddr > 223) {\r
1222 continue;\r
1223 }\r
1224\r
1225 if (CompareMem (\r
1226 DhcpRxBuf->u.Dhcpv4.chaddr,\r
1227 DHCPV4_TRANSMIT_BUFFER.chaddr,\r
1228 sizeof DhcpRxBuf->u.Dhcpv4.chaddr\r
1229 )) {\r
1230 //\r
1231 // no good\r
1232 //\r
1233 continue;\r
1234 }\r
1235\r
1236 Parse (DhcpRxBuf, RxBufLen);\r
1237\r
1238 if (!(*ExtraEdit) (DhcpRxBuf)) {\r
1239 continue;\r
1240 }\r
1241 //\r
1242 // Good DHCP packet.\r
1243 //\r
1244 StatCode = EFI_SUCCESS;\r
1245 break;\r
1246 }\r
1247\r
1248 return StatCode;\r
1249}\r
1250\r
1251/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1252\r
1253//\r
1254// get DHCPOFFER's\r
1255//\r
1cc8ee78 1256STATIC\r
878ddf1f 1257EFI_STATUS\r
1258GetOffers (\r
1259 PXE_BASECODE_DEVICE *Private\r
1260 )\r
1261{\r
1262 EFI_IP_ADDRESS ClientIp;\r
1263 EFI_IP_ADDRESS ServerIp;\r
1264 EFI_STATUS StatCode;\r
1265 EFI_EVENT TimeoutEvent;\r
1266 INTN NumOffers;\r
1267 INTN Index;\r
1268\r
1269 //\r
1270 //\r
1271 //\r
1272 ZeroMem (&ServerIp, sizeof (EFI_IP_ADDRESS));\r
1273 NumOffers = 0;\r
1274\r
1275 for (Index = 0; Index < (sizeof Private->ServerCount) / sizeof Private->ServerCount[0]; ++Index) {\r
1276 Private->ServerCount[Index] = 0;\r
1277 Private->GotProxy[Index] = 0;\r
1278 }\r
1279\r
1280 Private->GotBootp = 0;\r
1281 //\r
1282 // these we throw away\r
1283 //\r
1284 Private->GotProxy[DHCP_ONLY_IX] = 1;\r
1285 StatCode = gBS->CreateEvent (\r
93b0fbc8 1286 EVT_TIMER,\r
1287 TPL_CALLBACK,\r
878ddf1f 1288 NULL,\r
1289 NULL,\r
1290 &TimeoutEvent\r
1291 );\r
1292\r
1293 if (EFI_ERROR (StatCode)) {\r
1294 return StatCode;\r
1295 }\r
1296\r
1297 StatCode = gBS->SetTimer (\r
1298 TimeoutEvent,\r
1299 TimerRelative,\r
1300 Private->Timeout * 10000000 + 1000000\r
1301 );\r
1302\r
1303 if (EFI_ERROR (StatCode)) {\r
1304 gBS->CloseEvent (TimeoutEvent);\r
1305 return StatCode;\r
1306 }\r
1307 //\r
1308 // get offers\r
1309 //\r
1310 for (;;) {\r
1311 DHCP_RECEIVE_BUFFER *DhcpRxBuf;\r
1312 UNION_PTR LocalPtr;\r
1313\r
1314 DhcpRxBuf = &RxBuf[NumOffers];\r
1315\r
1316 if ((\r
1317 StatCode = GetOfferAck (\r
1318 Private,\r
1319 DHCPOfferAckEdit,\r
1320 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_IP |\r
1321 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_DEST_IP |\r
1322 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_PORT,\r
1323 &ServerIp,\r
1324 &DhcpServerPort,\r
1325 &ClientIp,\r
1326 &DHCPClientPort,\r
1327 DhcpRxBuf,\r
1328 TimeoutEvent\r
1329 )\r
1330) != EFI_SUCCESS\r
1331 ) {\r
1332 break;\r
1333 }\r
1334\r
1335 LocalPtr.OpPtr = DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_MESSAGE_TYPE_IX - 1];\r
1336\r
1337 //\r
1338 // check type of offer\r
1339 //\r
1340 if (LocalPtr.OpPtr == NULL) {\r
1341 //\r
1342 // bootp - we only need one and make sure has bootfile\r
1343 //\r
1344 if (Private->GotBootp || !DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1]) {\r
1345 continue;\r
1346 }\r
1347\r
1348 Private->GotBootp = (UINT8) (NumOffers + 1);\r
1349 }\r
1350 //\r
1351 // if a DHCP type, must be DHCPOFFER and must have server id\r
1352 //\r
1353 else if (LocalPtr.MessageType->Type != DHCPOFFER || !DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1]) {\r
1354 continue;\r
1355 } else {\r
1356 INTN TypeIx;\r
1357\r
1358 //\r
1359 // get type - PXE10, WfM11a, or BINL\r
1360 //\r
1361 if (DhcpRxBuf->OpAdds.Status & DISCOVER_TYPE) {\r
1362 TypeIx = PXE10_IX;\r
1363 } else if (DhcpRxBuf->OpAdds.Status & WfM11a_TYPE) {\r
1364 //\r
1365 // WfM - make sure it has a bootfile\r
1366 //\r
1367 if (!DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1]) {\r
1368 continue;\r
1369 }\r
1370\r
1371 TypeIx = WfM11a_IX;\r
1372 } else {\r
1373 TypeIx = (DhcpRxBuf->OpAdds.Status & PXE_TYPE) ? BINL_IX : DHCP_ONLY_IX;\r
1374 }\r
1375 //\r
1376 // check DHCP or proxy\r
1377 //\r
1378 if (DhcpRxBuf->u.Dhcpv4.yiaddr == 0) {\r
1379 //\r
1380 // proxy - only need one of each type if not BINL\r
1381 // and must have at least PXE_TYPE\r
1382 //\r
1383 if (TypeIx == BINL_IX) {\r
1384 Private->BinlProxies[Private->GotProxy[BINL_IX]++] = (UINT8) NumOffers;\r
1385 } else if (Private->GotProxy[TypeIx]) {\r
1386 continue;\r
1387 } else {\r
1388 Private->GotProxy[TypeIx] = (UINT8) (NumOffers + 1);\r
1389 }\r
1390 } else {\r
1391 Private->OfferCount[TypeIx][Private->ServerCount[TypeIx]++] = (UINT8) NumOffers;\r
1392 }\r
1393 }\r
1394\r
1395 if (++NumOffers == MAX_OFFERS) {\r
1396 break;\r
1397 }\r
1398 }\r
1399\r
1400 gBS->CloseEvent (TimeoutEvent);\r
1401 Private->NumOffersReceived = NumOffers;\r
1402\r
1403 return (Private->NumOffersReceived) ? EFI_SUCCESS : EFI_NO_RESPONSE;\r
1404}\r
1405\r
1406/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1407\r
1408//\r
1409// send DHCPDECLINE\r
1410//\r
1411STATIC\r
1412VOID\r
1413DeclineOffer (\r
1414 PXE_BASECODE_DEVICE *Private\r
1415 )\r
1416{\r
1417 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
1418 UINT16 SaveSecs;\r
1419\r
1420 PxebcMode = Private->EfiBc.Mode;\r
1421 SaveSecs = DHCPV4_TRANSMIT_BUFFER.secs;\r
1422\r
1423 DHCPV4_TRANSMIT_BUFFER.secs = 0;\r
1424 DHCPV4_TRANSMIT_BUFFER.flags = 0;\r
1425 SetMem (\r
1426 DHCPV4_TRANSMIT_BUFFER.options + sizeof (struct opdeclinestr),\r
1427 sizeof (DHCPOpStart) - sizeof (struct opdeclinestr),\r
1428 OP_PAD\r
1429 );\r
1430 DHCPDECLINEoptions.DhcpMessageType.Type = DHCPDECLINE;\r
1431 CopyMem (&DHCPDECLINEoptions.OpDeclineEnd, &DHCP_REQ_OPTIONS, sizeof (struct requestopendstr));\r
1432 // DHCPDECLINEoptions.OpDeclineEnd = DHCP_REQ_OPTIONS;\r
1433\r
1434 {\r
1435 EFI_IP_ADDRESS TmpIp;\r
1436\r
1437 CopyMem (&TmpIp, &DHCP_REQ_OPTIONS.DhcServerIpPtr.Ip, sizeof TmpIp);\r
1438\r
1439 DoUdpWrite (\r
1440 Private,\r
1441 &TmpIp,\r
1442 &DhcpServerPort,\r
1443 &PxebcMode->StationIp,\r
1444 &DHCPClientPort\r
1445 );\r
1446 }\r
1447\r
1448 InitDhcpv4TxBuf (Private);\r
1449 DHCPV4_TRANSMIT_BUFFER.secs = SaveSecs;\r
1450 Private->GoodStationIp = FALSE;\r
1451}\r
1452\r
1453/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1454\r
1455//\r
1456// send DHCPRELEASE\r
1457//\r
1458STATIC\r
1459BOOLEAN\r
1460Release (\r
1461 PXE_BASECODE_DEVICE *Private\r
1462 )\r
1463{\r
1464 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
1465 UINT16 SaveSecs;\r
1466 DHCPV4_OP_SERVER_IP *Point;\r
1467\r
1468 PxebcMode = Private->EfiBc.Mode;\r
1469 SaveSecs = DHCPV4_TRANSMIT_BUFFER.secs;\r
1470 DHCPV4_TRANSMIT_BUFFER.secs = 0;\r
1471\r
1472 SetMem (\r
1473 DHCPV4_TRANSMIT_BUFFER.options + sizeof (struct opreleasestr),\r
1474 sizeof (DHCPOpStart) - sizeof (struct opreleasestr),\r
1475 OP_PAD\r
1476 );\r
1477\r
1478 DHCPRELEASEoptions.DhcpMessageType.Type = DHCPRELEASE;\r
1479 Point = (DHCPV4_OP_SERVER_IP *) DHCPV4_ACK_BUFFER.OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1];\r
1480 CopyMem (\r
1481 &DHCPRELEASEoptions.DhcServerIpPtr,\r
1482 &Point,\r
1483 sizeof DHCPRELEASEoptions.DhcServerIpPtr\r
1484 );\r
1485\r
1486 DHCPRELEASEoptions.End[0] = OP_END;\r
1487\r
1488 {\r
1489 EFI_IP_ADDRESS TmpIp;\r
1490\r
1491 CopyMem (&TmpIp, &DHCPRELEASEoptions.DhcServerIpPtr.Ip, sizeof TmpIp);\r
1492\r
1493 DoUdpWrite (\r
1494 Private,\r
1495 &TmpIp,\r
1496 &DhcpServerPort,\r
1497 &PxebcMode->StationIp,\r
1498 &DHCPClientPort\r
1499 );\r
1500 }\r
1501\r
1502 InitDhcpv4TxBuf (Private);\r
1503\r
1504 DHCPV4_TRANSMIT_BUFFER.secs = SaveSecs;\r
1505 Private->GoodStationIp = FALSE;\r
1506 return FALSE;\r
1507}\r
1508\r
1509/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1510STATIC\r
1511BOOLEAN\r
1512GetBINLAck (\r
1513 PXE_BASECODE_DEVICE *Private,\r
1514 EFI_IP_ADDRESS *ServerIpPtr\r
1515 )\r
1516{\r
1517 DHCP_RECEIVE_BUFFER *DhcpRxBuf;\r
1518 EFI_STATUS StatCode;\r
1519 EFI_EVENT TimeoutEvent;\r
1520\r
1521 //\r
1522 //\r
1523 //\r
1524 StatCode = gBS->CreateEvent (\r
93b0fbc8 1525 EVT_TIMER,\r
1526 TPL_CALLBACK,\r
878ddf1f 1527 NULL,\r
1528 NULL,\r
1529 &TimeoutEvent\r
1530 );\r
1531\r
1532 if (EFI_ERROR (StatCode)) {\r
1533 return FALSE;\r
1534 }\r
1535\r
1536 StatCode = gBS->SetTimer (\r
1537 TimeoutEvent,\r
1538 TimerRelative,\r
1539 Private->Timeout * 10000000 + 1000000\r
1540 );\r
1541\r
1542 if (EFI_ERROR (StatCode)) {\r
1543 gBS->CloseEvent (TimeoutEvent);\r
1544 return FALSE;\r
1545 }\r
1546 //\r
1547 //\r
1548 //\r
1549 DhcpRxBuf = &PXE_BINL_BUFFER;\r
1550\r
1551 for (;;) {\r
1552 EFI_PXE_BASE_CODE_UDP_PORT BINLSrvPort;\r
1553\r
1554 BINLSrvPort = 0;\r
1555\r
1556 if (GetOfferAck (\r
1557 Private,\r
1558 AckEdit,\r
1559 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_PORT,\r
1560 ServerIpPtr,\r
1561 &BINLSrvPort,\r
1562 &Private->EfiBc.Mode->StationIp,\r
1563 &PSEUDO_DHCP_CLIENT_PORT,\r
1564 DhcpRxBuf,\r
1565 TimeoutEvent\r
1566 ) != EFI_SUCCESS) {\r
1567 break;\r
1568 }\r
1569 //\r
1570 // make sure from whom we wanted\r
1571 //\r
1572 if (!DhcpRxBuf->u.Dhcpv4.yiaddr && !CompareMem (\r
1573 &ServerIpPtr->v4,\r
1574 &((DHCPV4_OP_SERVER_IP *) DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1])->Ip,\r
1575 sizeof (ServerIpPtr->v4)\r
1576 )) {\r
1577 gBS->CloseEvent (TimeoutEvent);\r
1578 //\r
1579 // got an ACK from server\r
1580 //\r
1581 return TRUE;\r
1582 }\r
1583 }\r
1584\r
1585 gBS->CloseEvent (TimeoutEvent);\r
1586 return FALSE;\r
1587}\r
1588\r
1589/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1590\r
1591//\r
1592// make sure we can get BINL\r
1593// send DHCPREQUEST to PXE server\r
1594//\r
1595STATIC\r
1596BOOLEAN\r
1597TryBINL (\r
1598 PXE_BASECODE_DEVICE *Private,\r
1599 INTN OfferIx\r
1600 )\r
1601{\r
1602 DHCP_RECEIVE_BUFFER *DhcpRxBuf;\r
1603 EFI_IP_ADDRESS ServerIp;\r
1604 UINT16 SaveSecs;\r
1605 INTN Index;\r
1606\r
1607 DhcpRxBuf = &RxBuf[OfferIx];\r
1608\r
1609 //\r
9887e0d0 1610 // use next server address first.\r
878ddf1f 1611 //\r
9887e0d0 1612 ServerIp.Addr[0] = DhcpRxBuf->u.Dhcpv4.siaddr;\r
1613 if (ServerIp.Addr[0] == 0) {\r
1614 //\r
1615 // next server address is NULL, use option 54.\r
1616 //\r
1617 CopyMem (\r
1618 ((EFI_IPv4_ADDRESS *) &ServerIp),\r
1619 &((DHCPV4_OP_SERVER_IP *) DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1])->Ip,\r
1620 sizeof (EFI_IPv4_ADDRESS)\r
1621 );\r
1622 }\r
878ddf1f 1623\r
1624 //\r
1625 // client IP address - filled in by client if it knows it\r
1626 //\r
1627 CopyMem (\r
1628 ((EFI_IPv4_ADDRESS *) &DHCPV4_TRANSMIT_BUFFER.ciaddr),\r
1629 &DHCP_REQ_OPTIONS.OpReqIP.Ip,\r
1630 sizeof (EFI_IPv4_ADDRESS)\r
1631 );\r
1632\r
1633 SetMem (&DHCP_REQ_OPTIONS, sizeof DHCP_REQ_OPTIONS, OP_PAD);\r
1634 DHCPV4_TRANSMIT_BUFFER.flags = 0;\r
1635 DHCPV4_OPTIONS_BUFFER.End[0] = OP_END;\r
1636 AddRouters (Private, DhcpRxBuf);\r
1637 SaveSecs = DHCPV4_TRANSMIT_BUFFER.secs;\r
1638\r
1639 for (Index = 0; Index < 3; Private->TotalSeconds = (UINT16) (Private->TotalSeconds + Private->Timeout), ++Index) {\r
1640 DHCPV4_TRANSMIT_BUFFER.secs = HTONS (Private->TotalSeconds);\r
1641\r
1642 //\r
1643 // unicast DHCPREQUEST to PXE server\r
1644 //\r
1645 if (DoUdpWrite (\r
1646 Private,\r
1647 &ServerIp,\r
1648 &PseudoDhcpServerPort,\r
1649 (EFI_IP_ADDRESS *) &DHCPV4_TRANSMIT_BUFFER.ciaddr,\r
1650 &PSEUDO_DHCP_CLIENT_PORT\r
1651 ) != EFI_SUCCESS) {\r
1652 break;\r
1653 }\r
1654\r
1655 if (!GetBINLAck (Private, &ServerIp)) {\r
1656 continue;\r
1657 }\r
1658 //\r
1659 // early exit failures\r
1660 // make sure a good ACK\r
1661 //\r
1662 if (!DHCPOfferAckEdit (&PXE_BINL_BUFFER) || (\r
1663 !(PXE_BINL_BUFFER.OpAdds.Status & DISCOVER_TYPE) && !PXE_BINL_BUFFER.OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1]\r
1664 )\r
1665 ) {\r
1666 break;\r
1667 }\r
1668\r
1669 Private->EfiBc.Mode->ProxyOfferReceived = TRUE;\r
1670 return TRUE;\r
1671 }\r
1672 //\r
1673 // failed - reset seconds field, etc.\r
1674 //\r
1675 Private->EfiBc.Mode->RouteTableEntries = 0;\r
1676 //\r
1677 // reset\r
1678 //\r
1679 DHCPV4_TRANSMIT_BUFFER.secs = SaveSecs;\r
1680 return FALSE;\r
1681}\r
1682\r
1683/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1684STATIC\r
1685BOOLEAN\r
1686TryFinishBINL (\r
1687 PXE_BASECODE_DEVICE *Private,\r
1688 INTN OfferIx\r
1689 )\r
1690{\r
1691 if (TryBINL (Private, OfferIx)) {\r
1692 return TRUE;\r
1693 }\r
1694\r
1695 return Release (Private);\r
1696}\r
1697\r
1698/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1699STATIC\r
1700BOOLEAN\r
1701TryFinishProxyBINL (\r
1702 PXE_BASECODE_DEVICE *Private\r
1703 )\r
1704{\r
1705 INTN Index;\r
1706\r
1707 for (Index = 0; Index < Private->GotProxy[BINL_IX]; ++Index) {\r
1708 if (TryBINL (Private, Private->BinlProxies[Index])) {\r
1709 return TRUE;\r
1710 }\r
1711 }\r
1712\r
1713 return Release (Private);\r
1714}\r
1715\r
1716/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1717\r
1718//\r
1719// try to finish DORA - send DHCP request, wait for ACK, check with ARP\r
1720//\r
1721STATIC\r
1722BOOLEAN\r
1723TryFinishDORA (\r
1724 PXE_BASECODE_DEVICE *Private,\r
1725 INTN OfferIx\r
1726 )\r
1727{\r
1728 DHCP_RECEIVE_BUFFER *DhcpRxBuf;\r
1729 EFI_IP_ADDRESS ClientIp;\r
1730 EFI_IP_ADDRESS ServerIp;\r
1731 EFI_STATUS StatCode;\r
1732 UNION_PTR LocalPtr;\r
1733 EFI_EVENT TimeoutEvent;\r
1734\r
1735 //\r
1736 // send DHCP request\r
1737 // if fail return false\r
1738 //\r
1739 DhcpRxBuf = &DHCPV4_ACK_BUFFER;\r
1740 DHCPV4_OPTIONS_BUFFER.DhcpMessageType.Type = DHCPREQUEST;\r
1741 CopyMem (&DHCP_REQ_OPTIONS, &RequestOpEndStr, sizeof (RequestOpEndStr));\r
1742// DHCP_REQ_OPTIONS = RequestOpEndStr;\r
1743 DHCP_REQ_OPTIONS.OpReqIP.Ip = *(EFI_IPv4_ADDRESS *) &RxBuf[OfferIx].u.Dhcpv4.yiaddr;\r
1744\r
1745 CopyMem (\r
1746 &DHCP_REQ_OPTIONS.DhcServerIpPtr.Ip,\r
1747 &((DHCPV4_OP_SERVER_IP *) RxBuf[OfferIx].OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1])->Ip,\r
1748 sizeof DHCP_REQ_OPTIONS.DhcServerIpPtr.Ip\r
1749 );\r
1750\r
1751 CopyMem (\r
1752 Private->EfiBc.Mode->SubnetMask.Addr,\r
1753 &DefaultSubnetMask,\r
1754 4\r
1755 );\r
1756\r
1757 //\r
1758 // broadcast DHCPREQUEST\r
1759 //\r
1760 if (DoUdpWrite (\r
1761 Private,\r
1762 &BroadcastIP,\r
1763 &DhcpServerPort,\r
1764 (EFI_IP_ADDRESS *) &DHCPV4_TRANSMIT_BUFFER.ciaddr,\r
1765 &DHCPClientPort\r
1766 ) != EFI_SUCCESS) {\r
1767 return FALSE;\r
1768 }\r
1769 //\r
1770 //\r
1771 //\r
1772 StatCode = gBS->CreateEvent (\r
93b0fbc8 1773 EVT_TIMER,\r
1774 TPL_CALLBACK,\r
878ddf1f 1775 NULL,\r
1776 NULL,\r
1777 &TimeoutEvent\r
1778 );\r
1779\r
1780 if (EFI_ERROR (StatCode)) {\r
1781 return FALSE;\r
1782 }\r
1783\r
1784 StatCode = gBS->SetTimer (\r
1785 TimeoutEvent,\r
1786 TimerPeriodic,\r
1787 Private->Timeout * 10000000 + 1000000\r
1788 );\r
1789\r
1790 if (EFI_ERROR (StatCode)) {\r
1791 gBS->CloseEvent (TimeoutEvent);\r
1792 return FALSE;\r
1793 }\r
1794 //\r
1795 // wait for ACK\r
1796 //\r
1797 for (;;) {\r
1798 if (GetOfferAck (\r
1799 Private,\r
1800 DHCPAckEdit,\r
1801 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_DEST_IP | EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_IP,\r
1802 &ServerIp,\r
1803 &DhcpServerPort,\r
1804 &ClientIp,\r
1805 &DHCPClientPort,\r
1806 DhcpRxBuf,\r
1807 TimeoutEvent\r
1808 ) != EFI_SUCCESS) {\r
1809 break;\r
1810 }\r
1811 //\r
1812 // check type of response - need DHCPACK\r
1813 //\r
1814 if (CompareMem (\r
1815 &DHCP_REQ_OPTIONS.OpReqIP.Ip,\r
1816 &DhcpRxBuf->u.Dhcpv4.yiaddr,\r
1817 sizeof (EFI_IPv4_ADDRESS)\r
1818 ) || CompareMem (\r
1819 &DHCP_REQ_OPTIONS.DhcServerIpPtr.Ip,\r
1820 &((DHCPV4_OP_SERVER_IP *) DhcpRxBuf->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1])->Ip,\r
1821 sizeof (EFI_IPv4_ADDRESS)\r
1822 )) {\r
1823 continue;\r
1824 }\r
1825 //\r
1826 // got ACK\r
1827 // check with ARP that IP unused - good return true\r
1828 //\r
1829 if (!SetStationIP (Private)) {\r
1830 //\r
1831 // fail - send DHCPDECLINE and return false\r
1832 //\r
1833 DeclineOffer (Private);\r
1834 break;\r
1835 }\r
1836\r
1837 LocalPtr.OpPtr = DHCPV4_ACK_BUFFER.OpAdds.PktOptAdds[OP_SUBNET_MASK_IX - 1];\r
1838\r
1839 if (LocalPtr.OpPtr != NULL) {\r
1840 CopyMem (\r
1841 (EFI_IPv4_ADDRESS *) &Private->EfiBc.Mode->SubnetMask,\r
1842 &LocalPtr.SubnetMaskStr->Ip,\r
1843 sizeof (EFI_IPv4_ADDRESS)\r
1844 );\r
1845 }\r
1846\r
1847 AddRouters (Private, DhcpRxBuf);\r
1848 gBS->CloseEvent (TimeoutEvent);\r
1849 return TRUE;\r
1850 }\r
1851\r
1852 gBS->CloseEvent (TimeoutEvent);\r
1853 return FALSE;\r
1854}\r
1855\r
1856/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1857\r
1858//\r
1859// try a DHCP server of appropriate type\r
1860//\r
1861STATIC\r
1862BOOLEAN\r
1863TryDHCPFinishDORA (\r
1864 PXE_BASECODE_DEVICE *Private,\r
1865 INTN TypeIx\r
1866 )\r
1867{\r
1868 INTN Index;\r
1869\r
1870 //\r
1871 // go through the DHCP servers of the requested type\r
1872 //\r
1873 for (Index = 0; Index < Private->ServerCount[TypeIx]; ++Index) {\r
1874 if (TryFinishDORA (Private, Index = Private->OfferCount[TypeIx][Index])) {\r
1875 if (TypeIx == BINL_IX && !TryFinishBINL (Private, Index)) {\r
1876 continue;\r
1877 }\r
1878\r
1879 return TRUE;\r
1880 }\r
1881 }\r
1882\r
1883 return FALSE;\r
1884}\r
1885\r
1886/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1887\r
1888//\r
1889// try a DHCP only server and a proxy of appropriate type\r
1890//\r
1891STATIC\r
1892BOOLEAN\r
1893TryProxyFinishDORA (\r
1894 PXE_BASECODE_DEVICE *Private,\r
1895 INTN TypeIx\r
1896 )\r
1897{\r
1898 INTN Index;\r
1899\r
1900 if (!Private->GotProxy[TypeIx]) {\r
1901 //\r
1902 // no proxies of the type wanted\r
1903 //\r
1904 return FALSE;\r
1905 }\r
1906 //\r
1907 // go through the DHCP only servers\r
1908 //\r
1909 for (Index = 0; Index < Private->ServerCount[DHCP_ONLY_IX]; ++Index) {\r
1910 if (TryFinishDORA (Private, Private->OfferCount[DHCP_ONLY_IX][Index])) {\r
1911 if (TypeIx != BINL_IX) {\r
1912 CopyProxyRxBuf (Private, Private->GotProxy[TypeIx] - 1);\r
1913 } else if (!TryFinishProxyBINL (Private)) {\r
1914 //\r
1915 // if didn't work with this DHCP, won't work with any\r
1916 //\r
1917 return FALSE;\r
1918 }\r
1919\r
1920 return TRUE;\r
1921 }\r
1922 }\r
1923\r
1924 return FALSE;\r
1925}\r
1926\r
1927/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1928\r
1929//\r
1930// getting to the bottom of the barrel\r
1931//\r
1932STATIC\r
1933BOOLEAN\r
1934TryAnyWithBootfileFinishDORA (\r
1935 PXE_BASECODE_DEVICE *Private\r
1936 )\r
1937{\r
1938 //\r
1939 // try a DHCP only server who has a bootfile\r
1940 //\r
1941 UNION_PTR LocalPtr;\r
1942 INTN Index;\r
1943\r
1944 for (Index = 0; Index < Private->ServerCount[DHCP_ONLY_IX]; ++Index) {\r
1945 INTN offer;\r
1946\r
1947 offer = Private->OfferCount[DHCP_ONLY_IX][Index];\r
1948\r
1949 if (RxBuf[offer].OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1] && TryFinishDORA (Private, offer)) {\r
1950 return TRUE;\r
1951 }\r
1952 }\r
1953 //\r
1954 // really at bottom - see if be have any bootps\r
1955 //\r
1956 if (!Private->GotBootp) {\r
1957 return FALSE;\r
1958 }\r
1959\r
1960 DHCP_REQ_OPTIONS.OpReqIP.Ip = *(EFI_IPv4_ADDRESS *) &RxBuf[Private->GotBootp - 1].u.Dhcpv4.yiaddr;\r
1961\r
1962 if (!SetStationIP (Private)) {\r
1963 return FALSE;\r
1964 }\r
1965 //\r
1966 // treat BOOTP response as DHCP ACK packet\r
1967 //\r
1968 CopyParseRxBuf (Private, Private->GotBootp - 1, DHCPV4_ACK_INDEX);\r
1969\r
1970 LocalPtr.OpPtr = RxBuf[Private->GotBootp - 1].OpAdds.PktOptAdds[OP_SUBNET_MASK_IX - 1];\r
1971\r
1972 if (LocalPtr.OpPtr != NULL) {\r
1973 *(EFI_IPv4_ADDRESS *) &Private->EfiBc.Mode->SubnetMask = LocalPtr.SubnetMaskStr->Ip;\r
1974 }\r
1975\r
1976 return TRUE;\r
1977}\r
1978\r
1979/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1980\r
1981/* DoDhcpDora()\r
1982 */\r
1983STATIC\r
1984EFI_STATUS\r
1985DoDhcpDora (\r
1986 PXE_BASECODE_DEVICE *Private,\r
1987 BOOLEAN SortOffers\r
1988 )\r
1989{\r
1990 EFI_PXE_BASE_CODE_IP_FILTER Filter;\r
1991 EFI_STATUS StatCode;\r
1992 INTN NumOffers;\r
1993\r
1994 Filter.Filters = EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP | EFI_PXE_BASE_CODE_IP_FILTER_BROADCAST;\r
1995\r
1996 Filter.IpCnt = 0;\r
1997 Filter.reserved = 0;\r
1998\r
1999 //\r
2000 // set filter unicast or broadcast\r
2001 //\r
2002 if ((StatCode = IpFilter (Private, &Filter)) != EFI_SUCCESS) {\r
2003 return StatCode;\r
2004 }\r
2005 //\r
2006 // seed random number with hardware address\r
2007 //\r
2008 SeedRandom (Private, *(UINT16 *) &Private->SimpleNetwork->Mode->CurrentAddress);\r
2009\r
2010 for (Private->Timeout = 1;\r
2011 Private->Timeout < 17;\r
2012 Private->TotalSeconds = (UINT16) (Private->TotalSeconds + Private->Timeout), Private->Timeout <<= 1\r
2013 ) {\r
2014 INTN Index;\r
2015\r
2016 InitDhcpv4TxBuf (Private);\r
2017 DHCPV4_TRANSMIT_BUFFER.xid = Random (Private);\r
2018 DHCPV4_TRANSMIT_BUFFER.secs = HTONS (Private->TotalSeconds);\r
2019\r
2020 //\r
2021 // broadcast DHCPDISCOVER\r
2022 //\r
2023 StatCode = DoUdpWrite (\r
2024 Private,\r
2025 &BroadcastIP,\r
2026 &DhcpServerPort,\r
2027 (EFI_IP_ADDRESS *) &DHCPV4_TRANSMIT_BUFFER.ciaddr,\r
2028 &DHCPClientPort\r
2029 );\r
2030\r
2031 if (StatCode != EFI_SUCCESS) {\r
2032 return StatCode;\r
2033 }\r
2034\r
2035 CopyMem (\r
2036 &Private->EfiBc.Mode->DhcpDiscover,\r
2037 (EFI_PXE_BASE_CODE_PACKET *) &DHCPV4_TRANSMIT_BUFFER,\r
2038 sizeof (EFI_PXE_BASE_CODE_PACKET)\r
2039 );\r
2040\r
2041 //\r
2042 // get DHCPOFFER's\r
2043 //\r
2044 if ((StatCode = GetOffers (Private)) != EFI_SUCCESS) {\r
2045 if (StatCode != EFI_NO_RESPONSE) {\r
2046 return StatCode;\r
2047 }\r
2048\r
2049 continue;\r
2050 }\r
2051 //\r
2052 // select offer and reply DHCPREQUEST\r
2053 //\r
2054 if (SortOffers) {\r
2055 if (TryDHCPFinishDORA(Private, PXE10_IX) || // try DHCP with PXE10\r
2056 TryDHCPFinishDORA(Private, WfM11a_IX) || // no - try with WfM\r
2057 TryProxyFinishDORA(Private, PXE10_IX) || // no - try DHCP only and proxy with PXE10\r
2058 TryProxyFinishDORA(Private, WfM11a_IX) || // no - try DHCP only and proxy with WfM\r
2059 TryDHCPFinishDORA(Private, BINL_IX) || // no - try with WfM\r
2060 TryProxyFinishDORA(Private, BINL_IX) || // no - try DHCP only and proxy with PXE10\r
2061 TryAnyWithBootfileFinishDORA(Private))\r
2062 {\r
2063 return EFI_SUCCESS;\r
2064 }\r
2065\r
2066 continue;\r
2067 }\r
2068 //\r
2069 // FIFO order\r
2070 //\r
2071 NumOffers = Private->NumOffersReceived;\r
2072\r
2073 for (Index = 0; Index < NumOffers; ++Index) {\r
2074 //\r
2075 // ignore proxies\r
2076 //\r
2077 if (!RxBuf[Index].u.Dhcpv4.yiaddr) {\r
2078 continue;\r
2079 }\r
2080 //\r
2081 // check if a bootp server\r
2082 //\r
2083 if (!RxBuf[Index].OpAdds.PktOptAdds[OP_DHCP_MESSAGE_TYPE_IX - 1]) {\r
2084 //\r
2085 // it is - just check ARP\r
2086 //\r
2087 if (!SetStationIP (Private)) {\r
2088 continue;\r
2089 }\r
2090 }\r
2091 //\r
2092 // else check if a DHCP only server\r
2093 //\r
2094 else if (!(RxBuf[Index].OpAdds.Status & (DISCOVER_TYPE | WfM11a_TYPE | PXE_TYPE))) {\r
2095 //\r
2096 // it is a normal DHCP offer (without any PXE options), just finish the D.O.R.A by sending DHCP request.\r
2097 //\r
2098 if (!TryFinishDORA (Private, Index)) {\r
2099 continue;\r
2100 }\r
2101 } else if (TryFinishDORA (Private, Index)) {\r
2102 if (!(RxBuf[Index].OpAdds.Status & (DISCOVER_TYPE | WfM11a_TYPE)) && !TryFinishBINL (Private, Index)) {\r
2103 continue;\r
2104 }\r
2105 }\r
2106\r
2107 DEBUG ((EFI_D_WARN, "\nDoDhcpDora() Got packets. "));\r
2108 return EFI_SUCCESS;\r
2109 }\r
2110 //\r
2111 // now look for DHCP onlys and a Proxy\r
2112 //\r
2113 for (Index = 0; Index < NumOffers; ++Index) {\r
0cc82df1 2114 UINT8 Index2;\r
878ddf1f 2115\r
2116 //\r
2117 // ignore proxies, bootps, non DHCP onlys, and bootable DHCPS\r
2118 //\r
2119 if (!RxBuf[Index].u.Dhcpv4.yiaddr ||\r
2120 !RxBuf[Index].OpAdds.PktOptAdds[OP_DHCP_MESSAGE_TYPE_IX - 1] ||\r
2121 RxBuf[Index].OpAdds.Status & (DISCOVER_TYPE | WfM11a_TYPE | PXE_TYPE) ||\r
2122 RxBuf[Index].OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1]\r
2123 ) {\r
2124 continue;\r
2125 }\r
2126 //\r
2127 // found non bootable DHCP only - try to find a proxy\r
2128 //\r
2129 for (Index2 = 0; Index2 < NumOffers; ++Index2) {\r
2130 if (!RxBuf[Index2].u.Dhcpv4.yiaddr) {\r
2131 if (!TryFinishDORA (Private, Index)) {\r
2132 //\r
2133 // DHCP no ACK\r
2134 //\r
2135 break;\r
2136 }\r
2137\r
2138 if (RxBuf[Index2].OpAdds.Status & (DISCOVER_TYPE | WfM11a_TYPE)) {\r
2139 CopyProxyRxBuf (Private, Index2);\r
2140 } else if (!TryFinishBINL (Private, Index2)) {\r
2141 continue;\r
2142 }\r
2143\r
2144 DEBUG ((EFI_D_WARN, "\nDoDhcpDora() Got packets. "));\r
2145 return EFI_SUCCESS;\r
2146 }\r
2147 }\r
2148 }\r
2149 }\r
2150\r
2151 return EFI_NO_RESPONSE;\r
2152}\r
2153\r
2154/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
2155\r
2156//\r
2157// determine if the server ip is in the ip list\r
2158//\r
1cc8ee78 2159STATIC\r
878ddf1f 2160BOOLEAN\r
2161InServerList (\r
2162 EFI_IP_ADDRESS *ServerIpPtr,\r
2163 PXE_SERVER_LISTS *ServerListPtr\r
2164 )\r
2165{\r
2166 UINTN Index;\r
2167\r
2168 if (!ServerListPtr || !ServerListPtr->Ipv4List.IpCount) {\r
2169 return TRUE;\r
2170 }\r
2171\r
2172 for (Index = 0; Index < ServerListPtr->Ipv4List.IpCount; ++Index) {\r
2173 if (!CompareMem (\r
2174 ServerIpPtr,\r
2175 &ServerListPtr->Ipv4List.IpList[Index],\r
2176 sizeof (EFI_IPv4_ADDRESS)\r
2177 )) {\r
2178 return TRUE;\r
2179 }\r
2180 }\r
2181\r
2182 return FALSE;\r
2183}\r
2184\r
2185/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 2186STATIC\r
878ddf1f 2187BOOLEAN\r
2188ExtractBootServerList (\r
2189 UINT16 Type,\r
2190 DHCPV4_OP_STRUCT *ptr,\r
2191 PXE_SERVER_LISTS **ServerListPtr\r
2192 )\r
2193{\r
2194 UNION_PTR LocalPtr;\r
2195 INTN ServerListLen;\r
2196\r
2197 LocalPtr.OpPtr = ptr;\r
2198 ServerListLen = LocalPtr.BootServersStr->Header.Length;\r
2199\r
2200 //\r
2201 // find type\r
2202 //\r
2203 LocalPtr.BootServerList = LocalPtr.BootServersStr->ServerList;\r
2204\r
2205 while (ServerListLen) {\r
2206 INTN ServerEntryLen;\r
2207\r
2208 ServerEntryLen = sizeof (PXEV4_SERVER_LIST) + 2 + (LocalPtr.BootServerList->u.Ipv4List.IpCount - 1) *\r
2209 sizeof (EFI_IPv4_ADDRESS);\r
2210\r
2211 if (NTOHS (LocalPtr.BootServerList->Type) == Type) {\r
2212 *ServerListPtr = &LocalPtr.BootServerList->u;\r
2213 return TRUE;\r
2214 }\r
2215\r
2216 (LocalPtr.BytePtr) += ServerEntryLen;\r
2217 ServerListLen -= ServerEntryLen;\r
2218 }\r
2219\r
2220 return FALSE;\r
2221}\r
2222\r
2223/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 2224STATIC\r
878ddf1f 2225VOID\r
2226FreeMem (\r
2227 PXE_BASECODE_DEVICE *Private\r
2228 )\r
2229{\r
2230 if (Private->TransmitBuffer != NULL) {\r
26aa0c2f 2231 FreePool (Private->TransmitBuffer);\r
878ddf1f 2232 Private->TransmitBuffer = NULL;\r
2233 }\r
2234\r
2235 if (Private->ReceiveBuffers != NULL) {\r
26aa0c2f 2236 FreePool (Private->ReceiveBuffers);\r
878ddf1f 2237 Private->ReceiveBuffers = NULL;\r
2238 }\r
2239}\r
2240\r
2241/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 2242STATIC\r
878ddf1f 2243BOOLEAN\r
2244GetMem (\r
2245 PXE_BASECODE_DEVICE *Private\r
2246 )\r
2247{\r
878ddf1f 2248\r
2249 if (Private->DhcpPacketBuffer == NULL) {\r
26aa0c2f
LG		 2250 Private->DhcpPacketBuffer = AllocatePool (sizeof (DHCP_RECEIVE_BUFFER) * (PXE_BIS_INDEX + 1));\r
2251 if (Private->DhcpPacketBuffer == NULL) {\r
878ddf1f 2252 FreeMem (Private);\r
2253 return FALSE;\r
2254 }\r
2255 }\r
2256\r
26aa0c2f
LG		 2257 Private->TransmitBuffer = AllocatePool (sizeof (EFI_PXE_BASE_CODE_PACKET));\r
2258 if (Private->TransmitBuffer == NULL) {\r
2259 FreePool (Private->DhcpPacketBuffer);\r
878ddf1f 2260 Private->DhcpPacketBuffer = NULL;\r
878ddf1f 2261 FreeMem (Private);\r
2262 return FALSE;\r
2263 }\r
2264\r
26aa0c2f
LG		 2265 Private->ReceiveBuffers = AllocatePool (sizeof (DHCP_RECEIVE_BUFFER) * (MAX_OFFERS));\r
2266 if (Private->ReceiveBuffers == NULL) {\r
2267 FreePool (Private->TransmitBuffer);\r
2268 FreePool (Private->DhcpPacketBuffer);\r
878ddf1f 2269 Private->DhcpPacketBuffer = NULL;\r
2270 Private->TransmitBuffer = NULL;\r
878ddf1f 2271 FreeMem (Private);\r
2272 return FALSE;\r
2273 }\r
2274\r
2275 return TRUE;\r
2276}\r
2277\r
2278/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
2279EFI_STATUS\r
2280EFIAPI\r
2281BcDhcp (\r
2282 IN EFI_PXE_BASE_CODE_PROTOCOL *This,\r
2283 IN BOOLEAN SortOffers\r
2284 )\r
2285/*++\r
2286Routine description:\r
2287 standard DHCP Discover/Offer/Request/Ack session\r
2288 broadcast DHCPDISCOVER\r
2289 receive DHCPOFFER's\r
2290 broadcast DHCPREQUEST\r
2291 receive DHCPACK\r
2292 check (ARP) good IP\r
2293\r
2294Parameters:\r
2295 This := Pointer to PxeBc interface\r
3aaddf92 2296 SortOffers :=\r
878ddf1f 2297\r
2298Returns:\r
2299--*/\r
2300{\r
2301 EFI_PXE_BASE_CODE_IP_FILTER Filter;\r
2302 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
2303 PXE_BASECODE_DEVICE *Private;\r
2304 EFI_STATUS StatCode;\r
2305\r
2306 //\r
2307 // Lock the instance data and make sure started\r
2308 //\r
2309 StatCode = EFI_SUCCESS;\r
2310\r
2311 if (This == NULL) {\r
2312 DEBUG ((EFI_D_ERROR, "BC *This pointer == NULL"));\r
2313 return EFI_INVALID_PARAMETER;\r
2314 }\r
2315\r
2316 Private = CR (This, PXE_BASECODE_DEVICE, EfiBc, PXE_BASECODE_DEVICE_SIGNATURE);\r
2317\r
2318 if (Private == NULL) {\r
2319 DEBUG ((EFI_D_ERROR, "PXE_BASECODE_DEVICE pointer == NULL"));\r
2320 return EFI_INVALID_PARAMETER;\r
2321 }\r
2322\r
2323 EfiAcquireLock (&Private->Lock);\r
2324\r
2325 if (This->Mode == NULL || !This->Mode->Started) {\r
2326 DEBUG ((EFI_D_ERROR, "BC was not started."));\r
2327 EfiReleaseLock (&Private->Lock);\r
2328 return EFI_NOT_STARTED;\r
2329 }\r
2330\r
2331 Filter.Filters = EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP;\r
2332 Filter.IpCnt = 0;\r
2333 Filter.reserved = 0;\r
2334\r
2335 DEBUG ((EFI_D_INFO, "\nBcDhcp() Enter. "));\r
2336\r
2337 PxebcMode = Private->EfiBc.Mode;\r
2338\r
2339 if (!GetMem (Private)) {\r
2340 DEBUG ((EFI_D_ERROR, "\nBcDhcp() GetMem() failed.\n"));\r
2341 EfiReleaseLock (&Private->Lock);\r
2342 return EFI_OUT_OF_RESOURCES;\r
2343 }\r
2344\r
2345 PxebcMode->DhcpDiscoverValid = FALSE;\r
2346 PxebcMode->DhcpAckReceived = FALSE;\r
2347 PxebcMode->ProxyOfferReceived = FALSE;\r
2348\r
2349 Private->Function = EFI_PXE_BASE_CODE_FUNCTION_DHCP;\r
2350\r
2351 //\r
2352 // Issue BC command\r
2353 //\r
2354 if (Private->TotalSeconds == 0) {\r
2355 //\r
2356 // put in seconds field of DHCP send packets\r
2357 //\r
2358 Private->TotalSeconds = 4;\r
2359 }\r
2360\r
2361 if ((StatCode = DoDhcpDora (Private, SortOffers)) == EFI_SUCCESS) {\r
2362 //\r
2363 // success - copy packets\r
2364 //\r
2365 PxebcMode->DhcpDiscoverValid = PxebcMode->DhcpAckReceived = TRUE;\r
2366\r
2367 CopyMem (\r
2368 &PxebcMode->DhcpAck,\r
2369 (EFI_PXE_BASE_CODE_PACKET *) &DHCPV4_ACK_PACKET,\r
2370 sizeof (EFI_PXE_BASE_CODE_PACKET)\r
2371 );\r
2372\r
2373 if (PxebcMode->ProxyOfferReceived) {\r
2374 CopyMem (\r
2375 &PxebcMode->ProxyOffer,\r
2376 (EFI_PXE_BASE_CODE_PACKET *) &PXE_OFFER_PACKET,\r
2377 sizeof (EFI_PXE_BASE_CODE_PACKET)\r
2378 );\r
2379 }\r
2380 }\r
2381 //\r
2382 // set filter back to unicast\r
2383 //\r
2384 IpFilter (Private, &Filter);\r
2385\r
2386 FreeMem (Private);\r
2387\r
2388 //\r
2389 // Unlock the instance data\r
2390 //\r
2391 DEBUG ((EFI_D_WARN, "\nBcDhcp() Exit = %xh ", StatCode));\r
2392\r
2393 EfiReleaseLock (&Private->Lock);\r
2394 return StatCode;\r
2395}\r
2396\r
2397/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
2398STATIC\r
2399BOOLEAN\r
2400VerifyCredentialOption (\r
2401 UINT8 *tx,\r
2402 UINT8 *rx\r
2403 )\r
2404{\r
2405 UINTN n;\r
2406\r
2407 //\r
2408 // Fail verification if either pointer is NULL.\r
2409 //\r
2410 if (tx == NULL || rx == NULL) {\r
2411 return FALSE;\r
2412 }\r
2413 //\r
2414 // Fail verification if tx[0] is not a credential type option\r
2415 // or if the length is zero or not a multiple of four.\r
2416 //\r
2417 if (tx[0] != VEND_PXE_CREDENTIAL_TYPES || tx[1] == 0 || tx[1] % 4 != 0) {\r
2418 return FALSE;\r
2419 }\r
2420 //\r
2421 // Fail verification if rx[0] is not a credential type option\r
2422 // or if the length is not equal to four.\r
2423 //\r
2424 if (rx[0] != VEND_PXE_CREDENTIAL_TYPES || rx[1] != 4) {\r
2425 return FALSE;\r
2426 }\r
2427 //\r
2428 // Look through transmitted credential types for a copy\r
2429 // of the received credential type.\r
2430 //\r
2431 for (n = 0; n < tx[1]; n += 4) {\r
2432 if (!CompareMem (&tx[n + 2], &rx[2], 4)) {\r
2433 return TRUE;\r
2434 }\r
2435 }\r
2436\r
2437 return FALSE;\r
2438}\r
2439\r
2440/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
1cc8ee78 2441STATIC\r
878ddf1f 2442EFI_STATUS\r
2443DoDiscover (\r
2444 PXE_BASECODE_DEVICE *Private,\r
2445 UINT16 OpFlags,\r
2446 IN UINT16 Type,\r
2447 IN UINT16 *LayerPtr,\r
2448 IN BOOLEAN UseBis,\r
2449 EFI_IP_ADDRESS *DestPtr,\r
2450 PXE_SERVER_LISTS *ServerListPtr\r
2451 )\r
2452/*++\r
3aaddf92 2453Routine description:\r
878ddf1f 2454 This function tries to complete the PXE Bootserver and/or boot image\r
2455 discovery sequence. When this command completes successfully, the\r
2456 PXEdiscover and PXEreply fields in the BC instance data structure are\r
2457 updated. If the Info pointer is set to NULL, the discovery information\r
3aaddf92 2458 in the DHCPack and ProxyOffer packets must be valid and will be used.\r
878ddf1f 2459 If Info is not set to NULL, the discovery methods in the Info field\r
2460 must be set and will be used. When discovering any layer number other\r
2461 than zero (the credential flag does not count), only unicast discovery\r
2462 is used.\r
2463\r
2464Parameters:\r
2465 Private := Pointer to PxeBc interface\r
3aaddf92 2466 OpFlags :=\r
2467 Type :=\r
2468 LayerPtr :=\r
2469 UseBis :=\r
2470 DestPtr :=\r
2471 ServerListPtr :=\r
878ddf1f 2472\r
2473Returns:\r
2474--*/\r
2475{\r
2476 EFI_PXE_BASE_CODE_UDP_PORT ClientPort;\r
2477 EFI_PXE_BASE_CODE_UDP_PORT ServerPort;\r
2478 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
2479 EFI_STATUS StatCode;\r
2480 EFI_EVENT TimeoutEvent;\r
2481 UINT8 OpLen;\r
2482\r
2483 PxebcMode = Private->EfiBc.Mode;\r
2484\r
2485 if (DestPtr->Addr[0] == 0) {\r
2486 DEBUG ((EFI_D_WARN, "\nDoDiscover() !DestPtr->Addr[0]"));\r
2487 return EFI_INVALID_PARAMETER;\r
2488 }\r
2489 //\r
2490 // seed random number with hardware address\r
2491 //\r
2492 SeedRandom (Private, *(UINT16 *) &Private->SimpleNetwork->Mode->CurrentAddress);\r
2493\r
2494 if (DestPtr->Addr[0] == BroadcastIP.Addr[0]) {\r
2495 ClientPort = DHCPClientPort;\r
2496 ServerPort = DhcpServerPort;\r
2497 } else {\r
2498 ClientPort = PSEUDO_DHCP_CLIENT_PORT;\r
2499 ServerPort = PseudoDhcpServerPort;\r
2500 }\r
2501\r
2502 if (UseBis) {\r
2503 *LayerPtr |= PXE_BOOT_LAYER_CREDENTIAL_FLAG;\r
2504 } else {\r
2505 *LayerPtr &= PXE_BOOT_LAYER_MASK;\r
2506 }\r
2507\r
2508 for (Private->Timeout = 1;\r
2509 Private->Timeout < 5;\r
2510 Private->TotalSeconds = (UINT16) (Private->TotalSeconds + Private->Timeout), ++Private->Timeout\r
2511 ) {\r
2512 InitDhcpv4TxBuf (Private);\r
2513 //\r
2514 // initialize DHCP message structure\r
2515 //\r
2516 DHCPV4_TRANSMIT_BUFFER.xid = Random (Private);\r
2517 DHCPV4_TRANSMIT_BUFFER.secs = HTONS (Private->TotalSeconds);\r
2518 CopyMem (\r
2519 &DHCPV4_TRANSMIT_BUFFER.ciaddr,\r
2520 &PxebcMode->StationIp,\r
2521 sizeof DHCPV4_TRANSMIT_BUFFER.ciaddr\r
2522 );\r
2523\r
2524 DHCPV4_OPTIONS_BUFFER.DhcpMessageType.Type = DHCPREQUEST;\r
2525 DISCOVERoptions.Header.OpCode = OP_VENDOR_SPECIFIC;\r
2526 DISCOVERoptions.BootItem.Header.OpCode = VEND_PXE_BOOT_ITEM;\r
2527 DISCOVERoptions.BootItem.Header.Length = DHCPV4_OPTION_LENGTH (PXE_OP_BOOT_ITEM);\r
2528 DISCOVERoptions.BootItem.Type = HTONS (Type);\r
2529 DISCOVERoptions.BootItem.Layer = HTONS (*LayerPtr);\r
2530\r
2531 if (UseBis) {\r
2532 EFI_BIS_PROTOCOL *BisPtr;\r
2533 BIS_APPLICATION_HANDLE BisAppHandle;\r
2534 EFI_BIS_DATA *BisDataSigInfo;\r
2535 EFI_BIS_SIGNATURE_INFO *BisSigInfo;\r
2536 UINTN Index;\r
2537 UINTN Index2;\r
2538\r
2539 BisPtr = PxebcBisStart (\r
2540 Private,\r
2541 &BisAppHandle,\r
2542 &BisDataSigInfo\r
2543 );\r
2544\r
2545 if (BisPtr == NULL) {\r
2546 //\r
2547 // %%TBD - In order to get here, BIS must have\r
2548 // been present when PXEBC.Start() was called.\r
2549 // BIS had to be shutdown/removed/damaged\r
2550 // before PXEBC.Discover() was called.\r
2551 // Do we need to document a specific error\r
2552 // for this case?\r
2553 //\r
2554 return EFI_OUT_OF_RESOURCES;\r
2555 }\r
2556 //\r
2557 // Compute number of credential types.\r
2558 //\r
2559 Index2 = BisDataSigInfo->Length / sizeof (EFI_BIS_SIGNATURE_INFO);\r
2560\r
2561 DISCREDoptions.Header.OpCode = VEND_PXE_CREDENTIAL_TYPES;\r
2562\r
2563 DISCREDoptions.Header.Length = (UINT8) (Index2 * sizeof (PXE_CREDENTIAL));\r
2564\r
2565 OpLen = (UINT8) (DHCPV4_OPTION_LENGTH (PXE_DISCOVER_OPTIONS) + sizeof (DHCPV4_OP_HEADER) + DISCREDoptions.Header.Length);\r
2566\r
2567 BisSigInfo = (EFI_BIS_SIGNATURE_INFO *) BisDataSigInfo->Data;\r
2568\r
2569 for (Index = 0; Index < Index2; ++Index) {\r
2570 UINT32 x;\r
2571\r
2572 CopyMem (&x, &BisSigInfo[Index], sizeof x);\r
2573 x = HTONL (x);\r
2574 CopyMem (&DISCREDoptions.Credentials[Index], &x, sizeof x);\r
2575 }\r
2576\r
2577 PxebcBisStop (BisPtr, BisAppHandle, BisDataSigInfo);\r
2578 } else {\r
2579 OpLen = DHCPV4_OPTION_LENGTH (PXE_DISCOVER_OPTIONS);\r
2580 }\r
2581\r
2582 DISCOVERoptions.Header.Length = OpLen;\r
2583\r
2584 ((UINT8 *) &DISCOVERoptions)[sizeof (DHCPV4_OP_HEADER) + OpLen - 1] = OP_END;\r
2585 ((UINT8 *) &DISCOVERoptions)[sizeof (DHCPV4_OP_HEADER) + OpLen] = OP_END;\r
2586\r
2587 StatCode = DoUdpWrite (\r
2588 Private,\r
2589 DestPtr,\r
2590 &ServerPort,\r
2591 (EFI_IP_ADDRESS *) &DHCPV4_TRANSMIT_BUFFER.ciaddr,\r
2592 &ClientPort\r
2593 );\r
2594\r
2595 if (StatCode != EFI_SUCCESS) {\r
2596 return StatCode;\r
2597 }\r
2598 //\r
2599 //\r
2600 //\r
2601 StatCode = gBS->CreateEvent (\r
93b0fbc8 2602 EVT_TIMER,\r
2603 TPL_CALLBACK,\r
878ddf1f 2604 NULL,\r
2605 NULL,\r
2606 &TimeoutEvent\r
2607 );\r
2608\r
2609 if (EFI_ERROR (StatCode)) {\r
2610 return StatCode;\r
2611 }\r
2612\r
2613 StatCode = gBS->SetTimer (\r
2614 TimeoutEvent,\r
2615 TimerRelative,\r
2616 Private->Timeout * 10000000 + 1000000\r
2617 );\r
2618\r
2619 if (EFI_ERROR (StatCode)) {\r
2620 gBS->CloseEvent (TimeoutEvent);\r
2621 return StatCode;\r
2622 }\r
2623 //\r
2624 // wait for ACK\r
2625 //\r
2626 for (;;) {\r
2627 DHCP_RECEIVE_BUFFER *RxBufPtr;\r
2628 UINT16 TmpType;\r
2629 UINT16 TmpLayer;\r
2630\r
2631 RxBufPtr = UseBis ? &PXE_BIS_BUFFER : &PXE_ACK_BUFFER;\r
2632 ZeroMem (&Private->ServerIp, sizeof (EFI_IP_ADDRESS));\r
2633\r
2634 if (GetOfferAck (\r
2635 Private,\r
2636 AckEdit,\r
2637 OpFlags,\r
2638 (EFI_IP_ADDRESS *) &Private->ServerIp,\r
2639 0,\r
2640 (EFI_IP_ADDRESS *) &DHCPV4_TRANSMIT_BUFFER.ciaddr,\r
2641 &ClientPort,\r
2642 RxBufPtr,\r
2643 TimeoutEvent\r
2644 ) != EFI_SUCCESS) {\r
2645 break;\r
2646 }\r
2647 //\r
2648 // check type of response - need PXEClient DHCPACK of proper type with bootfile\r
2649 //\r
2650 if (!(RxBufPtr->OpAdds.Status & PXE_TYPE) ||\r
2651 (UseBis && (RxBufPtr->OpAdds.Status & USE_THREE_BYTE)) ||\r
2652 !RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_BOOTFILE_IX - 1] ||\r
2653 !RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX - 1] ||\r
2654 !InServerList((EFI_IP_ADDRESS *)&((DHCPV4_OP_SERVER_IP *)RxBufPtr->OpAdds.PktOptAdds[OP_DHCP_SERVER_IP_IX-1])->Ip, ServerListPtr)) {\r
2655\r
2656 continue;\r
2657 }\r
2658\r
2659 TmpType = TmpLayer = 0;\r
2660\r
2661 if (RxBufPtr->OpAdds.PxeOptAdds[VEND_PXE_BOOT_ITEM_IX - 1]) {\r
2662 TmpType = NTOHS (((PXE_OP_BOOT_ITEM *) RxBufPtr->OpAdds.PxeOptAdds[VEND_PXE_BOOT_ITEM_IX - 1])->Type);\r
2663\r
2664 if (RxBufPtr->OpAdds.Status & USE_THREE_BYTE) {\r
2665 TmpLayer = (UINT16) (((PXE_OP_BOOT_ITEM *) RxBufPtr->OpAdds.PxeOptAdds[VEND_PXE_BOOT_ITEM_IX - 1])->Layer >> 8);\r
2666 } else {\r
2667 TmpLayer = NTOHS (((PXE_OP_BOOT_ITEM *) RxBufPtr->OpAdds.PxeOptAdds[VEND_PXE_BOOT_ITEM_IX - 1])->Layer);\r
2668 }\r
2669 }\r
2670\r
2671 if (TmpType != Type) {\r
2672 continue;\r
2673 }\r
2674\r
2675 if (UseBis) {\r
2676 if (!RxBufPtr->OpAdds.PxeOptAdds[VEND_PXE_CREDENTIAL_TYPES_IX - 1]) {\r
2677 continue;\r
2678 }\r
2679\r
2680 if (!VerifyCredentialOption (\r
2681 (UINT8 *) &DISCREDoptions.Header,\r
2682 (UINT8 *) RxBufPtr->OpAdds.PxeOptAdds[VEND_PXE_CREDENTIAL_TYPES_IX - 1]\r
2683 )) {\r
2684 continue;\r
2685 }\r
2686 }\r
2687\r
2688 *LayerPtr = TmpLayer;\r
2689\r
2690 if (UseBis) {\r
2691 CopyMem (\r
2692 &PxebcMode->PxeBisReply,\r
2693 &RxBufPtr->u.Dhcpv4,\r
2694 sizeof (EFI_PXE_BASE_CODE_PACKET)\r
2695 );\r
2696\r
2697 PxebcMode->PxeBisReplyReceived = TRUE;\r
2698\r
2699 StatCode = DoDiscover (\r
2700 Private,\r
2701 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_PORT,\r
2702 Type,\r
2703 LayerPtr,\r
2704 FALSE,\r
2705 &Private->ServerIp,\r
2706 0\r
2707 );\r
2708\r
2709 gBS->CloseEvent (TimeoutEvent);\r
2710 return StatCode;\r
2711 }\r
2712\r
2713 PxebcMode->PxeDiscoverValid = PxebcMode->PxeReplyReceived = TRUE;\r
2714\r
2715 CopyMem (\r
2716 &PxebcMode->PxeDiscover,\r
2717 &*(EFI_PXE_BASE_CODE_PACKET *) &DHCPV4_TRANSMIT_BUFFER,\r
2718 sizeof (*(EFI_PXE_BASE_CODE_PACKET *) &DHCPV4_TRANSMIT_BUFFER)\r
2719 );\r
2720\r
2721 CopyMem (\r
2722 &PxebcMode->PxeReply,\r
2723 &*(EFI_PXE_BASE_CODE_PACKET *) &RxBufPtr->u.Dhcpv4,\r
2724 sizeof (*(EFI_PXE_BASE_CODE_PACKET *) &RxBufPtr->u.Dhcpv4)\r
2725 );\r
2726\r
2727 AddRouters (Private, RxBufPtr);\r
2728\r
2729 gBS->CloseEvent (TimeoutEvent);\r
2730 return EFI_SUCCESS;\r
2731 }\r
2732\r
2733 gBS->CloseEvent (TimeoutEvent);\r
2734 }\r
2735 //\r
2736 // end for loop\r
2737 //\r
2738 return EFI_TIMEOUT;\r
2739}\r
2740\r
2741/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
2742STATIC\r
2743EFI_STATUS\r
2744Discover (\r
2745 PXE_BASECODE_DEVICE *Private,\r
2746 IN UINT16 Type,\r
2747 IN UINT16 *LayerPtr,\r
2748 IN BOOLEAN UseBis,\r
2749 IN EFI_PXE_BASE_CODE_DISCOVER_INFO *DiscoverInfoPtr,\r
2750 PXE_SERVER_LISTS *McastServerListPtr,\r
2751 PXE_SERVER_LISTS *ServerListPtr\r
2752 )\r
2753/*++\r
2754Routine Description:\r
2755\r
2756Parameters:\r
2757 Private := Pointer to PxeBc interface\r
3aaddf92 2758 Type :=\r
2759 LayerPtr :=\r
2760 UseBis :=\r
2761 DiscoverInfoPtr :=\r
2762 McastServerListPtr :=\r
2763 ServerListPtr :=\r
878ddf1f 2764\r
2765Returns:\r
2766--*/\r
2767{\r
2768 EFI_IP_ADDRESS DestIp;\r
2769 EFI_STATUS StatCode;\r
2770\r
2771 DEBUG ((EFI_D_INFO, "\nDiscover() Type=%d Layer=%d ", Type, *LayerPtr));\r
2772\r
2773 if (UseBis) {\r
2774 DEBUG ((EFI_D_INFO, "BIS "));\r
2775 }\r
2776 //\r
2777 // get dest IP addr - mcast, bcast, or unicast\r
2778 //\r
2779 if (DiscoverInfoPtr->UseMCast) {\r
2780 DestIp.v4 = DiscoverInfoPtr->ServerMCastIp.v4;\r
2781\r
2782 DEBUG (\r
2783 (EFI_D_INFO,\r
2784 "\nDiscover() MCast %d.%d.%d.%d ",\r
2785 DestIp.v4.Addr[0],\r
2786 DestIp.v4.Addr[1],\r
2787 DestIp.v4.Addr[2],\r
2788 DestIp.v4.Addr[3])\r
2789 );\r
2790\r
2791 if ((StatCode = DoDiscover (\r
2792 Private,\r
2793 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_IP | EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_PORT,\r
2794 Type,\r
2795 LayerPtr,\r
2796 UseBis,\r
2797 &DestIp,\r
2798 McastServerListPtr\r
2799 )) != EFI_TIMEOUT) {\r
2800 DEBUG (\r
2801 (EFI_D_WARN,\r
2802 "\nDiscover() status == %r (%Xh)",\r
2803 StatCode,\r
2804 StatCode)\r
2805 );\r
2806\r
2807 return StatCode;\r
2808 }\r
2809 }\r
2810\r
2811 if (DiscoverInfoPtr->UseBCast) {\r
2812 DEBUG ((EFI_D_INFO, "\nDiscver() BCast "));\r
2813\r
2814 if ((StatCode = DoDiscover (\r
2815 Private,\r
2816 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_IP | EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_PORT,\r
2817 Type,\r
2818 LayerPtr,\r
2819 UseBis,\r
2820 &BroadcastIP,\r
2821 McastServerListPtr\r
2822 )) != EFI_TIMEOUT) {\r
2823\r
2824 DEBUG ((EFI_D_WARN, "\nDiscover() status == %r (%Xh)", StatCode, StatCode));\r
2825\r
2826 return StatCode;\r
2827 }\r
2828 }\r
2829\r
2830 if (DiscoverInfoPtr->UseUCast) {\r
2831 UINTN Index;\r
2832\r
2833 DEBUG (\r
2834 (EFI_D_INFO,\r
2835 "\nDiscover() UCast IP#=%d ",\r
2836 ServerListPtr->Ipv4List.IpCount)\r
2837 );\r
2838\r
2839 for (Index = 0; Index < ServerListPtr->Ipv4List.IpCount; ++Index) {\r
2840 CopyMem (&DestIp, &ServerListPtr->Ipv4List.IpList[Index], 4);\r
2841\r
2842 DEBUG (\r
2843 (EFI_D_INFO,\r
2844 "\nDiscover() UCast %d.%d.%d.%d ",\r
2845 DestIp.v4.Addr[0],\r
2846 DestIp.v4.Addr[1],\r
2847 DestIp.v4.Addr[2],\r
2848 DestIp.v4.Addr[3])\r
2849 );\r
2850\r
2851 if ((StatCode = DoDiscover (\r
2852 Private,\r
2853 EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_IP | EFI_PXE_BASE_CODE_UDP_OPFLAGS_ANY_SRC_PORT,\r
2854 Type,\r
2855 LayerPtr,\r
2856 UseBis,\r
2857 &DestIp,\r
2858 0\r
2859 )) != EFI_TIMEOUT) {\r
2860 DEBUG (\r
2861 (EFI_D_WARN,\r
2862 "\nDiscover() status == %r (%Xh)",\r
2863 StatCode,\r
2864 StatCode)\r
2865 );\r
2866\r
2867 return StatCode;\r
2868 }\r
2869 }\r
2870 }\r
2871\r
2872 DEBUG ((EFI_D_WARN, "\nDiscover() TIMEOUT"));\r
2873\r
2874 return EFI_TIMEOUT;\r
2875}\r
2876\r
2877/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
2878\r
2879/* BcDiscover()\r
2880 */\r
2881EFI_STATUS\r
2882EFIAPI\r
2883BcDiscover (\r
2884 IN EFI_PXE_BASE_CODE_PROTOCOL * This,\r
2885 IN UINT16 Type,\r
2886 IN UINT16 *LayerPtr,\r
2887 IN BOOLEAN UseBis,\r
2888 IN EFI_PXE_BASE_CODE_DISCOVER_INFO * DiscoverInfoPtr OPTIONAL\r
2889 )\r
2890/*++\r
2891Routine description:\r
2892\r
2893Parameters:\r
3aaddf92 2894 This :=\r
2895 Type :=\r
2896 LayerPtr :=\r
2897 UseBis :=\r
2898 DiscoverInfoPtr :=\r
878ddf1f 2899\r
2900Returns:\r
2901--*/\r
2902{\r
2903 EFI_PXE_BASE_CODE_DISCOVER_INFO DefaultInfo;\r
2904 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
2905 DHCP_RECEIVE_BUFFER *DhcpRxBuf;\r
2906 PXE_SERVER_LISTS DefaultSrvList;\r
2907 PXE_SERVER_LISTS *ServerListPtr;\r
2908 PXE_SERVER_LISTS *McastServerListPtr;\r
878ddf1f 2909 UNION_PTR LocalPtr;\r
2910 UINTN Index;\r
2911 UINTN Index2;\r
2912 BOOLEAN AcquiredSrvList;\r
2913 EFI_STATUS StatCode;\r
2914 PXE_BASECODE_DEVICE *Private;\r
2915\r
2916 //\r
2917 // Lock the instance data and make sure started\r
2918 //\r
2919 StatCode = EFI_SUCCESS;\r
2920\r
2921 if (This == NULL) {\r
2922 DEBUG ((EFI_D_ERROR, "BC *This pointer == NULL"));\r
2923 return EFI_INVALID_PARAMETER;\r
2924 }\r
2925\r
2926 Private = CR (This, PXE_BASECODE_DEVICE, EfiBc, PXE_BASECODE_DEVICE_SIGNATURE);\r
2927\r
2928 if (Private == NULL) {\r
2929 DEBUG ((EFI_D_ERROR, "PXE_BASECODE_DEVICE poiner == NULL"));\r
2930 return EFI_INVALID_PARAMETER;\r
2931 }\r
2932\r
2933 EfiAcquireLock (&Private->Lock);\r
2934\r
2935 if (This->Mode == NULL || !This->Mode->Started) {\r
2936 DEBUG ((EFI_D_ERROR, "BC was not started."));\r
2937 EfiReleaseLock (&Private->Lock);\r
2938 return EFI_NOT_STARTED;\r
2939 }\r
2940\r
2941 ServerListPtr = NULL;\r
2942 McastServerListPtr = NULL;\r
2943 AcquiredSrvList = FALSE;\r
2944\r
2945 PxebcMode = Private->EfiBc.Mode;\r
2946\r
2947 if (!GetMem (Private)) {\r
2948 EfiReleaseLock (&Private->Lock);\r
2949 return EFI_OUT_OF_RESOURCES;\r
2950 }\r
2951\r
2952 if (UseBis) {\r
2953 if (!PxebcMode->BisSupported) {\r
2954 EfiReleaseLock (&Private->Lock);\r
2955 return EFI_INVALID_PARAMETER;\r
2956 }\r
2957 }\r
2958\r
2959 Private->Function = EFI_PXE_BASE_CODE_FUNCTION_DISCOVER;\r
2960\r
2961 if (Private->TotalSeconds == 0) {\r
2962 //\r
2963 // put in seconds field of DHCP send packets\r
2964 //\r
2965 Private->TotalSeconds = 4;\r
2966 }\r
2967\r
2968 ZeroMem (&DefaultInfo, sizeof (EFI_PXE_BASE_CODE_DISCOVER_INFO));\r
2969\r
2970 //\r
2971 // if layer number not zero, use previous discover\r
2972 //\r
2973 if (*LayerPtr != 0) {\r
2974 DEBUG ((EFI_D_WARN, "\nBcDiscover() layer != 0"));\r
2975\r
2976 if (DiscoverInfoPtr != NULL) {\r
2977 DEBUG ((EFI_D_WARN, "\nBcDiscover() layer != 0 && DiscoverInfoPtr != NULL\n"));\r
2978\r
2979 EfiReleaseLock (&Private->Lock);\r
2980 return EFI_INVALID_PARAMETER;\r
2981 }\r
2982\r
2983 if (!PxebcMode->PxeDiscoverValid) {\r
2984 DEBUG ((EFI_D_WARN, "\nBcDiscover() layer != 0 && PxeDiscoverValid == 0\n"));\r
2985\r
2986 EfiReleaseLock (&Private->Lock);\r
2987 return EFI_INVALID_PARAMETER;\r
2988 }\r
2989\r
2990 if (!PxebcMode->PxeReplyReceived) {\r
2991 DEBUG ((EFI_D_WARN, "\nBcDiscover() layer != 0 && PxeReplyReceived == 0\n"));\r
2992\r
2993 EfiReleaseLock (&Private->Lock);\r
2994 return EFI_INVALID_PARAMETER;\r
2995 }\r
2996\r
2997 if (UseBis && !PxebcMode->PxeBisReplyReceived) {\r
2998 DEBUG ((EFI_D_WARN, "\nBcDiscover() layer != 0 && PxeBisReplyReceived == 0\n"));\r
2999\r
3000 EfiReleaseLock (&Private->Lock);\r
3001 return EFI_INVALID_PARAMETER;\r
3002 }\r
3003\r
3004 DefaultInfo.UseUCast = TRUE;\r
3005 DiscoverInfoPtr = &DefaultInfo;\r
3006\r
3007 DefaultSrvList.Ipv4List.IpCount = 1;\r
3008 CopyMem (&DefaultSrvList.Ipv4List.IpList[0], &Private->ServerIp, 4);\r
3009\r
3010 ServerListPtr = &DefaultSrvList;\r
3011 }\r
3012 //\r
3013 // layer is zero - see if info is supplied or if we need to use info from a cached offer\r
3014 //\r
3015 else if (!DiscoverInfoPtr) {\r
3016 //\r
3017 // not supplied - generate it\r
3018 // make sure that there is cached, appropriate information\r
3019 // if neither DhcpAck packet nor ProxyOffer packet has pxe info, fail\r
3020 //\r
3021 DhcpRxBuf = (PxebcMode->ProxyOfferReceived) ? &PXE_OFFER_BUFFER : &DHCPV4_ACK_BUFFER;\r
3022\r
3023 if (!PxebcMode->DhcpAckReceived || !(DhcpRxBuf->OpAdds.Status & DISCOVER_TYPE)) {\r
3024 DEBUG ((EFI_D_WARN, "\nBcDiscover() !ack && !proxy"));\r
3025 EfiReleaseLock (&Private->Lock);\r
3026 return EFI_INVALID_PARAMETER;\r
3027 }\r
3028\r
3029 DiscoverInfoPtr = &DefaultInfo;\r
3030\r
3031 LocalPtr.OpPtr = DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_DISCOVERY_CONTROL_IX - 1];\r
3032\r
3033 //\r
3034 // if multicast enabled, need multicast address\r
3035 //\r
3036 if (!(LocalPtr.DiscoveryControl->ControlBits & DISABLE_MCAST)) {\r
3037 DefaultInfo.UseMCast = TRUE;\r
3038\r
3039 CopyMem (\r
3040 ((EFI_IPv4_ADDRESS *) &DefaultInfo.ServerMCastIp),\r
3041 &((DHCPV4_OP_IP_ADDRESS *) DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_DISCOVERY_MCAST_ADDR_IX - 1])->Ip,\r
3042 sizeof (EFI_IPv4_ADDRESS)\r
3043 );\r
3044 }\r
3045\r
3046 DefaultInfo.UseBCast = (BOOLEAN) ((LocalPtr.DiscoveryControl->ControlBits & DISABLE_BCAST) == 0);\r
3047\r
3048 DefaultInfo.MustUseList = (BOOLEAN) ((LocalPtr.DiscoveryControl->ControlBits & USE_ACCEPT_LIST) != 0);\r
3049\r
3050 DefaultInfo.UseUCast = (BOOLEAN)\r
3051 (\r
3052 (DefaultInfo.MustUseList) ||\r
3053 ((LocalPtr.DiscoveryControl->ControlBits & (DISABLE_MCAST | DISABLE_BCAST)) == (DISABLE_MCAST | DISABLE_BCAST))\r
3054 );\r
3055\r
3056 if ((DefaultInfo.UseUCast | DefaultInfo.MustUseList) && !ExtractBootServerList (\r
3057 Type,\r
3058 DhcpRxBuf->OpAdds.PxeOptAdds[VEND_PXE_BOOT_SERVERS_IX - 1],\r
3059 &ServerListPtr\r
3060 )) {\r
3061 DEBUG ((EFI_D_WARN, "\nBcDiscover() type not in list"));\r
3062 EfiReleaseLock (&Private->Lock);\r
3063 return EFI_INVALID_PARAMETER;\r
3064 }\r
3065 }\r
3066 //\r
3067 // Info supplied - make SrvList if required\r
3068 // if we use ucast discovery or must use list, there better be one\r
3069 //\r
3070 else if (DiscoverInfoPtr->UseUCast || DiscoverInfoPtr->MustUseList) {\r
3071 //\r
3072 // there better be a list\r
3073 //\r
3074 if (DiscoverInfoPtr->IpCnt == 0) {\r
3075 DEBUG ((EFI_D_WARN, "\nBcDiscover() no bootserver list"));\r
3076 EfiReleaseLock (&Private->Lock);\r
3077 return EFI_INVALID_PARAMETER;\r
3078 }\r
3079 //\r
3080 // get its size\r
3081 //\r
3082 for (Index = Index2 = 0; Index < DiscoverInfoPtr->IpCnt; ++Index) {\r
3083 if (DiscoverInfoPtr->SrvList[Index].Type == Type) {\r
3084 if (DiscoverInfoPtr->SrvList[Index].AcceptAnyResponse) {\r
3085 if (Index2 != 0) {\r
3086 DEBUG ((EFI_D_WARN, "\nBcDiscover() accept any?"));\r
3087 EfiReleaseLock (&Private->Lock);\r
3088 return EFI_INVALID_PARAMETER;\r
3089 } else {\r
3090 Index2 = 1;\r
3091 DefaultSrvList.Ipv4List.IpCount = 0;\r
3092 ServerListPtr = &DefaultSrvList;\r
3093 break;\r
3094 }\r
3095 } else {\r
3096 ++Index2;\r
3097 }\r
3098 }\r
3099 }\r
3100\r
3101 if (Index2 == 0) {\r
3102 DEBUG ((EFI_D_WARN, "\nBcDiscover() !Index2?"));\r
3103 EfiReleaseLock (&Private->Lock);\r
3104 return EFI_INVALID_PARAMETER;\r
3105 }\r
3106\r
3107 if (ServerListPtr == NULL) {\r
26aa0c2f
LG		 3108 ServerListPtr = AllocatePool (\r
3109 sizeof (PXEV4_SERVER_LIST) + (Index2 - 1) * sizeof (EFI_IPv4_ADDRESS)\r
878ddf1f 3110 );\r
3111\r
26aa0c2f 3112 if (ServerListPtr == NULL) {\r
878ddf1f 3113 EfiReleaseLock (&Private->Lock);\r
3114 return EFI_OUT_OF_RESOURCES;\r
3115 }\r
3116 //\r
3117 // build an array of IP addresses from the server list\r
3118 //\r
3119 AcquiredSrvList = TRUE;\r
3120 ServerListPtr->Ipv4List.IpCount = (UINT8) Index2;\r
3121\r
3122 for (Index = Index2 = 0; Index < DiscoverInfoPtr->IpCnt; ++Index) {\r
3123 if (DiscoverInfoPtr->SrvList[Index].Type == Type) {\r
3124 CopyMem (\r
3125 &ServerListPtr->Ipv4List.IpList[Index2++],\r
3126 &DiscoverInfoPtr->SrvList[Index].IpAddr.v4,\r
3127 sizeof ServerListPtr->Ipv4List.IpList[0]\r
3128 );\r
3129 }\r
3130 }\r
3131 }\r
3132 }\r
3133\r
3134 if (DiscoverInfoPtr->MustUseList) {\r
3135 McastServerListPtr = ServerListPtr;\r
3136 }\r
3137\r
3138 if (!(DiscoverInfoPtr->UseMCast || DiscoverInfoPtr->UseBCast || DiscoverInfoPtr->UseUCast)) {\r
3139 DEBUG ((EFI_D_WARN, "\nBcDiscover() Nothing to use!\n"));\r
3140\r
3141 EfiReleaseLock (&Private->Lock);\r
3142 return EFI_INVALID_PARAMETER;\r
3143 }\r
3144\r
3145 PxebcMode->PxeDiscoverValid = PxebcMode->PxeReplyReceived = PxebcMode->PxeBisReplyReceived = FALSE;\r
3146\r
3147 StatCode = Discover (\r
3148 Private,\r
3149 Type,\r
3150 LayerPtr,\r
3151 UseBis,\r
3152 DiscoverInfoPtr,\r
3153 McastServerListPtr,\r
3154 ServerListPtr\r
3155 );\r
3156\r
3157 if (AcquiredSrvList) {\r
26aa0c2f 3158 FreePool (ServerListPtr);\r
878ddf1f 3159 }\r
3160\r
3161 FreeMem (Private);\r
3162\r
3163 //\r
3164 // Unlock the instance data\r
3165 //\r
3166 DEBUG (\r
3167 (EFI_D_INFO,\r
3168 "\nBcDiscover() status == %r (%Xh)\n",\r
3169 StatCode,\r
3170 StatCode)\r
3171 );\r
3172\r
3173 EfiReleaseLock (&Private->Lock);\r
3174 return StatCode;\r
3175}\r
3176\r
3177/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
3178EFI_STATUS\r
3179EFIAPI\r
3180BcSetPackets (\r
3181 IN EFI_PXE_BASE_CODE_PROTOCOL * This,\r
3182 BOOLEAN *NewDhcpDiscoverValid, OPTIONAL\r
3183 BOOLEAN *NewDhcpAckReceived, OPTIONAL\r
3184 BOOLEAN *NewProxyOfferReceived, OPTIONAL\r
3185 BOOLEAN *NewPxeDiscoverValid, OPTIONAL\r
3186 BOOLEAN *NewPxeReplyReceived, OPTIONAL\r
3187 BOOLEAN *NewPxeBisReplyReceived, OPTIONAL\r
3188 IN EFI_PXE_BASE_CODE_PACKET * NewDhcpDiscover, OPTIONAL\r
3189 IN EFI_PXE_BASE_CODE_PACKET * NewDhcpAck, OPTIONAL\r
3190 IN EFI_PXE_BASE_CODE_PACKET * NewProxyOffer, OPTIONAL\r
3191 IN EFI_PXE_BASE_CODE_PACKET * NewPxeDiscover, OPTIONAL\r
3192 IN EFI_PXE_BASE_CODE_PACKET * NewPxeReply, OPTIONAL\r
3193 IN EFI_PXE_BASE_CODE_PACKET * NewPxeBisReply OPTIONAL\r
3194 )\r
3195/*++\r
3196Routine description:\r
3197\r
3198Parameters:\r
3199\r
3200Returns:\r
3201--*/\r
3202{\r
3203 EFI_PXE_BASE_CODE_MODE *PxebcMode;\r
878ddf1f 3204 PXE_BASECODE_DEVICE *Private;\r
3205\r
3206 //\r
3207 // Lock the instance data and make sure started\r
3208 //\r
878ddf1f 3209\r
3210 if (This == NULL) {\r
3211 DEBUG ((EFI_D_ERROR, "BC *This pointer == NULL"));\r
3212 return EFI_INVALID_PARAMETER;\r
3213 }\r
3214\r
3215 Private = CR (This, PXE_BASECODE_DEVICE, EfiBc, PXE_BASECODE_DEVICE_SIGNATURE);\r
3216\r
3217 if (Private == NULL) {\r
3218 DEBUG ((EFI_D_ERROR, "PXE_BASECODE_DEVICE poiner == NULL"));\r
3219 return EFI_INVALID_PARAMETER;\r
3220 }\r
3221\r
3222 EfiAcquireLock (&Private->Lock);\r
3223\r
3224 if (This->Mode == NULL || !This->Mode->Started) {\r
3225 DEBUG ((EFI_D_ERROR, "BC was not started."));\r
3226 EfiReleaseLock (&Private->Lock);\r
3227 return EFI_NOT_STARTED;\r
3228 }\r
3229\r
3230 PxebcMode = Private->EfiBc.Mode;\r
3231\r
3232 if (Private->DhcpPacketBuffer == NULL) {\r
26aa0c2f
LG		 3233 Private->DhcpPacketBuffer = AllocatePool (sizeof (DHCP_RECEIVE_BUFFER) * (PXE_BIS_INDEX + 1));\r
3234 if (Private->DhcpPacketBuffer == NULL) {\r
878ddf1f 3235 EfiReleaseLock (&Private->Lock);\r
3236 return EFI_OUT_OF_RESOURCES;\r
3237 }\r
3238 }\r
3239 //\r
3240 // Issue BC command\r
3241 //\r
3242 //\r
3243 // reset\r
3244 //\r
3245 Private->FileSize = 0;\r
3246 if (NewDhcpDiscoverValid != NULL) {\r
3247 PxebcMode->DhcpDiscoverValid = *NewDhcpDiscoverValid;\r
3248 }\r
3249\r
3250 if (NewDhcpAckReceived != NULL) {\r
3251 PxebcMode->DhcpAckReceived = *NewDhcpAckReceived;\r
3252 }\r
3253\r
3254 if (NewProxyOfferReceived != NULL) {\r
3255 PxebcMode->ProxyOfferReceived = *NewProxyOfferReceived;\r
3256 }\r
3257\r
3258 if (NewPxeDiscoverValid != NULL) {\r
3259 PxebcMode->PxeDiscoverValid = *NewPxeDiscoverValid;\r
3260 }\r
3261\r
3262 if (NewPxeReplyReceived != NULL) {\r
3263 PxebcMode->PxeReplyReceived = *NewPxeReplyReceived;\r
3264 }\r
3265\r
3266 if (NewPxeBisReplyReceived != NULL) {\r
3267 PxebcMode->PxeBisReplyReceived = *NewPxeBisReplyReceived;\r
3268 }\r
3269\r
3270 if (NewDhcpDiscover != NULL) {\r
3271 CopyMem (\r
3272 &PxebcMode->DhcpDiscover,\r
3273 NewDhcpDiscover,\r
3274 sizeof *NewDhcpDiscover\r
3275 );\r
3276 }\r
3277\r
3278 if (NewDhcpAck != NULL) {\r
3279 CopyParse (Private, &PxebcMode->DhcpAck, NewDhcpAck, DHCPV4_ACK_INDEX);\r
3280 }\r
3281\r
3282 if (NewProxyOffer != NULL) {\r
3283 CopyParse (Private, &PxebcMode->ProxyOffer, NewProxyOffer, PXE_OFFER_INDEX);\r
3284 }\r
3285\r
3286 if (NewPxeDiscover != NULL) {\r
3287 CopyMem (\r
3288 &PxebcMode->PxeDiscover,\r
3289 NewPxeDiscover,\r
3290 sizeof *NewPxeDiscover\r
3291 );\r
3292 }\r
3293\r
3294 if (NewPxeReply != NULL) {\r
3295 CopyParse (Private, &PxebcMode->PxeReply, NewPxeReply, PXE_ACK_INDEX);\r
3296 }\r
3297\r
3298 if (NewPxeBisReply != NULL) {\r
3299 CopyParse (Private, &PxebcMode->PxeBisReply, NewPxeBisReply, PXE_BIS_INDEX);\r
3300 }\r
3301 //\r
3302 // Unlock the instance data\r
3303 //\r
3304 EfiReleaseLock (&Private->Lock);\r
3305 return EFI_SUCCESS;\r
3306}\r
3307\r
3308/* eof - pxe_bc_dhcp.c */\r
EFI Development Kit II mirror for PVE