Fix GCC build breaks.

[mirror_edk2.git] / StdLib / EfiSocketLib / UseEfiSocketLib.c

/** @file\r
  Implement the connection to the EFI socket library\r
\r
  Copyright (c) 2011, Intel Corporation\r
  All rights reserved. This program and the accompanying materials\r
  are licensed and made available under the terms and conditions of the BSD License\r
  which accompanies this distribution.  The full text of the license may be found at\r
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "Socket.h"\r
\r
\r
CONST EFI_GUID mEslRawServiceGuid = {\r
  0xc31bf4a5, 0x2c7, 0x49d2, { 0xa5, 0x58, 0xfe, 0x62, 0x6f, 0x7e, 0xd4, 0x77 }\r
};\r
\r
CONST EFI_GUID mEslTcp4ServiceGuid = {\r
  0xffc659c2, 0x4ef2, 0x4532, { 0xb8, 0x75, 0xcd, 0x9a, 0xa4, 0x27, 0x4c, 0xde }\r
};\r
\r
CONST EFI_GUID mEslUdp4ServiceGuid = {\r
  0x44e03a55, 0x8d97, 0x4511, { 0xbf, 0xef, 0xa, 0x8b, 0xc6, 0x2c, 0x25, 0xae }\r
};\r
\r
\r
/**\r
  Connect to the EFI socket library\r
\r
  @param [in] ppSocketProtocol  Address to receive the socket protocol address\r
\r
  @retval 0             Successfully returned the socket protocol\r
  @retval other         Value for errno\r
 **/\r
int\r
EslServiceGetProtocol (\r
  IN EFI_SOCKET_PROTOCOL ** ppSocketProtocol\r
  )\r
{\r
  EFI_HANDLE ChildHandle;\r
  DT_SOCKET * pSocket;\r
  int RetVal;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Assume success\r
  //\r
  RetVal = 0;\r
\r
  //\r
  //  Locate the socket protocol\r
  //\r
  ChildHandle = NULL;\r
  Status = EslSocketAllocate ( &ChildHandle,\r
                               DEBUG_SOCKET,\r
                               &pSocket );\r
  if ( !EFI_ERROR ( Status )) {\r
    *ppSocketProtocol = &pSocket->SocketProtocol;\r
  }\r
  else {\r
    //\r
    //  No resources\r
    //\r
    RetVal = ENOMEM;\r
  }\r
\r
  //\r
  //  Return the operation status\r
  //\r
  DBG_EXIT_DEC ( RetVal );\r
  return RetVal;\r
}\r
\r
\r
/**\r
  Connect to the network layer\r
\r
  @retval EFI_SUCCESS   Successfully connected to the network layer\r
\r
 **/\r
EFI_STATUS\r
EslServiceNetworkConnect (\r
  VOID\r
  )\r
{\r
  UINTN HandleCount;\r
  EFI_HANDLE * pHandles;\r
  UINTN Index;\r
  CONST DT_SOCKET_BINDING * pSocketBinding;\r
  CONST DT_SOCKET_BINDING * pEnd;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Initialize the socket layer\r
  //\r
  Status = EFI_SUCCESS;\r
  EslServiceLoad ( gImageHandle );\r
\r
  //\r
  //  Connect the network devices\r
  //\r
  pSocketBinding = &cEslSocketBinding [0];\r
  pEnd = &pSocketBinding [ cEslSocketBindingEntries ];\r
  while ( pEnd > pSocketBinding ) {\r
    //\r
    //  Attempt to locate the network adapters\r
    //\r
    HandleCount = 0;\r
    pHandles = NULL;\r
    Status = gBS->LocateHandleBuffer ( ByProtocol,\r
                                       pSocketBinding->pNetworkBinding,\r
                                       NULL,\r
                                       &HandleCount,\r
                                       &pHandles );\r
    if ( EFI_ERROR ( Status )) {\r
      break;\r
    }\r
    if ( NULL != pHandles ) {\r
      //\r
      //  Attempt to connect to this network adapter\r
      //\r
      for ( Index = 0; HandleCount > Index; Index++ ) {\r
        Status = EslServiceConnect ( gImageHandle,\r
                                     pHandles [ Index ]);\r
        if ( EFI_ERROR ( Status )) {\r
          break;\r
        }\r
      }\r
\r
      //\r
      //  Done with the handles\r
      //\r
      gBS->FreePool ( pHandles );\r
    }\r
\r
    //\r
    //  Set the next network protocol\r
    //\r
    pSocketBinding += 1;\r
  }\r
\r
  //\r
  //  Return the network connection status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r
\r
\r
/**\r
  Disconnect from the network layer\r
\r
  @retval EFI_SUCCESS   Successfully disconnected from the network layer\r
\r
 **/\r
EFI_STATUS\r
EslServiceNetworkDisconnect (\r
  VOID\r
  )\r
{\r
  UINTN HandleCount;\r
  EFI_HANDLE * pHandles;\r
  UINTN Index;\r
  CONST DT_SOCKET_BINDING * pSocketBinding;\r
  CONST DT_SOCKET_BINDING * pEnd;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Assume success\r
  //\r
  Status = EFI_SUCCESS;\r
\r
  //\r
  //  Disconnect the network devices\r
  //\r
  pSocketBinding = &cEslSocketBinding [0];\r
  pEnd = &pSocketBinding [ cEslSocketBindingEntries ];\r
  while ( pEnd > pSocketBinding ) {\r
    //\r
    //  Attempt to locate the network adapters\r
    //\r
    HandleCount = 0;\r
    pHandles = NULL;\r
    Status = gBS->LocateHandleBuffer ( ByProtocol,\r
                                       pSocketBinding->pNetworkBinding,\r
                                       NULL,\r
                                       &HandleCount,\r
                                       &pHandles );\r
    if ( EFI_ERROR ( Status )) {\r
      break;\r
    }\r
    if ( NULL != pHandles ) {\r
      //\r
      //  Attempt to disconnect from this network adapter\r
      //\r
      for ( Index = 0; HandleCount > Index; Index++ ) {\r
        Status = EslServiceDisconnect ( gImageHandle,\r
                                        pHandles [ Index ]);\r
        if ( EFI_ERROR ( Status )) {\r
          break;\r
        }\r
      }\r
\r
      //\r
      //  Done with the handles\r
      //\r
      gBS->FreePool ( pHandles );\r
    }\r
\r
    //\r
    //  Set the next network protocol\r
    //\r
    pSocketBinding += 1;\r
  }\r
\r
  //\r
  //  Finish the disconnect operation\r
  //\r
  if ( !EFI_ERROR ( Status )) {\r
    EslServiceUnload ( );\r
  }\r
\r
  //\r
  //  Return the network connection status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r
\r
\r
PFN_ESL_xSTRUCTOR mpfnEslConstructor = EslServiceNetworkConnect;\r
PFN_ESL_xSTRUCTOR mpfnEslDestructor = EslServiceNetworkDisconnect;\r