Merge pull request #3 from V1ManagedServices/p/MXDR-3374_fix_DefaultAzureCredential

[MXDR-3374] fix default azure credential

Author: V1ManagedServices

Solutions/Trend Micro Vision One/Data Connectors/AzureFunctionTrendMicroXDR.zip

@@ -5,7 +5,7 @@
 
 from shared_code.models.oat import RiskLevel
 
-VERSION = '1.2.5'
+VERSION = '1.2.6'
 SIEM_NAME = 'SentinelAddon'
 XDR_HOSTS = {
     'us': 'https://api.xdr.trendmicro.com',
@@ -14,8 +14,8 @@
     'jp': 'https://api.xdr.trendmicro.co.jp',
     'sg': 'https://api.sg.xdr.trendmicro.com',
     'au': 'https://api.au.xdr.trendmicro.com',
-    'uae': 'https://api.uae.xdr.trendmicro.com/',
     'mea': 'https://api.mea.xdr.trendmicro.com',
+    'uk': 'https://api.uk.xdr.trendmicro.com',
 }
 
 
@@ -35,13 +35,14 @@ def get_api_tokens():
     )
     if is_key_vault_enabled:
         # get tokens from key vault
-        from azure.identity import DefaultAzureCredential
+        from azure.identity import ManagedIdentityCredential
         from azure.keyvault.secrets import SecretClient
 
         clp_ids = set(filter(None, os.getenv('clpIds').split(',')))
-        credential = DefaultAzureCredential(
-            managed_identity_client_id=os.getenv('keyVaultIdentityClientId')
-        )
+
+        managed_identity_client_id = os.getenv('keyVaultIdentityClientId')
+        credential = ManagedIdentityCredential(client_id=managed_identity_client_id)
+
         client = SecretClient(vault_url=os.getenv('keyVaultUrl'), credential=credential)
 
         tokens = []

Solutions/Trend Micro Vision One/Data Connectors/AzureFunctionTrendMicroXDR/shared_code/configurations.py

@@ -1,4 +1,4 @@
-# ver: 1.2.5
+# ver: 1.2.6
 import typing
 import json
 

Solutions/Trend Micro Vision One/Data Connectors/AzureFunctionTrendMicroXDR/timer_trigger/__init__.py

@@ -1,4 +1,4 @@
-# ver: 1.2.5
+# ver: 1.2.6
 import uuid
 from datetime import datetime, timedelta, timezone
 from typing import List, Optional, Tuple

Solutions/Trend Micro Vision One/Data Connectors/AzureFunctionTrendMicroXDR/timer_trigger_oat/__init__.py

@@ -27,8 +27,8 @@
                 "sg",
                 "au",
                 "in",
-                "uae",
-                "mea"
+                "mea",
+                "uk"
             ],
             "defaultValue": "us"
         },