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

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

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

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

#include <fs/ptyfs/ptyfs.h>
#include <miscfs/nullfs/null.h>

#include "h_macros.h"

static void
mountptyfs(const char *mp, int flags)
{
        struct ptyfs_args args;

        if (rump_sys_mkdir(mp, 0777) == -1) {
                if (errno != EEXIST)
                        atf_tc_fail_errno("null create %s", mp);
        }
        memset(&args, 0, sizeof(args));
        args.version = PTYFS_ARGSVERSION;
        args.mode = 0777;
        if (rump_sys_mount(MOUNT_PTYFS, mp, flags, &args, sizeof(args)) == -1)
                atf_tc_fail_errno("could not mount ptyfs");
}

static void
mountnull(const char *what, const char *mp, int flags)
{
        struct null_args nargs;

        if (rump_sys_mkdir(what, 0777) == -1) {
                if (errno != EEXIST)
                        atf_tc_fail_errno("null create %s", what);
        }
        if (rump_sys_mkdir(mp, 0777) == -1) {
                if (errno != EEXIST)
                        atf_tc_fail_errno("null create %s", mp);
        }
        memset(&nargs, 0, sizeof(nargs));
        nargs.nulla_target = __UNCONST(what);
        if (rump_sys_mount(MOUNT_NULL, mp, flags, &nargs, sizeof(nargs)) == -1)
                atf_tc_fail_errno("could not mount nullfs");
}

ATF_TC(nullrevoke);
ATF_TC_HEAD(nullrevoke, tc)
{
        atf_tc_set_md_var(tc, "descr", "null mount ptyfs and revoke");
}

ATF_TC_BODY(nullrevoke, tc)
{
        char path[MAXPATHLEN];
        struct ptmget ptg;
        int ptm;

        rump_init();

        /*
         * mount /dev/pts
         */
        mountptyfs("/dev/pts", 0);

        /*
         * null mount /dev/pts to /null/dev/pts
         */
        if (rump_sys_mkdir("/null", 0777) == -1) {
                if (errno != EEXIST)
                        atf_tc_fail_errno("null create /null");
        }
        if (rump_sys_mkdir("/null/dev", 0777) == -1) {
                if (errno != EEXIST)
                        atf_tc_fail_errno("null create /null/dev");
        }

        mountnull("/dev/pts", "/null/dev/pts", 0);

        /*
         * get slave/master pair.
         */
        ptm = rump_sys_open("/dev/ptm", O_RDWR);
        if (rump_sys_ioctl(ptm, TIOCPTMGET, &ptg) == -1)
                atf_tc_fail_errno("get pty");

        /*
         * Build nullfs path to slave.
         */
        strcpy(path, "/null");
        strcat(path, ptg.sn);

        /*
         * Open slave tty via nullfs.
         */
        if (rump_sys_open(path, O_RDWR) == -1)
                atf_tc_fail_errno("slave null open");

        /*
         * Close slave opened with /dev/ptm.  Need purely non-null refs to it.
         */
        rump_sys_close(ptg.sfd);

        /* revoke slave tty. */
        rump_sys_revoke(path);

        /* done */
}

ATF_TP_ADD_TCS(tp)
{

        ATF_TP_ADD_TC(tp, nullrevoke);

        return atf_no_error();
}