Fix issue with truststore password being removed too early for XA transaction

src/main/java/com/microsoft/sqlserver/jdbc/SQLServerDataSource.java

@@ -358,6 +358,10 @@ public void setTrustStorePassword(String trustStorePassword) {
                 trustStorePassword);
     }
 
+    String getTrustStorePassword() {
+        return getStringProperty(connectionProps, SQLServerDriverStringProperty.TRUST_STORE_PASSWORD.toString(), null);
+    }
+
     @Override
     public void setHostNameInCertificate(String hostName) {
         setStringProperty(connectionProps, SQLServerDriverStringProperty.HOSTNAME_IN_CERTIFICATE.toString(), hostName);

src/main/java/com/microsoft/sqlserver/jdbc/SQLServerXAConnection.java

@@ -52,6 +52,20 @@ public final class SQLServerXAConnection extends SQLServerPooledConnection imple
             controlConnectionProperties.setProperty(SQLServerDriverStringProperty.PASSWORD.toString(), pwd);
         }
 
+        // Add truststore password property for creating the control connection. This will be removed again
+        String trustStorePassword = ds.getTrustStorePassword();
+        if (null == trustStorePassword) {
+                // trustStorePassword can either come from the connection string or added via SQLServerXADataSource::setTrustStorePassword.
+                // if trustStorePassword is null at this point, then check the connection string.
+                Properties urlProps = Util.parseUrl(ds.getURL(), xaLogger);
+                trustStorePassword = urlProps.getProperty(SQLServerDriverStringProperty.TRUST_STORE_PASSWORD.toString());
+        }
+
+        // if trustStorePassword is still null, it wasn't provided. Do not set the property as null to avoid NPE.
+        if (null != trustStorePassword) {
+            controlConnectionProperties.setProperty(SQLServerDriverStringProperty.TRUST_STORE_PASSWORD.toString(), trustStorePassword);
+        }
+
         if (xaLogger.isLoggable(Level.FINER))
             xaLogger.finer("Creating an internal control connection for" + toString());
         physicalControlConnection = null;