Login handler for login (#2730)

* adding the Tds Stream APIs

* Fix the NS

* Fix the enums

* adding the change underlying stream

* Address CR comments

* Initial Prelogin

* Adding prelogin handler

* Prelogin handler

* Adding the ssl work in progress

* Prelogin WIP

* Prelogin WIP

* Adding the prelogin handler changes for encryption

* Fix object Id

* Clean up the handler

* adding the prelogin handler sub handlers

* Adding the sub handlers for prelogin :

* Simplify

* Extract constants

* Adding the prelogin handler comment

* adding the GUID for the transport

* Adding the Login handler

* Adding the server info

* adding the login handler details

* Adding prelogin initial

* Checkout unintended

* Refactor the handlers and move some methods to context.

* Usage of Binary Primitives

* adding the prelogin bug fixes

* adding the refactored changes

* Improve Exception handling

* Fix the handler class name casing

* Add comments about SNI error and simplify code

* Throw the errors

* add comments

* Adding tests

* Preallocate prelogin write buffer

* Fix bad commit

* adding the comments on the context object

* Commit for Exceptions handling

* adding the simplification of wirte

* Using only the offset

* Move to binary primitives

* Fix the tests

* Adding the tests for Tds Prelogin handler

* Address CR comments

* address CR comments

* Handler Changes for login

* adding the csproj changes

* Address CR comments

* Remove not needed TODO

* Move write to BinaryPrimitives

* Move the password change request

* adding the login handler changes

* adding the feature extension

* adding the login handler port

* adding the chain

* More changes

* adding the changes

* adding the login changes

* Finish the writing

* Clean up

* Adding the login complete

* adding the complete login flow

* Adding the code

* Adding the right compilation

* Adding the Login handler

* adding the compilation fix

* Fixing The NRE

* Fixing the Password

* Fix bugs

* Fix the bugs and added tests

* Remove unused code. Refactor

* Refactor the login context

* Adding the additional refactor

* Refactor the code for login

* Remove creds

* Round 1 of CR comments

* Adding the CR changes part 2

* Adding tests and more refactoring

* Adding unit tests

---------

Co-authored-by: Saurabh Singh <singhsaura@microsoft.com>

Author: saurabh500

src/Microsoft.Data.SqlClient/netcore/src/Microsoft.Data.SqlClient.csproj

@@ -929,6 +929,10 @@
     <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\ConnectionHandlerContext.cs" />
     <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\DataSourceParsingHandler.cs" />
     <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\PreloginHandler.cs" />
+    <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\LoginHandler.cs" />
+    <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\Login\LoginHandlerContext.cs" />
+    <Compile Include="Microsoft\Data\SqlClientX\Handlers\PasswordChangeRequest.cs" />
+    <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\Login\FeatureExtensions.cs" />
     <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\PreloginSubHandlers\PreloginHandlerContext.cs" />
     <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\PreloginSubHandlers\PreloginPacketHandler.cs" />
     <Compile Include="Microsoft\Data\SqlClientX\Handlers\Connection\PreloginSubHandlers\Tds8TlsHandler.cs" />

src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClientX/Handlers/Connection/ConnectionHandlerContext.cs

@@ -8,8 +8,10 @@
 using System.Net.Security;
 using System.Threading;
 using System.Threading.Tasks;
+using Microsoft.Data.ProviderBase;
 using Microsoft.Data.SqlClient;
 using Microsoft.Data.SqlClient.SNI;
+using Microsoft.Data.SqlClientX.Handlers.Connection.Login;
 using Microsoft.Data.SqlClientX.IO;
 
 namespace Microsoft.Data.SqlClientX.Handlers.Connection
@@ -18,13 +20,18 @@ namespace Microsoft.Data.SqlClientX.Handlers.Connection
      /// </summary>
     internal class ConnectionHandlerContext : HandlerRequest, ICloneable
     {
-
         // TODO: Decide if we need a default constructor depending on the latest design 
         /// <summary>
         /// Stream used by readers.
         /// </summary>
         public Stream ConnectionStream { get; set; }
 
+        /// <summary>
+        /// A timer representing the timeout for the connection.
+        /// TODO: This might require rethinking.
+        /// </summary>
+        internal TimeoutTimer TimeoutTimer { get; set; }
+
         /// <summary>
         /// Class that contains data required to handle a connection request.
         /// </summary>
@@ -64,7 +71,7 @@ internal class ConnectionHandlerContext : HandlerRequest, ICloneable
         /// <summary>
         /// Indicates if fed auth needed for this connection.
         /// </summary>
-        public bool IsFedAuthRequired { get; internal set; }
+        public bool IsFedAuthNegotiatedInPrelogin { get; internal set; }
 
         /// <summary>
         /// The access token in bytes.
@@ -96,6 +103,18 @@ internal class ConnectionHandlerContext : HandlerRequest, ICloneable
         /// </summary>
         internal Func<SqlAuthenticationParameters, CancellationToken, Task<SqlAuthenticationToken>> AccessTokenCallback { get; set; }
 
+        /// <summary>
+        /// Represents a password change request on this connection.
+        /// </summary>
+        public PasswordChangeRequest PasswordChangeRequest { get; internal set; }
+
+        /// <summary>
+        /// Features in the login request.
+        /// This needn't be cloned, since after routing, the feature extensions need to be 
+        /// re-requested like it was for a fresh connection.
+        /// </summary>
+        public FeatureExtensions Features { get; } = new();
+
         /// <summary>
         /// Clone <see cref="ConnectionHandlerContext"/> as part of history along the chain.
         /// </summary>
@@ -112,11 +131,12 @@ public object Clone()
                 SslOverTdsStream = this.SslOverTdsStream, 
                 TdsStream = this.TdsStream, 
                 IsMarsCapable = this.IsMarsCapable,
-                IsFedAuthRequired = this.IsFedAuthRequired,
+                IsFedAuthNegotiatedInPrelogin = this.IsFedAuthNegotiatedInPrelogin,
                 AccessTokenInBytes = this.AccessTokenInBytes,
                 ServerInfo = this.ServerInfo, 
                 ErrorCollection = this.ErrorCollection, 
-                AccessTokenCallback = this.AccessTokenCallback // Assuming delegate cloning is not required
+                AccessTokenCallback = this.AccessTokenCallback, // Assuming delegate cloning is not required
+                PasswordChangeRequest = this.PasswordChangeRequest,
             };
         }