Process.c 28.5 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
} ProcessField;

struct ProcessList_;

typedef struct Process_ {
   Object super;

   struct ProcessList_ *pl;

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

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

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

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

Hisham Muhammad's avatar
Hisham Muhammad committed
143
   int processor;
144
145
146
147
148
149
150
   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
151

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

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

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

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

}*/

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

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

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

284
285
static int Process_getuid = -1;

286
287
288
289
290
291
292
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;
293
   fscanf(file, "%32d", &maxPid);
294
295
   fclose(file);
   if (maxPid > 99999) {
296
297
298
299
300
301
      Process_fieldTitles[PID] =     "    PID ";
      Process_fieldTitles[PPID] =    "   PPID ";
      Process_fieldTitles[TPGID] =   "  TPGID ";
      Process_fieldTitles[TGID] =    "   TGID ";
      Process_fieldTitles[PGRP] =    "   PGRP ";
      Process_fieldTitles[SESSION] = "   SESN ";
302
303
304
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "    OOM ";
      #endif
305
306
307
      Process_pidFormat = "%7u ";
      Process_tpgidFormat = "%7d ";
   } else {
308
309
310
311
312
313
      Process_fieldTitles[PID] =     "  PID ";
      Process_fieldTitles[PPID] =    " PPID ";
      Process_fieldTitles[TPGID] =   "TPGID ";
      Process_fieldTitles[TGID] =    " TGID ";
      Process_fieldTitles[PGRP] =    " PGRP ";
      Process_fieldTitles[SESSION] = " SESN ";
314
315
316
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "  OOM ";
      #endif
317
318
319
320
321
      Process_pidFormat = "%5u ";
      Process_tpgidFormat = "%5d ";
   }
}

322
#define ONE_K 1024L
Hisham Muhammad's avatar
Hisham Muhammad committed
323
324
325
#define ONE_M (ONE_K * ONE_K)
#define ONE_G (ONE_M * ONE_K)

326
#define ONE_DECIMAL_K 1000L
327
#define ONE_DECIMAL_M (ONE_DECIMAL_K * ONE_DECIMAL_K)
328
#define ONE_DECIMAL_G (ONE_DECIMAL_M * ONE_DECIMAL_K)
329

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

378
static void Process_colorNumber(RichString* str, unsigned long long number, bool coloring) {
379
   char buffer[14];
380
381
382
383
384
385
386
387
388
389
390

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

391
392
   if (number > 10000000000) {
      snprintf(buffer, 13, "%11lld ", number / 1000);
393
394
395
      RichString_appendn(str, largeNumberColor, buffer, 5);
      RichString_appendn(str, processMegabytesColor, buffer+5, 3);
      RichString_appendn(str, processColor, buffer+8, 4);
396
   } else {
397
      snprintf(buffer, 13, "%11llu ", number);
398
399
400
401
      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);
402
403
404
   }
}

405
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
406

407
static void Process_printTime(RichString* str, unsigned long long t) {
Hisham Muhammad's avatar
Hisham Muhammad committed
408
409
410
411
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;

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

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

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

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

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

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

644
645
646
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
   ProcessField* fields = this->pl->fields;
647
   RichString_prune(out);
648
649
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
650
   if (this->pl->shadowOtherUsers && (int)this->st_uid != Process_getuid)
651
652
653
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
654
   assert(out->chlen > 0);
655
656
657
658
659
660
}

void Process_delete(Object* cast) {
   Process* this = (Process*) cast;
   assert (this != NULL);
   if (this->comm) free(this->comm);
661
662
663
#ifdef HAVE_CGROUP
   if (this->cgroup) free(this->cgroup);
#endif
664
665
666
   free(this);
}

667
668
669
670
671
672
673
ObjectClass Process_class = {
   .extends = Class(Object),
   .display = Process_display,
   .delete = Process_delete,
   .compare = Process_compare
};

674
Process* Process_new(struct ProcessList_ *pl) {
675
   Process* this = calloc(1, sizeof(Process));
676
   Object_setClass(this, Class(Process));
677
678
679
   this->pid = 0;
   this->pl = pl;
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
680
   this->showChildren = true;
681
   this->show = true;
682
683
684
685
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
686
   this->basenameOffset = -1;
687
   this->indent = 0;
688
689
690
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
   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);
}

708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
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)

733
#ifdef HAVE_LIBHWLOC
734

735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
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;
754
755
}

756
757
758
759
760
761
762
763
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;
764
}
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789

#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
790
#endif
791

792
793
void Process_sendSignal(Process* this, size_t sgn) {
   kill(this->pid, (int) sgn);
794
795
}

796
int Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
797
798
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
799
800
801
802
803
804
805
806
807
808
809
810
811
   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;
   }
812
   long long diff;
813
   switch (pl->sortKey) {
Hisham Muhammad's avatar
Hisham Muhammad committed
814
   case PID:
815
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
816
   case PPID:
817
      return (p1->ppid - p2->ppid);
Hisham Muhammad's avatar
Hisham Muhammad committed
818
   case USER:
Hisham Muhammad's avatar
Hisham Muhammad committed
819
      return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : "");
Hisham Muhammad's avatar
Hisham Muhammad committed
820
   case PRIORITY:
821
      return (p1->priority - p2->priority);
822
823
824
825
   case PROCESSOR:
      return (p1->processor - p2->processor);
   case SESSION:
      return (p1->session - p2->session);
Hisham Muhammad's avatar
Hisham Muhammad committed
826
   case STATE:
827
      return (p1->state - p2->state);
Hisham Muhammad's avatar
Hisham Muhammad committed
828
   case NICE:
829
      return (p1->nice - p2->nice);
830
831
832
833
834
835
836
837
   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
838
   case M_SIZE:
839
      return (p2->m_size - p1->m_size);
Hisham Muhammad's avatar
Hisham Muhammad committed
840
   case M_RESIDENT:
841
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
842
   case M_SHARE:
843
      return (p2->m_share - p1->m_share);
Hisham Muhammad's avatar
Hisham Muhammad committed
844
   case PERCENT_CPU:
845
      return (p2->percent_cpu > p1->percent_cpu ? 1 : -1);
Hisham Muhammad's avatar
Hisham Muhammad committed
846
   case PERCENT_MEM:
847
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
848
   case UTIME:
849
      return (p2->utime - p1->utime);
Hisham Muhammad's avatar
Hisham Muhammad committed
850
   case STIME:
851
      return (p2->stime - p1->stime);
Hisham Muhammad's avatar
Hisham Muhammad committed
852
   case TIME:
853
      return ((p2->utime+p2->stime) - (p1->utime+p1->stime));
Hisham Muhammad's avatar
Hisham Muhammad committed
854
   case COMM:
855
      return strcmp(p1->comm, p2->comm);
856
857
   case NLWP:
      return (p1->nlwp - p2->nlwp);
858
859
860
861
862
863
   case STARTTIME: {
      if (p1->starttime_ctime == p2->starttime_ctime)
         return (p1->pid - p2->pid);
      else
         return (p1->starttime_ctime - p2->starttime_ctime);
   }
864
   #ifdef HAVE_OPENVZ
865
866
   case CTID:
      return (p1->ctid - p2->ctid);
867
868
869
   case VPID:
      return (p1->vpid - p2->vpid);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
870
871
872
873
   #ifdef HAVE_VSERVER
   case VXID:
      return (p1->vxid - p2->vxid);
   #endif
874
875
876
877
878
879
880
881
882
883
   #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;
884
   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;
885
   #endif
886
887
888
889
   #ifdef HAVE_CGROUP
   case CGROUP:
      return strcmp(p1->cgroup ? p1->cgroup : "", p2->cgroup ? p2->cgroup : "");
   #endif
890
891
892
893
   #ifdef HAVE_OOM
   case OOM:
      return (p1->oom - p2->oom);
   #endif
894
895
   case IO_PRIORITY:
      return Process_effectiveIOPriority(p1) - Process_effectiveIOPriority(p2);
Hisham Muhammad's avatar
Hisham Muhammad committed
896
   default:
897
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
898
   }
899
900
   test_diff:
   return (diff > 0) ? 1 : (diff < 0 ? -1 : 0);
Hisham Muhammad's avatar
Hisham Muhammad committed
901
}