huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Sync shared code from runtime (#24845) 

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2020-08-13 03:08:49 +00:00 committed by GitHub
parent 93cfbe46fd
commit da6aa6fc6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 30 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Shared/runtime/Quic/Implementations/MsQuic/Internal/MsQuicSession.cs
View File

@ -56,7 +56,7 @@ namespace System.Net.Quic.Implementations.MsQuic.Internal
QuicExceptionHelpers.ThrowIfFailed(MsQuicApi.Api.ListenerOpenDelegate(
_nativeObjPtr,
MsQuicListener.NativeCallbackHandler,
MsQuicListener.s_listenerDelegate,
IntPtr.Zero,
out IntPtr listenerPointer),
"Could not open listener.");

8
src/Shared/runtime/Quic/Implementations/MsQuic/MsQuicConnection.cs
View File

@ -37,8 +37,8 @@ namespace System.Net.Quic.Implementations.MsQuic
private SslApplicationProtocol _negotiatedAlpnProtocol;
// TODO: only allocate these when there is an outstanding connect/shutdown.
private readonly TaskCompletionSource<uint> _connectTcs = new TaskCompletionSource<uint>();
private readonly TaskCompletionSource<uint> _shutdownTcs = new TaskCompletionSource<uint>();
private readonly TaskCompletionSource<uint> _connectTcs = new TaskCompletionSource<uint>(TaskCreationOptions.RunContinuationsAsynchronously);
private readonly TaskCompletionSource<uint> _shutdownTcs = new TaskCompletionSource<uint>(TaskCreationOptions.RunContinuationsAsynchronously);
private bool _disposed;
private bool _connected;
@ -291,6 +291,8 @@ namespace System.Net.Quic.Implementations.MsQuic
ErrorCode);
QuicExceptionHelpers.ThrowIfFailed(status, "Failed to shutdown connection.");
Debug.Assert(_shutdownTcs.Task.IsCompleted == false);
return new ValueTask(_shutdownTcs.Task);
}
@ -349,6 +351,8 @@ namespace System.Net.Quic.Implementations.MsQuic
_disposed = true;
}
// TODO: this appears abortive and will cause prior successfully shutdown and closed streams to drop data.
// It's unclear how to gracefully wait for a connection to be 100% done.
internal override ValueTask CloseAsync(long errorCode, CancellationToken cancellationToken = default)
{
ThrowIfDisposed();

4
src/Shared/runtime/Quic/Implementations/MsQuic/MsQuicListener.cs
View File

@ -26,7 +26,7 @@ namespace System.Net.Quic.Implementations.MsQuic
private GCHandle _handle;
// Delegate that wraps the static function that will be called when receiving an event.
private static readonly ListenerCallbackDelegate s_listenerDelegate = new ListenerCallbackDelegate(NativeCallbackHandler);
internal static readonly ListenerCallbackDelegate s_listenerDelegate = new ListenerCallbackDelegate(NativeCallbackHandler);
// Ssl listening options (ALPN, cert, etc)
private readonly SslServerAuthenticationOptions _sslOptions;
@ -189,7 +189,7 @@ namespace System.Net.Quic.Implementations.MsQuic
_acceptConnectionQueue.Writer.TryComplete();
}
internal static uint NativeCallbackHandler(
private static uint NativeCallbackHandler(
IntPtr listener,
IntPtr context,
ref ListenerEvent connectionEventStruct)

42
src/Shared/runtime/Quic/Implementations/MsQuic/MsQuicStream.cs
View File

@ -34,7 +34,8 @@ namespace System.Net.Quic.Implementations.MsQuic
// Resettable completions to be used for multiple calls to receive.
private readonly ResettableCompletionSource<uint> _receiveResettableCompletionSource;
private readonly ResettableCompletionSource<uint> _shutdownWriteResettableCompletionSource;
// Set once writes have been shutdown.
private readonly TaskCompletionSource _shutdownWriteCompletionSource = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
// Buffers to hold during a call to send.
private MemoryHandle[] _bufferArrays = new MemoryHandle[1];
@ -49,7 +50,7 @@ namespace System.Net.Quic.Implementations.MsQuic
private ReadState _readState;
private long _readErrorCode = -1;
private ShutdownWriteState _shutdownState;
private ShutdownWriteState _shutdownWriteState;
private SendState _sendState;
private long _sendErrorCode = -1;
@ -76,7 +77,6 @@ namespace System.Net.Quic.Implementations.MsQuic
_sendResettableCompletionSource = new ResettableCompletionSource<uint>();
_receiveResettableCompletionSource = new ResettableCompletionSource<uint>();
_shutdownWriteResettableCompletionSource = new ResettableCompletionSource<uint>();
SetCallbackHandler();
bool isBidirectional = !flags.HasFlag(QUIC_STREAM_OPEN_FLAG.UNIDIRECTIONAL);
@ -323,16 +323,16 @@ namespace System.Net.Quic.Implementations.MsQuic
lock (_sync)
{
if (_shutdownState == ShutdownWriteState.None)
if (_shutdownWriteState == ShutdownWriteState.None)
{
_shutdownState = ShutdownWriteState.Canceled;
_shutdownWriteState = ShutdownWriteState.Canceled;
shouldComplete = true;
}
}
if (shouldComplete)
{
_shutdownWriteResettableCompletionSource.CompleteException(new QuicStreamAbortedException("Shutdown was aborted.", errorCode));
_shutdownWriteCompletionSource.SetException(new QuicStreamAbortedException("Shutdown was aborted.", errorCode));
}
MsQuicApi.Api.StreamShutdownDelegate(_ptr, (uint)QUIC_STREAM_SHUTDOWN_FLAG.ABORT_SEND, errorCode);
@ -348,20 +348,20 @@ namespace System.Net.Quic.Implementations.MsQuic
bool shouldComplete = false;
lock (_sync)
{
if (_shutdownState == ShutdownWriteState.None)
if (_shutdownWriteState == ShutdownWriteState.None)
{
_shutdownState = ShutdownWriteState.Canceled;
_shutdownWriteState = ShutdownWriteState.Canceled;
shouldComplete = true;
}
}
if (shouldComplete)
{
_shutdownWriteResettableCompletionSource.CompleteException(new OperationCanceledException("Shutdown was canceled", cancellationToken));
_shutdownWriteCompletionSource.SetException(new OperationCanceledException("Shutdown was canceled", cancellationToken));
}
});
return _shutdownWriteResettableCompletionSource.GetTypelessValueTask();
return new ValueTask(_shutdownWriteCompletionSource.Task);
}
internal override void Shutdown()
@ -408,8 +408,6 @@ namespace System.Net.Quic.Implementations.MsQuic
return default;
}
CleanupSendState();
if (_ptr != IntPtr.Zero)
{
// TODO resolve graceful vs abortive dispose here. Will file a separate issue.
@ -419,6 +417,8 @@ namespace System.Net.Quic.Implementations.MsQuic
_handle.Free();
CleanupSendState();
_disposed = true;
return default;
@ -442,8 +442,6 @@ namespace System.Net.Quic.Implementations.MsQuic
return;
}
CleanupSendState();
if (_ptr != IntPtr.Zero)
{
// TODO resolve graceful vs abortive dispose here. Will file a separate issue.
@ -453,6 +451,8 @@ namespace System.Net.Quic.Implementations.MsQuic
_handle.Free();
CleanupSendState();
_disposed = true;
}
@ -461,7 +461,7 @@ namespace System.Net.Quic.Implementations.MsQuic
MsQuicApi.Api.StreamReceiveSetEnabledDelegate(_ptr, enabled: true);
}
internal static uint NativeCallbackHandler(
private static uint NativeCallbackHandler(
IntPtr stream,
IntPtr context,
ref StreamEvent streamEvent)
@ -614,16 +614,16 @@ namespace System.Net.Quic.Implementations.MsQuic
bool shouldComplete = false;
lock (_sync)
{
if (_shutdownState == ShutdownWriteState.None)
if (_shutdownWriteState == ShutdownWriteState.None)
{
_shutdownState = ShutdownWriteState.Finished;
_shutdownWriteState = ShutdownWriteState.Finished;
shouldComplete = true;
}
}
if (shouldComplete)
{
_shutdownWriteResettableCompletionSource.Complete(MsQuicStatusCodes.Success);
_shutdownWriteCompletionSource.TrySetResult();
}
return MsQuicStatusCodes.Success;
@ -646,9 +646,9 @@ namespace System.Net.Quic.Implementations.MsQuic
_readState = ReadState.ReadsCompleted;
if (_shutdownState == ShutdownWriteState.None)
if (_shutdownWriteState == ShutdownWriteState.None)
{
_shutdownState = ShutdownWriteState.Finished;
_shutdownWriteState = ShutdownWriteState.Finished;
shouldShutdownWriteComplete = true;
}
}
@ -660,7 +660,7 @@ namespace System.Net.Quic.Implementations.MsQuic
if (shouldShutdownWriteComplete)
{
_shutdownWriteResettableCompletionSource.Complete(MsQuicStatusCodes.Success);
_shutdownWriteCompletionSource.TrySetResult();
}
return MsQuicStatusCodes.Success;