LinuxProcessList.c 24.3 KB
Newer Older
1
2
3
4
5
6
7
/*
htop - LinuxProcessList.c
(C) 2014 Hisham H. Muhammad
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/

Hisham Muhammad's avatar
Hisham Muhammad committed
8
9
10
#include "LinuxProcessList.h"
#include "LinuxProcess.h"
#include "CRT.h"
David Hunt's avatar
David Hunt committed
11
#include "StringUtils.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <errno.h>
#include <sys/time.h>
#include <sys/utsname.h>
#include <sys/stat.h>
#include <unistd.h>
#include <dirent.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <stdbool.h>
#include <stdarg.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <assert.h>
#include <sys/types.h>
#include <fcntl.h>
29
30
31

/*{

Hisham Muhammad's avatar
Hisham Muhammad committed
32
33
#include "ProcessList.h"

Hisham Muhammad's avatar
Hisham Muhammad committed
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
typedef struct CPUData_ {
   unsigned long long int totalTime;
   unsigned long long int userTime;
   unsigned long long int systemTime;
   unsigned long long int systemAllTime;
   unsigned long long int idleAllTime;
   unsigned long long int idleTime;
   unsigned long long int niceTime;
   unsigned long long int ioWaitTime;
   unsigned long long int irqTime;
   unsigned long long int softIrqTime;
   unsigned long long int stealTime;
   unsigned long long int guestTime;
   
   unsigned long long int totalPeriod;
   unsigned long long int userPeriod;
   unsigned long long int systemPeriod;
   unsigned long long int systemAllPeriod;
   unsigned long long int idleAllPeriod;
   unsigned long long int idlePeriod;
   unsigned long long int nicePeriod;
   unsigned long long int ioWaitPeriod;
   unsigned long long int irqPeriod;
   unsigned long long int softIrqPeriod;
   unsigned long long int stealPeriod;
   unsigned long long int guestPeriod;
} CPUData;

typedef struct LinuxProcessList_ {
   ProcessList super;

   CPUData* cpus;

} LinuxProcessList;

69
70
71
72
73
74
75
76
77
78
79
80
#ifndef PROCDIR
#define PROCDIR "/proc"
#endif

#ifndef PROCSTATFILE
#define PROCSTATFILE PROCDIR "/stat"
#endif

#ifndef PROCMEMINFOFILE
#define PROCMEMINFOFILE PROCDIR "/meminfo"
#endif

81
82
83
84
#ifndef PROC_LINE_LENGTH
#define PROC_LINE_LENGTH 512
#endif

85
}*/
86
87
88
89

#ifndef CLAMP
#define CLAMP(x,low,high) (((x)>(high))?(high):(((x)<(low))?(low):(x)))
#endif
90
   
Hisham Muhammad's avatar
Hisham Muhammad committed
91
ProcessList* ProcessList_new(UsersTable* usersTable, Hashtable* pidWhiteList, uid_t userId) {
Hisham's avatar
Hisham committed
92
   LinuxProcessList* this = xCalloc(1, sizeof(LinuxProcessList));
Hisham Muhammad's avatar
Hisham Muhammad committed
93
   ProcessList* pl = &(this->super);
94
   ProcessList_init(pl, Class(LinuxProcess), usersTable, pidWhiteList, userId);
95
96
97
98
99
100

   // Update CPU count:
   FILE* file = fopen(PROCSTATFILE, "r");
   if (file == NULL) {
      CRT_fatalError("Cannot open " PROCSTATFILE);
   }
101
   char buffer[PROC_LINE_LENGTH + 1];
102
103
104
   int cpus = -1;
   do {
      cpus++;
105
      char * s = fgets(buffer, PROC_LINE_LENGTH, file);
Christian Hesse's avatar
Christian Hesse committed
106
      (void) s;
107
108
109
   } while (String_startsWith(buffer, "cpu"));
   fclose(file);

Hisham Muhammad's avatar
Hisham Muhammad committed
110
   pl->cpuCount = MAX(cpus - 1, 1);
Hisham's avatar
Hisham committed
111
   this->cpus = xCalloc(cpus, sizeof(CPUData));
112
113
114
115
116
117

   for (int i = 0; i < cpus; i++) {
      this->cpus[i].totalTime = 1;
      this->cpus[i].totalPeriod = 1;
   }

Hisham Muhammad's avatar
Hisham Muhammad committed
118
   return pl;
119
120
}

Hisham Muhammad's avatar
Hisham Muhammad committed
121
122
123
124
void ProcessList_delete(ProcessList* pl) {
   LinuxProcessList* this = (LinuxProcessList*) pl;
   ProcessList_done(pl);
   free(this->cpus);
125
126
127
   free(this);
}

128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
static ssize_t xread(int fd, void *buf, size_t count) {
  // Read some bytes. Retry on EINTR and when we don't get as many bytes as we requested.
  size_t alreadyRead = 0;
  for(;;) {
     ssize_t res = read(fd, buf, count);
     if (res == -1 && errno == EINTR) continue;
     if (res > 0) {
       buf = ((char*)buf)+res;
       count -= res;
       alreadyRead += res;
     }
     if (res == -1) return -1;
     if (count == 0 || res == 0) return alreadyRead;
  }
}

144
145
146
147
148
149
150
151
static double jiffy = 0.0;

static inline unsigned long long LinuxProcess_adjustTime(unsigned long long t) {
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;
   return (unsigned long long) t * jiffytime * 100;
}

152
static bool LinuxProcessList_readStatFile(Process *process, const char* dirname, const char* name, char* command) {
153
   LinuxProcess* lp = (LinuxProcess*) process;
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/stat", dirname, name);
   int fd = open(filename, O_RDONLY);
   if (fd == -1)
      return false;

   static char buf[MAX_READ+1];

   int size = xread(fd, buf, MAX_READ);
   close(fd);
   if (size <= 0) return false;
   buf[size] = '\0';

   assert(process->pid == atoi(buf));
   char *location = strchr(buf, ' ');
   if (!location) return false;

   location += 2;
   char *end = strrchr(location, ')');
   if (!end) return false;
   
   int commsize = end - location;
   memcpy(command, location, commsize);
   command[commsize] = '\0';
   location = end + 2;

   process->state = location[0];
   location += 2;
   process->ppid = strtol(location, &location, 10);
   location += 1;
   process->pgrp = strtoul(location, &location, 10);
   location += 1;
   process->session = strtoul(location, &location, 10);
   location += 1;
   process->tty_nr = strtoul(location, &location, 10);
   location += 1;
   process->tpgid = strtol(location, &location, 10);
   location += 1;
   process->flags = strtoul(location, &location, 10);
   location += 1;
   process->minflt = strtoull(location, &location, 10);
   location += 1;
196
   lp->cminflt = strtoull(location, &location, 10);
197
198
199
   location += 1;
   process->majflt = strtoull(location, &location, 10);
   location += 1;
200
   lp->cmajflt = strtoull(location, &location, 10);
201
   location += 1;
202
   lp->utime = LinuxProcess_adjustTime(strtoull(location, &location, 10));
203
   location += 1;
204
   lp->stime = LinuxProcess_adjustTime(strtoull(location, &location, 10));
205
   location += 1;
206
   lp->cutime = LinuxProcess_adjustTime(strtoull(location, &location, 10));
207
   location += 1;
208
   lp->cstime = LinuxProcess_adjustTime(strtoull(location, &location, 10));
209
210
211
212
213
214
215
216
217
218
219
   location += 1;
   process->priority = strtol(location, &location, 10);
   location += 1;
   process->nice = strtol(location, &location, 10);
   location += 1;
   process->nlwp = strtol(location, &location, 10);
   location += 1;
   for (int i=0; i<17; i++) location = strchr(location, ' ')+1;
   process->exit_signal = strtol(location, &location, 10);
   location += 1;
   assert(location != NULL);
220
   process->processor = strtol(location, &location, 10);
221
222
   
   process->time = lp->utime + lp->stime;
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249

   return true;
}


static bool LinuxProcessList_statProcessDir(Process* process, const char* dirname, char* name, time_t curTime) {
   char filename[MAX_NAME+1];
   filename[MAX_NAME] = '\0';

   snprintf(filename, MAX_NAME, "%s/%s", dirname, name);
   struct stat sstat;
   int statok = stat(filename, &sstat);
   if (statok == -1)
      return false;
   process->st_uid = sstat.st_uid;
  
   struct tm date;
   time_t ctime = sstat.st_ctime;
   process->starttime_ctime = ctime;
   (void) localtime_r((time_t*) &ctime, &date);
   strftime(process->starttime_show, 7, ((ctime > curTime - 86400) ? "%R " : "%b%d "), &date);
   
   return true;
}

#ifdef HAVE_TASKSTATS

250
static void LinuxProcessList_readIoFile(LinuxProcess* process, const char* dirname, char* name, unsigned long long now) {
251
252
253
254
255
   char filename[MAX_NAME+1];
   filename[MAX_NAME] = '\0';

   snprintf(filename, MAX_NAME, "%s/%s/io", dirname, name);
   int fd = open(filename, O_RDONLY);
Hisham Muhammad's avatar
Hisham Muhammad committed
256
257
258
   if (fd == -1) {
      process->io_rate_read_bps = -1;
      process->io_rate_write_bps = -1;
259
      return;
Hisham Muhammad's avatar
Hisham Muhammad committed
260
   }
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
   
   char buffer[1024];
   ssize_t buflen = xread(fd, buffer, 1023);
   close(fd);
   if (buflen < 1) return;
   buffer[buflen] = '\0';
   unsigned long long last_read = process->io_read_bytes;
   unsigned long long last_write = process->io_write_bytes;
   char *buf = buffer;
   char *line = NULL;
   while ((line = strsep(&buf, "\n")) != NULL) {
      switch (line[0]) {
      case 'r':
         if (line[1] == 'c' && strncmp(line+2, "har: ", 5) == 0)
            process->io_rchar = strtoull(line+7, NULL, 10);
         else if (strncmp(line+1, "ead_bytes: ", 11) == 0) {
            process->io_read_bytes = strtoull(line+12, NULL, 10);
            process->io_rate_read_bps = 
               ((double)(process->io_read_bytes - last_read))/(((double)(now - process->io_rate_read_time))/1000);
            process->io_rate_read_time = now;
         }
         break;
      case 'w':
         if (line[1] == 'c' && strncmp(line+2, "har: ", 5) == 0)
            process->io_wchar = strtoull(line+7, NULL, 10);
         else if (strncmp(line+1, "rite_bytes: ", 12) == 0) {
            process->io_write_bytes = strtoull(line+13, NULL, 10);
            process->io_rate_write_bps = 
               ((double)(process->io_write_bytes - last_write))/(((double)(now - process->io_rate_write_time))/1000);
            process->io_rate_write_time = now;
         }
         break;
      case 's':
         if (line[5] == 'r' && strncmp(line+1, "yscr: ", 6) == 0)
            process->io_syscr = strtoull(line+7, NULL, 10);
         else if (strncmp(line+1, "yscw: ", 6) == 0)
            sscanf(line, "syscw: %32llu", &process->io_syscw);
            process->io_syscw = strtoull(line+7, NULL, 10);
         break;
      case 'c':
         if (strncmp(line+1, "ancelled_write_bytes: ", 22) == 0)
           process->io_cancelled_write_bytes = strtoull(line+23, NULL, 10);
      }
   }
}

#endif



311
static bool LinuxProcessList_readStatmFile(LinuxProcess* process, const char* dirname, const char* name) {
312
313
314
315
316
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/statm", dirname, name);
   int fd = open(filename, O_RDONLY);
   if (fd == -1)
      return false;
317
318
   char buf[PROC_LINE_LENGTH + 1];
   ssize_t rres = xread(fd, buf, PROC_LINE_LENGTH);
319
320
321
322
323
   close(fd);
   if (rres < 1) return false;

   char *p = buf;
   errno = 0;
324
325
   process->super.m_size = strtol(p, &p, 10); if (*p == ' ') p++;
   process->super.m_resident = strtol(p, &p, 10); if (*p == ' ') p++;
326
327
328
329
330
331
332
333
334
335
   process->m_share = strtol(p, &p, 10); if (*p == ' ') p++;
   process->m_trs = strtol(p, &p, 10); if (*p == ' ') p++;
   process->m_lrs = strtol(p, &p, 10); if (*p == ' ') p++;
   process->m_drs = strtol(p, &p, 10); if (*p == ' ') p++;
   process->m_dt = strtol(p, &p, 10);
   return (errno == 0);
}

#ifdef HAVE_OPENVZ

336
337
338
static void LinuxProcessList_readOpenVZData(LinuxProcess* process, const char* dirname, const char* name) {
   if ( (access("/proc/vz", R_OK) != 0)) {
      process->vpid = process->super.pid;
339
340
341
342
343
344
      process->ctid = 0;
      return;
   }
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/stat", dirname, name);
   FILE* file = fopen(filename, "r");
345
   if (!file)
346
      return;
347
   (void) fscanf(file,
348
349
350
351
352
353
354
355
356
      "%*32u %*32s %*1c %*32u %*32u %*32u %*32u %*32u %*32u %*32u "
      "%*32u %*32u %*32u %*32u %*32u %*32u %*32u %*32u "
      "%*32u %*32u %*32u %*32u %*32u %*32u %*32u %*32u "
      "%*32u %*32u %*32u %*32u %*32u %*32u %*32u %*32u "
      "%*32u %*32u %*32u %*32u %*32u %*32u %*32u %*32u "
      "%*32u %*32u %*32u %*32u %*32u %*32u %*32u "
      "%*32u %*32u %32u %32u",
      &process->vpid, &process->ctid);
   fclose(file);
357
   return;
358
359
360
361
362
363
}

#endif

#ifdef HAVE_CGROUP

364
static void LinuxProcessList_readCGroupFile(LinuxProcess* process, const char* dirname, const char* name) {
365
366
367
368
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/cgroup", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file) {
Hisham's avatar
Hisham committed
369
      process->cgroup = xStrdup("");
370
371
      return;
   }
372
   char output[PROC_LINE_LENGTH + 1];
Hisham Muhammad's avatar
Hisham Muhammad committed
373
374
   output[0] = '\0';
   char* at = output;
375
   int left = PROC_LINE_LENGTH;
Hisham Muhammad's avatar
Hisham Muhammad committed
376
   while (!feof(file) && left > 0) {
377
378
      char buffer[PROC_LINE_LENGTH + 1];
      char *ok = fgets(buffer, PROC_LINE_LENGTH, file);
Hisham Muhammad's avatar
Hisham Muhammad committed
379
380
381
382
383
384
385
      if (!ok) break;
      char* group = strchr(buffer, ':');
      if (!group) break;
      if (at != output) {
         *at = ';';
         at++;
         left--;
386
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
387
388
      int wrote = snprintf(at, left, "%s", group);
      left -= wrote;
389
390
   }
   fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
391
   free(process->cgroup);
Hisham's avatar
Hisham committed
392
   process->cgroup = xStrdup(output);
393
394
395
396
397
398
}

#endif

#ifdef HAVE_VSERVER

399
static void LinuxProcessList_readVServerData(LinuxProcess* process, const char* dirname, const char* name) {
400
401
402
403
404
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/status", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return;
405
   char buffer[PROC_LINE_LENGTH + 1];
406
   process->vxid = 0;
407
   while (fgets(buffer, PROC_LINE_LENGTH, file)) {
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
      if (String_startsWith(buffer, "VxID:")) {
         int vxid;
         int ok = sscanf(buffer, "VxID:\t%32d", &vxid);
         if (ok >= 1) {
            process->vxid = vxid;
         }
      }
      #if defined HAVE_ANCIENT_VSERVER
      else if (String_startsWith(buffer, "s_context:")) {
         int vxid;
         int ok = sscanf(buffer, "s_context:\t%32d", &vxid);
         if (ok >= 1) {
            process->vxid = vxid;
         }
      }
      #endif
   }
   fclose(file);
}

#endif

430
static void LinuxProcessList_readOomData(LinuxProcess* process, const char* dirname, const char* name) {
431
432
433
434
435
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/oom_score", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return;
436
437
   char buffer[PROC_LINE_LENGTH + 1];
   if (fgets(buffer, PROC_LINE_LENGTH, file)) {
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
      unsigned int oom;
      int ok = sscanf(buffer, "%32u", &oom);
      if (ok >= 1) {
         process->oom = oom;
      }
   }
   fclose(file);
}

static bool LinuxProcessList_readCmdlineFile(Process* process, const char* dirname, const char* name) {
   if (Process_isKernelThread(process))
      return true;

   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/cmdline", dirname, name);
   int fd = open(filename, O_RDONLY);
   if (fd == -1)
      return false;
         
   char command[4096+1]; // max cmdline length on Linux
   int amtRead = xread(fd, command, sizeof(command) - 1);
   close(fd);
   int tokenEnd = 0; 
   if (amtRead > 0) {
      for (int i = 0; i < amtRead; i++)
         if (command[i] == '\0' || command[i] == '\n') {
            if (tokenEnd == 0) {
               tokenEnd = i;
            }
            command[i] = ' ';
         }
   }
   if (tokenEnd == 0) {
      tokenEnd = amtRead;
   }
   command[amtRead] = '\0';
   free(process->comm);
   process->comm = strdup(command);
   process->basenameOffset = tokenEnd;

   return true;
}

481
static bool LinuxProcessList_recurseProcTree(LinuxProcessList* this, const char* dirname, Process* parent, double period, struct timeval tv) {
Hisham Muhammad's avatar
Hisham Muhammad committed
482
   ProcessList* pl = (ProcessList*) this;
483
484
   DIR* dir;
   struct dirent* entry;
485
   Settings* settings = pl->settings;
486
487
488
489
490
491
492
493

   time_t curTime = tv.tv_sec;
   #ifdef HAVE_TASKSTATS
   unsigned long long now = tv.tv_sec*1000LL+tv.tv_usec/1000LL;
   #endif

   dir = opendir(dirname);
   if (!dir) return false;
494
   int cpus = pl->cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
495
496
   bool hideKernelThreads = settings->hideKernelThreads;
   bool hideUserlandThreads = settings->hideUserlandThreads;
497
498
499
500
501
   while ((entry = readdir(dir)) != NULL) {
      char* name = entry->d_name;

      // The RedHat kernel hides threads with a dot.
      // I believe this is non-standard.
Hisham Muhammad's avatar
Hisham Muhammad committed
502
      if ((!settings->hideThreads) && name[0] == '.') {
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
         name++;
      }

      // Just skip all non-number directories.
      if (name[0] < '0' || name[0] > '9') {
         continue;
      }

      // filename is a number: process directory
      int pid = atoi(name);
     
      if (parent && pid == parent->pid)
         continue;

      if (pid <= 0) 
         continue;

520
      bool preExisting = false;
521
      Process* proc = ProcessList_getProcess(pl, pid, &preExisting, (Process_New) LinuxProcess_new);
522
      proc->tgid = parent ? parent->pid : pid;
523
524
      
      LinuxProcess* lp = (LinuxProcess*) proc;
525
526
527

      char subdirname[MAX_NAME+1];
      snprintf(subdirname, MAX_NAME, "%s/%s/task", dirname, name);
528
      LinuxProcessList_recurseProcTree(this, subdirname, proc, period, tv);
529
530

      #ifdef HAVE_TASKSTATS
Hisham Muhammad's avatar
Hisham Muhammad committed
531
      if (settings->flags & PROCESS_FLAG_IO)
532
         LinuxProcessList_readIoFile(lp, dirname, name, now);
533
534
      #endif

535
      if (! LinuxProcessList_readStatmFile(lp, dirname, name))
536
537
         goto errorReadingProcess;

538
      proc->show = ! ((hideKernelThreads && Process_isKernelThread(proc)) || (hideUserlandThreads && Process_isUserlandThread(proc)));
539
540

      char command[MAX_NAME+1];
541
542
      unsigned long long int lasttimes = (lp->utime + lp->stime);
      if (! LinuxProcessList_readStatFile(proc, dirname, name, command))
543
         goto errorReadingProcess;
544
      if (settings->flags & PROCESS_FLAG_LINUX_IOPRIO)
545
546
         LinuxProcess_updateIOPriority(lp);
      float percent_cpu = (lp->utime + lp->stime - lasttimes) / period * 100.0;
547
      proc->percent_cpu = CLAMP(percent_cpu, 0.0, cpus * 100.0);
548
      if (isnan(proc->percent_cpu)) proc->percent_cpu = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
549
      proc->percent_mem = (proc->m_resident * PAGE_SIZE_KB) / (double)(pl->totalMem) * 100.0;
550

551
      if(!preExisting) {
552

553
         if (! LinuxProcessList_statProcessDir(proc, dirname, name, curTime))
554
555
            goto errorReadingProcess;

556
         proc->user = UsersTable_getRef(pl->usersTable, proc->st_uid);
557
558

         #ifdef HAVE_OPENVZ
559
560
561
         if (settings->flags & PROCESS_FLAG_LINUX_OPENVZ) {
            LinuxProcessList_readOpenVZData(lp, dirname, name);
         }
562
563
564
         #endif
         
         #ifdef HAVE_VSERVER
565
         if (settings->flags & PROCESS_FLAG_LINUX_VSERVER) {
566
            LinuxProcessList_readVServerData(lp, dirname, name);
567
         }
568
569
         #endif

570
         if (! LinuxProcessList_readCmdlineFile(proc, dirname, name)) {
571
            goto errorReadingProcess;
572
         }
573

574
         ProcessList_add(pl, proc);
575
      } else {
Hisham Muhammad's avatar
Hisham Muhammad committed
576
         if (settings->updateProcessNames) {
577
            if (! LinuxProcessList_readCmdlineFile(proc, dirname, name)) {
578
               goto errorReadingProcess;
579
            }
580
581
582
583
         }
      }

      #ifdef HAVE_CGROUP
584
      if (settings->flags & PROCESS_FLAG_LINUX_CGROUP)
585
         LinuxProcessList_readCGroupFile(lp, dirname, name);
586
587
      #endif
      
588
589
      if (settings->flags & PROCESS_FLAG_LINUX_OOM)
         LinuxProcessList_readOomData(lp, dirname, name);
590

591
592
593
594
595
596
597
598
599
      if (proc->state == 'Z') {
         free(proc->comm);
         proc->basenameOffset = -1;
         proc->comm = strdup(command);
      } else if (Process_isThread(proc)) {
         if (settings->showThreadNames || Process_isKernelThread(proc) || proc->state == 'Z') {
            free(proc->comm);
            proc->basenameOffset = -1;
            proc->comm = strdup(command);
Hisham Muhammad's avatar
Hisham Muhammad committed
600
         } else if (settings->showThreadNames) {
601
            if (! LinuxProcessList_readCmdlineFile(proc, dirname, name))
602
603
               goto errorReadingProcess;
         }
604
         if (Process_isKernelThread(proc)) {
605
            pl->kernelThreads++;
606
         } else {
607
            pl->userlandThreads++;
608
609
610
         }
      }

611
      pl->totalTasks++;
612
      if (proc->state == 'R')
613
         pl->runningTasks++;
614
      proc->updated = true;
615
616
617
618
      continue;

      // Exception handler.
      errorReadingProcess: {
619
         if (preExisting) {
620
            ProcessList_remove(pl, proc);
621
         } else {
622
            Process_delete((Object*)proc);
623
         }
624
625
626
627
628
629
      }
   }
   closedir(dir);
   return true;
}

Hisham Muhammad's avatar
Hisham Muhammad committed
630
static inline void LinuxProcessList_scanMemoryInfo(ProcessList* this) {
Hisham Muhammad's avatar
Hisham Muhammad committed
631
   unsigned long long int swapFree = 0;
632
633
   unsigned long long int shmem = 0;
   unsigned long long int sreclaimable = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
634

635
636
637
638
   FILE* file = fopen(PROCMEMINFOFILE, "r");
   if (file == NULL) {
      CRT_fatalError("Cannot open " PROCMEMINFOFILE);
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
639
640
641
   char buffer[128];
   while (fgets(buffer, 128, file)) {

642
      #define tryRead(label, variable) (String_startsWith(buffer, label) && sscanf(buffer + strlen(label), " %32llu kB", variable))
Hisham Muhammad's avatar
Hisham Muhammad committed
643
644
      switch (buffer[0]) {
      case 'M':
645
646
647
         if (tryRead("MemTotal:", &this->totalMem)) {}
         else if (tryRead("MemFree:", &this->freeMem)) {}
         else if (tryRead("MemShared:", &this->sharedMem)) {}
Hisham Muhammad's avatar
Hisham Muhammad committed
648
649
         break;
      case 'B':
650
         if (tryRead("Buffers:", &this->buffersMem)) {}
Hisham Muhammad's avatar
Hisham Muhammad committed
651
652
         break;
      case 'C':
653
         if (tryRead("Cached:", &this->cachedMem)) {}
Hisham Muhammad's avatar
Hisham Muhammad committed
654
655
         break;
      case 'S':
656
657
658
659
660
661
662
663
664
665
666
667
         switch (buffer[1]) {
         case 'w':
            if (tryRead("SwapTotal:", &this->totalSwap)) {}
            else if (tryRead("SwapFree:", &swapFree)) {}
            break;
         case 'h':
            if (tryRead("Shmem:", &shmem)) {}
            break;
         case 'R':
            if (tryRead("SReclaimable:", &sreclaimable)) {}
            break;
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
668
         break;
669
      }
670
      #undef tryRead
671
672
   }

673
674
   this->usedMem = this->totalMem - this->freeMem;
   this->cachedMem = this->cachedMem + sreclaimable - shmem;
675
676
   this->usedSwap = this->totalSwap - swapFree;
   fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
677
}
678

Hisham Muhammad's avatar
Hisham Muhammad committed
679
static inline double LinuxProcessList_scanCPUTime(LinuxProcessList* this) {
Hisham Muhammad's avatar
Hisham Muhammad committed
680

Hisham Muhammad's avatar
Hisham Muhammad committed
681
   FILE* file = fopen(PROCSTATFILE, "r");
682
683
684
   if (file == NULL) {
      CRT_fatalError("Cannot open " PROCSTATFILE);
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
685
   int cpus = this->super.cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
686
   assert(cpus > 0);
687
   for (int i = 0; i <= cpus; i++) {
688
      char buffer[PROC_LINE_LENGTH + 1];
Hisham Muhammad's avatar
Hisham Muhammad committed
689
      unsigned long long int usertime, nicetime, systemtime, idletime;
690
691
      unsigned long long int ioWait, irq, softIrq, steal, guest, guestnice;
      ioWait = irq = softIrq = steal = guest = guestnice = 0;
peter-warhzner's avatar
peter-warhzner committed
692
      // Depending on your kernel version,
693
694
      // 5, 7, 8 or 9 of these fields will be set.
      // The rest will remain at zero.
695
      char* ok = fgets(buffer, PROC_LINE_LENGTH, file);
696
      if (!ok) buffer[0] = '\0';
697
      if (i == 0)
Hisham Muhammad's avatar
Hisham Muhammad committed
698
         sscanf(buffer,   "cpu  %16llu %16llu %16llu %16llu %16llu %16llu %16llu %16llu %16llu %16llu",         &usertime, &nicetime, &systemtime, &idletime, &ioWait, &irq, &softIrq, &steal, &guest, &guestnice);
699
      else {
Hisham Muhammad's avatar
Hisham Muhammad committed
700
         int cpuid;
701
702
703
704
705
706
707
708
         sscanf(buffer, "cpu%4d %16llu %16llu %16llu %16llu %16llu %16llu %16llu %16llu %16llu %16llu", &cpuid, &usertime, &nicetime, &systemtime, &idletime, &ioWait, &irq, &softIrq, &steal, &guest, &guestnice);
         assert(cpuid == i - 1);
      }
      // Guest time is already accounted in usertime
      usertime = usertime - guest;
      nicetime = nicetime - guestnice;
      // Fields existing on kernels >= 2.6
      // (and RHEL's patched kernel 2.4...)
Hisham Muhammad's avatar
Hisham Muhammad committed
709
710
711
712
      unsigned long long int idlealltime = idletime + ioWait;
      unsigned long long int systemalltime = systemtime + irq + softIrq;
      unsigned long long int virtalltime = guest + guestnice;
      unsigned long long int totaltime = usertime + nicetime + systemalltime + idlealltime + steal + virtalltime;
713
714
715
716
717
718
719
720
721
722
723
      CPUData* cpuData = &(this->cpus[i]);
      assert (systemtime >= cpuData->systemTime);
      assert (idletime >= cpuData->idleTime);
      assert (totaltime >= cpuData->totalTime);
      assert (systemalltime >= cpuData->systemAllTime);
      assert (idlealltime >= cpuData->idleAllTime);
      assert (ioWait >= cpuData->ioWaitTime);
      assert (irq >= cpuData->irqTime);
      assert (softIrq >= cpuData->softIrqTime);
      assert (steal >= cpuData->stealTime);
      assert (virtalltime >= cpuData->guestTime);
724
725
726
727
728
      // Since we do a subtraction (usertime - guest) and cputime64_to_clock_t()
      // used in /proc/stat rounds down numbers, it can lead to a case where the
      // integer overflow.
      cpuData->userPeriod = (usertime > cpuData->userTime) ? usertime - cpuData->userTime : 0;
      cpuData->nicePeriod = (nicetime > cpuData->niceTime) ? nicetime - cpuData->niceTime : 0;
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
      cpuData->systemPeriod = systemtime - cpuData->systemTime;
      cpuData->systemAllPeriod = systemalltime - cpuData->systemAllTime;
      cpuData->idleAllPeriod = idlealltime - cpuData->idleAllTime;
      cpuData->idlePeriod = idletime - cpuData->idleTime;
      cpuData->ioWaitPeriod = ioWait - cpuData->ioWaitTime;
      cpuData->irqPeriod = irq - cpuData->irqTime;
      cpuData->softIrqPeriod = softIrq - cpuData->softIrqTime;
      cpuData->stealPeriod = steal - cpuData->stealTime;
      cpuData->guestPeriod = virtalltime - cpuData->guestTime;
      cpuData->totalPeriod = totaltime - cpuData->totalTime;
      cpuData->userTime = usertime;
      cpuData->niceTime = nicetime;
      cpuData->systemTime = systemtime;
      cpuData->systemAllTime = systemalltime;
      cpuData->idleAllTime = idlealltime;
      cpuData->idleTime = idletime;
      cpuData->ioWaitTime = ioWait;
      cpuData->irqTime = irq;
      cpuData->softIrqTime = softIrq;
      cpuData->stealTime = steal;
      cpuData->guestTime = virtalltime;
      cpuData->totalTime = totaltime;
   }
752
753
   double period = (double)this->cpus[0].totalPeriod / cpus;
   fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
754
755
756
   return period;
}

757
void ProcessList_goThroughEntries(ProcessList* super) {
Hisham Muhammad's avatar
Hisham Muhammad committed
758
   LinuxProcessList* this = (LinuxProcessList*) super;
Hisham Muhammad's avatar
Hisham Muhammad committed
759

Hisham Muhammad's avatar
Hisham Muhammad committed
760
   LinuxProcessList_scanMemoryInfo(super);
Hisham Muhammad's avatar
Hisham Muhammad committed
761
   double period = LinuxProcessList_scanCPUTime(this);
762
763
764

   struct timeval tv;
   gettimeofday(&tv, NULL);
765
   LinuxProcessList_recurseProcTree(this, PROCDIR, NULL, period, tv);
766
}