I/O Detection program

[Smotang]

// SE Toolkit program - http://www.setoolkit.com
//
// psio.se - a "ps -ef" style tool that prints I/O. Solaris, SE Toolkit.
//
// This program could assist finding a process that is causing the most I/O.
// The I/O reported is the total of all types: disk I/O, network I/O,
// stream I/O, etc..
//
// This is an SE Toolkit version of the perl "psio" program. The perl version
// reports disk I/O only, and has many different switches and modes of
// operation. Future versions of this SE Toolkit program are unlikely to
// have any extra features (for more features, see "pea.se").
//
// 12-Mar-2004, ver 0.70
//
//
// USAGE: se psio.se [-h] | [interval [count]]
//
// se psio.se # "ps -ef" style output, 1 second sample
// se psio.se 10 # print a 10 second sample
// se psio.se 10 5 # print 5 x 10 second samples
//
// se psio.se 30 | sort -n +3 # sort by I/O, 30 sec sample
//
// se psio.se 5 12 | grep [f]sflush # examine "fsflush" for 1 min
//
//
// FIELDS:
// IOSIZE Size of I/O (Kbytes)
//
// SEE ALSO: se -DWIDE pea.se # SE Toolkit
// psio # http://www.brendangregg.com/psio.html
//
// COPYRIGHT: Copyright (c) 2004 Brendan Gregg.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// (http://www.gnu.org/copyleft/gpl.html)
//
// Author: Brendan Gregg [Sydney, Australia]
//
// 12-Mar-2004 Brendan Gregg Created this.


#define BLK 8192
#include <stdio.se>
#include <stdlib.se>
#include <unistd.se>
#include <string.se>
#include <dirent.se>
#include <time.se>
#include <fcntl.se>
#include <proc.se>
#include <process_class.se>
#include <pwd.se>


main(int argc, string argv[2]) {
//
// Variable declaration
//
psinfo_t p; /* raw proc info */
proc_class_t pp; /* advanced proc info */
passwd_t user; /* passwd structure, used for uid->name */
ulong user_ptr; /* passwd pointer */
tm_t time_tm; /* time structure, time conversions */
ulong time_ulong; /* temporary time variable */
char time[16]; /* TIME field */
char stime[16]; /* STIME field */
string username; /* username field */
double io_bytes; /* I/O bytes */
double io_kb; /* I/O Kbytes */
int i = 0; /* main loop counter */
int secs; /* interval seconds */
int loop; /* loop limit */

//
// Process command line args
//
if ((argc > 1) && (argv[1] == "-h" || argv[1] == "--help")) {
usage(argv[0]);
}
switch(argc) {
case 3:
secs = atoi(argv[1]);
loop = atoi(argv[2]);
break;
case 2:
secs = atoi(argv[1]);
loop = 1;
break;
default:
secs = 0;
loop = 1;
break;
}

//
// Prime initial I/O values
//
refresh$(pp);
pp.action$ = PROC_ACTION_NEXT_INDEX;
for (;;) {
refresh$(pp);
if (pp.index$ == -1) { break; }
// we read the following to reset the counters
io_bytes = pp.inblocks*BLK + pp.outblocks*BLK + pp.charios;
}


//
// Main loop
//
for (;;) {

// print header
if (i > 0) { printf("\n"); }
printf(" UID PID PPID IOSIZE STIME TTY TIME CMD\n");

// sleep while I/O occurs
sleep(secs);

//
// Print process listing
//
for (;;) {
refresh$(pp);
if (pp.index$ == -1) { break; }

// we need this structure to fetch the start time and tty
p = get_proc(pp.pid$);

// calculate start time
time_ulong = p.pr_start.tv_sec;
time_tm = localtime(&time_ulong);
strftime(stime, sizeof(stime), "%b %d", time_tm);

// calculate CPU time
time_ulong = pp.total_user + pp.total_system;
time_tm = gmtime(&time_ulong);
strftime(time, sizeof(time), "%M:%S", time_tm);

// get username
user_ptr = getpwuid(pp.uid);
user = *((passwd_t *) user_ptr);
username = user.pw_name;

// calculate I/O by bytes
io_bytes = pp.inblocks*BLK + pp.outblocks*BLK + pp.charios;
io_kb = io_bytes / 1024;

printf("%8s %5d %5d %7.1f %8s %-6s %6s %s\n",
username, pp.pid$, pp.ppid, io_kb, stime,
dev_to_tty(p.pr_ttydev), time, pp.args);
}

if (++i == loop) { break; }
}
}



//
// Subroutines
//

// usage - print usage and exit.
//
usage (string cmd)
{
printf("USAGE: se %s [-h] | [interval [count]]\n",cmd);
printf(" eg, se %s # print a 1 second sample\n",cmd);
printf(" eg, se %s 10 # print a 10 second sample\n",cmd);
printf(" eg, se %s 10 5 # print 5 x 10 second samples\n",cmd);
exit(1);
}




Very helpful piece of Code.