NetworkPkg: Support TCP Cancel function
This path is used to support TCP Cancel function to abort an asynchronous connection, listen, transmission or receive request. If any TCP CompletionToken is not signaled, it should not be closed directly by calling CloseEvent (Still in the TCP TokenList). If not, any exception behavior may be triggered. We should cancel it by calling Tcp->Cancel() first. In such a case, TCP Cancel function is necessary. Cc: Ye Ting <ting.ye@intel.com> Cc: Fu Siyuan <siyuan.fu@intel.com> Cc: Zhang Lubo <lubo.zhang@intel.com> Cc: Hegde Nagaraj P <nagaraj-p.hegde@hpe.com> Cc: Gary Lin <glin@suse.com> Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Jiaxin Wu <jiaxin.wu@intel.com> Reviewed-by: Gary Lin <glin@suse.com> Reviewed-by: Hegde Nagaraj P <nagaraj-p.hegde@hpe.com> Reviewed-by: Ye Ting <ting.ye@intel.com> Tested-by: Gary Lin <glin@suse.com> Tested-by: Hegde Nagaraj P <nagaraj-p.hegde@hpe.com>
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
/** @file
|
||||
Interface function of the Socket.
|
||||
|
||||
Copyright (c) 2009 - 2012, Intel Corporation. All rights reserved.<BR>
|
||||
Copyright (c) 2009 - 2016, Intel Corporation. All rights reserved.<BR>
|
||||
|
||||
This program and the accompanying materials
|
||||
are licensed and made available under the terms and conditions of the BSD License
|
||||
@@ -876,6 +876,96 @@ Exit:
|
||||
return Status;
|
||||
}
|
||||
|
||||
/**
|
||||
Abort the socket associated connection, listen, transmission or receive request.
|
||||
|
||||
@param[in, out] Sock Pointer to the socket to abort.
|
||||
@param[in] Token Pointer to a token that has been issued by
|
||||
Connect(), Accept(), Transmit() or Receive(). If
|
||||
NULL, all pending tokens issued by the four
|
||||
functions listed above will be aborted.
|
||||
|
||||
@retval EFI_UNSUPPORTED The operation is not supported in the current
|
||||
implementation.
|
||||
**/
|
||||
EFI_STATUS
|
||||
SockCancel (
|
||||
IN OUT SOCKET *Sock,
|
||||
IN VOID *Token
|
||||
)
|
||||
{
|
||||
EFI_STATUS Status;
|
||||
|
||||
Status = EFI_SUCCESS;
|
||||
|
||||
ASSERT (SockStream == Sock->Type);
|
||||
|
||||
Status = EfiAcquireLockOrFail (&(Sock->Lock));
|
||||
if (EFI_ERROR (Status)) {
|
||||
DEBUG (
|
||||
(EFI_D_ERROR,
|
||||
"SockCancel: Get the access for socket failed with %r",
|
||||
Status)
|
||||
);
|
||||
|
||||
return EFI_ACCESS_DENIED;
|
||||
}
|
||||
|
||||
if (SOCK_IS_UNCONFIGURED (Sock)) {
|
||||
Status = EFI_NOT_STARTED;
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
//
|
||||
// 1. Check ConnectionToken.
|
||||
//
|
||||
if (Token == NULL || (SOCK_COMPLETION_TOKEN *) Token == Sock->ConnectionToken) {
|
||||
if (Sock->ConnectionToken != NULL) {
|
||||
SIGNAL_TOKEN (Sock->ConnectionToken, EFI_ABORTED);
|
||||
Sock->ConnectionToken = NULL;
|
||||
}
|
||||
|
||||
if (Token != NULL) {
|
||||
Status = EFI_SUCCESS;
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// 2. Check ListenTokenList.
|
||||
//
|
||||
Status = SockCancelToken (Token, &Sock->ListenTokenList);
|
||||
if (Token != NULL && !EFI_ERROR (Status)) {
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
//
|
||||
// 3. Check RcvTokenList.
|
||||
//
|
||||
Status = SockCancelToken (Token, &Sock->RcvTokenList);
|
||||
if (Token != NULL && !EFI_ERROR (Status)) {
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
//
|
||||
// 4. Check SndTokenList.
|
||||
//
|
||||
Status = SockCancelToken (Token, &Sock->SndTokenList);
|
||||
if (Token != NULL && !EFI_ERROR (Status)) {
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
//
|
||||
// 5. Check ProcessingSndTokenList.
|
||||
//
|
||||
Status = SockCancelToken (Token, &Sock->ProcessingSndTokenList);
|
||||
|
||||
Exit:
|
||||
EfiReleaseLock (&(Sock->Lock));
|
||||
return Status;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
Get the mode data of the low layer protocol.
|
||||
|
||||
|
Reference in New Issue
Block a user