/*
 * trace/beauty/prctl.c
 *
 *  Copyright (C) 2017, Red Hat Inc, Arnaldo Carvalho de Melo <acme@redhat.com>
 *
 * Released under the GPL v2. (and only v2, not any later version)
 */

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

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

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

static size_t prctl__scnprintf_set_mm(int option, char *bf, size_t size)
{
	static DEFINE_STRARRAY(prctl_set_mm_options);
	return strarray__scnprintf(&strarray__prctl_set_mm_options, bf, size, "%d", 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);
	/*
	 * 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);
}