From d3c9975943df58e293359b87905d19ff1fd52061 Mon Sep 17 00:00:00 2001 From: Daniel Lange Date: Mon, 11 Apr 2016 13:00:15 +0200 Subject: Imported Upstream version 0.5 --- Process.c | 425 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 425 insertions(+) create mode 100644 Process.c (limited to 'Process.c') diff --git a/Process.c b/Process.c new file mode 100644 index 0000000..4483325 --- /dev/null +++ b/Process.c @@ -0,0 +1,425 @@ +/* +htop +(C) 2004 Hisham H. Muhammad +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" + +#include "debug.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +// TODO: wtf!? +int kill(pid_t pid, int signal); + +// 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 +#define PROCESS_USER_LEN 10 + +/*{ + +typedef enum ProcessField_ { + PID, 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, 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, LAST_PROCESSFIELD +} ProcessField; + +struct ProcessList_; + +typedef struct Process_ { + Object super; + + struct ProcessList_ *pl; + bool updated; + + int pid; + char comm[PROCESS_COMM_LEN + 2]; + int indent; + char state; + bool tag; + int ppid; + int pgrp; + int session; + int tty_nr; + int tpgid; + unsigned long int flags; + unsigned long int minflt; + unsigned long int cminflt; + unsigned long int majflt; + unsigned long int cmajflt; + unsigned long int utime; + unsigned long int stime; + long int cutime; + long int cstime; + long int priority; + long int nice; + 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; + 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_USER_LEN + 1]; +} Process; + +extern char* PROCESS_CLASS; + +extern char* Process_fieldNames[]; + +}*/ + +/* private property */ +char* PROCESS_CLASS = "Process"; + +/* private property */ +char *Process_fieldNames[] = { "PID", "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", "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", "LAST_PROCESSFIELD"}; + +Process* Process_new(struct ProcessList_ *pl) { + Process* this = malloc(sizeof(Process)); + ((Object*)this)->class = PROCESS_CLASS; + ((Object*)this)->display = Process_display; + ((Object*)this)->equals = Process_equals; + ((Object*)this)->delete = Process_delete; + this->pl = pl; + this->tag = false; + this->updated = false; + this->utime = 0; + this->stime = 0; + return this; +} + +Process* Process_clone(Process* this) { + Process* clone = malloc(sizeof(Process)); + memcpy(clone, this, sizeof(Process)); + return clone; +} + +void Process_delete(Object* cast) { + Process* this = (Process*) cast; + assert (this != NULL); + free(this); +} + +void Process_display(Object* cast, RichString* out) { + Process* this = (Process*) cast; + ProcessField* fields = this->pl->fields; + RichString_prune(out); + for (int i = 0; fields[i] != LAST_PROCESSFIELD; i++) + Process_writeField(this, out, fields[i]); + if (this->pl->shadowOtherUsers && this->st_uid != 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; +} + +bool Process_equals(const Object* o1, const Object* o2) { + Process* p1 = (Process*) o1; + Process* p2 = (Process*) o2; + if (p1 == NULL || p2 == NULL) + return false; + return (p1->pid == p2->pid); +} + +void Process_setPriority(Process* this, int priority) { + int err = setpriority(PRIO_PROCESS, this->pid, priority); + if (err == 0) { + 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) + +/* private */ +void Process_printLargeNumber(Process* this, RichString *str, unsigned int number) { + char buffer[10]; + if(number >= (1000 * ONE_M)) { + snprintf(buffer, 10, "%4.2fG ", (float)number / ONE_M); + RichString_append(str, CRT_colors[LARGE_NUMBER], buffer); + } else if(number >= (100000)) { + snprintf(buffer, 10, "%4dM ", number / ONE_K); + int attr = this->pl->highlightMegabytes + ? CRT_colors[PROCESS_MEGABYTES] + : CRT_colors[PROCESS]; + RichString_append(str, attr, buffer); + } else if (this->pl->highlightMegabytes && number >= 1000) { + snprintf(buffer, 10, "%2d", number/1000); + RichString_append(str, CRT_colors[PROCESS_MEGABYTES], buffer); + number %= 1000; + snprintf(buffer, 10, "%03d ", number); + RichString_append(str, CRT_colors[PROCESS], buffer); + } else { + snprintf(buffer, 10, "%5d ", number); + RichString_append(str, CRT_colors[PROCESS], buffer); + } +} + +/* private property */ +double jiffy = 0.0; + +/* private */ +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[10]; + 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); +} + +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 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 (getuid() != this->st_uid) + attr = CRT_colors[PROCESS_SHADOW]; + snprintf(buffer, n, "%-8s ", this->user); + 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_compare(const Object* v1, const Object* v2) { + Process* p1 = (Process*)v1; + Process* p2 = (Process*)v2; + int direction = p1->pl->direction; + switch (p1->pl->sortKey) { + case PID: + return (p2->pid - p1->pid) * direction; + case PPID: + return (p2->ppid - p1->ppid) * direction; + case USER: + return strcmp(p2->user, p1->user) * direction; + case PRIORITY: + return (p2->priority - p1->priority) * direction; + case STATE: + return (p2->state - p1->state) * direction; + case NICE: + return (p2->nice - p1->nice) * direction; + case M_SIZE: + return (p1->m_size - p2->m_size) * direction; + case M_RESIDENT: + return (p1->m_resident - p2->m_resident) * direction; + case M_SHARE: + return (p1->m_share - p2->m_share) * direction; + case PERCENT_CPU: + return (p1->percent_cpu < p2->percent_cpu ? -1 : 1) * direction; + case PERCENT_MEM: + return (p1->percent_mem < p2->percent_mem ? -1 : 1) * direction; + case UTIME: + return (p1->utime - p2->utime) * direction; + case STIME: + return (p1->stime - p2->stime) * direction; + case TIME: + return ((p1->utime+p1->stime) - (p2->utime+p2->stime)) * direction; + case COMM: + return strcmp(p2->comm, p1->comm) * direction; + default: + return (p2->pid - p1->pid) * direction; + } +} + +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 "- "; + } +} -- cgit v1.2.3