lnet/utils/routerstat.c

   1 #include <stdio.h>
   2 #include <errno.h>
   3 #include <string.h>
   4 #include <fcntl.h>
   5 #include <unistd.h>
   6 #include <stdlib.h>
   7 #include <sys/types.h>
   8 #include <sys/time.h>
   9
  10 double
  11 timenow ()
  12 {
  13    struct timeval tv;
  14
  15    gettimeofday (&tv, NULL);
  16    return (tv.tv_sec + tv.tv_usec / 1000000.0);
  17 }
  18
  19 typedef struct {
  20         unsigned long        msgs_alloc;
  21         unsigned long        msgs_max;
  22         unsigned long        errors;
  23         unsigned long        send_count;
  24         unsigned long        recv_count;
  25         unsigned long        route_count;
  26         unsigned long        drop_count;
  27         unsigned long long   send_length;
  28         unsigned long long   recv_length;
  29         unsigned long long   route_length;
  30         unsigned long long   drop_length;
  31 } counters_t;
  32
  33 unsigned long long subull(unsigned long long a, unsigned long long b)
  34 {
  35         if (a < b)
  36                 return -1ULL - b + a + 1;
  37
  38         return a - b;
  39 }
  40
  41 unsigned long long subul(unsigned long a, unsigned long b)
  42 {
  43         if (a < b)
  44                 return -1UL - b + a + 1;
  45
  46         return a - b;
  47 }
  48
  49 double rul(unsigned long a, double secs)
  50 {
  51         return (double)a/secs;
  52 }
  53
  54 double rull(unsigned long long a, double secs)
  55 {
  56         return (double)a/secs;
  57 }
  58
  59 void
  60 do_stat (int fd)
  61 {
  62    static char  buffer[1024];
  63    static double last = 0.0;
  64    static counters_t old_counter;
  65    double now;
  66    double t;
  67    counters_t new_counter;
  68    counters_t counter;
  69    int    n;
  70
  71    lseek (fd, 0, SEEK_SET);
  72    now = timenow();
  73    n = read (fd, buffer, sizeof (buffer));
  74    if (n < 0)
  75    {
  76       fprintf (stderr, "Can't read statfile\n");
  77       exit (1);
  78    }
  79    buffer[n] = 0;
  80
  81    n = sscanf (buffer, "%lu %lu %lu %lu %lu %lu %lu %Lu %Lu %Lu %Lu",
  82                &new_counter.msgs_alloc, &new_counter.msgs_max,
  83                &new_counter.errors,
  84                &new_counter.send_count, &new_counter.recv_count,
  85                &new_counter.route_count, &new_counter.drop_count,
  86                &new_counter.send_length, &new_counter.recv_length,
  87                &new_counter.route_length, &new_counter.drop_length);
  88    if (n < 11)
  89    {
  90       fprintf (stderr, "Can't parse statfile\n");
  91       exit (1);
  92    }
  93
  94    if (last == 0.0) {
  95            printf ("M %lu(%lu) E %lu S %lu/%llu R %lu/%llu F %lu/%llu D %lu/%llu\n",
  96                    new_counter.msgs_alloc, new_counter.msgs_max,
  97                    new_counter.errors,
  98                    new_counter.send_count, new_counter.send_length,
  99                    new_counter.recv_count, new_counter.recv_length,
 100                    new_counter.route_count, new_counter.route_length,
 101                    new_counter.drop_count, new_counter.drop_length);
 102    } else {
 103            t = now - last;
 104
 105            counter.msgs_alloc = new_counter.msgs_alloc;
 106            counter.msgs_max = new_counter.msgs_max;
 107
 108            counter.errors = subul(new_counter.errors, old_counter.errors);
 109            counter.send_count = subul(new_counter.send_count, old_counter.send_count);
 110            counter.recv_count = subul(new_counter.recv_count, old_counter.recv_count);
 111            counter.route_count = subul(new_counter.route_count, old_counter.route_count);
 112            counter.drop_count = subul(new_counter.drop_count, old_counter.drop_count);
 113            counter.send_length = subull(new_counter.send_length, old_counter.send_length);
 114            counter.recv_length = subull(new_counter.recv_length, old_counter.recv_length);
 115            counter.route_length = subull(new_counter.route_length, old_counter.route_length);
 116            counter.drop_length = subull(new_counter.drop_length, old_counter.drop_length);
 117
 118            printf ("M %3lu(%3lu) E %0.0f S %7.2f/%6.0f R %7.2f/%6.0f F %7.2f/%6.0f D %4.2f/%0.0f\n",
 119                    counter.msgs_alloc, counter.msgs_max,
 120                    rul(counter.errors,t),
 121                    rull(counter.send_length,t*1024.0*1024.0), rul(counter.send_count, t),
 122                    rull(counter.recv_length,t*1024.0*1024.0), rul(counter.recv_count, t),
 123                    rull(counter.route_length,t*1024.0*1024.0), rul(counter.route_count, t),
 124                    rull(counter.drop_length,t*1024.0*1024.0), rul(counter.drop_count, t));
 125    }
 126
 127    old_counter = new_counter;
 128    fflush (stdout);
 129
 130    lseek (fd, 0, SEEK_SET);
 131    last = timenow();
 132 }
 133
 134 int main (int argc, char **argv)
 135 {
 136    int  interval = 0;
 137    int  fd;
 138
 139    if (argc > 1)
 140       interval = atoi (argv[1]);
 141
 142    fd = open ("/proc/sys/lnet/stats", O_RDONLY);
 143    if (fd < 0)
 144    {
 145       fprintf (stderr, "Can't open stat: %s\n", strerror (errno));
 146       return (1);
 147    }
 148
 149    do_stat (fd);
 150    if (interval == 0)
 151       return (0);
 152
 153    for (;;)
 154    {
 155       sleep (interval);
 156       do_stat (fd);
 157    }
 158 }