/*	$NetBSD: t_basic.c,v 1.14 2017/01/13 21:30:40 christos Exp $	*/

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

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

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

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

/*
 * Do a synchronous operation.  When this returns, all FAF operations
 * have at least been delivered to the file system.
 *
 * XXX: is this really good enough considering puffs(9)-issued
 * callback operations?
 */
static void
syncbar(const char *fs)
{
	struct statvfs svb;

	if (rump_sys_statvfs1(fs, &svb, ST_WAIT) == -1)
		atf_tc_fail_errno("statvfs");
}

#ifdef PUFFSDUMP
static void __unused
dumpopcount(struct puffstestargs *args)
{
	size_t i;

	printf("VFS OPS:\n");
	for (i = 0; i < MIN(puffsdump_vfsop_count, PUFFS_VFS_MAX); i++) {
		printf("\t%s: %d\n",
		    puffsdump_vfsop_revmap[i], args->pta_vfs_toserv_ops[i]);
	}

	printf("VN OPS:\n");
	for (i = 0; i < MIN(puffsdump_vnop_count, PUFFS_VN_MAX); i++) {
		printf("\t%s: %d\n",
		    puffsdump_vnop_revmap[i], args->pta_vn_toserv_ops[i]);
	}
}
#endif

ATF_TC(mount);
ATF_TC_HEAD(mount, tc)
{

	atf_tc_set_md_var(tc, "descr", "puffs+dtfs un/mount test");
}

ATF_TC_BODY(mount, tc)
{
	void *args;

	FSTEST_CONSTRUCTOR(tc, puffs, args);
	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(root_reg);
ATF_TC_HEAD(root_reg, tc)
{
	atf_tc_set_md_var(tc, "descr", "root is a regular file");
}

#define MAKEOPTS(...) \
    char *theopts[] = {NULL, "-s", __VA_ARGS__, "dtfs", "n/a", NULL}

ATF_TC_BODY(root_reg, tc)
{
	MAKEOPTS("-r", "reg");
	void *args;
	int fd, rv;

	FSTEST_CONSTRUCTOR_FSPRIV(tc, puffs, args, theopts);

	fd = rump_sys_open(FSTEST_MNTNAME, O_RDWR);
	if (fd == -1)
		atf_tc_fail_errno("open root");
	if (rump_sys_write(fd, &fd, sizeof(fd)) != sizeof(fd))
		atf_tc_fail_errno("write to root");
	rv = rump_sys_mkdir(FSTEST_MNTNAME "/test", 0777);
	ATF_REQUIRE(errno == ENOTDIR);
	ATF_REQUIRE(rv == -1);
	rump_sys_close(fd);

	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(root_lnk);
ATF_TC_HEAD(root_lnk, tc)
{

	atf_tc_set_md_var(tc, "descr", "root is a symbolic link");
}

#define LINKSTR "/path/to/nowhere"
ATF_TC_BODY(root_lnk, tc)
{
	MAKEOPTS("-r", "lnk " LINKSTR);
	void *args;
	char buf[PATH_MAX];
	ssize_t len;

	FSTEST_CONSTRUCTOR_FSPRIV(tc, puffs, args, theopts);

	if ((len = rump_sys_readlink(FSTEST_MNTNAME, buf, sizeof(buf)-1)) == -1)
		atf_tc_fail_errno("readlink");
	buf[len] = '\0';

	ATF_REQUIRE_STREQ(buf, LINKSTR);

#if 0 /* XXX: unmount uses FOLLOW */
	if (rump_sys_unmount("/mp", 0) == -1)
		atf_tc_fail_errno("unmount");
#endif
}

ATF_TC(root_fifo);
ATF_TC_HEAD(root_fifo, tc)
{

	atf_tc_set_md_var(tc, "descr", "root is a symbolic link");
}

#define MAGICSTR "nakit ja muusiperunat maustevoilla"
static void *
dofifow(void *arg)
{
	int fd = (int)(uintptr_t)arg;
	char buf[512];

	printf("writing\n");
	strcpy(buf, MAGICSTR);
	if (rump_sys_write(fd, buf, strlen(buf)+1) != strlen(buf)+1)
		atf_tc_fail_errno("write to fifo");

	return NULL;
}

ATF_TC_BODY(root_fifo, tc)
{
	MAKEOPTS("-r", "fifo");
	void *args;
	pthread_t pt;
	char buf[512];
	int fd;

	FSTEST_CONSTRUCTOR_FSPRIV(tc, puffs, args, theopts);

	fd = rump_sys_open(FSTEST_MNTNAME, O_RDWR);
	if (fd == -1)
		atf_tc_fail_errno("open fifo");

	pthread_create(&pt, NULL, dofifow, (void *)(uintptr_t)fd);

	memset(buf, 0, sizeof(buf));
	if (rump_sys_read(fd, buf, sizeof(buf)) == -1)
		atf_tc_fail_errno("read fifo");

	ATF_REQUIRE_STREQ(buf, MAGICSTR);
	rump_sys_close(fd);

	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(root_chrdev);
ATF_TC_HEAD(root_chrdev, tc)
{

	atf_tc_set_md_var(tc, "descr", "root is /dev/null");
}

ATF_TC_BODY(root_chrdev, tc)
{
	MAKEOPTS("-r", "chr 2 2");
	void *args;
	ssize_t rv;
	char buf[512];
	int fd;

	FSTEST_CONSTRUCTOR_FSPRIV(tc, puffs, args, theopts);

	fd = rump_sys_open(FSTEST_MNTNAME, O_RDWR);
	if (fd == -1)
		atf_tc_fail_errno("open null");

	rv = rump_sys_write(fd, buf, sizeof(buf));
	ATF_REQUIRE(rv == sizeof(buf));

	rv = rump_sys_read(fd, buf, sizeof(buf));
	ATF_REQUIRE(rv == 0);

	rump_sys_close(fd);

	FSTEST_DESTRUCTOR(tc, puffs, args);
}

/*
 * Inactive/reclaim tests
 */

ATF_TC(inactive_basic);
ATF_TC_HEAD(inactive_basic, tc)
{

	atf_tc_set_md_var(tc, "descr", "inactive gets called");
}

ATF_TC_BODY(inactive_basic, tc)
{
	struct puffstestargs *pargs;
	void *args;
	int fd;

	FSTEST_CONSTRUCTOR(tc, puffs, args);
	FSTEST_ENTER();
	pargs = args;

	fd = rump_sys_open("file", O_CREAT | O_RDWR, 0777);
	if (fd == -1)
		atf_tc_fail_errno("create");

	/* none yet */
	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE], 0);

	rump_sys_close(fd);

	/* one for file */
	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE], 1);

	FSTEST_EXIT();

	/* another for the mountpoint */
	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE], 2);

	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(inactive_reclaim);
ATF_TC_HEAD(inactive_reclaim, tc)
{

	atf_tc_set_md_var(tc, "descr", "inactive/reclaim gets called");
}

ATF_TC_BODY(inactive_reclaim, tc)
{
	struct puffstestargs *pargs;
	void *args;
	int fd;

	FSTEST_CONSTRUCTOR(tc, puffs, args);
	FSTEST_ENTER();
	pargs = args;

	fd = rump_sys_open("file", O_CREAT | O_RDWR, 0777);
	if (fd == -1)
		atf_tc_fail_errno("create");

	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE], 0);

	if (rump_sys_unlink("file") == -1)
		atf_tc_fail_errno("remove");

	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE], 0);
	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 0);

	rump_sys_close(fd);
	syncbar(FSTEST_MNTNAME);

	ATF_REQUIRE(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE] > 0);
	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 1);

	FSTEST_EXIT();
	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(reclaim_hardlink);
ATF_TC_HEAD(reclaim_hardlink, tc)
{

	atf_tc_set_md_var(tc, "descr", "reclaim gets called only after "
	    "final link is gone");
}

ATF_TC_BODY(reclaim_hardlink, tc)
{
	struct puffstestargs *pargs;
	void *args;
	int fd;
	int ianow;

	FSTEST_CONSTRUCTOR(tc, puffs, args);
	FSTEST_ENTER();
	pargs = args;

	fd = rump_sys_open("file", O_CREAT | O_RDWR, 0777);
	if (fd == -1)
		atf_tc_fail_errno("create");

	if (rump_sys_link("file", "anotherfile") == -1)
		atf_tc_fail_errno("create link");
	rump_sys_close(fd);

	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 0);

	/* unlink first hardlink */
	if (rump_sys_unlink("file") == -1)
		atf_tc_fail_errno("unlink 1");

	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 0);
	ianow = pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE];

	/* unlink second hardlink */
	if (rump_sys_unlink("anotherfile") == -1)
		atf_tc_fail_errno("unlink 2");

	syncbar(FSTEST_MNTNAME);

	ATF_REQUIRE(ianow < pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE]);
	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 1);

	FSTEST_EXIT();
	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(unlink_accessible);
ATF_TC_HEAD(unlink_accessible, tc)
{

	atf_tc_set_md_var(tc, "descr", "open file is accessible after "
	    "having been unlinked");
}

ATF_TC_BODY(unlink_accessible, tc)
{
	MAKEOPTS("-i", "-o", "nopagecache");
	struct puffstestargs *pargs;
	void *args;
	char buf[512];
	int fd, ianow;

	assert(sizeof(buf) > sizeof(MAGICSTR));

	FSTEST_CONSTRUCTOR_FSPRIV(tc, puffs, args, theopts);
	FSTEST_ENTER();
	pargs = args;

	fd = rump_sys_open("file", O_CREAT | O_RDWR, 0777);
	if (fd == -1)
		atf_tc_fail_errno("create");

	if (rump_sys_write(fd, MAGICSTR, sizeof(MAGICSTR)) != sizeof(MAGICSTR))
		atf_tc_fail_errno("write");
	if (rump_sys_unlink("file") == -1)
		atf_tc_fail_errno("unlink");

	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 0);
	ianow = pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE];

	if (rump_sys_pread(fd, buf, sizeof(buf), 0) == -1)
		atf_tc_fail_errno("read");
	rump_sys_close(fd);

	syncbar(FSTEST_MNTNAME);

	ATF_REQUIRE_EQ(pargs->pta_vn_toserv_ops[PUFFS_VN_RECLAIM], 1);
	ATF_REQUIRE(pargs->pta_vn_toserv_ops[PUFFS_VN_INACTIVE] > ianow);

	ATF_REQUIRE_STREQ(buf, MAGICSTR);

	FSTEST_EXIT();
	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TC(signals);
ATF_TC_HEAD(signals, tc)
{

	atf_tc_set_md_var(tc, "descr", "Checks that sending a signal can "
	    "cause an interrupt to puffs wait");
}

extern struct proc *rumpns_initproc;
extern void rumpns_psignal(struct proc *, int);
extern void rumpns_sigclearall(struct proc *, void *, void *);
ATF_TC_BODY(signals, tc)
{
	struct stat sb;
	void *args;

	rump_boot_setsigmodel(RUMP_SIGMODEL_RECORD);

	FSTEST_CONSTRUCTOR(tc, puffs, args);
	FSTEST_ENTER();
	RL(rump_sys_stat(".", &sb));

	/* send SIGUSR1, should not affect puffs ops */
	rump_schedule();
	rumpns_psignal(rumpns_initproc, SIGUSR1);
	rump_unschedule();
	RL(rump_sys_stat(".", &sb));

	/* send SIGTERM, should get EINTR */
	rump_schedule();
	rumpns_psignal(rumpns_initproc, SIGTERM);
	rump_unschedule();
	ATF_REQUIRE_ERRNO(EINTR, rump_sys_stat(".", &sb) == -1);

	/* clear sigmask so that we can unmount */
	rump_schedule();
	rumpns_sigclearall(rumpns_initproc, NULL, NULL);
	rump_unschedule();

	FSTEST_EXIT();
	FSTEST_DESTRUCTOR(tc, puffs, args);
}

ATF_TP_ADD_TCS(tp)
{

	ATF_TP_ADD_TC(tp, mount);

	ATF_TP_ADD_TC(tp, root_fifo);
	ATF_TP_ADD_TC(tp, root_lnk);
	ATF_TP_ADD_TC(tp, root_reg);
	ATF_TP_ADD_TC(tp, root_chrdev);

	ATF_TP_ADD_TC(tp, inactive_basic);
	ATF_TP_ADD_TC(tp, inactive_reclaim);
	ATF_TP_ADD_TC(tp, reclaim_hardlink);
	ATF_TP_ADD_TC(tp, unlink_accessible);

	ATF_TP_ADD_TC(tp, signals);

	return atf_no_error();
}