UNIX 7th Ed. - /usr/src/cmd/ac.c

   1: /*
   2:  * acct [ -w wtmp ] [ -d ] [ -p ] [ people ]
   3:  */
   4: 
   5: #include <stdio.h>
   6: #include <ctype.h>
   7: #include <time.h>
   8: #include <utmp.h>
   9: #include <sys/types.h>
  10: #include <sys/timeb.h>
  11: 
  12: #define TSIZE   33
  13: #define USIZE   200
  14: struct  utmp ibuf;
  15: 
  16: struct ubuf {
  17:     char    uname[8];
  18:     long    utime;
  19: } ubuf[USIZE];
  20: 
  21: struct tbuf {
  22:     struct  ubuf    *userp;
  23:     long    ttime;
  24: } tbuf[TSIZE];
  25: 
  26: char    *wtmp;
  27: int pflag, byday;
  28: long    dtime;
  29: long    midnight;
  30: long    lastime;
  31: long    day = 86400L;
  32: int pcount;
  33: char    **pptr;
  34: 
  35: main(argc, argv)
  36: char **argv;
  37: {
  38:     int c, fl;
  39:     register i;
  40:     FILE *wf;
  41: 
  42:     wtmp = "/usr/adm/wtmp";
  43:     while (--argc > 0 && **++argv == '-')
  44:     switch(*++*argv) {
  45:     case 'd':
  46:         byday++;
  47:         continue;
  48: 
  49:     case 'w':
  50:         if (--argc>0)
  51:             wtmp = *++argv;
  52:         continue;
  53: 
  54:     case 'p':
  55:         pflag++;
  56:         continue;
  57:     }
  58:     pcount = argc;
  59:     pptr = argv;
  60:     if ((wf = fopen(wtmp, "r")) == NULL) {
  61:         printf("No %s\n", wtmp);
  62:         exit(1);
  63:     }
  64:     for(;;) {
  65:         if (fread((char *)&ibuf, sizeof(ibuf), 1, wf) != 1)
  66:             break;
  67:         fl = 0;
  68:         for (i=0; i<8; i++) {
  69:             c = ibuf.ut_name[i];
  70:             if(isdigit(c) || isalpha(c)) {
  71:                 if (fl)
  72:                     goto skip;
  73:                 continue;
  74:             }
  75:             if (c==' ' || c=='\0') {
  76:                 fl++;
  77:                 ibuf.ut_name[i] = '\0';
  78:             } else
  79:                 goto skip;
  80:         }
  81:         loop();
  82:     skip:;
  83:     }
  84:     ibuf.ut_name[0] = '\0';
  85:     ibuf.ut_line[0] = '~';
  86:     time(&ibuf.ut_time);
  87:     loop();
  88:     print();
  89:     exit(0);
  90: }
  91: 
  92: loop()
  93: {
  94:     register i;
  95:     register struct tbuf *tp;
  96:     register struct ubuf *up;
  97: 
  98:     if(ibuf.ut_line[0] == '|') {
  99:         dtime = ibuf.ut_time;
 100:         return;
 101:     }
 102:     if(ibuf.ut_line[0] == '}') {
 103:         if(dtime == 0)
 104:             return;
 105:         for(tp = tbuf; tp < &tbuf[TSIZE]; tp++)
 106:             tp->ttime += ibuf.ut_time-dtime;
 107:         dtime = 0;
 108:         return;
 109:     }
 110:     if (lastime>ibuf.ut_time || lastime+(1.5*day)<ibuf.ut_time)
 111:         midnight = 0;
 112:     if (midnight==0)
 113:         newday();
 114:     lastime = ibuf.ut_time;
 115:     if (byday && ibuf.ut_time > midnight) {
 116:         upall(1);
 117:         print();
 118:         newday();
 119:         for (up=ubuf; up < &ubuf[USIZE]; up++)
 120:             up->utime = 0;
 121:     }
 122:     if (ibuf.ut_line[0] == '~') {
 123:         ibuf.ut_name[0] = '\0';
 124:         upall(0);
 125:         return;
 126:     }
 127:     if (ibuf.ut_line[0]=='t')
 128:         i = (ibuf.ut_line[3]-'0')*10 + (ibuf.ut_line[4]-'0');
 129:     else
 130:         i = TSIZE-1;
 131:     if (i<0 || i>=TSIZE)
 132:         i = TSIZE-1;
 133:     tp = &tbuf[i];
 134:     update(tp, 0);
 135: }
 136: 
 137: print()
 138: {
 139:     int i;
 140:     long ttime, t;
 141: 
 142:     ttime = 0;
 143:     for (i=0; i<USIZE; i++) {
 144:         if(!among(i))
 145:             continue;
 146:         t = ubuf[i].utime;
 147:         if (t>0)
 148:             ttime += t;
 149:         if (pflag && ubuf[i].utime > 0) {
 150:             printf("\t%-8.8s%6.2f\n",
 151:                 ubuf[i].uname, ubuf[i].utime/3600.);
 152:         }
 153:     }
 154:     if (ttime > 0) {
 155:         pdate();
 156:         printf("\ttotal%9.2f\n", ttime/3600.);
 157:     }
 158: }
 159: 
 160: upall(f)
 161: {
 162:     register struct tbuf *tp;
 163: 
 164:     for (tp=tbuf; tp < &tbuf[TSIZE]; tp++)
 165:         update(tp, f);
 166: }
 167: 
 168: update(tp, f)
 169: struct tbuf *tp;
 170: {
 171:     int j;
 172:     struct ubuf *up;
 173:     long t, t1;
 174: 
 175:     if (f)
 176:         t = midnight;
 177:     else
 178:         t = ibuf.ut_time;
 179:     if (tp->userp) {
 180:         t1 = t - tp->ttime;
 181:         if (t1>0 && t1 < 1.5*day)
 182:             tp->userp->utime += t1;
 183:     }
 184:     tp->ttime = t;
 185:     if (f)
 186:         return;
 187:     if (ibuf.ut_name[0]=='\0') {
 188:         tp->userp = 0;
 189:         return;
 190:     }
 191:     for (up=ubuf; up < &ubuf[USIZE]; up++) {
 192:         if (up->uname[0] == '\0')
 193:             break;
 194:         for (j=0; j<8 && up->uname[j]==ibuf.ut_name[j]; j++);
 195:         if (j>=8)
 196:             break;
 197:     }
 198:     for (j=0; j<8; j++)
 199:         up->uname[j] = ibuf.ut_name[j];
 200:     tp->userp = up;
 201: }
 202: 
 203: among(i)
 204: {
 205:     register j, k;
 206:     register char *p;
 207: 
 208:     if (pcount==0)
 209:         return(1);
 210:     for (j=0; j<pcount; j++) {
 211:         p = pptr[j];
 212:         for (k=0; k<8; k++) {
 213:             if (*p == ubuf[i].uname[k]) {
 214:                 if (*p++ == '\0')
 215:                     return(1);
 216:             } else
 217:                 break;
 218:         }
 219:     }
 220:     return(0);
 221: }
 222: 
 223: newday()
 224: {
 225:     long ttime;
 226:     struct timeb tb;
 227:     struct tm *localtime();
 228: 
 229:     time(&ttime);
 230:     if (midnight == 0) {
 231:         ftime(&tb);
 232:         midnight = 60*(long)tb.timezone;
 233:         if (localtime(&ttime)->tm_isdst)
 234:             midnight -= 3600;
 235:     }
 236:     while (midnight <= ibuf.ut_time)
 237:         midnight += day;
 238: }
 239: 
 240: pdate()
 241: {
 242:     long x;
 243:     char *ctime();
 244: 
 245:     if (byday==0)
 246:         return;
 247:     x = midnight-1;
 248:     printf("%.6s", ctime(&x)+4);
 249: }

Defined functions

Defined variables

Defined struct's

Defined macros