// Copyright 2015 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/password_manager/content/browser/bad_message.h"

#include "base/logging.h"
#include "base/metrics/histogram_functions.h"
#include "base/syslog_logging.h"
#include "components/password_manager/core/browser/password_form.h"
#include "content/public/browser/child_process_security_policy.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_process_host.h"

namespace password_manager::bad_message {
namespace {

// Called when the browser receives a bad IPC message from a renderer process on
// the UI thread. Logs the event, records a histogram metric for the |reason|,
// and terminates the process for |host|.
void ReceivedBadMessage(content::RenderProcessHost* host,
                        BadMessageReason reason) {
  LOG(ERROR)
      << "Terminating renderer for bad PasswordManager IPC message, reason "
      << static_cast<int>(reason);
  base::UmaHistogramSparse("Stability.BadMessageTerminated.PasswordManager",
                           static_cast<int>(reason));
  host->ShutdownForBadMessage(
      content::RenderProcessHost::CrashReportMode::GENERATE_CRASH_DUMP);
}

}  // namespace

bool CheckForIllegalURL(content::RenderFrameHost* frame,
                        const GURL& form_url,
                        BadMessageReason reason,
                        bool may_kill_renderer) {
  if (form_url.SchemeIs(url::kAboutScheme) ||
      form_url.SchemeIs(url::kDataScheme)) {
    if (may_kill_renderer) {
      SYSLOG(WARNING)
          << "Killing renderer: illegal password access from about: "
          << "or data: URL. Reason: " << static_cast<int>(reason);
      bad_message::ReceivedBadMessage(frame->GetProcess(), reason);
    }
    return false;
  }

  return true;
}

bool CheckChildProcessSecurityPolicyForURL(content::RenderFrameHost* frame,
                                           const GURL& form_url,
                                           BadMessageReason reason,
                                           bool may_kill_renderer) {
  if (!CheckForIllegalURL(frame, form_url, reason, may_kill_renderer)) {
    return false;
  }

  content::ChildProcessSecurityPolicy* policy =
      content::ChildProcessSecurityPolicy::GetInstance();
  if (!policy->CanAccessDataForOrigin(frame->GetProcess()->GetDeprecatedID(),
                                      url::Origin::Create(form_url))) {
    if (may_kill_renderer) {
      SYSLOG(WARNING) << "Killing renderer: illegal password access. Reason: "
                      << static_cast<int>(reason);
      bad_message::ReceivedBadMessage(frame->GetProcess(), reason);
    }
    return false;
  }

  return true;
}

bool CheckFrameNotPrerendering(content::RenderFrameHost* frame) {
  if (frame->GetLifecycleState() ==
      content::RenderFrameHost::LifecycleState::kPrerendering) {
    bad_message::ReceivedBadMessage(
        frame->GetProcess(), BadMessageReason::CPMD_BAD_ORIGIN_PRERENDERING);
    return false;
  }
  return true;
}

bool CheckGeneratedPassword(content::RenderFrameHost* frame,
                            const std::u16string& generated_password) {
  if (generated_password.empty()) {
    ReceivedBadMessage(
        frame->GetProcess(),
        BadMessageReason::CPMD_BAD_ORIGIN_NO_GENERATED_PASSWORD_TO_EDIT);
    return false;
  }
  return true;
}

}  // namespace password_manager::bad_message