// Copyright 2017 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 "base/process/kill.h"

#include <magenta/syscalls.h>

#include "base/process/process_iterator.h"
#include "base/task_scheduler/post_task.h"
#include "base/threading/platform_thread.h"

namespace base {

bool KillProcessGroup(ProcessHandle process_group_id) {
  // |process_group_id| is really a job on Fuchsia.
  mx_status_t status = mx_task_kill(process_group_id);
  DLOG_IF(ERROR, status != NO_ERROR)
      << "unable to terminate job " << process_group_id;
  return status == NO_ERROR;
}

TerminationStatus GetTerminationStatus(ProcessHandle handle, int* exit_code) {
  mx_info_process_t process_info;
  mx_status_t status =
      mx_object_get_info(handle, MX_INFO_PROCESS, &process_info,
                         sizeof(process_info), nullptr, nullptr);
  if (status != NO_ERROR) {
    DLOG(ERROR) << "unable to get termination status for " << handle;
    *exit_code = 0;
    return TERMINATION_STATUS_NORMAL_TERMINATION;
  }
  if (!process_info.started) {
    return TERMINATION_STATUS_LAUNCH_FAILED;
  }
  if (!process_info.exited) {
    return TERMINATION_STATUS_STILL_RUNNING;
  }

  // TODO(fuchsia): Is there more information about types of crashes, OOM, etc.
  // available? https://crbug.com/706592.

  *exit_code = process_info.return_code;
  return process_info.return_code == 0
             ? TERMINATION_STATUS_NORMAL_TERMINATION
             : TERMINATION_STATUS_ABNORMAL_TERMINATION;
}

void EnsureProcessTerminated(Process process) {
  DCHECK(!process.is_current());

  mx_signals_t signals;
  if (mx_object_wait_one(process.Handle(), MX_TASK_TERMINATED, 0, &signals) ==
      NO_ERROR) {
    DCHECK(signals & MX_TASK_TERMINATED);
    // If already signaled, then the process is terminated.
    return;
  }

  PostDelayedTaskWithTraits(
      FROM_HERE,
      {TaskPriority::BACKGROUND, TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN},
      Bind(
          [](Process process) {
            mx_signals_t signals;
            if (mx_object_wait_one(process.Handle(), MX_TASK_TERMINATED, 0,
                                   &signals) == NO_ERROR) {
              return;
            }
            process.Terminate(1, false);
          },
          Passed(&process)),
      TimeDelta::FromSeconds(2));
}

}  // namespace base