1 files changed, 116 insertions, 0 deletions
diff --git a/StdLib/BsdSocketLib/close.c b/StdLib/BsdSocketLib/close.c
new file mode 100644
index 0000000000..7d70e4f126
--- /dev/null
+++ b/StdLib/BsdSocketLib/close.c
@@ -0,0 +1,116 @@
+/** @file
+  Implement the close API.
+
+  Copyright (c) 2011, Intel Corporation
+  All rights reserved. This program and the accompanying materials
+  are licensed and made available under the terms and conditions of the BSD License
+  which accompanies this distribution.  The full text of the license may be found at
+  http://opensource.org/licenses/bsd-license.php
+
+  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+
+**/
+
+#include <SocketInternals.h>
+
+
+/**
+  Worker routine to close the socket.
+
+  @param [in] pSocketProtocol   Socket protocol structure address
+
+  @param [in] pErrno            Address of the errno variable
+
+  @retval EFI_SUCCESS   Successfully closed the socket
+
+**/
+EFI_STATUS
+BslSocketCloseWork (
+  IN EFI_SOCKET_PROTOCOL * pSocketProtocol,
+  IN int * pErrno
+  )
+{
+  EFI_SERVICE_BINDING_PROTOCOL * pServiceBinding;
+  EFI_STATUS Status;
+
+  //
+  //  Start closing the socket
+  //
+  Status = pSocketProtocol->pfnCloseStart ( pSocketProtocol,
+                                            FALSE,
+                                            pErrno );
+
+  //
+  //  Wait for the socket to close or an error
+  //
+  while ( EFI_NOT_READY == Status ) {
+    Status = pSocketProtocol->pfnClosePoll ( pSocketProtocol,
+                                             pErrno );
+  }
+  if ( !EFI_ERROR ( Status )) {
+    //
+    //  Locate the socket protocol
+    //
+    Status = gBS->LocateProtocol ( &gEfiSocketServiceBindingProtocolGuid,
+                                   NULL,
+                                   (VOID **) &pServiceBinding );
+    if ( !EFI_ERROR ( Status )) {
+      //
+      //  Release the handle
+      //
+      Status = pServiceBinding->DestroyChild ( pServiceBinding,
+                                               pSocketProtocol->SocketHandle );
+    }
+    if ( EFI_ERROR ( Status )) {
+      *pErrno = EIO;
+    }
+  }
+  else {
+    DEBUG (( DEBUG_ERROR,
+              "ERROR - Failed to close the socket: %r\r\n",
+              Status ));
+    *pErrno = EIO;
+  }
+
+  //
+  //  Return the close status
+  //
+  return Status;
+}
+
+
+/**
+  Close the socket
+
+  @param [in] pDescriptor Descriptor address for the file
+
+  @returns  This routine returns 0 upon success and -1 upon failure.
+            In the case of failure, errno contains more information.
+
+**/
+int
+BslSocketClose (
+  struct __filedes * pDescriptor
+  )
+{
+  int CloseStatus;
+  EFI_SOCKET_PROTOCOL * pSocketProtocol;
+
+  //
+  //  Locate the socket protocol
+  //
+  pSocketProtocol = BslValidateSocketFd ( pDescriptor, &errno );
+  if ( NULL != pSocketProtocol ) {
+    //
+    //  Close the socket
+    //
+    BslSocketCloseWork ( pSocketProtocol, &errno );
+  }
+
+  //
+  //  Return the close status
+  //
+  CloseStatus = ( errno == 0 ) ? 0 : -1;
+  return CloseStatus;
+}