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

/* $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();
}