// SPDX-License-Identifier: GPL-2.0-or-later
/* Copyright (c) 2021 SUSE LLC */

#include <stdlib.h>
#include <stdio.h>

#include "tst_test.h"

static char *only_mount_v1;
static char *no_cleanup;
static struct tst_option opts[] = {
	{"v", &only_mount_v1, "-v\tOnly try to mount CGroups V1"},
	{"n", &no_cleanup, "-n\tLeave CGroups created by test"},
	{NULL, NULL, NULL},
};
static struct tst_cg_opts cgopts;
static struct tst_cg_group *cg_child;

static void do_test(void)
{
	char buf[BUFSIZ];
	size_t mem;

	if (!TST_CG_VER_IS_V1(tst_cg, "memory"))
		SAFE_CG_PRINT(tst_cg, "cgroup.subtree_control", "+memory");
	if (!TST_CG_VER_IS_V1(tst_cg, "cpuset"))
		SAFE_CG_PRINT(tst_cg, "cgroup.subtree_control", "+cpuset");

	cg_child = tst_cg_group_mk(tst_cg, "child");
	if (!SAFE_FORK()) {
		SAFE_CG_PRINTF(cg_child, "cgroup.procs", "%d", getpid());

		SAFE_CG_SCANF(cg_child, "memory.current", "%zu", &mem);
		tst_res(TPASS, "child/memory.current = %zu", mem);
		SAFE_CG_PRINTF(cg_child, "memory.max",
				   "%zu", (1UL << 24) - 1);
		SAFE_CG_PRINTF(cg_child, "memory.swap.max",
				   "%zu", 1UL << 31);

		SAFE_CG_READ(cg_child, "cpuset.mems", buf, sizeof(buf));
		tst_res(TPASS, "child/cpuset.mems = %s", buf);
		SAFE_CG_PRINT(cg_child, "cpuset.mems", buf);

		exit(0);
	}

	SAFE_CG_PRINTF(tst_cg, "memory.max", "%zu", (1UL << 24) - 1);
	SAFE_CG_PRINTF(cg_child, "cgroup.procs", "%d", getpid());
	SAFE_CG_SCANF(tst_cg, "memory.current", "%zu", &mem);
	tst_res(TPASS, "memory.current = %zu", mem);

	tst_reap_children();
	SAFE_CG_PRINTF(tst_cg_drain, "cgroup.procs", "%d", getpid());
	cg_child = tst_cg_group_rm(cg_child);
}

static void setup(void)
{
	cgopts.needs_ver = !!only_mount_v1 ? TST_CG_V1 : 0;

	tst_cg_scan();
	tst_cg_print_config();

	tst_cg_require("memory", &cgopts);
	tst_cg_require("cpuset", &cgopts);

	tst_cg_init();
}

static void cleanup(void)
{
	if (cg_child) {
		SAFE_CG_PRINTF(tst_cg_drain,
				   "cgroup.procs", "%d", getpid());
		cg_child = tst_cg_group_rm(cg_child);
	}
	if (!no_cleanup)
		tst_cg_cleanup();
}

static struct tst_test test = {
	.test_all = do_test,
	.setup = setup,
	.cleanup = cleanup,
	.options = opts,
	.forks_child = 1,
};