src/bootchart/bootchart.c

   1 /*
   2  * bootchart.c
   3  *
   4  * Copyright (C) 2009-2012 Intel Coproration
   5  *
   6  * Authors:
   7  *   Auke Kok <auke-jan.h.kok@intel.com>
   8  *
   9  * This program is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU General Public License
  11  * as published by the Free Software Foundation; version 2
  12  * of the License.
  13  */
  14
  15
  16 #include <sys/time.h>
  17 #include <sys/types.h>
  18 #include <sys/resource.h>
  19 #include <stdio.h>
  20 #include <signal.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <unistd.h>
  24 #include <time.h>
  25 #include <getopt.h>
  26 #include <limits.h>
  27 #include <errno.h>
  28
  29
  30 #include "bootchart.h"
  31
  32 double graph_start;
  33 double log_start;
  34 double sampletime[MAXSAMPLES];
  35 struct ps_struct *ps_first;
  36 struct block_stat_struct blockstat[MAXSAMPLES];
  37 int entropy_avail[MAXSAMPLES];
  38 struct cpu_stat_struct cpustat[MAXCPUS];
  39 int pscount;
  40 int cpus;
  41 double interval;
  42 FILE *of;
  43 int overrun = 0;
  44 static int exiting = 0;
  45
  46 /* graph defaults */
  47 int entropy = 0;
  48 int initcall = 1;
  49 int relative;
  50 int filter = 1;
  51 int pss = 0;
  52 int samples;
  53 int len = 500; /* we record len+1 (1 start sample) */
  54 double hz = 25.0;   /* 20 seconds log time */
  55 double scale_x = 100.0; /* 100px = 1sec */
  56 double scale_y = 20.0;  /* 16px = 1 process bar */
  57
  58 char init_path[PATH_MAX] = "/sbin/init";
  59 char output_path[PATH_MAX] = "/var/log";
  60
  61 static struct rlimit rlim;
  62
  63 static void signal_handler(int sig)
  64 {
  65         if (sig++)
  66                 sig--;
  67         exiting = 1;
  68 }
  69
  70
  71 int main(int argc, char *argv[])
  72 {
  73         struct sigaction sig;
  74         struct ps_struct *ps;
  75         char output_file[PATH_MAX];
  76         char datestr[200];
  77         time_t t;
  78         FILE *f;
  79         int gind;
  80         int i;
  81
  82         memset(&t, 0, sizeof(time_t));
  83
  84         rlim.rlim_cur = 4096;
  85         rlim.rlim_max = 4096;
  86         (void) setrlimit(RLIMIT_NOFILE, &rlim);
  87
  88         f = fopen("/etc/systemd/bootchart.conf", "r");
  89         if (f) {
  90                 char buf[256];
  91                 char *key;
  92                 char *val;
  93
  94                 while (fgets(buf, 80, f) != NULL) {
  95                         char *c;
  96
  97                         c = strchr(buf, '\n');
  98                         if (c) *c = 0; /* remove trailing \n */
  99
 100                         if (buf[0] == '#')
 101                                 continue; /* comment line */
 102
 103                         key = strtok(buf, "=");
 104                         if (!key)
 105                                 continue;
 106                         val = strtok(NULL, "=");
 107                         if (!val)
 108                                 continue;
 109
 110                         // todo: filter leading/trailing whitespace
 111
 112                         if (!strcmp(key, "samples"))
 113                                 len = atoi(val);
 114                         if (!strcmp(key, "freq"))
 115                                 hz = atof(val);
 116                         if (!strcmp(key, "rel"))
 117                                 relative = atoi(val);
 118                         if (!strcmp(key, "filter"))
 119                                 filter = atoi(val);
 120                         if (!strcmp(key, "pss"))
 121                                 pss = atoi(val);
 122                         if (!strcmp(key, "output"))
 123                                 strncpy(output_path, val, PATH_MAX - 1);
 124                         if (!strcmp(key, "init"))
 125                                 strncpy(init_path, val, PATH_MAX - 1);
 126                         if (!strcmp(key, "scale_x"))
 127                                 scale_x = atof(val);
 128                         if (!strcmp(key, "scale_y"))
 129                                 scale_y = atof(val);
 130                         if (!strcmp(key, "entropy"))
 131                                 entropy = atoi(val);
 132                 }
 133                 fclose(f);
 134         }
 135
 136         while (1) {
 137                 static struct option opts[] = {
 138                         {"rel", 0, NULL, 'r'},
 139                         {"freq", 1, NULL, 'f'},
 140                         {"samples", 1, NULL, 'n'},
 141                         {"pss", 0, NULL, 'p'},
 142                         {"output", 1, NULL, 'o'},
 143                         {"init", 1, NULL, 'i'},
 144                         {"filter", 0, NULL, 'F'},
 145                         {"help", 0, NULL, 'h'},
 146                         {"scale-x", 1, NULL, 'x'},
 147                         {"scale-y", 1, NULL, 'y'},
 148                         {"entropy", 0, NULL, 'e'},
 149                         {NULL, 0, NULL, 0}
 150                 };
 151
 152                 gind = 0;
 153
 154                 i = getopt_long(argc, argv, "erpf:n:o:i:Fhx:y:", opts, &gind);
 155                 if (i == -1)
 156                         break;
 157                 switch (i) {
 158                 case 'r':
 159                         relative = 1;
 160                         break;
 161                 case 'f':
 162                         hz = atof(optarg);
 163                         break;
 164                 case 'F':
 165                         filter = 0;
 166                         break;
 167                 case 'n':
 168                         len = atoi(optarg);
 169                         break;
 170                 case 'o':
 171                         strncpy(output_path, optarg, PATH_MAX - 1);
 172                         break;
 173                 case 'i':
 174                         strncpy(init_path, optarg, PATH_MAX - 1);
 175                         break;
 176                 case 'p':
 177                         pss = 1;
 178                         break;
 179                 case 'x':
 180                         scale_x = atof(optarg);
 181                         break;
 182                 case 'y':
 183                         scale_y = atof(optarg);
 184                         break;
 185                 case 'e':
 186                         entropy = 1;
 187                         break;
 188                 case 'h':
 189                         fprintf(stderr, "Usage: %s [OPTIONS]\n", argv[0]);
 190                         fprintf(stderr, " --rel,     -r            Record time relative to recording\n");
 191                         fprintf(stderr, " --freq,    -f N          Sample frequency [%f]\n", hz);
 192                         fprintf(stderr, " --samples, -n N          Stop sampling at [%d] samples\n", len);
 193                         fprintf(stderr, " --scale-x, -x N          Scale the graph horizontally [%f] \n", scale_x);
 194                         fprintf(stderr, " --scale-y, -y N          Scale the graph vertically [%f] \n", scale_y);
 195                         fprintf(stderr, " --pss,     -p            Enable PSS graph (CPU intensive)\n");
 196                         fprintf(stderr, " --entropy, -e            Enable the entropy_avail graph\n");
 197                         fprintf(stderr, " --output,  -o [PATH]     Path to output files [%s]\n", output_path);
 198                         fprintf(stderr, " --init,    -i [PATH]     Path to init executable [%s]\n", init_path);
 199                         fprintf(stderr, " --filter,  -F            Disable filtering of processes from the graph\n");
 200                         fprintf(stderr, "                          that are of less importance or short-lived\n");
 201                         fprintf(stderr, " --help,    -h            Display this message\n");
 202                         fprintf(stderr, "See the installed README and bootchartd.conf.example for more information.\n");
 203                         exit (EXIT_SUCCESS);
 204                         break;
 205                 default:
 206                         break;
 207                 }
 208         }
 209
 210         if (len > MAXSAMPLES) {
 211                 fprintf(stderr, "Error: samples exceeds maximum\n");
 212                 exit(EXIT_FAILURE);
 213         }
 214
 215         if (hz <= 0.0) {
 216                 fprintf(stderr, "Error: Frequency needs to be > 0\n");
 217                 exit(EXIT_FAILURE);
 218         }
 219
 220         /*
 221          * If the kernel executed us through init=/sbin/bootchartd, then
 222          * fork:
 223          * - parent execs executable specified via init_path[] (/sbin/init by default) as pid=1
 224          * - child logs data
 225          */
 226         if (getpid() == 1) {
 227                 if (fork()) {
 228                         /* parent */
 229                         execl(init_path, init_path, NULL);
 230                 }
 231         }
 232
 233         /* start with empty ps LL */
 234         ps_first = malloc(sizeof(struct ps_struct));
 235         if (!ps_first) {
 236                 perror("malloc(ps_struct)");
 237                 exit(EXIT_FAILURE);
 238         }
 239         memset(ps_first, 0, sizeof(struct ps_struct));
 240
 241         /* handle TERM/INT nicely */
 242         memset(&sig, 0, sizeof(struct sigaction));
 243         sig.sa_handler = signal_handler;
 244         sigaction(SIGHUP, &sig, NULL);
 245
 246         interval = (1.0 / hz) * 1000000000.0;
 247
 248         log_uptime();
 249
 250         /* main program loop */
 251         while (!exiting) {
 252                 int res;
 253                 double sample_stop;
 254                 struct timespec req;
 255                 time_t newint_s;
 256                 long newint_ns;
 257                 double elapsed;
 258                 double timeleft;
 259
 260                 sampletime[samples] = gettime_ns();
 261
 262                 /* wait for /proc to become available, discarding samples */
 263                 if (!(graph_start > 0.0))
 264                         log_uptime();
 265                 else
 266                         log_sample(samples);
 267
 268                 sample_stop = gettime_ns();
 269
 270                 elapsed = (sample_stop - sampletime[samples]) * 1000000000.0;
 271                 timeleft = interval - elapsed;
 272
 273                 newint_s = (time_t)(timeleft / 1000000000.0);
 274                 newint_ns = (long)(timeleft - (newint_s * 1000000000.0));
 275
 276                 /*
 277                  * check if we have not consumed our entire timeslice. If we
 278                  * do, don't sleep and take a new sample right away.
 279                  * we'll lose all the missed samples and overrun our total
 280                  * time
 281                  */
 282                 if ((newint_ns > 0) || (newint_s > 0)) {
 283                         req.tv_sec = newint_s;
 284                         req.tv_nsec = newint_ns;
 285
 286                         res = nanosleep(&req, NULL);
 287                         if (res) {
 288                                 if (errno == EINTR) {
 289                                         /* caught signal, probably HUP! */
 290                                         break;
 291                                 }
 292                                 perror("nanosleep()");
 293                                 exit (EXIT_FAILURE);
 294                         }
 295                 } else {
 296                         overrun++;
 297                         /* calculate how many samples we lost and scrap them */
 298                         len = len + ((int)(newint_ns / interval));
 299                 }
 300
 301                 samples++;
 302
 303                 if (samples > len)
 304                         break;
 305
 306         }
 307
 308         /* do some cleanup, close fd's */
 309         ps = ps_first;
 310         while (ps->next_ps) {
 311                 ps = ps->next_ps;
 312                 if (ps->schedstat)
 313                         close(ps->schedstat);
 314                 if (ps->sched)
 315                         close(ps->sched);
 316                 if (ps->smaps)
 317                         fclose(ps->smaps);
 318         }
 319         closedir(proc);
 320
 321         t = time(NULL);
 322         strftime(datestr, sizeof(datestr), "%Y%m%d-%H%M", localtime(&t));
 323         snprintf(output_file, PATH_MAX, "%s/bootchart-%s.svg", output_path, datestr);
 324
 325         of = fopen(output_file, "w");
 326         if (!of) {
 327                 perror("open output_file");
 328                 exit (EXIT_FAILURE);
 329         }
 330
 331         svg_do();
 332
 333         fprintf(stderr, "bootchartd: Wrote %s\n", output_file);
 334         fclose(of);
 335
 336         /* nitpic cleanups */
 337         ps = ps_first;
 338         while (ps->next_ps) {
 339                 struct ps_struct *old = ps;
 340                 ps = ps->next_ps;
 341                 free(old->sample);
 342                 free(old);
 343         }
 344         free(ps->sample);
 345         free(ps);
 346
 347         /* don't complain when overrun once, happens most commonly on 1st sample */
 348         if (overrun > 1)
 349                 fprintf(stderr, "bootchartd: Warning: sample time overrun %i times\n", overrun);
 350
 351         return 0;
 352 }