3 Copyright (c) 2006 - 2009, Intel Corporation.<BR>
4 All rights reserved. This program and the accompanying materials
5 are licensed and made available under the terms and conditions of the BSD License
6 which accompanies this distribution. The full text of the license may be found at
7 http://opensource.org/licenses/bsd-license.php
9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
14 #include "Dhcp4Impl.h"
15 #include "Dhcp4Driver.h"
17 EFI_DRIVER_BINDING_PROTOCOL gDhcp4DriverBinding
= {
18 Dhcp4DriverBindingSupported
,
19 Dhcp4DriverBindingStart
,
20 Dhcp4DriverBindingStop
,
26 EFI_SERVICE_BINDING_PROTOCOL mDhcp4ServiceBindingTemplate
= {
27 Dhcp4ServiceBindingCreateChild
,
28 Dhcp4ServiceBindingDestroyChild
32 This is the declaration of an EFI image entry point. This entry point is
33 the same for UEFI Applications, UEFI OS Loaders, and UEFI Drivers including
34 both device drivers and bus drivers.
36 Entry point of the DHCP driver to install various protocols.
38 @param[in] ImageHandle The firmware allocated handle for the UEFI image.
39 @param[in] SystemTable A pointer to the EFI System Table.
41 @retval EFI_SUCCESS The operation completed successfully.
42 @retval EFI_OUT_OF_RESOURCES The request could not be completed due to a lack of resources.
47 Dhcp4DriverEntryPoint (
48 IN EFI_HANDLE ImageHandle
,
49 IN EFI_SYSTEM_TABLE
*SystemTable
52 return EfiLibInstallDriverBindingComponentName2 (
64 Test to see if this driver supports ControllerHandle. This service
65 is called by the EFI boot service ConnectController(). In
66 order to make drivers as small as possible, there are a few calling
67 restrictions for this service. ConnectController() must
68 follow these calling restrictions. If any other agent wishes to call
69 Supported() it must also follow these calling restrictions.
71 @param[in] This Protocol instance pointer.
72 @param[in] ControllerHandle Handle of device to test
73 @param[in] RemainingDevicePath Optional parameter use to pick a specific child
76 @retval EFI_SUCCESS This driver supports this device
77 @retval EFI_ALREADY_STARTED This driver is already running on this device
78 @retval other This driver does not support this device
83 Dhcp4DriverBindingSupported (
84 IN EFI_DRIVER_BINDING_PROTOCOL
*This
,
85 IN EFI_HANDLE ControllerHandle
,
86 IN EFI_DEVICE_PATH_PROTOCOL
*RemainingDevicePath OPTIONAL
91 Status
= gBS
->OpenProtocol (
93 &gEfiUdp4ServiceBindingProtocolGuid
,
95 This
->DriverBindingHandle
,
97 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
106 Configure the default UDP child to receive all the DHCP traffics
107 on this network interface.
109 @param[in] UdpIo The UDP IO port to configure
110 @param[in] Context The context to the function
112 @retval EFI_SUCCESS The UDP IO port is successfully configured.
113 @retval Others Failed to configure the UDP child.
118 IN UDP_IO_PORT
*UdpIo
,
122 EFI_UDP4_CONFIG_DATA UdpConfigData
;
124 UdpConfigData
.AcceptBroadcast
= TRUE
;
125 UdpConfigData
.AcceptPromiscuous
= FALSE
;
126 UdpConfigData
.AcceptAnyPort
= FALSE
;
127 UdpConfigData
.AllowDuplicatePort
= TRUE
;
128 UdpConfigData
.TypeOfService
= 0;
129 UdpConfigData
.TimeToLive
= 64;
130 UdpConfigData
.DoNotFragment
= FALSE
;
131 UdpConfigData
.ReceiveTimeout
= 0;
132 UdpConfigData
.TransmitTimeout
= 0;
134 UdpConfigData
.UseDefaultAddress
= FALSE
;
135 UdpConfigData
.StationPort
= DHCP_CLIENT_PORT
;
136 UdpConfigData
.RemotePort
= DHCP_SERVER_PORT
;
138 ZeroMem (&UdpConfigData
.StationAddress
, sizeof (EFI_IPv4_ADDRESS
));
139 ZeroMem (&UdpConfigData
.SubnetMask
, sizeof (EFI_IPv4_ADDRESS
));
140 ZeroMem (&UdpConfigData
.RemoteAddress
, sizeof (EFI_IPv4_ADDRESS
));
142 return UdpIo
->Udp
->Configure (UdpIo
->Udp
, &UdpConfigData
);;
148 Destory the DHCP service. The Dhcp4 service may be partly initialized,
149 or partly destroyed. If a resource is destroyed, it is marked as so in
150 case the destroy failed and being called again later.
152 @param[in] DhcpSb The DHCP service instance to destory.
154 @retval EFI_SUCCESS Always return success.
159 IN DHCP_SERVICE
*DhcpSb
162 DhcpCleanLease (DhcpSb
);
164 if (DhcpSb
->UdpIo
!= NULL
) {
165 UdpIoFreePort (DhcpSb
->UdpIo
);
166 DhcpSb
->UdpIo
= NULL
;
169 if (DhcpSb
->Timer
!= NULL
) {
170 gBS
->SetTimer (DhcpSb
->Timer
, TimerCancel
, 0);
171 gBS
->CloseEvent (DhcpSb
->Timer
);
173 DhcpSb
->Timer
= NULL
;
182 Create a new DHCP service binding instance for the controller.
184 @param[in] Controller The controller to install DHCP service binding
186 @param[in] ImageHandle The driver's image handle
187 @param[out] Service The variable to receive the created DHCP service
190 @retval EFI_OUT_OF_RESOURCES Failed to allocate resource .
191 @retval EFI_SUCCESS The DHCP service instance is created.
192 @retval other Other error occurs.
197 IN EFI_HANDLE Controller
,
198 IN EFI_HANDLE ImageHandle
,
199 OUT DHCP_SERVICE
**Service
202 DHCP_SERVICE
*DhcpSb
;
206 DhcpSb
= AllocateZeroPool (sizeof (DHCP_SERVICE
));
208 if (DhcpSb
== NULL
) {
209 return EFI_OUT_OF_RESOURCES
;
212 DhcpSb
->Signature
= DHCP_SERVICE_SIGNATURE
;
213 DhcpSb
->ServiceState
= DHCP_UNCONFIGED
;
214 DhcpSb
->InDestory
= FALSE
;
215 DhcpSb
->Controller
= Controller
;
216 DhcpSb
->Image
= ImageHandle
;
217 InitializeListHead (&DhcpSb
->Children
);
218 DhcpSb
->DhcpState
= Dhcp4Stopped
;
219 DhcpSb
->Xid
= NET_RANDOM (NetRandomInitSeed ());
221 &DhcpSb
->ServiceBinding
,
222 &mDhcp4ServiceBindingTemplate
,
223 sizeof (EFI_SERVICE_BINDING_PROTOCOL
)
226 // Create various resources, UdpIo, Timer, and get Mac address
228 Status
= gBS
->CreateEvent (
229 EVT_NOTIFY_SIGNAL
| EVT_TIMER
,
236 if (EFI_ERROR (Status
)) {
240 DhcpSb
->UdpIo
= UdpIoCreatePort (Controller
, ImageHandle
, DhcpConfigUdpIo
, NULL
);
242 if (DhcpSb
->UdpIo
== NULL
) {
243 Status
= EFI_OUT_OF_RESOURCES
;
247 DhcpSb
->HwLen
= (UINT8
) DhcpSb
->UdpIo
->SnpMode
.HwAddressSize
;
248 DhcpSb
->HwType
= DhcpSb
->UdpIo
->SnpMode
.IfType
;
249 CopyMem (&DhcpSb
->Mac
, &DhcpSb
->UdpIo
->SnpMode
.CurrentAddress
, sizeof (DhcpSb
->Mac
));
255 Dhcp4CloseService (DhcpSb
);
263 Start this driver on ControllerHandle. This service is called by the
264 EFI boot service ConnectController(). In order to make
265 drivers as small as possible, there are a few calling restrictions for
266 this service. ConnectController() must follow these
267 calling restrictions. If any other agent wishes to call Start() it
268 must also follow these calling restrictions.
270 @param[in] This Protocol instance pointer.
271 @param[in] ControllerHandle Handle of device to bind driver to
272 @param[in] RemainingDevicePath Optional parameter use to pick a specific child
275 @retval EFI_SUCCESS This driver is added to ControllerHandle
276 @retval EFI_ALREADY_STARTED This driver is already running on ControllerHandle
277 @retval other This driver does not support this device
282 Dhcp4DriverBindingStart (
283 IN EFI_DRIVER_BINDING_PROTOCOL
*This
,
284 IN EFI_HANDLE ControllerHandle
,
285 IN EFI_DEVICE_PATH_PROTOCOL
*RemainingDevicePath OPTIONAL
288 DHCP_SERVICE
*DhcpSb
;
292 // First: test for the DHCP4 Protocol
294 Status
= gBS
->OpenProtocol (
296 &gEfiDhcp4ServiceBindingProtocolGuid
,
298 This
->DriverBindingHandle
,
300 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
303 if (Status
== EFI_SUCCESS
) {
304 return EFI_ALREADY_STARTED
;
307 Status
= Dhcp4CreateService (ControllerHandle
, This
->DriverBindingHandle
, &DhcpSb
);
309 if (EFI_ERROR (Status
)) {
314 // Start the receiving
316 Status
= UdpIoRecvDatagram (DhcpSb
->UdpIo
, DhcpInput
, DhcpSb
, 0);
318 if (EFI_ERROR (Status
)) {
321 Status
= gBS
->SetTimer (DhcpSb
->Timer
, TimerPeriodic
, TICKS_PER_SECOND
);
323 if (EFI_ERROR (Status
)) {
328 // Install the Dhcp4ServiceBinding Protocol onto ControlerHandle
330 Status
= gBS
->InstallMultipleProtocolInterfaces (
332 &gEfiDhcp4ServiceBindingProtocolGuid
,
333 &DhcpSb
->ServiceBinding
,
337 if (EFI_ERROR (Status
)) {
344 Dhcp4CloseService (DhcpSb
);
351 Stop this driver on ControllerHandle. This service is called by the
352 EFI boot service DisconnectController(). In order to
353 make drivers as small as possible, there are a few calling
354 restrictions for this service. DisconnectController()
355 must follow these calling restrictions. If any other agent wishes
356 to call Stop() it must also follow these calling restrictions.
358 @param[in] This Protocol instance pointer.
359 @param[in] ControllerHandle Handle of device to stop driver on
360 @param[in] NumberOfChildren Number of Handles in ChildHandleBuffer. If number of
361 children is zero stop the entire bus driver.
362 @param[in] ChildHandleBuffer List of Child Handles to Stop.
364 @retval EFI_SUCCESS This driver is removed ControllerHandle
365 @retval other This driver was not removed from this device
370 Dhcp4DriverBindingStop (
371 IN EFI_DRIVER_BINDING_PROTOCOL
*This
,
372 IN EFI_HANDLE ControllerHandle
,
373 IN UINTN NumberOfChildren
,
374 IN EFI_HANDLE
*ChildHandleBuffer
377 EFI_SERVICE_BINDING_PROTOCOL
*ServiceBinding
;
378 DHCP_SERVICE
*DhcpSb
;
379 DHCP_PROTOCOL
*Instance
;
380 EFI_HANDLE NicHandle
;
385 // DHCP driver opens UDP child, So, the ControllerHandle is the
386 // UDP child handle. locate the Nic handle first.
388 NicHandle
= NetLibGetNicHandle (ControllerHandle
, &gEfiUdp4ProtocolGuid
);
390 if (NicHandle
== NULL
) {
391 return EFI_DEVICE_ERROR
;
394 Status
= gBS
->OpenProtocol (
396 &gEfiDhcp4ServiceBindingProtocolGuid
,
397 (VOID
**) &ServiceBinding
,
398 This
->DriverBindingHandle
,
400 EFI_OPEN_PROTOCOL_GET_PROTOCOL
403 if (EFI_ERROR (Status
)) {
404 return EFI_DEVICE_ERROR
;
407 DhcpSb
= DHCP_SERVICE_FROM_THIS (ServiceBinding
);
409 if (DhcpSb
->InDestory
) {
413 OldTpl
= gBS
->RaiseTPL (TPL_CALLBACK
);
415 if (NumberOfChildren
== 0) {
417 DhcpSb
->InDestory
= TRUE
;
418 DhcpSb
->ServiceState
= DHCP_DESTORY
;
420 gBS
->UninstallProtocolInterface (
422 &gEfiDhcp4ServiceBindingProtocolGuid
,
426 Dhcp4CloseService (DhcpSb
);
431 // Don't use NET_LIST_FOR_EACH_SAFE here, Dhcp4ServiceBindingDestoryChild
432 // may cause other child to be deleted.
434 while (!IsListEmpty (&DhcpSb
->Children
)) {
435 Instance
= NET_LIST_HEAD (&DhcpSb
->Children
, DHCP_PROTOCOL
, Link
);
436 ServiceBinding
->DestroyChild (ServiceBinding
, Instance
->Handle
);
439 if (DhcpSb
->NumChildren
!= 0) {
440 Status
= EFI_DEVICE_ERROR
;
444 gBS
->RestoreTPL (OldTpl
);
451 Initialize a new DHCP instance.
453 @param DhcpSb The dhcp service instance
454 @param Instance The dhcp instance to initialize
459 IN DHCP_SERVICE
*DhcpSb
,
460 IN OUT DHCP_PROTOCOL
*Instance
463 Instance
->Signature
= DHCP_PROTOCOL_SIGNATURE
;
464 CopyMem (&Instance
->Dhcp4Protocol
, &mDhcp4ProtocolTemplate
, sizeof (Instance
->Dhcp4Protocol
));
465 InitializeListHead (&Instance
->Link
);
466 Instance
->Handle
= NULL
;
467 Instance
->Service
= DhcpSb
;
468 Instance
->InDestory
= FALSE
;
469 Instance
->CompletionEvent
= NULL
;
470 Instance
->RenewRebindEvent
= NULL
;
471 Instance
->Token
= NULL
;
472 Instance
->UdpIo
= NULL
;
473 NetbufQueInit (&Instance
->ResponseQueue
);
478 Creates a child handle and installs a protocol.
480 The CreateChild() function installs a protocol on ChildHandle.
481 If ChildHandle is a pointer to NULL, then a new handle is created and returned in ChildHandle.
482 If ChildHandle is not a pointer to NULL, then the protocol installs on the existing ChildHandle.
484 @param This Pointer to the EFI_SERVICE_BINDING_PROTOCOL instance.
485 @param ChildHandle Pointer to the handle of the child to create. If it is NULL,
486 then a new handle is created. If it is a pointer to an existing UEFI handle,
487 then the protocol is added to the existing UEFI handle.
489 @retval EFI_SUCCES The protocol was added to ChildHandle.
490 @retval EFI_INVALID_PARAMETER ChildHandle is NULL.
491 @retval EFI_OUT_OF_RESOURCES There are not enough resources availabe to create
493 @retval other The child handle was not created
498 Dhcp4ServiceBindingCreateChild (
499 IN EFI_SERVICE_BINDING_PROTOCOL
*This
,
500 IN EFI_HANDLE
*ChildHandle
503 DHCP_SERVICE
*DhcpSb
;
504 DHCP_PROTOCOL
*Instance
;
509 if ((This
== NULL
) || (ChildHandle
== NULL
)) {
510 return EFI_INVALID_PARAMETER
;
513 Instance
= AllocatePool (sizeof (*Instance
));
515 if (Instance
== NULL
) {
516 return EFI_OUT_OF_RESOURCES
;
519 DhcpSb
= DHCP_SERVICE_FROM_THIS (This
);
520 DhcpInitProtocol (DhcpSb
, Instance
);
523 // Install DHCP4 onto ChildHandle
525 Status
= gBS
->InstallMultipleProtocolInterfaces (
527 &gEfiDhcp4ProtocolGuid
,
528 &Instance
->Dhcp4Protocol
,
532 if (EFI_ERROR (Status
)) {
537 Instance
->Handle
= *ChildHandle
;
540 // Open the Udp4 protocol BY_CHILD.
542 Status
= gBS
->OpenProtocol (
543 DhcpSb
->UdpIo
->UdpHandle
,
544 &gEfiUdp4ProtocolGuid
,
546 gDhcp4DriverBinding
.DriverBindingHandle
,
548 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
550 if (EFI_ERROR (Status
)) {
551 gBS
->UninstallMultipleProtocolInterfaces (
553 &gEfiDhcp4ProtocolGuid
,
554 &Instance
->Dhcp4Protocol
,
562 OldTpl
= gBS
->RaiseTPL (TPL_CALLBACK
);
564 InsertTailList (&DhcpSb
->Children
, &Instance
->Link
);
565 DhcpSb
->NumChildren
++;
567 gBS
->RestoreTPL (OldTpl
);
574 Destroys a child handle with a protocol installed on it.
576 The DestroyChild() function does the opposite of CreateChild(). It removes a protocol
577 that was installed by CreateChild() from ChildHandle. If the removed protocol is the
578 last protocol on ChildHandle, then ChildHandle is destroyed.
580 @param This Pointer to the EFI_SERVICE_BINDING_PROTOCOL instance.
581 @param ChildHandle Handle of the child to destroy
583 @retval EFI_SUCCES The protocol was removed from ChildHandle.
584 @retval EFI_UNSUPPORTED ChildHandle does not support the protocol that is being removed.
585 @retval EFI_INVALID_PARAMETER Child handle is not a valid UEFI Handle.
586 @retval EFI_ACCESS_DENIED The protocol could not be removed from the ChildHandle
587 because its services are being used.
588 @retval other The child handle was not destroyed
593 Dhcp4ServiceBindingDestroyChild (
594 IN EFI_SERVICE_BINDING_PROTOCOL
*This
,
595 IN EFI_HANDLE ChildHandle
598 DHCP_SERVICE
*DhcpSb
;
599 DHCP_PROTOCOL
*Instance
;
600 EFI_DHCP4_PROTOCOL
*Dhcp
;
604 if ((This
== NULL
) || (ChildHandle
== NULL
)) {
605 return EFI_INVALID_PARAMETER
;
609 // Retrieve the private context data structures
611 Status
= gBS
->OpenProtocol (
613 &gEfiDhcp4ProtocolGuid
,
615 gDhcp4DriverBinding
.DriverBindingHandle
,
617 EFI_OPEN_PROTOCOL_GET_PROTOCOL
620 if (EFI_ERROR (Status
)) {
621 return EFI_UNSUPPORTED
;
624 Instance
= DHCP_INSTANCE_FROM_THIS (Dhcp
);
625 DhcpSb
= DHCP_SERVICE_FROM_THIS (This
);
627 if (Instance
->Service
!= DhcpSb
) {
628 return EFI_INVALID_PARAMETER
;
632 // A child can be destroyed more than once. For example,
633 // Dhcp4DriverBindingStop will destroy all of its children.
634 // when caller driver is being stopped, it will destory the
635 // dhcp child it opens.
637 if (Instance
->InDestory
) {
641 OldTpl
= gBS
->RaiseTPL (TPL_CALLBACK
);
642 Instance
->InDestory
= TRUE
;
645 // Close the Udp4 protocol.
648 DhcpSb
->UdpIo
->UdpHandle
,
649 &gEfiUdp4ProtocolGuid
,
650 gDhcp4DriverBinding
.DriverBindingHandle
,
655 // Uninstall the DHCP4 protocol first to enable a top down destruction.
657 Status
= gBS
->UninstallProtocolInterface (
659 &gEfiDhcp4ProtocolGuid
,
663 if (EFI_ERROR (Status
)) {
664 Instance
->InDestory
= FALSE
;
666 gBS
->RestoreTPL (OldTpl
);
670 if (DhcpSb
->ActiveChild
== Instance
) {
671 DhcpYieldControl (DhcpSb
);
674 RemoveEntryList (&Instance
->Link
);
675 DhcpSb
->NumChildren
--;
677 gBS
->RestoreTPL (OldTpl
);