/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <strings.h>
#include <rpc/rpc.h>
#include <stdlib.h>
#include <sys/param.h>
#include <rpcsvc/mount.h>

#include "rpcsvc/nfs_prot.h"

char sharedpath[MAXPATHLEN];
fhandle3 *rootfh;

/*
 * The waiting() function returns the value passed in, until something
 * external modifies it.  In this case, the D script tst.call.d will
 * modify the value of *a, and thus break the while loop in dotest().
 *
 * This serves the purpose of not making the RPC calls until tst.call.d
 * is active.  Thus, the probes in tst.call.d can fire as a result of
 * the RPC call in dotest().
 */

int
waiting(volatile int *a)
{
        return (*a);
}

static void
getattr_arginit(void *argp)
{
        GETATTR3args *args = argp;

        args->object.data.data_len = rootfh->fhandle3_len;
        args->object.data.data_val = rootfh->fhandle3_val;
}

static void
setattr_arginit(void *argp)
{
        SETATTR3args *args = argp;

        bzero(args, sizeof (*args));
        args->object.data.data_len = rootfh->fhandle3_len;
        args->object.data.data_val = rootfh->fhandle3_val;
}

static void
lookup_arginit(void *argp)
{
        LOOKUP3args *args = argp;

        args->what.name = "giant-skunk";
        args->what.dir.data.data_len = rootfh->fhandle3_len;
        args->what.dir.data.data_val = rootfh->fhandle3_val;
}

static void
access_arginit(void *argp)
{
        ACCESS3args *args = argp;

        args->object.data.data_len = rootfh->fhandle3_len;
        args->object.data.data_val = rootfh->fhandle3_val;
}

static void
commit_arginit(void *argp)
{
        COMMIT3args *args = argp;

        bzero(args, sizeof (*args));
        args->file.data.data_len = rootfh->fhandle3_len;
        args->file.data.data_val = rootfh->fhandle3_val;
}

static void
create_arginit(void *argp)
{
        CREATE3args *args = argp;

        bzero(args, sizeof (*args));
        args->where.name = "pinky-blue";
        args->where.dir.data.data_len = rootfh->fhandle3_len;
        args->where.dir.data.data_val = rootfh->fhandle3_val;
}

static void
fsinfo_arginit(void *argp)
{
        FSINFO3args *args = argp;

        args->fsroot.data.data_len = rootfh->fhandle3_len;
        args->fsroot.data.data_val = rootfh->fhandle3_val;
}

static void
fsstat_arginit(void *argp)
{
        FSSTAT3args *args = argp;

        args->fsroot.data.data_len = rootfh->fhandle3_len;
        args->fsroot.data.data_val = rootfh->fhandle3_val;
}

static void
link_arginit(void *argp)
{
        LINK3args *args = argp;

        args->file.data.data_len = rootfh->fhandle3_len;
        args->file.data.data_val = rootfh->fhandle3_val;
        args->link.dir.data.data_len = rootfh->fhandle3_len;
        args->link.dir.data.data_val = rootfh->fhandle3_val;
        args->link.name = "samf";
}

static void
mkdir_arginit(void *argp)
{
        MKDIR3args *args = argp;

        bzero(args, sizeof (*args));
        args->where.dir.data.data_len = rootfh->fhandle3_len;
        args->where.dir.data.data_val = rootfh->fhandle3_val;
        args->where.name = "cookie";
}

static void
mknod_arginit(void *argp)
{
        MKNOD3args *args = argp;

        bzero(args, sizeof (*args));
        args->where.dir.data.data_len = rootfh->fhandle3_len;
        args->where.dir.data.data_val = rootfh->fhandle3_val;
        args->where.name = "pookie";
}

static void
null_arginit(void *argp)
{
}

static void
pathconf_arginit(void *argp)
{
        PATHCONF3args *args = argp;

        args->object.data.data_len = rootfh->fhandle3_len;
        args->object.data.data_val = rootfh->fhandle3_val;
}

static void
read_arginit(void *argp)
{
        READ3args *args = argp;

        bzero(args, sizeof (*args));
        args->file.data.data_len = rootfh->fhandle3_len;
        args->file.data.data_val = rootfh->fhandle3_val;
}

static void
readdir_arginit(void *argp)
{
        READDIR3args *args = argp;

        bzero(args, sizeof (*args));
        args->dir.data.data_len = rootfh->fhandle3_len;
        args->dir.data.data_val = rootfh->fhandle3_val;
        args->count = 1024;
}

static void
readdirplus_arginit(void *argp)
{
        READDIRPLUS3args *args = argp;

        bzero(args, sizeof (*args));
        args->dir.data.data_len = rootfh->fhandle3_len;
        args->dir.data.data_val = rootfh->fhandle3_val;
        args->dircount = 1024;
        args->maxcount = 1024;
}

static void
readlink_arginit(void *argp)
{
        READLINK3args *args = argp;

        args->symlink.data.data_len = rootfh->fhandle3_len;
        args->symlink.data.data_val = rootfh->fhandle3_val;
}

static void
remove_arginit(void *argp)
{
        REMOVE3args *args = argp;

        args->object.dir.data.data_len = rootfh->fhandle3_len;
        args->object.dir.data.data_val = rootfh->fhandle3_val;
        args->object.name = "antelope";
}

static void
rename_arginit(void *argp)
{
        RENAME3args *args = argp;

        args->from.dir.data.data_len = rootfh->fhandle3_len;
        args->from.dir.data.data_val = rootfh->fhandle3_val;
        args->from.name = "walter";
        args->to.dir.data.data_len = rootfh->fhandle3_len;
        args->to.dir.data.data_val = rootfh->fhandle3_val;
        args->to.name = "wendy";
}

static void
rmdir_arginit(void *argp)
{
        RMDIR3args *args = argp;

        args->object.dir.data.data_len = rootfh->fhandle3_len;
        args->object.dir.data.data_val = rootfh->fhandle3_val;
        args->object.name = "bunny";
}

static void
symlink_arginit(void *argp)
{
        SYMLINK3args *args = argp;

        bzero(args, sizeof (*args));
        args->where.dir.data.data_len = rootfh->fhandle3_len;
        args->where.dir.data.data_val = rootfh->fhandle3_val;
        args->where.name = "parlor";
        args->symlink.symlink_data = "interior";
}

static void
write_arginit(void *argp)
{
        WRITE3args *args = argp;

        bzero(args, sizeof (*args));
        args->file.data.data_len = rootfh->fhandle3_len;
        args->file.data.data_val = rootfh->fhandle3_val;
}

typedef void (*call3_arginit_t)(void *);

typedef struct {
        call3_arginit_t arginit;
        rpcproc_t proc;
        xdrproc_t xdrargs;
        size_t argsize;
        xdrproc_t xdrres;
        size_t ressize;
} call3_test_t;
call3_test_t call3_tests[] = {
        {getattr_arginit, NFSPROC3_GETATTR, xdr_GETATTR3args,
            sizeof (GETATTR3args), xdr_GETATTR3res, sizeof (GETATTR3res)},
        {setattr_arginit, NFSPROC3_SETATTR, xdr_SETATTR3args,
            sizeof (SETATTR3args), xdr_SETATTR3res, sizeof (SETATTR3res)},
        {lookup_arginit, NFSPROC3_LOOKUP, xdr_LOOKUP3args,
            sizeof (LOOKUP3args), xdr_LOOKUP3res, sizeof (LOOKUP3res)},
        {access_arginit, NFSPROC3_ACCESS, xdr_ACCESS3args,
            sizeof (ACCESS3args), xdr_ACCESS3res, sizeof (ACCESS3res)},
        {commit_arginit, NFSPROC3_COMMIT, xdr_COMMIT3args,
            sizeof (COMMIT3args), xdr_COMMIT3res, sizeof (COMMIT3res)},
        {create_arginit, NFSPROC3_CREATE, xdr_CREATE3args,
            sizeof (CREATE3args), xdr_CREATE3res, sizeof (CREATE3res)},
        {fsinfo_arginit, NFSPROC3_FSINFO, xdr_FSINFO3args,
            sizeof (FSINFO3args), xdr_FSINFO3res, sizeof (FSINFO3res)},
        {fsstat_arginit, NFSPROC3_FSSTAT, xdr_FSSTAT3args,
            sizeof (FSSTAT3args), xdr_FSSTAT3res, sizeof (FSSTAT3res)},
        {link_arginit, NFSPROC3_LINK, xdr_LINK3args,
            sizeof (LINK3args), xdr_LINK3res, sizeof (LINK3res)},
        {mkdir_arginit, NFSPROC3_MKDIR, xdr_MKDIR3args,
            sizeof (MKDIR3args), xdr_MKDIR3res, sizeof (MKDIR3res)},
        {mknod_arginit, NFSPROC3_MKNOD, xdr_MKNOD3args,
            sizeof (MKNOD3args), xdr_MKNOD3res, sizeof (MKNOD3res)},
        /*
         * NULL proc is special.  Rather than special case its zero-sized
         * args/results, we give it a small nonzero size, so as to not
         * make realloc() do the wrong thing.
         */
        {null_arginit, NFSPROC3_NULL, xdr_void, sizeof (int), xdr_void,
            sizeof (int)},
        {pathconf_arginit, NFSPROC3_PATHCONF, xdr_PATHCONF3args,
            sizeof (PATHCONF3args), xdr_PATHCONF3res, sizeof (PATHCONF3res)},
        {read_arginit, NFSPROC3_READ, xdr_READ3args,
            sizeof (READ3args), xdr_READ3res, sizeof (READ3res)},
        {readdir_arginit, NFSPROC3_READDIR, xdr_READDIR3args,
            sizeof (READDIR3args), xdr_READDIR3res, sizeof (READDIR3res)},
        {readdirplus_arginit, NFSPROC3_READDIRPLUS, xdr_READDIRPLUS3args,
            sizeof (READDIRPLUS3args), xdr_READDIRPLUS3res,
            sizeof (READDIRPLUS3res)},
        {readlink_arginit, NFSPROC3_READLINK, xdr_READLINK3args,
            sizeof (READLINK3args), xdr_READLINK3res, sizeof (READLINK3res)},
        {remove_arginit, NFSPROC3_REMOVE, xdr_REMOVE3args,
            sizeof (REMOVE3args), xdr_REMOVE3res, sizeof (REMOVE3res)},
        {rename_arginit, NFSPROC3_RENAME, xdr_RENAME3args,
            sizeof (RENAME3args), xdr_RENAME3res, sizeof (RENAME3res)},
        {rmdir_arginit, NFSPROC3_RMDIR, xdr_RMDIR3args,
            sizeof (RMDIR3args), xdr_RMDIR3res, sizeof (RMDIR3res)},
        {symlink_arginit, NFSPROC3_SYMLINK, xdr_SYMLINK3args,
            sizeof (SYMLINK3args), xdr_SYMLINK3res, sizeof (SYMLINK3res)},
        {write_arginit, NFSPROC3_WRITE, xdr_WRITE3args,
            sizeof (WRITE3args), xdr_WRITE3res, sizeof (WRITE3res)},
        {NULL}
};

int
dotest(void)
{
        CLIENT *client, *mountclient;
        AUTH *auth;
        struct timeval timeout;
        caddr_t args, res;
        enum clnt_stat status;
        rpcproc_t proc;
        call3_test_t *test;
        void *argbuf = NULL;
        void *resbuf = NULL;
        struct mountres3 mountres3;
        char *sp;
        volatile int a = 0;

        while (waiting(&a) == 0)
                continue;

        timeout.tv_sec = 30;
        timeout.tv_usec = 0;

        mountclient = clnt_create("localhost", MOUNTPROG, MOUNTVERS3, "tcp");
        if (mountclient == NULL) {
                clnt_pcreateerror("clnt_create mount");
                return (1);
        }
        auth = authsys_create_default();
        mountclient->cl_auth = auth;
        sp = sharedpath;
        bzero(&mountres3, sizeof (mountres3));
        status = clnt_call(mountclient, MOUNTPROC_MNT,
            xdr_dirpath, (char *)&sp,
            xdr_mountres3, (char *)&mountres3,
            timeout);
        if (status != RPC_SUCCESS) {
                clnt_perror(mountclient, "mnt");
                return (1);
        }
        if (mountres3.fhs_status != 0) {
                fprintf(stderr, "MOUNTPROG/MOUNTVERS3 failed %d\n",
                    mountres3.fhs_status);
                return (1);
        }
        rootfh = &mountres3.mountres3_u.mountinfo.fhandle;

        client = clnt_create("localhost", NFS3_PROGRAM, NFS_V3, "tcp");
        if (client == NULL) {
                clnt_pcreateerror("clnt_create");
                return (1);
        }
        client->cl_auth = auth;

        for (test = call3_tests; test->arginit; ++test) {
                argbuf = realloc(argbuf, test->argsize);
                resbuf = realloc(resbuf, test->ressize);
                if ((argbuf == NULL) || (resbuf == NULL)) {
                        perror("realloc() failed");
                        return (1);
                }
                (test->arginit)(argbuf);
                bzero(resbuf, test->ressize);
                status = clnt_call(client, test->proc,
                    test->xdrargs, argbuf,
                    test->xdrres, resbuf,
                    timeout);
                if (status != RPC_SUCCESS)
                        clnt_perror(client, "call");
        }

        status = clnt_call(mountclient, MOUNTPROC_UMNT,
            xdr_dirpath, (char *)&sp,
            xdr_void, NULL,
            timeout);
        if (status != RPC_SUCCESS)
                clnt_perror(mountclient, "umnt");

        return (0);
}

/*ARGSUSED*/
int
main(int argc, char **argv)
{
        char shareline[BUFSIZ], unshareline[BUFSIZ];
        int rc;

        (void) snprintf(sharedpath, sizeof (sharedpath),
            "/tmp/nfsv3test.%d", getpid());
        (void) snprintf(shareline, sizeof (shareline),
            "mkdir %s ; share %s", sharedpath, sharedpath);
        (void) snprintf(unshareline, sizeof (unshareline),
            "unshare %s ; rmdir %s", sharedpath, sharedpath);

        (void) system(shareline);
        rc = dotest();
        (void) system(unshareline);

        return (rc);
}