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

Hisham Muhammad's avatar
Hisham Muhammad committed
8
9
#include "Process.h"

Hisham Muhammad's avatar
Hisham Muhammad committed
10
11
12
#include "ProcessList.h"
#include "CRT.h"
#include "String.h"
13
#include "RichString.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
14
15
16
17
18
19
20

#include <stdio.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <unistd.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
21
#include <stdlib.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
22
23
24
25
#include <signal.h>
#include <string.h>
#include <stdbool.h>
#include <pwd.h>
26
#include <sched.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
27
#include <time.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
28
#include <assert.h>
29
#include <sys/syscall.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
30

31
#ifdef HAVE_LIBHWLOC
32
#include <hwloc/linux.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
33
#endif
34

Hisham Muhammad's avatar
Hisham Muhammad committed
35
36
// This works only with glibc 2.1+. On earlier versions
// the behavior is similar to have a hardcoded page size.
37
#ifndef PAGE_SIZE
Hisham Muhammad's avatar
Hisham Muhammad committed
38
#define PAGE_SIZE ( sysconf(_SC_PAGESIZE) )
39
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
40
#define PAGE_SIZE_KB ( PAGE_SIZE / ONE_K )
Hisham Muhammad's avatar
Hisham Muhammad committed
41
42

/*{
Hisham Muhammad's avatar
Hisham Muhammad committed
43
44
#include "Object.h"
#include "Affinity.h"
45
#include "IOPriority.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
46
#include <sys/types.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
47

48
49
50
51
52
53
#define PROCESS_FLAG_IO 1
#define PROCESS_FLAG_IOPRIO 2
#define PROCESS_FLAG_OPENVZ 4
#define PROCESS_FLAG_VSERVER 8
#define PROCESS_FLAG_CGROUP 16

54
55
56
57
58
59
60
61
#ifndef Process_isKernelThread
#define Process_isKernelThread(_process) (_process->pgrp == 0)
#endif

#ifndef Process_isUserlandThread
#define Process_isUserlandThread(_process) (_process->pid != _process->tgid)
#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
62
#ifndef Process_isThread
63
#define Process_isThread(_process) (Process_isUserlandThread(_process) || Process_isKernelThread(_process))
Hisham Muhammad's avatar
Hisham Muhammad committed
64
65
#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
66
67
68
69
70
typedef enum ProcessField_ {
   PID = 1, COMM, STATE, PPID, PGRP, SESSION, TTY_NR, TPGID, FLAGS, MINFLT, CMINFLT, MAJFLT, CMAJFLT, UTIME,
   STIME, CUTIME, CSTIME, PRIORITY, NICE, ITREALVALUE, STARTTIME, VSIZE, RSS, RLIM, STARTCODE, ENDCODE,
   STARTSTACK, KSTKESP, KSTKEIP, SIGNAL, BLOCKED, SSIGIGNORE, SIGCATCH, WCHAN, NSWAP, CNSWAP, EXIT_SIGNAL,
   PROCESSOR, M_SIZE, M_RESIDENT, M_SHARE, M_TRS, M_DRS, M_LRS, M_DT, ST_UID, PERCENT_CPU, PERCENT_MEM,
71
   USER, TIME, NLWP, TGID,
72
   #ifdef HAVE_OPENVZ
73
   CTID, VPID,
74
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
75
76
77
   #ifdef HAVE_VSERVER
   VXID,
   #endif
78
   #ifdef HAVE_TASKSTATS
79
   RCHAR, WCHAR, SYSCR, SYSCW, RBYTES, WBYTES, CNCLWB, IO_READ_RATE, IO_WRITE_RATE, IO_RATE,
80
   #endif
81
82
83
   #ifdef HAVE_CGROUP
   CGROUP,
   #endif
84
85
86
   #ifdef HAVE_OOM
   OOM,
   #endif
87
   IO_PRIORITY,
88
   LAST_PROCESSFIELD
Hisham Muhammad's avatar
Hisham Muhammad committed
89
90
91
92
93
94
95
96
97
98
} ProcessField;

struct ProcessList_;

typedef struct Process_ {
   Object super;

   struct ProcessList_ *pl;
   bool updated;

Hisham Muhammad's avatar
Hisham Muhammad committed
99
   pid_t pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
100
101
102
103
   char* comm;
   int indent;
   char state;
   bool tag;
Hisham Muhammad's avatar
Hisham Muhammad committed
104
   bool showChildren;
105
   bool show;
Hisham Muhammad's avatar
Hisham Muhammad committed
106
   pid_t ppid;
107
108
109
   unsigned int pgrp;
   unsigned int session;
   unsigned int tty_nr;
Hisham Muhammad's avatar
Hisham Muhammad committed
110
   pid_t tgid;
111
   int tpgid;
Hisham Muhammad's avatar
Hisham Muhammad committed
112
   unsigned long int flags;
113
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
114
115
116
117
   unsigned long int minflt;
   unsigned long int cminflt;
   unsigned long int majflt;
   unsigned long int cmajflt;
118
   #endif
119
120
121
122
   unsigned long long int utime;
   unsigned long long int stime;
   unsigned long long int cutime;
   unsigned long long int cstime;
Hisham Muhammad's avatar
Hisham Muhammad committed
123
124
   long int priority;
   long int nice;
125
   long int nlwp;
126
   IOPriority ioPriority;
Hisham Muhammad's avatar
Hisham Muhammad committed
127
128
   char starttime_show[8];
   time_t starttime_ctime;
129
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
   long int itrealvalue;
   unsigned long int vsize;
   long int rss;
   unsigned long int rlim;
   unsigned long int startcode;
   unsigned long int endcode;
   unsigned long int startstack;
   unsigned long int kstkesp;
   unsigned long int kstkeip;
   unsigned long int signal;
   unsigned long int blocked;
   unsigned long int sigignore;
   unsigned long int sigcatch;
   unsigned long int wchan;
   unsigned long int nswap;
   unsigned long int cnswap;
146
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
147
148
149
150
151
152
153
154
155
156
157
158
   int exit_signal;
   int processor;
   int m_size;
   int m_resident;
   int m_share;
   int m_trs;
   int m_drs;
   int m_lrs;
   int m_dt;
   uid_t st_uid;
   float percent_cpu;
   float percent_mem;
159
   char* user;
160
   #ifdef HAVE_OPENVZ
161
   unsigned int ctid;
162
163
   unsigned int vpid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
164
165
166
   #ifdef HAVE_VSERVER
   unsigned int vxid;
   #endif
167
168
169
170
171
172
173
174
175
176
177
178
179
   #ifdef HAVE_TASKSTATS
   unsigned long long io_rchar;
   unsigned long long io_wchar;
   unsigned long long io_syscr;
   unsigned long long io_syscw;
   unsigned long long io_read_bytes;
   unsigned long long io_write_bytes;
   unsigned long long io_cancelled_write_bytes;
   double io_rate_read_bps;
   unsigned long long io_rate_read_time;
   double io_rate_write_bps;
   unsigned long long io_rate_write_time;   
   #endif
180
181
182
   #ifdef HAVE_CGROUP
   char* cgroup;
   #endif
183
184
185
   #ifdef HAVE_OOM
   unsigned int oom;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
186
187
188
189
} Process;

}*/

Hisham Muhammad's avatar
Hisham Muhammad committed
190
const char *Process_fieldNames[] = {
191
192
193
194
195
196
197
198
   "", "PID", "Command", "STATE", "PPID", "PGRP", "SESSION",
   "TTY_NR", "TPGID", "FLAGS", "MINFLT", "CMINFLT", "MAJFLT", "CMAJFLT",
   "UTIME", "STIME", "CUTIME", "CSTIME", "PRIORITY", "NICE", "ITREALVALUE",
   "STARTTIME", "VSIZE", "RSS", "RLIM", "STARTCODE", "ENDCODE", "STARTSTACK",
   "KSTKESP", "KSTKEIP", "SIGNAL", "BLOCKED", "SIGIGNORE", "SIGCATCH", "WCHAN",
   "NSWAP", "CNSWAP", "EXIT_SIGNAL", "PROCESSOR", "M_SIZE", "M_RESIDENT", "M_SHARE",
   "M_TRS", "M_DRS", "M_LRS", "M_DT", "ST_UID", "PERCENT_CPU", "PERCENT_MEM",
   "USER", "TIME", "NLWP", "TGID", 
199
#ifdef HAVE_OPENVZ
200
   "CTID", "VPID",
201
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
202
203
204
#ifdef HAVE_VSERVER
   "VXID",
#endif
205
#ifdef HAVE_TASKSTATS
206
207
   "RCHAR", "WCHAR", "SYSCR", "SYSCW", "RBYTES", "WBYTES", "CNCLWB",
   "IO_READ_RATE", "IO_WRITE_RATE", "IO_RATE",
208
#endif
209
210
#ifdef HAVE_CGROUP
   "CGROUP",
211
212
213
#endif
#ifdef HAVE_OOM
   "OOM",
214
#endif
215
   "IO_PRIORITY",
216
"*** report bug! ***"
217
};
Hisham Muhammad's avatar
Hisham Muhammad committed
218

219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
const int Process_fieldFlags[] = {
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0,
#ifdef HAVE_OPENVZ
   PROCESS_FLAG_OPENVZ, PROCESS_FLAG_OPENVZ,
#endif
#ifdef HAVE_VSERVER
   PROCESS_FLAG_VSERVER,
#endif
#ifdef HAVE_TASKSTATS
   PROCESS_FLAG_IO, PROCESS_FLAG_IO, PROCESS_FLAG_IO, PROCESS_FLAG_IO, PROCESS_FLAG_IO, PROCESS_FLAG_IO, PROCESS_FLAG_IO, 
   PROCESS_FLAG_IO, PROCESS_FLAG_IO, PROCESS_FLAG_IO, 
#endif
#ifdef HAVE_CGROUP
   PROCESS_FLAG_CGROUP,
240
241
242
#endif
#ifdef HAVE_OOM
   0,
243
244
245
246
#endif
   PROCESS_FLAG_IOPRIO
};

Hisham Muhammad's avatar
Hisham Muhammad committed
247
const char *Process_fieldTitles[] = {
248
   "", "    PID ", "Command ", "S ", "   PPID ", "   PGRP ", "   SESN ",
249
   "  TTY ", "  TPGID ", "- ", "- ", "- ", "- ", "- ",
250
   " UTIME+  ", " STIME+  ", " CUTIME+ ", " CSTIME+ ", "PRI ", " NI ", "- ",
Hisham Muhammad's avatar
Hisham Muhammad committed
251
   "START ", "- ", "- ", "- ", "- ", "- ", "- ",
252
253
254
   "- ", "- ", "- ", "- ", "- ", "- ", "- ",
   "- ", "- ", "- ", "CPU ", " VIRT ", "  RES ", "  SHR ",
   " CODE ", " DATA ", " LIB ", " DIRTY ", " UID ", "CPU% ", "MEM% ",
255
   "USER      ", "  TIME+  ", "NLWP ", "   TGID ",
256
#ifdef HAVE_OPENVZ
257
   " CTID ", " VPID ",
258
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
259
260
261
#ifdef HAVE_VSERVER
   " VXID ",
#endif
262
#ifdef HAVE_TASKSTATS
263
   "     RD_CHAR ", "     WR_CHAR ", "   RD_SYSC ", "   WR_SYSC ", "  IO_RBYTES ", "  IO_WBYTES ", " IO_CANCEL ",
264
265
   " IORR ", " IOWR ", "   IO ",
#endif
266
267
#ifdef HAVE_CGROUP
   "    CGROUP ",
268
269
270
#endif
#ifdef HAVE_OOM
   "    OOM ",
271
#endif
272
   "IO ",
273
"*** report bug! ***"
274
275
};

276
277
static int Process_getuid = -1;

278
279
280
281
282
283
284
static char* Process_pidFormat = "%7u ";
static char* Process_tpgidFormat = "%7u ";

void Process_getMaxPid() {
   FILE* file = fopen(PROCDIR "/sys/kernel/pid_max", "r");
   if (!file) return;
   int maxPid = 4194303;
285
   fscanf(file, "%32d", &maxPid);
286
287
   fclose(file);
   if (maxPid > 99999) {
288
289
290
291
292
293
      Process_fieldTitles[PID] =     "    PID ";
      Process_fieldTitles[PPID] =    "   PPID ";
      Process_fieldTitles[TPGID] =   "  TPGID ";
      Process_fieldTitles[TGID] =    "   TGID ";
      Process_fieldTitles[PGRP] =    "   PGRP ";
      Process_fieldTitles[SESSION] = "   SESN ";
294
295
296
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "    OOM ";
      #endif
297
298
299
      Process_pidFormat = "%7u ";
      Process_tpgidFormat = "%7d ";
   } else {
300
301
302
303
304
305
      Process_fieldTitles[PID] =     "  PID ";
      Process_fieldTitles[PPID] =    " PPID ";
      Process_fieldTitles[TPGID] =   "TPGID ";
      Process_fieldTitles[TGID] =    " TGID ";
      Process_fieldTitles[PGRP] =    " PGRP ";
      Process_fieldTitles[SESSION] = " SESN ";
306
307
308
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "  OOM ";
      #endif
309
310
311
312
313
      Process_pidFormat = "%5u ";
      Process_tpgidFormat = "%5d ";
   }
}

Hisham Muhammad's avatar
Hisham Muhammad committed
314
315
316
317
#define ONE_K 1024
#define ONE_M (ONE_K * ONE_K)
#define ONE_G (ONE_M * ONE_K)

318
319
320
#define ONE_DECIMAL_K 1000
#define ONE_DECIMAL_M (ONE_DECIMAL_K * ONE_DECIMAL_K)

321
static void Process_humanNumber(Process* this, RichString* str, unsigned long number) {
Hisham Muhammad's avatar
Hisham Muhammad committed
322
323
   char buffer[11];
   int len;
324
325
   if(number >= (10 * ONE_DECIMAL_M)) {
      if(number >= (100 * ONE_DECIMAL_M)) {
326
327
328
329
330
331
332
         len = snprintf(buffer, 10, "%4ldG ", number / ONE_M);
         RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, len);
      } else {
         len = snprintf(buffer, 10, "%3.1fG ", (float)number / ONE_M);
         RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, len);
      }
   } else if (number >= 100000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
333
      len = snprintf(buffer, 10, "%4ldM ", number / ONE_K);
Hisham Muhammad's avatar
Hisham Muhammad committed
334
335
336
337
338
      int attr = this->pl->highlightMegabytes
               ? CRT_colors[PROCESS_MEGABYTES]
               : CRT_colors[PROCESS];
      RichString_appendn(str, attr, buffer, len);
   } else if (this->pl->highlightMegabytes && number >= 1000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
339
      len = snprintf(buffer, 10, "%2ld", number/1000);
Hisham Muhammad's avatar
Hisham Muhammad committed
340
341
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer, len);
      number %= 1000;
Hisham Muhammad's avatar
Hisham Muhammad committed
342
      len = snprintf(buffer, 10, "%03ld ", number);
Hisham Muhammad's avatar
Hisham Muhammad committed
343
344
      RichString_appendn(str, CRT_colors[PROCESS], buffer, len);
   } else {
Hisham Muhammad's avatar
Hisham Muhammad committed
345
      len = snprintf(buffer, 10, "%5ld ", number);
Hisham Muhammad's avatar
Hisham Muhammad committed
346
347
348
349
      RichString_appendn(str, CRT_colors[PROCESS], buffer, len);
   }
}

350
static void Process_colorNumber(RichString* str, unsigned long long number) {
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
   char buffer[14];
   if (number > 10000000000) {
      snprintf(buffer, 13, "%11lld ", number / 1000);
      RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, 5);
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer+5, 3);
      RichString_appendn(str, CRT_colors[PROCESS], buffer+8, 4);
   } else {
      snprintf(buffer, 13, "%11lld ", number);
      RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, 2);
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer+2, 3);
      RichString_appendn(str, CRT_colors[PROCESS], buffer+5, 3);
      RichString_appendn(str, CRT_colors[PROCESS_SHADOW], buffer+8, 4);
   }
}

366
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
367

368
static void Process_printTime(RichString* str, unsigned long long t) {
Hisham Muhammad's avatar
Hisham Muhammad committed
369
370
371
372
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;

   double realTime = t * jiffytime;
373
   unsigned long long iRealTime = (unsigned long long) realTime;
Hisham Muhammad's avatar
Hisham Muhammad committed
374

375
   unsigned long long hours = iRealTime / 3600;
Hisham Muhammad's avatar
Hisham Muhammad committed
376
377
378
379
   int minutes = (iRealTime / 60) % 60;
   int seconds = iRealTime % 60;
   int hundredths = (realTime - iRealTime) * 100;
   char buffer[11];
380
381
   if (hours >= 100) {
      snprintf(buffer, 10, "%7lluh ", hours);
Hisham Muhammad's avatar
Hisham Muhammad committed
382
383
      RichString_append(str, CRT_colors[LARGE_NUMBER], buffer);
   } else {
384
385
386
387
388
389
390
391
      if (hours) {
         snprintf(buffer, 10, "%2lluh", hours);
         RichString_append(str, CRT_colors[LARGE_NUMBER], buffer);
         snprintf(buffer, 10, "%02d:%02d ", minutes, seconds);
      } else {
         snprintf(buffer, 10, "%2d:%02d.%02d ", minutes, seconds, hundredths);
      }
      RichString_append(str, CRT_colors[DEFAULT_COLOR], buffer);
Hisham Muhammad's avatar
Hisham Muhammad committed
392
393
394
   }
}

395
static inline void Process_writeCommand(Process* this, int attr, int baseattr, RichString* str) {
396
   int start = RichString_size(str);
397
   RichString_append(str, attr, this->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
398
   if (this->pl->highlightBaseName) {
399
      int finish = RichString_size(str) - 1;
400
401
402
403
404
405
406
407
      int space = RichString_findChar(str, ' ', start);
      if (space != -1)
         finish = space - 1;
      for (;;) {
         int slash = RichString_findChar(str, '/', start);
         if (slash == -1 || slash > finish)
            break;
         start = slash + 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
408
      }
409
      RichString_setAttrn(str, baseattr, start, finish);
Hisham Muhammad's avatar
Hisham Muhammad committed
410
411
412
   }
}

413
414
415
416
417
418
419
420
421
static inline void Process_outputRate(Process* this, RichString* str, int attr, char* buffer, int n, double rate) {
   rate = rate / 1024;
   if (rate < 0.01)
      snprintf(buffer, n, "    0 ");
   else if (rate <= 10)
      snprintf(buffer, n, "%5.2f ", rate);
   else if (rate <= 100)
      snprintf(buffer, n, "%5.1f ", rate);
   else {
422
      Process_humanNumber(this, str, rate);
423
424
425
426
427
      return;
   }
   RichString_append(str, attr, buffer);
}

428
static void Process_writeField(Process* this, RichString* str, ProcessField field) {
429
   char buffer[128]; buffer[127] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
430
   int attr = CRT_colors[DEFAULT_COLOR];
431
   int baseattr = CRT_colors[PROCESS_BASENAME];
432
   int n = sizeof(buffer) - 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
433
434

   switch (field) {
435
436
   case PID: snprintf(buffer, n, Process_pidFormat, this->pid); break;
   case PPID: snprintf(buffer, n, Process_pidFormat, this->ppid); break;
437
438
   case PGRP: snprintf(buffer, n, Process_pidFormat, this->pgrp); break;
   case SESSION: snprintf(buffer, n, Process_pidFormat, this->session); break;
439
   case TTY_NR: snprintf(buffer, n, "%5u ", this->tty_nr); break;
440
441
   case TGID: snprintf(buffer, n, Process_pidFormat, this->tgid); break;
   case TPGID: snprintf(buffer, n, Process_tpgidFormat, this->tpgid); break;
442
   case PROCESSOR: snprintf(buffer, n, "%3d ", ProcessList_cpuId(this->pl, this->processor)); break;
443
   case NLWP: snprintf(buffer, n, "%4ld ", this->nlwp); break;
Hisham Muhammad's avatar
Hisham Muhammad committed
444
   case COMM: {
Hisham Muhammad's avatar
Hisham Muhammad committed
445
      if (this->pl->highlightThreads && Process_isThread(this)) {
446
447
448
         attr = CRT_colors[PROCESS_THREAD];
         baseattr = CRT_colors[PROCESS_THREAD_BASENAME];
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
449
      if (!this->pl->treeView || this->indent == 0) {
450
         Process_writeCommand(this, attr, baseattr, str);
Hisham Muhammad's avatar
Hisham Muhammad committed
451
452
453
454
         return;
      } else {
         char* buf = buffer;
         int maxIndent = 0;
455
456
457
458
459
460
         const char **treeStr = this->pl->treeStr;
         bool lastItem = (this->indent < 0);
         int indent = (this->indent < 0 ? -this->indent : this->indent);
         if (treeStr == NULL)
             treeStr = ProcessList_treeStrAscii;

Hisham Muhammad's avatar
Hisham Muhammad committed
461
         for (int i = 0; i < 32; i++)
462
            if (indent & (1 << i))
Hisham Muhammad's avatar
Hisham Muhammad committed
463
               maxIndent = i+1;
464
          for (int i = 0; i < maxIndent - 1; i++) {
465
466
467
            int written;
            if (indent & (1 << i))
               written = snprintf(buf, n, "%s  ", treeStr[TREE_STR_VERT]);
Hisham Muhammad's avatar
Hisham Muhammad committed
468
            else
469
470
471
               written = snprintf(buf, n, "   ");
            buf += written;
            n -= written;
472
         }
473
474
         const char* draw = treeStr[lastItem ? (this->pl->direction == 1 ? TREE_STR_BEND : TREE_STR_TEND) : TREE_STR_RTEE];
         snprintf(buf, n, "%s%s ", draw, this->showChildren ? treeStr[TREE_STR_SHUT] : treeStr[TREE_STR_OPEN] );
Hisham Muhammad's avatar
Hisham Muhammad committed
475
         RichString_append(str, CRT_colors[PROCESS_TREE], buffer);
476
         Process_writeCommand(this, attr, baseattr, str);
477
         return;
Hisham Muhammad's avatar
Hisham Muhammad committed
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
      }
   }
   case STATE: {
      snprintf(buffer, n, "%c ", this->state);
      attr = this->state == 'R'
           ? CRT_colors[PROCESS_R_STATE]
           : attr;
      break;
   }
   case PRIORITY: {
      if(this->priority == -100)
         snprintf(buffer, n, " RT ");
      else
         snprintf(buffer, n, "%3ld ", this->priority);
      break;
   }
   case NICE: {
      snprintf(buffer, n, "%3ld ", this->nice);
      attr = this->nice < 0 ? CRT_colors[PROCESS_HIGH_PRIORITY]
           : this->nice > 0 ? CRT_colors[PROCESS_LOW_PRIORITY]
           : attr;
      break;
   }
501
502
503
504
505
506
507
   case M_DRS: Process_humanNumber(this, str, this->m_drs * PAGE_SIZE_KB); return;
   case M_DT: Process_humanNumber(this, str, this->m_dt * PAGE_SIZE_KB); return;
   case M_LRS: Process_humanNumber(this, str, this->m_lrs * PAGE_SIZE_KB); return;
   case M_TRS: Process_humanNumber(this, str, this->m_trs * PAGE_SIZE_KB); return;
   case M_SIZE: Process_humanNumber(this, str, this->m_size * PAGE_SIZE_KB); return;
   case M_RESIDENT: Process_humanNumber(this, str, this->m_resident * PAGE_SIZE_KB); return;
   case M_SHARE: Process_humanNumber(this, str, this->m_share * PAGE_SIZE_KB); return;
Hisham Muhammad's avatar
Hisham Muhammad committed
508
509
   case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break;
   case USER: {
Hisham Muhammad's avatar
Hisham Muhammad committed
510
      if (Process_getuid != (int) this->st_uid)
Hisham Muhammad's avatar
Hisham Muhammad committed
511
         attr = CRT_colors[PROCESS_SHADOW];
512
      if (this->user) {
513
         snprintf(buffer, n, "%-9s ", this->user);
514
      } else {
515
         snprintf(buffer, n, "%-9d ", this->st_uid);
516
      }
517
518
519
      if (buffer[9] != '\0') {
         buffer[9] = ' ';
         buffer[10] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
520
521
522
523
524
525
526
527
528
      }
      break;
   }
   case UTIME: Process_printTime(str, this->utime); return;
   case STIME: Process_printTime(str, this->stime); return;
   case CUTIME: Process_printTime(str, this->cutime); return;
   case CSTIME: Process_printTime(str, this->cstime); return;
   case TIME: Process_printTime(str, this->utime + this->stime); return;
   case PERCENT_CPU: {
529
530
531
532
      if (this->percent_cpu > 999.9) {
         snprintf(buffer, n, "%4d ", (unsigned int)this->percent_cpu); 
      } else if (this->percent_cpu > 99.9) {
         snprintf(buffer, n, "%3d. ", (unsigned int)this->percent_cpu); 
Hisham Muhammad's avatar
Hisham Muhammad committed
533
534
535
536
537
538
539
540
541
542
543
544
545
      } else {
         snprintf(buffer, n, "%4.1f ", this->percent_cpu);
      }
      break;
   }
   case PERCENT_MEM: {
      if (this->percent_mem > 99.9) {
         snprintf(buffer, n, "100. "); 
      } else {
         snprintf(buffer, n, "%4.1f ", this->percent_mem);
      }
      break;
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
546
   case STARTTIME: snprintf(buffer, n, "%s", this->starttime_show); break;
547
   #ifdef HAVE_OPENVZ
548
   case CTID: snprintf(buffer, n, "%5u ", this->ctid); break;
549
550
   case VPID: snprintf(buffer, n, "%5u ", this->vpid); break;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
551
552
553
   #ifdef HAVE_VSERVER
   case VXID: snprintf(buffer, n, "%5u ", this->vxid); break;
   #endif
554
   #ifdef HAVE_TASKSTATS
555
556
   case RCHAR:  snprintf(buffer, n, "%12llu ", this->io_rchar); break;
   case WCHAR:  snprintf(buffer, n, "%12llu ", this->io_wchar); break;   
557
558
   case SYSCR:  snprintf(buffer, n, "%10llu ", this->io_syscr); break;   
   case SYSCW:  snprintf(buffer, n, "%10llu ", this->io_syscw); break; 
559
560
   case RBYTES: Process_colorNumber(str, this->io_read_bytes); return;
   case WBYTES: Process_colorNumber(str, this->io_write_bytes); return;
561
   case CNCLWB: snprintf(buffer, n, "%10llu ", this->io_cancelled_write_bytes); break; 
562
563
564
   case IO_READ_RATE:  Process_outputRate(this, str, attr, buffer, n, this->io_rate_read_bps); return;
   case IO_WRITE_RATE: Process_outputRate(this, str, attr, buffer, n, this->io_rate_write_bps); return;
   case IO_RATE: Process_outputRate(this, str, attr, buffer, n, this->io_rate_read_bps + this->io_rate_write_bps); return;
565
   #endif
566
567
568
   #ifdef HAVE_CGROUP
   case CGROUP: snprintf(buffer, n, "%-10s ", this->cgroup); break;
   #endif
569
570
571
   #ifdef HAVE_OOM
   case OOM: snprintf(buffer, n, Process_pidFormat, this->oom); break;
   #endif
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
   case IO_PRIORITY: {
      int klass = IOPriority_class(this->ioPriority);
      if (klass == IOPRIO_CLASS_NONE) {
         // see note [1] above
         snprintf(buffer, n, "B%1d ", (int) (this->nice + 20) / 5);
      } else if (klass == IOPRIO_CLASS_BE) {
         snprintf(buffer, n, "B%1d ", IOPriority_data(this->ioPriority));
      } else if (klass == IOPRIO_CLASS_RT) {
         attr = CRT_colors[PROCESS_HIGH_PRIORITY];
         snprintf(buffer, n, "R%1d ", IOPriority_data(this->ioPriority));
      } else if (this->ioPriority == IOPriority_Idle) {
         attr = CRT_colors[PROCESS_LOW_PRIORITY]; 
         snprintf(buffer, n, "id ");
      } else {
         snprintf(buffer, n, "?? ");
      }
      break;
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
590
591
592
593
594
595
   default:
      snprintf(buffer, n, "- ");
   }
   RichString_append(str, attr, buffer);
}

596
597
598
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
   ProcessField* fields = this->pl->fields;
599
   RichString_prune(out);
600
601
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
602
   if (this->pl->shadowOtherUsers && (int)this->st_uid != Process_getuid)
603
604
605
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
606
   assert(out->chlen > 0);
607
608
609
610
611
612
}

void Process_delete(Object* cast) {
   Process* this = (Process*) cast;
   assert (this != NULL);
   if (this->comm) free(this->comm);
613
614
615
#ifdef HAVE_CGROUP
   if (this->cgroup) free(this->cgroup);
#endif
616
617
618
   free(this);
}

619
620
621
622
623
624
625
ObjectClass Process_class = {
   .extends = Class(Object),
   .display = Process_display,
   .delete = Process_delete,
   .compare = Process_compare
};

626
Process* Process_new(struct ProcessList_ *pl) {
627
   Process* this = calloc(1, sizeof(Process));
628
   Object_setClass(this, Class(Process));
629
630
631
   this->pid = 0;
   this->pl = pl;
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
632
   this->showChildren = true;
633
   this->show = true;
634
635
636
637
638
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
   this->indent = 0;
639
640
641
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
   if (Process_getuid == -1) Process_getuid = getuid();
   return this;
}

void Process_toggleTag(Process* this) {
   this->tag = this->tag == true ? false : true;
}

bool Process_setPriority(Process* this, int priority) {
   int old_prio = getpriority(PRIO_PROCESS, this->pid);
   int err = setpriority(PRIO_PROCESS, this->pid, priority);
   if (err == 0 && old_prio != getpriority(PRIO_PROCESS, this->pid)) {
      this->nice = priority;
   }
   return (err == 0);
}

659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
bool Process_changePriorityBy(Process* this, size_t delta) {
   return Process_setPriority(this, this->nice + delta);
}

IOPriority Process_updateIOPriority(Process* this) {
   IOPriority ioprio = syscall(SYS_ioprio_get, IOPRIO_WHO_PROCESS, this->pid);
   this->ioPriority = ioprio;
   return ioprio;
}

bool Process_setIOPriority(Process* this, IOPriority ioprio) {
   syscall(SYS_ioprio_set, IOPRIO_WHO_PROCESS, this->pid, ioprio);
   return (Process_updateIOPriority(this) == ioprio);
}

/*
[1] Note that before kernel 2.6.26 a process that has not asked for
an io priority formally uses "none" as scheduling class, but the
io scheduler will treat such processes as if it were in the best
effort class. The priority within the best effort class will  be
dynamically  derived  from  the  cpu  nice level of the process:
io_priority = (cpu_nice + 20) / 5. -- From ionice(1) man page
*/
#define Process_effectiveIOPriority(p_) (IOPriority_class(p_->ioPriority) == IOPRIO_CLASS_NONE ? IOPriority_tuple(IOPRIO_CLASS_BE, (p_->nice + 20) / 5) : p_->ioPriority)

684
#ifdef HAVE_LIBHWLOC
685

686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
Affinity* Process_getAffinity(Process* this) {
   hwloc_cpuset_t cpuset = hwloc_bitmap_alloc();
   bool ok = (hwloc_linux_get_tid_cpubind(this->pl->topology, this->pid, cpuset) == 0);
   Affinity* affinity = NULL;
   if (ok) {
      affinity = Affinity_new();
      if (hwloc_bitmap_last(cpuset) == -1) {
         for (int i = 0; i < this->pl->cpuCount; i++) {
            Affinity_add(affinity, i);
         }
      } else {
         unsigned int id;
         hwloc_bitmap_foreach_begin(id, cpuset);
            Affinity_add(affinity, id);
         hwloc_bitmap_foreach_end();
      }
   }
   hwloc_bitmap_free(cpuset);
   return affinity;
705
706
}

707
708
709
710
711
712
713
714
bool Process_setAffinity(Process* this, Affinity* affinity) {
   hwloc_cpuset_t cpuset = hwloc_bitmap_alloc();
   for (int i = 0; i < affinity->used; i++) {
      hwloc_bitmap_set(cpuset, affinity->cpus[i]);
   }
   bool ok = (hwloc_linux_set_tid_cpubind(this->pl->topology, this->pid, cpuset) == 0);
   hwloc_bitmap_free(cpuset);
   return ok;
715
}
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740

#elif HAVE_NATIVE_AFFINITY

Affinity* Process_getAffinity(Process* this) {
   cpu_set_t cpuset;
   bool ok = (sched_getaffinity(this->pid, sizeof(cpu_set_t), &cpuset) == 0);
   if (!ok) return NULL;
   Affinity* affinity = Affinity_new();
   for (int i = 0; i < this->pl->cpuCount; i++) {
      if (CPU_ISSET(i, &cpuset))
         Affinity_add(affinity, i);
   }
   return affinity;
}

bool Process_setAffinity(Process* this, Affinity* affinity) {
   cpu_set_t cpuset;
   CPU_ZERO(&cpuset);
   for (int i = 0; i < affinity->used; i++) {
      CPU_SET(affinity->cpus[i], &cpuset);
   }
   bool ok = (sched_setaffinity(this->pid, sizeof(unsigned long), &cpuset) == 0);
   return ok;
}

Hisham Muhammad's avatar
Hisham Muhammad committed
741
#endif
742

743
744
void Process_sendSignal(Process* this, size_t sgn) {
   kill(this->pid, (int) sgn);
745
746
}

747
int Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
748
749
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
750
751
752
753
754
755
756
757
758
759
760
761
762
   return (p1->pid - p2->pid);
}

int Process_compare(const void* v1, const void* v2) {
   Process *p1, *p2;
   ProcessList *pl = ((Process*)v1)->pl;
   if (pl->direction == 1) {
      p1 = (Process*)v1;
      p2 = (Process*)v2;
   } else {
      p2 = (Process*)v1;
      p1 = (Process*)v2;
   }
763
   long long diff;
764
   switch (pl->sortKey) {
Hisham Muhammad's avatar
Hisham Muhammad committed
765
   case PID:
766
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
767
   case PPID:
768
      return (p1->ppid - p2->ppid);
Hisham Muhammad's avatar
Hisham Muhammad committed
769
   case USER:
Hisham Muhammad's avatar
Hisham Muhammad committed
770
      return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : "");
Hisham Muhammad's avatar
Hisham Muhammad committed
771
   case PRIORITY:
772
      return (p1->priority - p2->priority);
773
774
775
776
   case PROCESSOR:
      return (p1->processor - p2->processor);
   case SESSION:
      return (p1->session - p2->session);
Hisham Muhammad's avatar
Hisham Muhammad committed
777
   case STATE:
778
      return (p1->state - p2->state);
Hisham Muhammad's avatar
Hisham Muhammad committed
779
   case NICE:
780
      return (p1->nice - p2->nice);
781
782
783
784
785
786
787
788
   case M_DRS:
      return (p2->m_drs - p1->m_drs);
   case M_DT:
      return (p2->m_dt - p1->m_dt);
   case M_LRS:
      return (p2->m_lrs - p1->m_lrs);
   case M_TRS:
      return (p2->m_trs - p1->m_trs);
Hisham Muhammad's avatar
Hisham Muhammad committed
789
   case M_SIZE:
790
      return (p2->m_size - p1->m_size);
Hisham Muhammad's avatar
Hisham Muhammad committed
791
   case M_RESIDENT:
792
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
793
   case M_SHARE:
794
      return (p2->m_share - p1->m_share);
Hisham Muhammad's avatar
Hisham Muhammad committed
795
   case PERCENT_CPU:
796
      return (p2->percent_cpu > p1->percent_cpu ? 1 : -1);
Hisham Muhammad's avatar
Hisham Muhammad committed
797
   case PERCENT_MEM:
798
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
799
   case UTIME:
800
      return (p2->utime - p1->utime);
Hisham Muhammad's avatar
Hisham Muhammad committed
801
   case STIME:
802
      return (p2->stime - p1->stime);
Hisham Muhammad's avatar
Hisham Muhammad committed
803
   case TIME:
804
      return ((p2->utime+p2->stime) - (p1->utime+p1->stime));
Hisham Muhammad's avatar
Hisham Muhammad committed
805
   case COMM:
806
      return strcmp(p1->comm, p2->comm);
807
808
   case NLWP:
      return (p1->nlwp - p2->nlwp);
809
810
811
812
813
814
   case STARTTIME: {
      if (p1->starttime_ctime == p2->starttime_ctime)
         return (p1->pid - p2->pid);
      else
         return (p1->starttime_ctime - p2->starttime_ctime);
   }
815
   #ifdef HAVE_OPENVZ
816
817
   case CTID:
      return (p1->ctid - p2->ctid);
818
819
820
   case VPID:
      return (p1->vpid - p2->vpid);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
821
822
823
824
   #ifdef HAVE_VSERVER
   case VXID:
      return (p1->vxid - p2->vxid);
   #endif
825
826
827
828
829
830
831
832
833
834
   #ifdef HAVE_TASKSTATS
   case RCHAR:  diff = p2->io_rchar - p1->io_rchar; goto test_diff;
   case WCHAR:  diff = p2->io_wchar - p1->io_wchar; goto test_diff;
   case SYSCR:  diff = p2->io_syscr - p1->io_syscr; goto test_diff;
   case SYSCW:  diff = p2->io_syscw - p1->io_syscw; goto test_diff;
   case RBYTES: diff = p2->io_read_bytes - p1->io_read_bytes; goto test_diff;
   case WBYTES: diff = p2->io_write_bytes - p1->io_write_bytes; goto test_diff;
   case CNCLWB: diff = p2->io_cancelled_write_bytes - p1->io_cancelled_write_bytes; goto test_diff;
   case IO_READ_RATE:  diff = p2->io_rate_read_bps - p1->io_rate_read_bps; goto test_diff;
   case IO_WRITE_RATE: diff = p2->io_rate_write_bps - p1->io_rate_write_bps; goto test_diff;
835
   case IO_RATE: diff = (p2->io_rate_read_bps + p2->io_rate_write_bps) - (p1->io_rate_read_bps + p1->io_rate_write_bps); goto test_diff;
836
   #endif
837
838
839
840
   #ifdef HAVE_CGROUP
   case CGROUP:
      return strcmp(p1->cgroup ? p1->cgroup : "", p2->cgroup ? p2->cgroup : "");
   #endif
841
842
843
844
   #ifdef HAVE_OOM
   case OOM:
      return (p1->oom - p2->oom);
   #endif
845
846
   case IO_PRIORITY:
      return Process_effectiveIOPriority(p1) - Process_effectiveIOPriority(p2);
Hisham Muhammad's avatar
Hisham Muhammad committed
847
   default:
848
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
849
   }
850
851
   test_diff:
   return (diff > 0) ? 1 : (diff < 0 ? -1 : 0);
Hisham Muhammad's avatar
Hisham Muhammad committed
852
}