// SPDX-License-Identifier: GPL-2.0
#include <linux/ptrace.h>
#include <linux/sched.h>
#include <linux/sched/task_stack.h>
#include <linux/export.h>
#include <asm/syscall.h>

static int collect_syscall(struct task_struct *target, struct syscall_info *info)
{
 unsigned long args[6] = { };
 struct pt_regs *regs;

 if (!try_get_task_stack(target)) {
  /* Task has no stack, so the task isn't in a syscall. */
  memset(info, 0, sizeof(*info));
  info->data.nr = -1;
  return 0;
 }

 regs = task_pt_regs(target);
 if (unlikely(!regs)) {
  put_task_stack(target);
  return -EAGAIN;
 }

 info->sp = user_stack_pointer(regs);
 info->data.instruction_pointer = instruction_pointer(regs);

 info->data.nr = syscall_get_nr(target, regs);
 if (info->data.nr != -1L)
  syscall_get_arguments(target, regs, args);

 info->data.args[0] = args[0];
 info->data.args[1] = args[1];
 info->data.args[2] = args[2];
 info->data.args[3] = args[3];
 info->data.args[4] = args[4];
 info->data.args[5] = args[5];

 put_task_stack(target);
 return 0;
}

/**
 * task_current_syscall - Discover what a blocked task is doing.
 * @target:  thread to examine
 * @info:  structure with the following fields:
 *    .sp        - filled with user stack pointer
 *    .data.nr   - filled with system call number or -1
 *    .data.args - filled with @maxargs system call arguments
 *    .data.instruction_pointer - filled with user PC
 *
 * If @target is blocked in a system call, returns zero with @info.data.nr
 * set to the call's number and @info.data.args filled in with its
 * arguments. Registers not used for system call arguments may not be available
 * and it is not kosher to use &struct user_regset calls while the system
 * call is still in progress.  Note we may get this result if @target
 * has finished its system call but not yet returned to user mode, such
 * as when it's stopped for signal handling or syscall exit tracing.
 *
 * If @target is blocked in the kernel during a fault or exception,
 * returns zero with *@info.data.nr set to -1 and does not fill in
 * @info.data.args. If so, it's now safe to examine @target using
 * &struct user_regset get() calls as long as we're sure @target won't return
 * to user mode.
 *
 * Returns -%EAGAIN if @target does not remain blocked.
 */
int task_current_syscall(struct task_struct *target, struct syscall_info *info)
{
 unsigned long ncsw;
 unsigned int state;

 if (target == current)
  return collect_syscall(target, info);

 state = READ_ONCE(target->__state);
 if (unlikely(!state))
  return -EAGAIN;

 ncsw = wait_task_inactive(target, state);
 if (unlikely(!ncsw) ||
     unlikely(collect_syscall(target, info)) ||
     unlikely(wait_task_inactive(target, state) != ncsw))
  return -EAGAIN;

 return 0;
}