/*
 * Copyright (c) 1995-2001 Silicon Graphics, Inc.  All Rights Reserved.
 * Copyright (c) 2016 Ken McDonell.  All Rights Reserved.
 *
 * Hard pmFetch loop, sort of like the core of pmlogger.
 */

#include <pcp/pmapi.h>
#include <sys/time.h>
#ifdef HAVE_SYS_TIMES_H
#include <sys/times.h>
#endif
#include <unistd.h>

static const char **namelist;
static pmID	*pmidlist;
static int	numpmid;

static void
dometric(const char *name)
{
    numpmid++;

    namelist = (const char **)realloc(namelist, numpmid*sizeof(char *));
    if (namelist == NULL) {
	pmNoMem("dometric: namelist", numpmid*sizeof(char *), PM_FATAL_ERR);
	/* NOTREACHED */
    }
    pmidlist = (pmID *)realloc(pmidlist, numpmid*sizeof(pmID));
    if (pmidlist == NULL) {
	pmNoMem("dometric: pmidlist", numpmid*sizeof(pmID), PM_FATAL_ERR);
	/* NOTREACHED */
    }
    namelist[numpmid-1] = strdup(name);
    if (namelist[numpmid-1] == NULL) {
	pmNoMem("dometric: namelist[]", strlen(name)+1, PM_FATAL_ERR);
	/* NOTREACHED */
    }
}

static void
dumpstate(int flags)
{
    char	buffer[64];

    sprintf(buffer, "PMCD state change(s): ");
    if (flags & PMCD_AGENT_CHANGE)
	strcat(buffer, "agent ");
    if (flags & PMCD_LABEL_CHANGE)
	strcat(buffer, "label ");
    if (flags & PMCD_NAMES_CHANGE)
	strcat(buffer, "names ");

    if (flags)
	buffer[strlen(buffer)-1] = '\0';
    else
	strcat(buffer, "none");

    puts(buffer);
}

int
main(int argc, char **argv)
{
    int		c;
    int		sts;
    int		errflag = 0;
    int		type = 0;
    char	*host = NULL;			/* pander to gcc */
    char 	*configfile = NULL;
    int		samples = 1000;
    int		interval = 0;
    int		verbose = 0;
    int		save_samples;
    char	*endnum;
    pmResult	*rp;
    struct tms	now, then;
    clock_t	ticks;
    long	hz = sysconf(_SC_CLK_TCK);

    pmSetProgname(argv[0]);

    while ((c = getopt(argc, argv, "c:D:h:Ln:s:t:v?")) != EOF) {
	switch (c) {

	case 'c':	/* configfile */
	    if (configfile != NULL) {
		fprintf(stderr, "%s: at most one -c option allowed\n", pmGetProgname());
		errflag++;
	    }
	    configfile = optarg;
	    break;	

	case 'D':	/* debug options */
	    sts = pmSetDebug(optarg);
	    if (sts < 0) {
		fprintf(stderr, "%s: unrecognized debug options specification (%s)\n",
		    pmGetProgname(), optarg);
		errflag++;
	    }
	    break;

	case 'h':	/* contact PMCD on this hostname */
	    if (type != 0) {
		fprintf(stderr, "%s: at most one of -h and -L allowed\n", pmGetProgname());
		errflag++;
	    }
	    host = optarg;
	    type = PM_CONTEXT_HOST;
	    break;

	case 'L':	/* LOCAL context mode, not PMCD */
	    if (type != 0) {
		fprintf(stderr, "%s: at most one of -h and -L allowed\n", pmGetProgname());
		errflag++;
	    }
	    host = NULL;
	    type = PM_CONTEXT_LOCAL;
	    putenv("PMDA_LOCAL_PROC=");		/* if proc PMDA needed */
	    putenv("PMDA_LOCAL_SAMPLE=");	/* if sampledso PMDA needed */
	    break;

	case 's':	/* sample count */
	    samples = (int)strtol(optarg, &endnum, 10);
	    if (*endnum != '\0') {
		fprintf(stderr, "%s: -s requires numeric argument\n", pmGetProgname());
		errflag++;
	    }
	    break;

	case 't':	/* sample delay */
	    interval = (int)strtol(optarg, &endnum, 10);
	    if (*endnum != '\0' || interval < 0) {
		fprintf(stderr, "%s: -t requires numeric argument\n", pmGetProgname());
		errflag++;
	    }
	    break;

	case 'v':	/* report collector state (change) on each fetch */
	    verbose++;
	    break;

	case '?':
	default:
	    errflag++;
	    break;
	}
    }

    if (errflag) {
	fprintf(stderr,
"Usage: %s [options] [metrics ...]\n\
\n\
Options:\n\
  -c configfile  file to load configuration from\n\
  -h host        metrics source is PMCD on host\n\
  -L             use local context instead of PMCD\n\
  -s samples     terminate after this many samples [default 1000]\n\
  -t interval    sleep for interval seconds samples [default 0]\n\
  -v             verbose mode, report any state changes\n", pmGetProgname());
        exit(1);
    }

    if (type == 0) {
	type = PM_CONTEXT_HOST;
	host = "unix:";
    }

    if ((sts = pmNewContext(type, host)) < 0) {
	if (type == PM_CONTEXT_HOST)
	    fprintf(stderr, "%s: Cannot connect to PMCD on host \"%s\": %s\n",
		pmGetProgname(), host, pmErrStr(sts));
	else
	    fprintf(stderr, "%s: Cannot open archive \"%s\": %s\n",
		pmGetProgname(), host, pmErrStr(sts));
	exit(1);
    }

    /* command line metrics are argv[optind] ... argv[argc-1] */
    while (optind < argc) {
	pmTraversePMNS(argv[optind], dometric);
	optind++;
    }

    if (configfile != NULL) {
	FILE	*f;
	char	buf[1024];
	char	*cp;
	/*
	 * metric names (terminal or non-terminal) are one per line in
	 * configfile
	 */

	if ((f = fopen(configfile, "r")) == NULL) {
	    fprintf(stderr, "fopen: %s: failed: %s\n", configfile, pmErrStr(-oserror()));
	    exit(1);
	}
	while (fgets(buf, sizeof(buf), f) != NULL) {
	    /* strip \n */
	    for (cp = buf; *cp && *cp != '\n'; cp++)
		;
	    *cp = '\0';
	    pmTraversePMNS(buf, dometric);
	}
	fclose(f);
    }

    sts = pmLookupName(numpmid, namelist, pmidlist);
    if (sts != numpmid) {
	int		i;
	if (sts < 0)
	    fprintf(stderr, "%s: pmLookupName: %s\n", pmGetProgname(), pmErrStr(sts));
	else
	    fprintf(stderr, "%s: pmLookupName: returned %d, expected %d\n", pmGetProgname(), sts, numpmid);
	for (i = 0; i < numpmid; i++) {
	    if (pmidlist[i] == PM_ID_NULL)
		fprintf(stderr, "   %s is bad\n", namelist[i]);
	}
	exit(1);
    }

    save_samples = samples;
    times(&then);
    while (samples < 0 || samples-- > 0) {
	sts = pmFetch(numpmid, pmidlist, &rp);
	if (sts < 0) {
	    fprintf(stderr, "%s: pmFetch: %s\n", pmGetProgname(), pmErrStr(sts));
	    exit(1);
	}
	pmFreeResult(rp);
	if (verbose)
	    dumpstate(sts);
	if (interval && samples != 0)
	    sleep(interval);
    }
    if (save_samples < 0)
	return 0;
    times(&now);
    ticks = now.tms_utime - then.tms_utime + now.tms_stime - then.tms_stime;
    printf("pmFetch time: %d iterations, CPU time %.3f sec (%.1f usec / fetch)\n", save_samples, ((double)(ticks))/hz, 1000000*((double)(ticks))/(hz*save_samples));

    return 0;
}
