Process.c 26.3 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
   IO_PRIORITY,
85
   LAST_PROCESSFIELD
Hisham Muhammad's avatar
Hisham Muhammad committed
86
87
88
89
90
91
92
93
94
95
} ProcessField;

struct ProcessList_;

typedef struct Process_ {
   Object super;

   struct ProcessList_ *pl;
   bool updated;

Hisham Muhammad's avatar
Hisham Muhammad committed
96
   pid_t pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
97
98
99
100
   char* comm;
   int indent;
   char state;
   bool tag;
Hisham Muhammad's avatar
Hisham Muhammad committed
101
   bool showChildren;
102
   bool show;
Hisham Muhammad's avatar
Hisham Muhammad committed
103
   pid_t ppid;
104
105
106
   unsigned int pgrp;
   unsigned int session;
   unsigned int tty_nr;
Hisham Muhammad's avatar
Hisham Muhammad committed
107
   pid_t tgid;
108
   int tpgid;
Hisham Muhammad's avatar
Hisham Muhammad committed
109
   unsigned long int flags;
110
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
111
112
113
114
   unsigned long int minflt;
   unsigned long int cminflt;
   unsigned long int majflt;
   unsigned long int cmajflt;
115
   #endif
116
117
118
119
   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
120
121
   long int priority;
   long int nice;
122
   long int nlwp;
123
   IOPriority ioPriority;
Hisham Muhammad's avatar
Hisham Muhammad committed
124
125
   char starttime_show[8];
   time_t starttime_ctime;
126
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
   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;
143
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
144
145
146
147
148
149
150
151
152
153
154
155
   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;
156
   char* user;
157
   #ifdef HAVE_OPENVZ
158
   unsigned int ctid;
159
160
   unsigned int vpid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
161
162
163
   #ifdef HAVE_VSERVER
   unsigned int vxid;
   #endif
164
165
166
167
168
169
170
171
172
173
174
175
176
   #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
177
178
179
   #ifdef HAVE_CGROUP
   char* cgroup;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
180
181
182
183
} Process;

}*/

Hisham Muhammad's avatar
Hisham Muhammad committed
184
const char *Process_fieldNames[] = {
185
186
187
188
189
190
191
192
   "", "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", 
193
#ifdef HAVE_OPENVZ
194
   "CTID", "VPID",
195
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
196
197
198
#ifdef HAVE_VSERVER
   "VXID",
#endif
199
#ifdef HAVE_TASKSTATS
200
201
   "RCHAR", "WCHAR", "SYSCR", "SYSCW", "RBYTES", "WBYTES", "CNCLWB",
   "IO_READ_RATE", "IO_WRITE_RATE", "IO_RATE",
202
#endif
203
204
205
#ifdef HAVE_CGROUP
   "CGROUP",
#endif
206
   "IO_PRIORITY",
207
"*** report bug! ***"
208
};
Hisham Muhammad's avatar
Hisham Muhammad committed
209

210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
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,
#endif
   PROCESS_FLAG_IOPRIO
};

Hisham Muhammad's avatar
Hisham Muhammad committed
235
const char *Process_fieldTitles[] = {
236
   "", "    PID ", "Command ", "S ", "   PPID ", "   PGRP ", "   SESN ",
237
   "  TTY ", "  TPGID ", "- ", "- ", "- ", "- ", "- ",
238
   " UTIME+  ", " STIME+  ", " CUTIME+ ", " CSTIME+ ", "PRI ", " NI ", "- ",
Hisham Muhammad's avatar
Hisham Muhammad committed
239
   "START ", "- ", "- ", "- ", "- ", "- ", "- ",
240
241
242
   "- ", "- ", "- ", "- ", "- ", "- ", "- ",
   "- ", "- ", "- ", "CPU ", " VIRT ", "  RES ", "  SHR ",
   " CODE ", " DATA ", " LIB ", " DIRTY ", " UID ", "CPU% ", "MEM% ",
243
   "USER      ", "  TIME+  ", "NLWP ", "   TGID ",
244
#ifdef HAVE_OPENVZ
245
   " CTID ", " VPID ",
246
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
247
248
249
#ifdef HAVE_VSERVER
   " VXID ",
#endif
250
#ifdef HAVE_TASKSTATS
251
   "     RD_CHAR ", "     WR_CHAR ", "   RD_SYSC ", "   WR_SYSC ", "  IO_RBYTES ", "  IO_WBYTES ", " IO_CANCEL ",
252
253
   " IORR ", " IOWR ", "   IO ",
#endif
254
255
256
#ifdef HAVE_CGROUP
   "    CGROUP ",
#endif
257
   "IO ",
258
"*** report bug! ***"
259
260
};

261
262
static int Process_getuid = -1;

263
264
265
266
267
268
269
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;
270
   fscanf(file, "%32d", &maxPid);
271
272
   fclose(file);
   if (maxPid > 99999) {
273
274
275
276
277
278
      Process_fieldTitles[PID] =     "    PID ";
      Process_fieldTitles[PPID] =    "   PPID ";
      Process_fieldTitles[TPGID] =   "  TPGID ";
      Process_fieldTitles[TGID] =    "   TGID ";
      Process_fieldTitles[PGRP] =    "   PGRP ";
      Process_fieldTitles[SESSION] = "   SESN ";
279
280
281
      Process_pidFormat = "%7u ";
      Process_tpgidFormat = "%7d ";
   } else {
282
283
284
285
286
287
      Process_fieldTitles[PID] =     "  PID ";
      Process_fieldTitles[PPID] =    " PPID ";
      Process_fieldTitles[TPGID] =   "TPGID ";
      Process_fieldTitles[TGID] =    " TGID ";
      Process_fieldTitles[PGRP] =    " PGRP ";
      Process_fieldTitles[SESSION] = " SESN ";
288
289
290
291
292
      Process_pidFormat = "%5u ";
      Process_tpgidFormat = "%5d ";
   }
}

Hisham Muhammad's avatar
Hisham Muhammad committed
293
294
295
296
#define ONE_K 1024
#define ONE_M (ONE_K * ONE_K)
#define ONE_G (ONE_M * ONE_K)

297
static void Process_humanNumber(Process* this, RichString* str, unsigned long number) {
Hisham Muhammad's avatar
Hisham Muhammad committed
298
299
   char buffer[11];
   int len;
Hisham Muhammad's avatar
Hisham Muhammad committed
300
   if(number >= (10 * ONE_M)) {
301
302
303
304
305
306
307
308
      if(number >= (100 * ONE_M)) {
         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
309
      len = snprintf(buffer, 10, "%4ldM ", number / ONE_K);
Hisham Muhammad's avatar
Hisham Muhammad committed
310
311
312
313
314
      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
315
      len = snprintf(buffer, 10, "%2ld", number/1000);
Hisham Muhammad's avatar
Hisham Muhammad committed
316
317
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer, len);
      number %= 1000;
Hisham Muhammad's avatar
Hisham Muhammad committed
318
      len = snprintf(buffer, 10, "%03ld ", number);
Hisham Muhammad's avatar
Hisham Muhammad committed
319
320
      RichString_appendn(str, CRT_colors[PROCESS], buffer, len);
   } else {
Hisham Muhammad's avatar
Hisham Muhammad committed
321
      len = snprintf(buffer, 10, "%5ld ", number);
Hisham Muhammad's avatar
Hisham Muhammad committed
322
323
324
325
      RichString_appendn(str, CRT_colors[PROCESS], buffer, len);
   }
}

326
static void Process_colorNumber(RichString* str, unsigned long long number) {
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
   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);
   }
}

342
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
343

344
static void Process_printTime(RichString* str, unsigned long long t) {
Hisham Muhammad's avatar
Hisham Muhammad committed
345
346
347
348
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;

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

351
   unsigned long long hours = iRealTime / 3600;
Hisham Muhammad's avatar
Hisham Muhammad committed
352
353
354
355
   int minutes = (iRealTime / 60) % 60;
   int seconds = iRealTime % 60;
   int hundredths = (realTime - iRealTime) * 100;
   char buffer[11];
356
357
   if (hours >= 100) {
      snprintf(buffer, 10, "%7lluh ", hours);
Hisham Muhammad's avatar
Hisham Muhammad committed
358
359
      RichString_append(str, CRT_colors[LARGE_NUMBER], buffer);
   } else {
360
361
362
363
364
365
366
367
      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
368
369
370
   }
}

371
static inline void Process_writeCommand(Process* this, int attr, int baseattr, RichString* str) {
372
   int start = RichString_size(str);
373
   RichString_append(str, attr, this->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
374
   if (this->pl->highlightBaseName) {
375
      int finish = RichString_size(str) - 1;
376
377
378
379
380
381
382
383
      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
384
      }
385
      RichString_setAttrn(str, baseattr, start, finish);
Hisham Muhammad's avatar
Hisham Muhammad committed
386
387
388
   }
}

389
390
391
392
393
394
395
396
397
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 {
398
      Process_humanNumber(this, str, rate);
399
400
401
402
403
      return;
   }
   RichString_append(str, attr, buffer);
}

404
static void Process_writeField(Process* this, RichString* str, ProcessField field) {
405
   char buffer[128]; buffer[127] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
406
   int attr = CRT_colors[DEFAULT_COLOR];
407
   int baseattr = CRT_colors[PROCESS_BASENAME];
408
   int n = sizeof(buffer) - 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
409
410

   switch (field) {
411
412
   case PID: snprintf(buffer, n, Process_pidFormat, this->pid); break;
   case PPID: snprintf(buffer, n, Process_pidFormat, this->ppid); break;
413
414
   case PGRP: snprintf(buffer, n, Process_pidFormat, this->pgrp); break;
   case SESSION: snprintf(buffer, n, Process_pidFormat, this->session); break;
415
   case TTY_NR: snprintf(buffer, n, "%5u ", this->tty_nr); break;
416
417
   case TGID: snprintf(buffer, n, Process_pidFormat, this->tgid); break;
   case TPGID: snprintf(buffer, n, Process_tpgidFormat, this->tpgid); break;
418
   case PROCESSOR: snprintf(buffer, n, "%3d ", ProcessList_cpuId(this->pl, this->processor)); break;
419
   case NLWP: snprintf(buffer, n, "%4ld ", this->nlwp); break;
Hisham Muhammad's avatar
Hisham Muhammad committed
420
   case COMM: {
Hisham Muhammad's avatar
Hisham Muhammad committed
421
      if (this->pl->highlightThreads && Process_isThread(this)) {
422
423
424
         attr = CRT_colors[PROCESS_THREAD];
         baseattr = CRT_colors[PROCESS_THREAD_BASENAME];
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
425
      if (!this->pl->treeView || this->indent == 0) {
426
         Process_writeCommand(this, attr, baseattr, str);
Hisham Muhammad's avatar
Hisham Muhammad committed
427
428
429
430
         return;
      } else {
         char* buf = buffer;
         int maxIndent = 0;
431
432
433
434
435
436
         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
437
         for (int i = 0; i < 32; i++)
438
            if (indent & (1 << i))
Hisham Muhammad's avatar
Hisham Muhammad committed
439
               maxIndent = i+1;
440
          for (int i = 0; i < maxIndent - 1; i++) {
441
442
443
            int written;
            if (indent & (1 << i))
               written = snprintf(buf, n, "%s  ", treeStr[TREE_STR_VERT]);
Hisham Muhammad's avatar
Hisham Muhammad committed
444
            else
445
446
447
               written = snprintf(buf, n, "   ");
            buf += written;
            n -= written;
448
         }
449
450
         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
451
         RichString_append(str, CRT_colors[PROCESS_TREE], buffer);
452
         Process_writeCommand(this, attr, baseattr, str);
453
         return;
Hisham Muhammad's avatar
Hisham Muhammad committed
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
      }
   }
   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;
   }
477
478
479
480
481
482
483
   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
484
485
   case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break;
   case USER: {
Hisham Muhammad's avatar
Hisham Muhammad committed
486
      if (Process_getuid != (int) this->st_uid)
Hisham Muhammad's avatar
Hisham Muhammad committed
487
         attr = CRT_colors[PROCESS_SHADOW];
488
      if (this->user) {
489
         snprintf(buffer, n, "%-9s ", this->user);
490
      } else {
491
         snprintf(buffer, n, "%-9d ", this->st_uid);
492
      }
493
494
495
      if (buffer[9] != '\0') {
         buffer[9] = ' ';
         buffer[10] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
496
497
498
499
500
501
502
503
504
      }
      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: {
505
506
507
508
      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
509
510
511
512
513
514
515
516
517
518
519
520
521
      } 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
522
   case STARTTIME: snprintf(buffer, n, "%s", this->starttime_show); break;
523
   #ifdef HAVE_OPENVZ
524
   case CTID: snprintf(buffer, n, "%5u ", this->ctid); break;
525
526
   case VPID: snprintf(buffer, n, "%5u ", this->vpid); break;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
527
528
529
   #ifdef HAVE_VSERVER
   case VXID: snprintf(buffer, n, "%5u ", this->vxid); break;
   #endif
530
   #ifdef HAVE_TASKSTATS
531
532
   case RCHAR:  snprintf(buffer, n, "%12llu ", this->io_rchar); break;
   case WCHAR:  snprintf(buffer, n, "%12llu ", this->io_wchar); break;   
533
534
   case SYSCR:  snprintf(buffer, n, "%10llu ", this->io_syscr); break;   
   case SYSCW:  snprintf(buffer, n, "%10llu ", this->io_syscw); break; 
535
536
   case RBYTES: Process_colorNumber(str, this->io_read_bytes); return;
   case WBYTES: Process_colorNumber(str, this->io_write_bytes); return;
537
   case CNCLWB: snprintf(buffer, n, "%10llu ", this->io_cancelled_write_bytes); break; 
538
539
540
   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;
541
   #endif
542
543
544
   #ifdef HAVE_CGROUP
   case CGROUP: snprintf(buffer, n, "%-10s ", this->cgroup); break;
   #endif
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
   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
563
564
565
566
567
568
   default:
      snprintf(buffer, n, "- ");
   }
   RichString_append(str, attr, buffer);
}

569
570
571
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
   ProcessField* fields = this->pl->fields;
572
   RichString_prune(out);
573
574
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
575
   if (this->pl->shadowOtherUsers && (int)this->st_uid != Process_getuid)
576
577
578
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
579
   assert(out->chlen > 0);
580
581
582
583
584
585
}

void Process_delete(Object* cast) {
   Process* this = (Process*) cast;
   assert (this != NULL);
   if (this->comm) free(this->comm);
586
587
588
#ifdef HAVE_CGROUP
   if (this->cgroup) free(this->cgroup);
#endif
589
590
591
   free(this);
}

592
593
594
595
596
597
598
ObjectClass Process_class = {
   .extends = Class(Object),
   .display = Process_display,
   .delete = Process_delete,
   .compare = Process_compare
};

599
600
Process* Process_new(struct ProcessList_ *pl) {
   Process* this = calloc(sizeof(Process), 1);
601
   Object_setClass(this, Class(Process));
602
603
604
   this->pid = 0;
   this->pl = pl;
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
605
   this->showChildren = true;
606
   this->show = true;
607
608
609
610
611
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
   this->indent = 0;
612
613
614
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
   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);
}

632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
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)

657
#ifdef HAVE_LIBHWLOC
658

659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
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;
678
679
}

680
681
682
683
684
685
686
687
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;
688
}
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713

#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
714
#endif
715

716
717
void Process_sendSignal(Process* this, size_t sgn) {
   kill(this->pid, (int) sgn);
718
719
}

720
int Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
721
722
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
723
724
725
726
727
728
729
730
731
732
733
734
735
   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;
   }
736
   long long diff;
737
   switch (pl->sortKey) {
Hisham Muhammad's avatar
Hisham Muhammad committed
738
   case PID:
739
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
740
   case PPID:
741
      return (p1->ppid - p2->ppid);
Hisham Muhammad's avatar
Hisham Muhammad committed
742
   case USER:
Hisham Muhammad's avatar
Hisham Muhammad committed
743
      return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : "");
Hisham Muhammad's avatar
Hisham Muhammad committed
744
   case PRIORITY:
745
      return (p1->priority - p2->priority);
746
747
748
749
   case PROCESSOR:
      return (p1->processor - p2->processor);
   case SESSION:
      return (p1->session - p2->session);
Hisham Muhammad's avatar
Hisham Muhammad committed
750
   case STATE:
751
      return (p1->state - p2->state);
Hisham Muhammad's avatar
Hisham Muhammad committed
752
   case NICE:
753
      return (p1->nice - p2->nice);
754
755
756
757
758
759
760
761
   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
762
   case M_SIZE:
763
      return (p2->m_size - p1->m_size);
Hisham Muhammad's avatar
Hisham Muhammad committed
764
   case M_RESIDENT:
765
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
766
   case M_SHARE:
767
      return (p2->m_share - p1->m_share);
Hisham Muhammad's avatar
Hisham Muhammad committed
768
   case PERCENT_CPU:
769
      return (p2->percent_cpu > p1->percent_cpu ? 1 : -1);
Hisham Muhammad's avatar
Hisham Muhammad committed
770
   case PERCENT_MEM:
771
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
772
   case UTIME:
773
      return (p2->utime - p1->utime);
Hisham Muhammad's avatar
Hisham Muhammad committed
774
   case STIME:
775
      return (p2->stime - p1->stime);
Hisham Muhammad's avatar
Hisham Muhammad committed
776
   case TIME:
777
      return ((p2->utime+p2->stime) - (p1->utime+p1->stime));
Hisham Muhammad's avatar
Hisham Muhammad committed
778
   case COMM:
779
      return strcmp(p1->comm, p2->comm);
780
781
   case NLWP:
      return (p1->nlwp - p2->nlwp);
782
783
784
785
786
787
   case STARTTIME: {
      if (p1->starttime_ctime == p2->starttime_ctime)
         return (p1->pid - p2->pid);
      else
         return (p1->starttime_ctime - p2->starttime_ctime);
   }
788
   #ifdef HAVE_OPENVZ
789
790
   case CTID:
      return (p1->ctid - p2->ctid);
791
792
793
   case VPID:
      return (p1->vpid - p2->vpid);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
794
795
796
797
   #ifdef HAVE_VSERVER
   case VXID:
      return (p1->vxid - p2->vxid);
   #endif
798
799
800
801
802
803
804
805
806
807
   #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;
808
   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;
809
   #endif
810
811
812
813
   #ifdef HAVE_CGROUP
   case CGROUP:
      return strcmp(p1->cgroup ? p1->cgroup : "", p2->cgroup ? p2->cgroup : "");
   #endif
814
815
   case IO_PRIORITY:
      return Process_effectiveIOPriority(p1) - Process_effectiveIOPriority(p2);
Hisham Muhammad's avatar
Hisham Muhammad committed
816
   default:
817
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
818
   }
819
820
   test_diff:
   return (diff > 0) ? 1 : (diff < 0 ? -1 : 0);
Hisham Muhammad's avatar
Hisham Muhammad committed
821
}