-/* Conky, a system monitor, based on torsmo
+/* -*- mode: c; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
+ * vim: ts=4 sw=4 noet ai cindent syntax=c
+ *
+ * Conky, a system monitor, based on torsmo
*
* Any original torsmo code is licensed under the BSD license
*
*
* Copyright (c) 2005 Adi Zaimi, Dan Piponi <dan@tanelorn.demon.co.uk>,
* Dave Clark <clarkd@skynet.ca>
- * Copyright (c) 2005-2009 Brenden Matthews, Philip Kovacs, et. al.
+ * Copyright (c) 2005-2010 Brenden Matthews, Philip Kovacs, et. al.
* (see AUTHORS)
* All rights reserved.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
- * vim: ts=4 sw=4 noet ai cindent syntax=c
- *
*/
#include "top.h"
+#include "logging.h"
static unsigned long g_time = 0;
static unsigned long long previous_total = 0;
static struct process *first_process = 0;
+/* a simple hash table to speed up find_process() */
+struct proc_hash_entry {
+ struct proc_hash_entry *next;
+ struct process *proc;
+};
+static struct proc_hash_entry proc_hash_table[256];
+
+static void hash_process(struct process *p)
+{
+ struct proc_hash_entry *phe;
+ static char first_run = 1;
+
+ /* better make sure all next pointers are zero upon first access */
+ if (first_run) {
+ memset(proc_hash_table, 0, sizeof(struct proc_hash_entry) * 256);
+ first_run = 0;
+ }
+
+ /* get the bucket head */
+ phe = &proc_hash_table[p->pid % 256];
+
+ /* find the bucket's end */
+ while (phe->next)
+ phe = phe->next;
+
+ /* append process */
+ phe->next = malloc(sizeof(struct proc_hash_entry));
+ memset(phe->next, 0, sizeof(struct proc_hash_entry));
+ phe->next->proc = p;
+}
+
+static void unhash_process(struct process *p)
+{
+ struct proc_hash_entry *phe, *tmp;
+
+ /* get the bucket head */
+ phe = &proc_hash_table[p->pid % 256];
+
+ /* find the entry pointing to p and drop it */
+ while (phe->next) {
+ if (phe->next->proc == p) {
+ tmp = phe->next;
+ phe->next = phe->next->next;
+ free(tmp);
+ return;
+ }
+ phe = phe->next;
+ }
+}
+
+static void __unhash_all_processes(struct proc_hash_entry *phe)
+{
+ if (phe->next)
+ __unhash_all_processes(phe->next);
+ free(phe->next);
+}
+
+static void unhash_all_processes(void)
+{
+ int i;
+
+ for (i = 0; i < 256; i++) {
+ __unhash_all_processes(&proc_hash_table[i]);
+ proc_hash_table[i].next = NULL;
+ }
+}
+
struct process *get_first_process(void)
{
return first_process;
pr = next;
}
first_process = NULL;
+
+ /* drop the whole hash table */
+ unhash_all_processes();
}
struct process *get_process_by_name(const char *name)
struct process *p = first_process;
while (p) {
- if (!strcmp(p->name, name))
+ if (p->name && !strcmp(p->name, name))
return p;
p = p->next;
}
static struct process *find_process(pid_t pid)
{
- struct process *p = first_process;
+ struct proc_hash_entry *phe;
- while (p) {
- if (p->pid == pid) {
- return p;
- }
- p = p->next;
+ phe = &proc_hash_table[pid % 256];
+ while (phe->next) {
+ if (phe->next->proc->pid == pid)
+ return phe->next->proc;
+ phe = phe->next;
}
return 0;
}
#ifdef IOSTATS
process->previous_read_bytes = ULLONG_MAX;
process->previous_write_bytes = ULLONG_MAX;
-#endif
+#endif /* IOSTATS */
process->counted = 1;
/* process_find_name(process); */
+ /* add the process to the hash table */
+ hash_process(process);
+
return process;
}
* Anyone hoping to port wmtop should look here first. */
static int process_parse_stat(struct process *process)
{
- struct information *cur = &info;
char line[BUFFER_LEN] = { 0 }, filename[BUFFER_LEN], procname[BUFFER_LEN];
+ char state[4];
int ps;
unsigned long user_time = 0;
unsigned long kernel_time = 0;
rc = MIN((unsigned)(rparen - lparen - 1), sizeof(procname) - 1);
strncpy(procname, lparen + 1, rc);
procname[rc] = '\0';
- rc = sscanf(rparen + 1, "%*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %lu "
- "%lu %*s %*s %*s %d %*s %*s %*s %u %u", &process->user_time,
- &process->kernel_time, &nice_val, &process->vsize, &process->rss);
- if (rc < 5) {
+ rc = sscanf(rparen + 1, "%3s %*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %lu "
+ "%lu %*s %*s %*s %d %*s %*s %*s %u %u", state, &process->user_time,
+ &process->kernel_time, &nice_val, &process->vsize, &process->rss);
+ if (rc < 6) {
+ NORM_ERR("scaning data for %s failed, got only %d fields", procname, rc);
return 1;
}
+
+ if(state[0]=='R')
+ ++ info.run_procs;
+
/* remove any "kdeinit: " */
if (procname == strstr(procname, "kdeinit")) {
snprintf(filename, sizeof(filename), PROCFS_CMDLINE_TEMPLATE,
- process->pid);
+ process->pid);
ps = open(filename, O_RDONLY);
if (ps < 0) {
process->name = strndup(procname, text_buffer_size);
process->rss *= getpagesize();
- if (!cur->memmax) {
- update_total_processes();
- }
-
process->total_cpu_time = process->user_time + process->kernel_time;
- process->totalmem = (float) (((float) process->rss / cur->memmax) / 10);
if (process->previous_user_time == ULONG_MAX) {
process->previous_user_time = process->user_time;
}
process->previous_kernel_time = process->kernel_time;
}
+ /* strangely, the values aren't monotonous */
+ if (process->previous_user_time > process->user_time)
+ process->previous_user_time = process->user_time;
+
+ if (process->previous_kernel_time > process->kernel_time)
+ process->previous_kernel_time = process->kernel_time;
+
/* store the difference of the user_time */
user_time = process->user_time - process->previous_user_time;
kernel_time = process->kernel_time - process->previous_kernel_time;
return 0;
}
-#endif
+#endif /* IOSTATS */
/******************************************
* Get process structure for process pid *
/* compute each process cpu usage by reading /proc/<proc#>/stat */
rc = process_parse_stat(process);
- if (rc)
- return 1;
+ if (rc) return 1;
/* rc = process_parse_statm(process); if (rc) return 1; */
#ifdef IOSTATS
rc = process_parse_io(process);
- if (rc)
- return 1;
-#endif
+ if (rc) return 1;
+#endif /* IOSTATS */
/*
* Check name against the exclusion list
return 1;
}
+ info.run_procs = 0;
++g_time;
/* Get list of processes from /proc directory */
if (p->name) {
free(p->name);
}
+ /* remove the process from the hash table */
+ unhash_process(p);
free(p);
}
}
sscanf(line, template, &cpu, &niceval, &systemval, &idle, &iowait, &irq,
- &softirq, &steal);
+ &softirq, &steal);
total = cpu + niceval + systemval + idle + iowait + irq + softirq + steal;
t = total - previous_total;
{
struct process *p;
unsigned long long sum = 0;
-
+
for (p = first_process; p; p = p->next)
sum += p->read_bytes + p->write_bytes;
for (p = first_process; p; p = p->next)
p->io_perc = 100.0 * (p->read_bytes + p->write_bytes) / (float) sum;
}
-#endif
+#endif /* IOSTATS */
/******************************************
* Find the top processes *
{
struct sorted_process *sp;
sp = malloc(sizeof(struct sorted_process));
- sp->greater = NULL;
- sp->less = NULL;
+ memset(sp, 0, sizeof(struct sorted_process));
sp->proc = proc;
return sp;
}
/* mem comparison function for insert_sp_element */
static int compare_mem(struct process *a, struct process *b)
{
- if (a->totalmem < b->totalmem) {
+ if (a->rss < b->rss) {
return 1;
- } else if (a->totalmem > b->totalmem) {
+ } else if (a->rss > b->rss) {
return -1;
} else {
return 0;
return 0;
}
}
-#endif
+#endif /* IOSTATS */
/* insert this process into the list in a sorted fashion,
* or destroy it if it doesn't fit on the list */
struct process **ptime
#ifdef IOSTATS
, struct process **io
-#endif
+#endif /* IOSTATS */
)
{
struct sorted_process *spc_head = NULL, *spc_tail = NULL, *spc_cur = NULL;
struct sorted_process *spt_head = NULL, *spt_tail = NULL, *spt_cur = NULL;
#ifdef IOSTATS
struct sorted_process *spi_head = NULL, *spi_tail = NULL, *spi_cur = NULL;
-#endif
+#endif /* IOSTATS */
struct process *cur_proc = NULL;
unsigned long long total = 0;
if (!top_cpu && !top_mem && !top_time
#ifdef IOSTATS
&& !top_io
-#endif
+#endif /* IOSTATS */
+ && !top_running
) {
return;
}
process_cleanup(); /* cleanup list from exited processes */
#ifdef IOSTATS
calc_io_each(); /* percentage of I/O for each task */
-#endif
+#endif /* IOSTATS */
cur_proc = first_process;
if (top_cpu) {
spc_cur = malloc_sp(cur_proc);
insert_sp_element(spc_cur, &spc_head, &spc_tail, MAX_SP,
- &compare_cpu);
+ &compare_cpu);
}
if (top_mem) {
spm_cur = malloc_sp(cur_proc);
insert_sp_element(spm_cur, &spm_head, &spm_tail, MAX_SP,
- &compare_mem);
+ &compare_mem);
}
if (top_time) {
spt_cur = malloc_sp(cur_proc);
insert_sp_element(spt_cur, &spt_head, &spt_tail, MAX_SP,
- &compare_time);
+ &compare_time);
}
#ifdef IOSTATS
if (top_io) {
spi_cur = malloc_sp(cur_proc);
insert_sp_element(spi_cur, &spi_head, &spi_tail, MAX_SP,
- &compare_io);
+ &compare_io);
}
-#endif
+#endif /* IOSTATS */
cur_proc = cur_proc->next;
}
- if (top_cpu)
- sp_acopy(spc_head, cpu, MAX_SP);
- if (top_mem)
- sp_acopy(spm_head, mem, MAX_SP);
- if (top_time)
- sp_acopy(spt_head, ptime, MAX_SP);
+ if (top_cpu) sp_acopy(spc_head, cpu, MAX_SP);
+ if (top_mem) sp_acopy(spm_head, mem, MAX_SP);
+ if (top_time) sp_acopy(spt_head, ptime, MAX_SP);
+#ifdef IOSTATS
+ if (top_io) sp_acopy(spi_head, io, MAX_SP);
+#endif /* IOSTATS */
+}
+
+struct top_data {
+ int num;
+ int type;
+ int was_parsed;
+ char *s;
+};
+
+int parse_top_args(const char *s, const char *arg, struct text_object *obj)
+{
+ struct top_data *td;
+ char buf[64];
+ int n;
+
+ if (!arg) {
+ NORM_ERR("top needs arguments");
+ return 0;
+ }
+
+ if (obj->data.opaque) {
+ return 1;
+ }
+
+ if (s[3] == 0) {
+ obj->type = OBJ_top;
+ top_cpu = 1;
+ } else if (strcmp(&s[3], "_mem") == EQUAL) {
+ obj->type = OBJ_top_mem;
+ top_mem = 1;
+ } else if (strcmp(&s[3], "_time") == EQUAL) {
+ obj->type = OBJ_top_time;
+ top_time = 1;
+#ifdef IOSTATS
+ } else if (strcmp(&s[3], "_io") == EQUAL) {
+ obj->type = OBJ_top_io;
+ top_io = 1;
+#endif /* IOSTATS */
+ } else {
+#ifdef IOSTATS
+ NORM_ERR("Must be top, top_mem, top_time or top_io");
+#else /* IOSTATS */
+ NORM_ERR("Must be top, top_mem or top_time");
+#endif /* IOSTATS */
+ return 0;
+ }
+
+ obj->data.opaque = td = malloc(sizeof(struct top_data));
+ memset(td, 0, sizeof(struct top_data));
+ td->s = strndup(arg, text_buffer_size);
+
+ if (sscanf(arg, "%63s %i", buf, &n) == 2) {
+ if (strcmp(buf, "name") == EQUAL) {
+ td->type = TOP_NAME;
+ } else if (strcmp(buf, "cpu") == EQUAL) {
+ td->type = TOP_CPU;
+ } else if (strcmp(buf, "pid") == EQUAL) {
+ td->type = TOP_PID;
+ } else if (strcmp(buf, "mem") == EQUAL) {
+ td->type = TOP_MEM;
+ } else if (strcmp(buf, "time") == EQUAL) {
+ td->type = TOP_TIME;
+ } else if (strcmp(buf, "mem_res") == EQUAL) {
+ td->type = TOP_MEM_RES;
+ } else if (strcmp(buf, "mem_vsize") == EQUAL) {
+ td->type = TOP_MEM_VSIZE;
+#ifdef IOSTATS
+ } else if (strcmp(buf, "io_read") == EQUAL) {
+ td->type = TOP_READ_BYTES;
+ } else if (strcmp(buf, "io_write") == EQUAL) {
+ td->type = TOP_WRITE_BYTES;
+ } else if (strcmp(buf, "io_perc") == EQUAL) {
+ td->type = TOP_IO_PERC;
+#endif /* IOSTATS */
+ } else {
+ NORM_ERR("invalid type arg for top");
+#ifdef IOSTATS
+ NORM_ERR("must be one of: name, cpu, pid, mem, time, mem_res, mem_vsize, "
+ "io_read, io_write, io_perc");
+#else /* IOSTATS */
+ NORM_ERR("must be one of: name, cpu, pid, mem, time, mem_res, mem_vsize");
+#endif /* IOSTATS */
+ free(td->s);
+ free(obj->data.opaque);
+ return 0;
+ }
+ if (n < 1 || n > 10) {
+ NORM_ERR("invalid num arg for top. Must be between 1 and 10.");
+ free(td->s);
+ free(obj->data.opaque);
+ return 0;
+ } else {
+ td->num = n - 1;
+ }
+ } else {
+ NORM_ERR("invalid argument count for top");
+ free(td->s);
+ free(obj->data.opaque);
+ return 0;
+ }
+ return 1;
+}
+
+static char *format_time(unsigned long timeval, const int width)
+{
+ char buf[10];
+ unsigned long nt; // narrow time, for speed on 32-bit
+ unsigned cc; // centiseconds
+ unsigned nn; // multi-purpose whatever
+
+ nt = timeval;
+ cc = nt % 100; // centiseconds past second
+ nt /= 100; // total seconds
+ nn = nt % 60; // seconds past the minute
+ nt /= 60; // total minutes
+ if (width >= snprintf(buf, sizeof buf, "%lu:%02u.%02u",
+ nt, nn, cc)) {
+ return strndup(buf, text_buffer_size);
+ }
+ if (width >= snprintf(buf, sizeof buf, "%lu:%02u", nt, nn)) {
+ return strndup(buf, text_buffer_size);
+ }
+ nn = nt % 60; // minutes past the hour
+ nt /= 60; // total hours
+ if (width >= snprintf(buf, sizeof buf, "%lu,%02u", nt, nn)) {
+ return strndup(buf, text_buffer_size);
+ }
+ nn = nt; // now also hours
+ if (width >= snprintf(buf, sizeof buf, "%uh", nn)) {
+ return strndup(buf, text_buffer_size);
+ }
+ nn /= 24; // now days
+ if (width >= snprintf(buf, sizeof buf, "%ud", nn)) {
+ return strndup(buf, text_buffer_size);
+ }
+ nn /= 7; // now weeks
+ if (width >= snprintf(buf, sizeof buf, "%uw", nn)) {
+ return strndup(buf, text_buffer_size);
+ }
+ // well shoot, this outta' fit...
+ return strndup("<inf>", text_buffer_size);
+}
+
+static unsigned int top_name_width = 15;
+
+/* return zero on success, non-zero otherwise */
+int set_top_name_width(const char *s)
+{
+ if (!s)
+ return 0;
+ return !(sscanf(s, "%u", &top_name_width) == 1);
+}
+
+void print_top(struct text_object *obj, char *p, int p_max_size)
+{
+ struct information *cur = &info;
+ struct top_data *td = obj->data.opaque;
+ struct process **needed = 0;
+ int width;
+
+ if (!td)
+ return;
+
+ switch (obj->type) {
+ case OBJ_top:
+ needed = cur->cpu;
+ break;
+ case OBJ_top_mem:
+ needed = cur->memu;
+ break;
+ case OBJ_top_time:
+ needed = cur->time;
+ break;
#ifdef IOSTATS
- if (top_io)
- sp_acopy(spi_head, io,MAX_SP);
+ case OBJ_top_io:
+ needed = cur->io;
+ break;
+#endif /* IOSTATS */
+ default:
+ return;
+ }
+
+
+ if (needed[td->num]) {
+ char *timeval;
+
+ switch (td->type) {
+ case TOP_NAME:
+ if (needed[td->num]->name) {
+ width = MIN(p_max_size, (int)top_name_width + 1);
+ snprintf(p, width + 1, "%-*s", width,
+ needed[td->num]->name);
+ }
+ break;
+ case TOP_CPU:
+ width = MIN(p_max_size, 7);
+ snprintf(p, width, "%6.2f",
+ needed[td->num]->amount);
+ break;
+ case TOP_PID:
+ width = MIN(p_max_size, 6);
+ snprintf(p, width, "%5i",
+ needed[td->num]->pid);
+ break;
+ case TOP_MEM:
+ /* Calculate a percentage of residential mem from total mem available.
+ * Since rss is bytes and memmax kilobytes, dividing by 10 suffices here. */
+ width = MIN(p_max_size, 7);
+ snprintf(p, width, "%6.2f",
+ (float) ((float)needed[td->num]->rss / cur->memmax) / 10);
+ break;
+ case TOP_TIME:
+ width = MIN(p_max_size, 10);
+ timeval = format_time(
+ needed[td->num]->total_cpu_time, 9);
+ snprintf(p, width, "%9s", timeval);
+ free(timeval);
+ break;
+ case TOP_MEM_RES:
+ human_readable(needed[td->num]->rss,
+ p, p_max_size);
+ break;
+ case TOP_MEM_VSIZE:
+ human_readable(needed[td->num]->vsize,
+ p, p_max_size);
+ break;
+#ifdef IOSTATS
+ case TOP_READ_BYTES:
+ human_readable(needed[td->num]->read_bytes / update_interval,
+ p, p_max_size);
+ break;
+ case TOP_WRITE_BYTES:
+ human_readable(needed[td->num]->write_bytes / update_interval,
+ p, p_max_size);
+ break;
+ case TOP_IO_PERC:
+ width = MIN(p_max_size, 7);
+ snprintf(p, width, "%6.2f",
+ needed[td->num]->io_perc);
+ break;
#endif
+ }
+ }
+}
+
+void free_top(struct text_object *obj, int internal)
+{
+ struct top_data *td = obj->data.opaque;
+
+ if (info.first_process && !internal) {
+ free_all_processes();
+ info.first_process = NULL;
+ }
+
+ if (!td)
+ return;
+ if (td->s)
+ free(td->s);
+ free(obj->data.opaque);
+ obj->data.opaque = NULL;
}