/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/* All Rights Reserved */

#include <sys/types.h>
#include <sys/param.h>
#include <sys/fss.h>
#include <sys/priocntl.h>
#include <sys/fsspriocntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include "dispadmin.h"

/*
 * This file contains the class specific code implementing the fair-share
 * scheduler dispadmin sub-command.
 */

#define BASENMSZ        16

extern char *basename();
static void getadmin(), setadmin();

static char usage[] = "usage:   dispadmin -l\n"
        "dispadmin -c FSS -g [-r res]\n"
        "dispadmin -c FSS -s infile\n";

static char basenm[BASENMSZ];
static char cmdpath[256];

int
main(int argc, char **argv)
{
        int c;
        int lflag, gflag, rflag, sflag;
        ulong_t res;
        char *infile;
        char *endp;

        (void) strcpy(cmdpath, argv[0]);
        (void) strcpy(basenm, basename(argv[0]));
        lflag = gflag = rflag = sflag = 0;
        while ((c = getopt(argc, argv, "lc:gr:s:")) != -1) {
                switch (c) {

                case 'l':
                        lflag++;
                        break;

                case 'c':
                        if (strcmp(optarg, "FSS") != 0)
                                fatalerr("error: %s executed for %s class, "
                                    "%s is actually sub-command for FSS "
                                    "class\n", cmdpath, optarg, cmdpath);
                        break;

                case 'g':
                        gflag++;
                        break;

                case 'r':
                        rflag++;
                        errno = 0;
                        res = strtoul(optarg, &endp, 10);
                        if (res == 0 || errno != 0 || *endp != '\0')
                                fatalerr("%s: Can't convert to requested "
                                    "resolution\n", basenm);
                        break;

                case 's':
                        sflag++;
                        infile = optarg;
                        break;

                case '?':
                        fatalerr(usage);

                default:
                        break;
                }
        }

        if (lflag) {
                if (gflag || rflag || sflag)
                        fatalerr(usage);

                (void) printf("FSS\t(Fair Share)\n");
                return (0);

        } else if (gflag) {
                if (lflag || sflag)
                        fatalerr(usage);

                if (rflag == 0)
                        res = 1000;

                getadmin(res);
                return (0);

        } else if (sflag) {
                if (lflag || gflag || rflag)
                        fatalerr(usage);

                setadmin(infile);
                return (0);

        } else {
                fatalerr(usage);
        }
        return (1);
}

/*
 * Retrieve the current settings from kernel, convert the time quantum
 * value to the resolution specified by res and print out the results.
 */
static void
getadmin(ulong_t res)
{
        pcinfo_t pcinfo;
        pcadmin_t pcadmin;
        fssadmin_t fssadmin;
        hrtimer_t hrtime;
        long fss_quantum;

        (void) strcpy(pcinfo.pc_clname, "FSS");
        if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
                fatalerr("%s: Can't get FSS class ID, priocntl system call "
                    "failed (%s)\n", basenm, strerror(errno));

        pcadmin.pc_cid = pcinfo.pc_cid;
        pcadmin.pc_cladmin = (char *)&fssadmin;
        fssadmin.fss_cmd = FSS_GETADMIN;

        if (priocntl(0, 0, PC_ADMIN, (caddr_t)&pcadmin) == -1)
                fatalerr("%s: Can't get scheduler configuration, priocntl "
                    "system call failed (%s)\n", basenm, strerror(errno));

        hrtime.hrt_secs = 0;
        hrtime.hrt_rem = fssadmin.fss_quantum;
        hrtime.hrt_res = HZ;
        if (_hrtnewres(&hrtime, res, HRT_RNDUP) == -1)
                fatalerr("%s: Can't convert to requested resolution\n", basenm);
        if ((fss_quantum = hrtconvert(&hrtime)) == -1)
                fatalerr("%s: Can't express time quantum in "
                    "requested resolution,\n"
                    "try coarser resolution\n", basenm);

        (void) printf("#\n# Fair Share Scheduler Configuration\n#\n");
        (void) printf("RES=%ld\n", res);
        (void) printf("#\n# Time Quantum\n#\n");
        (void) printf("QUANTUM=%ld\n", fss_quantum);
}

/*
 * Read the scheduler settings from infile, convert the time quantum values
 * to HZ resolution, do a little sanity checking and overwrite kernel settings
 * with the values from the file.
 */
static void
setadmin(char *infile)
{
        int i;
        pcinfo_t pcinfo;
        pcadmin_t pcadmin;
        fssadmin_t fssadmin;
        int line;
        ulong_t res;
        long fss_quantum;
        hrtimer_t hrtime;
        FILE *fp;
        char buf[512];
        char *endp;
        int nparams = 0;

        (void) strcpy(pcinfo.pc_clname, "FSS");
        if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
                fatalerr("%s: Can't get FSS class ID, priocntl system call "
                    "failed (%s)\n", basenm, strerror(errno));

        if ((fp = fopen(infile, "r")) == NULL)
                fatalerr("%s: Can't open %s for input\n", basenm, infile);

        for (line = 1; fgets(buf, sizeof (buf), fp) != NULL; line++) {
                char name[512], value[512];
                int len;

                if (buf[0] == '#' || buf[0] == '\n')
                        continue;
                /*
                 * Look for "name=value", with optional whitespaces on either
                 * side, terminated by a newline, and consuming the whole line.
                 */
                /* LINTED - unbounded string specifier */
                if (sscanf(buf, " %[^=]=%s \n%n", name, value, &len) == 2 &&
                    name[0] != '\0' && value[0] != '\0' && len == strlen(buf)) {

                        if (strcmp(name, "RES") == 0) {
                                errno = 0;
                                i = (int)strtol(value, &endp, 10);
                                if (errno != 0 || endp == value ||
                                    i < 0 || *endp != '\0')
                                        fatalerr("%s, line %d: illegal "
                                            "resolution value\n", infile, line);
                                else
                                        res = i;
                                nparams++;
                        } else if (strcmp(name, "QUANTUM") == 0) {
                                errno = 0;
                                i = (int)strtol(value, &endp, 10);
                                if (errno != 0 || endp == value ||
                                    i < 0 || *endp != '\0')
                                        fatalerr("%s, line %d: illegal time "
                                            "quantum value\n", infile, line);
                                else
                                        fss_quantum = i;
                                nparams++;
                        } else {
                                fatalerr("%s, line %d: invalid token\n",
                                    infile, line);
                        }
                } else {
                        fatalerr("%s, line %d: syntax error\n", infile, line);
                }
        }
        if (line == 1 || nparams < 2)
                fatalerr("cannot read settings from %s\n", infile);
        if (res != HZ) {
                hrtime.hrt_secs = 0;
                hrtime.hrt_rem = fss_quantum;
                hrtime.hrt_res = res;
                if (_hrtnewres(&hrtime, HZ, HRT_RNDUP) == -1)
                        fatalerr("%s: Can't convert specified "
                            "resolution to ticks\n", basenm);
                if ((fssadmin.fss_quantum = hrtconvert(&hrtime)) == -1)
                        fatalerr("%s, line %d: time quantum value out of "
                            "valid range\n", infile, line);
        } else {
                fssadmin.fss_quantum = (short)fss_quantum;
        }

        pcadmin.pc_cid = pcinfo.pc_cid;
        pcadmin.pc_cladmin = (char *)&fssadmin;
        fssadmin.fss_cmd = FSS_SETADMIN;

        if (priocntl(0, 0, PC_ADMIN, (caddr_t)&pcadmin) == -1)
                fatalerr("%s: Can't set scheduler parameters, priocntl "
                    "system call failed (%s)\n", basenm, strerror(errno));
}