--- projects/libstatgrab/src/saidar/saidar.c 2004/04/05 15:40:17 1.27 +++ projects/libstatgrab/src/saidar/saidar.c 2007/01/05 13:30:36 1.41 @@ -1,5 +1,5 @@ /* - * i-scream central monitoring system + * i-scream libstatgrab * http://www.i-scream.org * Copyright (C) 2000-2004 i-scream * @@ -17,7 +17,7 @@ * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * - * $Id: saidar.c,v 1.27 2004/04/05 15:40:17 ats Exp $ + * $Id: saidar.c,v 1.41 2007/01/05 13:30:36 tdb Exp $ */ #ifdef HAVE_CONFIG_H @@ -36,13 +36,31 @@ #include #include #include +#include #ifdef HAVE_NCURSES_H -#include -#else -#include +#define COLOR_SUPPORT #endif +#include CURSES_HEADER_FILE +#define THRESHOLD_LOAD 1.0 + +#define THRESHOLD_WARN_ZMB 0 + +#define THRESHOLD_WARN_CPU 60.0 +#define THRESHOLD_ALERT_CPU 90.0 + +#define THRESHOLD_WARN_SWAP 75.0 +#define THRESHOLD_ALERT_SWAP 90.0 + +#define THRESHOLD_WARN_MEM 75.0 +#define THRESHOLD_ALERT_MEM 90.0 + +#define THRESHOLD_WARN_DISK 75.0 +#define THRESHOLD_ALERT_DISK 90.0 + +int sig_winch_flag = 0; + typedef struct{ sg_cpu_percents *cpu_percents; sg_mem_stats *mem_stats; @@ -63,14 +81,20 @@ typedef struct{ sg_host_info *host_info; sg_user_stats *user_stats; }stats_t; - + stats_t stats; char *size_conv(long long number){ char type[] = {'B', 'K', 'M', 'G', 'T'}; int x=0; + int sign=1; static char string[10]; + if(number < 0){ + sign=-1; + number=-number; + } + for(;x<5;x++){ if( (number/1024) < (100)) { break; @@ -78,16 +102,18 @@ char *size_conv(long long number){ number = (number/1024); } - snprintf(string, 10, "%lld%c", number, type[x]); + number = number*sign; + + snprintf(string, 10, "%lld%c", number, type[x]); return string; - + } char *hr_uptime(time_t time){ int day = 0, hour = 0, min = 0; static char uptime_str[25]; int sec = (int) time; - + day = sec / (24*60*60); sec = sec % (24*60*60); hour = sec / (60*60); @@ -152,12 +178,12 @@ void display_headings(){ printw("Mem Free :"); /* Swap */ - move(6, 21); - printw("Swap Total:"); - move(7, 21); - printw("Swap Used :"); - move(8, 21); - printw("Swap Free :"); + move(6, 21); + printw("Swap Total:"); + move(7, 21); + printw("Swap Used :"); + move(8, 21); + printw("Swap Free :"); /* VM */ move(6, 42); @@ -179,7 +205,7 @@ void display_headings(){ move(10,15); printw("Read"); move(10,28); - printw("Write"); + printw("Write"); line = 10; if (stats.network_io_stats != NULL) { @@ -203,7 +229,7 @@ void display_headings(){ refresh(); } -void display_data(){ +void display_data(int colors){ char cur_time[20]; struct tm *tm_time; time_t epoc_time; @@ -228,7 +254,10 @@ void display_data(){ */ if (ptr != NULL){ *ptr = '\0'; - } + } + if (colors) { + attron(COLOR_PAIR(1)); + } printw("%s", hostname); move(0,36); printw("%s", hr_uptime(stats.host_info->uptime)); @@ -237,16 +266,40 @@ void display_data(){ strftime(cur_time, 20, "%Y-%m-%d %T", tm_time); move(0,61); printw("%s", cur_time); + if (colors) { + attroff(COLOR_PAIR(1)); + } } if (stats.load_stats != NULL) { /* Load */ + if (colors) { + attron(COLOR_PAIR(6)); + } move(2,12); + if (colors && fabs(stats.load_stats->min1 - stats.load_stats->min5) > THRESHOLD_LOAD) { + attron(A_BOLD); + } printw("%6.2f", stats.load_stats->min1); + if (colors) { + attroff(A_BOLD); + } move(3,12); + if (colors && fabs(stats.load_stats->min5 - stats.load_stats->min15) > THRESHOLD_LOAD) { + attron(A_BOLD); + } printw("%6.2f", stats.load_stats->min5); + if (colors) { + attroff(A_BOLD); + } move(4,12); + if (colors && fabs(stats.load_stats->min1 - stats.load_stats->min15) > THRESHOLD_LOAD) { + attron(A_BOLD); + } printw("%6.2f", stats.load_stats->min15); + if (colors) { + attroff(A_BOLD); + } } if (stats.cpu_percents != NULL) { @@ -256,7 +309,19 @@ void display_data(){ move(3,33); printw("%6.2f%%", (stats.cpu_percents->kernel + stats.cpu_percents->iowait + stats.cpu_percents->swap)); move(4,33); + if (colors && stats.cpu_percents->user + stats.cpu_percents->nice > THRESHOLD_ALERT_CPU) { + attron(A_STANDOUT); + attron(A_BOLD); + } + else if (colors && stats.cpu_percents->user + stats.cpu_percents->nice > THRESHOLD_WARN_CPU) { + attron(A_BOLD); + } printw("%6.2f%%", (stats.cpu_percents->user + stats.cpu_percents->nice)); + if(colors) { + attroff(A_BOLD); + attroff(A_STANDOUT); + attron(COLOR_PAIR(6)); + } } if (stats.process_count != NULL) { @@ -264,7 +329,15 @@ void display_data(){ move(2, 54); printw("%5d", stats.process_count->running); move(2,74); + if (colors && stats.process_count->zombie > THRESHOLD_WARN_ZMB) { + attron(A_STANDOUT); + attron(A_BOLD); + } printw("%5d", stats.process_count->zombie); + if(colors) { + attroff(A_STANDOUT); + attroff(A_BOLD); + } move(3, 54); printw("%5d", stats.process_count->sleeping); move(3, 74); @@ -277,17 +350,21 @@ void display_data(){ printw("%5d", stats.user_stats->num_entries); } + if(colors) { + attroff(COLOR_PAIR(6)); + attron(COLOR_PAIR(5)); + } if (stats.mem_stats != NULL) { /* Mem */ move(6, 12); - printw("%7s", size_conv(stats.mem_stats->total)); + printw("%7s", size_conv(stats.mem_stats->total)); move(7, 12); - printw("%7s", size_conv(stats.mem_stats->used)); + printw("%7s", size_conv(stats.mem_stats->used)); move(8, 12); printw("%7s", size_conv(stats.mem_stats->free)); } - - if (stats.swap_stats != NULL) { + + if (stats.swap_stats != NULL) { /* Swap */ move(6, 32); printw("%8s", size_conv(stats.swap_stats->total)); @@ -298,39 +375,74 @@ void display_data(){ } /* VM */ - if (stats.mem_stats != NULL && stats.mem_stats->total != 0) { + if (stats.mem_stats != NULL && stats.mem_stats->total != 0) { + float f = 100.00 * (float)(stats.mem_stats->used)/stats.mem_stats->total; + if (colors && f > THRESHOLD_ALERT_MEM) { + attron(A_STANDOUT); + attron(A_BOLD); + } + else if (colors && f > THRESHOLD_WARN_MEM) { + attron(A_BOLD); + } move(6, 54); - printw("%5.2f%%", (100.00 * (float)(stats.mem_stats->used)/stats.mem_stats->total)); + printw("%5.2f%%", f); + if (colors) { + attroff(A_STANDOUT); + attroff(A_BOLD); + attron(COLOR_PAIR(5)); + } } - if (stats.swap_stats != NULL && stats.swap_stats->total != 0) { + if (stats.swap_stats != NULL && stats.swap_stats->total != 0) { + float f = 100.00 * (float)(stats.swap_stats->used)/stats.swap_stats->total; + if (colors && f > THRESHOLD_ALERT_SWAP) { + attron(A_STANDOUT); + attron(A_BOLD); + } + else if (colors && f > THRESHOLD_WARN_SWAP) { + attron(A_BOLD); + } move(7, 54); - printw("%5.2f%%", (100.00 * (float)(stats.swap_stats->used)/stats.swap_stats->total)); + printw("%5.2f%%", f); + if (colors) { + attroff(A_STANDOUT); + attroff(A_BOLD); + attron(COLOR_PAIR(5)); + } } if (stats.mem_stats != NULL && stats.swap_stats != NULL && - stats.mem_stats->total != 0 && stats.swap_stats->total != 0) { + stats.mem_stats->total != 0 && stats.swap_stats->total != 0) { move(8, 54); printw("%5.2f%%", (100.00 * (float)(stats.mem_stats->used+stats.swap_stats->used)/(stats.mem_stats->total+stats.swap_stats->total))); } - if (stats.page_stats != NULL) { + if (stats.page_stats != NULL) { /* Paging */ move(6, 74); printw("%5d", (stats.page_stats->systime)? (stats.page_stats->pages_pagein / stats.page_stats->systime): stats.page_stats->pages_pagein); move(7, 74); printw("%5d", (stats.page_stats->systime)? (stats.page_stats->pages_pageout / stats.page_stats->systime) : stats.page_stats->pages_pageout); } + if (colors) { + attroff(COLOR_PAIR(5)); + } line = 11; - if (stats.disk_io_stats != NULL) { + if (stats.disk_io_stats != NULL) { /* Disk IO */ disk_io_stat_ptr = stats.disk_io_stats; r=0; w=0; for(counter=0;counterdisk_name, sizeof(name)); + name[sizeof(name)-1] = '\0'; /* strncpy doesn't terminate longer strings */ move(line, 0); - printw("%s", disk_io_stat_ptr->disk_name); + printw("%s", name); move(line, 12); rt = (disk_io_stat_ptr->systime)? (disk_io_stat_ptr->read_bytes/disk_io_stat_ptr->systime): disk_io_stat_ptr->read_bytes; + if(colors) { + attron(COLOR_PAIR(4)); + } printw("%7s", size_conv(rt)); r+=rt; move(line, 26); @@ -339,58 +451,91 @@ void display_data(){ w+=wt; disk_io_stat_ptr++; line++; + if(colors) { + attroff(COLOR_PAIR(4)); + } } line++; move(line, 0); printw("Total"); move(line, 12); + if(colors) { + attron(COLOR_PAIR(4)); + } printw("%7s", size_conv(r)); move(line, 26); printw("%7s", size_conv(w)); + if(colors) { + attroff(COLOR_PAIR(4)); + } } line = 11; - if (stats.network_io_stats != NULL) { + if (stats.network_io_stats != NULL) { /* Network */ network_stat_ptr = stats.network_io_stats; for(counter=0;counterinterface_name); + char name[20]; + strncpy(name, network_stat_ptr->interface_name, sizeof(name)); + name[sizeof(name)-1] = '\0'; /* strncpy doesn't terminate longer strings */ + move(line, 42); + printw("%s", name); move(line, 62); rt = (network_stat_ptr->systime)? (network_stat_ptr->rx / network_stat_ptr->systime): network_stat_ptr->rx; + if(colors) { + attron(COLOR_PAIR(4)); + } printw("%7s", size_conv(rt)); move(line, 72); - wt = (network_stat_ptr->systime)? (network_stat_ptr->tx / network_stat_ptr->systime): network_stat_ptr->tx; + wt = (network_stat_ptr->systime)? (network_stat_ptr->tx / network_stat_ptr->systime): network_stat_ptr->tx; printw("%7s", size_conv(wt)); network_stat_ptr++; line++; + if(colors) { + attroff(COLOR_PAIR(4)); + } } line += 2; } - if (stats.fs_stats != NULL) { + if (stats.fs_stats != NULL) { /* Disk */ disk_stat_ptr = stats.fs_stats; for(counter=0;countermnt_point, sizeof(name)); + name[sizeof(name)-1] = '\0'; /* strncpy doesn't terminate longer strings */ move(line, 42); - printw("%s", disk_stat_ptr->mnt_point); + printw("%s", name); move(line, 62); + if(colors) { + attron(COLOR_PAIR(2)); + } printw("%7s", size_conv(disk_stat_ptr->avail)); move(line, 73); - printw("%5.2f%%", 100.00 * ((float) disk_stat_ptr->used / (float) (disk_stat_ptr->used + disk_stat_ptr->avail))); + if(colors && 100.00 * ((float) disk_stat_ptr->used / (float) (disk_stat_ptr->used + disk_stat_ptr->avail)) > THRESHOLD_ALERT_DISK) { + attron(A_STANDOUT); + attron(A_BOLD); + } else if (colors && 100.00 * ((float) disk_stat_ptr->used / (float) (disk_stat_ptr->used + disk_stat_ptr->avail)) > THRESHOLD_WARN_DISK) { + attron(A_BOLD); + } + printw("%6.2f%%", 100.00 * ((float) disk_stat_ptr->used / (float) (disk_stat_ptr->used + disk_stat_ptr->avail))); disk_stat_ptr++; line++; + if(colors) { + attroff(COLOR_PAIR(2)); + attroff(A_STANDOUT); + attroff(A_BOLD); + } } } refresh(); } -void sig_winch_handler(int sig){ - clear(); - display_headings(); - display_data(); - signal(SIGWINCH, sig_winch_handler); +void sig_winch_handler(int dummy){ + sig_winch_flag = 1; + signal(SIGWINCH, sig_winch_handler); } int get_stats(){ @@ -416,19 +561,26 @@ void version_num(char *progname){ } void usage(char *progname){ - fprintf(stderr, "Usage: %s [-d delay] [-v] [-h]\n\n", progname); - fprintf(stderr, " -d Sets the update time in seconds\n"); +#ifdef COLOR_SUPPORT + fprintf(stderr, "Usage: %s [-d delay] [-c] [-v] [-h]\n\n", progname); +#else + fprintf(stderr, "Usage: %s [-d delay] [-v] [-h]\n\n", progname); +#endif + fprintf(stderr, " -d Sets the update time in seconds\n"); +#ifdef COLOR_SUPPORT + fprintf(stderr, " -c Enables coloured output\n"); +#endif fprintf(stderr, " -v Prints version number\n"); - fprintf(stderr, " -h Displays this help information.\n"); - fprintf(stderr, "\nReport bugs to <%s>.\n", PACKAGE_BUGREPORT); - exit(1); + fprintf(stderr, " -h Displays this help information.\n"); + fprintf(stderr, "\nReport bugs to <%s>.\n", PACKAGE_BUGREPORT); + exit(1); } int main(int argc, char **argv){ + extern char *optarg; + int c; + int colouron = 0; - extern char *optarg; - int c; - time_t last_update = 0; WINDOW *window; @@ -442,34 +594,62 @@ int main(int argc, char **argv){ fprintf(stderr, "Failed to drop setuid/setgid privileges\n"); return 1; } - - while ((c = getopt(argc, argv, "vhd:")) != -1){ - switch (c){ - case 'd': - delay = atoi(optarg); + +#ifdef COLOR_SUPPORT + while ((c = getopt(argc, argv, "d:cvh")) != -1){ +#else + while ((c = getopt(argc, argv, "d:vh")) != -1){ +#endif + switch (c){ + case 'd': + delay = atoi(optarg); if (delay < 1){ fprintf(stderr, "Time must be 1 second or greater\n"); exit(1); } - break; + break; +#ifdef COLOR_SUPPORT + case 'c': + colouron = 1; + break; +#endif case 'v': - version_num(argv[0]); + version_num(argv[0]); break; case 'h': default: usage(argv[0]); return 1; break; - } - } + } + } signal(SIGWINCH, sig_winch_handler); - initscr(); - nonl(); - cbreak(); - noecho(); + initscr(); +#ifdef COLOR_SUPPORT + /* turn on colour */ + if (colouron) { + if (has_colors()) { + start_color(); + use_default_colors(); + init_pair(1,COLOR_RED,-1); + init_pair(2,COLOR_GREEN,-1); + init_pair(3,COLOR_YELLOW,-1); + init_pair(4,COLOR_BLUE,-1); + init_pair(5,COLOR_MAGENTA,-1); + init_pair(6,COLOR_CYAN,-1); + } else { + fprintf(stderr, "Colour support disabled: your terminal does not support colour."); + colouron = 0; + } + } +#endif + nonl(); + curs_set(0); + cbreak(); + noecho(); timeout(delay * 1000); - window=newwin(0, 0, 0, 0); + window=newwin(0, 0, 0, 0); clear(); if(!get_stats()){ @@ -497,9 +677,15 @@ int main(int argc, char **argv){ } last_update = now; - display_data(); - } + if(sig_winch_flag) { + clear(); + display_headings(); + sig_winch_flag = 0; + } + display_data(colouron); + } + endwin(); return 0; -} +}