Process.c 21.3 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
#include "Settings.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
11
12
#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>
Hisham Muhammad's avatar
Hisham Muhammad committed
27
#include <time.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
28
#include <assert.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
29

30
// On Linux, this works only with glibc 2.1+. On earlier versions
Hisham Muhammad's avatar
Hisham Muhammad committed
31
// the behavior is similar to have a hardcoded page size.
32
#ifndef PAGE_SIZE
Hisham Muhammad's avatar
Hisham Muhammad committed
33
#define PAGE_SIZE ( sysconf(_SC_PAGESIZE) )
34
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
35
#define PAGE_SIZE_KB ( PAGE_SIZE / ONE_K )
Hisham Muhammad's avatar
Hisham Muhammad committed
36
37

/*{
Hisham Muhammad's avatar
Hisham Muhammad committed
38
#include "Object.h"
39

Hisham Muhammad's avatar
Hisham Muhammad committed
40
#include <sys/types.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
41

42
#define PROCESS_FLAG_IO 0x0001
43

44
45
46
47
48
49
50
51
#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
52
#ifndef Process_isThread
53
#define Process_isThread(_process) (Process_isUserlandThread(_process) || Process_isKernelThread(_process))
Hisham Muhammad's avatar
Hisham Muhammad committed
54
55
#endif

56
typedef int ProcessField;
Hisham Muhammad's avatar
Hisham Muhammad committed
57
58
59
60

typedef struct Process_ {
   Object super;

Hisham Muhammad's avatar
Hisham Muhammad committed
61
   struct Settings_* settings;
Hisham Muhammad's avatar
Hisham Muhammad committed
62

Hisham Muhammad's avatar
Hisham Muhammad committed
63
   pid_t pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
64
65
   char* comm;
   int indent;
66
67
68
69

   int basenameOffset;
   bool updated;

Hisham Muhammad's avatar
Hisham Muhammad committed
70
71
   char state;
   bool tag;
Hisham Muhammad's avatar
Hisham Muhammad committed
72
   bool showChildren;
73
   bool show;
Hisham Muhammad's avatar
Hisham Muhammad committed
74
   pid_t ppid;
75
76
77
   unsigned int pgrp;
   unsigned int session;
   unsigned int tty_nr;
Hisham Muhammad's avatar
Hisham Muhammad committed
78
   pid_t tgid;
79
   int tpgid;
Hisham Muhammad's avatar
Hisham Muhammad committed
80
   unsigned long int flags;
Hisham Muhammad's avatar
Hisham Muhammad committed
81
82
83
84
85
86

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

87
88
89
90
   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
91
92
   long int priority;
   long int nice;
93
   long int nlwp;
Hisham Muhammad's avatar
Hisham Muhammad committed
94
95
   char starttime_show[8];
   time_t starttime_ctime;
Hisham Muhammad's avatar
Hisham Muhammad committed
96
97
98
99
100
101
102
103
104
105
106

   #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;
   unsigned long long io_rate_read_time;
   unsigned long long io_rate_write_time;   
Hisham Muhammad's avatar
Hisham Muhammad committed
107
108
   double io_rate_read_bps;
   double io_rate_write_bps;
109
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
110

Hisham Muhammad's avatar
Hisham Muhammad committed
111
   int processor;
112
113
114
115
116
117
118
   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
119

120
   #ifdef HAVE_OPENVZ
121
   unsigned int ctid;
122
123
   unsigned int vpid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
124
125
126
   #ifdef HAVE_VSERVER
   unsigned int vxid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
127

128
129
130
   #ifdef HAVE_CGROUP
   char* cgroup;
   #endif
131
132
133
   #ifdef HAVE_OOM
   unsigned int oom;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
134
135
136
137
138
139
140

   int exit_signal;

   unsigned long int minflt;
   unsigned long int cminflt;
   unsigned long int majflt;
   unsigned long int cmajflt;
141
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
   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
160
161
} Process;

Hisham Muhammad's avatar
Hisham Muhammad committed
162
163
164
165
166
167
typedef struct ProcessFieldData_ {
   const char* name;
   const char* title;
   const char* description;
   int flags;
} ProcessFieldData;
Hisham Muhammad's avatar
Hisham Muhammad committed
168

Hisham Muhammad's avatar
Hisham Muhammad committed
169
170
void Process_writeField(Process* this, RichString* str, ProcessField field);
long Process_compare(const void* v1, const void* v2);
Hisham Muhammad's avatar
Hisham Muhammad committed
171

Hisham Muhammad's avatar
Hisham Muhammad committed
172
}*/
173

174
175
static int Process_getuid = -1;

176
#define ONE_K 1024L
Hisham Muhammad's avatar
Hisham Muhammad committed
177
178
179
#define ONE_M (ONE_K * ONE_K)
#define ONE_G (ONE_M * ONE_K)

180
#define ONE_DECIMAL_K 1000L
181
#define ONE_DECIMAL_M (ONE_DECIMAL_K * ONE_DECIMAL_K)
182
#define ONE_DECIMAL_G (ONE_DECIMAL_M * ONE_DECIMAL_K)
183

184
static void Process_humanNumber(RichString* str, unsigned long number, bool coloring) {
Hisham Muhammad's avatar
Hisham Muhammad committed
185
186
   char buffer[11];
   int len;
187
188
189
190
191
192
193
194
   
   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
195
 
196
   if(number >= (10 * ONE_DECIMAL_M)) {
197
198
199
      #ifdef __LP64__
      if(number >= (100 * ONE_DECIMAL_G)) {
         len = snprintf(buffer, 10, "%4ldT ", number / ONE_G);
200
         RichString_appendn(str, largeNumberColor, buffer, len);
201
         return;
202
      } else if (number >= (1000 * ONE_DECIMAL_M)) {
203
         len = snprintf(buffer, 10, "%4.1lfT ", (double)number / ONE_G);
204
         RichString_appendn(str, largeNumberColor, buffer, len);
205
206
         return;
      }
207
      #endif
208
      if(number >= (100 * ONE_DECIMAL_M)) {
209
         len = snprintf(buffer, 10, "%4ldG ", number / ONE_M);
210
         RichString_appendn(str, largeNumberColor, buffer, len);
211
         return;
212
      }
213
      len = snprintf(buffer, 10, "%4.1lfG ", (double)number / ONE_M);
214
      RichString_appendn(str, largeNumberColor, buffer, len);
215
      return;
216
   } else if (number >= 100000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
217
      len = snprintf(buffer, 10, "%4ldM ", number / ONE_K);
218
      RichString_appendn(str, processMegabytesColor, buffer, len);
219
      return;
220
   } else if (number >= 1000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
221
      len = snprintf(buffer, 10, "%2ld", number/1000);
222
      RichString_appendn(str, processMegabytesColor, buffer, len);
Hisham Muhammad's avatar
Hisham Muhammad committed
223
      number %= 1000;
224
      len = snprintf(buffer, 10, "%03lu ", number);
225
      RichString_appendn(str, processColor, buffer, len);
226
      return;
Hisham Muhammad's avatar
Hisham Muhammad committed
227
   }
228
   len = snprintf(buffer, 10, "%5lu ", number);
229
   RichString_appendn(str, processColor, buffer, len);
Hisham Muhammad's avatar
Hisham Muhammad committed
230
231
}

232
static void Process_colorNumber(RichString* str, unsigned long long number, bool coloring) {
233
   char buffer[14];
234
235
236
237
238
239
240
241
242
243
244

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

245
246
   if (number > 10000000000) {
      snprintf(buffer, 13, "%11lld ", number / 1000);
247
248
249
      RichString_appendn(str, largeNumberColor, buffer, 5);
      RichString_appendn(str, processMegabytesColor, buffer+5, 3);
      RichString_appendn(str, processColor, buffer+8, 4);
250
   } else {
251
      snprintf(buffer, 13, "%11llu ", number);
252
253
254
255
      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);
256
257
258
   }
}

259
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
260

261
static void Process_printTime(RichString* str, unsigned long long t) {
Hisham Muhammad's avatar
Hisham Muhammad committed
262
263
264
265
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;

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

268
   unsigned long long hours = iRealTime / 3600;
Hisham Muhammad's avatar
Hisham Muhammad committed
269
270
271
272
   int minutes = (iRealTime / 60) % 60;
   int seconds = iRealTime % 60;
   int hundredths = (realTime - iRealTime) * 100;
   char buffer[11];
273
274
   if (hours >= 100) {
      snprintf(buffer, 10, "%7lluh ", hours);
Hisham Muhammad's avatar
Hisham Muhammad committed
275
276
      RichString_append(str, CRT_colors[LARGE_NUMBER], buffer);
   } else {
277
278
279
280
281
282
283
284
      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
285
286
287
   }
}

288
static inline void Process_writeCommand(Process* this, int attr, int baseattr, RichString* str) {
289
   int start = RichString_size(str);
290
   RichString_append(str, attr, this->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
291
   if (this->settings->highlightBaseName) {
292
      int finish = RichString_size(str) - 1;
293
294
      if (this->basenameOffset != -1)
         finish = (start + this->basenameOffset) - 1;
295
296
297
298
      int colon = RichString_findChar(str, ':', start);
      if (colon != -1 && colon < finish) {
         finish = colon;
      } else {
299
         for (int i = finish - start; i >= 0; i--) {
300
301
302
303
304
            if (this->comm[i] == '/') {
               start += i+1;
               break;
            }
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
305
      }
306
      RichString_setAttrn(str, baseattr, start, finish);
Hisham Muhammad's avatar
Hisham Muhammad committed
307
308
309
   }
}

Hisham Muhammad's avatar
Hisham Muhammad committed
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
static inline void Process_outputRate(RichString* str, char* buffer, int n, double rate, int coloring) {
   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];
   }
   if (rate < ONE_K) {
      int len = snprintf(buffer, n, "%7.2f B/s ", rate);
      RichString_appendn(str, processColor, buffer, len);
   } else if (rate < ONE_K * ONE_K) {
      int len = snprintf(buffer, n, "%7.2f K/s ", rate / ONE_K);
      RichString_appendn(str, processColor, buffer, len);
   } else if (rate < ONE_K * ONE_K * ONE_K) {
      int len = snprintf(buffer, n, "%7.2f M/s ", rate / ONE_K / ONE_K);
      RichString_appendn(str, processMegabytesColor, buffer, len);
   } else {
      int len = snprintf(buffer, n, "%7.2f G/s ", rate / ONE_K / ONE_K / ONE_K);
      RichString_appendn(str, largeNumberColor, buffer, len);
330
331
332
   }
}

Hisham Muhammad's avatar
Hisham Muhammad committed
333
void Process_writeDefaultField(Process* this, RichString* str, ProcessField field) {
334
   char buffer[256]; buffer[255] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
335
   int attr = CRT_colors[DEFAULT_COLOR];
336
   int baseattr = CRT_colors[PROCESS_BASENAME];
337
   int n = sizeof(buffer) - 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
338
   bool coloring = this->settings->highlightMegabytes;
Hisham Muhammad's avatar
Hisham Muhammad committed
339
340

   switch (field) {
341
342
   case PID: snprintf(buffer, n, Process_pidFormat, this->pid); break;
   case PPID: snprintf(buffer, n, Process_pidFormat, this->ppid); break;
343
344
   case PGRP: snprintf(buffer, n, Process_pidFormat, this->pgrp); break;
   case SESSION: snprintf(buffer, n, Process_pidFormat, this->session); break;
345
   case TTY_NR: snprintf(buffer, n, "%5u ", this->tty_nr); break;
346
347
   case TGID: snprintf(buffer, n, Process_pidFormat, this->tgid); break;
   case TPGID: snprintf(buffer, n, Process_tpgidFormat, this->tpgid); break;
348
349
350
351
   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;
Hisham Muhammad's avatar
Hisham Muhammad committed
352
   case PROCESSOR: snprintf(buffer, n, "%3d ", Settings_cpuId(this->settings, this->processor)); break;
353
   case NLWP: snprintf(buffer, n, "%4ld ", this->nlwp); break;
Hisham Muhammad's avatar
Hisham Muhammad committed
354
   case COMM: {
Hisham Muhammad's avatar
Hisham Muhammad committed
355
      if (this->settings->highlightThreads && Process_isThread(this)) {
356
357
358
         attr = CRT_colors[PROCESS_THREAD];
         baseattr = CRT_colors[PROCESS_THREAD_BASENAME];
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
359
      if (!this->settings->treeView || this->indent == 0) {
360
         Process_writeCommand(this, attr, baseattr, str);
Hisham Muhammad's avatar
Hisham Muhammad committed
361
362
363
364
         return;
      } else {
         char* buf = buffer;
         int maxIndent = 0;
365
366
367
         bool lastItem = (this->indent < 0);
         int indent = (this->indent < 0 ? -this->indent : this->indent);

Hisham Muhammad's avatar
Hisham Muhammad committed
368
         for (int i = 0; i < 32; i++)
369
            if (indent & (1 << i))
Hisham Muhammad's avatar
Hisham Muhammad committed
370
               maxIndent = i+1;
371
          for (int i = 0; i < maxIndent - 1; i++) {
372
373
            int written;
            if (indent & (1 << i))
Hisham Muhammad's avatar
Hisham Muhammad committed
374
               written = snprintf(buf, n, "%s  ", CRT_treeStr[TREE_STR_VERT]);
Hisham Muhammad's avatar
Hisham Muhammad committed
375
            else
376
377
378
               written = snprintf(buf, n, "   ");
            buf += written;
            n -= written;
379
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
380
381
         const char* draw = CRT_treeStr[lastItem ? (this->settings->direction == 1 ? TREE_STR_BEND : TREE_STR_TEND) : TREE_STR_RTEE];
         snprintf(buf, n, "%s%s ", draw, this->showChildren ? CRT_treeStr[TREE_STR_SHUT] : CRT_treeStr[TREE_STR_OPEN] );
Hisham Muhammad's avatar
Hisham Muhammad committed
382
         RichString_append(str, CRT_colors[PROCESS_TREE], buffer);
383
         Process_writeCommand(this, attr, baseattr, str);
384
         return;
Hisham Muhammad's avatar
Hisham Muhammad committed
385
386
387
388
      }
   }
   case STATE: {
      snprintf(buffer, n, "%c ", this->state);
389
390
391
392
393
394
395
396
      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
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
      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;
   }
413
414
415
416
417
418
419
   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
420
421
   case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break;
   case USER: {
Hisham Muhammad's avatar
Hisham Muhammad committed
422
      if (Process_getuid != (int) this->st_uid)
Hisham Muhammad's avatar
Hisham Muhammad committed
423
         attr = CRT_colors[PROCESS_SHADOW];
424
      if (this->user) {
425
         snprintf(buffer, n, "%-9s ", this->user);
426
      } else {
427
         snprintf(buffer, n, "%-9d ", this->st_uid);
428
      }
429
430
431
      if (buffer[9] != '\0') {
         buffer[9] = ' ';
         buffer[10] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
432
433
434
435
436
437
438
439
440
      }
      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: {
441
442
443
444
      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
445
446
447
448
449
450
451
452
453
454
455
456
457
      } 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
458
   case STARTTIME: snprintf(buffer, n, "%s", this->starttime_show); break;
459
   #ifdef HAVE_OPENVZ
460
461
   case CTID: snprintf(buffer, n, "%7u ", this->ctid); break;
   case VPID: snprintf(buffer, n, Process_pidFormat, this->vpid); break;
462
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
463
464
465
   #ifdef HAVE_VSERVER
   case VXID: snprintf(buffer, n, "%5u ", this->vxid); break;
   #endif
466
   #ifdef HAVE_TASKSTATS
467
468
469
470
471
472
473
   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;
Hisham Muhammad's avatar
Hisham Muhammad committed
474
475
476
   case IO_READ_RATE:  Process_outputRate(str, buffer, n, this->io_rate_read_bps, coloring); return;
   case IO_WRITE_RATE: Process_outputRate(str, buffer, n, this->io_rate_write_bps, coloring); return;
   case IO_RATE: Process_outputRate(str, buffer, n, this->io_rate_read_bps + this->io_rate_write_bps, coloring); return;
477
   #endif
478
479
480
   #ifdef HAVE_CGROUP
   case CGROUP: snprintf(buffer, n, "%-10s ", this->cgroup); break;
   #endif
481
482
483
   #ifdef HAVE_OOM
   case OOM: snprintf(buffer, n, Process_pidFormat, this->oom); break;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
484
485
486
487
488
489
   default:
      snprintf(buffer, n, "- ");
   }
   RichString_append(str, attr, buffer);
}

490
491
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
Hisham Muhammad's avatar
Hisham Muhammad committed
492
   ProcessField* fields = this->settings->fields;
493
   RichString_prune(out);
494
495
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
496
   if (this->settings->shadowOtherUsers && (int)this->st_uid != Process_getuid)
497
498
499
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
500
   assert(out->chlen > 0);
501
502
}

503
void Process_done(Process* this) {
504
   assert (this != NULL);
505
   free(this->comm);
506
#ifdef HAVE_CGROUP
507
   free(this->cgroup);
508
#endif
509
510
}

511
512
513
514
515
516
517
ObjectClass Process_class = {
   .extends = Class(Object),
   .display = Process_display,
   .delete = Process_delete,
   .compare = Process_compare
};

518
void Process_init(Process* this, struct Settings_* settings) {
519
   this->pid = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
520
   this->settings = settings;
521
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
522
   this->showChildren = true;
523
   this->show = true;
524
525
526
527
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
528
   this->basenameOffset = -1;
529
   this->indent = 0;
530
531
532
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
   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);
}

549
550
551
552
553
554
bool Process_changePriorityBy(Process* this, size_t delta) {
   return Process_setPriority(this, this->nice + delta);
}

void Process_sendSignal(Process* this, size_t sgn) {
   kill(this->pid, (int) sgn);
555
556
}

557
long Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
558
559
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
560
561
562
   return (p1->pid - p2->pid);
}

Hisham Muhammad's avatar
Hisham Muhammad committed
563
long Process_defaultCompare(const void* v1, const void* v2) {
564
   Process *p1, *p2;
Hisham Muhammad's avatar
Hisham Muhammad committed
565
566
   Settings *settings = ((Process*)v1)->settings;
   if (settings->direction == 1) {
567
568
569
570
571
572
      p1 = (Process*)v1;
      p2 = (Process*)v2;
   } else {
      p2 = (Process*)v1;
      p1 = (Process*)v2;
   }
573
   long long diff;
Hisham Muhammad's avatar
Hisham Muhammad committed
574
   switch (settings->sortKey) {
Hisham Muhammad's avatar
Hisham Muhammad committed
575
   case PID:
576
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
577
   case PPID:
578
      return (p1->ppid - p2->ppid);
Hisham Muhammad's avatar
Hisham Muhammad committed
579
   case USER:
Hisham Muhammad's avatar
Hisham Muhammad committed
580
      return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : "");
Hisham Muhammad's avatar
Hisham Muhammad committed
581
   case PRIORITY:
582
      return (p1->priority - p2->priority);
583
584
585
586
   case PROCESSOR:
      return (p1->processor - p2->processor);
   case SESSION:
      return (p1->session - p2->session);
Hisham Muhammad's avatar
Hisham Muhammad committed
587
   case STATE:
588
      return (p1->state - p2->state);
Hisham Muhammad's avatar
Hisham Muhammad committed
589
   case NICE:
590
      return (p1->nice - p2->nice);
591
592
593
594
595
596
597
598
   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
599
   case M_SIZE:
600
      return (p2->m_size - p1->m_size);
Hisham Muhammad's avatar
Hisham Muhammad committed
601
   case M_RESIDENT:
602
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
603
   case M_SHARE:
604
      return (p2->m_share - p1->m_share);
Hisham Muhammad's avatar
Hisham Muhammad committed
605
   case PERCENT_CPU:
606
      return (p2->percent_cpu > p1->percent_cpu ? 1 : -1);
Hisham Muhammad's avatar
Hisham Muhammad committed
607
   case PERCENT_MEM:
608
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
609
610
611
612
   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
613
   case TIME:
614
      return ((p2->utime+p2->stime) - (p1->utime+p1->stime));
Hisham Muhammad's avatar
Hisham Muhammad committed
615
   case COMM:
616
      return strcmp(p1->comm, p2->comm);
617
618
   case NLWP:
      return (p1->nlwp - p2->nlwp);
619
620
621
622
623
624
   case STARTTIME: {
      if (p1->starttime_ctime == p2->starttime_ctime)
         return (p1->pid - p2->pid);
      else
         return (p1->starttime_ctime - p2->starttime_ctime);
   }
625
   #ifdef HAVE_OPENVZ
626
627
   case CTID:
      return (p1->ctid - p2->ctid);
628
629
630
   case VPID:
      return (p1->vpid - p2->vpid);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
631
632
633
634
   #ifdef HAVE_VSERVER
   case VXID:
      return (p1->vxid - p2->vxid);
   #endif
635
636
637
638
639
640
641
642
643
644
   #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;
645
   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;
646
   #endif
647
648
649
650
   #ifdef HAVE_CGROUP
   case CGROUP:
      return strcmp(p1->cgroup ? p1->cgroup : "", p2->cgroup ? p2->cgroup : "");
   #endif
651
652
653
654
   #ifdef HAVE_OOM
   case OOM:
      return (p1->oom - p2->oom);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
655
   default:
656
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
657
   }
658
659
   test_diff:
   return (diff > 0) ? 1 : (diff < 0 ? -1 : 0);
Hisham Muhammad's avatar
Hisham Muhammad committed
660
}