Merge branch 'access-token-callback' of https://github.com/microsoft/mssql-jdbc into access-token-callback

Author: tkyc

build.gradle

@@ -130,7 +130,7 @@ dependencies {
     implementation 'org.osgi:org.osgi.core:6.0.0',
 			'org.osgi:org.osgi.compendium:5.0.0'
 	compileOnly 'com.azure:azure-security-keyvault-keys:4.2.8',
-			'com.azure:azure-identity:1.3.3',
+			'com.azure:azure-identity:1.7.0-beta.2',
 			'org.antlr:antlr4-runtime:4.9.2',
 			'com.google.code.gson:gson:2.8.7',
 			'org.bouncycastle:bcprov-jdk15on:1.69',
@@ -152,7 +152,7 @@ dependencies {
 			'com.google.code.gson:gson:2.8.7',
 			'org.bouncycastle:bcprov-jdk15on:1.69',
 			'com.azure:azure-security-keyvault-keys:4.2.8',
-			'com.azure:azure-identity:1.3.3',
+			'com.azure:azure-identity:1.7.0-beta.2',
 			'com.microsoft.azure:adal4j:1.6.7',
 			'com.h2database:h2:1.4.200'
 }

pom.xml

@@ -63,9 +63,9 @@
 
 		<!-- Driver Dependencies -->
 		<org.osgi.core.version>6.0.0</org.osgi.core.version>
-		<azure-security-keyvault-keys.version>4.4.4</azure-security-keyvault-keys.version>
+		<azure-security-keyvault-keys.version>4.5.1</azure-security-keyvault-keys.version>
 		<azure-identity.version>1.7.0-beta.2</azure-identity.version>
-		<msal.version>1.13.0</msal.version>
+		<msal.version>1.13.3</msal.version>
 		<org.osgi.compendium.version>5.0.0</org.osgi.compendium.version>
 		<antlr-runtime.version>4.9.3</antlr-runtime.version>
 		<com.google.code.gson.version>2.9.0</com.google.code.gson.version>
@@ -109,7 +109,7 @@
 		<dependency>
 			<groupId>com.microsoft.azure</groupId>
 			<artifactId>msal4j</artifactId>
-			<version>1.13.0</version>
+			<version>1.13.3</version>
 			<optional>true</optional>
 		</dependency>
 

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

@@ -2416,7 +2416,7 @@ enum Result {
 
     // list of addresses for ip selection by type preference
     private static ArrayList<InetAddress> addressList = new ArrayList<>();
-    private static final Lock ADDRESS_LIST_LOCK = new ReentrantLock();
+    private static final Lock addressListLock = new ReentrantLock();
 
     /**
      * Constructs a new SocketFinder object with appropriate traceId
@@ -2765,7 +2765,7 @@ private SocketFactory getSocketFactory() throws IOException {
      */
     private InetSocketAddress getInetAddressByIPPreference(String hostName,
             int portNumber) throws IOException, SQLServerException {
-        ADDRESS_LIST_LOCK.lock();
+        addressListLock.lock();
         try {
             InetSocketAddress addr = InetSocketAddress.createUnresolved(hostName, portNumber);
             for (int i = 0; i < addressList.size(); i++) {
@@ -2775,7 +2775,7 @@ private InetSocketAddress getInetAddressByIPPreference(String hostName,
             }
             return addr;
         } finally {
-            ADDRESS_LIST_LOCK.unlock();
+            addressListLock.unlock();
         }
     }
 
@@ -2838,7 +2838,7 @@ private Socket getSocketByIPPreference(String hostName, int portNumber, int time
         // Note that Socket(host, port) throws an UnknownHostException if the host name
         // cannot be resolved, but that InetSocketAddress(host, port) does not - it sets
         // the returned InetSocketAddress as unresolved.
-        if (addr.isUnresolved()) {
+        if (addr != null && addr.isUnresolved()) {
             if (logger.isLoggable(Level.FINER)) {
                 logger.finer(this.toString() + "Failed to resolve host name: " + hostName
                         + ". Using IP address from DNS cache.");
@@ -2859,7 +2859,7 @@ private Socket getSocketByIPPreference(String hostName, int portNumber, int time
      *        Boolean switch for IPv6 first
      */
     private void fillAddressList(InetAddress[] addresses, boolean ipv6first) {
-        ADDRESS_LIST_LOCK.lock();
+        addressListLock.lock();
         try {
             addressList.clear();
             if (ipv6first) {
@@ -2876,7 +2876,7 @@ private void fillAddressList(InetAddress[] addresses, boolean ipv6first) {
                 }
             }
         } finally {
-            ADDRESS_LIST_LOCK.unlock();
+            addressListLock.unlock();
         }
     }
 
@@ -4155,7 +4155,7 @@ void writeString(String value) throws SQLServerException {
         int charsCopied = 0;
         int length = value.length();
         while (charsCopied < length) {
-            long bytesToCopy = 2 * (length - charsCopied);
+            long bytesToCopy = 2 * ((long) length - charsCopied);
 
             if (bytesToCopy > valueBytes.length)
                 bytesToCopy = valueBytes.length;
@@ -6053,7 +6053,8 @@ private void writeScaledTemporal(GregorianCalendar cal, int subSecondNanos, int
                     + 60 * 60 * cal.get(Calendar.HOUR_OF_DAY);
 
             // Scale nanos since midnight to the desired scale, rounding the value as necessary
-            long divisor = Nanos.PER_MAX_SCALE_INTERVAL * (long) Math.pow(10, TDS.MAX_FRACTIONAL_SECONDS_SCALE - scale);
+            long divisor = Nanos.PER_MAX_SCALE_INTERVAL
+                    * (long) Math.pow(10, TDS.MAX_FRACTIONAL_SECONDS_SCALE - (double) scale);
 
             // The scaledNanos variable represents the fractional seconds of the value at the scale
             // indicated by the scale variable. So, for example, scaledNanos = 3 means 300 nanoseconds
@@ -6195,7 +6196,8 @@ byte[] writeEncryptedScaledTemporal(GregorianCalendar cal, int subSecondNanos, i
                     + 60 * 60 * cal.get(Calendar.HOUR_OF_DAY);
 
             // Scale nanos since midnight to the desired scale, rounding the value as necessary
-            divisor = Nanos.PER_MAX_SCALE_INTERVAL * (long) Math.pow(10, TDS.MAX_FRACTIONAL_SECONDS_SCALE - scale);
+            divisor = Nanos.PER_MAX_SCALE_INTERVAL
+                    * (long) Math.pow(10, TDS.MAX_FRACTIONAL_SECONDS_SCALE - (double) scale);
 
             // The scaledNanos variable represents the fractional seconds of the value at the scale
             // indicated by the scale variable. So, for example, scaledNanos = 3 means 300 nanoseconds

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

@@ -469,7 +469,7 @@ public void run() {
                     } else {
                         try {
                             if (connectRetryCount > 1) {
-                                Thread.sleep(con.getRetryInterval() * 1000);
+                                Thread.sleep((long) (con.getRetryInterval()) * 1000);
                             }
                         } catch (InterruptedException ie) {
                             // re-interrupt thread

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

@@ -1365,7 +1365,7 @@ private int findColumn(String columnName) throws SQLServerException {
 
         // 1. Search using case-sensitive non-locale specific (binary) compare first.
         // 2. Search using case-insensitive, non-locale specific (binary) compare last.
-        Integer matchPos = parameterNames.get(columnNameWithSign);
+        Integer matchPos = (parameterNames != null) ? parameterNames.get(columnNameWithSign) : null;
         if (null == matchPos) {
             matchPos = insensitiveParameterNames.get(columnNameWithSign);
         }

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

@@ -227,10 +227,11 @@ public class SQLServerConnection implements ISQLServerConnection, java.io.Serial
      * lock instance for "this"
      **/
     private final Lock lock = new ReentrantLock();
+
     /**
      * static lock instance for the class
      **/
-    private static final Lock LOCK = new ReentrantLock();
+    private static final Lock sLock = new ReentrantLock();
 
     /**
      * Return an existing cached SharedTimer associated with this Connection or create a new one.
@@ -947,7 +948,7 @@ public static void registerColumnEncryptionKeyStoreProviders(
         loggerExternal.entering(loggingClassNameBase, "registerColumnEncryptionKeyStoreProviders",
                 "Registering Column Encryption Key Store Providers");
 
-        LOCK.lock();
+        sLock.lock();
         try {
             if (null == clientKeyStoreProviders) {
                 throw new SQLServerException(null, SQLServerException.getErrString("R_CustomKeyStoreProviderMapNull"),
@@ -988,7 +989,7 @@ public static void registerColumnEncryptionKeyStoreProviders(
                 globalCustomColumnEncryptionKeyStoreProviders.put(providerName, provider);
             }
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
 
         loggerExternal.exiting(loggingClassNameBase, "registerColumnEncryptionKeyStoreProviders",
@@ -1004,14 +1005,14 @@ public static void unregisterColumnEncryptionKeyStoreProviders() {
         loggerExternal.entering(loggingClassNameBase, "unregisterColumnEncryptionKeyStoreProviders",
                 "Removing Column Encryption Key Store Provider");
 
-        LOCK.lock();
+        sLock.lock();
         try {
             if (null != globalCustomColumnEncryptionKeyStoreProviders) {
                 globalCustomColumnEncryptionKeyStoreProviders.clear();
                 globalCustomColumnEncryptionKeyStoreProviders = null;
             }
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
 
         loggerExternal.exiting(loggingClassNameBase, "unregisterColumnEncryptionKeyStoreProviders",
@@ -1233,15 +1234,15 @@ public static void setColumnEncryptionTrustedMasterKeyPaths(Map<String, List<Str
         loggerExternal.entering(loggingClassNameBase, "setColumnEncryptionTrustedMasterKeyPaths",
                 "Setting Trusted Master Key Paths");
 
-        LOCK.lock();
+        sLock.lock();
         try {
             // Use upper case for server and instance names.
             columnEncryptionTrustedMasterKeyPaths.clear();
             for (Map.Entry<String, List<String>> entry : trustedKeyPaths.entrySet()) {
                 columnEncryptionTrustedMasterKeyPaths.put(entry.getKey().toUpperCase(), entry.getValue());
             }
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
 
         loggerExternal.exiting(loggingClassNameBase, "setColumnEncryptionTrustedMasterKeyPaths",
@@ -1260,12 +1261,12 @@ public static void updateColumnEncryptionTrustedMasterKeyPaths(String server, Li
         loggerExternal.entering(loggingClassNameBase, "updateColumnEncryptionTrustedMasterKeyPaths",
                 "Updating Trusted Master Key Paths");
 
-        LOCK.lock();
+        sLock.lock();
         try {
             // Use upper case for server and instance names.
             columnEncryptionTrustedMasterKeyPaths.put(server.toUpperCase(), trustedKeyPaths);
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
 
         loggerExternal.exiting(loggingClassNameBase, "updateColumnEncryptionTrustedMasterKeyPaths",
@@ -1282,12 +1283,12 @@ public static void removeColumnEncryptionTrustedMasterKeyPaths(String server) {
         loggerExternal.entering(loggingClassNameBase, "removeColumnEncryptionTrustedMasterKeyPaths",
                 "Removing Trusted Master Key Paths");
 
-        LOCK.lock();
+        sLock.lock();
         try {
             // Use upper case for server and instance names.
             columnEncryptionTrustedMasterKeyPaths.remove(server.toUpperCase());
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
 
         loggerExternal.exiting(loggingClassNameBase, "removeColumnEncryptionTrustedMasterKeyPaths",
@@ -1303,7 +1304,7 @@ public static Map<String, List<String>> getColumnEncryptionTrustedMasterKeyPaths
         loggerExternal.entering(loggingClassNameBase, "getColumnEncryptionTrustedMasterKeyPaths",
                 "Getting Trusted Master Key Paths");
 
-        LOCK.lock();
+        sLock.lock();
         try {
             Map<String, List<String>> masterKeyPathCopy = new HashMap<>();
 
@@ -1316,12 +1317,12 @@ public static Map<String, List<String>> getColumnEncryptionTrustedMasterKeyPaths
 
             return masterKeyPathCopy;
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
     }
 
     static List<String> getColumnEncryptionTrustedMasterKeyPaths(String server, Boolean[] hasEntry) {
-        LOCK.lock();
+        sLock.lock();
         try {
             if (columnEncryptionTrustedMasterKeyPaths.containsKey(server)) {
                 hasEntry[0] = true;
@@ -1331,7 +1332,7 @@ static List<String> getColumnEncryptionTrustedMasterKeyPaths(String server, Bool
                 return null;
             }
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
     }
 
@@ -1340,11 +1341,11 @@ static List<String> getColumnEncryptionTrustedMasterKeyPaths(String server, Bool
      * request to acquire an access token.
      */
     public static void clearUserTokenCache() {
-        LOCK.lock();
+        sLock.lock();
         try {
             PersistentTokenCacheAccessAspect.clearUserTokenCache();
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
     }
 
@@ -7478,7 +7479,7 @@ void doSecurityCheck() {
      */
     public static void setColumnEncryptionKeyCacheTtl(int columnEncryptionKeyCacheTTL,
             TimeUnit unit) throws SQLServerException {
-        LOCK.lock();
+        sLock.lock();
         try {
             if (columnEncryptionKeyCacheTTL < 0 || unit.equals(TimeUnit.MILLISECONDS)
                     || unit.equals(TimeUnit.MICROSECONDS) || unit.equals(TimeUnit.NANOSECONDS)) {
@@ -7488,16 +7489,16 @@ public static void setColumnEncryptionKeyCacheTtl(int columnEncryptionKeyCacheTT
 
             columnEncryptionKeyCacheTtl = TimeUnit.SECONDS.convert(columnEncryptionKeyCacheTTL, unit);
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
     }
 
     static long getColumnEncryptionKeyCacheTtl() {
-        LOCK.lock();
+        sLock.lock();
         try {
             return columnEncryptionKeyCacheTtl;
         } finally {
-            LOCK.unlock();
+            sLock.unlock();
         }
     }