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 * $FreeBSD: src/usr.sbin/sa/pdb.c,v 1.7 1999/08/28 01:19:53 peter Exp $ 31 * $DragonFly: src/usr.sbin/sa/pdb.c,v 1.3 2003/11/03 19:31:43 eirikn Exp $ 32 */ 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 <stdio.h> 40 #include <string.h> 41 #include "extern.h" 42 #include "pathnames.h" 43 44 static int check_junk(struct cmdinfo *); 45 static void add_ci(const struct cmdinfo *, struct cmdinfo *); 46 static void print_ci(const struct cmdinfo *, const struct cmdinfo *); 47 48 static DB *pacct_db; 49 50 int 51 pacct_init() 52 { 53 DB *saved_pacct_db; 54 int error; 55 56 pacct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, NULL); 57 if (pacct_db == NULL) 58 return (-1); 59 60 error = 0; 61 if (!iflag) { 62 DBT key, data; 63 int serr, nerr; 64 65 saved_pacct_db = dbopen(_PATH_SAVACCT, O_RDONLY, 0, DB_BTREE, 66 NULL); 67 if (saved_pacct_db == NULL) { 68 error = errno == ENOENT ? 0 : -1; 69 if (error) 70 warn("retrieving process accounting summary"); 71 goto out; 72 } 73 74 serr = DB_SEQ(saved_pacct_db, &key, &data, R_FIRST); 75 if (serr < 0) { 76 warn("retrieving process accounting summary"); 77 error = -1; 78 goto closeout; 79 } 80 while (serr == 0) { 81 nerr = DB_PUT(pacct_db, &key, &data, 0); 82 if (nerr < 0) { 83 warn("initializing process accounting stats"); 84 error = -1; 85 break; 86 } 87 88 serr = DB_SEQ(saved_pacct_db, &key, &data, R_NEXT); 89 if (serr < 0) { 90 warn("retrieving process accounting summary"); 91 error = -1; 92 break; 93 } 94 } 95 96 closeout: if (DB_CLOSE(saved_pacct_db) < 0) { 97 warn("closing process accounting summary"); 98 error = -1; 99 } 100 } 101 102 out: if (error != 0) 103 pacct_destroy(); 104 return (error); 105 } 106 107 void 108 pacct_destroy() 109 { 110 if (DB_CLOSE(pacct_db) < 0) 111 warn("destroying process accounting stats"); 112 } 113 114 int 115 pacct_add(ci) 116 const struct cmdinfo *ci; 117 { 118 DBT key, data; 119 struct cmdinfo newci; 120 char keydata[sizeof ci->ci_comm]; 121 int rv; 122 123 bcopy(ci->ci_comm, &keydata, sizeof keydata); 124 key.data = &keydata; 125 key.size = strlen(keydata); 126 127 rv = DB_GET(pacct_db, &key, &data, 0); 128 if (rv < 0) { 129 warn("get key %s from process accounting stats", ci->ci_comm); 130 return (-1); 131 } else if (rv == 0) { /* it's there; copy whole thing */ 132 /* XXX compare size if paranoid */ 133 /* add the old data to the new data */ 134 bcopy(data.data, &newci, data.size); 135 } else { /* it's not there; zero it and copy the key */ 136 bzero(&newci, sizeof newci); 137 bcopy(key.data, newci.ci_comm, key.size); 138 } 139 140 add_ci(ci, &newci); 141 142 data.data = &newci; 143 data.size = sizeof newci; 144 rv = DB_PUT(pacct_db, &key, &data, 0); 145 if (rv < 0) { 146 warn("add key %s to process accounting stats", ci->ci_comm); 147 return (-1); 148 } else if (rv == 1) { 149 warnx("duplicate key %s in process accounting stats", 150 ci->ci_comm); 151 return (-1); 152 } 153 154 return (0); 155 } 156 157 int 158 pacct_update() 159 { 160 DB *saved_pacct_db; 161 DBT key, data; 162 int error, serr, nerr; 163 164 saved_pacct_db = dbopen(_PATH_SAVACCT, O_RDWR|O_CREAT|O_TRUNC, 0644, 165 DB_BTREE, NULL); 166 if (saved_pacct_db == NULL) { 167 warn("creating process accounting summary"); 168 return (-1); 169 } 170 171 error = 0; 172 173 serr = DB_SEQ(pacct_db, &key, &data, R_FIRST); 174 if (serr < 0) { 175 warn("retrieving process accounting stats"); 176 error = -1; 177 } 178 while (serr == 0) { 179 nerr = DB_PUT(saved_pacct_db, &key, &data, 0); 180 if (nerr < 0) { 181 warn("saving process accounting summary"); 182 error = -1; 183 break; 184 } 185 186 serr = DB_SEQ(pacct_db, &key, &data, R_NEXT); 187 if (serr < 0) { 188 warn("retrieving process accounting stats"); 189 error = -1; 190 break; 191 } 192 } 193 194 if (DB_SYNC(saved_pacct_db, 0) < 0) { 195 warn("syncing process accounting summary"); 196 error = -1; 197 } 198 if (DB_CLOSE(saved_pacct_db) < 0) { 199 warn("closing process accounting summary"); 200 error = -1; 201 } 202 return error; 203 } 204 205 void 206 pacct_print() 207 { 208 BTREEINFO bti; 209 DBT key, data, ndata; 210 DB *output_pacct_db; 211 struct cmdinfo *cip, ci, ci_total, ci_other, ci_junk; 212 int rv; 213 214 bzero(&ci_total, sizeof ci_total); 215 strcpy(ci_total.ci_comm, ""); 216 bzero(&ci_other, sizeof ci_other); 217 strcpy(ci_other.ci_comm, "***other"); 218 bzero(&ci_junk, sizeof ci_junk); 219 strcpy(ci_junk.ci_comm, "**junk**"); 220 221 /* 222 * Retrieve them into new DB, sorted by appropriate key. 223 * At the same time, cull 'other' and 'junk' 224 */ 225 bzero(&bti, sizeof bti); 226 bti.compare = sa_cmp; 227 output_pacct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, &bti); 228 if (output_pacct_db == NULL) { 229 warn("couldn't sort process accounting stats"); 230 return; 231 } 232 233 ndata.data = NULL; 234 ndata.size = 0; 235 rv = DB_SEQ(pacct_db, &key, &data, R_FIRST); 236 if (rv < 0) 237 warn("retrieving process accounting stats"); 238 while (rv == 0) { 239 cip = (struct cmdinfo *) data.data; 240 bcopy(cip, &ci, sizeof ci); 241 242 /* add to total */ 243 add_ci(&ci, &ci_total); 244 245 if (vflag && ci.ci_calls <= cutoff && 246 (fflag || check_junk(&ci))) { 247 /* put it into **junk** */ 248 add_ci(&ci, &ci_junk); 249 goto next; 250 } 251 if (!aflag && 252 ((ci.ci_flags & CI_UNPRINTABLE) != 0 || ci.ci_calls <= 1)) { 253 /* put into ***other */ 254 add_ci(&ci, &ci_other); 255 goto next; 256 } 257 rv = DB_PUT(output_pacct_db, &data, &ndata, 0); 258 if (rv < 0) 259 warn("sorting process accounting stats"); 260 261 next: rv = DB_SEQ(pacct_db, &key, &data, R_NEXT); 262 if (rv < 0) 263 warn("retrieving process accounting stats"); 264 } 265 266 /* insert **junk** and ***other */ 267 if (ci_junk.ci_calls != 0) { 268 data.data = &ci_junk; 269 data.size = sizeof ci_junk; 270 rv = DB_PUT(output_pacct_db, &data, &ndata, 0); 271 if (rv < 0) 272 warn("sorting process accounting stats"); 273 } 274 if (ci_other.ci_calls != 0) { 275 data.data = &ci_other; 276 data.size = sizeof ci_other; 277 rv = DB_PUT(output_pacct_db, &data, &ndata, 0); 278 if (rv < 0) 279 warn("sorting process accounting stats"); 280 } 281 282 /* print out the total */ 283 print_ci(&ci_total, &ci_total); 284 285 /* print out; if reversed, print first (smallest) first */ 286 rv = DB_SEQ(output_pacct_db, &data, &ndata, rflag ? R_FIRST : R_LAST); 287 if (rv < 0) 288 warn("retrieving process accounting report"); 289 while (rv == 0) { 290 cip = (struct cmdinfo *) data.data; 291 bcopy(cip, &ci, sizeof ci); 292 293 print_ci(&ci, &ci_total); 294 295 rv = DB_SEQ(output_pacct_db, &data, &ndata, 296 rflag ? R_NEXT : R_PREV); 297 if (rv < 0) 298 warn("retrieving process accounting report"); 299 } 300 DB_CLOSE(output_pacct_db); 301 } 302 303 static int 304 check_junk(cip) 305 struct cmdinfo *cip; 306 { 307 char *cp; 308 size_t len; 309 310 fprintf(stderr, "%s (%qu) -- ", cip->ci_comm, cip->ci_calls); 311 cp = fgetln(stdin, &len); 312 313 return (cp && (cp[0] == 'y' || cp[0] == 'Y')) ? 1 : 0; 314 } 315 316 static void 317 add_ci(fromcip, tocip) 318 const struct cmdinfo *fromcip; 319 struct cmdinfo *tocip; 320 { 321 tocip->ci_calls += fromcip->ci_calls; 322 tocip->ci_etime += fromcip->ci_etime; 323 tocip->ci_utime += fromcip->ci_utime; 324 tocip->ci_stime += fromcip->ci_stime; 325 tocip->ci_mem += fromcip->ci_mem; 326 tocip->ci_io += fromcip->ci_io; 327 } 328 329 static void 330 print_ci(cip, totalcip) 331 const struct cmdinfo *cip, *totalcip; 332 { 333 double t, c; 334 int uflow; 335 336 c = cip->ci_calls ? cip->ci_calls : 1; 337 t = (cip->ci_utime + cip->ci_stime) / (double) AHZ; 338 if (t < 0.01) { 339 t = 0.01; 340 uflow = 1; 341 } else 342 uflow = 0; 343 344 printf("%8qu ", cip->ci_calls); 345 if (cflag) { 346 if (cip != totalcip) 347 printf(" %4.2f%% ", 348 cip->ci_calls / (double) totalcip->ci_calls); 349 else 350 printf(" %4s ", ""); 351 } 352 353 if (jflag) 354 printf("%11.2fre ", cip->ci_etime / (double) (AHZ * c)); 355 else 356 printf("%11.2fre ", cip->ci_etime / (60.0 * AHZ)); 357 if (cflag) { 358 if (cip != totalcip) 359 printf(" %4.2f%% ", 360 cip->ci_etime / (double) totalcip->ci_etime); 361 else 362 printf(" %4s ", ""); 363 } 364 365 if (!lflag) { 366 if (jflag) 367 printf("%11.2fcp ", t / (double) cip->ci_calls); 368 else 369 printf("%11.2fcp ", t / 60.0); 370 if (cflag) { 371 if (cip != totalcip) 372 printf(" %4.2f%% ", 373 (cip->ci_utime + cip->ci_stime) / (double) 374 (totalcip->ci_utime + totalcip->ci_stime)); 375 else 376 printf(" %4s ", ""); 377 } 378 } else { 379 if (jflag) 380 printf("%11.2fu ", cip->ci_utime / (double) (AHZ * c)); 381 else 382 printf("%11.2fu ", cip->ci_utime / (60.0 * AHZ)); 383 if (cflag) { 384 if (cip != totalcip) 385 printf(" %4.2f%% ", cip->ci_utime / (double) totalcip->ci_utime); 386 else 387 printf(" %4s ", ""); 388 } 389 if (jflag) 390 printf("%11.2fs ", cip->ci_stime / (double) (AHZ * c)); 391 else 392 printf("%11.2fs ", cip->ci_stime / (60.0 * AHZ)); 393 if (cflag) { 394 if (cip != totalcip) 395 printf(" %4.2f%% ", cip->ci_stime / (double) totalcip->ci_stime); 396 else 397 printf(" %4s ", ""); 398 } 399 } 400 401 if (tflag) 402 if (!uflow) 403 printf("%8.2fre/cp ", cip->ci_etime / (double) (cip->ci_utime + cip->ci_stime)); 404 else 405 printf("*ignore* "); 406 407 if (Dflag) 408 printf("%10qutio ", cip->ci_io); 409 else 410 printf("%8.0favio ", cip->ci_io / c); 411 412 if (Kflag) 413 printf("%10quk*sec ", cip->ci_mem); 414 else 415 printf("%8.0fk ", cip->ci_mem / t); 416 417 printf(" %s\n", cip->ci_comm); 418 } 419