From 8f7f89e757f0ac8f7a2da54d2001bb2ff88269cc Mon Sep 17 00:00:00 2001
From: Fup Duck <fupduck@sacknagel.com>
Date: Thu, 8 Feb 2018 14:34:20 +0100
Subject: 8827 - fix tests

---
 .../se/leap/bitmaskclient/ProviderApiManager.java  | 74 ++++++++++------------
 1 file changed, 35 insertions(+), 39 deletions(-)

(limited to 'app/src/production/java')

diff --git a/app/src/production/java/se/leap/bitmaskclient/ProviderApiManager.java b/app/src/production/java/se/leap/bitmaskclient/ProviderApiManager.java
index ac58d005..a4b3e491 100644
--- a/app/src/production/java/se/leap/bitmaskclient/ProviderApiManager.java
+++ b/app/src/production/java/se/leap/bitmaskclient/ProviderApiManager.java
@@ -70,44 +70,35 @@ public class ProviderApiManager extends ProviderApiManagerBase {
         int progress = 0;
         Bundle currentDownload = new Bundle();
 
-        if (task != null) {
-            String mainUrlString = provider.getMainUrlString();
-            if (isEmpty(mainUrlString)) {
-                currentDownload.putBoolean(BROADCAST_RESULT_KEY, false);
-                setErrorResult(currentDownload, malformed_url, null);
-                return currentDownload;
-            }
-
-            getPersistedProviderUpdates(provider);
-            currentDownload = validateProviderDetails(provider);
+        if (isEmpty(provider.getMainUrlString()) || provider.getMainUrl().isDefault()) {
+            currentDownload.putBoolean(BROADCAST_RESULT_KEY, false);
+            setErrorResult(currentDownload, malformed_url, null);
+            return currentDownload;
+        }
 
-            //provider details invalid
-            if (currentDownload.containsKey(ERRORS)) {
-                return currentDownload;
-            }
+        getPersistedProviderUpdates(provider);
+        currentDownload = validateProviderDetails(provider);
 
-            //no provider certificate available
-            if (currentDownload.containsKey(BROADCAST_RESULT_KEY) && !currentDownload.getBoolean(BROADCAST_RESULT_KEY)) {
-                resetProviderDetails(provider);
-            }
+        //provider details invalid
+        if (currentDownload.containsKey(ERRORS)) {
+            return currentDownload;
+        }
 
-            go_ahead = true;
+        //no provider certificate available
+        if (currentDownload.containsKey(BROADCAST_RESULT_KEY) && !currentDownload.getBoolean(BROADCAST_RESULT_KEY)) {
+            resetProviderDetails(provider);
         }
 
+        go_ahead = true;
+
         if (!provider.hasDefinition()) {
             currentDownload = getAndSetProviderJson(provider);
         }
         if (provider.hasDefinition() || (currentDownload.containsKey(BROADCAST_RESULT_KEY) && currentDownload.getBoolean(BROADCAST_RESULT_KEY))) {
-            broadcastProgress(++progress);
-
             if (!provider.hasCaCert())
                 currentDownload = downloadCACert(provider);
             if (provider.hasCaCert() || (currentDownload.containsKey(BROADCAST_RESULT_KEY) && currentDownload.getBoolean(BROADCAST_RESULT_KEY))) {
-                broadcastProgress(++progress);
                 currentDownload = getAndSetEipServiceJson(provider);
-                if (currentDownload.containsKey(BROADCAST_RESULT_KEY) && currentDownload.getBoolean(BROADCAST_RESULT_KEY)) {
-                    broadcastProgress(++progress);
-                }
             }
         }
 
@@ -123,9 +114,10 @@ public class ProviderApiManager extends ProviderApiManagerBase {
 
         if (go_ahead) {
             String providerDotJsonString;
-            if(providerDefinition.length() == 0 || caCert.isEmpty())
-                providerDotJsonString = downloadWithCommercialCA(provider);
-            else {
+            if(providerDefinition.length() == 0 || caCert.isEmpty()) {
+                String providerJsonUrl = provider.getMainUrlString() + "/provider.json";
+                providerDotJsonString = downloadWithCommercialCA(providerJsonUrl, provider);
+            } else {
                 providerDotJsonString = downloadFromApiUrlWithProviderCA("/provider.json", caCert, providerDefinition);
             }
 
@@ -207,15 +199,20 @@ public class ProviderApiManager extends ProviderApiManagerBase {
 
     private Bundle downloadCACert(Provider provider) {
         Bundle result = new Bundle();
-        String providerDomain = getDomainFromMainURL(provider.getMainUrlString());
-        String certString = downloadWithCommercialCA(provider);
-
-        if (validCertificate(provider, certString) && go_ahead) {
-            provider.setCaCert(certString);
-            preferences.edit().putString(Provider.CA_CERT + "." + providerDomain, certString).apply();
-            result.putBoolean(BROADCAST_RESULT_KEY, true);
-        } else {
-            setErrorResult(result, warning_corrupted_provider_cert, ERROR_CERTIFICATE_PINNING.toString());
+        try {
+            String caCertUrl = provider.getDefinition().getString(Provider.CA_CERT_URI);
+            String providerDomain = getDomainFromMainURL(provider.getMainUrlString());
+            String certString = downloadWithCommercialCA(caCertUrl, provider);
+
+            if (validCertificate(provider, certString) && go_ahead) {
+                provider.setCaCert(certString);
+                preferences.edit().putString(Provider.CA_CERT + "." + providerDomain, certString).apply();
+                result.putBoolean(BROADCAST_RESULT_KEY, true);
+            } else {
+                setErrorResult(result, warning_corrupted_provider_cert, ERROR_CERTIFICATE_PINNING.toString());
+            }
+        } catch (JSONException e) {
+            e.printStackTrace();
         }
 
         return result;
@@ -225,8 +222,7 @@ public class ProviderApiManager extends ProviderApiManagerBase {
      * Tries to download the contents of the provided url using commercially validated CA certificate from chosen provider.
      *
      */
-    private String downloadWithCommercialCA(Provider provider) {
-        String stringUrl = provider.getMainUrlString() + "/provider.json";
+    private String downloadWithCommercialCA(String stringUrl, Provider provider) {
         String responseString;
         JSONObject errorJson = new JSONObject();
 
-- 
cgit v1.2.3