/// <summary>
/// Handler for pre-login request
/// </summary>
public override TDSMessageCollection OnPreLoginRequest(ITDSServerSession session, TDSMessage request)
{
// Delegate to the base class
TDSMessageCollection response = base.OnPreLoginRequest(session, request);
// Check if arguments are of the routing server
if (Arguments is RoutingTDSServerArguments)
{
// Cast to routing server arguments
RoutingTDSServerArguments serverArguments = Arguments as RoutingTDSServerArguments;
// Check if routing is configured during login
if (serverArguments.RouteOnPacket == TDSMessageType.TDS7Login)
{
// Check if pre-login response is contained inside the first message
if (response.Count > 0 && response[0].Any(t => t is TDSPreLoginToken))
{
// Find the first prelogin token
TDSPreLoginToken preLoginResponse = (TDSPreLoginToken)response[0].Where(t => t is TDSPreLoginToken).First();
// Inflate pre-login request from the message
TDSPreLoginToken preLoginRequest = request[0] as TDSPreLoginToken;
// Update MARS with the requested value
preLoginResponse.IsMARS = preLoginRequest.IsMARS;
}
}
}
return(response);
}
/// <summary>
/// Handler for pre-login request
/// </summary>
public virtual TDSMessageCollection OnPreLoginRequest(ITDSServerSession session, TDSMessage request)
{
// Inflate pre-login request from the message
TDSPreLoginToken preLoginRequest = request[0] as TDSPreLoginToken;
// Log request
TDSUtilities.Log(Arguments.Log, "Request", preLoginRequest);
// Generate server response for encryption
TDSPreLoginTokenEncryptionType serverResponse = TDSUtilities.GetEncryptionResponse(preLoginRequest.Encryption, Arguments.Encryption);
// Update client state with encryption resolution
session.Encryption = TDSUtilities.ResolveEncryption(preLoginRequest.Encryption, serverResponse);
// Create TDS prelogin packet
TDSPreLoginToken preLoginToken = new TDSPreLoginToken(Arguments.ServerVersion, serverResponse, false); // TDS server doesn't support MARS
// Cache the recieved Nonce into the session
(session as GenericTDSServerSession).ClientNonce = preLoginRequest.Nonce;
// Check if the server has been started up as requiring FedAuth when choosing between SSPI and FedAuth
if (Arguments.FedAuthRequiredPreLoginOption == TdsPreLoginFedAuthRequiredOption.FedAuthRequired)
{
if (preLoginRequest.FedAuthRequired == TdsPreLoginFedAuthRequiredOption.FedAuthRequired)
{
// Set the FedAuthRequired option
preLoginToken.FedAuthRequired = TdsPreLoginFedAuthRequiredOption.FedAuthRequired;
}
// Keep the federated authentication required flag in the server session
(session as GenericTDSServerSession).FedAuthRequiredPreLoginServerResponse = preLoginToken.FedAuthRequired;
if (preLoginRequest.Nonce != null)
{
// Generate Server Nonce
preLoginToken.Nonce = _GenerateRandomBytes(32);
}
}
// Cache the server Nonce in a session
(session as GenericTDSServerSession).ServerNonce = preLoginToken.Nonce;
// Log response
TDSUtilities.Log(Arguments.Log, "Response", preLoginToken);
// Reset authentication information
session.SQLUserID = null;
session.NTUserAuthenticationContext = null;
// Respond with a single message that contains only one token
return(new TDSMessageCollection(new TDSMessage(TDSMessageType.Response, preLoginToken)));
}
/// <summary>
/// Handler for login request
/// </summary>
public override TDSMessageCollection OnPreLoginRequest(ITDSServerSession session, TDSMessage request)
{
// Get the collection from a valid On PreLogin Request
TDSMessageCollection preLoginCollection = base.OnPreLoginRequest(session, request);
// Check if arguments are of the Federated Authentication server
if (Arguments is FederatedAuthenticationNegativeTDSServerArguments)
{
// Cast to federated authentication server arguments
FederatedAuthenticationNegativeTDSServerArguments ServerArguments = Arguments as FederatedAuthenticationNegativeTDSServerArguments;
// Find the is token carrying on TDSPreLoginToken
TDSPreLoginToken preLoginToken = preLoginCollection.Find(message => message.Exists(packetToken => packetToken is TDSPreLoginToken)).
Find(packetToken => packetToken is TDSPreLoginToken) as TDSPreLoginToken;
switch (ServerArguments.Scenario)
{
case FederatedAuthenticationNegativeTDSScenarioType.NonceMissingInFedAuthPreLogin:
{
// If we have the prelogin token
if (preLoginToken != null && preLoginToken.Nonce != null)
{
// Nullify the nonce from the Token
preLoginToken.Nonce = null;
}
break;
}
case FederatedAuthenticationNegativeTDSScenarioType.InvalidB_FEDAUTHREQUIREDResponse:
{
// If we have the prelogin token
if (preLoginToken != null)
{
// Set an illegal value for B_FEDAUTHREQURED
preLoginToken.FedAuthRequired = TdsPreLoginFedAuthRequiredOption.Illegal;
}
break;
}
}
}
// Return the collection
return(preLoginCollection);
}
