| /* |
| htop - Process.c |
| (C) 2004-2006 Hisham H. Muhammad |
| Released under the GNU GPL, see the COPYING file |
| in the source distribution for its full text. |
| */ |
| |
| #define _GNU_SOURCE |
| #include "ProcessList.h" |
| #include "Object.h" |
| #include "CRT.h" |
| #include "String.h" |
| #include "Process.h" |
| |
| #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> |
| |
| // This works only with glibc 2.1+. On earlier versions |
| // the behavior is similar to have a hardcoded page size. |
| #define PAGE_SIZE ( sysconf(_SC_PAGESIZE) / 1024 ) |
| |
| #define PROCESS_COMM_LEN 300 |
| |
| /*{ |
| |
| 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, |
| USER, TIME, LAST_PROCESSFIELD |
| } ProcessField; |
| |
| struct ProcessList_; |
| |
| typedef struct Process_ { |
| Object super; |
| |
| struct ProcessList_ *pl; |
| bool updated; |
| |
| int pid; |
| char* comm; |
| int indent; |
| char state; |
| bool tag; |
| int ppid; |
| int pgrp; |
| int session; |
| int tty_nr; |
| int tpgid; |
| unsigned long int flags; |
| #ifdef DEBUG |
| unsigned long int minflt; |
| unsigned long int cminflt; |
| unsigned long int majflt; |
| unsigned long int cmajflt; |
| #endif |
| unsigned long int utime; |
| unsigned long int stime; |
| long int cutime; |
| long int cstime; |
| long int priority; |
| long int nice; |
| #ifdef DEBUG |
| long int itrealvalue; |
| unsigned long int starttime; |
| 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 |
| 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; |
| char* user; |
| } Process; |
| |
| }*/ |
| |
| #ifdef DEBUG |
| char* PROCESS_CLASS = "Process"; |
| #else |
| #define PROCESS_CLASS NULL |
| #endif |
| |
| char *Process_fieldNames[] = { |
| "", "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", "*** report bug! ***" |
| }; |
| |
| static int Process_getuid = -1; |
| |
| Process* Process_new(struct ProcessList_ *pl) { |
| Process* this = malloc(sizeof(Process)); |
| Object_setClass(this, PROCESS_CLASS); |
| ((Object*)this)->display = Process_display; |
| ((Object*)this)->delete = Process_delete; |
| this->pid = 0; |
| this->pl = pl; |
| this->tag = false; |
| this->updated = false; |
| this->utime = 0; |
| this->stime = 0; |
| this->comm = NULL; |
| this->indent = 0; |
| if (Process_getuid == -1) Process_getuid = getuid(); |
| return this; |
| } |
| |
| Process* Process_clone(Process* this) { |
| Process* clone = malloc(sizeof(Process)); |
| memcpy(clone, this, sizeof(Process)); |
| this->comm = NULL; |
| this->pid = 0; |
| return clone; |
| } |
| |
| void Process_delete(Object* cast) { |
| Process* this = (Process*) cast; |
| assert (this != NULL); |
| if (this->comm) free(this->comm); |
| free(this); |
| } |
| |
| void Process_display(Object* cast, RichString* out) { |
| Process* this = (Process*) cast; |
| ProcessField* fields = this->pl->fields; |
| RichString_init(out); |
| for (int i = 0; fields[i]; i++) |
| Process_writeField(this, out, fields[i]); |
| if (this->pl->shadowOtherUsers && 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->len > 0); |
| } |
| |
| void Process_toggleTag(Process* this) { |
| this->tag = this->tag == true ? false : true; |
| } |
| |
| void 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; |
| } |
| } |
| |
| void Process_sendSignal(Process* this, int signal) { |
| kill(this->pid, signal); |
| } |
| |
| #define ONE_K 1024 |
| #define ONE_M (ONE_K * ONE_K) |
| #define ONE_G (ONE_M * ONE_K) |
| |
| static void Process_printLargeNumber(Process* this, RichString *str, unsigned long number) { |
| char buffer[11]; |
| int len; |
| if(number >= (1000 * ONE_M)) { |
| len = snprintf(buffer, 10, "%4.2fG ", (float)number / ONE_M); |
| RichString_appendn(str, CRT_colors[LARGE_NUMBER], buffer, len); |
| } else if(number >= (100000)) { |
| len = snprintf(buffer, 10, "%4dM ", number / ONE_K); |
| 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) { |
| len = snprintf(buffer, 10, "%2d", number/1000); |
| RichString_appendn(str, CRT_colors[PROCESS_MEGABYTES], buffer, len); |
| number %= 1000; |
| len = snprintf(buffer, 10, "%03d ", number); |
| RichString_appendn(str, CRT_colors[PROCESS], buffer, len); |
| } else { |
| len = snprintf(buffer, 10, "%5d ", number); |
| RichString_appendn(str, CRT_colors[PROCESS], buffer, len); |
| } |
| } |
| |
| static double jiffy = 0.0; |
| |
| static void Process_printTime(RichString* str, unsigned long t) { |
| 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); |
| } |
| |
| static inline void Process_writeCommand(Process* this, int attr, RichString* str) { |
| if (this->pl->highlightBaseName) { |
| char* firstSpace = strchr(this->comm, ' '); |
| if (firstSpace) { |
| char* slash = firstSpace; |
| while (slash > this->comm && *slash != '/') |
| slash--; |
| if (slash > this->comm) { |
| slash++; |
| RichString_appendn(str, attr, this->comm, slash - this->comm); |
| } |
| RichString_appendn(str, CRT_colors[PROCESS_BASENAME], slash, firstSpace - slash); |
| RichString_append(str, attr, firstSpace); |
| } else { |
| RichString_append(str, CRT_colors[PROCESS_BASENAME], this->comm); |
| } |
| } else { |
| RichString_append(str, attr, this->comm); |
| } |
| } |
| |
| void Process_writeField(Process* this, RichString* str, ProcessField field) { |
| char buffer[PROCESS_COMM_LEN]; |
| int attr = CRT_colors[DEFAULT_COLOR]; |
| int n = PROCESS_COMM_LEN; |
| |
| switch (field) { |
| case PID: snprintf(buffer, n, "%5d ", this->pid); break; |
| case PPID: snprintf(buffer, n, "%5d ", this->ppid); break; |
| case PGRP: snprintf(buffer, n, "%5d ", this->pgrp); break; |
| case SESSION: snprintf(buffer, n, "%5d ", this->session); break; |
| case TTY_NR: snprintf(buffer, n, "%5d ", this->tty_nr); break; |
| case TPGID: snprintf(buffer, n, "%5d ", this->tpgid); break; |
| case PROCESSOR: snprintf(buffer, n, "%3d ", this->processor+1); break; |
| case COMM: { |
| if (!this->pl->treeView || this->indent == 0) { |
| Process_writeCommand(this, attr, str); |
| return; |
| } else { |
| char* buf = buffer; |
| int maxIndent = 0; |
| for (int i = 0; i < 32; i++) |
| if (this->indent & (1 << i)) |
| maxIndent = i+1; |
| for (int i = 0; i < maxIndent - 1; i++) { |
| if (this->indent & (1 << i)) |
| snprintf(buf, n, " | "); |
| else |
| snprintf(buf, n, " "); |
| buf += 4; |
| n -= 4; |
| } |
| if (this->pl->direction == 1) |
| snprintf(buf, n, " `- "); |
| else |
| snprintf(buf, n, " ,- "); |
| RichString_append(str, CRT_colors[PROCESS_TREE], buffer); |
| Process_writeCommand(this, attr, str); |
| return; |
| } |
| } |
| 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; |
| } |
| case M_SIZE: Process_printLargeNumber(this, str, this->m_size * PAGE_SIZE); return; |
| case M_RESIDENT: Process_printLargeNumber(this, str, this->m_resident * PAGE_SIZE); return; |
| case M_SHARE: Process_printLargeNumber(this, str, this->m_share * PAGE_SIZE); return; |
| case ST_UID: snprintf(buffer, n, "%4d ", this->st_uid); break; |
| case USER: { |
| if (Process_getuid != this->st_uid) |
| attr = CRT_colors[PROCESS_SHADOW]; |
| if (this->user) { |
| snprintf(buffer, n, "%-8s ", this->user); |
| } else { |
| snprintf(buffer, n, "%-8d ", this->st_uid); |
| } |
| if (buffer[8] != '\0') { |
| buffer[8] = ' '; |
| buffer[9] = '\0'; |
| } |
| 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: { |
| if (this->percent_cpu > 99.9) { |
| snprintf(buffer, n, "100. "); |
| } 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; |
| } |
| default: |
| snprintf(buffer, n, "- "); |
| } |
| RichString_append(str, attr, buffer); |
| return; |
| } |
| |
| int Process_pidCompare(const void* v1, const void* v2) { |
| Process* p1 = (Process*)v1; |
| Process* p2 = (Process*)v2; |
| 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; |
| } |
| switch (pl->sortKey) { |
| case PID: |
| return (p1->pid - p2->pid); |
| case PPID: |
| return (p1->ppid - p2->ppid); |
| case USER: |
| return strcmp(p1->user, p2->user); |
| case PRIORITY: |
| return (p1->priority - p2->priority); |
| case STATE: |
| return (p1->state - p2->state); |
| case NICE: |
| return (p1->nice - p2->nice); |
| case M_SIZE: |
| return (p2->m_size - p1->m_size); |
| case M_RESIDENT: |
| return (p2->m_resident - p1->m_resident); |
| case M_SHARE: |
| return (p2->m_share - p1->m_share); |
| case PERCENT_CPU: |
| return (p2->percent_cpu > p1->percent_cpu ? 1 : -1); |
| case PERCENT_MEM: |
| return (p2->m_resident - p1->m_resident); |
| case UTIME: |
| return (p2->utime - p1->utime); |
| case STIME: |
| return (p2->stime - p1->stime); |
| case TIME: |
| return ((p2->utime+p2->stime) - (p1->utime+p1->stime)); |
| case COMM: |
| return strcmp(p1->comm, p2->comm); |
| default: |
| return (p1->pid - p2->pid); |
| } |
| |
| } |
| |
| char* Process_printField(ProcessField field) { |
| switch (field) { |
| case PID: return " PID "; |
| case PPID: return " PPID "; |
| case PGRP: return " PGRP "; |
| case SESSION: return " SESN "; |
| case TTY_NR: return " TTY "; |
| case TPGID: return " TGID "; |
| case COMM: return "Command "; |
| case STATE: return "S "; |
| case PRIORITY: return "PRI "; |
| case NICE: return " NI "; |
| case M_SIZE: return " VIRT "; |
| case M_RESIDENT: return " RES "; |
| case M_SHARE: return " SHR "; |
| case ST_UID: return " UID "; |
| case USER: return "USER "; |
| case UTIME: return " UTIME+ "; |
| case STIME: return " STIME+ "; |
| case TIME: return " TIME+ "; |
| case PERCENT_CPU: return "CPU% "; |
| case PERCENT_MEM: return "MEM% "; |
| case PROCESSOR: return "CPU "; |
| default: return "- "; |
| } |
| } |