Bug 1525119: Check if this is the default install and if so lock the profile. r=froydnj

Previously we attempted to do this when XPCOM wasn't available so it always
failed to get the shell service. Instead set a flag telling us to do it later
when we choose the old default profile.

Also includes a block of code to attempt to fix the issue for existing nightly
users, this can be removed before betas.

Differential Revision: https://phabricator.services.mozilla.com/D18596

UltraBlame original commit: c132ef1b7c8ab954be0049c95f8b42039029c3e1

Author: marco-c

toolkit/profile/nsToolkitProfileService.cpp

@@ -304,7 +304,8 @@ nsToolkitProfileService::nsToolkitProfileService()
       mUseDedicatedProfile(false),
 #endif
       mCreatedAlternateProfile(false),
-      mStartupReason(NS_LITERAL_STRING("unknown")) {
+      mStartupReason(NS_LITERAL_STRING("unknown")),
+      mMaybeLockProfile(false) {
 #ifdef MOZ_DEV_EDITION
   mUseDevEditionProfile = true;
 #endif
@@ -313,13 +314,30 @@ nsToolkitProfileService::nsToolkitProfileService()
 
 nsToolkitProfileService::~nsToolkitProfileService() { gService = nullptr; }
 
-void nsToolkitProfileService::RecordStartupTelemetry() {
+void nsToolkitProfileService::CompleteStartup() {
   if (!mStartupProfileSelected) {
     return;
   }
 
   ScalarSet(mozilla::Telemetry::ScalarID::STARTUP_PROFILE_SELECTION_REASON,
             mStartupReason);
+
+  if (mMaybeLockProfile) {
+    nsCOMPtr<nsIToolkitShellService> shell =
+        do_GetService(NS_TOOLKITSHELLSERVICE_CONTRACTID);
+    if (!shell) {
+      return;
+    }
+
+    bool isDefaultApp;
+    nsresult rv = shell->IsDefaultApplication(&isDefaultApp);
+    NS_ENSURE_SUCCESS_VOID(rv);
+
+    if (isDefaultApp) {
+      mInstallData.SetString(mInstallHash.get(), "Locked", "1");
+      Flush();
+    }
+  }
 }
 
 
@@ -457,29 +475,18 @@ bool nsToolkitProfileService::MaybeMakeDefaultDedicatedProfile(
 
   SetDefaultProfile(aProfile);
 
-  bool isDefaultApp = false;
-
-  nsCOMPtr<nsIToolkitShellService> shell =
-      do_GetService(NS_TOOLKITSHELLSERVICE_CONTRACTID);
-  if (shell) {
-    rv = shell->IsDefaultApplication(&isDefaultApp);
-    
-    
-    if (NS_FAILED(rv)) {
-      isDefaultApp = false;
-    }
-  }
-
-  if (!isDefaultApp) {
-    
-    
-    
-    mInstallData.DeleteString(mInstallHash.get(), "Locked");
-  }
+  
+  
+  
+  mInstallData.DeleteString(mInstallHash.get(), "Locked");
 
 
   Flush();
 
+  
+  
+  mMaybeLockProfile = true;
+
   return true;
 }
 
@@ -818,7 +825,7 @@ nsToolkitProfileService::SelectStartupProfile(
 
 
   if (NS_SUCCEEDED(rv)) {
-    RecordStartupTelemetry();
+    CompleteStartup();
   }
 
   return rv;
@@ -1120,6 +1127,18 @@ nsresult nsToolkitProfileService::SelectStartupProfile(
   mStartupReason = NS_LITERAL_STRING("default");
 
 
+  if (mUseDedicatedProfile) {
+    nsCString locked;
+    rv = mInstallData.GetString(mInstallHash.get(), "Locked", locked);
+    if (NS_FAILED(rv) || !locked.EqualsLiteral("1")) {
+      
+      
+      
+      mMaybeLockProfile = true;
+    }
+  }
+
+  
   mCurrent->GetRootDir(aRootDir);
   mCurrent->GetLocalDir(aLocalDir);
   NS_ADDREF(*aProfile = mCurrent);

toolkit/profile/nsToolkitProfileService.h

@@ -79,7 +79,7 @@ class nsToolkitProfileService final : public nsIToolkitProfileService {
                                 nsIToolkitProfile** aProfile, bool* aDidCreate);
   nsresult CreateResetProfile(nsIToolkitProfile** aNewProfile);
   nsresult ApplyResetProfile(nsIToolkitProfile* aOldProfile);
-  void RecordStartupTelemetry();
+  void CompleteStartup();
 
  private:
   friend class nsToolkitProfile;
@@ -143,6 +143,7 @@ class nsToolkitProfileService final : public nsIToolkitProfileService {
 
   bool mCreatedAlternateProfile;
   nsString mStartupReason;
+  bool mMaybeLockProfile;
 
   static nsToolkitProfileService* gService;
 

toolkit/xre/nsAppRunner.cpp

@@ -4695,8 +4695,7 @@ nsresult XREMain::XRE_mainRun() {
   AddSandboxAnnotations();
 #endif 
 
-  static_cast<nsToolkitProfileService*>(mProfileSvc.get())
-      ->RecordStartupTelemetry();
+  static_cast<nsToolkitProfileService*>(mProfileSvc.get())->CompleteStartup();
 
   {
     rv = appStartup->Run();