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