// Copyright 2015 The Crashpad Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "handler/mac/crash_report_exception_handler.h"

#include <utility>
#include <vector>

#include "base/apple/mach_logging.h"
#include "base/apple/scoped_mach_port.h"
#include "base/logging.h"
#include "base/strings/stringprintf.h"
#include "client/settings.h"
#include "handler/mac/file_limit_annotation.h"
#include "minidump/minidump_file_writer.h"
#include "minidump/minidump_user_extension_stream_data_source.h"
#include "snapshot/crashpad_info_client_options.h"
#include "snapshot/mac/process_snapshot_mac.h"
#include "util/file/file_helper.h"
#include "util/file/file_writer.h"
#include "util/mach/bootstrap.h"
#include "util/mach/exc_client_variants.h"
#include "util/mach/exception_behaviors.h"
#include "util/mach/exception_ports.h"
#include "util/mach/exception_types.h"
#include "util/mach/mach_extensions.h"
#include "util/mach/mach_message.h"
#include "util/mach/scoped_task_suspend.h"
#include "util/mach/symbolic_constants_mach.h"
#include "util/misc/metrics.h"
#include "util/misc/tri_state.h"
#include "util/misc/uuid.h"

namespace crashpad {

CrashReportExceptionHandler::CrashReportExceptionHandler(
    CrashReportDatabase* database,
    CrashReportUploadThread* upload_thread,
    const std::map<std::string, std::string>* process_annotations,
    const std::vector<base::FilePath>* attachments,
    const UserStreamDataSources* user_stream_data_sources,
    bool wait_for_upload,
    const base::FilePath* crash_reporter,
    const base::FilePath* crash_envelope,
    const UUID* report_id)
    : database_(database),
      upload_thread_(upload_thread),
      process_annotations_(process_annotations),
      attachments_(attachments),
      user_stream_data_sources_(user_stream_data_sources),
      wait_for_upload_(wait_for_upload),
      crash_reporter_(crash_reporter),
      crash_envelope_(crash_envelope),
      report_id_(report_id) {}

CrashReportExceptionHandler::~CrashReportExceptionHandler() {
}

kern_return_t CrashReportExceptionHandler::CatchMachException(
    exception_behavior_t behavior,
    exception_handler_t exception_port,
    thread_t thread,
    task_t task,
    exception_type_t exception,
    const mach_exception_data_type_t* code,
    mach_msg_type_number_t code_count,
    thread_state_flavor_t* flavor,
    ConstThreadState old_state,
    mach_msg_type_number_t old_state_count,
    thread_state_t new_state,
    mach_msg_type_number_t* new_state_count,
    const mach_msg_trailer_t* trailer,
    bool* destroy_complex_request) {
  RecordFileLimitAnnotation();
  Metrics::ExceptionEncountered();
  Metrics::ExceptionCode(ExceptionCodeForMetrics(exception, code[0]));
  *destroy_complex_request = true;

  // The expected behavior is EXCEPTION_STATE_IDENTITY | MACH_EXCEPTION_CODES,
  // but it’s possible to deal with any exception behavior as long as it
  // carries identity information (valid thread and task ports).
  if (!ExceptionBehaviorHasIdentity(behavior)) {
    LOG(ERROR) << base::StringPrintf(
        "unexpected exception behavior %s, rejecting",
        ExceptionBehaviorToString(
            behavior, kUseFullName | kUnknownIsNumeric | kUseOr).c_str());
    Metrics::ExceptionCaptureResult(
        Metrics::CaptureResult::kUnexpectedExceptionBehavior);
    return KERN_FAILURE;
  } else if (behavior != (EXCEPTION_STATE_IDENTITY | kMachExceptionCodes)) {
    LOG(WARNING) << base::StringPrintf(
        "unexpected exception behavior %s, proceeding",
        ExceptionBehaviorToString(
            behavior, kUseFullName | kUnknownIsNumeric | kUseOr).c_str());
  }

  if (task == mach_task_self()) {
    LOG(ERROR) << "cannot suspend myself";
    Metrics::ExceptionCaptureResult(
        Metrics::CaptureResult::kFailedDueToSuspendSelf);
    return KERN_FAILURE;
  }

  ScopedTaskSuspend suspend(task);

  ProcessSnapshotMac process_snapshot;
  if (!process_snapshot.Initialize(task)) {
    Metrics::ExceptionCaptureResult(Metrics::CaptureResult::kSnapshotFailed);
    return KERN_FAILURE;
  }

  // Check for suspicious message sources. A suspicious exception message comes
  // from a source other than the kernel or the process that the exception
  // purportedly occurred in.
  //
  // TODO(mark): Consider exceptions outside of the range (0, 32) from the
  // kernel to be suspicious, and exceptions other than kMachExceptionSimulated
  // from the process itself to be suspicious.
  const pid_t pid = process_snapshot.ProcessID();
  pid_t audit_pid = AuditPIDFromMachMessageTrailer(trailer);
  if (audit_pid != -1 && audit_pid != 0) {
    if (audit_pid != pid) {
      LOG(WARNING) << "exception for pid " << pid << " sent by pid "
                   << audit_pid;
    }
  }

  CrashpadInfoClientOptions client_options;
  process_snapshot.GetCrashpadOptions(&client_options);

  if (client_options.crashpad_handler_behavior != TriState::kDisabled &&
      !IsExceptionNonfatalResource(exception, code[0], pid)) {
    // Non-fatal resource exceptions are never user-visible and are not
    // currently of interest to Crashpad.

    if (!process_snapshot.InitializeException(behavior,
                                              thread,
                                              exception,
                                              code,
                                              code_count,
                                              *flavor,
                                              old_state,
                                              old_state_count)) {
      Metrics::ExceptionCaptureResult(
          Metrics::CaptureResult::kExceptionInitializationFailed);
      return KERN_FAILURE;
    }

    UUID client_id;
    Settings* const settings = database_->GetSettings();
    if (settings && settings->GetClientID(&client_id)) {
      process_snapshot.SetClientID(client_id);
    }
    process_snapshot.SetAnnotationsSimpleMap(*process_annotations_);

    std::unique_ptr<CrashReportDatabase::NewReport> new_report;
    CrashReportDatabase::OperationStatus database_status =
        database_->PrepareNewCrashReport(&new_report, report_id_);
    if (database_status != CrashReportDatabase::kNoError) {
      Metrics::ExceptionCaptureResult(
          Metrics::CaptureResult::kPrepareNewCrashReportFailed);
      return KERN_FAILURE;
    }

    process_snapshot.SetReportID(new_report->ReportID());

    MinidumpFileWriter minidump;
    minidump.InitializeFromSnapshot(&process_snapshot);
    AddUserExtensionStreams(
        user_stream_data_sources_, &process_snapshot, &minidump);

    if (!minidump.WriteEverything(new_report->Writer())) {
      Metrics::ExceptionCaptureResult(
          Metrics::CaptureResult::kMinidumpWriteFailed);
      return KERN_FAILURE;
    }

    for (const auto& attachment : (*attachments_)) {
      FileReader file_reader;
      if (!file_reader.Open(attachment)) {
        LOG(ERROR) << "attachment " << attachment.value().c_str()
                   << " couldn't be opened, skipping";
        continue;
      }

      base::FilePath filename = attachment.BaseName();
      FileWriter* file_writer = new_report->AddAttachment(filename.value());
      if (file_writer == nullptr) {
        LOG(ERROR) << "attachment " << filename.value().c_str()
                   << " couldn't be created, skipping";
        continue;
      }

      CopyFileContent(&file_reader, file_writer);
    }

    bool has_crash_reporter = crash_reporter_ && !crash_reporter_->empty() &&
                              crash_envelope_ && !crash_envelope_->empty();
    if (has_crash_reporter) {
      CrashReportDatabase::Envelope envelope(new_report->ReportID());
      if (envelope.Initialize(*crash_envelope_)) {
        envelope.AddAttachments(*attachments_);
        if (auto reader = new_report->Reader()) {
          envelope.AddMinidump(reader);
        }
        envelope.Finish();
        database_->LaunchCrashReporter(*crash_reporter_, *crash_envelope_);
      }
    }

    UUID uuid;
    database_status =
        database_->FinishedWritingCrashReport(std::move(new_report), &uuid);
    if (database_status != CrashReportDatabase::kNoError) {
      Metrics::ExceptionCaptureResult(
          Metrics::CaptureResult::kFinishedWritingCrashReportFailed);
      return KERN_FAILURE;
    }

    if (has_crash_reporter) {
      database_->DeleteReport(uuid);
    } else if (upload_thread_) {
      if (wait_for_upload_) {
        upload_thread_->ReportPendingSync(uuid);
      } else {
        upload_thread_->ReportPending(uuid);
      }
    }
  }

  if (client_options.system_crash_reporter_forwarding != TriState::kDisabled) {
    if (exception == EXC_CRASH) {
      // For exception handlers that respond to state-carrying behaviors, when
      // the handler is called by the kernel (as it is normally), the kernel
      // will attempt to set a new thread state when the exception handler
      // returns successfully. Other code that mimics the kernel’s
      // exception-delivery semantics may implement the same or similar
      // behavior. In some situations, it is undesirable to set a new thread
      // state. If the exception handler were to return unsuccessfully, however,
      // the kernel would continue searching for an exception handler at a wider
      // (task or host) scope. This may also be undesirable.
      //
      // If such exception handlers return `MACH_RCV_PORT_DIED`, the kernel will
      // not set a new thread state and will also not search for another
      // exception handler. See 15.3 xnu-11215.84.4/osfmk/kern/exception.c.
      // `exception_deliver()` will only set a new thread state if the handler’s
      // return code was `MACH_MSG_SUCCESS` (a synonym for `KERN_SUCCESS`), and
      // subsequently, `exception_triage()` will not search for a new handler if
      // the handler’s return code was `KERN_SUCCESS` or `MACH_RCV_PORT_DIED`.
      //
      // Another effect of returning `MACH_RCV_PORT_DIED` for `EXC_CRASH` is
      // that an `EXC_CORPSE_NOTIFY` exception is generated. Starting with macOS
      // 10.15, for the system crash reporter to generate a report,
      // `EXC_CORPSE_NOTIFY` *must* be generated and forwarding `EXC_CRASH` (as
      // we do below with `EXC_RESOURCE` and pre-macOS 13 `EXC_GUARD`) is not
      // sufficient. Between macOS 10.11 and macOS 10.14 (inclusive), both
      // forwarding as below, and causing `EXC_CORPSE_NOTIFY` to be generated
      // are sufficient (and in fact, if we do both, two crash reports are
      // generated).
      return MACH_RCV_PORT_DIED;
    }
    if (exception == EXC_RESOURCE || exception == EXC_GUARD) {
      // Only forward the types of exceptions that the crash reporter would
      // receive under normal conditions. Otherwise, system crash reporter could
      // present the system’s crash UI for processes that haven’t actually
      // crashed, and could result in reports not actually associated with
      // crashes being sent to the operating system vendor.
      base::apple::ScopedMachSendRight system_crash_reporter_handler(
          SystemCrashReporterHandler());
      if (system_crash_reporter_handler.get()) {
        // Make copies of mutable out parameters so that the system crash
        // reporter can’t influence the state returned by this method.
        thread_state_flavor_t flavor_forward = *flavor;
        mach_msg_type_number_t new_state_forward_count = *new_state_count;
        std::vector<natural_t> new_state_forward;
        if (new_state_forward_count) {
          new_state_forward.assign(new_state,
                                   new_state + new_state_forward_count);
        }
        kern_return_t kr = UniversalExceptionRaise(
            EXCEPTION_STATE_IDENTITY | MACH_EXCEPTION_CODES,
            system_crash_reporter_handler.get(),
            thread,
            task,
            exception,
            code,
            code_count,
            &flavor_forward,
            old_state,
            old_state_count,
            new_state_forward_count ? &new_state_forward[0] : nullptr,
            &new_state_forward_count);
        MACH_LOG_IF(WARNING, kr != KERN_SUCCESS, kr)
            << "UniversalExceptionRaise";
      }
    }
  }

  ExcServerCopyState(
      behavior, old_state, old_state_count, new_state, new_state_count);

  Metrics::ExceptionCaptureResult(Metrics::CaptureResult::kSuccess);
  return KERN_SUCCESS;
}

void CrashReportExceptionHandler::RequestRetry() {
  if (upload_thread_) {
    upload_thread_->RetryPending();
  }
}

}  // namespace crashpad
