// SPDX-License-Identifier: LGPL-2.1
/*
 * trace/beauty/prctl.c
 *
 *  Copyright (C) 2017, Red Hat Inc, Arnaldo Carvalho de Melo <acme@redhat.com>
 */

#include "trace/beauty/beauty.h"
#include <linux/kernel.h>
#include <linux/prctl.h>

#include "trace/beauty/generated/prctl_option_array.c"

DEFINE_STRARRAY(prctl_options, "PR_");

static size_t prctl__scnprintf_option(int option, char *bf, size_t size, bool show_prefix)
{
 return strarray__scnprintf(&strarray__prctl_options, bf, size, "%d", show_prefix, option);
}

static size_t prctl__scnprintf_set_mm(int option, char *bf, size_t size, bool show_prefix)
{
 static DEFINE_STRARRAY(prctl_set_mm_options, "PR_SET_MM_");
 return strarray__scnprintf(&strarray__prctl_set_mm_options, bf, size, "%d", show_prefix, option);
}

size_t syscall_arg__scnprintf_prctl_arg2(char *bf, size_t size, struct syscall_arg *arg)
{
 int option = syscall_arg__val(arg, 0);

 if (option == PR_SET_MM)
  return prctl__scnprintf_set_mm(arg->val, bf, size, arg->show_string_prefix);
 /*
  * We still don't grab the contents of pointers on entry or exit,
  * so just print them as hex numbers
 */
 if (option == PR_SET_NAME)
  return syscall_arg__scnprintf_hex(bf, size, arg);

 return syscall_arg__scnprintf_long(bf, size, arg);
}

size_t syscall_arg__scnprintf_prctl_arg3(char *bf, size_t size, struct syscall_arg *arg)
{
 int option = syscall_arg__val(arg, 0);

 if (option == PR_SET_MM)
  return syscall_arg__scnprintf_hex(bf, size, arg);

 return syscall_arg__scnprintf_long(bf, size, arg);
}

size_t syscall_arg__scnprintf_prctl_option(char *bf, size_t size, struct syscall_arg *arg)
{
 unsigned long option = arg->val;
 enum {
                SPO_ARG2 = (1 << 1),
                SPO_ARG3 = (1 << 2),
                SPO_ARG4 = (1 << 3),
                SPO_ARG5 = (1 << 4),
                SPO_ARG6 = (1 << 5),
        };
 const u8 all_but2 = SPO_ARG3 | SPO_ARG4 | SPO_ARG5 | SPO_ARG6;
 const u8 all = SPO_ARG2 | all_but2;
 const u8 masks[] = {
  [PR_GET_DUMPABLE]  = all,
  [PR_SET_DUMPABLE]  = all_but2,
  [PR_SET_NAME]   = all_but2,
  [PR_GET_CHILD_SUBREAPER] = all_but2,
  [PR_SET_CHILD_SUBREAPER] = all_but2,
  [PR_GET_SECUREBITS]  = all,
  [PR_SET_SECUREBITS]  = all_but2,
  [PR_SET_MM]   = SPO_ARG4 | SPO_ARG5 | SPO_ARG6,
  [PR_GET_PDEATHSIG]  = all,
  [PR_SET_PDEATHSIG]  = all_but2,
 };

 if (option < ARRAY_SIZE(masks))
  arg->mask |= masks[option];

 return prctl__scnprintf_option(option, bf, size, arg->show_string_prefix);
}