setup_install_details.cc | Explore in Territory

// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif

#include "chrome/installer/setup/setup_install_details.h"

#include <optional>
#include <string>
#include <string_view>

#include "base/command_line.h"
#include "base/win/registry.h"
#include "chrome/install_static/install_constants.h"
#include "chrome/install_static/install_details.h"
#include "chrome/install_static/install_modes.h"
#include "chrome/install_static/install_util.h"
#include "chrome/installer/util/google_update_constants.h"
#include "chrome/installer/util/initial_preferences.h"
#include "chrome/installer/util/initial_preferences_constants.h"
#include "chrome/installer/util/util_constants.h"

namespace {

const install_static::InstallConstants* FindInstallMode(
    const base::CommandLine& command_line) {
  // Search for a mode whose switch is on the command line.
  for (int i = 1; i < install_static::NUM_INSTALL_MODES; ++i) {
    const install_static::InstallConstants& mode =
        install_static::kInstallModes[i];
    if (command_line.HasSwitch(mode.install_switch))
      return &mode;
  }
  // The first mode is always the default if all else fails.
  return &install_static::kInstallModes[0];
}

// Returns the value of `switch_name` from `command_line` if it is present, or
// nullopt otherwise.
std::optional<std::wstring> GetSwitchValue(
    const base::CommandLine& command_line,
    std::string_view switch_name) {
  std::optional<std::wstring> result;
  if (command_line.HasSwitch(switch_name))
    result = command_line.GetSwitchValueNative(switch_name);
  return result;
}

}  // namespace

void InitializeInstallDetails(
    const base::CommandLine& command_line,
    const installer::InitialPreferences& initial_preferences) {
  install_static::InstallDetails::SetForProcess(
      MakeInstallDetails(command_line, initial_preferences));
}

std::unique_ptr<install_static::PrimaryInstallDetails> MakeInstallDetails(
    const base::CommandLine& command_line,
    const installer::InitialPreferences& initial_preferences) {
  std::unique_ptr<install_static::PrimaryInstallDetails> details(
      std::make_unique<install_static::PrimaryInstallDetails>());

  // The mode is determined by brand-specific command line switches.
  const install_static::InstallConstants* const mode =
      FindInstallMode(command_line);
  details->set_mode(mode);

  // The install level may be set by any of:
  // - distribution.system_level=true in initial_preferences,
  // - --system-level on the command line, or
  // - the GoogleUpdateIsMachine=1 environment variable.
  // In all three cases the value is sussed out in InitialPreferences
  // initialization.
  bool system_level = false;
  initial_preferences.GetBool(installer::initial_preferences::kSystemLevel,
                              &system_level);
  details->set_system_level(system_level);

  // The channel is determined based on the brand and the mode's
  // ChannelStrategy. For brands that do not support Google Update, the channel
  // is an empty string. For modes using the FIXED strategy, the channel is the
  // default_channel_name in the mode. For modes using the FLOATING strategy,
  // the channel is dictated by the --channel switch on the command line or the
  // mode's default if none is provided. An override on the command line will be
  // written to the "ap" value for the sake of subsequent update checks.

  // Cache the ap and cohort name values found in the registry for use in crash
  // keys.
  std::wstring update_ap;
  std::wstring update_cohort_name;

  std::optional<std::wstring> channel_from_cmd_line =
      GetSwitchValue(command_line, installer::switches::kChannel);

  auto channel = install_static::DetermineChannel(
      *mode, system_level,
      channel_from_cmd_line ? channel_from_cmd_line->c_str() : nullptr,
      &update_ap, &update_cohort_name);
  details->set_channel(channel.channel_name);
  details->set_channel_origin(channel.origin);
  if (channel.origin == install_static::ChannelOrigin::kPolicy)
    details->set_channel_override(*channel_from_cmd_line);
  details->set_is_extended_stable_channel(channel.is_extended_stable);
  details->set_update_ap(update_ap);
  details->set_update_cohort_name(update_cohort_name);

  return details;
}
chromium/chrome/installer/setup/setup_install_details.cc
