// SPDX-License-Identifier: GPL-2.0
/* 
 * Copyright (C) 2000 - 2007 Jeff Dike (jdike@{addtoit,linux.intel}.com)
 */

#include <linux/sched/signal.h>
#include <linux/sched/task.h>
#include <linux/sched/mm.h>
#include <linux/spinlock.h>
#include <linux/slab.h>
#include <linux/oom.h>
#include <linux/reboot.h>
#include <kern_util.h>
#include <os.h>
#include <skas.h>

void (*pm_power_off)(void);
EXPORT_SYMBOL(pm_power_off);

static void kill_off_processes(void)
{
 struct task_struct *p;
 int pid;

 read_lock(&tasklist_lock);
 for_each_process(p) {
  struct task_struct *t;

  t = find_lock_task_mm(p);
  if (!t)
   continue;
  pid = t->mm->context.id.pid;
  task_unlock(t);
  os_kill_ptraced_process(pid, 1);
 }
 read_unlock(&tasklist_lock);
}

void uml_cleanup(void)
{
 kmalloc_ok = 0;
 do_uml_exitcalls();
 kill_off_processes();
}

void machine_restart(char * __unused)
{
 uml_cleanup();
 reboot_skas();
}

void machine_power_off(void)
{
 uml_cleanup();
 halt_skas();
}

void machine_halt(void)
{
 machine_power_off();
}

static int sys_power_off_handler(struct sys_off_data *data)
{
 machine_power_off();
 return 0;
}

static int register_power_off(void)
{
 register_sys_off_handler(SYS_OFF_MODE_POWER_OFF,
     SYS_OFF_PRIO_DEFAULT,
     sys_power_off_handler, NULL);
 return 0;
}
__initcall(register_power_off);