3 Copyright (c) 2006 - 2015, Intel Corporation. All rights reserved.<BR>
4 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 to configure
110 @param[in] Context The context to the function
112 @retval EFI_SUCCESS The UDP IO is successfully configured.
113 @retval Others Failed to configure the UDP child.
123 EFI_UDP4_CONFIG_DATA UdpConfigData
;
125 UdpConfigData
.AcceptBroadcast
= TRUE
;
126 UdpConfigData
.AcceptPromiscuous
= FALSE
;
127 UdpConfigData
.AcceptAnyPort
= FALSE
;
128 UdpConfigData
.AllowDuplicatePort
= TRUE
;
129 UdpConfigData
.TypeOfService
= 0;
130 UdpConfigData
.TimeToLive
= 64;
131 UdpConfigData
.DoNotFragment
= FALSE
;
132 UdpConfigData
.ReceiveTimeout
= 0;
133 UdpConfigData
.TransmitTimeout
= 0;
135 UdpConfigData
.UseDefaultAddress
= FALSE
;
136 UdpConfigData
.StationPort
= DHCP_CLIENT_PORT
;
137 UdpConfigData
.RemotePort
= DHCP_SERVER_PORT
;
139 ZeroMem (&UdpConfigData
.StationAddress
, sizeof (EFI_IPv4_ADDRESS
));
140 ZeroMem (&UdpConfigData
.SubnetMask
, sizeof (EFI_IPv4_ADDRESS
));
141 ZeroMem (&UdpConfigData
.RemoteAddress
, sizeof (EFI_IPv4_ADDRESS
));
143 return UdpIo
->Protocol
.Udp4
->Configure (UdpIo
->Protocol
.Udp4
, &UdpConfigData
);;
149 Destroy the DHCP service. The Dhcp4 service may be partly initialized,
150 or partly destroyed. If a resource is destroyed, it is marked as so in
151 case the destroy failed and being called again later.
153 @param[in] DhcpSb The DHCP service instance to destroy.
155 @retval EFI_SUCCESS Always return success.
160 IN DHCP_SERVICE
*DhcpSb
163 DhcpCleanLease (DhcpSb
);
165 if (DhcpSb
->UdpIo
!= NULL
) {
166 UdpIoFreeIo (DhcpSb
->UdpIo
);
167 DhcpSb
->UdpIo
= NULL
;
170 if (DhcpSb
->Timer
!= NULL
) {
171 gBS
->SetTimer (DhcpSb
->Timer
, TimerCancel
, 0);
172 gBS
->CloseEvent (DhcpSb
->Timer
);
174 DhcpSb
->Timer
= NULL
;
183 Create a new DHCP service binding instance for the controller.
185 @param[in] Controller The controller to install DHCP service binding
187 @param[in] ImageHandle The driver's image handle
188 @param[out] Service The variable to receive the created DHCP service
191 @retval EFI_OUT_OF_RESOURCES Failed to allocate resource .
192 @retval EFI_SUCCESS The DHCP service instance is created.
193 @retval other Other error occurs.
198 IN EFI_HANDLE Controller
,
199 IN EFI_HANDLE ImageHandle
,
200 OUT DHCP_SERVICE
**Service
203 DHCP_SERVICE
*DhcpSb
;
207 DhcpSb
= AllocateZeroPool (sizeof (DHCP_SERVICE
));
209 if (DhcpSb
== NULL
) {
210 return EFI_OUT_OF_RESOURCES
;
213 DhcpSb
->Signature
= DHCP_SERVICE_SIGNATURE
;
214 DhcpSb
->ServiceState
= DHCP_UNCONFIGED
;
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
= UdpIoCreateIo (
248 if (DhcpSb
->UdpIo
== NULL
) {
249 Status
= EFI_OUT_OF_RESOURCES
;
253 DhcpSb
->HwLen
= (UINT8
) DhcpSb
->UdpIo
->SnpMode
.HwAddressSize
;
254 DhcpSb
->HwType
= DhcpSb
->UdpIo
->SnpMode
.IfType
;
255 CopyMem (&DhcpSb
->Mac
, &DhcpSb
->UdpIo
->SnpMode
.CurrentAddress
, sizeof (DhcpSb
->Mac
));
261 Dhcp4CloseService (DhcpSb
);
269 Start this driver on ControllerHandle. This service is called by the
270 EFI boot service ConnectController(). In order to make
271 drivers as small as possible, there are a few calling restrictions for
272 this service. ConnectController() must follow these
273 calling restrictions. If any other agent wishes to call Start() it
274 must also follow these calling restrictions.
276 @param[in] This Protocol instance pointer.
277 @param[in] ControllerHandle Handle of device to bind driver to
278 @param[in] RemainingDevicePath Optional parameter use to pick a specific child
281 @retval EFI_SUCCESS This driver is added to ControllerHandle
282 @retval EFI_ALREADY_STARTED This driver is already running on ControllerHandle
283 @retval other This driver does not support this device
288 Dhcp4DriverBindingStart (
289 IN EFI_DRIVER_BINDING_PROTOCOL
*This
,
290 IN EFI_HANDLE ControllerHandle
,
291 IN EFI_DEVICE_PATH_PROTOCOL
*RemainingDevicePath OPTIONAL
294 DHCP_SERVICE
*DhcpSb
;
298 // First: test for the DHCP4 Protocol
300 Status
= gBS
->OpenProtocol (
302 &gEfiDhcp4ServiceBindingProtocolGuid
,
304 This
->DriverBindingHandle
,
306 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
309 if (Status
== EFI_SUCCESS
) {
310 return EFI_ALREADY_STARTED
;
313 Status
= Dhcp4CreateService (ControllerHandle
, This
->DriverBindingHandle
, &DhcpSb
);
315 if (EFI_ERROR (Status
)) {
318 ASSERT (DhcpSb
!= NULL
);
321 // Start the receiving
323 Status
= UdpIoRecvDatagram (DhcpSb
->UdpIo
, DhcpInput
, DhcpSb
, 0);
325 if (EFI_ERROR (Status
)) {
328 Status
= gBS
->SetTimer (DhcpSb
->Timer
, TimerPeriodic
, TICKS_PER_SECOND
);
330 if (EFI_ERROR (Status
)) {
335 // Install the Dhcp4ServiceBinding Protocol onto ControlerHandle
337 Status
= gBS
->InstallMultipleProtocolInterfaces (
339 &gEfiDhcp4ServiceBindingProtocolGuid
,
340 &DhcpSb
->ServiceBinding
,
344 if (EFI_ERROR (Status
)) {
351 Dhcp4CloseService (DhcpSb
);
357 Callback function which provided by user to remove one node in NetDestroyLinkList process.
359 @param[in] Entry The entry to be removed.
360 @param[in] Context Pointer to the callback context corresponds to the Context in NetDestroyLinkList.
362 @retval EFI_SUCCESS The entry has been removed successfully.
363 @retval Others Fail to remove the entry.
368 Dhcp4DestroyChildEntry (
369 IN LIST_ENTRY
*Entry
,
373 DHCP_PROTOCOL
*Instance
;
374 EFI_SERVICE_BINDING_PROTOCOL
*ServiceBinding
;
376 if (Entry
== NULL
|| Context
== NULL
) {
377 return EFI_INVALID_PARAMETER
;
380 Instance
= NET_LIST_USER_STRUCT_S (Entry
, DHCP_PROTOCOL
, Link
, DHCP_PROTOCOL_SIGNATURE
);
381 ServiceBinding
= (EFI_SERVICE_BINDING_PROTOCOL
*) Context
;
383 return ServiceBinding
->DestroyChild (ServiceBinding
, Instance
->Handle
);
388 Stop this driver on ControllerHandle. This service is called by the
389 EFI boot service DisconnectController(). In order to
390 make drivers as small as possible, there are a few calling
391 restrictions for this service. DisconnectController()
392 must follow these calling restrictions. If any other agent wishes
393 to call Stop() it must also follow these calling restrictions.
395 @param[in] This Protocol instance pointer.
396 @param[in] ControllerHandle Handle of device to stop driver on
397 @param[in] NumberOfChildren Number of Handles in ChildHandleBuffer. If number of
398 children is zero stop the entire bus driver.
399 @param[in] ChildHandleBuffer List of Child Handles to Stop.
401 @retval EFI_SUCCESS This driver is removed ControllerHandle
402 @retval other This driver was not removed from this device
407 Dhcp4DriverBindingStop (
408 IN EFI_DRIVER_BINDING_PROTOCOL
*This
,
409 IN EFI_HANDLE ControllerHandle
,
410 IN UINTN NumberOfChildren
,
411 IN EFI_HANDLE
*ChildHandleBuffer
414 EFI_SERVICE_BINDING_PROTOCOL
*ServiceBinding
;
415 DHCP_SERVICE
*DhcpSb
;
416 EFI_HANDLE NicHandle
;
422 // DHCP driver opens UDP child, So, the ControllerHandle is the
423 // UDP child handle. locate the Nic handle first.
425 NicHandle
= NetLibGetNicHandle (ControllerHandle
, &gEfiUdp4ProtocolGuid
);
427 if (NicHandle
== NULL
) {
431 Status
= gBS
->OpenProtocol (
433 &gEfiDhcp4ServiceBindingProtocolGuid
,
434 (VOID
**) &ServiceBinding
,
435 This
->DriverBindingHandle
,
437 EFI_OPEN_PROTOCOL_GET_PROTOCOL
440 if (EFI_ERROR (Status
)) {
441 return EFI_DEVICE_ERROR
;
444 DhcpSb
= DHCP_SERVICE_FROM_THIS (ServiceBinding
);
445 if (!IsListEmpty (&DhcpSb
->Children
)) {
447 // Destroy all the children instances before destory the service.
449 List
= &DhcpSb
->Children
;
450 Status
= NetDestroyLinkList (
452 Dhcp4DestroyChildEntry
,
456 if (EFI_ERROR (Status
) || ListLength
!= 0) {
457 Status
= EFI_DEVICE_ERROR
;
461 if (NumberOfChildren
== 0 && !IsListEmpty (&DhcpSb
->Children
)) {
462 Status
= EFI_DEVICE_ERROR
;
465 if (NumberOfChildren
== 0 && IsListEmpty (&DhcpSb
->Children
)) {
467 // Destroy the service itself if no child instance left.
469 DhcpSb
->ServiceState
= DHCP_DESTROY
;
471 gBS
->UninstallProtocolInterface (
473 &gEfiDhcp4ServiceBindingProtocolGuid
,
477 Dhcp4CloseService (DhcpSb
);
479 if (gDhcpControllerNameTable
!= NULL
) {
480 FreeUnicodeStringTable (gDhcpControllerNameTable
);
481 gDhcpControllerNameTable
= NULL
;
485 Status
= EFI_SUCCESS
;
493 Initialize a new DHCP instance.
495 @param DhcpSb The dhcp service instance
496 @param Instance The dhcp instance to initialize
501 IN DHCP_SERVICE
*DhcpSb
,
502 IN OUT DHCP_PROTOCOL
*Instance
505 Instance
->Signature
= DHCP_PROTOCOL_SIGNATURE
;
506 CopyMem (&Instance
->Dhcp4Protocol
, &mDhcp4ProtocolTemplate
, sizeof (Instance
->Dhcp4Protocol
));
507 InitializeListHead (&Instance
->Link
);
508 Instance
->Handle
= NULL
;
509 Instance
->Service
= DhcpSb
;
510 Instance
->InDestroy
= FALSE
;
511 Instance
->CompletionEvent
= NULL
;
512 Instance
->RenewRebindEvent
= NULL
;
513 Instance
->Token
= NULL
;
514 Instance
->UdpIo
= NULL
;
515 Instance
->ElaspedTime
= 0;
516 NetbufQueInit (&Instance
->ResponseQueue
);
521 Creates a child handle and installs a protocol.
523 The CreateChild() function installs a protocol on ChildHandle.
524 If ChildHandle is a pointer to NULL, then a new handle is created and returned in ChildHandle.
525 If ChildHandle is not a pointer to NULL, then the protocol installs on the existing ChildHandle.
527 @param This Pointer to the EFI_SERVICE_BINDING_PROTOCOL instance.
528 @param ChildHandle Pointer to the handle of the child to create. If it is NULL,
529 then a new handle is created. If it is a pointer to an existing UEFI handle,
530 then the protocol is added to the existing UEFI handle.
532 @retval EFI_SUCCES The protocol was added to ChildHandle.
533 @retval EFI_INVALID_PARAMETER ChildHandle is NULL.
534 @retval EFI_OUT_OF_RESOURCES There are not enough resources availabe to create
536 @retval other The child handle was not created
541 Dhcp4ServiceBindingCreateChild (
542 IN EFI_SERVICE_BINDING_PROTOCOL
*This
,
543 IN EFI_HANDLE
*ChildHandle
546 DHCP_SERVICE
*DhcpSb
;
547 DHCP_PROTOCOL
*Instance
;
552 if ((This
== NULL
) || (ChildHandle
== NULL
)) {
553 return EFI_INVALID_PARAMETER
;
556 Instance
= AllocatePool (sizeof (*Instance
));
558 if (Instance
== NULL
) {
559 return EFI_OUT_OF_RESOURCES
;
562 DhcpSb
= DHCP_SERVICE_FROM_THIS (This
);
563 DhcpInitProtocol (DhcpSb
, Instance
);
566 // Install DHCP4 onto ChildHandle
568 Status
= gBS
->InstallMultipleProtocolInterfaces (
570 &gEfiDhcp4ProtocolGuid
,
571 &Instance
->Dhcp4Protocol
,
575 if (EFI_ERROR (Status
)) {
580 Instance
->Handle
= *ChildHandle
;
583 // Open the Udp4 protocol BY_CHILD.
585 Status
= gBS
->OpenProtocol (
586 DhcpSb
->UdpIo
->UdpHandle
,
587 &gEfiUdp4ProtocolGuid
,
589 gDhcp4DriverBinding
.DriverBindingHandle
,
591 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
593 if (EFI_ERROR (Status
)) {
594 gBS
->UninstallMultipleProtocolInterfaces (
596 &gEfiDhcp4ProtocolGuid
,
597 &Instance
->Dhcp4Protocol
,
605 OldTpl
= gBS
->RaiseTPL (TPL_CALLBACK
);
607 InsertTailList (&DhcpSb
->Children
, &Instance
->Link
);
608 DhcpSb
->NumChildren
++;
610 gBS
->RestoreTPL (OldTpl
);
617 Destroys a child handle with a protocol installed on it.
619 The DestroyChild() function does the opposite of CreateChild(). It removes a protocol
620 that was installed by CreateChild() from ChildHandle. If the removed protocol is the
621 last protocol on ChildHandle, then ChildHandle is destroyed.
623 @param This Pointer to the EFI_SERVICE_BINDING_PROTOCOL instance.
624 @param ChildHandle Handle of the child to destroy
626 @retval EFI_SUCCES The protocol was removed from ChildHandle.
627 @retval EFI_UNSUPPORTED ChildHandle does not support the protocol that is being removed.
628 @retval EFI_INVALID_PARAMETER Child handle is NULL.
629 @retval EFI_ACCESS_DENIED The protocol could not be removed from the ChildHandle
630 because its services are being used.
631 @retval other The child handle was not destroyed
636 Dhcp4ServiceBindingDestroyChild (
637 IN EFI_SERVICE_BINDING_PROTOCOL
*This
,
638 IN EFI_HANDLE ChildHandle
641 DHCP_SERVICE
*DhcpSb
;
642 DHCP_PROTOCOL
*Instance
;
643 EFI_DHCP4_PROTOCOL
*Dhcp
;
647 if ((This
== NULL
) || (ChildHandle
== NULL
)) {
648 return EFI_INVALID_PARAMETER
;
652 // Retrieve the private context data structures
654 Status
= gBS
->OpenProtocol (
656 &gEfiDhcp4ProtocolGuid
,
658 gDhcp4DriverBinding
.DriverBindingHandle
,
660 EFI_OPEN_PROTOCOL_GET_PROTOCOL
663 if (EFI_ERROR (Status
)) {
664 return EFI_UNSUPPORTED
;
667 Instance
= DHCP_INSTANCE_FROM_THIS (Dhcp
);
668 DhcpSb
= DHCP_SERVICE_FROM_THIS (This
);
670 if (Instance
->Service
!= DhcpSb
) {
671 return EFI_INVALID_PARAMETER
;
675 // A child can be destroyed more than once. For example,
676 // Dhcp4DriverBindingStop will destroy all of its children.
677 // when caller driver is being stopped, it will destroy the
678 // dhcp child it opens.
680 if (Instance
->InDestroy
) {
684 OldTpl
= gBS
->RaiseTPL (TPL_CALLBACK
);
685 Instance
->InDestroy
= TRUE
;
688 // Close the Udp4 protocol.
691 DhcpSb
->UdpIo
->UdpHandle
,
692 &gEfiUdp4ProtocolGuid
,
693 gDhcp4DriverBinding
.DriverBindingHandle
,
698 // Uninstall the DHCP4 protocol first to enable a top down destruction.
700 gBS
->RestoreTPL (OldTpl
);
701 Status
= gBS
->UninstallProtocolInterface (
703 &gEfiDhcp4ProtocolGuid
,
706 OldTpl
= gBS
->RaiseTPL (TPL_CALLBACK
);
707 if (EFI_ERROR (Status
)) {
708 Instance
->InDestroy
= FALSE
;
710 gBS
->RestoreTPL (OldTpl
);
714 if (DhcpSb
->ActiveChild
== Instance
) {
715 DhcpYieldControl (DhcpSb
);
718 RemoveEntryList (&Instance
->Link
);
719 DhcpSb
->NumChildren
--;
721 if (Instance
->UdpIo
!= NULL
) {
722 UdpIoCleanIo (Instance
->UdpIo
);
724 Instance
->UdpIo
->UdpHandle
,
725 &gEfiUdp4ProtocolGuid
,
726 Instance
->Service
->Image
,
729 UdpIoFreeIo (Instance
->UdpIo
);
730 Instance
->UdpIo
= NULL
;
731 Instance
->Token
= NULL
;
734 gBS
->RestoreTPL (OldTpl
);