/*	$NetBSD: t_renamerace.c,v 1.44 2022/01/31 17:23:37 ryo Exp $	*/

/*
 * Modified for rump and atf from a program supplied
 * by Nicolas Joly in kern/40948
 */

#include <sys/types.h>
#include <sys/mount.h>
#include <sys/utsname.h>

#include <atf-c.h>
#include <errno.h>
#include <fcntl.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <rump/rump.h>
#include <rump/rump_syscalls.h>

/* Bump the size of the test file system image to a larger value.
 *
 * These tests cause a lot of churn in the file system by creating and
 * deleting files/directories in quick succession.  A faster CPU will cause
 * more churn because the tests are capped by a run time period in seconds,
 * not number of operations.
 *
 * This is all fine except for LFS, because the lfs_cleanerd cannot keep up
 * with the churn and thus causes the test to fail on fast machines.  Hence
 * the reason for this hack. */
#define FSTEST_IMGSIZE (50000 * 512)

#include "../common/h_fsmacros.h"
#include "h_macros.h"

static volatile int quittingtime;
pid_t wrkpid;

static void *
w1(void *arg)
{
	int fd;

	rump_pub_lwproc_newlwp(wrkpid);

	while (!quittingtime) {
		fd = rump_sys_open("rename.test1",
		    O_WRONLY|O_CREAT|O_TRUNC, 0666);
		if (fd == -1 && errno != EEXIST)
			atf_tc_fail_errno("create");
		rump_sys_unlink("rename.test1");
		rump_sys_close(fd);
	}

	return NULL;
}

static void *
w1_dirs(void *arg)
{

	rump_pub_lwproc_newlwp(wrkpid);

	while (!quittingtime) {
		if (rump_sys_mkdir("rename.test1", 0777) == -1)
			atf_tc_fail_errno("mkdir");
		rump_sys_rmdir("rename.test1");
	}

	return NULL;
}

static void *
w2(void *arg)
{

	rump_pub_lwproc_newlwp(wrkpid);

	while (!quittingtime) {
		rump_sys_rename("rename.test1", "rename.test2");
	}

	return NULL;
}

static void
w3_mkdir(void)
{

	if (rump_sys_mkdir("c", 0777) == -1)
		ATF_CHECK_MSG(errno == ENOENT || errno == EEXIST,
		    "mkdir: %s (errno=%d)\n", strerror(errno), errno);
	if (rump_sys_mkdir("c/d", 0777) == -1)
		ATF_CHECK_MSG(errno == ENOENT || errno == EEXIST,
		    "mkdir: %s (errno=%d)\n", strerror(errno), errno);
	if (rump_sys_mkdir("c/d/e", 0777) == -1)
		ATF_CHECK_MSG(errno == ENOENT || errno == EEXIST,
		    "mkdir: %s (errno=%d)\n", strerror(errno), errno);
}

static void *
w3_rmdir(void *arg)
{

	rump_pub_lwproc_newlwp(wrkpid);

	while (!quittingtime) {
		w3_mkdir();
		rump_sys_rmdir("c/d/e");
		rump_sys_rmdir("c/d");
	}

	return NULL;
}

static void *
w3_rename1(void *arg)
{

	rump_pub_lwproc_newlwp(wrkpid);

	while (!quittingtime) {
		w3_mkdir();
		rump_sys_rename("c", "c/d/e");
	}

	return NULL;
}

static void *
w3_rename2(void *arg)
{

	rump_pub_lwproc_newlwp(wrkpid);

	while (!quittingtime) {
		w3_mkdir();
		rump_sys_rename("c/d/e", "c");
	}

	return NULL;
}

#define NWRK 8
static void
renamerace(const atf_tc_t *tc, const char *mp)
{
	pthread_t pt1[NWRK], pt2[NWRK];
	int i;

	/*
	 * Sysvbfs supports only 8 inodes so this test would exhaust
	 * the inode table and creating files would fail with ENOSPC.
	 */
	if (FSTYPE_SYSVBFS(tc))
		atf_tc_skip("filesystem has not enough inodes");
	if (FSTYPE_RUMPFS(tc))
		atf_tc_skip("rename not supported by file system");
	if (FSTYPE_UDF(tc))
		atf_tc_expect_fail("PR kern/53865");

	RZ(rump_pub_lwproc_rfork(RUMP_RFCFDG));
	RL(wrkpid = rump_sys_getpid());

	RL(rump_sys_chdir(mp));
	for (i = 0; i < NWRK; i++)
		pthread_create(&pt1[i], NULL, w1, NULL);

	for (i = 0; i < NWRK; i++)
		pthread_create(&pt2[i], NULL, w2, NULL);

	sleep(5);
	quittingtime = 1;

	for (i = 0; i < NWRK; i++)
		pthread_join(pt1[i], NULL);
	for (i = 0; i < NWRK; i++)
		pthread_join(pt2[i], NULL);
	RL(rump_sys_chdir("/"));

	if (FSTYPE_UDF(tc))
		atf_tc_fail("race did not trigger this time");
}

static void
renamerace_dirs(const atf_tc_t *tc, const char *mp)
{
	pthread_t pt1, pt2;

	if (FSTYPE_SYSVBFS(tc))
		atf_tc_skip("directories not supported by file system");
	if (FSTYPE_RUMPFS(tc))
		atf_tc_skip("rename not supported by file system");
	if (FSTYPE_UDF(tc))
		atf_tc_expect_fail("PR kern/53865");

	RZ(rump_pub_lwproc_rfork(RUMP_RFCFDG));
	RL(wrkpid = rump_sys_getpid());

	RL(rump_sys_chdir(mp));
	pthread_create(&pt1, NULL, w1_dirs, NULL);
	pthread_create(&pt2, NULL, w2, NULL);

	sleep(5);
	quittingtime = 1;

	pthread_join(pt1, NULL);
	pthread_join(pt2, NULL);
	RL(rump_sys_chdir("/"));

	if (FSTYPE_UDF(tc))
		atf_tc_fail("race did not trigger this time");
}

static void
renamerace_cycle(const atf_tc_t *tc, const char *mp)
{
	pthread_t pt_rmdir, pt_rename1, pt_rename2;

	if (FSTYPE_SYSVBFS(tc))
		atf_tc_skip("directories not supported by file system");
	if (FSTYPE_RUMPFS(tc))
		atf_tc_skip("rename not supported by file system");
	if (FSTYPE_P2K_FFS(tc))
		atf_tc_expect_fail("assertion \"vp->v_size == ip->i_size\" failed");
	if (FSTYPE_PUFFS(tc))
		atf_tc_expect_fail("assertion \"dfd\" failed");
	if (FSTYPE_NFS(tc))
		atf_tc_expect_fail("mkdir fails with ESTALE");
	if (FSTYPE_UDF(tc))
		atf_tc_expect_fail("sometimes fails with ENOSPC, PR kern/56253");

	RZ(rump_pub_lwproc_rfork(RUMP_RFCFDG));
	RL(wrkpid = rump_sys_getpid());

	RL(rump_sys_chdir(mp));
	pthread_create(&pt_rmdir, NULL, w3_rmdir, NULL);
	pthread_create(&pt_rename1, NULL, w3_rename1, NULL);
	pthread_create(&pt_rename2, NULL, w3_rename2, NULL);

	sleep(10);
	quittingtime = 1;

	alarm(5);
	pthread_join(pt_rmdir, NULL);
	pthread_join(pt_rename1, NULL);
	pthread_join(pt_rename2, NULL);
	alarm(0);
	RL(rump_sys_chdir("/"));

	if (FSTYPE_UDF(tc))
		atf_tc_fail("PR kern/56253 did not trigger this time");
	if (FSTYPE_P2K_FFS(tc))
		atf_tc_fail("did not fail this time");
	if (FSTYPE_PUFFS(tc))
		atf_tc_fail("did not fail this time");
	if (FSTYPE_NFS(tc))
		atf_tc_fail("did not fail this time");
}

ATF_TC_FSAPPLY(renamerace, "rename(2) race with file unlinked mid-operation");
ATF_TC_FSAPPLY(renamerace_dirs, "rename(2) race with directories");
ATF_TC_FSAPPLY(renamerace_cycle, "rename(2) race making directory cycles");

ATF_TP_ADD_TCS(tp)
{

	ATF_TP_FSAPPLY(renamerace); /* PR kern/41128 */
	ATF_TP_FSAPPLY(renamerace_dirs);
	ATF_TP_FSAPPLY(renamerace_cycle);

	return atf_no_error();
}