add msi detecting

Azure · Nov 2, 2018 · 717ff62 · 717ff62
1 parent 421d0a3
commit 717ff62
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/msrestazure/azure_local_creds_prober.py b/msrestazure/azure_local_creds_prober.py
@@ -85,6 +85,13 @@ def probe(self):
         if not self.enabled:
             return None
         try:
+            # detect we are inside a VM with identity
+            _ = requests.get('http://169.254.169.254/metadata/identity/oauth2/token',
+                                  params={'api-version': '2018-02-01',
+                                          'resource':'https://management.core.windows.net/'},
+                                  headers={'Metadata': 'true'},
+                                  timeout=1)
+
             creds = MSIAuthentication()
              # MSI is not yet supported in sovereigns
             setattr(creds, 'cloud_environment', AZURE_PUBLIC_CLOUD)