OpenSolaris_b135/cmd/acct/acctprc1.c
/*
* 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.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
/*
* 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);
}