]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Library/DxeTcpIoLib/DxeTcpIoLib.c
MdeModulePkg: Add UefiBootManagerLib
[mirror_edk2.git] / MdeModulePkg / Library / DxeTcpIoLib / DxeTcpIoLib.c
index 730ccb2cafe8b36fda3d7a89898588dee12a90be..17183e1a6c818df61f2e1715b02f4d0259de140b 100644 (file)
@@ -22,92 +22,92 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Library/MemoryAllocationLib.h>\r
 #include <Library/BaseMemoryLib.h>\r
 \r
-/**
+/**\r
   The common notify function associated with various TcpIo events. \r
-
-  @param[in]  Event   The event signaled.
-  @param[in]  Context The context.
-
-**/
-VOID
-EFIAPI
+\r
+  @param[in]  Event   The event signaled.\r
+  @param[in]  Context The context.\r
+\r
+**/\r
+VOID\r
+EFIAPI\r
 TcpIoCommonNotify (\r
-  IN EFI_EVENT  Event,
-  IN VOID       *Context
-  )
+  IN EFI_EVENT  Event,\r
+  IN VOID       *Context\r
+  )\r
 {\r
   if ((Event == NULL) || (Context == NULL)) {\r
     return ;\r
   }\r
 \r
-  *((BOOLEAN *) Context) = TRUE;
+  *((BOOLEAN *) Context) = TRUE;\r
 }\r
 \r
-/**
+/**\r
   The internal function for delay configuring TCP6 when IP6 driver is still in DAD.\r
-
-  @param[in]  Tcp6               The EFI_TCP6_PROTOCOL protocol instance.
-  @param[in]  Tcp6ConfigData     The Tcp6 configuration data.
-
-  @retval EFI_SUCCESS            The operational settings successfully
-                                 completed.
+\r
+  @param[in]  Tcp6               The EFI_TCP6_PROTOCOL protocol instance.\r
+  @param[in]  Tcp6ConfigData     The Tcp6 configuration data.\r
+\r
+  @retval EFI_SUCCESS            The operational settings successfully\r
+                                 completed.\r
   @retval EFI_INVALID_PARAMETER  One or more parameters are invalid.\r
   @retval Others                 Failed to finish the operation.\r
 \r
-**/
-EFI_STATUS
+**/\r
+EFI_STATUS\r
 TcpIoGetMapping (\r
-  IN EFI_TCP6_PROTOCOL    *Tcp6,
-  IN EFI_TCP6_CONFIG_DATA *Tcp6ConfigData
-  )
-{
-  EFI_STATUS              Status;
-  EFI_EVENT               Event;
+  IN EFI_TCP6_PROTOCOL    *Tcp6,\r
+  IN EFI_TCP6_CONFIG_DATA *Tcp6ConfigData\r
+  )\r
+{\r
+  EFI_STATUS              Status;\r
+  EFI_EVENT               Event;\r
 \r
   if ((Tcp6 == NULL) || (Tcp6ConfigData == NULL)) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  Event  = NULL;
-  Status = gBS->CreateEvent (
-                  EVT_TIMER,
-                  TPL_CALLBACK,
-                  NULL,
-                  NULL,
-                  &Event
-                  );
-  if (EFI_ERROR (Status)) {
-    goto ON_EXIT;
-  }
-
-  Status = gBS->SetTimer (
-                  Event,
-                  TimerRelative,
+  Event  = NULL;\r
+  Status = gBS->CreateEvent (\r
+                  EVT_TIMER,\r
+                  TPL_CALLBACK,\r
+                  NULL,\r
+                  NULL,\r
+                  &Event\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_EXIT;\r
+  }\r
+\r
+  Status = gBS->SetTimer (\r
+                  Event,\r
+                  TimerRelative,\r
                   TCP_GET_MAPPING_TIMEOUT\r
-                  );
-
-  if (EFI_ERROR (Status)) {
-    goto ON_EXIT;
-  }
-
-  while (EFI_ERROR (gBS->CheckEvent (Event))) {
-
-    Tcp6->Poll (Tcp6);
-
-    Status = Tcp6->Configure (Tcp6, Tcp6ConfigData);
-
-    if (!EFI_ERROR (Status)) {
-      break;
-    }
-  }
-
-ON_EXIT:
-
-  if (Event != NULL) {
-    gBS->CloseEvent (Event);
-  }
-
-  return Status;
+                  );\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_EXIT;\r
+  }\r
+\r
+  while (EFI_ERROR (gBS->CheckEvent (Event))) {\r
+\r
+    Tcp6->Poll (Tcp6);\r
+\r
+    Status = Tcp6->Configure (Tcp6, Tcp6ConfigData);\r
+\r
+    if (!EFI_ERROR (Status)) {\r
+      break;\r
+    }\r
+  }\r
+\r
+ON_EXIT:\r
+\r
+  if (Event != NULL) {\r
+    gBS->CloseEvent (Event);\r
+  }\r
+\r
+  return Status;\r
 }\r
 \r
 /**\r
@@ -121,7 +121,7 @@ ON_EXIT:
   \r
   @retval EFI_SUCCESS            The TCP socket is created and configured.\r
   @retval EFI_INVALID_PARAMETER  One or more parameters are invalid.\r
-  @retval EFI_UNSUPPORTED        One or more of the control options are not
+  @retval EFI_UNSUPPORTED        One or more of the control options are not\r
                                  supported in the implementation.\r
   @retval EFI_OUT_OF_RESOURCES   Failed to allocate memory.\r
   @retval Others                 Failed to create the TCP socket or configure it.\r
@@ -177,26 +177,26 @@ TcpIoCreateSocket (
   //\r
   // Create the TCP child instance and get the TCP protocol.\r
   //  \r
-  Status = NetLibCreateServiceChild (
-             Controller,
-             Image,
+  Status = NetLibCreateServiceChild (\r
+             Controller,\r
+             Image,\r
              ServiceBindingGuid,\r
              &TcpIo->Handle\r
-             );
-  if (EFI_ERROR (Status)) {
-    return Status;
-  }
-
-  Status = gBS->OpenProtocol (
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  Status = gBS->OpenProtocol (\r
                   TcpIo->Handle,\r
                   ProtocolGuid,\r
                   Interface,\r
-                  Image,
-                  Controller,
-                  EFI_OPEN_PROTOCOL_BY_DRIVER
-                  );
+                  Image,\r
+                  Controller,\r
+                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
+                  );\r
   if (EFI_ERROR (Status) || (*Interface == NULL)) {\r
-    goto ON_ERROR;
+    goto ON_ERROR;\r
   }\r
 \r
   if (TcpVersion == TCP_VERSION_4) {\r
@@ -204,28 +204,28 @@ TcpIoCreateSocket (
   } else {\r
     Tcp6             = TcpIo->Tcp.Tcp6;\r
   }\r
-
+\r
   TcpIo->Image       = Image;\r
   TcpIo->Controller  = Controller;\r
-
-  //
-  // Set the configuration parameters.
-  //
-  ControlOption.ReceiveBufferSize       = 0x200000;
-  ControlOption.SendBufferSize          = 0x200000;
-  ControlOption.MaxSynBackLog           = 0;
-  ControlOption.ConnectionTimeout       = 0;
-  ControlOption.DataRetries             = 6;
-  ControlOption.FinTimeout              = 0;
-  ControlOption.TimeWaitTimeout         = 0;
-  ControlOption.KeepAliveProbes         = 4;
-  ControlOption.KeepAliveTime           = 0;
-  ControlOption.KeepAliveInterval       = 0;
-  ControlOption.EnableNagle             = FALSE;
-  ControlOption.EnableTimeStamp         = FALSE;
-  ControlOption.EnableWindowScaling     = TRUE;
-  ControlOption.EnableSelectiveAck      = FALSE;
-  ControlOption.EnablePathMtuDiscovery  = FALSE;
+\r
+  //\r
+  // Set the configuration parameters.\r
+  //\r
+  ControlOption.ReceiveBufferSize       = 0x200000;\r
+  ControlOption.SendBufferSize          = 0x200000;\r
+  ControlOption.MaxSynBackLog           = 0;\r
+  ControlOption.ConnectionTimeout       = 0;\r
+  ControlOption.DataRetries             = 6;\r
+  ControlOption.FinTimeout              = 0;\r
+  ControlOption.TimeWaitTimeout         = 0;\r
+  ControlOption.KeepAliveProbes         = 4;\r
+  ControlOption.KeepAliveTime           = 0;\r
+  ControlOption.KeepAliveInterval       = 0;\r
+  ControlOption.EnableNagle             = FALSE;\r
+  ControlOption.EnableTimeStamp         = FALSE;\r
+  ControlOption.EnableWindowScaling     = TRUE;\r
+  ControlOption.EnableSelectiveAck      = FALSE;\r
+  ControlOption.EnablePathMtuDiscovery  = FALSE;\r
 \r
   if (TcpVersion == TCP_VERSION_4) {\r
     Tcp4ConfigData.TypeOfService        = 8;\r
@@ -309,59 +309,59 @@ TcpIoCreateSocket (
     }\r
   }\r
 \r
-  //
-  // Create events for variuos asynchronous operations.
+  //\r
+  // Create events for variuos asynchronous operations.\r
   //\r
   Status = gBS->CreateEvent (\r
-                  EVT_NOTIFY_SIGNAL,
-                  TPL_NOTIFY,
+                  EVT_NOTIFY_SIGNAL,\r
+                  TPL_NOTIFY,\r
                   TcpIoCommonNotify,\r
                   &TcpIo->IsConnDone,\r
                   &Event\r
-                  );
-  if (EFI_ERROR (Status)) {
-    goto ON_ERROR;
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_ERROR;\r
   }\r
 \r
   TcpIo->ConnToken.Tcp4Token.CompletionToken.Event = Event;\r
 \r
-  Status = gBS->CreateEvent (
-                  EVT_NOTIFY_SIGNAL,
-                  TPL_NOTIFY,
+  Status = gBS->CreateEvent (\r
+                  EVT_NOTIFY_SIGNAL,\r
+                  TPL_NOTIFY,\r
                   TcpIoCommonNotify,\r
                   &TcpIo->IsListenDone,\r
                   &Event\r
-                  );
-  if (EFI_ERROR (Status)) {
-    goto ON_ERROR;
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_ERROR;\r
   }\r
 \r
   TcpIo->ListenToken.Tcp4Token.CompletionToken.Event = Event;\r
 \r
-  Status = gBS->CreateEvent (
-                  EVT_NOTIFY_SIGNAL,
-                  TPL_NOTIFY,
+  Status = gBS->CreateEvent (\r
+                  EVT_NOTIFY_SIGNAL,\r
+                  TPL_NOTIFY,\r
                   TcpIoCommonNotify,\r
                   &TcpIo->IsTxDone,\r
                   &Event\r
-                  );
-  if (EFI_ERROR (Status)) {
-    goto ON_ERROR;
-  }
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_ERROR;\r
+  }\r
 \r
   TcpIo->TxToken.Tcp4Token.CompletionToken.Event = Event;\r
 \r
 \r
-  Status = gBS->CreateEvent (
-                  EVT_NOTIFY_SIGNAL,
-                  TPL_NOTIFY,
+  Status = gBS->CreateEvent (\r
+                  EVT_NOTIFY_SIGNAL,\r
+                  TPL_NOTIFY,\r
                   TcpIoCommonNotify,\r
                   &TcpIo->IsRxDone,\r
                   &Event\r
-                  );
-  if (EFI_ERROR (Status)) {
-    goto ON_ERROR;
-  }
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_ERROR;\r
+  }\r
 \r
   TcpIo->RxToken.Tcp4Token.CompletionToken.Event = Event;\r
 \r
@@ -373,26 +373,26 @@ TcpIoCreateSocket (
 \r
   TcpIo->RxToken.Tcp4Token.Packet.RxData = RxData;\r
 \r
-  Status = gBS->CreateEvent (
-                  EVT_NOTIFY_SIGNAL,
-                  TPL_NOTIFY,
+  Status = gBS->CreateEvent (\r
+                  EVT_NOTIFY_SIGNAL,\r
+                  TPL_NOTIFY,\r
                   TcpIoCommonNotify,\r
                   &TcpIo->IsCloseDone,\r
                   &Event\r
-                  );
-  if (EFI_ERROR (Status)) {
-    goto ON_ERROR;
-  }
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_ERROR;\r
+  }\r
 \r
   TcpIo->CloseToken.Tcp4Token.CompletionToken.Event = Event;\r
 \r
 \r
   return EFI_SUCCESS;\r
-
-ON_ERROR:
+\r
+ON_ERROR:\r
 \r
   TcpIoDestroySocket (TcpIo);\r
-
+\r
   return Status;\r
 }\r
   \r
@@ -482,7 +482,7 @@ TcpIoDestroySocket (
 \r
   if ((Tcp4 != NULL) || (Tcp6 != NULL)) {\r
 \r
-    gBS->CloseProtocol (
+    gBS->CloseProtocol (\r
            TcpIo->Handle,\r
            ProtocolGuid,\r
            TcpIo->Image,\r
@@ -518,7 +518,7 @@ TcpIoDestroySocket (
     }\r
   }\r
 \r
-  NetLibDestroyServiceChild (
+  NetLibDestroyServiceChild (\r
     TcpIo->Controller,\r
     TcpIo->Image,\r
     ServiceBindingGuid,\r
@@ -572,24 +572,24 @@ TcpIoConnect (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  if (EFI_ERROR (Status)) {
-    return Status;
-  }
-
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
   while (!TcpIo->IsConnDone && EFI_ERROR (gBS->CheckEvent (Timeout))) {\r
     if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
       Tcp4->Poll (Tcp4);\r
     } else {\r
       Tcp6->Poll (Tcp6);\r
     }\r
-  }
-
+  }\r
+\r
   if (!TcpIo->IsConnDone) {\r
-    Status = EFI_TIMEOUT;
+    Status = EFI_TIMEOUT;\r
   } else {\r
     Status = TcpIo->ConnToken.Tcp4Token.CompletionToken.Status;\r
   }\r
-
+\r
   return Status;\r
 }\r
 \r
@@ -642,8 +642,8 @@ TcpIoAccept (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  if (EFI_ERROR (Status)) {
-    return Status;
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
   }\r
 \r
   while (!TcpIo->IsListenDone && EFI_ERROR (gBS->CheckEvent (Timeout))) {\r
@@ -652,18 +652,18 @@ TcpIoAccept (
     } else {\r
       Tcp6->Poll (Tcp6);\r
     }\r
-  }
+  }\r
 \r
   if (!TcpIo->IsListenDone) {\r
-    Status = EFI_TIMEOUT;
+    Status = EFI_TIMEOUT;\r
   } else {\r
     Status = TcpIo->ListenToken.Tcp4Token.CompletionToken.Status;\r
   }\r
 \r
-  //
-  // The new TCP instance handle created for the established connection is 
-  // in ListenToken.
-  //
+  //\r
+  // The new TCP instance handle created for the established connection is \r
+  // in ListenToken.\r
+  //\r
   if (!EFI_ERROR (Status)) {\r
     if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
       ProtocolGuid = &gEfiTcp4ProtocolGuid;\r
@@ -671,16 +671,16 @@ TcpIoAccept (
       ProtocolGuid = &gEfiTcp6ProtocolGuid;\r
     }\r
     \r
-    Status = gBS->OpenProtocol (
+    Status = gBS->OpenProtocol (\r
                     TcpIo->ListenToken.Tcp4Token.NewChildHandle,\r
                     ProtocolGuid,\r
                     (VOID **) (&TcpIo->NewTcp.Tcp4),\r
                     TcpIo->Image,\r
                     TcpIo->Controller,\r
-                    EFI_OPEN_PROTOCOL_BY_DRIVER
-                    );
-
-  }
+                    EFI_OPEN_PROTOCOL_BY_DRIVER\r
+                    );\r
+\r
+  }\r
 \r
   return Status;\r
 }\r
@@ -721,10 +721,10 @@ TcpIoReset (
     return ;\r
   }\r
 \r
-  if (EFI_ERROR (Status)) {
-    return ;
-  }
-
+  if (EFI_ERROR (Status)) {\r
+    return ;\r
+  }\r
+\r
   while (!TcpIo->IsCloseDone) {\r
     if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
       Tcp4->Poll (Tcp4);\r
@@ -780,16 +780,16 @@ TcpIoTransmit (
 \r
   Data = AllocatePool (Size);\r
   if (Data == NULL) {\r
-    return EFI_OUT_OF_RESOURCES;
+    return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
   ((EFI_TCP4_TRANSMIT_DATA *) Data)->Push        = TRUE;\r
   ((EFI_TCP4_TRANSMIT_DATA *) Data)->Urgent      = FALSE;\r
   ((EFI_TCP4_TRANSMIT_DATA *) Data)->DataLength  = Packet->TotalSize;\r
 \r
-  //
-  // Build the fragment table.
-  //
+  //\r
+  // Build the fragment table.\r
+  //\r
   ((EFI_TCP4_TRANSMIT_DATA *) Data)->FragmentCount = Packet->BlockOpNum;\r
 \r
   NetbufBuildExt (\r
@@ -802,8 +802,8 @@ TcpIoTransmit (
   Tcp6   = NULL;\r
   Status = EFI_DEVICE_ERROR;\r
 \r
-  //
-  // Trasnmit the packet.
+  //\r
+  // Trasnmit the packet.\r
   //\r
   if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
     TcpIo->TxToken.Tcp4Token.Packet.TxData = (EFI_TCP4_TRANSMIT_DATA *) Data;\r
@@ -831,25 +831,25 @@ TcpIoTransmit (
     Status  = Tcp6->Transmit (Tcp6, &TcpIo->TxToken.Tcp6Token);\r
   }\r
 \r
-  if (EFI_ERROR (Status)) {
-    goto ON_EXIT;
-  }
-
+  if (EFI_ERROR (Status)) {\r
+    goto ON_EXIT;\r
+  }\r
+\r
   while (!TcpIo->IsTxDone) {\r
     if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
       Tcp4->Poll (Tcp4);\r
     } else {\r
       Tcp6->Poll (Tcp6);\r
     }\r
-  }
-
+  }\r
+\r
   TcpIo->IsTxDone  = FALSE;\r
   Status           = TcpIo->TxToken.Tcp4Token.CompletionToken.Status;\r
 \r
-ON_EXIT:
-
+ON_EXIT:\r
+\r
   FreePool (Data);\r
-
+\r
   return Status;\r
 }\r
 \r
@@ -926,22 +926,22 @@ TcpIoReceive (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  FragmentCount = Packet->BlockOpNum;
-  Fragment      = AllocatePool (FragmentCount * sizeof (NET_FRAGMENT));
+  FragmentCount = Packet->BlockOpNum;\r
+  Fragment      = AllocatePool (FragmentCount * sizeof (NET_FRAGMENT));\r
   if (Fragment == NULL) {\r
     Status = EFI_OUT_OF_RESOURCES;\r
     goto ON_EXIT;\r
-  }
-  //
-  // Build the fragment table.
-  //
-  NetbufBuildExt (Packet, Fragment, &FragmentCount);
+  }\r
+  //\r
+  // Build the fragment table.\r
+  //\r
+  NetbufBuildExt (Packet, Fragment, &FragmentCount);\r
 \r
   RxData->FragmentCount         = 1;\r
   CurrentFragment               = 0;\r
-  Status                        = EFI_SUCCESS;
-
-  while (CurrentFragment < FragmentCount) {
+  Status                        = EFI_SUCCESS;\r
+\r
+  while (CurrentFragment < FragmentCount) {\r
     RxData->DataLength                       = Fragment[CurrentFragment].Len;\r
     RxData->FragmentTable[0].FragmentLength  = Fragment[CurrentFragment].Len;\r
     RxData->FragmentTable[0].FragmentBuffer  = Fragment[CurrentFragment].Bulk;\r
@@ -952,9 +952,9 @@ TcpIoReceive (
       Status = Tcp6->Receive (Tcp6, &TcpIo->RxToken.Tcp6Token);\r
     }\r
     \r
-    if (EFI_ERROR (Status)) {
-      goto ON_EXIT;
-    }
+    if (EFI_ERROR (Status)) {\r
+      goto ON_EXIT;\r
+    }\r
     \r
     while (!TcpIo->IsRxDone && ((Timeout == NULL) || EFI_ERROR (gBS->CheckEvent (Timeout)))) {\r
       //\r
@@ -966,42 +966,42 @@ TcpIoReceive (
         Tcp6->Poll (Tcp6);\r
       }\r
     }\r
-
+\r
     if (!TcpIo->IsRxDone) {\r
-      //
-      // Timeout occurs, cancel the receive request.
-      //
+      //\r
+      // Timeout occurs, cancel the receive request.\r
+      //\r
       if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
         Tcp4->Cancel (Tcp4, &TcpIo->RxToken.Tcp4Token.CompletionToken);\r
       } else {\r
         Tcp6->Cancel (Tcp6, &TcpIo->RxToken.Tcp6Token.CompletionToken);\r
       }\r
-
-      Status = EFI_TIMEOUT;
-      goto ON_EXIT;
-    } else {
+\r
+      Status = EFI_TIMEOUT;\r
+      goto ON_EXIT;\r
+    } else {\r
       TcpIo->IsRxDone = FALSE;\r
     }\r
 \r
     Status = TcpIo->RxToken.Tcp4Token.CompletionToken.Status;\r
 \r
     if (EFI_ERROR (Status)) {\r
-      goto ON_EXIT;
-    }
-
+      goto ON_EXIT;\r
+    }\r
+\r
     Fragment[CurrentFragment].Len -= RxData->FragmentTable[0].FragmentLength;\r
-    if (Fragment[CurrentFragment].Len == 0) {
-      CurrentFragment++;
-    } else {
+    if (Fragment[CurrentFragment].Len == 0) {\r
+      CurrentFragment++;\r
+    } else {\r
       Fragment[CurrentFragment].Bulk += RxData->FragmentTable[0].FragmentLength;\r
-    }
-  }
-
-ON_EXIT:
+    }\r
+  }\r
+\r
+ON_EXIT:\r
 \r
   if (Fragment != NULL) {\r
     FreePool (Fragment);\r
   }\r
-
+\r
   return Status;\r
 }\r