3 Copyright (c) 2005 - 2007, Intel Corporation
\r
4 All rights reserved. This program and the accompanying materials
\r
5 are licensed and made available under the terms and conditions of the BSD License
\r
6 which accompanies this distribution. The full text of the license may be found at
\r
7 http://opensource.org/licenses/bsd-license.php
\r
9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
\r
10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
\r
21 #include "Tcp4Main.h"
\r
24 UINT16 mTcp4RandomPort;
\r
25 extern EFI_COMPONENT_NAME_PROTOCOL gTcp4ComponentName;
\r
26 extern EFI_COMPONENT_NAME2_PROTOCOL gTcp4ComponentName2;
\r
28 TCP4_HEARTBEAT_TIMER mTcp4Timer = {
\r
33 EFI_TCP4_PROTOCOL mTcp4ProtocolTemplate = {
\r
46 SOCK_INIT_DATA mTcp4DefaultSockData = {
\r
53 &mTcp4ProtocolTemplate,
\r
58 EFI_DRIVER_BINDING_PROTOCOL mTcp4DriverBinding = {
\r
59 Tcp4DriverBindingSupported,
\r
60 Tcp4DriverBindingStart,
\r
61 Tcp4DriverBindingStop,
\r
67 EFI_SERVICE_BINDING_PROTOCOL mTcp4ServiceBinding = {
\r
68 Tcp4ServiceBindingCreateChild,
\r
69 Tcp4ServiceBindingDestroyChild
\r
74 Create and start the heartbeat timer for TCP driver.
\r
78 @retval EFI_SUCCESS The timer is successfully created and started.
\r
79 @retval other The timer is not created.
\r
90 Status = EFI_SUCCESS;
\r
92 if (mTcp4Timer.RefCnt == 0) {
\r
94 Status = gBS->CreateEvent (
\r
95 EVT_TIMER | EVT_NOTIFY_SIGNAL,
\r
99 &mTcp4Timer.TimerEvent
\r
101 if (!EFI_ERROR (Status)) {
\r
103 Status = gBS->SetTimer (
\r
104 mTcp4Timer.TimerEvent,
\r
106 (UINT64) (TICKS_PER_SECOND / TCP_TICK_HZ)
\r
111 if (!EFI_ERROR (Status)) {
\r
113 mTcp4Timer.RefCnt++;
\r
121 Stop and destroy the heartbeat timer for TCP driver.
\r
134 ASSERT (mTcp4Timer.RefCnt > 0);
\r
136 mTcp4Timer.RefCnt--;
\r
138 if (mTcp4Timer.RefCnt > 0) {
\r
142 gBS->SetTimer (mTcp4Timer.TimerEvent, TimerCancel, 0);
\r
143 gBS->CloseEvent (mTcp4Timer.TimerEvent);
\r
144 mTcp4Timer.TimerEvent = NULL;
\r
150 Tcp4DriverEntryPoint (
\r
151 IN EFI_HANDLE ImageHandle,
\r
152 IN EFI_SYSTEM_TABLE *SystemTable
\r
156 Routine Description:
\r
158 The entry point for Tcp4 driver. used to install
\r
159 Tcp4 driver on the ImageHandle.
\r
163 ImageHandle - The firmware allocated handle for this
\r
165 SystemTable - Pointer to the EFI system table.
\r
169 EFI_SUCCESS - Driver loaded.
\r
170 other - Driver not loaded.
\r
178 // Install the TCP4 Driver Binding Protocol
\r
180 Status = EfiLibInstallDriverBindingComponentName2 (
\r
183 &mTcp4DriverBinding,
\r
185 &gTcp4ComponentName,
\r
186 &gTcp4ComponentName2
\r
188 ASSERT_EFI_ERROR (Status);
\r
190 // Initialize ISS and random port.
\r
192 Seed = NetRandomInitSeed ();
\r
193 mTcpGlobalIss = NET_RANDOM (Seed) % mTcpGlobalIss;
\r
194 mTcp4RandomPort = (UINT16) ( TCP4_PORT_KNOWN +
\r
195 (UINT16) (NET_RANDOM(Seed) % TCP4_PORT_KNOWN));
\r
202 Test to see if this driver supports ControllerHandle.
\r
204 @param This Protocol instance pointer.
\r
205 @param ControllerHandle Handle of device to test.
\r
206 @param RemainingDevicePath Optional parameter use to pick a specific child
\r
209 @retval EFI_SUCCESS This driver supports this device.
\r
210 @retval EFI_ALREADY_STARTED This driver is already running on this device.
\r
211 @retval other This driver does not support this device.
\r
216 Tcp4DriverBindingSupported (
\r
217 IN EFI_DRIVER_BINDING_PROTOCOL * This,
\r
218 IN EFI_HANDLE ControllerHandle,
\r
219 IN EFI_DEVICE_PATH_PROTOCOL * RemainingDevicePath OPTIONAL
\r
225 // Test for the Tcp4ServiceBinding Protocol
\r
227 Status = gBS->OpenProtocol (
\r
229 &gEfiTcp4ServiceBindingProtocolGuid,
\r
231 This->DriverBindingHandle,
\r
233 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
\r
235 if (!EFI_ERROR (Status)) {
\r
236 return EFI_ALREADY_STARTED;
\r
240 // Test for the Ip4 Protocol
\r
242 Status = gBS->OpenProtocol (
\r
244 &gEfiIp4ServiceBindingProtocolGuid,
\r
246 This->DriverBindingHandle,
\r
248 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
\r
256 Start this driver on ControllerHandle.
\r
258 @param This Protocol instance pointer.
\r
259 @param ControllerHandle Handle of device to bind driver to.
\r
260 @param RemainingDevicePath Optional parameter use to pick a specific child
\r
263 @retval EFI_SUCCESS The driver is added to ControllerHandle.
\r
264 @retval EFI_OUT_OF_RESOURCES There are not enough resources to start the
\r
266 @retval other The driver cannot be added to ControllerHandle.
\r
271 Tcp4DriverBindingStart (
\r
272 IN EFI_DRIVER_BINDING_PROTOCOL * This,
\r
273 IN EFI_HANDLE ControllerHandle,
\r
274 IN EFI_DEVICE_PATH_PROTOCOL * RemainingDevicePath OPTIONAL
\r
278 TCP4_SERVICE_DATA *TcpServiceData;
\r
279 IP_IO_OPEN_DATA OpenData;
\r
281 TcpServiceData = NetAllocateZeroPool (sizeof (TCP4_SERVICE_DATA));
\r
283 if (NULL == TcpServiceData) {
\r
284 TCP4_DEBUG_ERROR (("Tcp4DriverBindingStart: Have no enough"
\r
285 " resource to create a Tcp Servcie Data!\n"));
\r
287 return EFI_OUT_OF_RESOURCES;
\r
291 // Create a new IP IO to Consume it
\r
293 TcpServiceData->IpIo = IpIoCreate (This->DriverBindingHandle, ControllerHandle);
\r
294 if (NULL == TcpServiceData->IpIo) {
\r
296 TCP4_DEBUG_ERROR (("Tcp4DriverBindingStart: Have no enough"
\r
297 " resource to create an Ip Io!\n"));
\r
299 Status = EFI_OUT_OF_RESOURCES;
\r
304 // Configure and start IpIo.
\r
306 NetZeroMem (&OpenData, sizeof (IP_IO_OPEN_DATA));
\r
308 CopyMem (&OpenData.IpConfigData, &mIpIoDefaultIpConfigData, sizeof (OpenData.IpConfigData));
\r
309 OpenData.IpConfigData.DefaultProtocol = EFI_IP_PROTO_TCP;
\r
311 OpenData.PktRcvdNotify = Tcp4RxCallback;
\r
312 Status = IpIoOpen (TcpServiceData->IpIo, &OpenData);
\r
314 if (EFI_ERROR (Status)) {
\r
319 // Create the timer event used by TCP driver
\r
321 Status = Tcp4CreateTimer ();
\r
322 if (EFI_ERROR (Status)) {
\r
324 TCP4_DEBUG_ERROR (("Tcp4DriverBindingStart: Create TcpTimer"
\r
325 " Event failed with %r\n", Status));
\r
331 // Install the Tcp4ServiceBinding Protocol on the
\r
332 // controller handle
\r
334 TcpServiceData->Tcp4ServiceBinding = mTcp4ServiceBinding;
\r
336 Status = gBS->InstallMultipleProtocolInterfaces (
\r
338 &gEfiTcp4ServiceBindingProtocolGuid,
\r
339 &TcpServiceData->Tcp4ServiceBinding,
\r
342 if (EFI_ERROR (Status)) {
\r
344 TCP4_DEBUG_ERROR (("Tcp4DriverBindingStart: Install Tcp4 Service Binding"
\r
345 " Protocol failed for %r\n", Status));
\r
347 Tcp4DestroyTimer ();
\r
352 // Initialize member in TcpServiceData
\r
354 TcpServiceData->ControllerHandle = ControllerHandle;
\r
355 TcpServiceData->Signature = TCP4_DRIVER_SIGNATURE;
\r
356 TcpServiceData->DriverBindingHandle = This->DriverBindingHandle;
\r
358 NetListInit (&TcpServiceData->SocketList);
\r
360 TcpSetVariableData (TcpServiceData);
\r
362 return EFI_SUCCESS;
\r
366 if (TcpServiceData->IpIo != NULL) {
\r
367 IpIoDestroy (TcpServiceData->IpIo);
\r
370 NetFreePool (TcpServiceData);
\r
377 Stop this driver on ControllerHandle.
\r
379 @param This Protocol instance pointer.
\r
380 @param ControllerHandle Handle of device to stop driver on.
\r
381 @param NumberOfChildren Number of Handles in ChildHandleBuffer. If number
\r
382 of children is zero stop the entire bus driver.
\r
383 @param ChildHandleBuffer List of Child Handles to Stop.
\r
385 @retval EFI_SUCCESS This driver is removed from ControllerHandle.
\r
386 @retval other This driver is not removed from ControllerHandle.
\r
391 Tcp4DriverBindingStop (
\r
392 IN EFI_DRIVER_BINDING_PROTOCOL *This,
\r
393 IN EFI_HANDLE ControllerHandle,
\r
394 IN UINTN NumberOfChildren,
\r
395 IN EFI_HANDLE *ChildHandleBuffer
\r
399 EFI_HANDLE NicHandle;
\r
400 EFI_SERVICE_BINDING_PROTOCOL *ServiceBinding;
\r
401 TCP4_SERVICE_DATA *TcpServiceData;
\r
404 // Find the NicHandle where Tcp4 ServiceBinding Protocol is installed.
\r
406 NicHandle = NetLibGetNicHandle (ControllerHandle, &gEfiIp4ProtocolGuid);
\r
407 if (NicHandle == NULL) {
\r
408 return EFI_DEVICE_ERROR;
\r
412 // Retrieve the TCP driver Data Structure
\r
414 Status = gBS->OpenProtocol (
\r
416 &gEfiTcp4ServiceBindingProtocolGuid,
\r
417 (VOID **) &ServiceBinding,
\r
418 This->DriverBindingHandle,
\r
420 EFI_OPEN_PROTOCOL_GET_PROTOCOL
\r
422 if (EFI_ERROR (Status)) {
\r
424 TCP4_DEBUG_ERROR (("Tcp4DriverBindingStop: Locate Tcp4 Service "
\r
425 " Binding Protocol failed with %r\n", Status));
\r
427 return EFI_DEVICE_ERROR;
\r
430 TcpServiceData = TCP4_FROM_THIS (ServiceBinding);
\r
432 if (NumberOfChildren == 0) {
\r
434 // Uninstall TCP servicebinding protocol
\r
436 gBS->UninstallMultipleProtocolInterfaces (
\r
438 &gEfiTcp4ServiceBindingProtocolGuid,
\r
444 // Destroy the IpIO consumed by TCP driver
\r
446 IpIoDestroy (TcpServiceData->IpIo);
\r
449 // Destroy the heartbeat timer.
\r
451 Tcp4DestroyTimer ();
\r
454 // Clear the variable.
\r
456 TcpClearVariableData (TcpServiceData);
\r
459 // Release the TCP service data
\r
461 NetFreePool (TcpServiceData);
\r
464 while (!NetListIsEmpty (&TcpServiceData->SocketList)) {
\r
465 Sock = NET_LIST_HEAD (&TcpServiceData->SocketList, SOCKET, Link);
\r
467 ServiceBinding->DestroyChild (ServiceBinding, Sock->SockHandle);
\r
475 Creates a child handle with a set of TCP4 services.
\r
477 @param This Protocol instance pointer.
\r
478 @param ChildHandle Pointer to the handle of the child to create. If
\r
479 it is NULL, then a new handle is created. If it is
\r
480 not NULL, then the I/O services are added to the
\r
481 existing child handle.
\r
483 @retval EFI_SUCCESS The child handle is created.
\r
484 @retval EFI_INVALID_PARAMETER One or more parameters are invalid.
\r
485 @retval EFI_OUT_OF_RESOURCES There are not enough resources to create the
\r
491 Tcp4ServiceBindingCreateChild (
\r
492 IN EFI_SERVICE_BINDING_PROTOCOL *This,
\r
493 IN EFI_HANDLE *ChildHandle
\r
497 TCP4_SERVICE_DATA *TcpServiceData;
\r
498 TCP4_PROTO_DATA TcpProto;
\r
503 if (NULL == This || NULL == ChildHandle) {
\r
504 return EFI_INVALID_PARAMETER;
\r
507 OldTpl = NET_RAISE_TPL (NET_TPL_LOCK);
\r
508 TcpServiceData = TCP4_FROM_THIS (This);
\r
509 TcpProto.TcpService = TcpServiceData;
\r
510 TcpProto.TcpPcb = NULL;
\r
513 // Create a tcp instance with defualt Tcp default
\r
514 // sock init data and TcpProto
\r
516 mTcp4DefaultSockData.DriverBinding = TcpServiceData->DriverBindingHandle;
\r
518 Sock = SockCreateChild (&mTcp4DefaultSockData, &TcpProto, sizeof (TCP4_PROTO_DATA));
\r
519 if (NULL == Sock) {
\r
520 TCP4_DEBUG_ERROR (("Tcp4DriverBindingCreateChild: "
\r
521 "No resource to create a Tcp Child\n"));
\r
523 Status = EFI_OUT_OF_RESOURCES;
\r
527 *ChildHandle = Sock->SockHandle;
\r
530 // Open the default Ip4 protocol of IP_IO BY_DRIVER.
\r
532 Status = gBS->OpenProtocol (
\r
533 TcpServiceData->IpIo->ChildHandle,
\r
534 &gEfiIp4ProtocolGuid,
\r
536 TcpServiceData->DriverBindingHandle,
\r
538 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
\r
540 if (EFI_ERROR (Status)) {
\r
541 SockDestroyChild (Sock);
\r
546 // Open the device path on the handle where service binding resides on.
\r
548 Status = gBS->OpenProtocol (
\r
549 TcpServiceData->ControllerHandle,
\r
550 &gEfiDevicePathProtocolGuid,
\r
551 (VOID **) &Sock->ParentDevicePath,
\r
552 TcpServiceData->DriverBindingHandle,
\r
554 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
\r
556 if (EFI_ERROR (Status)) {
\r
557 gBS->CloseProtocol (
\r
558 TcpServiceData->IpIo->ChildHandle,
\r
559 &gEfiIp4ProtocolGuid,
\r
560 TcpServiceData->DriverBindingHandle,
\r
563 SockDestroyChild (Sock);
\r
565 NetListInsertTail (&TcpServiceData->SocketList, &Sock->Link);
\r
570 NET_RESTORE_TPL (OldTpl);
\r
576 Destroys a child handle with a set of UDP4 services.
\r
578 @param This Protocol instance pointer.
\r
579 @param ChildHandle Handle of the child to be destroyed.
\r
581 @retval EFI_SUCCESS The TCP4 services are removed from the child
\r
583 @retval EFI_INVALID_PARAMETER One or more parameters are invalid.
\r
584 @retval other The child handle is not destroyed.
\r
589 Tcp4ServiceBindingDestroyChild (
\r
590 IN EFI_SERVICE_BINDING_PROTOCOL *This,
\r
591 IN EFI_HANDLE ChildHandle
\r
595 EFI_TCP4_PROTOCOL *Tcp4;
\r
597 TCP4_PROTO_DATA *TcpProtoData;
\r
598 TCP4_SERVICE_DATA *TcpServiceData;
\r
601 if (NULL == This || NULL == ChildHandle) {
\r
602 return EFI_INVALID_PARAMETER;
\r
605 OldTpl = NET_RAISE_TPL (NET_TPL_LOCK);
\r
608 // retrieve the Tcp4 protocol from ChildHandle
\r
610 Status = gBS->OpenProtocol (
\r
612 &gEfiTcp4ProtocolGuid,
\r
614 mTcp4DriverBinding.DriverBindingHandle,
\r
616 EFI_OPEN_PROTOCOL_GET_PROTOCOL
\r
618 if (EFI_ERROR (Status)) {
\r
619 Status = EFI_UNSUPPORTED;
\r
624 // destroy this sock and related Tcp protocol control
\r
627 Sock = SOCK_FROM_THIS (Tcp4);
\r
628 TcpProtoData = (TCP4_PROTO_DATA *) Sock->ProtoReserved;
\r
629 TcpServiceData = TcpProtoData->TcpService;
\r
631 NetListRemoveEntry (&Sock->Link);
\r
633 SockDestroyChild (Sock);
\r
636 // Close the device path protocol
\r
638 gBS->CloseProtocol (
\r
639 TcpServiceData->ControllerHandle,
\r
640 &gEfiDevicePathProtocolGuid,
\r
641 TcpServiceData->DriverBindingHandle,
\r
646 // Close the Ip4 protocol.
\r
648 gBS->CloseProtocol (
\r
649 TcpServiceData->IpIo->ChildHandle,
\r
650 &gEfiIp4ProtocolGuid,
\r
651 TcpServiceData->DriverBindingHandle,
\r
656 NET_RESTORE_TPL (OldTpl);
\r