Relocate Windows ProcessNeedsProfileDir to install_static

chrome/install_static/install_util.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/install_static/install_util.h"
 
 #include <windows.h>
 #include <assert.h>
 #include <stdlib.h>
 #include <string.h>
 
 #include <algorithm>
 #include <iterator>
 #include <limits>
 #include <memory>
 #include <sstream>
 
 #include "chrome/install_static/install_details.h"
 #include "chrome/install_static/install_modes.h"
 #include "chrome/install_static/policy_path_parser.h"
 #include "chrome/install_static/user_data_dir.h"
 #include "chrome_elf/nt_registry/nt_registry.h"
 #include "components/version_info/channel.h"
 
 namespace install_static {
 
+enum class ProcessType {
+  UNINITIALIZED,
+  OTHER_PROCESS,
+  BROWSER_PROCESS,
+  CLOUD_PRINT_SERVICE_PROCESS,
+#if !defined(DISABLE_NACL)
+  NACL_BROKER_PROCESS,
+  NACL_LOADER_PROCESS,
+#endif
+};
+
+// Caches the |ProcessType| of the current process.
 ProcessType g_process_type = ProcessType::UNINITIALIZED;
 
 const wchar_t kRegValueChromeStatsSample[] = L"UsageStatsInSample";
 
 // TODO(ananta)
 // http://crbug.com/604923
 // The constants defined in this file are also defined in chrome/installer and
 // other places. we need to unify them.
 const wchar_t kHeadless[] = L"CHROME_HEADLESS";
 const wchar_t kShowRestart[] = L"CHROME_CRASHED";
@@ skipping 20 matching lines @@
 
 // TODO(ananta)
 // http://crbug.com/604923
 // These constants are defined in the chrome/installer directory as well. We
 // need to unify them.
 constexpr wchar_t kRegValueAp[] = L"ap";
 constexpr wchar_t kRegValueName[] = L"name";
 constexpr wchar_t kRegValueUsageStats[] = L"usagestats";
 constexpr wchar_t kMetricsReportingEnabled[] = L"MetricsReportingEnabled";
 
+constexpr wchar_t kCloudPrintServiceProcess[] = L"cloud-print-service";
+#if !defined(DISABLE_NACL)
+constexpr wchar_t kNaClBrokerProcess[] = L"nacl-broker";
+constexpr wchar_t kNaClLoaderProcess[] = L"nacl-loader";
+#endif
+
 void Trace(const wchar_t* format_string, ...) {
   static const int kMaxLogBufferSize = 1024;
   static wchar_t buffer[kMaxLogBufferSize] = {};
 
   va_list args = {};
 
   va_start(args, format_string);
   vswprintf(buffer, kMaxLogBufferSize, format_string, args);
   OutputDebugStringW(buffer);
   va_end(args);
@@ skipping 204 matching lines @@
   }
   if (MatchPattern(value, kChromeChannelBetaPattern) ||
       MatchPattern(value, kChromeChannelBetaX64Pattern)) {
     return kChromeChannelBeta;
   }
   // Else report values with garbage as stable since they will match the stable
   // rules in the update configs.
   return std::wstring();
 }
 
+// Converts a process type specified as a string to the ProcessType enum.
+ProcessType GetProcessType(const std::wstring& process_type) {
+  if (process_type.empty())
+    return ProcessType::BROWSER_PROCESS;
+  if (process_type == kCloudPrintServiceProcess)
+    return ProcessType::CLOUD_PRINT_SERVICE_PROCESS;
+#if !defined(DISABLE_NACL)
+  if (process_type == kNaClBrokerProcess)
+    return ProcessType::NACL_BROKER_PROCESS;
+  if (process_type == kNaClLoaderProcess)
+    return ProcessType::NACL_LOADER_PROCESS;
+#endif
+  return ProcessType::OTHER_PROCESS;
+}
+
+// Returns whether |process_type| needs the profile directory.
+bool ProcessNeedsProfileDir(ProcessType process_type) {
+  // On Windows we don't want subprocesses other than the browser process and
+  // service processes to be able to use the profile directory because if it
+  // lies on a network share the sandbox will prevent us from accessing it.
+  switch (process_type) {
+    case ProcessType::BROWSER_PROCESS:
+    case ProcessType::CLOUD_PRINT_SERVICE_PROCESS:
+#if !defined(DISABLE_NACL)
+    case ProcessType::NACL_BROKER_PROCESS:
+    case ProcessType::NACL_LOADER_PROCESS:
+#endif
+      return true;
+    case ProcessType::OTHER_PROCESS:
+      return false;
+    case ProcessType::UNINITIALIZED:
+      assert(false);
+      return false;
+  }
+  assert(false);
+  return false;
+}
+
 }  // namespace
 
 bool IsSystemInstall() {
   return InstallDetails::Get().system_level();
 }
 
 std::wstring GetChromeInstallSubDirectory() {
   std::wstring result;
   AppendChromeInstallSubDirectory(InstallDetails::Get().mode(),
                                   true /* include_suffix */, &result);
@@ skipping 146 matching lines @@
                              kMetricsReportingEnabled, &value)) {
     *crash_reporting_enabled = (value != 0);
     return true;
   }
 
   return false;
 }
 
 void InitializeProcessType() {
   assert(g_process_type == ProcessType::UNINITIALIZED);
-  typedef bool (*IsSandboxedProcessFunc)();
-  IsSandboxedProcessFunc is_sandboxed_process_func =
-      reinterpret_cast<IsSandboxedProcessFunc>(
-          ::GetProcAddress(::GetModuleHandle(nullptr), "IsSandboxedProcess"));
-  if (is_sandboxed_process_func && is_sandboxed_process_func()) {
-    g_process_type = ProcessType::NON_BROWSER_PROCESS;
-    return;
-  }
+  std::wstring process_type =
+      GetSwitchValueFromCommandLine(::GetCommandLine(), kProcessType);
+  g_process_type = GetProcessType(process_type);
+}
 
-  // TODO(robertshield): Drop the command line check when we drop support for
-  // enabling chrome_elf in unsandboxed processes.
-  const wchar_t* command_line = GetCommandLine();
-  if (command_line && ::wcsstr(command_line, L"--type")) {
-    g_process_type = ProcessType::NON_BROWSER_PROCESS;
-    return;
-  }
-
-  g_process_type = ProcessType::BROWSER_PROCESS;
+bool IsProcessTypeInitialized() {
+  return g_process_type != ProcessType::UNINITIALIZED;
 }
 
 bool IsNonBrowserProcess() {
   assert(g_process_type != ProcessType::UNINITIALIZED);
-  return g_process_type == ProcessType::NON_BROWSER_PROCESS;
+  return g_process_type != ProcessType::BROWSER_PROCESS;
+}
+
+bool ProcessNeedsProfileDir(const std::string& process_type) {
+  return ProcessNeedsProfileDir(GetProcessType(UTF8ToUTF16(process_type)));
 }
 
 std::wstring GetCrashDumpLocation() {
   // In order to be able to start crash handling very early and in chrome_elf,
   // we cannot rely on chrome's PathService entries (for DIR_CRASH_DUMPS) being
   // available on Windows. See https://crbug.com/564398.
   std::wstring user_data_dir;
   bool ret = GetUserDataDirectory(&user_data_dir, nullptr);
   assert(ret);
   IgnoreUnused(ret);
@@ skipping 348 matching lines @@
     case ChannelStrategy::ADDITIONAL_PARAMETERS:
       return ChannelFromAdditionalParameters(mode, ap_value);
     case ChannelStrategy::FIXED:
       return mode.default_channel_name;
   }
 
   return std::wstring();
 }
 
 }  // namespace install_static