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

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

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

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

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

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

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

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

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

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

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

Hisham Muhammad's avatar
Hisham Muhammad committed
66
67
68
69
70
typedef enum ProcessField_ {
   PID = 1, COMM, STATE, PPID, PGRP, SESSION, TTY_NR, TPGID, FLAGS, MINFLT, CMINFLT, MAJFLT, CMAJFLT, UTIME,
   STIME, CUTIME, CSTIME, PRIORITY, NICE, ITREALVALUE, STARTTIME, VSIZE, RSS, RLIM, STARTCODE, ENDCODE,
   STARTSTACK, KSTKESP, KSTKEIP, SIGNAL, BLOCKED, SSIGIGNORE, SIGCATCH, WCHAN, NSWAP, CNSWAP, EXIT_SIGNAL,
   PROCESSOR, M_SIZE, M_RESIDENT, M_SHARE, M_TRS, M_DRS, M_LRS, M_DT, ST_UID, PERCENT_CPU, PERCENT_MEM,
71
   USER, TIME, NLWP, TGID,
72
   #ifdef HAVE_OPENVZ
73
   CTID, VPID,
74
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
75
76
77
   #ifdef HAVE_VSERVER
   VXID,
   #endif
78
   #ifdef HAVE_TASKSTATS
79
   RCHAR, WCHAR, SYSCR, SYSCW, RBYTES, WBYTES, CNCLWB, IO_READ_RATE, IO_WRITE_RATE, IO_RATE,
80
   #endif
81
82
83
   #ifdef HAVE_CGROUP
   CGROUP,
   #endif
84
85
86
   #ifdef HAVE_OOM
   OOM,
   #endif
87
   IO_PRIORITY,
88
   LAST_PROCESSFIELD
Hisham Muhammad's avatar
Hisham Muhammad committed
89
90
91
92
93
94
95
96
97
} 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
      Process_fieldTitles[PID] =     "    PID ";
      Process_fieldTitles[PPID] =    "   PPID ";
298
299
300
      #ifdef HAVE_OPENVZ
      Process_fieldTitles[VPID] =    "   VPID ";
      #endif
301
302
303
304
      Process_fieldTitles[TPGID] =   "  TPGID ";
      Process_fieldTitles[TGID] =    "   TGID ";
      Process_fieldTitles[PGRP] =    "   PGRP ";
      Process_fieldTitles[SESSION] = "   SESN ";
305
306
307
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "    OOM ";
      #endif
308
309
310
      Process_pidFormat = "%7u ";
      Process_tpgidFormat = "%7d ";
   } else {
311
312
      Process_fieldTitles[PID] =     "  PID ";
      Process_fieldTitles[PPID] =    " PPID ";
313
314
315
      #ifdef HAVE_OPENVZ
      Process_fieldTitles[VPID] =    " VPID ";
      #endif
316
317
318
319
      Process_fieldTitles[TPGID] =   "TPGID ";
      Process_fieldTitles[TGID] =    " TGID ";
      Process_fieldTitles[PGRP] =    " PGRP ";
      Process_fieldTitles[SESSION] = " SESN ";
320
321
322
      #ifdef HAVE_OOM
      Process_fieldTitles[OOM] =     "  OOM ";
      #endif
323
324
325
326
327
      Process_pidFormat = "%5u ";
      Process_tpgidFormat = "%5d ";
   }
}

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

332
#define ONE_DECIMAL_K 1000L
333
#define ONE_DECIMAL_M (ONE_DECIMAL_K * ONE_DECIMAL_K)
334
#define ONE_DECIMAL_G (ONE_DECIMAL_M * ONE_DECIMAL_K)
335

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

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

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

397
398
   if (number > 10000000000) {
      snprintf(buffer, 13, "%11lld ", number / 1000);
399
400
401
      RichString_appendn(str, largeNumberColor, buffer, 5);
      RichString_appendn(str, processMegabytesColor, buffer+5, 3);
      RichString_appendn(str, processColor, buffer+8, 4);
402
   } else {
403
      snprintf(buffer, 13, "%11llu ", number);
404
405
406
407
      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);
408
409
410
   }
}

411
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
412

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

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

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

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

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

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

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

655
656
657
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
   ProcessField* fields = this->pl->fields;
658
   RichString_prune(out);
659
660
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
661
   if (this->pl->shadowOtherUsers && (int)this->st_uid != Process_getuid)
662
663
664
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
665
   assert(out->chlen > 0);
666
667
668
669
670
}

void Process_delete(Object* cast) {
   Process* this = (Process*) cast;
   assert (this != NULL);
671
   free(this->comm);
672
#ifdef HAVE_CGROUP
673
   free(this->cgroup);
674
#endif
675
676
677
   free(this);
}

678
679
680
681
682
683
684
ObjectClass Process_class = {
   .extends = Class(Object),
   .display = Process_display,
   .delete = Process_delete,
   .compare = Process_compare
};

685
Process* Process_new(struct ProcessList_ *pl) {
686
   Process* this = calloc(1, sizeof(Process));
687
   Object_setClass(this, Class(Process));
688
689
690
   this->pid = 0;
   this->pl = pl;
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
691
   this->showChildren = true;
692
   this->show = true;
693
694
695
696
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
697
   this->basenameOffset = -1;
698
   this->indent = 0;
699
700
701
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
   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);
}

719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
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)

744
#ifdef HAVE_LIBHWLOC
745

746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
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;
765
766
}

767
768
769
770
771
772
773
774
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;
775
}
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800

#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
801
#endif
802

803
804
void Process_sendSignal(Process* this, size_t sgn) {
   kill(this->pid, (int) sgn);
805
806
}

807
long Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
808
809
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
810
811
812
   return (p1->pid - p2->pid);
}

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