ProcessList.c 23.7 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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

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

/*{
#ifndef PROCDIR
#define PROCDIR "/proc"
#endif

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

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

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

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

}*/

/*{

59
60
61
62
#ifdef DEBUG
typedef int(*vxscanf)(void*, const char*, va_list);
#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
63
typedef struct ProcessList_ {
64
65
   Vector* processes;
   Vector* processes2;
Hisham Muhammad's avatar
Hisham Muhammad committed
66
67
68
69
70
71
72
73
   Hashtable* processTable;
   Process* prototype;
   UsersTable* usersTable;

   int processorCount;
   int totalTasks;
   int runningTasks;

74
75
76
   unsigned long long int* totalTime;
   unsigned long long int* userTime;
   unsigned long long int* systemTime;
77
   unsigned long long int* systemAllTime;
78
79
   unsigned long long int* idleTime;
   unsigned long long int* niceTime;
80
81
82
83
   unsigned long long int* ioWaitTime;
   unsigned long long int* irqTime;
   unsigned long long int* softIrqTime;
   unsigned long long int* stealTime;
84
85
86
   unsigned long long int* totalPeriod;
   unsigned long long int* userPeriod;
   unsigned long long int* systemPeriod;
87
   unsigned long long int* systemAllPeriod;
88
89
   unsigned long long int* idlePeriod;
   unsigned long long int* nicePeriod;
90
91
92
93
   unsigned long long int* ioWaitPeriod;
   unsigned long long int* irqPeriod;
   unsigned long long int* softIrqPeriod;
   unsigned long long int* stealPeriod;
94
95
96
97
98
99
100
101
102
103

   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
104
105
106
107
108
109
110
111
112
113
114

   ProcessField* fields;
   ProcessField sortKey;
   int direction;
   bool hideThreads;
   bool shadowOtherUsers;
   bool hideKernelThreads;
   bool hideUserlandThreads;
   bool treeView;
   bool highlightBaseName;
   bool highlightMegabytes;
Hisham Muhammad's avatar
Hisham Muhammad committed
115
116
117
   #ifdef DEBUG
   FILE* traceFile;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
118
119
120
121

} ProcessList;
}*/

122
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
123

Hisham Muhammad's avatar
Hisham Muhammad committed
124
125
126
127
128
#ifdef DEBUG

#define ProcessList_read(this, buffer, format, ...) ProcessList_xread(this, (vxscanf) vsscanf, buffer, format, ## __VA_ARGS__ )
#define ProcessList_fread(this, file, format, ...)  ProcessList_xread(this, (vxscanf) vfscanf, file, format, ## __VA_ARGS__ )

129
static FILE* ProcessList_fopen(ProcessList* this, const char* path, const char* mode) {
Hisham Muhammad's avatar
Hisham Muhammad committed
130
131
132
133
134
135
136
137
138
139
140
141
   fprintf(this->traceFile, "[%s]\n", path);
   return fopen(path, mode);
}

static inline int ProcessList_xread(ProcessList* this, vxscanf fn, void* buffer, char* format, ...) {
   va_list ap;
   va_start(ap, format);
   int num = fn(buffer, format, ap);
   va_end(format);
   va_start(ap, format);
   while (*format) {
      char ch = *format;
142
143
144
145
      char* c; int* d;
      long int* ld; unsigned long int* lu;
      long long int* lld; unsigned long long int* llu;
      char** s;
Hisham Muhammad's avatar
Hisham Muhammad committed
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
      if (ch != '%') {
         fprintf(this->traceFile, "%c", ch);
         format++;
         continue;
      }
      format++;
      switch(*format) {
      case 'c': c = va_arg(ap, char*);  fprintf(this->traceFile, "%c", *c); break;
      case 'd': d = va_arg(ap, int*);   fprintf(this->traceFile, "%d", *d); break;
      case 's': s = va_arg(ap, char**); fprintf(this->traceFile, "%s", *s); break;
      case 'l':
         format++;
         switch (*format) {
         case 'd': ld = va_arg(ap, long int*); fprintf(this->traceFile, "%ld", *ld); break;
         case 'u': lu = va_arg(ap, unsigned long int*); fprintf(this->traceFile, "%lu", *lu); break;
161
162
163
164
165
166
         case 'l':
            format++;
            switch (*format) {
            case 'd': lld = va_arg(ap, long long int*); fprintf(this->traceFile, "%lld", *lld); break;
            case 'u': llu = va_arg(ap, unsigned long long int*); fprintf(this->traceFile, "%llu", *llu); break;
            }
Hisham Muhammad's avatar
Hisham Muhammad committed
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
         }
      }
      format++;
   }
   fprintf(this->traceFile, "\n");
   va_end(format);
   return num;
}

#else

#ifndef ProcessList_read
#define ProcessList_fopen(this, path, mode) fopen(path, mode)
#define ProcessList_read(this, buffer, format, ...) sscanf(buffer, format, ## __VA_ARGS__ )
#define ProcessList_fread(this, file, format, ...) fscanf(file, format, ## __VA_ARGS__ )
#endif

#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
186
187
188
ProcessList* ProcessList_new(UsersTable* usersTable) {
   ProcessList* this;
   this = malloc(sizeof(ProcessList));
189
   this->processes = Vector_new(PROCESS_CLASS, true, DEFAULT_SIZE, Process_compare);
190
   this->processTable = Hashtable_new(70, false);
Hisham Muhammad's avatar
Hisham Muhammad committed
191
192
193
194
   this->prototype = Process_new(this);
   this->usersTable = usersTable;
   
   /* tree-view auxiliary buffers */
195
   this->processes2 = Vector_new(PROCESS_CLASS, true, DEFAULT_SIZE, Process_compare);
Hisham Muhammad's avatar
Hisham Muhammad committed
196
197
198
199
   
   #ifdef DEBUG
   this->traceFile = fopen("/tmp/htop-proc-trace", "w");
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
200
201
202
203
204
205
206
207
208
209
210

   FILE* status = fopen(PROCSTATFILE, "r");
   assert(status != NULL);
   char buffer[256];
   int procs = -1;
   do {
      procs++;
      fgets(buffer, 255, status);
   } while (String_startsWith(buffer, "cpu"));
   fclose(status);
   this->processorCount = procs - 1;
211
212
213
   this->totalTime = calloc(procs * FIELDS, sizeof(long long int));
   this->userTime = 

214
215
216
   this->totalTime = calloc(procs, sizeof(long long int));
   this->userTime = calloc(procs, sizeof(long long int));
   this->systemTime = calloc(procs, sizeof(long long int));
217
   this->systemAllTime = calloc(procs, sizeof(long long int));
218
219
   this->niceTime = calloc(procs, sizeof(long long int));
   this->idleTime = calloc(procs, sizeof(long long int));
220
221
222
223
   this->ioWaitTime = calloc(procs, sizeof(long long int));
   this->irqTime = calloc(procs, sizeof(long long int));
   this->softIrqTime = calloc(procs, sizeof(long long int));
   this->stealTime = calloc(procs, sizeof(long long int));
224
225
226
   this->totalPeriod = calloc(procs, sizeof(long long int));
   this->userPeriod = calloc(procs, sizeof(long long int));
   this->systemPeriod = calloc(procs, sizeof(long long int));
227
   this->systemAllPeriod = calloc(procs, sizeof(long long int));
228
229
   this->nicePeriod = calloc(procs, sizeof(long long int));
   this->idlePeriod = calloc(procs, sizeof(long long int));
230
231
232
233
   this->ioWaitPeriod = calloc(procs, sizeof(long long int));
   this->irqPeriod = calloc(procs, sizeof(long long int));
   this->softIrqPeriod = calloc(procs, sizeof(long long int));
   this->stealPeriod = calloc(procs, sizeof(long long int));
Hisham Muhammad's avatar
Hisham Muhammad committed
234
235
236
237
238
239
   for (int i = 0; i < procs; i++) {
      this->totalTime[i] = 1;
      this->totalPeriod[i] = 1;
   }

   this->fields = calloc(sizeof(ProcessField), LAST_PROCESSFIELD+1);
240
   // TODO: turn 'fields' into a Vector,
Hisham Muhammad's avatar
Hisham Muhammad committed
241
242
243
244
245
246
247
248
249
250
251
252
253
   // (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;
   this->hideKernelThreads = false;
   this->hideUserlandThreads = false;
   this->treeView = false;
   this->highlightBaseName = false;
   this->highlightMegabytes = false;
254
   this->expandSystemTime = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
255
256
257
258
259
260

   return this;
}

void ProcessList_delete(ProcessList* this) {
   Hashtable_delete(this->processTable);
261
262
   Vector_delete(this->processes);
   Vector_delete(this->processes2);
Hisham Muhammad's avatar
Hisham Muhammad committed
263
264
265
266
267
   Process_delete((Object*)this->prototype);

   free(this->totalTime);
   free(this->userTime);
   free(this->systemTime);
268
   free(this->systemAllTime);
Hisham Muhammad's avatar
Hisham Muhammad committed
269
270
   free(this->niceTime);
   free(this->idleTime);
271
272
273
274
   free(this->ioWaitTime);
   free(this->irqTime);
   free(this->softIrqTime);
   free(this->stealTime);
Hisham Muhammad's avatar
Hisham Muhammad committed
275
276
277
   free(this->totalPeriod);
   free(this->userPeriod);
   free(this->systemPeriod);
278
   free(this->systemAllPeriod);
Hisham Muhammad's avatar
Hisham Muhammad committed
279
280
   free(this->nicePeriod);
   free(this->idlePeriod);
281
282
283
284
   free(this->ioWaitPeriod);
   free(this->irqPeriod);
   free(this->softIrqPeriod);
   free(this->stealPeriod);
Hisham Muhammad's avatar
Hisham Muhammad committed
285

Hisham Muhammad's avatar
Hisham Muhammad committed
286
287
288
289
   #ifdef DEBUG
   fclose(this->traceFile);
   #endif

Hisham Muhammad's avatar
Hisham Muhammad committed
290
291
292
293
294
295
296
297
298
299
300
301
   free(this->fields);
   free(this);
}

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

RichString ProcessList_printHeader(ProcessList* this) {
302
303
   RichString out;
   RichString_init(&out);
Hisham Muhammad's avatar
Hisham Muhammad committed
304
305
306
307
308
309
310
311
312
313
314
315
316
   ProcessField* fields = this->fields;
   for (int i = 0; fields[i]; i++) {
      char* field = Process_printField(fields[i]);
      if (this->sortKey == fields[i])
         RichString_append(&out, CRT_colors[PANEL_HIGHLIGHT_FOCUS], field);
      else
         RichString_append(&out, CRT_colors[PANEL_HEADER_FOCUS], field);
   }
   return out;
}


void ProcessList_prune(ProcessList* this) {
317
   Vector_prune(this->processes);
Hisham Muhammad's avatar
Hisham Muhammad committed
318
319
320
}

void ProcessList_add(ProcessList* this, Process* p) {
321
   Vector_add(this->processes, p);
Hisham Muhammad's avatar
Hisham Muhammad committed
322
323
324
325
326
   Hashtable_put(this->processTable, p->pid, p);
}

void ProcessList_remove(ProcessList* this, Process* p) {
   Hashtable_remove(this->processTable, p->pid);
327
   int index = Vector_indexOf(this->processes, p, Process_pidCompare);
328
   Vector_remove(this->processes, index);
Hisham Muhammad's avatar
Hisham Muhammad committed
329
330
331
}

Process* ProcessList_get(ProcessList* this, int index) {
332
   return (Process*) (Vector_get(this->processes, index));
Hisham Muhammad's avatar
Hisham Muhammad committed
333
334
335
}

int ProcessList_size(ProcessList* this) {
336
   return (Vector_size(this->processes));
Hisham Muhammad's avatar
Hisham Muhammad committed
337
338
}

339
static void ProcessList_buildTree(ProcessList* this, int pid, int level, int indent, int direction) {
340
   Vector* children = Vector_new(PROCESS_CLASS, false, DEFAULT_SIZE, Process_compare);
Hisham Muhammad's avatar
Hisham Muhammad committed
341

342
343
   for (int i = 0; i < Vector_size(this->processes); i++) {
      Process* process = (Process*) (Vector_get(this->processes, i));
Hisham Muhammad's avatar
Hisham Muhammad committed
344
      if (process->ppid == pid) {
345
346
         Process* process = (Process*) (Vector_take(this->processes, i));
         Vector_add(children, process);
Hisham Muhammad's avatar
Hisham Muhammad committed
347
         i--;
Hisham Muhammad's avatar
Hisham Muhammad committed
348
349
      }
   }
350
   int size = Vector_size(children);
Hisham Muhammad's avatar
Hisham Muhammad committed
351
   for (int i = 0; i < size; i++) {
352
      Process* process = (Process*) (Vector_get(children, i));
Hisham Muhammad's avatar
Hisham Muhammad committed
353
      if (direction == 1)
354
         Vector_add(this->processes2, process);
Hisham Muhammad's avatar
Hisham Muhammad committed
355
      else
356
         Vector_insert(this->processes2, 0, process);
Hisham Muhammad's avatar
Hisham Muhammad committed
357
358
359
360
361
362
      int nextIndent = indent;
      if (i < size - 1)
         nextIndent = indent | (1 << level);
      ProcessList_buildTree(this, process->pid, level+1, nextIndent, direction);
      process->indent = indent | (1 << level);
   }
363
   Vector_delete(children);
Hisham Muhammad's avatar
Hisham Muhammad committed
364
365
366
367
}

void ProcessList_sort(ProcessList* this) {
   if (!this->treeView) {
368
      Vector_sort(this->processes);
Hisham Muhammad's avatar
Hisham Muhammad committed
369
370
371
372
373
   } else {
      int direction = this->direction;
      int sortKey = this->sortKey;
      this->sortKey = PID;
      this->direction = 1;
374
      Vector_sort(this->processes);
Hisham Muhammad's avatar
Hisham Muhammad committed
375
376
      this->sortKey = sortKey;
      this->direction = direction;
377
      Process* init = (Process*) (Vector_take(this->processes, 0));
Hisham Muhammad's avatar
Hisham Muhammad committed
378
379
      assert(init->pid == 1);
      init->indent = 0;
380
      Vector_add(this->processes2, init);
Hisham Muhammad's avatar
Hisham Muhammad committed
381
      ProcessList_buildTree(this, init->pid, 0, 0, direction);
382
      Vector* t = this->processes;
Hisham Muhammad's avatar
Hisham Muhammad committed
383
384
385
386
387
      this->processes = this->processes2;
      this->processes2 = t;
   }
}

388
static int ProcessList_readStatFile(ProcessList* this, Process *proc, FILE *f, char *command) {
Hisham Muhammad's avatar
Hisham Muhammad committed
389
   static char buf[MAX_READ];
390
   unsigned long int zero;
Hisham Muhammad's avatar
Hisham Muhammad committed
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407

   int size = fread(buf, 1, MAX_READ, f);
   if(!size) return 0;

   proc->pid = atoi(buf);
   char *location = strchr(buf, ' ');
   if(!location) return 0;

   location += 2;
   char *end = strrchr(location, ')');
   if(!end) return 0;
   
   int commsize = end - location;
   memcpy(command, location, commsize);
   command[commsize] = '\0';
   location = end + 2;
   
408
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
409
   int num = ProcessList_read(this, location, 
Hisham Muhammad's avatar
Hisham Muhammad committed
410
411
412
413
414
415
      "%c %d %d %d %d %d %lu %lu %lu %lu "
      "%lu %lu %lu %ld %ld %ld %ld %ld %ld "
      "%lu %lu %ld %lu %lu %lu %lu %lu "
      "%lu %lu %lu %lu %lu %lu %lu %lu "
      "%d %d",
      &proc->state, &proc->ppid, &proc->pgrp, &proc->session, &proc->tty_nr, 
416
417
418
      &proc->tpgid, &proc->flags,
      &proc->minflt, &proc->cminflt, &proc->majflt, &proc->cmajflt,
      &proc->utime, &proc->stime, &proc->cutime, &proc->cstime, 
Hisham Muhammad's avatar
Hisham Muhammad committed
419
420
421
422
423
424
      &proc->priority, &proc->nice, &zero, &proc->itrealvalue, 
      &proc->starttime, &proc->vsize, &proc->rss, &proc->rlim, 
      &proc->startcode, &proc->endcode, &proc->startstack, &proc->kstkesp, 
      &proc->kstkeip, &proc->signal, &proc->blocked, &proc->sigignore, 
      &proc->sigcatch, &proc->wchan, &proc->nswap, &proc->cnswap, 
      &proc->exit_signal, &proc->processor);
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
   #else
   long int uzero;
   int num = ProcessList_read(this, location, 
      "%c %d %d %d %d %d %lu %lu %lu %lu "
      "%lu %lu %lu %ld %ld %ld %ld %ld %ld "
      "%lu %lu %ld %lu %lu %lu %lu %lu "
      "%lu %lu %lu %lu %lu %lu %lu %lu "
      "%d %d",
      &proc->state, &proc->ppid, &proc->pgrp, &proc->session, &proc->tty_nr, 
      &proc->tpgid, &proc->flags,
      &zero, &zero, &zero, &zero,
      &proc->utime, &proc->stime, &proc->cutime, &proc->cstime, 
      &proc->priority, &proc->nice, &uzero, &uzero, 
      &zero, &zero, &uzero, &zero, 
      &zero, &zero, &zero, &zero, 
      &zero, &zero, &zero, &zero, 
      &zero, &zero, &zero, &zero, 
      &proc->exit_signal, &proc->processor);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
444
445
446
447
448
449
450
451
452
   
   // This assert is always valid on 2.4, but reportedly not always valid on 2.6.
   // TODO: Check if the semantics of this field has changed.
   // assert(zero == 0);
   
   if(num != 37) return 0;
   return 1;
}

Hisham Muhammad's avatar
Hisham Muhammad committed
453
bool ProcessList_readStatusFile(ProcessList* this, Process* proc, char* dirname, char* name) {
Hisham Muhammad's avatar
Hisham Muhammad committed
454
455
   char statusfilename[MAX_NAME+1];
   statusfilename[MAX_NAME] = '\0';
456
457
458
459
   /*
   bool success = false;
   char buffer[256];
   buffer[255] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
460
   snprintf(statusfilename, MAX_NAME, "%s/%s/status", dirname, name);
Hisham Muhammad's avatar
Hisham Muhammad committed
461
   FILE* status = ProcessList_fopen(this, statusfilename, "r");
Hisham Muhammad's avatar
Hisham Muhammad committed
462
463
   if (status) {
      while (!feof(status)) {
464
         char* ok = fgets(buffer, 255, status);
Hisham Muhammad's avatar
Hisham Muhammad committed
465
466
467
468
469
         if (!ok)
            break;
         if (String_startsWith(buffer, "Uid:")) {
            int uid1, uid2, uid3, uid4;
            // TODO: handle other uid's.
Hisham Muhammad's avatar
Hisham Muhammad committed
470
            int ok = ProcessList_read(this, buffer, "Uid:\t%d\t%d\t%d\t%d", &uid1, &uid2, &uid3, &uid4);
Hisham Muhammad's avatar
Hisham Muhammad committed
471
472
473
474
475
476
477
478
479
480
            if (ok >= 1) {
               proc->st_uid = uid1;
               success = true;
            }
            break;
         }
      }
      fclose(status);
   }
   if (!success) {
481
482
   */
      snprintf(statusfilename, MAX_NAME, "%s/%s", dirname, name);
Hisham Muhammad's avatar
Hisham Muhammad committed
483
484
485
486
487
      struct stat sstat;
      int statok = stat(statusfilename, &sstat);
      if (statok == -1)
         return false;
      proc->st_uid = sstat.st_uid;
488
489
490
491
492
      return true;
   /*
   } else
      return true;
   */
Hisham Muhammad's avatar
Hisham Muhammad committed
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
}

void ProcessList_processEntries(ProcessList* this, char* dirname, int parent, float period) {
   DIR* dir;
   struct dirent* entry;
   Process* prototype = this->prototype;

   dir = opendir(dirname);
   assert(dir != NULL);
   while ((entry = readdir(dir)) != NULL) {
      char* name = entry->d_name;
      int pid;
      // filename is a number: process directory
      pid = atoi(name);

      // The RedHat kernel hides threads with a dot.
      // I believe this is non-standard.
      bool isThread = false;
      if ((!this->hideThreads) && pid == 0 && name[0] == '.') {
         char* tname = name + 1;
         pid = atoi(tname);
         if (pid > 0)
            isThread = true;
      }

      if (pid > 0 && pid != parent) {
         if (!this->hideUserlandThreads) {
            char subdirname[MAX_NAME+1];
            snprintf(subdirname, MAX_NAME, "%s/%s/task", dirname, name);
   
            if (access(subdirname, X_OK) == 0) {
               ProcessList_processEntries(this, subdirname, pid, period);
            }
Hisham Muhammad's avatar
Hisham Muhammad committed
526
527
         }

Hisham Muhammad's avatar
Hisham Muhammad committed
528
529
530
531
532
         FILE* status;
         char statusfilename[MAX_NAME+1];
         char command[PROCESS_COMM_LEN + 1];

         Process* process;
533

Hisham Muhammad's avatar
Hisham Muhammad committed
534
         Process* existingProcess = (Process*) Hashtable_get(this->processTable, pid);
535
536
537
538
539
         if (existingProcess) {
            process = existingProcess;
         } else {
            process = prototype;
            process->comm = NULL;
Hisham Muhammad's avatar
Hisham Muhammad committed
540
            process->pid = pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
541
            if (! ProcessList_readStatusFile(this, process, dirname, name))
Hisham Muhammad's avatar
Hisham Muhammad committed
542
543
544
545
               goto errorReadingProcess;
         }
         process->updated = true;

546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
         snprintf(statusfilename, MAX_NAME, "%s/%s/statm", dirname, name);
         status = ProcessList_fopen(this, statusfilename, "r");

         if(!status) {
            goto errorReadingProcess;
         }
         int num = ProcessList_fread(this, status, "%d %d %d %d %d %d %d",
             &process->m_size, &process->m_resident, &process->m_share, 
             &process->m_trs, &process->m_drs, &process->m_lrs, 
             &process->m_dt);

         fclose(status);
         if(num != 7)
            goto errorReadingProcess;

         if (this->hideKernelThreads && process->m_size == 0)
            goto errorReadingProcess;

564
565
566
567
568
569
570
571
572
573
574
575
576
         int lasttimes = (process->utime + process->stime);

         snprintf(statusfilename, MAX_NAME, "%s/%s/stat", dirname, name);
         
         status = ProcessList_fopen(this, statusfilename, "r");
         if (status == NULL) 
            goto errorReadingProcess;

         int success = ProcessList_readStatFile(this, process, status, command);
         fclose(status);
         if(!success)
            goto errorReadingProcess;

Hisham Muhammad's avatar
Hisham Muhammad committed
577
         if(!existingProcess) {
578
            process->user = UsersTable_getRef(this->usersTable, process->st_uid);
579
 
Hisham Muhammad's avatar
Hisham Muhammad committed
580
            snprintf(statusfilename, MAX_NAME, "%s/%s/cmdline", dirname, name);
Hisham Muhammad's avatar
Hisham Muhammad committed
581
            status = ProcessList_fopen(this, statusfilename, "r");
Hisham Muhammad's avatar
Hisham Muhammad committed
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
            if (!status) {
               goto errorReadingProcess;
            }

            int amtRead = fread(command, 1, PROCESS_COMM_LEN - 1, status);
            if (amtRead > 0) {
               for (int i = 0; i < amtRead; i++)
                  if (command[i] == '\0' || command[i] == '\n')
                     command[i] = ' ';
               command[amtRead] = '\0';
            }
            command[PROCESS_COMM_LEN] = '\0';
            process->comm = String_copy(command);
            fclose(status);
         }

598
599
600
         process->percent_cpu = (process->utime + process->stime - lasttimes) / 
            period * 100.0;

Hisham Muhammad's avatar
Hisham Muhammad committed
601
602
603
604
605
         process->percent_mem = process->m_resident / 
            (float)(this->usedMem - this->cachedMem - this->buffersMem) * 
            100.0;

         this->totalTasks++;
Hisham Muhammad's avatar
Hisham Muhammad committed
606
607
608
         if (process->state == 'R') {
            this->runningTasks++;
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
609

610
611
612
613
         if (!existingProcess) {
            process = Process_clone(process);
            ProcessList_add(this, process);
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
614
615
616
617
618

         continue;

         // Exception handler.
         errorReadingProcess: {
619
620
621
622
623
624
            if (existingProcess)
               ProcessList_remove(this, process);
            else {
               if (process->comm)
                  free(process->comm);
            }
Hisham Muhammad's avatar
Hisham Muhammad committed
625
626
627
         }
      }
   }
628
   prototype->comm = NULL;
Hisham Muhammad's avatar
Hisham Muhammad committed
629
630
631
632
   closedir(dir);
}

void ProcessList_scan(ProcessList* this) {
633
   unsigned long long int usertime, nicetime, systemtime, systemalltime, idletime, totaltime;
634
   unsigned long long int swapFree;
Hisham Muhammad's avatar
Hisham Muhammad committed
635
636
637

   FILE* status;
   char buffer[128];
Hisham Muhammad's avatar
Hisham Muhammad committed
638
   status = ProcessList_fopen(this, PROCMEMINFOFILE, "r");
Hisham Muhammad's avatar
Hisham Muhammad committed
639
640
641
642
643
644
645
   assert(status != NULL);
   while (!feof(status)) {
      fgets(buffer, 128, status);

      switch (buffer[0]) {
      case 'M':
         if (String_startsWith(buffer, "MemTotal:"))
646
            ProcessList_read(this, buffer, "MemTotal: %llu kB", &this->totalMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
647
         else if (String_startsWith(buffer, "MemFree:"))
648
            ProcessList_read(this, buffer, "MemFree: %llu kB", &this->freeMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
649
         else if (String_startsWith(buffer, "MemShared:"))
650
            ProcessList_read(this, buffer, "MemShared: %llu kB", &this->sharedMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
651
652
653
         break;
      case 'B':
         if (String_startsWith(buffer, "Buffers:"))
654
            ProcessList_read(this, buffer, "Buffers: %llu kB", &this->buffersMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
655
656
657
         break;
      case 'C':
         if (String_startsWith(buffer, "Cached:"))
658
            ProcessList_read(this, buffer, "Cached: %llu kB", &this->cachedMem);
Hisham Muhammad's avatar
Hisham Muhammad committed
659
660
661
         break;
      case 'S':
         if (String_startsWith(buffer, "SwapTotal:"))
662
            ProcessList_read(this, buffer, "SwapTotal: %llu kB", &this->totalSwap);
Hisham Muhammad's avatar
Hisham Muhammad committed
663
         if (String_startsWith(buffer, "SwapFree:"))
664
            ProcessList_read(this, buffer, "SwapFree: %llu kB", &swapFree);
Hisham Muhammad's avatar
Hisham Muhammad committed
665
666
667
         break;
      }
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
668

Hisham Muhammad's avatar
Hisham Muhammad committed
669
670
671
672
   this->usedMem = this->totalMem - this->freeMem;
   this->usedSwap = this->totalSwap - swapFree;
   fclose(status);

Hisham Muhammad's avatar
Hisham Muhammad committed
673
674
   status = ProcessList_fopen(this, PROCSTATFILE, "r");

Hisham Muhammad's avatar
Hisham Muhammad committed
675
676
677
678
   assert(status != NULL);
   for (int i = 0; i <= this->processorCount; i++) {
      char buffer[256];
      int cpuid;
679
      unsigned long long int ioWait, irq, softIrq, steal;
Hisham Muhammad's avatar
Hisham Muhammad committed
680
681
682
683
684
685
      ioWait = irq = softIrq = steal = 0;
      // Dependending on your kernel version,
      // 5, 7 or 8 of these fields will be set.
      // The rest will remain at zero.
      fgets(buffer, 255, status);
      if (i == 0)
686
         ProcessList_read(this, buffer, "cpu  %llu %llu %llu %llu %llu %llu %llu %llu", &usertime, &nicetime, &systemtime, &idletime, &ioWait, &irq, &softIrq, &steal);
Hisham Muhammad's avatar
Hisham Muhammad committed
687
      else {
688
         ProcessList_read(this, buffer, "cpu%d %llu %llu %llu %llu %llu %llu %llu %llu", &cpuid, &usertime, &nicetime, &systemtime, &idletime, &ioWait, &irq, &softIrq, &steal);
Hisham Muhammad's avatar
Hisham Muhammad committed
689
690
691
692
         assert(cpuid == i - 1);
      }
      // Fields existing on kernels >= 2.6
      // (and RHEL's patched kernel 2.4...)
693
694
      systemalltime = systemtime + ioWait + irq + softIrq + steal;
      totaltime = usertime + nicetime + systemalltime + idletime;
Hisham Muhammad's avatar
Hisham Muhammad committed
695
696
697
698
699
      assert (usertime >= this->userTime[i]);
      assert (nicetime >= this->niceTime[i]);
      assert (systemtime >= this->systemTime[i]);
      assert (idletime >= this->idleTime[i]);
      assert (totaltime >= this->totalTime[i]);
700
701
702
703
704
      assert (systemalltime >= this->systemAllTime[i]);
      assert (ioWait >= this->ioWaitTime[i]);
      assert (irqTime >= this->irqTime[i]);
      assert (softIrqTime >= this->softIrqTime[i]);
      assert (stealTime >= this->stealTime[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
705
706
707
      this->userPeriod[i] = usertime - this->userTime[i];
      this->nicePeriod[i] = nicetime - this->niceTime[i];
      this->systemPeriod[i] = systemtime - this->systemTime[i];
708
      this->systemAllPeriod[i] = systemalltime - this->systemAllTime[i];
Hisham Muhammad's avatar
Hisham Muhammad committed
709
      this->idlePeriod[i] = idletime - this->idleTime[i];
710
711
712
713
      this->ioWaitPeriod[i] = ioWait - this->ioWaitTime[i];
      this->irqPeriod[i] = irq - this->irqTime[i];
      this->softIrqPeriod[i] = softIrq - this->softIrqTime[i];
      this->stealPeriod[i] = steal - this->stealTime[i];
Hisham Muhammad's avatar
Hisham Muhammad committed
714
715
716
717
      this->totalPeriod[i] = totaltime - this->totalTime[i];
      this->userTime[i] = usertime;
      this->niceTime[i] = nicetime;
      this->systemTime[i] = systemtime;
718
      this->systemAllTime[i] = systemalltime;
Hisham Muhammad's avatar
Hisham Muhammad committed
719
      this->idleTime[i] = idletime;
720
721
722
723
      this->ioWaitTime[i] = ioWait;
      this->irqTime[i] = irq;
      this->softIrqTime[i] = softIrq;
      this->stealTime[i] = steal;
Hisham Muhammad's avatar
Hisham Muhammad committed
724
725
726
727
728
729
      this->totalTime[i] = totaltime;
   }
   float period = (float)this->totalPeriod[0] / this->processorCount;
   fclose(status);

   // mark all process as "dirty"
730
731
   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
732
733
734
735
736
737
738
739
      p->updated = false;
   }
   
   this->totalTasks = 0;
   this->runningTasks = 0;
   
   ProcessList_processEntries(this, PROCDIR, 0, period);
   
740
741
   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
742
743
744
745
746
747
748
      if (p->updated == false)
         ProcessList_remove(this, p);
      else
         p->updated = false;
   }

}