Process.c 27 KB
Newer Older
Hisham Muhammad's avatar
Hisham Muhammad committed
1
2
/*
htop - Process.c
3
(C) 2004-2014 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"
14
#include "Platform.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
15
16
17
18
19
20
21

#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
22
#include <stdlib.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
23
24
25
26
#include <signal.h>
#include <string.h>
#include <stdbool.h>
#include <pwd.h>
27
#include <sched.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
28
#include <time.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
29
#include <assert.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

35
// On Linux, this works only with glibc 2.1+. On earlier versions
Hisham Muhammad's avatar
Hisham Muhammad committed
36
// 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

Hisham Muhammad's avatar
Hisham Muhammad committed
46
#include <sys/types.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
47

48
49
typedef struct Settings_ Settings;

50
51
52
53
54
55
#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

56
57
58
59
60
61
62
63
#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
64
#ifndef Process_isThread
65
#define Process_isThread(_process) (Process_isUserlandThread(_process) || Process_isKernelThread(_process))
Hisham Muhammad's avatar
Hisham Muhammad committed
66
67
#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
68
69
70
71
72
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,
73
   USER, TIME, NLWP, TGID,
74
   #ifdef HAVE_OPENVZ
75
   CTID, VPID,
76
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
77
78
79
   #ifdef HAVE_VSERVER
   VXID,
   #endif
80
   #ifdef HAVE_TASKSTATS
81
   RCHAR, WCHAR, SYSCR, SYSCW, RBYTES, WBYTES, CNCLWB, IO_READ_RATE, IO_WRITE_RATE, IO_RATE,
82
   #endif
83
84
85
   #ifdef HAVE_CGROUP
   CGROUP,
   #endif
86
87
88
   #ifdef HAVE_OOM
   OOM,
   #endif
89
   IO_PRIORITY,
90
   LAST_PROCESSFIELD
Hisham Muhammad's avatar
Hisham Muhammad committed
91
92
93
94
95
96
97
98
99
} ProcessField;

struct ProcessList_;

typedef struct Process_ {
   Object super;

   struct ProcessList_ *pl;

Hisham Muhammad's avatar
Hisham Muhammad committed
100
   pid_t pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
101
102
103
104
   char* comm;
   int indent;
   char state;
   bool tag;
Hisham Muhammad's avatar
Hisham Muhammad committed
105
   bool showChildren;
106
   bool show;
Hisham Muhammad's avatar
Hisham Muhammad committed
107
   pid_t ppid;
108
109
110
   unsigned int pgrp;
   unsigned int session;
   unsigned int tty_nr;
Hisham Muhammad's avatar
Hisham Muhammad committed
111
   pid_t tgid;
112
   int tpgid;
Hisham Muhammad's avatar
Hisham Muhammad committed
113
   unsigned long int flags;
Hisham Muhammad's avatar
Hisham Muhammad committed
114
115
116
117
118
119

   uid_t st_uid;
   float percent_cpu;
   float percent_mem;
   char* user;

120
121
122
123
   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
124
125
   long int priority;
   long int nice;
126
   long int nlwp;
Hisham Muhammad's avatar
Hisham Muhammad committed
127
128
   char starttime_show[8];
   time_t starttime_ctime;
Hisham Muhammad's avatar
Hisham Muhammad committed
129
130
131
132
133
134
135
136
137
138
139
140
141

   #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;   
142
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
143

Hisham Muhammad's avatar
Hisham Muhammad committed
144
   int processor;
145
146
147
148
149
150
151
   long m_size;
   long m_resident;
   long m_share;
   long m_trs;
   long m_drs;
   long m_lrs;
   long m_dt;
Hisham Muhammad's avatar
Hisham Muhammad committed
152

153
   #ifdef HAVE_OPENVZ
154
   unsigned int ctid;
155
156
   unsigned int vpid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
157
158
159
   #ifdef HAVE_VSERVER
   unsigned int vxid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
160

161
162
163
   #ifdef HAVE_CGROUP
   char* cgroup;
   #endif
164
165
166
   #ifdef HAVE_OOM
   unsigned int oom;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
167
168

   int exit_signal;
169
   int basenameOffset;
Hisham Muhammad's avatar
Hisham Muhammad committed
170
171
172
173
174
175
   bool updated;

   unsigned long int minflt;
   unsigned long int cminflt;
   unsigned long int majflt;
   unsigned long int cmajflt;
176
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
   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;
   #endif

Hisham Muhammad's avatar
Hisham Muhammad committed
195
196
197
198
} Process;

}*/

Hisham Muhammad's avatar
Hisham Muhammad committed
199
const char *Process_fieldNames[] = {
200
201
202
203
204
205
206
207
   "", "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", 
208
#ifdef HAVE_OPENVZ
209
   "CTID", "VPID",
210
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
211
212
213
#ifdef HAVE_VSERVER
   "VXID",
#endif
214
#ifdef HAVE_TASKSTATS
215
216
   "RCHAR", "WCHAR", "SYSCR", "SYSCW", "RBYTES", "WBYTES", "CNCLWB",
   "IO_READ_RATE", "IO_WRITE_RATE", "IO_RATE",
217
#endif
218
219
#ifdef HAVE_CGROUP
   "CGROUP",
220
221
222
#endif
#ifdef HAVE_OOM
   "OOM",
223
#endif
224
   "IO_PRIORITY",
225
"*** report bug! ***"
226
};
Hisham Muhammad's avatar
Hisham Muhammad committed
227

228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
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,
249
250
251
#endif
#ifdef HAVE_OOM
   0,
252
253
254
255
#endif
   PROCESS_FLAG_IOPRIO
};

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

285
286
static int Process_getuid = -1;

287
288
289
static char* Process_pidFormat = "%7u ";
static char* Process_tpgidFormat = "%7u ";

290
291
292
void Process_setupColumnWidths() {
   int maxPid = Platform_getMaxPid();
   if (maxPid == -1) return;
293
   if (maxPid > 99999) {
294
295
      Process_fieldTitles[PID] =     "    PID ";
      Process_fieldTitles[PPID] =    "   PPID ";
296
297
298
      #ifdef HAVE_OPENVZ
      Process_fieldTitles[VPID] =    "   VPID ";
      #endif
299
300
301
302
      Process_fieldTitles[TPGID] =   "  TPGID ";
      Process_fieldTitles[TGID] =    "   TGID ";
      Process_fieldTitles[PGRP] =    "   PGRP ";
      Process_fieldTitles[SESSION] = "   SESN ";
303
304
305
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "    OOM ";
      #endif
306
307
308
      Process_pidFormat = "%7u ";
      Process_tpgidFormat = "%7d ";
   } else {
309
310
      Process_fieldTitles[PID] =     "  PID ";
      Process_fieldTitles[PPID] =    " PPID ";
311
312
313
      #ifdef HAVE_OPENVZ
      Process_fieldTitles[VPID] =    " VPID ";
      #endif
314
315
316
317
      Process_fieldTitles[TPGID] =   "TPGID ";
      Process_fieldTitles[TGID] =    " TGID ";
      Process_fieldTitles[PGRP] =    " PGRP ";
      Process_fieldTitles[SESSION] = " SESN ";
318
319
320
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "  OOM ";
      #endif
321
322
323
324
325
      Process_pidFormat = "%5u ";
      Process_tpgidFormat = "%5d ";
   }
}

326
#define ONE_K 1024L
Hisham Muhammad's avatar
Hisham Muhammad committed
327
328
329
#define ONE_M (ONE_K * ONE_K)
#define ONE_G (ONE_M * ONE_K)

330
#define ONE_DECIMAL_K 1000L
331
#define ONE_DECIMAL_M (ONE_DECIMAL_K * ONE_DECIMAL_K)
332
#define ONE_DECIMAL_G (ONE_DECIMAL_M * ONE_DECIMAL_K)
333

334
static void Process_humanNumber(RichString* str, unsigned long number, bool coloring) {
Hisham Muhammad's avatar
Hisham Muhammad committed
335
336
   char buffer[11];
   int len;
337
338
339
340
341
342
343
344
   
   int largeNumberColor = CRT_colors[LARGE_NUMBER];
   int processMegabytesColor = CRT_colors[PROCESS_MEGABYTES];
   int processColor = CRT_colors[PROCESS];
   if (!coloring) {
      largeNumberColor = CRT_colors[PROCESS];
      processMegabytesColor = CRT_colors[PROCESS];
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
345
 
346
   if(number >= (10 * ONE_DECIMAL_M)) {
347
348
349
      #ifdef __LP64__
      if(number >= (100 * ONE_DECIMAL_G)) {
         len = snprintf(buffer, 10, "%4ldT ", number / ONE_G);
350
         RichString_appendn(str, largeNumberColor, buffer, len);
351
         return;
352
      } else if (number >= (1000 * ONE_DECIMAL_M)) {
353
         len = snprintf(buffer, 10, "%4.1lfT ", (double)number / ONE_G);
354
         RichString_appendn(str, largeNumberColor, buffer, len);
355
356
         return;
      }
357
      #endif
358
      if(number >= (100 * ONE_DECIMAL_M)) {
359
         len = snprintf(buffer, 10, "%4ldG ", number / ONE_M);
360
         RichString_appendn(str, largeNumberColor, buffer, len);
361
         return;
362
      }
363
      len = snprintf(buffer, 10, "%4.1lfG ", (double)number / ONE_M);
364
      RichString_appendn(str, largeNumberColor, buffer, len);
365
      return;
366
   } else if (number >= 100000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
367
      len = snprintf(buffer, 10, "%4ldM ", number / ONE_K);
368
      RichString_appendn(str, processMegabytesColor, buffer, len);
369
      return;
370
   } else if (number >= 1000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
371
      len = snprintf(buffer, 10, "%2ld", number/1000);
372
      RichString_appendn(str, processMegabytesColor, buffer, len);
Hisham Muhammad's avatar
Hisham Muhammad committed
373
      number %= 1000;
374
      len = snprintf(buffer, 10, "%03lu ", number);
375
      RichString_appendn(str, processColor, buffer, len);
376
      return;
Hisham Muhammad's avatar
Hisham Muhammad committed
377
   }
378
   len = snprintf(buffer, 10, "%5lu ", number);
379
   RichString_appendn(str, processColor, buffer, len);
Hisham Muhammad's avatar
Hisham Muhammad committed
380
381
}

382
static void Process_colorNumber(RichString* str, unsigned long long number, bool coloring) {
383
   char buffer[14];
384
385
386
387
388
389
390
391
392
393
394

   int largeNumberColor = CRT_colors[LARGE_NUMBER];
   int processMegabytesColor = CRT_colors[PROCESS_MEGABYTES];
   int processColor = CRT_colors[PROCESS];
   int processShadowColor = CRT_colors[PROCESS_SHADOW];
   if (!coloring) {
      largeNumberColor = CRT_colors[PROCESS];
      processMegabytesColor = CRT_colors[PROCESS];
      processShadowColor = CRT_colors[PROCESS];
   }

395
396
   if (number > 10000000000) {
      snprintf(buffer, 13, "%11lld ", number / 1000);
397
398
399
      RichString_appendn(str, largeNumberColor, buffer, 5);
      RichString_appendn(str, processMegabytesColor, buffer+5, 3);
      RichString_appendn(str, processColor, buffer+8, 4);
400
   } else {
401
      snprintf(buffer, 13, "%11llu ", number);
402
403
404
405
      RichString_appendn(str, largeNumberColor, buffer, 2);
      RichString_appendn(str, processMegabytesColor, buffer+2, 3);
      RichString_appendn(str, processColor, buffer+5, 3);
      RichString_appendn(str, processShadowColor, buffer+8, 4);
406
407
408
   }
}

409
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
410

411
static void Process_printTime(RichString* str, unsigned long long t) {
Hisham Muhammad's avatar
Hisham Muhammad committed
412
413
414
415
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;

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

418
   unsigned long long hours = iRealTime / 3600;
Hisham Muhammad's avatar
Hisham Muhammad committed
419
420
421
422
   int minutes = (iRealTime / 60) % 60;
   int seconds = iRealTime % 60;
   int hundredths = (realTime - iRealTime) * 100;
   char buffer[11];
423
424
   if (hours >= 100) {
      snprintf(buffer, 10, "%7lluh ", hours);
Hisham Muhammad's avatar
Hisham Muhammad committed
425
426
      RichString_append(str, CRT_colors[LARGE_NUMBER], buffer);
   } else {
427
428
429
430
431
432
433
434
      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
435
436
437
   }
}

438
static inline void Process_writeCommand(Process* this, int attr, int baseattr, RichString* str) {
439
   int start = RichString_size(str);
440
   RichString_append(str, attr, this->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
441
   if (this->pl->highlightBaseName) {
442
      int finish = RichString_size(str) - 1;
443
444
      if (this->basenameOffset != -1)
         finish = (start + this->basenameOffset) - 1;
445
446
447
448
      int colon = RichString_findChar(str, ':', start);
      if (colon != -1 && colon < finish) {
         finish = colon;
      } else {
449
         for (int i = finish - start; i >= 0; i--) {
450
451
452
453
454
            if (this->comm[i] == '/') {
               start += i+1;
               break;
            }
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
455
      }
456
      RichString_setAttrn(str, baseattr, start, finish);
Hisham Muhammad's avatar
Hisham Muhammad committed
457
458
459
   }
}

460
static inline void Process_outputRate(RichString* str, int attr, char* buffer, int n, double rate, int coloring) {
461
462
463
464
465
466
467
468
   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 {
469
      Process_humanNumber(str, rate, coloring);
470
471
472
473
474
      return;
   }
   RichString_append(str, attr, buffer);
}

475
static void Process_writeField(Process* this, RichString* str, ProcessField field) {
476
   char buffer[256]; buffer[255] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
477
   int attr = CRT_colors[DEFAULT_COLOR];
478
   int baseattr = CRT_colors[PROCESS_BASENAME];
479
   int n = sizeof(buffer) - 1;
480
   bool coloring = this->pl->highlightMegabytes;
Hisham Muhammad's avatar
Hisham Muhammad committed
481
482

   switch (field) {
483
484
   case PID: snprintf(buffer, n, Process_pidFormat, this->pid); break;
   case PPID: snprintf(buffer, n, Process_pidFormat, this->ppid); break;
485
486
   case PGRP: snprintf(buffer, n, Process_pidFormat, this->pgrp); break;
   case SESSION: snprintf(buffer, n, Process_pidFormat, this->session); break;
487
   case TTY_NR: snprintf(buffer, n, "%5u ", this->tty_nr); break;
488
489
   case TGID: snprintf(buffer, n, Process_pidFormat, this->tgid); break;
   case TPGID: snprintf(buffer, n, Process_tpgidFormat, this->tpgid); break;
490
491
492
493
   case MINFLT: Process_colorNumber(str, this->minflt, coloring); return;
   case CMINFLT: Process_colorNumber(str, this->cminflt, coloring); return;
   case MAJFLT: Process_colorNumber(str, this->majflt, coloring); return;
   case CMAJFLT: Process_colorNumber(str, this->cmajflt, coloring); return;
494
   case PROCESSOR: snprintf(buffer, n, "%3d ", ProcessList_cpuId(this->pl, this->processor)); break;
495
   case NLWP: snprintf(buffer, n, "%4ld ", this->nlwp); break;
Hisham Muhammad's avatar
Hisham Muhammad committed
496
   case COMM: {
Hisham Muhammad's avatar
Hisham Muhammad committed
497
      if (this->pl->highlightThreads && Process_isThread(this)) {
498
499
500
         attr = CRT_colors[PROCESS_THREAD];
         baseattr = CRT_colors[PROCESS_THREAD_BASENAME];
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
501
      if (!this->pl->treeView || this->indent == 0) {
502
         Process_writeCommand(this, attr, baseattr, str);
Hisham Muhammad's avatar
Hisham Muhammad committed
503
504
505
506
         return;
      } else {
         char* buf = buffer;
         int maxIndent = 0;
507
508
509
510
511
512
         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
513
         for (int i = 0; i < 32; i++)
514
            if (indent & (1 << i))
Hisham Muhammad's avatar
Hisham Muhammad committed
515
               maxIndent = i+1;
516
          for (int i = 0; i < maxIndent - 1; i++) {
517
518
519
            int written;
            if (indent & (1 << i))
               written = snprintf(buf, n, "%s  ", treeStr[TREE_STR_VERT]);
Hisham Muhammad's avatar
Hisham Muhammad committed
520
            else
521
522
523
               written = snprintf(buf, n, "   ");
            buf += written;
            n -= written;
524
         }
525
526
         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
527
         RichString_append(str, CRT_colors[PROCESS_TREE], buffer);
528
         Process_writeCommand(this, attr, baseattr, str);
529
         return;
Hisham Muhammad's avatar
Hisham Muhammad committed
530
531
532
533
      }
   }
   case STATE: {
      snprintf(buffer, n, "%c ", this->state);
534
535
536
537
538
539
540
541
      switch(this->state) {
          case 'R':
              attr = CRT_colors[PROCESS_R_STATE];
              break;
          case 'D':
              attr = CRT_colors[PROCESS_D_STATE];
              break;
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
      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;
   }
558
559
560
561
562
563
564
   case M_DRS: Process_humanNumber(str, this->m_drs * PAGE_SIZE_KB, coloring); return;
   case M_DT: Process_humanNumber(str, this->m_dt * PAGE_SIZE_KB, coloring); return;
   case M_LRS: Process_humanNumber(str, this->m_lrs * PAGE_SIZE_KB, coloring); return;
   case M_TRS: Process_humanNumber(str, this->m_trs * PAGE_SIZE_KB, coloring); return;
   case M_SIZE: Process_humanNumber(str, this->m_size * PAGE_SIZE_KB, coloring); return;
   case M_RESIDENT: Process_humanNumber(str, this->m_resident * PAGE_SIZE_KB, coloring); return;
   case M_SHARE: Process_humanNumber(str, this->m_share * PAGE_SIZE_KB, coloring); return;
Hisham Muhammad's avatar
Hisham Muhammad committed
565
566
   case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break;
   case USER: {
Hisham Muhammad's avatar
Hisham Muhammad committed
567
      if (Process_getuid != (int) this->st_uid)
Hisham Muhammad's avatar
Hisham Muhammad committed
568
         attr = CRT_colors[PROCESS_SHADOW];
569
      if (this->user) {
570
         snprintf(buffer, n, "%-9s ", this->user);
571
      } else {
572
         snprintf(buffer, n, "%-9d ", this->st_uid);
573
      }
574
575
576
      if (buffer[9] != '\0') {
         buffer[9] = ' ';
         buffer[10] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
577
578
579
580
581
582
583
584
585
      }
      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: {
586
587
588
589
      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
590
591
592
593
594
595
596
597
598
599
600
601
602
      } 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
603
   case STARTTIME: snprintf(buffer, n, "%s", this->starttime_show); break;
604
   #ifdef HAVE_OPENVZ
605
606
   case CTID: snprintf(buffer, n, "%7u ", this->ctid); break;
   case VPID: snprintf(buffer, n, Process_pidFormat, this->vpid); break;
607
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
608
609
610
   #ifdef HAVE_VSERVER
   case VXID: snprintf(buffer, n, "%5u ", this->vxid); break;
   #endif
611
   #ifdef HAVE_TASKSTATS
612
613
614
615
616
617
618
619
620
621
   case RCHAR:  Process_colorNumber(str, this->io_rchar, coloring); return;
   case WCHAR:  Process_colorNumber(str, this->io_wchar, coloring); return;
   case SYSCR:  Process_colorNumber(str, this->io_syscr, coloring); return;
   case SYSCW:  Process_colorNumber(str, this->io_syscw, coloring); return;
   case RBYTES: Process_colorNumber(str, this->io_read_bytes, coloring); return;
   case WBYTES: Process_colorNumber(str, this->io_write_bytes, coloring); return;
   case CNCLWB: Process_colorNumber(str, this->io_cancelled_write_bytes, coloring); return;
   case IO_READ_RATE:  Process_outputRate(str, attr, buffer, n, this->io_rate_read_bps, coloring); return;
   case IO_WRITE_RATE: Process_outputRate(str, attr, buffer, n, this->io_rate_write_bps, coloring); return;
   case IO_RATE: Process_outputRate(str, attr, buffer, n, this->io_rate_read_bps + this->io_rate_write_bps, coloring); return;
622
   #endif
623
624
625
   #ifdef HAVE_CGROUP
   case CGROUP: snprintf(buffer, n, "%-10s ", this->cgroup); break;
   #endif
626
627
628
   #ifdef HAVE_OOM
   case OOM: snprintf(buffer, n, Process_pidFormat, this->oom); break;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
629
630
631
632
633
634
   default:
      snprintf(buffer, n, "- ");
   }
   RichString_append(str, attr, buffer);
}

635
636
637
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
   ProcessField* fields = this->pl->fields;
638
   RichString_prune(out);
639
640
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
641
   if (this->pl->shadowOtherUsers && (int)this->st_uid != Process_getuid)
642
643
644
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
645
   assert(out->chlen > 0);
646
647
}

648
void Process_done(Process* this) {
649
   assert (this != NULL);
650
   free(this->comm);
651
#ifdef HAVE_CGROUP
652
   free(this->cgroup);
653
#endif
654
655
}

656
657
658
659
660
661
662
ObjectClass Process_class = {
   .extends = Class(Object),
   .display = Process_display,
   .delete = Process_delete,
   .compare = Process_compare
};

663
void Process_init(Process* this, struct Settings_* settings, struct ProcessList_* pl) {
664
665
666
   this->pid = 0;
   this->pl = pl;
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
667
   this->showChildren = true;
668
   this->show = true;
669
670
671
672
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
673
   this->basenameOffset = -1;
674
   this->indent = 0;
675
676
677
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
   if (Process_getuid == -1) Process_getuid = getuid();
}

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

694
695
696
697
bool Process_changePriorityBy(Process* this, size_t delta) {
   return Process_setPriority(this, this->nice + delta);
}

698
#ifdef HAVE_LIBHWLOC
699

700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
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;
719
720
}

721
722
723
724
725
726
727
728
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;
729
}
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754

#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
755
#endif
756

757
758
void Process_sendSignal(Process* this, size_t sgn) {
   kill(this->pid, (int) sgn);
759
760
}

761
long Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
762
763
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
764
765
766
   return (p1->pid - p2->pid);
}

767
long Process_compare(const void* v1, const void* v2) {
768
769
770
771
772
773
774
775
776
   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;
   }
777
   long long diff;
778
   switch (pl->sortKey) {
Hisham Muhammad's avatar
Hisham Muhammad committed
779
   case PID:
780
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
781
   case PPID:
782
      return (p1->ppid - p2->ppid);
Hisham Muhammad's avatar
Hisham Muhammad committed
783
   case USER:
Hisham Muhammad's avatar
Hisham Muhammad committed
784
      return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : "");
Hisham Muhammad's avatar
Hisham Muhammad committed
785
   case PRIORITY:
786
      return (p1->priority - p2->priority);
787
788
789
790
   case PROCESSOR:
      return (p1->processor - p2->processor);
   case SESSION:
      return (p1->session - p2->session);
Hisham Muhammad's avatar
Hisham Muhammad committed
791
   case STATE:
792
      return (p1->state - p2->state);
Hisham Muhammad's avatar
Hisham Muhammad committed
793
   case NICE:
794
      return (p1->nice - p2->nice);
795
796
797
798
799
800
801
802
   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
803
   case M_SIZE:
804
      return (p2->m_size - p1->m_size);
Hisham Muhammad's avatar
Hisham Muhammad committed
805
   case M_RESIDENT:
806
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
807
   case M_SHARE:
808
      return (p2->m_share - p1->m_share);
Hisham Muhammad's avatar
Hisham Muhammad committed
809
   case PERCENT_CPU:
810
      return (p2->percent_cpu > p1->percent_cpu ? 1 : -1);
Hisham Muhammad's avatar
Hisham Muhammad committed
811
   case PERCENT_MEM:
812
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
813
814
815
816
   case UTIME:  diff = p2->utime - p1->utime; goto test_diff;
   case CUTIME: diff = p2->cutime - p1->cutime; goto test_diff;
   case STIME:  diff = p2->stime - p1->stime; goto test_diff;
   case CSTIME: diff = p2->cstime - p2->cstime; goto test_diff;
Hisham Muhammad's avatar
Hisham Muhammad committed
817
   case TIME:
818
      return ((p2->utime+p2->stime) - (p1->utime+p1->stime));
Hisham Muhammad's avatar
Hisham Muhammad committed
819
   case COMM:
820
      return strcmp(p1->comm, p2->comm);
821
822
   case NLWP:
      return (p1->nlwp - p2->nlwp);
823
824
825
826
827
828
   case STARTTIME: {
      if (p1->starttime_ctime == p2->starttime_ctime)
         return (p1->pid - p2->pid);
      else
         return (p1->starttime_ctime - p2->starttime_ctime);
   }
829
   #ifdef HAVE_OPENVZ
830
831
   case CTID:
      return (p1->ctid - p2->ctid);
832
833
834
   case VPID:
      return (p1->vpid - p2->vpid);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
835
836
837
838
   #ifdef HAVE_VSERVER
   case VXID:
      return (p1->vxid - p2->vxid);
   #endif
839
840
841
842
843
844
845
846
847
848
   #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;
849
   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;
850
   #endif
851
852
853
854
   #ifdef HAVE_CGROUP
   case CGROUP:
      return strcmp(p1->cgroup ? p1->cgroup : "", p2->cgroup ? p2->cgroup : "");
   #endif
855
856
857
858
   #ifdef HAVE_OOM
   case OOM:
      return (p1->oom - p2->oom);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
859
   default:
860
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
861
   }
862
863
   test_diff:
   return (diff > 0) ? 1 : (diff < 0 ? -1 : 0);
Hisham Muhammad's avatar
Hisham Muhammad committed
864
}