components/omnibox/browser/location_bar_model_util.cc

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

#include "components/omnibox/browser/location_bar_model_util.h"

#include "base/feature_list.h"
#include "base/notreached.h"
#include "build/build_config.h"
#include "components/omnibox/browser/buildflags.h"
#include "components/omnibox/common/omnibox_features.h"
#include "components/safe_browsing/core/common/features.h"
#include "components/security_state/core/security_state.h"
#include "ui/base/ui_base_features.h"
#include "ui/gfx/vector_icon_types.h"

#if (!BUILDFLAG(IS_ANDROID) || BUILDFLAG(ENABLE_VR)) && !BUILDFLAG(IS_IOS)
#include "components/omnibox/browser/vector_icons.h"  // nogncheck
#include "components/vector_icons/vector_icons.h"     // nogncheck
#endif

namespace location_bar_model {

const gfx::VectorIcon& GetSecurityVectorIcon(
    security_state::SecurityLevel security_level,
    bool use_updated_connection_security_indicators,
    security_state::MaliciousContentStatus malicious_content_status) {
#if (!BUILDFLAG(IS_ANDROID) || BUILDFLAG(ENABLE_VR)) && !BUILDFLAG(IS_IOS)
  switch (security_level) {
    case security_state::NONE:
      return IsChromeRefreshIconsEnabled() ? omnibox::kHttpChromeRefreshIcon
                                           : omnibox::kHttpIcon;
    case security_state::SECURE:
      return IsChromeRefreshIconsEnabled()
                 ? omnibox::kSecurePageInfoChromeRefreshIcon
                 : vector_icons::kHttpsValidIcon;
    case security_state::SECURE_WITH_POLICY_INSTALLED_CERT:
      return IsChromeRefreshIconsEnabled()
                 ? vector_icons::kBusinessChromeRefreshIcon
                 : vector_icons::kBusinessIcon;
    case security_state::WARNING:
      return IsChromeRefreshIconsEnabled()
                 ? vector_icons::kNotSecureWarningChromeRefreshIcon
                 : vector_icons::kNotSecureWarningIcon;
    case security_state::DANGEROUS:
      if (malicious_content_status ==
              security_state::MALICIOUS_CONTENT_STATUS_MANAGED_POLICY_WARN ||
          malicious_content_status ==
              security_state::MALICIOUS_CONTENT_STATUS_MANAGED_POLICY_BLOCK) {
        return IsChromeRefreshIconsEnabled()
                   ? vector_icons::kBusinessChromeRefreshIcon
                   : vector_icons::kBusinessIcon;
      }
      if (malicious_content_status !=
          security_state::MALICIOUS_CONTENT_STATUS_BILLING) {
        return IsChromeRefreshIconsEnabled()
                   ? vector_icons::kDangerousChromeRefreshIcon
                   : vector_icons::kDangerousIcon;
      }
      return IsChromeRefreshIconsEnabled()
                 ? vector_icons::kNotSecureWarningChromeRefreshIcon
                 : vector_icons::kNotSecureWarningIcon;

    case security_state::SECURITY_LEVEL_COUNT:
      NOTREACHED();
      return IsChromeRefreshIconsEnabled() ? omnibox::kHttpChromeRefreshIcon
                                           : omnibox::kHttpIcon;
  }
  NOTREACHED();
  return IsChromeRefreshIconsEnabled() ? omnibox::kHttpChromeRefreshIcon
                                       : omnibox::kHttpIcon;
#else
  NOTREACHED();
  static const gfx::VectorIcon dummy = {};
  return dummy;
#endif
}

bool IsChromeRefreshIconsEnabled() {
  return omnibox::IsOmniboxCr23CustomizeGuardedFeatureEnabled(
      omnibox::kOmniboxCR23SteadyStateIcons);
}

}  // namespace location_bar_model