Process.c 23.1 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
8
9
10
11
12
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/

#include "ProcessList.h"
#include "Object.h"
#include "CRT.h"
#include "String.h"
#include "Process.h"
13
#include "RichString.h"
14
#include "Affinity.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
15
16
17
18
19
20
21
22
23
24
25
26
27
28

#include "debug.h"

#include <stdio.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/param.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <signal.h>
#include <string.h>
#include <stdbool.h>
#include <pwd.h>
29
#include <sched.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
30
#include <time.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
31

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

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

/*{

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

Hisham Muhammad's avatar
Hisham Muhammad committed
57
58
59
60
61
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,
62
   USER, TIME, NLWP, TGID,
63
   #ifdef HAVE_OPENVZ
64
   CTID, VPID,
65
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
66
67
68
   #ifdef HAVE_VSERVER
   VXID,
   #endif
69
   #ifdef HAVE_TASKSTATS
70
   RCHAR, WCHAR, SYSCR, SYSCW, RBYTES, WBYTES, CNCLWB, IO_READ_RATE, IO_WRITE_RATE, IO_RATE,
71
   #endif
72
73
74
   #ifdef HAVE_CGROUP
   CGROUP,
   #endif
75
   LAST_PROCESSFIELD
Hisham Muhammad's avatar
Hisham Muhammad committed
76
77
78
79
80
81
82
83
84
85
} ProcessField;

struct ProcessList_;

typedef struct Process_ {
   Object super;

   struct ProcessList_ *pl;
   bool updated;

Hisham Muhammad's avatar
Hisham Muhammad committed
86
   pid_t pid;
Hisham Muhammad's avatar
Hisham Muhammad committed
87
88
89
90
   char* comm;
   int indent;
   char state;
   bool tag;
Hisham Muhammad's avatar
Hisham Muhammad committed
91
   bool showChildren;
92
   bool show;
Hisham Muhammad's avatar
Hisham Muhammad committed
93
   pid_t ppid;
94
95
96
   unsigned int pgrp;
   unsigned int session;
   unsigned int tty_nr;
Hisham Muhammad's avatar
Hisham Muhammad committed
97
   pid_t tgid;
98
   int tpgid;
Hisham Muhammad's avatar
Hisham Muhammad committed
99
   unsigned long int flags;
100
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
101
102
103
104
   unsigned long int minflt;
   unsigned long int cminflt;
   unsigned long int majflt;
   unsigned long int cmajflt;
105
   #endif
106
107
108
109
   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
110
111
   long int priority;
   long int nice;
112
   long int nlwp;
Hisham Muhammad's avatar
Hisham Muhammad committed
113
114
   char starttime_show[8];
   time_t starttime_ctime;
115
   #ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
   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;
132
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
133
134
135
136
137
138
139
140
141
142
143
144
   int exit_signal;
   int processor;
   int m_size;
   int m_resident;
   int m_share;
   int m_trs;
   int m_drs;
   int m_lrs;
   int m_dt;
   uid_t st_uid;
   float percent_cpu;
   float percent_mem;
145
   char* user;
146
   #ifdef HAVE_OPENVZ
147
   unsigned int ctid;
148
149
   unsigned int vpid;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
150
151
152
   #ifdef HAVE_VSERVER
   unsigned int vxid;
   #endif
153
154
155
156
157
158
159
160
161
162
163
164
165
   #ifdef HAVE_TASKSTATS
   unsigned long long io_rchar;
   unsigned long long io_wchar;
   unsigned long long io_syscr;
   unsigned long long io_syscw;
   unsigned long long io_read_bytes;
   unsigned long long io_write_bytes;
   unsigned long long io_cancelled_write_bytes;
   double io_rate_read_bps;
   unsigned long long io_rate_read_time;
   double io_rate_write_bps;
   unsigned long long io_rate_write_time;   
   #endif
166
167
168
   #ifdef HAVE_CGROUP
   char* cgroup;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
169
170
171
172
} Process;

}*/

173
#ifdef DEBUG
Hisham Muhammad's avatar
Hisham Muhammad committed
174
char* PROCESS_CLASS = "Process";
175
176
177
#else
#define PROCESS_CLASS NULL
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
178

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

Hisham Muhammad's avatar
Hisham Muhammad committed
204
const char *Process_fieldTitles[] = {
205
206
   "", "    PID ", "Command ", "S ", "   PPID ", " PGRP ", " SESN ",
   "  TTY ", "  TPGID ", "- ", "- ", "- ", "- ", "- ",
207
   " UTIME+  ", " STIME+  ", " CUTIME+ ", " CSTIME+ ", "PRI ", " NI ", "- ",
Hisham Muhammad's avatar
Hisham Muhammad committed
208
   "START ", "- ", "- ", "- ", "- ", "- ", "- ",
209
210
211
   "- ", "- ", "- ", "- ", "- ", "- ", "- ",
   "- ", "- ", "- ", "CPU ", " VIRT ", "  RES ", "  SHR ",
   " CODE ", " DATA ", " LIB ", " DIRTY ", " UID ", "CPU% ", "MEM% ",
212
   "USER      ", "  TIME+  ", "NLWP ", "   TGID ",
213
#ifdef HAVE_OPENVZ
214
   " CTID ", " VPID ",
215
#endif
Hisham Muhammad's avatar
Hisham Muhammad committed
216
217
218
#ifdef HAVE_VSERVER
   " VXID ",
#endif
219
#ifdef HAVE_TASKSTATS
220
   "     RD_CHAR ", "     WR_CHAR ", "   RD_SYSC ", "   WR_SYSC ", "  IO_RBYTES ", "  IO_WBYTES ", " IO_CANCEL ",
221
222
   " IORR ", " IOWR ", "   IO ",
#endif
223
224
225
226
#ifdef HAVE_CGROUP
   "    CGROUP ",
#endif
"*** report bug! ***"
227
228
};

229
230
static int Process_getuid = -1;

231
232
233
234
235
236
237
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;
238
   fscanf(file, "%32d", &maxPid);
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
   fclose(file);
   if (maxPid > 99999) {
      Process_fieldTitles[PID] =   "    PID ";
      Process_fieldTitles[PPID] =  "   PPID ";
      Process_fieldTitles[TPGID] = "  TPGID ";
      Process_fieldTitles[TGID] =  "   TGID ";
      Process_pidFormat = "%7u ";
      Process_tpgidFormat = "%7d ";
   } else {
      Process_fieldTitles[PID] =   "  PID ";
      Process_fieldTitles[PPID] =  " PPID ";
      Process_fieldTitles[TPGID] = "TPGID ";
      Process_fieldTitles[TGID] =  " TGID ";
      Process_pidFormat = "%5u ";
      Process_tpgidFormat = "%5d ";
   }
}

Hisham Muhammad's avatar
Hisham Muhammad committed
257
258
259
260
#define ONE_K 1024
#define ONE_M (ONE_K * ONE_K)
#define ONE_G (ONE_M * ONE_K)

261
static void Process_humanNumber(Process* this, RichString* str, unsigned long number) {
Hisham Muhammad's avatar
Hisham Muhammad committed
262
263
   char buffer[11];
   int len;
Hisham Muhammad's avatar
Hisham Muhammad committed
264
   if(number >= (10 * ONE_M)) {
265
266
267
268
269
270
271
272
      if(number >= (100 * ONE_M)) {
         len = snprintf(buffer, 10, "%4ldG ", number / ONE_M);
         RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, len);
      } else {
         len = snprintf(buffer, 10, "%3.1fG ", (float)number / ONE_M);
         RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, len);
      }
   } else if (number >= 100000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
273
      len = snprintf(buffer, 10, "%4ldM ", number / ONE_K);
Hisham Muhammad's avatar
Hisham Muhammad committed
274
275
276
277
278
      int attr = this->pl->highlightMegabytes
               ? CRT_colors[PROCESS_MEGABYTES]
               : CRT_colors[PROCESS];
      RichString_appendn(str, attr, buffer, len);
   } else if (this->pl->highlightMegabytes && number >= 1000) {
Hisham Muhammad's avatar
Hisham Muhammad committed
279
      len = snprintf(buffer, 10, "%2ld", number/1000);
Hisham Muhammad's avatar
Hisham Muhammad committed
280
281
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer, len);
      number %= 1000;
Hisham Muhammad's avatar
Hisham Muhammad committed
282
      len = snprintf(buffer, 10, "%03ld ", number);
Hisham Muhammad's avatar
Hisham Muhammad committed
283
284
      RichString_appendn(str, CRT_colors[PROCESS], buffer, len);
   } else {
Hisham Muhammad's avatar
Hisham Muhammad committed
285
      len = snprintf(buffer, 10, "%5ld ", number);
Hisham Muhammad's avatar
Hisham Muhammad committed
286
287
288
289
      RichString_appendn(str, CRT_colors[PROCESS], buffer, len);
   }
}

290
static void Process_colorNumber(RichString* str, unsigned long long number) {
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
   char buffer[14];
   if (number > 10000000000) {
      snprintf(buffer, 13, "%11lld ", number / 1000);
      RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, 5);
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer+5, 3);
      RichString_appendn(str, CRT_colors[PROCESS], buffer+8, 4);
   } else {
      snprintf(buffer, 13, "%11lld ", number);
      RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, 2);
      RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer+2, 3);
      RichString_appendn(str, CRT_colors[PROCESS], buffer+5, 3);
      RichString_appendn(str, CRT_colors[PROCESS_SHADOW], buffer+8, 4);
   }
}

306
static double jiffy = 0.0;
Hisham Muhammad's avatar
Hisham Muhammad committed
307

308
static void Process_printTime(RichString* str, unsigned long long t) {
Hisham Muhammad's avatar
Hisham Muhammad committed
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
   if(jiffy == 0.0) jiffy = sysconf(_SC_CLK_TCK);
   double jiffytime = 1.0 / jiffy;

   double realTime = t * jiffytime;
   int iRealTime = (int) realTime;

   int hours = iRealTime / 3600;
   int minutes = (iRealTime / 60) % 60;
   int seconds = iRealTime % 60;
   int hundredths = (realTime - iRealTime) * 100;
   char buffer[11];
   if (hours) {
      snprintf(buffer, 10, "%2dh", 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);
}

330
static inline void Process_writeCommand(Process* this, int attr, int baseattr, RichString* str) {
331
   int start = RichString_size(str);
332
   RichString_append(str, attr, this->comm);
Hisham Muhammad's avatar
Hisham Muhammad committed
333
   if (this->pl->highlightBaseName) {
334
      int finish = RichString_size(str) - 1;
335
336
337
338
339
340
341
342
      int space = RichString_findChar(str, ' ', start);
      if (space != -1)
         finish = space - 1;
      for (;;) {
         int slash = RichString_findChar(str, '/', start);
         if (slash == -1 || slash > finish)
            break;
         start = slash + 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
343
      }
344
      RichString_setAttrn(str, baseattr, start, finish);
Hisham Muhammad's avatar
Hisham Muhammad committed
345
346
347
   }
}

348
349
350
351
352
353
354
355
356
static inline void Process_outputRate(Process* this, RichString* str, int attr, char* buffer, int n, double rate) {
   rate = rate / 1024;
   if (rate < 0.01)
      snprintf(buffer, n, "    0 ");
   else if (rate <= 10)
      snprintf(buffer, n, "%5.2f ", rate);
   else if (rate <= 100)
      snprintf(buffer, n, "%5.1f ", rate);
   else {
357
      Process_humanNumber(this, str, rate);
358
359
360
361
362
      return;
   }
   RichString_append(str, attr, buffer);
}

363
static void Process_writeField(Process* this, RichString* str, ProcessField field) {
364
   char buffer[128]; buffer[127] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
365
   int attr = CRT_colors[DEFAULT_COLOR];
366
   int baseattr = CRT_colors[PROCESS_BASENAME];
367
   int n = sizeof(buffer) - 1;
Hisham Muhammad's avatar
Hisham Muhammad committed
368
369

   switch (field) {
370
371
   case PID: snprintf(buffer, n, Process_pidFormat, this->pid); break;
   case PPID: snprintf(buffer, n, Process_pidFormat, this->ppid); break;
372
373
374
   case PGRP: snprintf(buffer, n, "%5u ", this->pgrp); break;
   case SESSION: snprintf(buffer, n, "%5u ", this->session); break;
   case TTY_NR: snprintf(buffer, n, "%5u ", this->tty_nr); break;
375
376
   case TGID: snprintf(buffer, n, Process_pidFormat, this->tgid); break;
   case TPGID: snprintf(buffer, n, Process_tpgidFormat, this->tpgid); break;
377
   case PROCESSOR: snprintf(buffer, n, "%3d ", ProcessList_cpuId(this->pl, this->processor)); break;
378
   case NLWP: snprintf(buffer, n, "%4ld ", this->nlwp); break;
Hisham Muhammad's avatar
Hisham Muhammad committed
379
   case COMM: {
Hisham Muhammad's avatar
Hisham Muhammad committed
380
      if (this->pl->highlightThreads && Process_isThread(this)) {
381
382
383
         attr = CRT_colors[PROCESS_THREAD];
         baseattr = CRT_colors[PROCESS_THREAD_BASENAME];
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
384
      if (!this->pl->treeView || this->indent == 0) {
385
         Process_writeCommand(this, attr, baseattr, str);
Hisham Muhammad's avatar
Hisham Muhammad committed
386
387
388
389
         return;
      } else {
         char* buf = buffer;
         int maxIndent = 0;
390
391
392
393
394
395
         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
396
         for (int i = 0; i < 32; i++)
397
            if (indent & (1 << i))
Hisham Muhammad's avatar
Hisham Muhammad committed
398
               maxIndent = i+1;
399
          for (int i = 0; i < maxIndent - 1; i++) {
400
401
402
            int written;
            if (indent & (1 << i))
               written = snprintf(buf, n, "%s  ", treeStr[TREE_STR_VERT]);
Hisham Muhammad's avatar
Hisham Muhammad committed
403
            else
404
405
406
               written = snprintf(buf, n, "   ");
            buf += written;
            n -= written;
407
         }
408
409
         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
410
         RichString_append(str, CRT_colors[PROCESS_TREE], buffer);
411
         Process_writeCommand(this, attr, baseattr, str);
412
         return;
Hisham Muhammad's avatar
Hisham Muhammad committed
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
      }
   }
   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;
   }
436
437
438
439
440
441
442
   case M_DRS: Process_humanNumber(this, str, this->m_drs * PAGE_SIZE_KB); return;
   case M_DT: Process_humanNumber(this, str, this->m_dt * PAGE_SIZE_KB); return;
   case M_LRS: Process_humanNumber(this, str, this->m_lrs * PAGE_SIZE_KB); return;
   case M_TRS: Process_humanNumber(this, str, this->m_trs * PAGE_SIZE_KB); return;
   case M_SIZE: Process_humanNumber(this, str, this->m_size * PAGE_SIZE_KB); return;
   case M_RESIDENT: Process_humanNumber(this, str, this->m_resident * PAGE_SIZE_KB); return;
   case M_SHARE: Process_humanNumber(this, str, this->m_share * PAGE_SIZE_KB); return;
Hisham Muhammad's avatar
Hisham Muhammad committed
443
444
   case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break;
   case USER: {
Hisham Muhammad's avatar
Hisham Muhammad committed
445
      if (Process_getuid != (int) this->st_uid)
Hisham Muhammad's avatar
Hisham Muhammad committed
446
         attr = CRT_colors[PROCESS_SHADOW];
447
      if (this->user) {
448
         snprintf(buffer, n, "%-9s ", this->user);
449
      } else {
450
         snprintf(buffer, n, "%-9d ", this->st_uid);
451
      }
452
453
454
      if (buffer[9] != '\0') {
         buffer[9] = ' ';
         buffer[10] = '\0';
Hisham Muhammad's avatar
Hisham Muhammad committed
455
456
457
458
459
460
461
462
463
      }
      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: {
464
465
466
467
      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
468
469
470
471
472
473
474
475
476
477
478
479
480
      } 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
481
   case STARTTIME: snprintf(buffer, n, "%s", this->starttime_show); break;
482
   #ifdef HAVE_OPENVZ
483
   case CTID: snprintf(buffer, n, "%5u ", this->ctid); break;
484
485
   case VPID: snprintf(buffer, n, "%5u ", this->vpid); break;
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
486
487
488
   #ifdef HAVE_VSERVER
   case VXID: snprintf(buffer, n, "%5u ", this->vxid); break;
   #endif
489
   #ifdef HAVE_TASKSTATS
490
491
   case RCHAR:  snprintf(buffer, n, "%12llu ", this->io_rchar); break;
   case WCHAR:  snprintf(buffer, n, "%12llu ", this->io_wchar); break;   
492
493
   case SYSCR:  snprintf(buffer, n, "%10llu ", this->io_syscr); break;   
   case SYSCW:  snprintf(buffer, n, "%10llu ", this->io_syscw); break; 
494
495
   case RBYTES: Process_colorNumber(str, this->io_read_bytes); return;
   case WBYTES: Process_colorNumber(str, this->io_write_bytes); return;
496
   case CNCLWB: snprintf(buffer, n, "%10llu ", this->io_cancelled_write_bytes); break; 
497
498
499
   case IO_READ_RATE:  Process_outputRate(this, str, attr, buffer, n, this->io_rate_read_bps); return;
   case IO_WRITE_RATE: Process_outputRate(this, str, attr, buffer, n, this->io_rate_write_bps); return;
   case IO_RATE: Process_outputRate(this, str, attr, buffer, n, this->io_rate_read_bps + this->io_rate_write_bps); return;
500
   #endif
501
502
503
   #ifdef HAVE_CGROUP
   case CGROUP: snprintf(buffer, n, "%-10s ", this->cgroup); break;
   #endif
504

Hisham Muhammad's avatar
Hisham Muhammad committed
505
506
507
508
509
510
   default:
      snprintf(buffer, n, "- ");
   }
   RichString_append(str, attr, buffer);
}

511
512
513
static void Process_display(Object* cast, RichString* out) {
   Process* this = (Process*) cast;
   ProcessField* fields = this->pl->fields;
514
   RichString_prune(out);
515
516
   for (int i = 0; fields[i]; i++)
      Process_writeField(this, out, fields[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
517
   if (this->pl->shadowOtherUsers && (int)this->st_uid != Process_getuid)
518
519
520
      RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]);
   if (this->tag == true)
      RichString_setAttr(out, CRT_colors[PROCESS_TAG]);
521
   assert(out->chlen > 0);
522
523
524
525
526
527
}

void Process_delete(Object* cast) {
   Process* this = (Process*) cast;
   assert (this != NULL);
   if (this->comm) free(this->comm);
528
529
530
#ifdef HAVE_CGROUP
   if (this->cgroup) free(this->cgroup);
#endif
531
532
533
534
535
536
537
538
539
540
541
   free(this);
}

Process* Process_new(struct ProcessList_ *pl) {
   Process* this = calloc(sizeof(Process), 1);
   Object_setClass(this, PROCESS_CLASS);
   ((Object*)this)->display = Process_display;
   ((Object*)this)->delete = Process_delete;
   this->pid = 0;
   this->pl = pl;
   this->tag = false;
Hisham Muhammad's avatar
Hisham Muhammad committed
542
   this->showChildren = true;
543
   this->show = true;
544
545
546
547
548
   this->updated = false;
   this->utime = 0;
   this->stime = 0;
   this->comm = NULL;
   this->indent = 0;
549
550
551
#ifdef HAVE_CGROUP
   this->cgroup = NULL;
#endif
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
   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);
}

569
#ifdef HAVE_HWLOC
570

571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
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;
590
591
}

592
593
594
595
596
597
598
599
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;
600
}
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625

#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
626
#endif
627

Hisham Muhammad's avatar
Hisham Muhammad committed
628
629
void Process_sendSignal(Process* this, int sgn) {
   kill(this->pid, sgn);
630
631
}

632
int Process_pidCompare(const void* v1, const void* v2) {
Hisham Muhammad's avatar
Hisham Muhammad committed
633
634
   Process* p1 = (Process*)v1;
   Process* p2 = (Process*)v2;
635
636
637
638
639
640
641
642
643
644
645
646
647
   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;
   }
648
   long long diff;
649
   switch (pl->sortKey) {
Hisham Muhammad's avatar
Hisham Muhammad committed
650
   case PID:
651
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
652
   case PPID:
653
      return (p1->ppid - p2->ppid);
Hisham Muhammad's avatar
Hisham Muhammad committed
654
   case USER:
Hisham Muhammad's avatar
Hisham Muhammad committed
655
      return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : "");
Hisham Muhammad's avatar
Hisham Muhammad committed
656
   case PRIORITY:
657
      return (p1->priority - p2->priority);
658
659
660
661
   case PROCESSOR:
      return (p1->processor - p2->processor);
   case SESSION:
      return (p1->session - p2->session);
Hisham Muhammad's avatar
Hisham Muhammad committed
662
   case STATE:
663
      return (p1->state - p2->state);
Hisham Muhammad's avatar
Hisham Muhammad committed
664
   case NICE:
665
      return (p1->nice - p2->nice);
666
667
668
669
670
671
672
673
   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
674
   case M_SIZE:
675
      return (p2->m_size - p1->m_size);
Hisham Muhammad's avatar
Hisham Muhammad committed
676
   case M_RESIDENT:
677
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
678
   case M_SHARE:
679
      return (p2->m_share - p1->m_share);
Hisham Muhammad's avatar
Hisham Muhammad committed
680
   case PERCENT_CPU:
681
      return (p2->percent_cpu > p1->percent_cpu ? 1 : -1);
Hisham Muhammad's avatar
Hisham Muhammad committed
682
   case PERCENT_MEM:
683
      return (p2->m_resident - p1->m_resident);
Hisham Muhammad's avatar
Hisham Muhammad committed
684
   case UTIME:
685
      return (p2->utime - p1->utime);
Hisham Muhammad's avatar
Hisham Muhammad committed
686
   case STIME:
687
      return (p2->stime - p1->stime);
Hisham Muhammad's avatar
Hisham Muhammad committed
688
   case TIME:
689
      return ((p2->utime+p2->stime) - (p1->utime+p1->stime));
Hisham Muhammad's avatar
Hisham Muhammad committed
690
   case COMM:
691
      return strcmp(p1->comm, p2->comm);
692
693
   case NLWP:
      return (p1->nlwp - p2->nlwp);
694
695
696
697
698
699
   case STARTTIME: {
      if (p1->starttime_ctime == p2->starttime_ctime)
         return (p1->pid - p2->pid);
      else
         return (p1->starttime_ctime - p2->starttime_ctime);
   }
700
   #ifdef HAVE_OPENVZ
701
702
   case CTID:
      return (p1->ctid - p2->ctid);
703
704
705
   case VPID:
      return (p1->vpid - p2->vpid);
   #endif
Hisham Muhammad's avatar
Hisham Muhammad committed
706
707
708
709
   #ifdef HAVE_VSERVER
   case VXID:
      return (p1->vxid - p2->vxid);
   #endif
710
711
712
713
714
715
716
717
718
719
   #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;
720
   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;
721
   #endif
722
723
724
725
   #ifdef HAVE_CGROUP
   case CGROUP:
      return strcmp(p1->cgroup ? p1->cgroup : "", p2->cgroup ? p2->cgroup : "");
   #endif
726

Hisham Muhammad's avatar
Hisham Muhammad committed
727
   default:
728
      return (p1->pid - p2->pid);
Hisham Muhammad's avatar
Hisham Muhammad committed
729
   }
730
731
   test_diff:
   return (diff > 0) ? 1 : (diff < 0 ? -1 : 0);
Hisham Muhammad's avatar
Hisham Muhammad committed
732
}