/*
* Copyright (c) 2003, 2011, Oracle and/or its affiliates. All rights reserved.
*
* U.S. Government Rights - Commercial software. Government users are subject
* to the Sun Microsystems, Inc. standard license agreement and applicable
* provisions of the FAR and its supplements.
*
*
* This distribution may include materials developed by third parties. Sun,
* Sun Microsystems, the Sun logo and Solaris are trademarks or registered
* trademarks of Sun Microsystems, Inc. in the U.S. and other countries.
*
*/



/*
 * nfsstat: Network File System statistics for client RPC
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdarg.h>
#include <string.h>
#include <errno.h>
#include <fcntl.h>
#include <kvm.h>
#include <kstat.h>
#include <sys/param.h>
#include <sys/types.h>
#include <sys/t_lock.h>
#include <sys/tiuser.h>
#include <sys/statvfs.h>
#include <sys/mntent.h>
#include <sys/mnttab.h>
#include <sys/sysmacros.h>
#include <sys/mkdev.h>
#include <rpc/types.h>
#include <rpc/xdr.h>
#include <rpc/auth.h>
#include <rpc/clnt.h>
#include <nfs/nfs.h>
#include <nfs/nfs_clnt.h>

static kstat_ctl_t *kc = NULL;		/* libkstat cookie */
static kstat_t *rpc_clts_client_kstat;
static kstat_t *rpc_cots_client_kstat;

static void getstats(void);
static void setup();
static void clear_setup_data();
static  void    fail(int, char *, ...);

static kid_t safe_kstat_read(kstat_ctl_t *, kstat_t *, void *);

#define	MAX_COLUMNS	80

static int field_width = 0;
static int ncolumns;

int cflag = 0;		/* client stats */
int rflag = 0;		/* rpc stats */
int zflag = 0;		/* zero stats after printing */

static void stat_width(kstat_t *);

static void
getstats(void)
{

	if (rpc_clts_client_kstat != NULL) {
		safe_kstat_read(kc, rpc_clts_client_kstat, NULL);
		stat_width(rpc_clts_client_kstat);
	}
	if (rpc_cots_client_kstat != NULL) {
		safe_kstat_read(kc, rpc_cots_client_kstat, NULL);
		stat_width(rpc_cots_client_kstat);
	}
}



static void
clear_setup_data()
{
	/*
	 * Verify and close the kstat control structure 
	 */
	if (kc)
		kstat_close(kc);

	/*
	 * Set all the related variables to null
	 */
	kc = (kstat_ctl_t *)NULL;
	rpc_clts_client_kstat = (kstat_t *)NULL;
	rpc_cots_client_kstat = (kstat_t *)NULL;
}

	
static void
setup()
{
	/* Initialize kstat control structures */
	if ((kc = kstat_open()) == NULL)
		fail(1, "kstat_open(): can't open /dev/kstat");

	/* Find a kstat by name */
	rpc_clts_client_kstat = kstat_lookup(kc, "unix", 0, "rpc_clts_client");
	rpc_cots_client_kstat = kstat_lookup(kc, "unix", 0, "rpc_cots_client");
}

static void
stat_width(kstat_t *req)
{
	int i, nreq, len;
	char fixlen[128];
	kstat_named_t *knp;

	knp = KSTAT_NAMED_PTR(req);
	nreq = req->ks_ndata;

	for (i = 0; i < nreq; i++) {
		len = strlen(knp[i].name) + 1;
		if (field_width < len)
			field_width = len;
		(void) sprintf(fixlen, "%lu", knp[i].value.ul);
		len = strlen(fixlen) + 1;
		if (field_width < len)
			field_width = len;
	}
}

/*
 * Get Client RPC data
 *   This function is called by Health Monitoring Module
 */
krgetclientrpcdetail(double *calls, int *badcalls, int *retrans, int *badxids, int *timeouts, int *newcreds, int *badverfs, int *timers, int *nomem, int *cantsend) 
{
        int i, j, nreq;
        char fixlen[128];
	char buf[1024];
        kstat_named_t *knp;
	cflag++;
	rflag++;

        buf[0]= '\0';

	setup();

	getstats();

	ncolumns = (MAX_COLUMNS - 1) / field_width;

	/* Kstat lookup client rpc info */
	if (rpc_clts_client_kstat != NULL) {

		knp = KSTAT_NAMED_PTR(rpc_clts_client_kstat);
        	nreq = rpc_clts_client_kstat->ks_ndata;

        	for (i = 0; i < nreq; i += ncolumns) {
               	 	/* Don't print heading */
               	 	/*
               		for (j = i; j < MIN(i + ncolumns, nreq); j++) {
             	           printf("%-*s", field_width, knp[j].name);
               	 	}
                	printf("\n");
                	*/
                	for (j = i; j < MIN(i + ncolumns, nreq); j++) {
#if (defined SOLARIS2_5 || defined SOLARIS2_6)
                        	(void) sprintf(fixlen, "%lu ", knp[j].value.ul);
#else
                                (void) sprintf(fixlen, "%llu ", knp[j].value.ui64);
#endif

                        	strcat(buf, fixlen);
                	}
        	}

	}
	*calls = atof(strtok(buf, " "));
	*badcalls = atoi (strtok(NULL, " "));
	*retrans = atoi (strtok(NULL, " "));
	*badxids = atoi (strtok(NULL, " "));
	*timeouts = atoi (strtok(NULL, " "));
	*newcreds = atoi (strtok(NULL, " "));
	*badverfs = atoi (strtok(NULL, " "));
	*timers = atoi (strtok(NULL, " "));
	*nomem = atoi (strtok(NULL, " "));
	*cantsend = atoi (strtok(NULL, " "));

	/*
	 * The following function closes the kstat control structure
	 *  and sets the related variables to NULL.
	 */
	clear_setup_data();

	return (0);
}

#define	MIN(a, b)	((a) < (b) ? (a) : (b))

static void
fail(int do_perror, char *message, ...)
{
	va_list args;

	va_start(args, message);
	fprintf(stderr, "nfsstat: ");
	vfprintf(stderr, message, args);
	va_end(args);
	if (do_perror)
		fprintf(stderr, ": %s", strerror(errno));
	fprintf(stderr, "\n");
	exit(1);
}

kid_t
safe_kstat_read(kstat_ctl_t *kc, kstat_t *ksp, void *data)
{
	kid_t kstat_chain_id = kstat_read(kc, ksp, data);

	if (kstat_chain_id == -1)
		fail(1, "kstat_read(%x, '%s') failed", kc, ksp->ks_name);
	return (kstat_chain_id);
}