Added IConnectionSocketFeature for exposing underlying Socket!

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!

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,8 @@ CancellationToken IConnectionLifetimeFeature.ConnectionClosed
             set => ConnectionClosed = value;
         }
 
+        Socket? IConnectionSocketFeature.Socket { get; }
+
         void IConnectionLifetimeFeature.Abort() => Abort(new ConnectionAbortedException("The connection was aborted by the application via IConnectionLifetimeFeature.Abort()."));
     }
 }

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

@@ -19,6 +19,7 @@ internal partial class TransportConnection : IFeatureCollection
         private object? _currentIConnectionItemsFeature;
         private object? _currentIMemoryPoolFeature;
         private object? _currentIConnectionLifetimeFeature;
+        private object? _currentIConnectionSocketFeature;
 
         private int _featureRevision;
 
@@ -31,6 +32,7 @@ private void FastReset()
             _currentIConnectionItemsFeature = this;
             _currentIMemoryPoolFeature = this;
             _currentIConnectionLifetimeFeature = this;
+            _currentIConnectionSocketFeature = this;
 
         }
 
@@ -123,6 +125,10 @@ private void ExtraFeatureSet(Type key, object? value)
                 {
                     feature = _currentIConnectionLifetimeFeature;
                 }
+                else if (key == typeof(IConnectionSocketFeature))
+                {
+                    feature = _currentIConnectionSocketFeature;
+                }
                 else if (MaybeExtra != null)
                 {
                     feature = ExtraFeatureGet(key);
@@ -155,6 +161,10 @@ private void ExtraFeatureSet(Type key, object? value)
                 {
                     _currentIConnectionLifetimeFeature = value;
                 }
+                else if (key == typeof(IConnectionSocketFeature))
+                {
+                    _currentIConnectionSocketFeature = value;
+                }
                 else
                 {
                     ExtraFeatureSet(key, value);
@@ -185,6 +195,10 @@ private void ExtraFeatureSet(Type key, object? value)
             {
                 feature = (TFeature?)_currentIConnectionLifetimeFeature;
             }
+            else if (typeof(TFeature) == typeof(IConnectionSocketFeature))
+            {
+                feature = (TFeature?)_currentIConnectionSocketFeature;
+            }
             else if (MaybeExtra != null)
             {
                 feature = (TFeature?)(ExtraFeatureGet(typeof(TFeature)));
@@ -216,6 +230,10 @@ private void ExtraFeatureSet(Type key, object? value)
             {
                 _currentIConnectionLifetimeFeature = feature;
             }
+            else if (typeof(TFeature) == typeof(IConnectionSocketFeature))
+            {
+                _currentIConnectionSocketFeature = feature;
+            }
             else
             {
                 ExtraFeatureSet(typeof(TFeature), feature);
@@ -244,6 +262,10 @@ private IEnumerable<KeyValuePair<Type, object>> FastEnumerable()
             {
                 yield return new KeyValuePair<Type, object>(typeof(IConnectionLifetimeFeature), _currentIConnectionLifetimeFeature);
             }
+            if (_currentIConnectionSocketFeature != null)
+            {
+                yield return new KeyValuePair<Type, object>(typeof(IConnectionSocketFeature), _currentIConnectionSocketFeature);
+            }
 
             if (MaybeExtra != null)
             {

src/Servers/Kestrel/shared/TransportConnection.cs

@@ -1,7 +1,6 @@
 // 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;

src/Servers/Kestrel/tools/CodeGenerator/TransportConnectionFeatureCollection.cs

@@ -15,7 +15,8 @@ public static string GenerateFile()
                 "IConnectionTransportFeature",
                 "IConnectionItemsFeature",
                 "IMemoryPoolFeature",
-                "IConnectionLifetimeFeature"
+                "IConnectionLifetimeFeature",
+                "IConnectionSocketFeature"
             };
 
             var usings = $@"