chrome/browser/policy/chrome_extension_policy_migrator.cc

// Copyright 2018 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/browser/policy/chrome_extension_policy_migrator.h"

#include "base/logging.h"
#include "base/strings/utf_string_conversions.h"
#include "components/strings/grit/components_strings.h"
#include "extensions/common/hashed_extension_id.h"
#include "ui/base/l10n/l10n_util.h"

namespace policy {

void ChromeExtensionPolicyMigrator::CopyPoliciesIfUnset(
    PolicyBundle* bundle,
    const std::string& hashed_extension_id,
    base::span<const Migration> migrations) {
  // HashedExtensionId gives an all-uppercase output, so make sure the input is
  // all uppercase.
  std::string hashed_extension_id_uppercase = hashed_extension_id;
  std::transform(hashed_extension_id_uppercase.begin(),
                 hashed_extension_id_uppercase.end(),
                 hashed_extension_id_uppercase.begin(), ::toupper);

  // Look for an extension with this hash.
  PolicyMap* extension_map = nullptr;
  for (auto& policy : *bundle) {
    const PolicyNamespace& policy_namespace = policy.first;

    if (policy_namespace.domain == PolicyDomain::POLICY_DOMAIN_EXTENSIONS &&
        extensions::HashedExtensionId(policy_namespace.component_id).value() ==
            hashed_extension_id_uppercase) {
      extension_map = policy.second.get();
      break;
    }
  }

  if (extension_map == nullptr || extension_map->empty()) {
    VLOG(3) << "Extension with hashed id '" << hashed_extension_id_uppercase
            << "' not installed. Aborting policy migration.";
    return;
  }

  PolicyMap& chrome_map = bundle->Get(PolicyNamespace(
      PolicyDomain::POLICY_DOMAIN_CHROME, /* component_id */ std::string()));

  for (const auto& migration : migrations) {
    PolicyMap::Entry* entry = extension_map->GetMutable(migration.old_name);
    if (entry) {
      if (!chrome_map.Get(migration.new_name)) {
        VLOG(3) << "Extension policy is configured: '" << migration.old_name
                << "'. Copied to '" << migration.new_name << "'.";
        auto new_entry = entry->DeepCopy();
        migration.transform.Run(new_entry.value());
        new_entry.AddError(
            l10n_util::GetStringFUTF8(IDS_POLICY_MIGRATED_NEW_POLICY,
                                      base::UTF8ToUTF16(migration.old_name)));
        chrome_map.Set(migration.new_name, std::move(new_entry));
      } else {
        VLOG(3) << "Extension policy is configured, but conflicts: '"
                << migration.old_name << "' (Chrome policy '"
                << migration.new_name << "' is also set). Skipped.";
      }
      entry->AddError(
          l10n_util::GetStringFUTF8(IDS_POLICY_MIGRATED_OLD_POLICY,
                                    base::UTF8ToUTF16(migration.new_name)));
    } else {
      VLOG(3) << "Extension policy is not configured: '" << migration.old_name
              << "'. Skipped.";
    }
  }
}

}  // namespace policy