/*
 * Copyright 2023, Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 */


#include <OS.h>
#include <vm_defs.h>

#include "strace.h"
#include "Syscall.h"
#include "TypeHandler.h"


#define FLAG_INFO_ENTRY(name) \
	{ name, #name }

static const FlagsTypeHandler::FlagInfo kAreaProtectionFlagInfos[] = {
	FLAG_INFO_ENTRY(B_READ_AREA),
	FLAG_INFO_ENTRY(B_WRITE_AREA),
	FLAG_INFO_ENTRY(B_EXECUTE_AREA),
	FLAG_INFO_ENTRY(B_STACK_AREA),

	FLAG_INFO_ENTRY(B_CLONEABLE_AREA),
	FLAG_INFO_ENTRY(B_OVERCOMMITTING_AREA),

	{ 0, NULL }
};


struct enum_info {
	unsigned int index;
	const char *name;
};

#define ENUM_INFO_ENTRY(name) \
	{ name, #name }

static const enum_info kAddressSpecs[] = {
	ENUM_INFO_ENTRY(B_ANY_ADDRESS),
	ENUM_INFO_ENTRY(B_EXACT_ADDRESS),
	ENUM_INFO_ENTRY(B_BASE_ADDRESS),
	ENUM_INFO_ENTRY(B_CLONE_ADDRESS),
	ENUM_INFO_ENTRY(B_ANY_KERNEL_ADDRESS),
	ENUM_INFO_ENTRY(B_RANDOMIZED_ANY_ADDRESS),
	ENUM_INFO_ENTRY(B_RANDOMIZED_BASE_ADDRESS),

	{ 0, NULL }
};


static FlagsTypeHandler::FlagsList kAreaProtectionFlags;
static EnumTypeHandler::EnumMap kAddressSpecsMap;


void
patch_area()
{
	for (int i = 0; kAreaProtectionFlagInfos[i].name != NULL; i++) {
		kAreaProtectionFlags.push_back(kAreaProtectionFlagInfos[i]);
	}
	for (int i = 0; kAddressSpecs[i].name != NULL; i++) {
		kAddressSpecsMap[kAddressSpecs[i].index] = kAddressSpecs[i].name;
	}

	Syscall *create = get_syscall("_kern_create_area");
	create->GetParameter("address")->SetInOut(true);
	create->GetParameter("addressSpec")->SetHandler(
		new EnumTypeHandler(kAddressSpecsMap));
	create->GetParameter("protection")->SetHandler(
		new FlagsTypeHandler(kAreaProtectionFlags));

	Syscall *transfer = get_syscall("_kern_transfer_area");
	transfer->GetParameter("_address")->SetInOut(true);
	transfer->GetParameter("addressSpec")->SetHandler(
		new EnumTypeHandler(kAddressSpecsMap));

	Syscall *clone = get_syscall("_kern_clone_area");
	clone->GetParameter("_address")->SetInOut(true);
	clone->GetParameter("addressSpec")->SetHandler(
		new EnumTypeHandler(kAddressSpecsMap));
	clone->GetParameter("protection")->SetHandler(
		new FlagsTypeHandler(kAreaProtectionFlags));

	Syscall *reserve_address_range = get_syscall("_kern_reserve_address_range");
	reserve_address_range->GetParameter("_address")->SetInOut(true);
	reserve_address_range->GetParameter("addressSpec")->SetHandler(
		new EnumTypeHandler(kAddressSpecsMap));

	Syscall *set_area_protection = get_syscall("_kern_set_area_protection");
	set_area_protection->GetParameter("newProtection")->SetHandler(
		new FlagsTypeHandler(kAreaProtectionFlags));

	Syscall *map_file = get_syscall("_kern_map_file");
	map_file->GetParameter("address")->SetInOut(true);
	map_file->GetParameter("addressSpec")->SetHandler(
		new EnumTypeHandler(kAddressSpecsMap));
	map_file->GetParameter("protection")->SetHandler(
		new FlagsTypeHandler(kAreaProtectionFlags));

	Syscall *set_memory_protection = get_syscall("_kern_set_memory_protection");
	set_memory_protection->GetParameter("protection")->SetHandler(
		new FlagsTypeHandler(kAreaProtectionFlags));
}