/*

 * 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.

 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#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);

}