usr.sbin/sa/pdb.c

   1 /*
   2  * Copyright (c) 1994 Christopher G. Demetriou
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by Christopher G. Demetriou.
  16  * 4. The name of the author may not be used to endorse or promote products
  17  *    derived from this software without specific prior written permission
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  24  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include <sys/cdefs.h>
  32 __FBSDID("$FreeBSD$");
  33
  34 #include <sys/types.h>
  35 #include <sys/acct.h>
  36 #include <err.h>
  37 #include <errno.h>
  38 #include <fcntl.h>
  39 #include <stdint.h>
  40 #include <stdio.h>
  41 #include <string.h>
  42 #include "extern.h"
  43 #include "pathnames.h"
  44
  45 static int check_junk __P((struct cmdinfo *));
  46 static void add_ci __P((const struct cmdinfo *, struct cmdinfo *));
  47 static void print_ci __P((const struct cmdinfo *, const struct cmdinfo *));
  48
  49 static DB       *pacct_db;
  50
  51 int
  52 pacct_init()
  53 {
  54         DB *saved_pacct_db;
  55         int error;
  56
  57         pacct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, NULL);
  58         if (pacct_db == NULL)
  59                 return (-1);
  60
  61         error = 0;
  62         if (!iflag) {
  63                 DBT key, data;
  64                 int serr, nerr;
  65
  66                 saved_pacct_db = dbopen(_PATH_SAVACCT, O_RDONLY, 0, DB_BTREE,
  67                     NULL);
  68                 if (saved_pacct_db == NULL) {
  69                         error = errno == ENOENT ? 0 : -1;
  70                         if (error)
  71                                 warn("retrieving process accounting summary");
  72                         goto out;
  73                 }
  74
  75                 serr = DB_SEQ(saved_pacct_db, &key, &data, R_FIRST);
  76                 if (serr < 0) {
  77                         warn("retrieving process accounting summary");
  78                         error = -1;
  79                         goto closeout;
  80                 }
  81                 while (serr == 0) {
  82                         nerr = DB_PUT(pacct_db, &key, &data, 0);
  83                         if (nerr < 0) {
  84                                 warn("initializing process accounting stats");
  85                                 error = -1;
  86                                 break;
  87                         }
  88
  89                         serr = DB_SEQ(saved_pacct_db, &key, &data, R_NEXT);
  90                         if (serr < 0) {
  91                                 warn("retrieving process accounting summary");
  92                                 error = -1;
  93                                 break;
  94                         }
  95                 }
  96
  97 closeout:       if (DB_CLOSE(saved_pacct_db) < 0) {
  98                         warn("closing process accounting summary");
  99                         error = -1;
 100                 }
 101         }
 102
 103 out:    if (error != 0)
 104                 pacct_destroy();
 105         return (error);
 106 }
 107
 108 void
 109 pacct_destroy()
 110 {
 111         if (DB_CLOSE(pacct_db) < 0)
 112                 warn("destroying process accounting stats");
 113 }
 114
 115 int
 116 pacct_add(ci)
 117         const struct cmdinfo *ci;
 118 {
 119         DBT key, data;
 120         struct cmdinfo newci;
 121         char keydata[sizeof ci->ci_comm];
 122         int rv;
 123
 124         bcopy(ci->ci_comm, &keydata, sizeof keydata);
 125         key.data = &keydata;
 126         key.size = strlen(keydata);
 127
 128         rv = DB_GET(pacct_db, &key, &data, 0);
 129         if (rv < 0) {
 130                 warn("get key %s from process accounting stats", ci->ci_comm);
 131                 return (-1);
 132         } else if (rv == 0) {   /* it's there; copy whole thing */
 133                 /* XXX compare size if paranoid */
 134                 /* add the old data to the new data */
 135                 bcopy(data.data, &newci, data.size);
 136         } else {                /* it's not there; zero it and copy the key */
 137                 bzero(&newci, sizeof newci);
 138                 bcopy(key.data, newci.ci_comm, key.size);
 139         }
 140
 141         add_ci(ci, &newci);
 142
 143         data.data = &newci;
 144         data.size = sizeof newci;
 145         rv = DB_PUT(pacct_db, &key, &data, 0);
 146         if (rv < 0) {
 147                 warn("add key %s to process accounting stats", ci->ci_comm);
 148                 return (-1);
 149         } else if (rv == 1) {
 150                 warnx("duplicate key %s in process accounting stats",
 151                     ci->ci_comm);
 152                 return (-1);
 153         }
 154
 155         return (0);
 156 }
 157
 158 int
 159 pacct_update()
 160 {
 161         DB *saved_pacct_db;
 162         DBT key, data;
 163         int error, serr, nerr;
 164
 165         saved_pacct_db = dbopen(_PATH_SAVACCT, O_RDWR|O_CREAT|O_TRUNC, 0644,
 166             DB_BTREE, NULL);
 167         if (saved_pacct_db == NULL) {
 168                 warn("creating process accounting summary");
 169                 return (-1);
 170         }
 171
 172         error = 0;
 173
 174         serr = DB_SEQ(pacct_db, &key, &data, R_FIRST);
 175         if (serr < 0) {
 176                 warn("retrieving process accounting stats");
 177                 error = -1;
 178         }
 179         while (serr == 0) {
 180                 nerr = DB_PUT(saved_pacct_db, &key, &data, 0);
 181                 if (nerr < 0) {
 182                         warn("saving process accounting summary");
 183                         error = -1;
 184                         break;
 185                 }
 186
 187                 serr = DB_SEQ(pacct_db, &key, &data, R_NEXT);
 188                 if (serr < 0) {
 189                         warn("retrieving process accounting stats");
 190                         error = -1;
 191                         break;
 192                 }
 193         }
 194
 195         if (DB_SYNC(saved_pacct_db, 0) < 0) {
 196                 warn("syncing process accounting summary");
 197                 error = -1;
 198         }
 199         if (DB_CLOSE(saved_pacct_db) < 0) {
 200                 warn("closing process accounting summary");
 201                 error = -1;
 202         }
 203         return error;
 204 }
 205
 206 void
 207 pacct_print()
 208 {
 209         BTREEINFO bti;
 210         DBT key, data, ndata;
 211         DB *output_pacct_db;
 212         struct cmdinfo *cip, ci, ci_total, ci_other, ci_junk;
 213         int rv;
 214
 215         bzero(&ci_total, sizeof ci_total);
 216         strcpy(ci_total.ci_comm, "");
 217         bzero(&ci_other, sizeof ci_other);
 218         strcpy(ci_other.ci_comm, "***other");
 219         bzero(&ci_junk, sizeof ci_junk);
 220         strcpy(ci_junk.ci_comm, "**junk**");
 221
 222         /*
 223          * Retrieve them into new DB, sorted by appropriate key.
 224          * At the same time, cull 'other' and 'junk'
 225          */
 226         bzero(&bti, sizeof bti);
 227         bti.compare = sa_cmp;
 228         output_pacct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, &bti);
 229         if (output_pacct_db == NULL) {
 230                 warn("couldn't sort process accounting stats");
 231                 return;
 232         }
 233
 234         ndata.data = NULL;
 235         ndata.size = 0;
 236         rv = DB_SEQ(pacct_db, &key, &data, R_FIRST);
 237         if (rv < 0)
 238                 warn("retrieving process accounting stats");
 239         while (rv == 0) {
 240                 cip = (struct cmdinfo *) data.data;
 241                 bcopy(cip, &ci, sizeof ci);
 242
 243                 /* add to total */
 244                 add_ci(&ci, &ci_total);
 245
 246                 if (vflag && ci.ci_calls <= cutoff &&
 247                     (fflag || check_junk(&ci))) {
 248                         /* put it into **junk** */
 249                         add_ci(&ci, &ci_junk);
 250                         goto next;
 251                 }
 252                 if (!aflag &&
 253                     ((ci.ci_flags & CI_UNPRINTABLE) != 0 || ci.ci_calls <= 1)) {
 254                         /* put into ***other */
 255                         add_ci(&ci, &ci_other);
 256                         goto next;
 257                 }
 258                 rv = DB_PUT(output_pacct_db, &data, &ndata, 0);
 259                 if (rv < 0)
 260                         warn("sorting process accounting stats");
 261
 262 next:           rv = DB_SEQ(pacct_db, &key, &data, R_NEXT);
 263                 if (rv < 0)
 264                         warn("retrieving process accounting stats");
 265         }
 266
 267         /* insert **junk** and ***other */
 268         if (ci_junk.ci_calls != 0) {
 269                 data.data = &ci_junk;
 270                 data.size = sizeof ci_junk;
 271                 rv = DB_PUT(output_pacct_db, &data, &ndata, 0);
 272                 if (rv < 0)
 273                         warn("sorting process accounting stats");
 274         }
 275         if (ci_other.ci_calls != 0) {
 276                 data.data = &ci_other;
 277                 data.size = sizeof ci_other;
 278                 rv = DB_PUT(output_pacct_db, &data, &ndata, 0);
 279                 if (rv < 0)
 280                         warn("sorting process accounting stats");
 281         }
 282
 283         /* print out the total */
 284         print_ci(&ci_total, &ci_total);
 285
 286         /* print out; if reversed, print first (smallest) first */
 287         rv = DB_SEQ(output_pacct_db, &data, &ndata, rflag ? R_FIRST : R_LAST);
 288         if (rv < 0)
 289                 warn("retrieving process accounting report");
 290         while (rv == 0) {
 291                 cip = (struct cmdinfo *) data.data;
 292                 bcopy(cip, &ci, sizeof ci);
 293
 294                 print_ci(&ci, &ci_total);
 295
 296                 rv = DB_SEQ(output_pacct_db, &data, &ndata,
 297                     rflag ? R_NEXT : R_PREV);
 298                 if (rv < 0)
 299                         warn("retrieving process accounting report");
 300         }
 301         DB_CLOSE(output_pacct_db);
 302 }
 303
 304 static int
 305 check_junk(cip)
 306         struct cmdinfo *cip;
 307 {
 308         char *cp;
 309         size_t len;
 310
 311         fprintf(stderr, "%s (%ju) -- ", cip->ci_comm, (uintmax_t)cip->ci_calls);
 312         cp = fgetln(stdin, &len);
 313
 314         return (cp && (cp[0] == 'y' || cp[0] == 'Y')) ? 1 : 0;
 315 }
 316
 317 static void
 318 add_ci(fromcip, tocip)
 319         const struct cmdinfo *fromcip;
 320         struct cmdinfo *tocip;
 321 {
 322         tocip->ci_calls += fromcip->ci_calls;
 323         tocip->ci_etime += fromcip->ci_etime;
 324         tocip->ci_utime += fromcip->ci_utime;
 325         tocip->ci_stime += fromcip->ci_stime;
 326         tocip->ci_mem += fromcip->ci_mem;
 327         tocip->ci_io += fromcip->ci_io;
 328 }
 329
 330 static void
 331 print_ci(cip, totalcip)
 332         const struct cmdinfo *cip, *totalcip;
 333 {
 334         double t, c;
 335         int uflow;
 336
 337         c = cip->ci_calls ? cip->ci_calls : 1;
 338         t = (cip->ci_utime + cip->ci_stime) / (double) AHZ;
 339         if (t < 0.01) {
 340                 t = 0.01;
 341                 uflow = 1;
 342         } else
 343                 uflow = 0;
 344
 345         printf("%8ju ", (uintmax_t)cip->ci_calls);
 346         if (cflag) {
 347                 if (cip != totalcip)
 348                         printf(" %4.2f%%  ",
 349                             cip->ci_calls / (double) totalcip->ci_calls);
 350                 else
 351                         printf(" %4s   ", "");
 352         }
 353
 354         if (jflag)
 355                 printf("%11.2fre ", cip->ci_etime / (double) (AHZ * c));
 356         else
 357                 printf("%11.2fre ", cip->ci_etime / (60.0 * AHZ));
 358         if (cflag) {
 359                 if (cip != totalcip)
 360                         printf(" %4.2f%%  ",
 361                             cip->ci_etime / (double) totalcip->ci_etime);
 362                 else
 363                         printf(" %4s   ", "");
 364         }
 365
 366         if (!lflag) {
 367                 if (jflag)
 368                         printf("%11.2fcp ", t / (double) cip->ci_calls);
 369                 else
 370                         printf("%11.2fcp ", t / 60.0);
 371                 if (cflag) {
 372                         if (cip != totalcip)
 373                                 printf(" %4.2f%%  ",
 374                                     (cip->ci_utime + cip->ci_stime) / (double)
 375                                     (totalcip->ci_utime + totalcip->ci_stime));
 376                         else
 377                                 printf(" %4s   ", "");
 378                 }
 379         } else {
 380                 if (jflag)
 381                         printf("%11.2fu ", cip->ci_utime / (double) (AHZ * c));
 382                 else
 383                         printf("%11.2fu ", cip->ci_utime / (60.0 * AHZ));
 384                 if (cflag) {
 385                         if (cip != totalcip)
 386                                 printf(" %4.2f%%  ", cip->ci_utime / (double) totalcip->ci_utime);
 387                         else
 388                                 printf(" %4s   ", "");
 389                 }
 390                 if (jflag)
 391                         printf("%11.2fs ", cip->ci_stime / (double) (AHZ * c));
 392                 else
 393                         printf("%11.2fs ", cip->ci_stime / (60.0 * AHZ));
 394                 if (cflag) {
 395                         if (cip != totalcip)
 396                                 printf(" %4.2f%%  ", cip->ci_stime / (double) totalcip->ci_stime);
 397                         else
 398                                 printf(" %4s   ", "");
 399                 }
 400         }
 401
 402         if (tflag) {
 403                 if (!uflow)
 404                         printf("%8.2fre/cp ",
 405                             cip->ci_etime /
 406                             (double) (cip->ci_utime + cip->ci_stime));
 407                 else
 408                         printf("*ignore*      ");
 409         }
 410
 411         if (Dflag)
 412                 printf("%10jutio ", (uintmax_t)cip->ci_io);
 413         else
 414                 printf("%8.0favio ", cip->ci_io / c);
 415
 416         if (Kflag)
 417                 printf("%10juk*sec ", (uintmax_t)cip->ci_mem);
 418         else
 419                 printf("%8.0fk ", cip->ci_mem / t);
 420
 421         printf("  %s\n", cip->ci_comm);
 422 }