1 /* -*- mode: c; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
2 * vim: ts=4 sw=4 noet ai cindent syntax=c
4 * Conky, a system monitor, based on torsmo
6 * Any original torsmo code is licensed under the BSD license
8 * All code written since the fork of torsmo is licensed under the GPL
10 * Please see COPYING for details
12 * Copyright (c) 2004, Hannu Saransaari and Lauri Hakkarainen
13 * Copyright (c) 2005-2010 Brenden Matthews, Philip Kovacs, et. al.
15 * All rights reserved.
17 * This program is free software: you can redistribute it and/or modify
18 * it under the terms of the GNU General Public License as published by
19 * the Free Software Foundation, either version 3 of the License, or
20 * (at your option) any later version.
22 * This program is distributed in the hope that it will be useful,
23 * but WITHOUT ANY WARRANTY; without even the implied warranty of
24 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
25 * GNU General Public License for more details.
26 * You should have received a copy of the GNU General Public License
27 * along with this program. If not, see <http://www.gnu.org/licenses/>.
32 #include "conky.h" /* text_buffer_size */
38 #include "text_object.h"
43 /* this is the root of all per disk stats,
44 * also containing the totals. */
45 struct diskio_stat stats = {
47 .sample = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0},
48 .sample_read = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0},
49 .sample_write = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0},
54 .last_read = UINT_MAX,
55 .last_write = UINT_MAX,
58 void clear_diskio_stats(void)
60 struct diskio_stat *cur;
63 stats.next = stats.next->next;
70 struct diskio_stat *prepare_diskio_stat(const char *s)
73 char stat_name[text_buffer_size], device_name[text_buffer_size];
74 struct diskio_stat *cur = &stats;
79 #if defined(__FreeBSD__)
80 if (strncmp(s, "/dev/", 5) == 0) {
81 // supplied a /dev/device arg, so cut off the /dev part
82 strncpy(device_name, s + 5, text_buffer_size);
85 strncpy(device_name, s, text_buffer_size);
87 snprintf(stat_name, text_buffer_size, "/dev/%s", device_name);
89 if (stat(stat_name, &sb)) {
90 NORM_ERR("diskio device '%s' does not exist", s);
96 if (!strcmp(cur->dev, device_name)) {
101 /* no existing found, make a new one */
102 cur->next = calloc(1, sizeof(struct diskio_stat));
104 cur->dev = strndup(device_name, text_buffer_size);
105 cur->last = UINT_MAX;
106 cur->last_read = UINT_MAX;
107 cur->last_write = UINT_MAX;
112 void parse_diskio_arg(struct text_object *obj, const char *arg)
114 obj->data.opaque = prepare_diskio_stat(arg);
117 /* dir indicates the direction:
122 static void print_diskio_dir(struct text_object *obj, int dir, char *p, int p_max_size)
124 struct diskio_stat *diskio = obj->data.opaque;
131 val = diskio->current_read;
133 val = diskio->current;
135 val = diskio->current_write;
137 /* TODO: move this correction from kB to kB/s elsewhere
138 * (or get rid of it??) */
139 human_readable((val / update_interval) * 1024LL, p, p_max_size);
142 void print_diskio(struct text_object *obj, char *p, int p_max_size)
144 print_diskio_dir(obj, 0, p, p_max_size);
147 void print_diskio_read(struct text_object *obj, char *p, int p_max_size)
149 print_diskio_dir(obj, -1, p, p_max_size);
152 void print_diskio_write(struct text_object *obj, char *p, int p_max_size)
154 print_diskio_dir(obj, 1, p, p_max_size);
158 void parse_diskiograph_arg(struct text_object *obj, const char *arg)
161 buf = scan_graph(obj, arg, 0);
163 obj->data.opaque = prepare_diskio_stat(dev_name(buf));
168 static void print_diskiograph_dir(struct text_object *obj, int dir, char *p, int p_max_size)
170 struct diskio_stat *diskio = obj->data.opaque;
180 val = diskio->current_read;
182 val = diskio->current;
184 val = diskio->current_write;
186 new_graph(obj, p, p_max_size, val);
189 void print_diskiograph(struct text_object *obj, char *p, int p_max_size)
191 print_diskiograph_dir(obj, 0, p, p_max_size);
194 void print_diskiograph_read(struct text_object *obj, char *p, int p_max_size)
196 print_diskiograph_dir(obj, -1, p, p_max_size);
199 void print_diskiograph_write(struct text_object *obj, char *p, int p_max_size)
201 print_diskiograph_dir(obj, 1, p, p_max_size);
205 void update_diskio_values(struct diskio_stat *ds,
206 unsigned int reads, unsigned int writes)
209 double sum=0, sum_r=0, sum_w=0;
211 if (reads < ds->last_read || writes < ds->last_write) {
212 /* counter overflow or reset - rebase to sane values */
213 ds->last = reads+writes;
214 ds->last_read = reads;
215 ds->last_write = writes;
217 /* since the values in /proc/diskstats are absolute, we have to substract
218 * our last reading. The numbers stand for "sectors read", and we therefore
219 * have to divide by two to get KB */
220 ds->sample_read[0] = (reads - ds->last_read) / 2;
221 ds->sample_write[0] = (writes - ds->last_write) / 2;
222 ds->sample[0] = ds->sample_read[0] + ds->sample_write[0];
224 /* compute averages */
225 for (i = 0; i < (signed) info.diskio_avg_samples; i++) {
226 sum += ds->sample[i];
227 sum_r += ds->sample_read[i];
228 sum_w += ds->sample_write[i];
230 ds->current = sum / (double) info.diskio_avg_samples;
231 ds->current_read = sum_r / (double) info.diskio_avg_samples;
232 ds->current_write = sum_w / (double) info.diskio_avg_samples;
234 /* shift sample history */
235 for (i = info.diskio_avg_samples-1; i > 0; i--) {
236 ds->sample[i] = ds->sample[i-1];
237 ds->sample_read[i] = ds->sample_read[i-1];
238 ds->sample_write[i] = ds->sample_write[i-1];
242 ds->last_read = reads;
243 ds->last_write = writes;
244 ds->last = ds->last_read + ds->last_write;