diff --git a/SnowflakePS/Commands/Authentication/ConnectAppCommand.cs b/SnowflakePS/Commands/Authentication/ConnectAppCommand.cs index 176c293..1174c2e 100644 --- a/SnowflakePS/Commands/Authentication/ConnectAppCommand.cs +++ b/SnowflakePS/Commands/Authentication/ConnectAppCommand.cs @@ -416,7 +416,7 @@ protected override void ProcessRecord() // Authenticate with SSO int localPort = GetRandomUnusedPort(); - string ssoLoginLinkResult = SnowflakeDriver.GetSSOLoginLinkForAccountAndUser(appUserContext.AccountUrl, appUserContext.AccountName, appUserContext.UserName, localPort); + string ssoLoginLinkResult = SnowflakeDriver.GetSSOLoginLinkForAccountAndUser(appUserContext.AccountUrl, appUserContext.AccountName, appUserContext.UserName, localPort, appUserContext.Cookies); if (ssoLoginLinkResult.Length == 0) { throw new InvalidCredentialException(String.Format("Invalid response on getting SSO for user {0}@{1}", appUserContext.UserName, appUserContext.AccountName)); diff --git a/SnowflakePS/UtilityDrivers/SnowflakeDriver.cs b/SnowflakePS/UtilityDrivers/SnowflakeDriver.cs index 16558ed..8a2889d 100644 --- a/SnowflakePS/UtilityDrivers/SnowflakeDriver.cs +++ b/SnowflakePS/UtilityDrivers/SnowflakeDriver.cs @@ -227,7 +227,7 @@ public static string GetMasterTokenAndSessionTokenFromSSOToken(string accountUrl cookies: cookies, snowflakeContext: String.Empty, referer: String.Empty, classicUIAuthToken: String.Empty); } - public static string GetSSOLoginLinkForAccountAndUser(string accountUrl, string accountName, string userName, int returnRedirectPortNumber) + public static string GetSSOLoginLinkForAccountAndUser(string accountUrl, string accountName, string userName, int returnRedirectPortNumber, CookieContainer cookies) { string requestJSONTemplate = @"{{ @@ -248,7 +248,8 @@ public static string GetSSOLoginLinkForAccountAndUser(string accountUrl, string "session/authenticator-request", "application/json", requestBody, - "application/json", null); + "application/json", + cookies: cookies); } #endregion