ProcessList.c 29.8 KB
Newer Older
Hisham Muhammad's avatar
Hisham Muhammad committed
1
2
3
4
5
6
7
8
/*
htop - ProcessList.c
(C) 2004,2005 Hisham H. Muhammad
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/

#include "ProcessList.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
9
10

#include "CRT.h"
11
#include "String.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
12

Hisham Muhammad's avatar
Hisham Muhammad committed
13
14
#include <sys/time.h>
#include <sys/utsname.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
15
16
17
18
19
20
21
#include <sys/stat.h>
#include <unistd.h>
#include <dirent.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <stdbool.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
22
#include <stdarg.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
23
#include <math.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
24
25
#include <string.h>
#include <time.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
26
27
28
#include <assert.h>

/*{
Hisham Muhammad's avatar
Hisham Muhammad committed
29
30
31
32
33
34
#include "Vector.h"
#include "Hashtable.h"
#include "UsersTable.h"
#include "Panel.h"
#include "Process.h"
#include <sys/types.h>
35

Hisham Muhammad's avatar
Hisham Muhammad committed
36
37
38
39
40
#ifndef PROCDIR
#define PROCDIR "/proc"
#endif

#ifndef PROCSTATFILE
41
#define PROCSTATFILE PROCDIR "/stat"
Hisham Muhammad's avatar
Hisham Muhammad committed
42
43
44
#endif

#ifndef PROCMEMINFOFILE
45
#define PROCMEMINFOFILE PROCDIR "/meminfo"
Hisham Muhammad's avatar
Hisham Muhammad committed
46
47
48
#endif

#ifndef MAX_NAME
Hisham Muhammad's avatar
Hisham Muhammad committed
49
50
51
52
#define MAX_NAME 128
#endif

#ifndef MAX_READ
53
#define MAX_READ 2048
Hisham Muhammad's avatar
Hisham Muhammad committed
54
55
#endif

56
#ifndef ProcessList_cpuId
57
#define ProcessList_cpuId(pl, cpu) ((pl)->countCPUsFromZero ? (cpu) : (cpu)+1)
58
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
59

60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
typedef enum TreeStr_ {
   TREE_STR_HORZ,
   TREE_STR_VERT,
   TREE_STR_RTEE,
   TREE_STR_BEND,
   TREE_STR_TEND,
   TREE_STR_OPEN,
   TREE_STR_SHUT,
   TREE_STR_COUNT
} TreeStr;

typedef enum TreeType_ {
   TREE_TYPE_AUTO,
   TREE_TYPE_ASCII,
   TREE_TYPE_UTF8,
} TreeType;
Hisham Muhammad's avatar
Hisham Muhammad committed
76

77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
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;

Hisham Muhammad's avatar
Hisham Muhammad committed
105
typedef struct ProcessList_ {
106
107
   Vector* processes;
   Vector* processes2;
Hisham Muhammad's avatar
Hisham Muhammad committed
108
109
110
   Hashtable* processTable;
   UsersTable* usersTable;

111
   Panel* panel;
112
   int following;
113
114
115
116
   bool userOnly;
   uid_t userId;
   bool filtering;
   const char* incFilter;
117
   Hashtable* pidWhiteList;
118

119
   int cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
120
   int totalTasks;
121
122
   int userlandThreads;
   int kernelThreads;
Hisham Muhammad's avatar
Hisham Muhammad committed
123
124
   int runningTasks;

125
   #ifdef HAVE_LIBHWLOC
126
127
128
   hwloc_topology_t topology;
   bool topologyOk;
   #endif
129
   CPUData* cpus;
130
131
132
133
134
135
136
137
138
139

   unsigned long long int totalMem;
   unsigned long long int usedMem;
   unsigned long long int freeMem;
   unsigned long long int sharedMem;
   unsigned long long int buffersMem;
   unsigned long long int cachedMem;
   unsigned long long int totalSwap;
   unsigned long long int usedSwap;
   unsigned long long int freeSwap;
Hisham Muhammad's avatar
Hisham Muhammad committed
140
141
142
143
144
145

   ProcessField* fields;
   ProcessField sortKey;
   int direction;
   bool hideThreads;
   bool shadowOtherUsers;
Hisham Muhammad's avatar
Hisham Muhammad committed
146
147
   bool showThreadNames;
   bool showingThreadNames;
Hisham Muhammad's avatar
Hisham Muhammad committed
148
149
150
151
152
   bool hideKernelThreads;
   bool hideUserlandThreads;
   bool treeView;
   bool highlightBaseName;
   bool highlightMegabytes;
153
   bool highlightThreads;
154
   bool detailedCPUTime;
155
   bool countCPUsFromZero;
156
   bool updateProcessNames;
157
   const char **treeStr;
Hisham Muhammad's avatar
Hisham Muhammad committed
158
159

} ProcessList;
160

Hisham Muhammad's avatar
Hisham Muhammad committed
161
162
}*/

163
static ProcessField defaultHeaders[] = { PID, USER, PRIORITY, NICE, M_SIZE, M_RESIDENT, M_SHARE, STATE, PERCENT_CPU, PERCENT_MEM, TIME, COMM, 0 };
Hisham Muhammad's avatar
Hisham Muhammad committed
164

165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
const char *ProcessList_treeStrAscii[TREE_STR_COUNT] = {
   "-", // TREE_STR_HORZ
   "|", // TREE_STR_VERT
   "`", // TREE_STR_RTEE
   "`", // TREE_STR_BEND
   ",", // TREE_STR_TEND
   "+", // TREE_STR_OPEN
   "-", // TREE_STR_SHUT
};

const char *ProcessList_treeStrUtf8[TREE_STR_COUNT] = {
   "\xe2\x94\x80", // TREE_STR_HORZ ─
   "\xe2\x94\x82", // TREE_STR_VERT │
   "\xe2\x94\x9c", // TREE_STR_RTEE ├
   "\xe2\x94\x94", // TREE_STR_BEND └
   "\xe2\x94\x8c", // TREE_STR_TEND ┌
   "+",            // TREE_STR_OPEN +
   "\xe2\x94\x80", // TREE_STR_SHUT ─
};

185
ProcessList* ProcessList_new(UsersTable* usersTable, Hashtable* pidWhiteList) {
Hisham Muhammad's avatar
Hisham Muhammad committed
186
   ProcessList* this;
187
   this = calloc(sizeof(ProcessList), 1);
188
   this->processes = Vector_new(PROCESS_CLASS, true, DEFAULT_SIZE, Process_compare);
189
   this->processTable = Hashtable_new(140, false);
Hisham Muhammad's avatar
Hisham Muhammad committed
190
   this->usersTable = usersTable;
191
   this->pidWhiteList = pidWhiteList;
Hisham Muhammad's avatar
Hisham Muhammad committed
192
193
   
   /* tree-view auxiliary buffers */
194
   this->processes2 = Vector_new(PROCESS_CLASS, true, DEFAULT_SIZE, Process_compare);
Hisham Muhammad's avatar
Hisham Muhammad committed
195
   
196
   FILE* file = fopen(PROCSTATFILE, "r");
197
198
199
   if (file == NULL) {
      CRT_fatalError("Cannot open " PROCSTATFILE);
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
200
   char buffer[256];
201
   int cpus = -1;
Hisham Muhammad's avatar
Hisham Muhammad committed
202
   do {
203
      cpus++;
204
      fgets(buffer, 255, file);
Hisham Muhammad's avatar
Hisham Muhammad committed
205
   } while (String_startsWith(buffer, "cpu"));
206
   fclose(file);
207
   this->cpuCount = cpus - 1;
208

209
#ifdef HAVE_LIBHWLOC
210
211
212
213
214
215
216
   this->topologyOk = false;
   int topoErr = hwloc_topology_init(&this->topology);
   if (topoErr == 0) {
      topoErr = hwloc_topology_load(this->topology);
      this->topologyOk = true;
   }
#endif
217
   this->cpus = calloc(sizeof(CPUData), cpus);
218

219
220
221
   for (int i = 0; i < cpus; i++) {
      this->cpus[i].totalTime = 1;
      this->cpus[i].totalPeriod = 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
222
223
224
   }

   this->fields = calloc(sizeof(ProcessField), LAST_PROCESSFIELD+1);
225
   // TODO: turn 'fields' into a Vector,
Hisham Muhammad's avatar
Hisham Muhammad committed
226
227
228
229
230
231
232
233
   // (and ProcessFields into proper objects).
   for (int i = 0; defaultHeaders[i]; i++) {
      this->fields[i] = defaultHeaders[i];
   }
   this->sortKey = PERCENT_CPU;
   this->direction = 1;
   this->hideThreads = false;
   this->shadowOtherUsers = false;
234
235
   this->showThreadNames = false;
   this->showingThreadNames = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
236
237
238
239
240
   this->hideKernelThreads = false;
   this->hideUserlandThreads = false;
   this->treeView = false;
   this->highlightBaseName = false;
   this->highlightMegabytes = false;
241
   this->detailedCPUTime = false;
242
   this->countCPUsFromZero = false;
243
   this->updateProcessNames = false;
244
   this->treeStr = NULL;
245
   this->following = -1;
Hisham Muhammad's avatar
Hisham Muhammad committed
246
247
248
249
250
251

   return this;
}

void ProcessList_delete(ProcessList* this) {
   Hashtable_delete(this->processTable);
252
253
   Vector_delete(this->processes);
   Vector_delete(this->processes2);
254
   free(this->cpus);
Hisham Muhammad's avatar
Hisham Muhammad committed
255
256
257
258
   free(this->fields);
   free(this);
}

259
260
261
262
void ProcessList_setPanel(ProcessList* this, Panel* panel) {
   this->panel = panel;
}

Hisham Muhammad's avatar
Hisham Muhammad committed
263
264
265
266
267
268
269
void ProcessList_invertSortOrder(ProcessList* this) {
   if (this->direction == 1)
      this->direction = -1;
   else
      this->direction = 1;
}

270
271
void ProcessList_printHeader(ProcessList* this, RichString* header) {
   RichString_prune(header);
Hisham Muhammad's avatar
Hisham Muhammad committed
272
273
   ProcessField* fields = this->fields;
   for (int i = 0; fields[i]; i++) {
Hisham Muhammad's avatar
Hisham Muhammad committed
274
      const char* field = Process_fieldTitles[fields[i]];
275
      if (!this->treeView && this->sortKey == fields[i])
276
         RichString_append(header, CRT_colors[PANEL_HIGHLIGHT_FOCUS], field);
Hisham Muhammad's avatar
Hisham Muhammad committed
277
      else
278
         RichString_append(header, CRT_colors[PANEL_HEADER_FOCUS], field);
Hisham Muhammad's avatar
Hisham Muhammad committed
279
280
281
   }
}

282
static void ProcessList_add(ProcessList* this, Process* p) {
283
284
   assert(Vector_indexOf(this->processes, p, Process_pidCompare) == -1);
   assert(Hashtable_get(this->processTable, p->pid) == NULL);
285
   
286
   Vector_add(this->processes, p);
Hisham Muhammad's avatar
Hisham Muhammad committed
287
   Hashtable_put(this->processTable, p->pid, p);
288
   
289
290
   assert(Vector_indexOf(this->processes, p, Process_pidCompare) != -1);
   assert(Hashtable_get(this->processTable, p->pid) != NULL);
291
   assert(Hashtable_count(this->processTable) == Vector_count(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
292
293
}

294
static void ProcessList_remove(ProcessList* this, Process* p) {
295
296
297
   assert(Vector_indexOf(this->processes, p, Process_pidCompare) != -1);
   assert(Hashtable_get(this->processTable, p->pid) != NULL);
   Process* pp = Hashtable_remove(this->processTable, p->pid);
298
   assert(pp == p); (void)pp;
299
   unsigned int pid = p->pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
300
301
302
   int idx = Vector_indexOf(this->processes, p, Process_pidCompare);
   assert(idx != -1);
   if (idx >= 0) Vector_remove(this->processes, idx);
303
   assert(Hashtable_get(this->processTable, pid) == NULL); (void)pid;
304
   assert(Hashtable_count(this->processTable) == Vector_count(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
305
306
}

Hisham Muhammad's avatar
Hisham Muhammad committed
307
308
Process* ProcessList_get(ProcessList* this, int idx) {
   return (Process*) (Vector_get(this->processes, idx));
Hisham Muhammad's avatar
Hisham Muhammad committed
309
310
311
}

int ProcessList_size(ProcessList* this) {
312
   return (Vector_size(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
313
314
}

Hisham Muhammad's avatar
Hisham Muhammad committed
315
static void ProcessList_buildTree(ProcessList* this, pid_t pid, int level, int indent, int direction, bool show) {
316
   Vector* children = Vector_new(PROCESS_CLASS, false, DEFAULT_SIZE, Process_compare);
Hisham Muhammad's avatar
Hisham Muhammad committed
317

318
   for (int i = Vector_size(this->processes) - 1; i >= 0; i--) {
319
      Process* process = (Process*) (Vector_get(this->processes, i));
320
      if (process->tgid == pid || (process->tgid == process->pid && process->ppid == pid)) {
Hisham Muhammad's avatar
Hisham Muhammad committed
321
         process = (Process*) (Vector_take(this->processes, i));
322
         Vector_add(children, process);
Hisham Muhammad's avatar
Hisham Muhammad committed
323
324
      }
   }
325
   int size = Vector_size(children);
Hisham Muhammad's avatar
Hisham Muhammad committed
326
   for (int i = 0; i < size; i++) {
327
      Process* process = (Process*) (Vector_get(children, i));
328
329
330
331
332
333
334
335
336
      if (!show)
         process->show = false;
      int s = this->processes2->items;
      if (direction == 1)
         Vector_add(this->processes2, process);
      else
         Vector_insert(this->processes2, 0, process);
      assert(this->processes2->items == s+1); (void)s;
      int nextIndent = indent | (1 << level);
337
      ProcessList_buildTree(this, process->pid, level+1, (i < size - 1) ? nextIndent : indent, direction, show ? process->showChildren : false);
338
339
340
341
      if (i == size - 1)
         process->indent = -nextIndent;
      else
         process->indent = nextIndent;
Hisham Muhammad's avatar
Hisham Muhammad committed
342
   }
343
   Vector_delete(children);
Hisham Muhammad's avatar
Hisham Muhammad committed
344
345
346
347
}

void ProcessList_sort(ProcessList* this) {
   if (!this->treeView) {
348
      Vector_insertionSort(this->processes);
Hisham Muhammad's avatar
Hisham Muhammad committed
349
   } else {
350
      // Save settings
Hisham Muhammad's avatar
Hisham Muhammad committed
351
352
      int direction = this->direction;
      int sortKey = this->sortKey;
353
      // Sort by PID
Hisham Muhammad's avatar
Hisham Muhammad committed
354
355
      this->sortKey = PID;
      this->direction = 1;
356
      Vector_quickSort(this->processes);
357
      // Restore settings
Hisham Muhammad's avatar
Hisham Muhammad committed
358
359
      this->sortKey = sortKey;
      this->direction = direction;
360
      // Take PID 1 as root and add to the new listing
361
      int vsize = Vector_size(this->processes);
362
      Process* init = (Process*) (Vector_take(this->processes, 0));
363
364
365
      // This assertion crashes on hardened kernels.
      // I wonder how well tree view works on those systems.
      // assert(init->pid == 1);
Hisham Muhammad's avatar
Hisham Muhammad committed
366
      init->indent = 0;
367
      Vector_add(this->processes2, init);
368
      // Recursively empty list
Hisham Muhammad's avatar
Hisham Muhammad committed
369
      ProcessList_buildTree(this, init->pid, 0, 0, direction, true);
370
      // Add leftovers
371
372
373
374
      while (Vector_size(this->processes)) {
         Process* p = (Process*) (Vector_take(this->processes, 0));
         p->indent = 0;
         Vector_add(this->processes2, p);
Hisham Muhammad's avatar
Hisham Muhammad committed
375
         ProcessList_buildTree(this, p->pid, 0, 0, direction, p->showChildren);
376
377
378
      }
      assert(Vector_size(this->processes2) == vsize); (void)vsize;
      assert(Vector_size(this->processes) == 0);
379
      // Swap listings around
380
      Vector* t = this->processes;
Hisham Muhammad's avatar
Hisham Muhammad committed
381
382
383
384
385
      this->processes = this->processes2;
      this->processes2 = t;
   }
}

386
387
388
389
390
391
392
static bool ProcessList_readStatFile(Process *process, const char* dirname, const char* name, char* command) {
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/stat", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return false;

Hisham Muhammad's avatar
Hisham Muhammad committed
393
394
   static char buf[MAX_READ];

395
396
   int size = fread(buf, 1, MAX_READ, file);
   if (!size) { fclose(file); return false; }
Hisham Muhammad's avatar
Hisham Muhammad committed
397

398
   assert(process->pid == atoi(buf));
Hisham Muhammad's avatar
Hisham Muhammad committed
399
   char *location = strchr(buf, ' ');
400
   if (!location) { fclose(file); return false; }
Hisham Muhammad's avatar
Hisham Muhammad committed
401
402
403

   location += 2;
   char *end = strrchr(location, ')');
404
   if (!end) { fclose(file); return false; }
Hisham Muhammad's avatar
Hisham Muhammad committed
405
406
407
408
409
   
   int commsize = end - location;
   memcpy(command, location, commsize);
   command[commsize] = '\0';
   location = end + 2;
410
411
412
413
414

   int num = sscanf(location, 
      "%c %d %u %u %u "
      "%d %lu "
      "%*u %*u %*u %*u "
415
      "%llu %llu %llu %llu "
416
417
      "%ld %ld %ld "
      "%*d %*u %*u %*d %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u "
418
      "%d %d",
419
420
421
422
423
424
425
      &process->state, &process->ppid, &process->pgrp, &process->session, &process->tty_nr, 
      &process->tpgid, &process->flags,
      &process->utime, &process->stime, &process->cutime, &process->cstime, 
      &process->priority, &process->nice, &process->nlwp,
      &process->exit_signal, &process->processor);
   fclose(file);
   return (num == 16);
Hisham Muhammad's avatar
Hisham Muhammad committed
426
427
}

428
429
430
static bool ProcessList_statProcessDir(Process* process, const char* dirname, char* name) {
   char filename[MAX_NAME+1];
   filename[MAX_NAME] = '\0';
431

432
   snprintf(filename, MAX_NAME, "%s/%s", dirname, name);
433
   struct stat sstat;
434
   int statok = stat(filename, &sstat);
435
436
   if (statok == -1)
      return false;
437
   process->st_uid = sstat.st_uid;
Hisham Muhammad's avatar
Hisham Muhammad committed
438
439
440
  
   struct tm date;
   time_t ctime = sstat.st_ctime;
441
   process->starttime_ctime = ctime;
Hisham Muhammad's avatar
Hisham Muhammad committed
442
   (void) localtime_r((time_t*) &ctime, &date);
443
   strftime(process->starttime_show, 7, ((ctime > time(NULL) - 86400) ? "%R " : "%b%d "), &date);
Hisham Muhammad's avatar
Hisham Muhammad committed
444
   
445
   return true;
Hisham Muhammad's avatar
Hisham Muhammad committed
446
447
}

448
#ifdef HAVE_TASKSTATS
449

450
451
452
static void ProcessList_readIoFile(Process* process, const char* dirname, char* name) {
   char filename[MAX_NAME+1];
   filename[MAX_NAME] = '\0';
453

454
455
456
457
458
459
460
461
462
463
464
465
466
   snprintf(filename, MAX_NAME, "%s/%s/io", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return;
   
   char buffer[256];
   buffer[255] = '\0';
   struct timeval tv;
   gettimeofday(&tv,NULL);
   unsigned long long now = tv.tv_sec*1000+tv.tv_usec/1000;
   unsigned long long last_read = process->io_read_bytes;
   unsigned long long last_write = process->io_write_bytes;
   while (fgets(buffer, 255, file)) {
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
      switch (buffer[0]) {
      case 'r':
         if (buffer[1] == 'c')
            sscanf(buffer, "rchar: %llu", &process->io_rchar);
         else if (sscanf(buffer, "read_bytes: %llu", &process->io_read_bytes)) {
            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 (buffer[1] == 'c')
            sscanf(buffer, "wchar: %llu", &process->io_wchar);
         else if (sscanf(buffer, "write_bytes: %llu", &process->io_write_bytes)) {
            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 (buffer[5] == 'r')
            sscanf(buffer, "syscr: %llu", &process->io_syscr);
         else
            sscanf(buffer, "syscw: %llu", &process->io_syscw);
         break;
      case 'c':
         sscanf(buffer, "cancelled_write_bytes: %llu", &process->io_cancelled_write_bytes);
494
495
496
497
498
499
500
501
502
503
504
505
506
507
      }
   }
   fclose(file);
}

#endif

static bool ProcessList_readStatmFile(Process* process, const char* dirname, const char* name) {
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/statm", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return false;

508
   int num = fscanf(file, "%32d %32d %32d %32d %32d %32d %32d",
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
       &process->m_size, &process->m_resident, &process->m_share, 
       &process->m_trs, &process->m_lrs, &process->m_drs, 
       &process->m_dt);
   fclose(file);
   return (num == 7);
}

#ifdef HAVE_OPENVZ

static void ProcessList_readOpenVZData(Process* process, const char* dirname, const char* name) {
   if (access("/proc/vz", R_OK) != 0) {
      process->vpid = process->pid;
      process->ctid = 0;
      return;
   }
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/stat", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file) 
      return;
   fscanf(file, 
530
531
532
533
534
535
536
      "%*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",
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
      &process->vpid, &process->ctid);
   fclose(file);
}

#endif

#ifdef HAVE_CGROUP

static void ProcessList_readCGroupFile(Process* process, const char* dirname, const char* name) {
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/cgroup", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file) {
      process->cgroup = strdup("");
      return;
   }
   char buffer[256];
   char *ok = fgets(buffer, 255, file);
   if (ok) {
      char* trimmed = String_trim(buffer);
557
558
      int nFields;
      char** fields = String_split(trimmed, ':', &nFields);
559
      free(trimmed);
560
561
562
563
564
      if (nFields >= 3) {
         process->cgroup = strndup(fields[2] + 1, 10);
      } else {
         process->cgroup = strdup("");
      }
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
      String_freeArray(fields);
   }
   fclose(file);
}

#endif

#ifdef HAVE_VSERVER

static void ProcessList_readVServerData(Process* process, const char* dirname, const char* name) {
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/status", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return;
   char buffer[256];
   process->vxid = 0;
   while (fgets(buffer, 255, file)) {
      if (String_startsWith(buffer, "VxID:")) {
         int vxid;
         int ok = sscanf(buffer, "VxID:\t%d", &vxid);
         if (ok >= 1) {
            process->vxid = vxid;
588
         }
589
590
591
592
593
594
595
      }
      #if defined HAVE_ANCIENT_VSERVER
      else if (String_startsWith(buffer, "s_context:")) {
         int vxid;
         int ok = sscanf(buffer, "s_context:\t%d", &vxid);
         if (ok >= 1) {
            process->vxid = vxid;
596
597
         }
      }
598
      #endif
599
   }
600
   fclose(file);
601
}
602

603
604
#endif

605
606
607
static bool ProcessList_readCmdlineFile(Process* process, const char* dirname, const char* name) {
   if (Process_isKernelThread(process))
      return true;
608

609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
   char filename[MAX_NAME+1];
   snprintf(filename, MAX_NAME, "%s/%s/cmdline", dirname, name);
   FILE* file = fopen(filename, "r");
   if (!file)
      return false;
         
   char command[4096+1]; // max cmdline length on Linux
   int amtRead = fread(command, 1, sizeof(command) - 1, file);
   if (amtRead > 0) {
      for (int i = 0; i < amtRead; i++)
         if (command[i] == '\0' || command[i] == '\n') {
            command[i] = ' ';
         }
   }
   command[amtRead] = '\0';
   fclose(file);
   free(process->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
626
   process->comm = strdup(command);
627

628
629
630
631
   return true;
}


Hisham Muhammad's avatar
Hisham Muhammad committed
632
static bool ProcessList_processEntries(ProcessList* this, const char* dirname, Process* parent, double period) {
Hisham Muhammad's avatar
Hisham Muhammad committed
633
634
635
636
   DIR* dir;
   struct dirent* entry;

   dir = opendir(dirname);
637
   if (!dir) return false;
638
   int cpus = this->cpuCount;
639
640
   bool hideKernelThreads = this->hideKernelThreads;
   bool hideUserlandThreads = this->hideUserlandThreads;
Hisham Muhammad's avatar
Hisham Muhammad committed
641
642
643
   while ((entry = readdir(dir)) != NULL) {
      char* name = entry->d_name;
      // filename is a number: process directory
644
645
646
      int pid = atoi(name);
     
      if (parent && pid == parent->pid)
647
         continue;
Hisham Muhammad's avatar
Hisham Muhammad committed
648
649
650
651

      // The RedHat kernel hides threads with a dot.
      // I believe this is non-standard.
      if ((!this->hideThreads) && pid == 0 && name[0] == '.') {
Hisham Muhammad's avatar
Hisham Muhammad committed
652
         pid = atoi(name + 1);
Hisham Muhammad's avatar
Hisham Muhammad committed
653
      }
654
655
      if (pid <= 0) 
         continue;
Hisham Muhammad's avatar
Hisham Muhammad committed
656

657
658
659
660
661
662
663
664
665
666
667
      Process* process = NULL;
      Process* existingProcess = (Process*) Hashtable_get(this->processTable, pid);

      if (existingProcess) {
         assert(Vector_indexOf(this->processes, existingProcess, Process_pidCompare) != -1);
         process = existingProcess;
         assert(process->pid == pid);
      } else {
         process = Process_new(this);
         assert(process->comm == NULL);
         process->pid = pid;
668
         process->tgid = parent ? parent->pid : pid;
669
      }
670

671
672
673
      char subdirname[MAX_NAME+1];
      snprintf(subdirname, MAX_NAME, "%s/%s/task", dirname, name);
      ProcessList_processEntries(this, subdirname, process, period);
674

675
676
677
      #ifdef HAVE_TASKSTATS
      ProcessList_readIoFile(process, dirname, name);
      #endif
678

679
680
      if (! ProcessList_readStatmFile(process, dirname, name))
         goto errorReadingProcess;
Hisham Muhammad's avatar
Hisham Muhammad committed
681

682
      process->show = ! ((hideKernelThreads && Process_isKernelThread(process)) || (hideUserlandThreads && Process_isUserlandThread(process)));
683

684
      char command[MAX_NAME+1];
Hisham Muhammad's avatar
Hisham Muhammad committed
685
      unsigned long long int lasttimes = (process->utime + process->stime);
686
687
      if (! ProcessList_readStatFile(process, dirname, name, command))
         goto errorReadingProcess;
688
      Process_updateIOPriority(process);
689
      float percent_cpu = (process->utime + process->stime - lasttimes) / period * 100.0;
690
691
      process->percent_cpu = MAX(MIN(percent_cpu, cpus*100.0), 0.0);
      if (isnan(process->percent_cpu)) process->percent_cpu = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
692
      process->percent_mem = (process->m_resident * PAGE_SIZE_KB) / (double)(this->totalMem) * 100.0;
693

694
      if(!existingProcess) {
695

696
         if (! ProcessList_statProcessDir(process, dirname, name))
697
698
            goto errorReadingProcess;

699
700
         process->user = UsersTable_getRef(this->usersTable, process->st_uid);

701
702
703
         #ifdef HAVE_OPENVZ
         ProcessList_readOpenVZData(process, dirname, name);
         #endif
704

705
706
707
         #ifdef HAVE_CGROUP
         ProcessList_readCGroupFile(process, dirname, name);
         #endif
708
         
709
710
711
712
713
         #ifdef HAVE_VSERVER
         ProcessList_readVServerData(process, dirname, name);
         #endif
         
         if (! ProcessList_readCmdlineFile(process, dirname, name))
714
715
            goto errorReadingProcess;

716
         ProcessList_add(this, process);
717
718
719
720
721
      } else {
         if (this->updateProcessNames) {
            if (! ProcessList_readCmdlineFile(process, dirname, name))
               goto errorReadingProcess;
         }
722
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
723

724
725
      if (process->state == 'Z') {
         free(process->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
726
         process->comm = strdup(command);
727
      } else if (Process_isThread(process)) {
728
         if (this->showThreadNames || Process_isKernelThread(process) || process->state == 'Z') {
729
            free(process->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
730
            process->comm = strdup(command);
731
732
733
         } else if (this->showingThreadNames) {
            if (! ProcessList_readCmdlineFile(process, dirname, name))
               goto errorReadingProcess;
Hisham Muhammad's avatar
Hisham Muhammad committed
734
         }
735
736
737
738
739
         if (Process_isKernelThread(process)) {
            this->kernelThreads++;
         } else {
            this->userlandThreads++;
         }
740
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
741

742
743
744
745
      this->totalTasks++;
      if (process->state == 'R')
         this->runningTasks++;
      process->updated = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
746

747
      continue;
Hisham Muhammad's avatar
Hisham Muhammad committed
748

749
750
751
752
753
      // Exception handler.
      errorReadingProcess: {
         if (process->comm) {
            free(process->comm);
            process->comm = NULL;
Hisham Muhammad's avatar
Hisham Muhammad committed
754
         }
755
756
757
758
         if (existingProcess)
            ProcessList_remove(this, process);
         else
            Process_delete((Object*)process);
Hisham Muhammad's avatar
Hisham Muhammad committed
759
760
761
      }
   }
   closedir(dir);
762
   return true;
Hisham Muhammad's avatar
Hisham Muhammad committed
763
764
765
}

void ProcessList_scan(ProcessList* this) {
766
   unsigned long long int usertime, nicetime, systemtime, systemalltime, idlealltime, idletime, totaltime, virtalltime;
Hisham Muhammad's avatar
Hisham Muhammad committed
767
   unsigned long long int swapFree = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
768

769
   FILE* file = fopen(PROCMEMINFOFILE, "r");
770
771
772
   if (file == NULL) {
      CRT_fatalError("Cannot open " PROCMEMINFOFILE);
   }
773
   int cpus = this->cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
774
775
   {
      char buffer[128];
776
      while (fgets(buffer, 128, file)) {
Hisham Muhammad's avatar
Hisham Muhammad committed
777
778
779
780
   
         switch (buffer[0]) {
         case 'M':
            if (String_startsWith(buffer, "MemTotal:"))
781
               sscanf(buffer, "MemTotal: %llu kB", &this->totalMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
782
            else if (String_startsWith(buffer, "MemFree:"))
783
               sscanf(buffer, "MemFree: %llu kB", &this->freeMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
784
            else if (String_startsWith(buffer, "MemShared:"))
785
               sscanf(buffer, "MemShared: %llu kB", &this->sharedMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
786
787
788
            break;
         case 'B':
            if (String_startsWith(buffer, "Buffers:"))
789
               sscanf(buffer, "Buffers: %llu kB", &this->buffersMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
790
791
792
            break;
         case 'C':
            if (String_startsWith(buffer, "Cached:"))
793
               sscanf(buffer, "Cached: %llu kB", &this->cachedMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
794
795
796
            break;
         case 'S':
            if (String_startsWith(buffer, "SwapTotal:"))
797
               sscanf(buffer, "SwapTotal: %llu kB", &this->totalSwap);
Hisham Muhammad's avatar
Hisham Muhammad committed
798
            if (String_startsWith(buffer, "SwapFree:"))
799
               sscanf(buffer, "SwapFree: %llu kB", &swapFree);
Hisham Muhammad's avatar
Hisham Muhammad committed
800
801
            break;
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
802
803
      }
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
804

Hisham Muhammad's avatar
Hisham Muhammad committed
805
806
   this->usedMem = this->totalMem - this->freeMem;
   this->usedSwap = this->totalSwap - swapFree;
807
   fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
808

809
   file = fopen(PROCSTATFILE, "r");
810
811
812
   if (file == NULL) {
      CRT_fatalError("Cannot open " PROCSTATFILE);
   }
813
   for (int i = 0; i <= cpus; i++) {
Hisham Muhammad's avatar
Hisham Muhammad committed
814
815
      char buffer[256];
      int cpuid;
816
817
      unsigned long long int ioWait, irq, softIrq, steal, guest;
      ioWait = irq = softIrq = steal = guest = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
818
819
820
      // Dependending on your kernel version,
      // 5, 7 or 8 of these fields will be set.
      // The rest will remain at zero.
821
      fgets(buffer, 255, file);
Hisham Muhammad's avatar
Hisham Muhammad committed
822
      if (i == 0)
823
         sscanf(buffer, "cpu  %llu %llu %llu %llu %llu %llu %llu %llu %llu", &usertime, &nicetime, &systemtime, &idletime, &ioWait, &irq, &softIrq, &steal, &guest);
Hisham Muhammad's avatar
Hisham Muhammad committed
824
      else {
825
         sscanf(buffer, "cpu%d %llu %llu %llu %llu %llu %llu %llu %llu %llu", &cpuid, &usertime, &nicetime, &systemtime, &idletime, &ioWait, &irq, &softIrq, &steal, &guest);
Hisham Muhammad's avatar
Hisham Muhammad committed
826
827
828
829
         assert(cpuid == i - 1);
      }
      // Fields existing on kernels >= 2.6
      // (and RHEL's patched kernel 2.4...)
830
      idlealltime = idletime + ioWait;
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
      systemalltime = systemtime + irq + softIrq;
      virtalltime = steal + guest;
      totaltime = usertime + nicetime + systemalltime + idlealltime + virtalltime;
      CPUData* cpuData = &(this->cpus[i]);
      assert (usertime >= cpuData->userTime);
      assert (nicetime >= cpuData->niceTime);
      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 (guest >= cpuData->guestTime);
      cpuData->userPeriod = usertime - cpuData->userTime;
      cpuData->nicePeriod = nicetime - cpuData->niceTime;
      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 = guest - 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 = guest;
      cpuData->totalTime = totaltime;
Hisham Muhammad's avatar
Hisham Muhammad committed
871
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
872
   double period = (double)this->cpus[0].totalPeriod / cpus; fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
873
874

   // mark all process as "dirty"
875
876
   for (int i = 0; i < Vector_size(this->processes); i++) {
      Process* p = (Process*) Vector_get(this->processes, i);
Hisham Muhammad's avatar
Hisham Muhammad committed
877
878
879
880
      p->updated = false;
   }
   
   this->totalTasks = 0;
881
882
   this->userlandThreads = 0;
   this->kernelThreads = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
883
   this->runningTasks = 0;
884

885
   ProcessList_processEntries(this, PROCDIR, NULL, period);
Hisham Muhammad's avatar
Hisham Muhammad committed
886
887
   
   this->showingThreadNames = this->showThreadNames;
Hisham Muhammad's avatar
Hisham Muhammad committed
888
   
889
890
   for (int i = Vector_size(this->processes) - 1; i >= 0; i--) {
      Process* p = (Process*) Vector_get(this->processes, i);
Hisham Muhammad's avatar
Hisham Muhammad committed
891
892
893
894
895
896
897
      if (p->updated == false)
         ProcessList_remove(this, p);
      else
         p->updated = false;
   }

}
898
899
900
901
902
903
904
905
906
907
908
909
910
911

ProcessField ProcessList_keyAt(ProcessList* this, int at) {
   int x = 0;
   ProcessField* fields = this->fields;
   ProcessField field;
   for (int i = 0; (field = fields[i]); i++) {
      int len = strlen(Process_fieldTitles[field]);
      if (at >= x && at <= x + len) {
         return field;
      }
      x += len;
   }
   return COMM;
}
912
913
914
915
916
917
918
919

void ProcessList_expandTree(ProcessList* this) {
   int size = Vector_size(this->processes);
   for (int i = 0; i < size; i++) {
      Process* process = (Process*) Vector_get(this->processes, i);
      process->showChildren = true;
   }
}
920

921
void ProcessList_rebuildPanel(ProcessList* this, bool flags, int following, bool userOnly, uid_t userId, bool filtering, const char* incFilter) {
922
   if (!flags) {
923
      following = this->following;
924
925
926
927
928
      userOnly = this->userOnly;
      userId = this->userId;
      filtering = this->filtering;
      incFilter = this->incFilter;
   } else {
929
      this->following = following;
930
931
932
933
934
935
936
      this->userOnly = userOnly;
      this->userId = userId;
      this->filtering = filtering;
      this->incFilter = incFilter;
   }

   int currPos = Panel_getSelectedIndex(this->panel);
937
   pid_t currPid = following != -1 ? following : 0;
938
939
940
941
942
943
944
945
946
947
948
   int currScrollV = this->panel->scrollV;

   Panel_prune(this->panel);
   int size = ProcessList_size(this);
   int idx = 0;
   for (int i = 0; i < size; i++) {
      bool hidden = false;
      Process* p = ProcessList_get(this, i);

      if ( (!p->show)
         || (userOnly && (p->st_uid != userId))
949
950
         || (filtering && !(String_contains_i(p->comm, incFilter)))
         || (this->pidWhiteList && !Hashtable_get(this->pidWhiteList, p->pid)) )
951
952
953
954
         hidden = true;

      if (!hidden) {
         Panel_set(this->panel, idx, (Object*)p);
955
         if ((following == -1 && idx == currPos) || (following != -1 && p->pid == currPid)) {
956
957
958
959
960
961
962
            Panel_setSelected(this->panel, idx);
            this->panel->scrollV = currScrollV;
         }
         idx++;
      }
   }
}