Added IConnectionSocketFeature for exposing underlying Socket!

Author: ShreyasJejurkar

src/Servers/Connections.Abstractions/src/BaseConnectionContext.cs

@@ -4,6 +4,7 @@
 using System;
 using System.Collections.Generic;
 using System.Net;
+using System.Net.Sockets;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Http.Features;
@@ -45,6 +46,11 @@ public abstract class BaseConnectionContext : IAsyncDisposable
         /// </summary>
         public virtual EndPoint? RemoteEndPoint { get; set; }
 
+        /// <summary>
+        /// Gets or sets the socket for this connection.
+        /// </summary>
+        public virtual Socket? Socket { get; set; }
+
         /// <summary>
         /// Aborts the underlying connection.
         /// </summary>

src/Servers/Connections.Abstractions/src/DefaultConnectionContext.cs

@@ -5,6 +5,7 @@
 using System.Collections.Generic;
 using System.IO.Pipelines;
 using System.Net;
+using System.Net.Sockets;
 using System.Security.Claims;
 using System.Threading;
 using System.Threading.Tasks;
@@ -22,7 +23,8 @@ public class DefaultConnectionContext : ConnectionContext,
                                             IConnectionTransportFeature,
                                             IConnectionUserFeature,
                                             IConnectionLifetimeFeature,
-                                            IConnectionEndPointFeature
+                                            IConnectionEndPointFeature,
+                                            IConnectionSocketFeature
     {
         private CancellationTokenSource _connectionClosedTokenSource = new CancellationTokenSource();
 
@@ -51,6 +53,7 @@ public DefaultConnectionContext(string id)
             Features.Set<IConnectionTransportFeature>(this);
             Features.Set<IConnectionLifetimeFeature>(this);
             Features.Set<IConnectionEndPointFeature>(this);
+            Features.Set<IConnectionSocketFeature>(this);
 
             ConnectionClosed = _connectionClosedTokenSource.Token;
         }
@@ -96,6 +99,9 @@ public DefaultConnectionContext(string id, IDuplexPipe transport, IDuplexPipe ap
         /// <inheritdoc />
         public override EndPoint? RemoteEndPoint { get; set; }
 
+        /// <inheritdoc />
+        public override Socket? Socket { get; set; }
+
         /// <inheritdoc />
         public override void Abort(ConnectionAbortedException abortReason)
         {

src/Servers/Connections.Abstractions/src/Features/IConnectionSocketFeature.cs

@@ -0,0 +1,18 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System.Net.Sockets;
+
+namespace Microsoft.AspNetCore.Connections.Features
+{
+    /// <summary>
+    /// The socket for the connection.
+    /// </summary>
+    public interface IConnectionSocketFeature
+    {
+        /// <summary>
+        /// Gets the underlying socket
+        /// </summary>
+        Socket? Socket { get; }
+    }
+}

src/Servers/Connections.Abstractions/src/PublicAPI.Unshipped.txt

@@ -1,5 +1,7 @@
 #nullable enable
 *REMOVED*Microsoft.AspNetCore.Connections.IConnectionListener.AcceptAsync(System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken)) -> System.Threading.Tasks.ValueTask<Microsoft.AspNetCore.Connections.ConnectionContext!>
+Microsoft.AspNetCore.Connections.Features.IConnectionSocketFeature
+Microsoft.AspNetCore.Connections.Features.IConnectionSocketFeature.Socket.get -> System.Net.Sockets.Socket?
 Microsoft.AspNetCore.Connections.IConnectionListener.AcceptAsync(System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken)) -> System.Threading.Tasks.ValueTask<Microsoft.AspNetCore.Connections.ConnectionContext?>
 Microsoft.AspNetCore.Connections.Experimental.IMultiplexedConnectionBuilder
 Microsoft.AspNetCore.Connections.Experimental.IMultiplexedConnectionBuilder.ApplicationServices.get -> System.IServiceProvider!
@@ -23,3 +25,7 @@ Microsoft.AspNetCore.Connections.Experimental.MultiplexedConnectionContext.Multi
 Microsoft.AspNetCore.Connections.Experimental.MultiplexedConnectionDelegate
 abstract Microsoft.AspNetCore.Connections.Experimental.MultiplexedConnectionContext.AcceptAsync(System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken)) -> System.Threading.Tasks.ValueTask<Microsoft.AspNetCore.Connections.ConnectionContext?>
 abstract Microsoft.AspNetCore.Connections.Experimental.MultiplexedConnectionContext.ConnectAsync(Microsoft.AspNetCore.Http.Features.IFeatureCollection? features = null, System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken)) -> System.Threading.Tasks.ValueTask<Microsoft.AspNetCore.Connections.ConnectionContext!>
+override Microsoft.AspNetCore.Connections.DefaultConnectionContext.Socket.get -> System.Net.Sockets.Socket?
+override Microsoft.AspNetCore.Connections.DefaultConnectionContext.Socket.set -> void
+virtual Microsoft.AspNetCore.Connections.BaseConnectionContext.Socket.get -> System.Net.Sockets.Socket?
+virtual Microsoft.AspNetCore.Connections.BaseConnectionContext.Socket.set -> void

src/Servers/Kestrel/Transport.Sockets/src/Internal/SocketConnection.cs

@@ -48,6 +48,7 @@ internal SocketConnection(Socket socket,
             MemoryPool = memoryPool;
             _trace = trace;
             _waitForData = waitForData;
+            Socket = _socket;
 
             LocalEndPoint = _socket.LocalEndPoint;
             RemoteEndPoint = _socket.RemoteEndPoint;

src/Servers/Kestrel/shared/TransportConnection.FeatureCollection.cs

@@ -4,6 +4,7 @@
 using System.Buffers;
 using System.Collections.Generic;
 using System.IO.Pipelines;
+using System.Net.Sockets;
 using System.Threading;
 using Microsoft.AspNetCore.Connections.Features;
 
@@ -15,7 +16,8 @@ internal partial class TransportConnection : IConnectionIdFeature,
                                                  IConnectionTransportFeature,
                                                  IConnectionItemsFeature,
                                                  IMemoryPoolFeature,
-                                                 IConnectionLifetimeFeature
+                                                 IConnectionLifetimeFeature,
+                                                 IConnectionSocketFeature
     {
         // NOTE: When feature interfaces are added to or removed from this TransportConnection class implementation,
         // then the list of `features` in the generated code project MUST also be updated.
@@ -41,6 +43,11 @@ CancellationToken IConnectionLifetimeFeature.ConnectionClosed
             set => ConnectionClosed = value;
         }
 
+        Socket? IConnectionSocketFeature.Socket
+        {
+            get => Socket;
+        }
+
         void IConnectionLifetimeFeature.Abort() => Abort(new ConnectionAbortedException("The connection was aborted by the application via IConnectionLifetimeFeature.Abort()."));
     }
 }

src/Servers/Kestrel/shared/TransportConnection.cs

@@ -1,12 +1,12 @@
 // Copyright (c) .NET Foundation. All rights reserved.
 // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
 
-using System;
 using System.Buffers;
 using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO.Pipelines;
 using System.Net;
+using System.Net.Sockets;
 using System.Threading;
 using Microsoft.AspNetCore.Http.Features;
 
@@ -65,6 +65,8 @@ public override string ConnectionId
             }
         }
 
+        public override Socket? Socket { get; set; }
+
         public override CancellationToken ConnectionClosed { get; set; }
 
         // DO NOT remove this override to ConnectionContext.Abort. Doing so would cause