| /* |
| htop - Process.c |
| (C) 2004-2015 Hisham H. Muhammad |
| Released under the GNU GPL, see the COPYING file |
| in the source distribution for its full text. |
| */ |
| |
| #include "Process.h" |
| #include "Settings.h" |
| |
| #include "CRT.h" |
| #include "StringUtils.h" |
| #include "RichString.h" |
| #include "Platform.h" |
| |
| #include <stdio.h> |
| #include <sys/time.h> |
| #include <sys/resource.h> |
| #include <sys/param.h> |
| #include <sys/stat.h> |
| #include <unistd.h> |
| #include <stdlib.h> |
| #include <signal.h> |
| #include <string.h> |
| #include <stdbool.h> |
| #include <pwd.h> |
| #include <time.h> |
| #include <assert.h> |
| #include <math.h> |
| |
| #ifdef __ANDROID__ |
| #define SYS_ioprio_get __NR_ioprio_get |
| #define SYS_ioprio_set __NR_ioprio_set |
| #endif |
| |
| // On Linux, this works only with glibc 2.1+. On earlier versions |
| // the behavior is similar to have a hardcoded page size. |
| #ifndef PAGE_SIZE |
| #define PAGE_SIZE ( sysconf(_SC_PAGESIZE) ) |
| #endif |
| #define PAGE_SIZE_KB ( PAGE_SIZE / ONE_K ) |
| |
| /*{ |
| #include "Object.h" |
| |
| #include <sys/types.h> |
| |
| #define PROCESS_FLAG_IO 0x0001 |
| |
| typedef enum ProcessFields { |
| PID = 1, |
| COMM = 2, |
| STATE = 3, |
| PPID = 4, |
| PGRP = 5, |
| SESSION = 6, |
| TTY_NR = 7, |
| TPGID = 8, |
| MINFLT = 10, |
| MAJFLT = 12, |
| PRIORITY = 18, |
| NICE = 19, |
| STARTTIME = 21, |
| PROCESSOR = 38, |
| M_SIZE = 39, |
| M_RESIDENT = 40, |
| ST_UID = 46, |
| PERCENT_CPU = 47, |
| PERCENT_MEM = 48, |
| USER = 49, |
| TIME = 50, |
| NLWP = 51, |
| TGID = 52, |
| } ProcessField; |
| |
| typedef struct ProcessPidColumn_ { |
| int id; |
| char* label; |
| } ProcessPidColumn; |
| |
| typedef struct Process_ { |
| Object super; |
| |
| struct Settings_* settings; |
| |
| unsigned long long int time; |
| pid_t pid; |
| pid_t ppid; |
| pid_t tgid; |
| char* comm; |
| int indent; |
| |
| int basenameOffset; |
| bool updated; |
| |
| char state; |
| bool tag; |
| bool showChildren; |
| bool show; |
| unsigned int pgrp; |
| unsigned int session; |
| unsigned int tty_nr; |
| int tpgid; |
| uid_t st_uid; |
| unsigned long int flags; |
| int processor; |
| |
| float percent_cpu; |
| float percent_mem; |
| char* user; |
| |
| long int priority; |
| long int nice; |
| long int nlwp; |
| char starttime_show[8]; |
| time_t starttime_ctime; |
| |
| long m_size; |
| long m_resident; |
| |
| int exit_signal; |
| |
| unsigned long int minflt; |
| unsigned long int majflt; |
| #ifdef DEBUG |
| 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 |
| |
| } Process; |
| |
| typedef struct ProcessFieldData_ { |
| const char* name; |
| const char* title; |
| const char* description; |
| int flags; |
| } ProcessFieldData; |
| |
| // Implemented in platform-specific code: |
| void Process_writeField(Process* this, RichString* str, ProcessField field); |
| long Process_compare(const void* v1, const void* v2); |
| void Process_delete(Object* cast); |
| bool Process_isThread(Process* this); |
| extern ProcessFieldData Process_fields[]; |
| extern ProcessPidColumn Process_pidColumns[]; |
| extern char Process_pidFormat[20]; |
| |
| typedef Process*(*Process_New)(struct Settings_*); |
| typedef void (*Process_WriteField)(Process*, RichString*, ProcessField); |
| |
| typedef struct ProcessClass_ { |
| const ObjectClass super; |
| const Process_WriteField writeField; |
| } ProcessClass; |
| |
| #define As_Process(this_) ((ProcessClass*)((this_)->super.klass)) |
| |
| }*/ |
| |
| static int Process_getuid = -1; |
| |
| #define ONE_K 1024L |
| #define ONE_M (ONE_K * ONE_K) |
| #define ONE_G (ONE_M * ONE_K) |
| |
| #define ONE_DECIMAL_K 1000L |
| #define ONE_DECIMAL_M (ONE_DECIMAL_K * ONE_DECIMAL_K) |
| #define ONE_DECIMAL_G (ONE_DECIMAL_M * ONE_DECIMAL_K) |
| |
| char Process_pidFormat[20] = "%7u "; |
| |
| static char Process_titleBuffer[20][20]; |
| |
| void Process_setupColumnWidths() { |
| int maxPid = Platform_getMaxPid(); |
| if (maxPid == -1) return; |
| int digits = ceil(log10(maxPid)); |
| assert(digits < 20); |
| for (int i = 0; Process_pidColumns[i].label; i++) { |
| assert(i < 20); |
| sprintf(Process_titleBuffer[i], "%*s ", digits, Process_pidColumns[i].label); |
| Process_fields[Process_pidColumns[i].id].title = Process_titleBuffer[i]; |
| } |
| sprintf(Process_pidFormat, "%%%du ", digits); |
| } |
| |
| void Process_humanNumber(RichString* str, unsigned long number, bool coloring) { |
| char buffer[11]; |
| int len; |
| |
| 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(number >= (10 * ONE_DECIMAL_M)) { |
| #ifdef __LP64__ |
| if(number >= (100 * ONE_DECIMAL_G)) { |
| len = snprintf(buffer, 10, "%4ldT ", number / ONE_G); |
| RichString_appendn(str, largeNumberColor, buffer, len); |
| return; |
| } else if (number >= (1000 * ONE_DECIMAL_M)) { |
| len = snprintf(buffer, 10, "%4.1lfT ", (double)number / ONE_G); |
| RichString_appendn(str, largeNumberColor, buffer, len); |
| return; |
| } |
| #endif |
| if(number >= (100 * ONE_DECIMAL_M)) { |
| len = snprintf(buffer, 10, "%4ldG ", number / ONE_M); |
| RichString_appendn(str, largeNumberColor, buffer, len); |
| return; |
| } |
| len = snprintf(buffer, 10, "%4.1lfG ", (double)number / ONE_M); |
| RichString_appendn(str, largeNumberColor, buffer, len); |
| return; |
| } else if (number >= 100000) { |
| len = snprintf(buffer, 10, "%4ldM ", number / ONE_K); |
| RichString_appendn(str, processMegabytesColor, buffer, len); |
| return; |
| } else if (number >= 1000) { |
| len = snprintf(buffer, 10, "%2ld", number/1000); |
| RichString_appendn(str, processMegabytesColor, buffer, len); |
| number %= 1000; |
| len = snprintf(buffer, 10, "%03lu ", number); |
| RichString_appendn(str, processColor, buffer, len); |
| return; |
| } |
| len = snprintf(buffer, 10, "%5lu ", number); |
| RichString_appendn(str, processColor, buffer, len); |
| } |
| |
| void Process_colorNumber(RichString* str, unsigned long long number, bool coloring) { |
| char buffer[14]; |
| |
| 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]; |
| } |
| |
| if (number > 10000000000) { |
| snprintf(buffer, 13, "%11lld ", number / 1000); |
| RichString_appendn(str, largeNumberColor, buffer, 5); |
| RichString_appendn(str, processMegabytesColor, buffer+5, 3); |
| RichString_appendn(str, processColor, buffer+8, 4); |
| } else { |
| snprintf(buffer, 13, "%11llu ", number); |
| 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); |
| } |
| } |
| |
| void Process_printTime(RichString* str, unsigned long long totalHundredths) { |
| unsigned long long totalSeconds = totalHundredths / 100; |
| |
| unsigned long long hours = totalSeconds / 3600; |
| int minutes = (totalSeconds / 60) % 60; |
| int seconds = totalSeconds % 60; |
| int hundredths = totalHundredths - (totalSeconds * 100); |
| char buffer[11]; |
| if (hours >= 100) { |
| snprintf(buffer, 10, "%7lluh ", hours); |
| RichString_append(str, CRT_colors[LARGE_NUMBER], buffer); |
| } else { |
| 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); |
| } |
| } |
| |
| static inline void Process_writeCommand(Process* this, int attr, int baseattr, RichString* str) { |
| int start = RichString_size(str), finish = 0; |
| char* comm = this->comm; |
| |
| if (this->settings->highlightBaseName || !this->settings->showProgramPath) { |
| int i, basename = 0; |
| for (i = 0; i < this->basenameOffset; i++) { |
| if (comm[i] == '/') { |
| basename = i + 1; |
| } else if (comm[i] == ':') { |
| finish = i + 1; |
| break; |
| } |
| } |
| if (!finish) { |
| if (this->settings->showProgramPath) |
| start += basename; |
| else |
| comm += basename; |
| finish = this->basenameOffset - basename; |
| } |
| finish += start - 1; |
| } |
| |
| RichString_append(str, attr, comm); |
| |
| if (this->settings->highlightBaseName) |
| RichString_setAttrn(str, baseattr, start, finish); |
| } |
| |
| 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); |
| } |
| } |
| |
| void Process_writeField(Process* this, RichString* str, ProcessField field) { |
| char buffer[256]; buffer[255] = '\0'; |
| int attr = CRT_colors[DEFAULT_COLOR]; |
| int baseattr = CRT_colors[PROCESS_BASENAME]; |
| int n = sizeof(buffer) - 1; |
| bool coloring = this->settings->highlightMegabytes; |
| |
| switch (field) { |
| case PERCENT_CPU: { |
| 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); |
| } 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; |
| } |
| case COMM: { |
| if (this->settings->highlightThreads && Process_isThread(this)) { |
| attr = CRT_colors[PROCESS_THREAD]; |
| baseattr = CRT_colors[PROCESS_THREAD_BASENAME]; |
| } |
| if (!this->settings->treeView || this->indent == 0) { |
| Process_writeCommand(this, attr, baseattr, str); |
| return; |
| } else { |
| char* buf = buffer; |
| int maxIndent = 0; |
| bool lastItem = (this->indent < 0); |
| int indent = (this->indent < 0 ? -this->indent : this->indent); |
| |
| for (int i = 0; i < 32; i++) |
| if (indent & (1 << i)) |
| maxIndent = i+1; |
| for (int i = 0; i < maxIndent - 1; i++) { |
| int written; |
| if (indent & (1 << i)) |
| written = snprintf(buf, n, "%s ", CRT_treeStr[TREE_STR_VERT]); |
| else |
| written = snprintf(buf, n, " "); |
| buf += written; |
| n -= written; |
| } |
| 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] ); |
| RichString_append(str, CRT_colors[PROCESS_TREE], buffer); |
| Process_writeCommand(this, attr, baseattr, str); |
| return; |
| } |
| } |
| case MAJFLT: Process_colorNumber(str, this->majflt, coloring); return; |
| case MINFLT: Process_colorNumber(str, this->minflt, coloring); return; |
| case M_RESIDENT: Process_humanNumber(str, this->m_resident * PAGE_SIZE_KB, coloring); return; |
| case M_SIZE: Process_humanNumber(str, this->m_size * PAGE_SIZE_KB, coloring); return; |
| 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; |
| } |
| case NLWP: snprintf(buffer, n, "%4ld ", this->nlwp); break; |
| case PGRP: snprintf(buffer, n, Process_pidFormat, this->pgrp); break; |
| case PID: snprintf(buffer, n, Process_pidFormat, this->pid); break; |
| case PPID: snprintf(buffer, n, Process_pidFormat, this->ppid); break; |
| case PRIORITY: { |
| if(this->priority == -100) |
| snprintf(buffer, n, " RT "); |
| else |
| snprintf(buffer, n, "%3ld ", this->priority); |
| break; |
| } |
| case PROCESSOR: snprintf(buffer, n, "%3d ", Settings_cpuId(this->settings, this->processor)); break; |
| case SESSION: snprintf(buffer, n, Process_pidFormat, this->session); break; |
| case STARTTIME: snprintf(buffer, n, "%s", this->starttime_show); break; |
| case STATE: { |
| snprintf(buffer, n, "%c ", this->state); |
| switch(this->state) { |
| case 'R': |
| attr = CRT_colors[PROCESS_R_STATE]; |
| break; |
| case 'D': |
| attr = CRT_colors[PROCESS_D_STATE]; |
| break; |
| } |
| break; |
| } |
| case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break; |
| case TIME: Process_printTime(str, this->time); return; |
| case TGID: snprintf(buffer, n, Process_pidFormat, this->tgid); break; |
| case TPGID: snprintf(buffer, n, Process_pidFormat, this->tpgid); break; |
| case TTY_NR: snprintf(buffer, n, "%5u ", this->tty_nr); break; |
| case USER: { |
| if (Process_getuid != (int) this->st_uid) |
| attr = CRT_colors[PROCESS_SHADOW]; |
| if (this->user) { |
| snprintf(buffer, n, "%-9s ", this->user); |
| } else { |
| snprintf(buffer, n, "%-9d ", this->st_uid); |
| } |
| if (buffer[9] != '\0') { |
| buffer[9] = ' '; |
| buffer[10] = '\0'; |
| } |
| break; |
| } |
| default: |
| snprintf(buffer, n, "- "); |
| } |
| RichString_append(str, attr, buffer); |
| } |
| |
| void Process_display(Object* cast, RichString* out) { |
| Process* this = (Process*) cast; |
| ProcessField* fields = this->settings->fields; |
| RichString_prune(out); |
| for (int i = 0; fields[i]; i++) |
| As_Process(this)->writeField(this, out, fields[i]); |
| if (this->settings->shadowOtherUsers && (int)this->st_uid != Process_getuid) |
| RichString_setAttr(out, CRT_colors[PROCESS_SHADOW]); |
| if (this->tag == true) |
| RichString_setAttr(out, CRT_colors[PROCESS_TAG]); |
| assert(out->chlen > 0); |
| } |
| |
| void Process_done(Process* this) { |
| assert (this != NULL); |
| free(this->comm); |
| } |
| |
| ProcessClass Process_class = { |
| .super = { |
| .extends = Class(Object), |
| .display = Process_display, |
| .delete = Process_delete, |
| .compare = Process_compare |
| }, |
| .writeField = Process_writeField, |
| }; |
| |
| void Process_init(Process* this, struct Settings_* settings) { |
| this->settings = settings; |
| this->tag = false; |
| this->showChildren = true; |
| this->show = true; |
| this->updated = false; |
| this->basenameOffset = -1; |
| 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); |
| } |
| |
| 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); |
| } |
| |
| long Process_pidCompare(const void* v1, const void* v2) { |
| Process* p1 = (Process*)v1; |
| Process* p2 = (Process*)v2; |
| return (p1->pid - p2->pid); |
| } |
| |
| long Process_compare(const void* v1, const void* v2) { |
| Process *p1, *p2; |
| Settings *settings = ((Process*)v1)->settings; |
| if (settings->direction == 1) { |
| p1 = (Process*)v1; |
| p2 = (Process*)v2; |
| } else { |
| p2 = (Process*)v1; |
| p1 = (Process*)v2; |
| } |
| switch (settings->sortKey) { |
| case PERCENT_CPU: |
| return (p2->percent_cpu > p1->percent_cpu ? 1 : -1); |
| case PERCENT_MEM: |
| return (p2->m_resident - p1->m_resident); |
| case COMM: |
| return strcmp(p1->comm, p2->comm); |
| case MAJFLT: |
| return (p2->majflt - p1->majflt); |
| case MINFLT: |
| return (p2->minflt - p1->minflt); |
| case M_RESIDENT: |
| return (p2->m_resident - p1->m_resident); |
| case M_SIZE: |
| return (p2->m_size - p1->m_size); |
| case NICE: |
| return (p1->nice - p2->nice); |
| case NLWP: |
| return (p1->nlwp - p2->nlwp); |
| case PGRP: |
| return (p1->pgrp - p2->pgrp); |
| case PID: |
| return (p1->pid - p2->pid); |
| case PPID: |
| return (p1->ppid - p2->ppid); |
| case PRIORITY: |
| return (p1->priority - p2->priority); |
| case PROCESSOR: |
| return (p1->processor - p2->processor); |
| case SESSION: |
| return (p1->session - p2->session); |
| case STARTTIME: { |
| if (p1->starttime_ctime == p2->starttime_ctime) |
| return (p1->pid - p2->pid); |
| else |
| return (p1->starttime_ctime - p2->starttime_ctime); |
| } |
| case STATE: |
| return (p1->state - p2->state); |
| case ST_UID: |
| return (p1->st_uid - p2->st_uid); |
| case TIME: |
| return ((p2->time) - (p1->time)); |
| case TGID: |
| return (p1->tgid - p2->tgid); |
| case TPGID: |
| return (p1->tpgid - p2->tpgid); |
| case TTY_NR: |
| return (p1->tty_nr - p2->tty_nr); |
| case USER: |
| return strcmp(p1->user ? p1->user : "", p2->user ? p2->user : ""); |
| default: |
| return (p1->pid - p2->pid); |
| } |
| } |