git.sur5r.net Git - i3/i3status/blob - src/print_cpu_usage.c

   1 // vim:sw=8:sts=8:ts=8:expandtab
   2 #include <stdlib.h>
   3 #include <limits.h>
   4 #include <stdio.h>
   5 #include <string.h>
   6
   7 #ifdef __FreeBSD__
   8 #include <sys/types.h>
   9 #include <sys/sysctl.h>
  10 #include <sys/dkstat.h>
  11 #endif
  12
  13 #include "i3status.h"
  14
  15 static int prev_total = 0;
  16 static int prev_idle  = 0;
  17
  18 /*
  19  * Reads the CPU utilization from /proc/stat and returns the usage as a
  20  * percentage.
  21  *
  22  */
  23 void print_cpu_usage(const char *format) {
  24         const char *walk;
  25         char buf[1024];
  26         int curr_user, curr_nice, curr_system, curr_idle, curr_total;
  27         int diff_idle, diff_total, diff_usage;
  28
  29 #if defined(LINUX)
  30         static char statpath[512];
  31         strcpy(statpath, "/proc/stat");
  32         if (!slurp(statpath, buf, sizeof(buf)))
  33                 die("could not read %s\n", statpath);
  34
  35         if (sscanf(buf, "cpu %d %d %d %d", &curr_user, &curr_nice, &curr_system, &curr_idle) != 4)
  36                 die("could not read cpu utilization\n");
  37
  38         curr_total = curr_user + curr_nice + curr_system + curr_idle;
  39         diff_idle  = curr_idle - prev_idle;
  40         diff_total = curr_total - prev_total;
  41         diff_usage = (1000 * (diff_total - diff_idle)/diff_total + 5)/10;
  42         prev_total = curr_total;
  43         prev_idle  = curr_idle;
  44 #endif
  45 #if defined(__FreeBSD__)
  46         size_t size;
  47         long cp_time[CPUSTATES];
  48         size = sizeof cp_time;
  49         if (sysctlbyname("kern.cp_time", &cp_time, &size, NULL, 0) < 0){
  50                 return;
  51         }
  52         curr_user = cp_time[CP_USER];
  53         curr_nice = cp_time[CP_NICE];
  54         curr_system = cp_time[CP_SYS];
  55         curr_idle = cp_time[CP_IDLE];
  56         curr_total = curr_user + curr_nice + curr_system + curr_idle;
  57         diff_idle  = curr_idle - prev_idle;
  58         diff_total = curr_total - prev_total;
  59         diff_usage = (1000 * (diff_total - diff_idle)/diff_total + 5)/10;
  60         prev_total = curr_total;
  61         prev_idle  = curr_idle;
  62
  63 #endif
  64         for (walk = format; *walk != '\0'; walk++) {
  65                 if (*walk != '%') {
  66                         putchar(*walk);
  67                         continue;
  68                 }
  69
  70                 if (strncmp(walk+1, "usage", strlen("usage")) == 0) {
  71                         printf("%02d%%", diff_usage);
  72                         walk += strlen("usage");
  73                 }
  74         }
  75 }