ProcessList.c 25.9 KB
Newer Older
Hisham Muhammad's avatar
Hisham Muhammad committed
1
2
3
4
5
6
7
/*
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.
*/

8
9
10
11
12
#ifndef CONFIG_H
#define CONFIG_H
#include "config.h"
#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
13
14
#include "ProcessList.h"
#include "Process.h"
15
#include "Vector.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
16
17
#include "UsersTable.h"
#include "Hashtable.h"
18
#include "String.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
19
20
21
22
23
24
25
26
27
28

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <dirent.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <stdbool.h>
#include <sys/utsname.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
29
#include <stdarg.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
30
#include <math.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
31
32
33
34
35

#include "debug.h"
#include <assert.h>

/*{
36

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

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

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

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

#ifndef MAX_READ
54
#define MAX_READ 2048
Hisham Muhammad's avatar
Hisham Muhammad committed
55
56
57
58
59
60
#endif

}*/

/*{

61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
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
89
typedef struct ProcessList_ {
90
91
   Vector* processes;
   Vector* processes2;
Hisham Muhammad's avatar
Hisham Muhammad committed
92
93
94
   Hashtable* processTable;
   UsersTable* usersTable;

95
   int cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
96
   int totalTasks;
97
98
   int userlandThreads;
   int kernelThreads;
Hisham Muhammad's avatar
Hisham Muhammad committed
99
100
   int runningTasks;

101
   CPUData* cpus;
102
103
104
105
106
107
108
109
110
111

   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
112
113
114
115
116
117

   ProcessField* fields;
   ProcessField sortKey;
   int direction;
   bool hideThreads;
   bool shadowOtherUsers;
Hisham Muhammad's avatar
Hisham Muhammad committed
118
119
   bool showThreadNames;
   bool showingThreadNames;
Hisham Muhammad's avatar
Hisham Muhammad committed
120
121
122
123
124
   bool hideKernelThreads;
   bool hideUserlandThreads;
   bool treeView;
   bool highlightBaseName;
   bool highlightMegabytes;
125
   bool highlightThreads;
126
   bool detailedCPUTime;
Hisham Muhammad's avatar
Hisham Muhammad committed
127
128
129
130

} ProcessList;
}*/

131
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
132
133
134
135

ProcessList* ProcessList_new(UsersTable* usersTable) {
   ProcessList* this;
   this = malloc(sizeof(ProcessList));
136
   this->processes = Vector_new(PROCESS_CLASS, true, DEFAULT_SIZE, Process_compare);
137
   this->processTable = Hashtable_new(70, false);
138
   assert(Hashtable_count(this->processTable) == Vector_count(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
139
140
141
   this->usersTable = usersTable;
   
   /* tree-view auxiliary buffers */
142
   this->processes2 = Vector_new(PROCESS_CLASS, true, DEFAULT_SIZE, Process_compare);
Hisham Muhammad's avatar
Hisham Muhammad committed
143
   
144
145
   FILE* file = fopen(PROCSTATFILE, "r");
   assert(file != NULL);
Hisham Muhammad's avatar
Hisham Muhammad committed
146
   char buffer[256];
147
   int cpus = -1;
Hisham Muhammad's avatar
Hisham Muhammad committed
148
   do {
149
      cpus++;
150
      fgets(buffer, 255, file);
Hisham Muhammad's avatar
Hisham Muhammad committed
151
   } while (String_startsWith(buffer, "cpu"));
152
   fclose(file);
153
   this->cpuCount = cpus - 1;
154
   
155
   this->cpus = calloc(sizeof(CPUData), cpus);
156

157
158
159
   for (int i = 0; i < cpus; i++) {
      this->cpus[i].totalTime = 1;
      this->cpus[i].totalPeriod = 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
160
161
162
   }

   this->fields = calloc(sizeof(ProcessField), LAST_PROCESSFIELD+1);
163
   // TODO: turn 'fields' into a Vector,
Hisham Muhammad's avatar
Hisham Muhammad committed
164
165
166
167
168
169
170
171
   // (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;
172
173
   this->showThreadNames = false;
   this->showingThreadNames = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
174
175
176
177
178
   this->hideKernelThreads = false;
   this->hideUserlandThreads = false;
   this->treeView = false;
   this->highlightBaseName = false;
   this->highlightMegabytes = false;
179
   this->detailedCPUTime = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
180
181
182
183
184
185

   return this;
}

void ProcessList_delete(ProcessList* this) {
   Hashtable_delete(this->processTable);
186
187
   Vector_delete(this->processes);
   Vector_delete(this->processes2);
188
   free(this->cpus);
Hisham Muhammad's avatar
Hisham Muhammad committed
189
190
191
192
193
194
195
196
197
198
199
   free(this->fields);
   free(this);
}

void ProcessList_invertSortOrder(ProcessList* this) {
   if (this->direction == 1)
      this->direction = -1;
   else
      this->direction = 1;
}

200
201
void ProcessList_printHeader(ProcessList* this, RichString* header) {
   RichString_prune(header);
Hisham Muhammad's avatar
Hisham Muhammad committed
202
203
   ProcessField* fields = this->fields;
   for (int i = 0; fields[i]; i++) {
Hisham Muhammad's avatar
Hisham Muhammad committed
204
      const char* field = Process_fieldTitles[fields[i]];
Hisham Muhammad's avatar
Hisham Muhammad committed
205
      if (this->sortKey == fields[i])
206
         RichString_append(header, CRT_colors[PANEL_HIGHLIGHT_FOCUS], field);
Hisham Muhammad's avatar
Hisham Muhammad committed
207
      else
208
         RichString_append(header, CRT_colors[PANEL_HEADER_FOCUS], field);
Hisham Muhammad's avatar
Hisham Muhammad committed
209
210
211
   }
}

212
static void ProcessList_add(ProcessList* this, Process* p) {
213
214
   assert(Vector_indexOf(this->processes, p, Process_pidCompare) == -1);
   assert(Hashtable_get(this->processTable, p->pid) == NULL);
215
   
216
   Vector_add(this->processes, p);
Hisham Muhammad's avatar
Hisham Muhammad committed
217
   Hashtable_put(this->processTable, p->pid, p);
218
   
219
220
   assert(Vector_indexOf(this->processes, p, Process_pidCompare) != -1);
   assert(Hashtable_get(this->processTable, p->pid) != NULL);
221
   assert(Hashtable_count(this->processTable) == Vector_count(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
222
223
}

224
static void ProcessList_remove(ProcessList* this, Process* p) {
225
226
227
   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);
228
   assert(pp == p); (void)pp;
229
   unsigned int pid = p->pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
230
231
232
   int idx = Vector_indexOf(this->processes, p, Process_pidCompare);
   assert(idx != -1);
   if (idx >= 0) Vector_remove(this->processes, idx);
233
   assert(Hashtable_get(this->processTable, pid) == NULL); (void)pid;
234
   assert(Hashtable_count(this->processTable) == Vector_count(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
235
236
}

Hisham Muhammad's avatar
Hisham Muhammad committed
237
238
Process* ProcessList_get(ProcessList* this, int idx) {
   return (Process*) (Vector_get(this->processes, idx));
Hisham Muhammad's avatar
Hisham Muhammad committed
239
240
241
}

int ProcessList_size(ProcessList* this) {
242
   return (Vector_size(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
243
244
}

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

248
   for (int i = Vector_size(this->processes) - 1; i >= 0; i--) {
249
      Process* process = (Process*) (Vector_get(this->processes, i));
250
      if (process->tgid == pid || (process->tgid == process->pid && process->ppid == pid)) {
Hisham Muhammad's avatar
Hisham Muhammad committed
251
         process = (Process*) (Vector_take(this->processes, i));
252
         Vector_add(children, process);
Hisham Muhammad's avatar
Hisham Muhammad committed
253
254
      }
   }
255
   int size = Vector_size(children);
Hisham Muhammad's avatar
Hisham Muhammad committed
256
   for (int i = 0; i < size; i++) {
257
      Process* process = (Process*) (Vector_get(children, i));
258
259
260
261
262
263
264
265
266
267
268
269
      if (!show)
         process->show = false;
      show = show ? process->showChildren : 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);
      ProcessList_buildTree(this, process->pid, level+1, (i < size - 1) ? nextIndent : indent, direction, show);
      process->indent = nextIndent;
Hisham Muhammad's avatar
Hisham Muhammad committed
270
   }
271
   Vector_delete(children);
Hisham Muhammad's avatar
Hisham Muhammad committed
272
273
274
275
}

void ProcessList_sort(ProcessList* this) {
   if (!this->treeView) {
276
      Vector_sort(this->processes);
Hisham Muhammad's avatar
Hisham Muhammad committed
277
   } else {
278
      // Save settings
Hisham Muhammad's avatar
Hisham Muhammad committed
279
280
      int direction = this->direction;
      int sortKey = this->sortKey;
281
      // Sort by PID
Hisham Muhammad's avatar
Hisham Muhammad committed
282
283
      this->sortKey = PID;
      this->direction = 1;
284
      Vector_sort(this->processes);
285
      // Restore settings
Hisham Muhammad's avatar
Hisham Muhammad committed
286
287
      this->sortKey = sortKey;
      this->direction = direction;
288
      // Take PID 1 as root and add to the new listing
289
      int vsize = Vector_size(this->processes);
290
      Process* init = (Process*) (Vector_take(this->processes, 0));
291
292
293
      // 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
294
      init->indent = 0;
295
      Vector_add(this->processes2, init);
296
      // Recursively empty list
Hisham Muhammad's avatar
Hisham Muhammad committed
297
      ProcessList_buildTree(this, init->pid, 0, 0, direction, true);
298
      // Add leftovers
299
300
301
302
      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
303
         ProcessList_buildTree(this, p->pid, 0, 0, direction, p->showChildren);
304
305
306
      }
      assert(Vector_size(this->processes2) == vsize); (void)vsize;
      assert(Vector_size(this->processes) == 0);
307
      // Swap listings around
308
      Vector* t = this->processes;
Hisham Muhammad's avatar
Hisham Muhammad committed
309
310
311
312
313
      this->processes = this->processes2;
      this->processes2 = t;
   }
}

314
315
316
317
318
319
320
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
321
322
   static char buf[MAX_READ];

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

326
   assert(process->pid == atoi(buf));
Hisham Muhammad's avatar
Hisham Muhammad committed
327
   char *location = strchr(buf, ' ');
328
   if (!location) { fclose(file); return false; }
Hisham Muhammad's avatar
Hisham Muhammad committed
329
330
331

   location += 2;
   char *end = strrchr(location, ')');
332
   if (!end) { fclose(file); return false; }
Hisham Muhammad's avatar
Hisham Muhammad committed
333
334
335
336
337
   
   int commsize = end - location;
   memcpy(command, location, commsize);
   command[commsize] = '\0';
   location = end + 2;
338
339
340
341
342
343
344
345

   int num = sscanf(location, 
      "%c %d %u %u %u "
      "%d %lu "
      "%*u %*u %*u %*u "
      "%lu %lu %ld %ld "
      "%ld %ld %ld "
      "%*d %*u %*u %*d %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u "
346
      "%d %d",
347
348
349
350
351
352
353
      &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
354
355
}

356
357
358
static bool ProcessList_statProcessDir(Process* process, const char* dirname, char* name) {
   char filename[MAX_NAME+1];
   filename[MAX_NAME] = '\0';
359

360
   snprintf(filename, MAX_NAME, "%s/%s", dirname, name);
361
   struct stat sstat;
362
   int statok = stat(filename, &sstat);
363
364
   if (statok == -1)
      return false;
365
   process->st_uid = sstat.st_uid;
Hisham Muhammad's avatar
Hisham Muhammad committed
366
367
368
  
   struct tm date;
   time_t ctime = sstat.st_ctime;
369
   process->starttime_ctime = ctime;
Hisham Muhammad's avatar
Hisham Muhammad committed
370
   (void) localtime_r((time_t*) &ctime, &date);
371
   strftime(process->starttime_show, 7, ((ctime > time(NULL) - 86400) ? "%R " : "%b%d "), &date);
Hisham Muhammad's avatar
Hisham Muhammad committed
372
   
373
   return true;
Hisham Muhammad's avatar
Hisham Muhammad committed
374
375
}

376
#ifdef HAVE_TASKSTATS
377

378
379
380
static void ProcessList_readIoFile(Process* process, const char* dirname, char* name) {
   char filename[MAX_NAME+1];
   filename[MAX_NAME] = '\0';
381

382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
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
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
   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)) {
      if (sscanf(buffer, "rchar: %llu", &process->io_rchar)) continue;
      if (sscanf(buffer, "wchar: %llu", &process->io_wchar)) continue;
      if (sscanf(buffer, "syscr: %llu", &process->io_syscr)) continue;
      if (sscanf(buffer, "syscw: %llu", &process->io_syscw)) continue;
      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;
         continue;
      }
      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;
         continue;
      }
      sscanf(buffer, "cancelled_write_bytes: %llu", &process->io_cancelled_write_bytes);
   }
   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;

   int num = fscanf(file, "%d %d %d %d %d %d %d",
       &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, 
      "%*u %*s %*c %*u %*u %*u %*u %*u %*u %*u "
      "%*u %*u %*u %*u %*u %*u %*u %*u "
      "%*u %*u %*u %*u %*u %*u %*u %*u "
      "%*u %*u %*u %*u %*u %*u %*u %*u "
      "%*u %*u %*u %*u %*u %*u %*u %*u "
      "%*u %*u %*u %*u %*u %*u %*u "
      "%*u %*u %u %u",
      &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);
      char** fields = String_split(trimmed, ':');
      free(trimmed);

      process->cgroup = strndup(fields[2] + 1, 10);
      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;
501
         }
502
503
504
505
506
507
508
      }
      #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;
509
510
         }
      }
511
      #endif
512
   }
513
   fclose(file);
514
}
515

516
517
#endif

518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
static bool ProcessList_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);
   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);
   process->comm = String_copy(command);
   return true;
}


static bool ProcessList_processEntries(ProcessList* this, const char* dirname, Process* parent, float period) {
Hisham Muhammad's avatar
Hisham Muhammad committed
544
545
546
547
   DIR* dir;
   struct dirent* entry;

   dir = opendir(dirname);
548
   if (!dir) return false;
549
   int cpus = this->cpuCount;
550
551
   bool hideKernelThreads = this->hideKernelThreads;
   bool hideUserlandThreads = this->hideUserlandThreads;
Hisham Muhammad's avatar
Hisham Muhammad committed
552
553
554
   while ((entry = readdir(dir)) != NULL) {
      char* name = entry->d_name;
      // filename is a number: process directory
555
556
557
      int pid = atoi(name);
     
      if (parent && pid == parent->pid)
558
         continue;
Hisham Muhammad's avatar
Hisham Muhammad committed
559

560
      bool isThread = parent;
Hisham Muhammad's avatar
Hisham Muhammad committed
561
562
563
564
565
      // The RedHat kernel hides threads with a dot.
      // I believe this is non-standard.
      if ((!this->hideThreads) && pid == 0 && name[0] == '.') {
         char* tname = name + 1;
         pid = atoi(tname);
566
         isThread = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
567
      }
568
569
      if (pid <= 0) 
         continue;
Hisham Muhammad's avatar
Hisham Muhammad committed
570

571
572
573
574
575
576
577
578
579
580
581
      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;
582
         process->tgid = parent ? parent->pid : pid;
583
      }
584

585
586
587
      char subdirname[MAX_NAME+1];
      snprintf(subdirname, MAX_NAME, "%s/%s/task", dirname, name);
      ProcessList_processEntries(this, subdirname, process, period);
588

589
590
591
      #ifdef HAVE_TASKSTATS
      ProcessList_readIoFile(process, dirname, name);
      #endif
592

593
594
      if (! ProcessList_readStatmFile(process, dirname, name))
         goto errorReadingProcess;
Hisham Muhammad's avatar
Hisham Muhammad committed
595

596
597
      isThread = Process_isThread(process);
      process->show = ! ((hideKernelThreads && Process_isKernelThread(process)) || (hideUserlandThreads && Process_isUserlandThread(process)));
598

599
600
601
602
603
604
605
606
      char command[MAX_NAME+1];
      int lasttimes = (process->utime + process->stime);
      if (! ProcessList_readStatFile(process, dirname, name, command))
         goto errorReadingProcess;
      int percent_cpu = (process->utime + process->stime - lasttimes) / period * 100.0;
      process->percent_cpu = MAX(MIN(percent_cpu, cpus*100.0), 0.0);
      if (isnan(process->percent_cpu)) process->percent_cpu = 0.0;
      process->percent_mem = (process->m_resident * PAGE_SIZE_KB) / (float)(this->totalMem) * 100.0;
607

608
609
      if(!existingProcess) {
         process->user = UsersTable_getRef(this->usersTable, process->st_uid);
610

611
         if (! ProcessList_statProcessDir(process, dirname, name))
612
613
            goto errorReadingProcess;

614
615
616
         #ifdef HAVE_OPENVZ
         ProcessList_readOpenVZData(process, dirname, name);
         #endif
617

618
619
620
         #ifdef HAVE_CGROUP
         ProcessList_readCGroupFile(process, dirname, name);
         #endif
621
         
622
623
624
625
626
         #ifdef HAVE_VSERVER
         ProcessList_readVServerData(process, dirname, name);
         #endif
         
         if (! ProcessList_readCmdlineFile(process, dirname, name))
627
628
            goto errorReadingProcess;

629
630
         ProcessList_add(this, process);
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
631

632
633
      if (isThread) {
         if (this->showThreadNames || Process_isKernelThread(process) || process->state == 'Z') {
634
            free(process->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
635
            process->comm = String_copy(command);
636
637
638
         } else if (this->showingThreadNames) {
            if (! ProcessList_readCmdlineFile(process, dirname, name))
               goto errorReadingProcess;
Hisham Muhammad's avatar
Hisham Muhammad committed
639
         }
640
641
642
643
644
         if (Process_isKernelThread(process)) {
            this->kernelThreads++;
         } else {
            this->userlandThreads++;
         }
645
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
646

647
648
649
650
      this->totalTasks++;
      if (process->state == 'R')
         this->runningTasks++;
      process->updated = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
651

652
      continue;
Hisham Muhammad's avatar
Hisham Muhammad committed
653

654
655
656
657
658
      // Exception handler.
      errorReadingProcess: {
         if (process->comm) {
            free(process->comm);
            process->comm = NULL;
Hisham Muhammad's avatar
Hisham Muhammad committed
659
         }
660
661
662
663
         if (existingProcess)
            ProcessList_remove(this, process);
         else
            Process_delete((Object*)process);
Hisham Muhammad's avatar
Hisham Muhammad committed
664
665
666
      }
   }
   closedir(dir);
667
   return true;
Hisham Muhammad's avatar
Hisham Muhammad committed
668
669
670
}

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

674
675
   FILE* file = fopen(PROCMEMINFOFILE, "r");
   assert(file != NULL);
676
   int cpus = this->cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
677
678
   {
      char buffer[128];
679
      while (fgets(buffer, 128, file)) {
Hisham Muhammad's avatar
Hisham Muhammad committed
680
681
682
683
   
         switch (buffer[0]) {
         case 'M':
            if (String_startsWith(buffer, "MemTotal:"))
684
               sscanf(buffer, "MemTotal: %llu kB", &this->totalMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
685
            else if (String_startsWith(buffer, "MemFree:"))
686
               sscanf(buffer, "MemFree: %llu kB", &this->freeMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
687
            else if (String_startsWith(buffer, "MemShared:"))
688
               sscanf(buffer, "MemShared: %llu kB", &this->sharedMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
689
690
691
            break;
         case 'B':
            if (String_startsWith(buffer, "Buffers:"))
692
               sscanf(buffer, "Buffers: %llu kB", &this->buffersMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
693
694
695
            break;
         case 'C':
            if (String_startsWith(buffer, "Cached:"))
696
               sscanf(buffer, "Cached: %llu kB", &this->cachedMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
697
698
699
            break;
         case 'S':
            if (String_startsWith(buffer, "SwapTotal:"))
700
               sscanf(buffer, "SwapTotal: %llu kB", &this->totalSwap);
Hisham Muhammad's avatar
Hisham Muhammad committed
701
            if (String_startsWith(buffer, "SwapFree:"))
702
               sscanf(buffer, "SwapFree: %llu kB", &swapFree);
Hisham Muhammad's avatar
Hisham Muhammad committed
703
704
            break;
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
705
706
      }
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
707

Hisham Muhammad's avatar
Hisham Muhammad committed
708
709
   this->usedMem = this->totalMem - this->freeMem;
   this->usedSwap = this->totalSwap - swapFree;
710
   fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
711

712
713
   file = fopen(PROCSTATFILE, "r");
   assert(file != NULL);
714
   for (int i = 0; i <= cpus; i++) {
Hisham Muhammad's avatar
Hisham Muhammad committed
715
716
      char buffer[256];
      int cpuid;
717
718
      unsigned long long int ioWait, irq, softIrq, steal, guest;
      ioWait = irq = softIrq = steal = guest = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
719
720
721
      // Dependending on your kernel version,
      // 5, 7 or 8 of these fields will be set.
      // The rest will remain at zero.
722
      fgets(buffer, 255, file);
Hisham Muhammad's avatar
Hisham Muhammad committed
723
      if (i == 0)
724
         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
725
      else {
726
         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
727
728
729
730
         assert(cpuid == i - 1);
      }
      // Fields existing on kernels >= 2.6
      // (and RHEL's patched kernel 2.4...)
731
      idlealltime = idletime + ioWait;
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
      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
772
   }
773
   float period = (float)this->cpus[0].totalPeriod / cpus; fclose(file);
Hisham Muhammad's avatar
Hisham Muhammad committed
774
775

   // mark all process as "dirty"
776
777
   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
778
779
780
781
      p->updated = false;
   }
   
   this->totalTasks = 0;
782
783
   this->userlandThreads = 0;
   this->kernelThreads = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
784
   this->runningTasks = 0;
785

786
   ProcessList_processEntries(this, PROCDIR, NULL, period);
Hisham Muhammad's avatar
Hisham Muhammad committed
787
788
   
   this->showingThreadNames = this->showThreadNames;
Hisham Muhammad's avatar
Hisham Muhammad committed
789
   
790
791
   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
792
793
794
795
796
797
798
      if (p->updated == false)
         ProcessList_remove(this, p);
      else
         p->updated = false;
   }

}
799
800
801
802
803
804
805
806
807
808
809
810
811
812

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;
}
813
814
815
816
817
818
819
820

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;
   }
}