/*
 * 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 (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/*        All Rights Reserved   */


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

/*
 *      acctprc1 [ctmpfile]
 *      reads std. input (acct.h format), adds login names
 *      writes std. output (ptmp.h/ascii format)
 *      if ctmpfile is given, it is expected have ctmp.h/ascii data,
 *      sorted by uid/name; it is used to make better guesses at login names
 */

#include <sys/types.h>
#include <sys/param.h>
#include "acctdef.h"
#include <stdio.h>
#include <errno.h>
#include <sys/acct.h>
#include <stdlib.h>

#define MYKIND(flag)    ((flag & ACCTF) == 0)

struct  acct    ab;
struct  ctmp    cb;
struct  ptmp    pb;

int     a_usize = A_USIZE;
struct urec {                           /* 1 for each distinct uid/name */
        uid_t   ur_uid;                 /* sorted by uid/name */
        char    ur_name[NSZ];
        struct srec     *ur_srec;               /* ptr to first session */
        short   ur_cnt;                 /* # sessions */
} * ur;

struct urec *urlast;

int     a_ssize = A_SSIZE;
int     ssize;

struct srec {                           /* 1 for each distinct session */
        dev_t   sr_tty;                 /* dev, used to connect with process*/
        time_t  sr_start;               /* start time of session */
        time_t  sr_end;                 /* end time of session */
} * sr;

char *getname(uid_t, dev_t, time_t);
void readctmp(char *);
char *getnamc(uid_t, dev_t, time_t);
int aread(int);

char    *uidtonam();

int
main(int argc, char **argv)
{
        long            elaps[2];
        ulong_t         etime, stime;
        unsigned long   mem;
        ulong_t         expand();
        int             ver;    /* version of acct struct */
        int             aread();

        if ((ur = (struct urec *) calloc(a_usize,
                sizeof (struct urec))) == NULL) {
                fprintf(stderr, "acctpr1: Cannot allocate memory\n");
                exit(3);
        }

        urlast = ur;
        if ((sr = (struct srec *) calloc(a_ssize,
                sizeof (struct srec))) == NULL) {
                fprintf(stderr, "acctpr1: Cannot allocate memory\n");
                exit(3);
        }

        while (--argc > 0) {
                if (**++argv == '-')
                        switch(*++*argv) {
                        }
                else {
                        readctmp(*argv);
                }
        }


        if (fread((char *)&ab, sizeof(struct acct), 1, stdin) != 1)
                exit(1);
        else if (ab.ac_flag & AEXPND)
                ver = 2;        /* 4.0 acct structure */
        else
                ver = 1;        /* 3.x acct structure */

        rewind(stdin);

        while (aread(ver) == 1) {
                if (!MYKIND(ab.ac_flag))
                        continue;
                pb.pt_uid = ab.ac_uid;
                CPYN(pb.pt_name, getname(ab.ac_uid, ab.ac_tty, ab.ac_btime));
                /*
                 * approximate cpu P/NP split as same as elapsed time
                 */
                if ((etime = SECS(expand(ab.ac_etime))) == 0)
                        etime = 1;
                stime = expand(ab.ac_stime) + expand(ab.ac_utime);
                mem = expand(ab.ac_mem);
                if(pnpsplit(ab.ac_btime, etime, elaps) == 0) {
                        fprintf(stderr, "acctprc1: could not calculate prime/non-prime hours\n");

                        exit(1);
                }
                pb.pt_cpu[0] = (double)stime * (double)elaps[0] / etime;
                pb.pt_cpu[1] = (stime > pb.pt_cpu[0])? stime - pb.pt_cpu[0] : 0;
                pb.pt_cpu[1] = stime - pb.pt_cpu[0];
                if (stime)
                        pb.pt_mem = (mem + stime - 1) / stime;
                else
                        pb.pt_mem = 0;  /* unlikely */
                printf("%ld\t%.*s\t%lu\t%lu\t%u\n",
                    pb.pt_uid,
                    OUTPUT_NSZ,
                    pb.pt_name,
                    pb.pt_cpu[0], pb.pt_cpu[1],
                    pb.pt_mem);
        }

        exit(0);
}

/*
 *      return ptr to name corresponding to uid
 *      try ctmp first, then use uidtonam (internal list or passwd file)
 */
char *
getname(uid_t uid, dev_t tty, time_t start)
{
        char *p;

        if ((p = getnamc(uid, tty, start)) != NULL)
                return (p);
        return (uidtonam(uid));
}

/*
 *      read ctmp file, build up urec-srec data structures for
 *      later use by getnamc
 */
void
readctmp(char *fname)
{
        FILE *fp;
        struct urec *up;
        struct srec *sp;
        int i = 0, j = 0, k=0;

        if ((fp = fopen(fname, "r")) == NULL) {
                fprintf(stderr, "acctprc1: can't open %s\n", fname);
                return;
        }

        up = NULL;
        sp = sr;

        while (fscanf(fp, "%hd\t%ld\t%s\t%lu\t%lu\t%lu\t%*[^\n]",
                &cb.ct_tty,
                &cb.ct_uid,
                cb.ct_name,
                &cb.ct_con[0],
                &cb.ct_con[1],
                &cb.ct_start) != EOF) {
                if (up == NULL || cb.ct_uid != up->ur_uid ||
                        !EQN(cb.ct_name, up->ur_name)) {
                        if (up == NULL)
                                up = ur;
                        if (++up >= &ur[a_usize]) {
                                a_usize = a_usize + A_USIZE;
                                if ((ur = (struct urec *) realloc(ur, a_usize *
                                        sizeof (struct urec))) == NULL) {
                                        fprintf(stderr, "acctprc1: 1 Cannot reallocate memory\n");
                                        exit(2);
                                }
                                up = &ur[a_usize - A_USIZE];
                        }
                        up->ur_uid = cb.ct_uid;
                        CPYN(up->ur_name, cb.ct_name);
                        up->ur_srec = sp;
                        up->ur_cnt = 0;
                }

                if (sp >= &sr[a_ssize-1]) {
                        a_ssize = a_ssize + A_SSIZE;
                        if ((sr = (struct srec *) realloc(sr, a_ssize *
                                sizeof (struct srec))) == NULL) {
                                fprintf(stderr, "acctprc1: 2 Cannot reallocate memory\n");
                                printf("errno=%d\n", errno);
                                exit(2);
                        }
                        sp = &sr[a_ssize - A_SSIZE];
                }

                sp->sr_tty = cb.ct_tty;
                sp->sr_start = cb.ct_start;
                sp->sr_end = cb.ct_start + cb.ct_con[0] + cb.ct_con[1];
                sp++;
                up->ur_cnt++;
        }
        if (up != NULL)
                urlast = ++up;
        fclose(fp);
}

/*
 *      using urec-srec data (if any), make best guess at login name
 *      corresponding to uid, return ptr to the name.
 *      must match on tty; use start time to help guess
 *      for any urec having same uid as uid, search array of associated
 *      srecs for those having same tty
 *      if start time of process is within range of session, that's it
 *      if none can be found within range, give it to person of same uid
 *      who last logged off on that terminal
 */
char *
getnamc(uid_t uid, dev_t tty, time_t start)
{
        struct urec *up;
        struct srec *sp;
        struct srec *splast;
        long latest;
        char *guess;

        latest = 0;
        guess = NULL;
        for (up = ur; up < urlast && uid >= up->ur_uid; up++)
                if (uid == up->ur_uid) {
                        sp = up->ur_srec;
                        splast = sp+up->ur_cnt;
                        for (; sp < splast; sp++)
                                if (tty == sp->sr_tty) {
                                        if (start >= sp->sr_start &&
                                                start <= sp->sr_end)
                                                return(up->ur_name);
                                        if (start >= sp->sr_start &&
                                                sp->sr_end > latest) {
                                                latest = sp->sr_end;
                                                guess = up->ur_name;
                                        }
                                }
                }

        return(guess);
}
int
aread(int ver)
{
        struct o_acct oab;
        int ret;

        if (ver != 2) {
                if ((ret = fread((char *)&oab, sizeof(struct o_acct), 1, stdin)) == 1){
                        /* copy SVR3 acct struct to SVR4 acct struct */
                        ab.ac_flag = oab.ac_flag | AEXPND;
                        ab.ac_stat = oab.ac_stat;
                        ab.ac_uid = (uid_t) oab.ac_uid;
                        ab.ac_gid = (gid_t) oab.ac_gid;
                        ab.ac_tty = (dev_t) oab.ac_tty;
                        ab.ac_btime = oab.ac_btime;
                        ab.ac_utime = oab.ac_utime;
                        ab.ac_stime = oab.ac_stime;
                        ab.ac_mem = oab.ac_mem;
                        ab.ac_io = oab.ac_io;
                        ab.ac_rw = oab.ac_rw;
                        strcpy(ab.ac_comm, oab.ac_comm);
                }
        } else
                ret = fread((char *)&ab, sizeof(struct acct), 1, stdin);


        return(ret != 1 ? 0 : 1);
}