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-2009 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/>.
35 #include "text_object.h"
36 #include "timed_thread.h"
38 #include <sys/types.h>
48 timed_thread *p_timed_thread;
52 /* FIXME: this will probably not work, since the variable is being reused
53 * between different text objects. So when a process really hangs, it's PID
54 * will be overwritten at the next iteration. */
57 //our own implementation of popen, the difference : the value of 'childpid' will be filled with
58 //the pid of the running 'command'. This is useful if want to kill it when it hangs while reading
59 //or writing to it. We have to kill it because pclose will wait until the process dies by itself
60 static FILE* pid_popen(const char *command, const char *mode, pid_t *child) {
62 int parentend, childend;
64 //by running pipe after the strcmp's we make sure that we don't have to create a pipe
65 //and close the ends if mode is something illegal
66 if(strcmp(mode, "r") == 0) {
72 } else if(strcmp(mode, "w") == 0) {
86 } else if(*child > 0) {
88 waitpid(*child, NULL, 0);
90 //don't read from both stdin and pipe or write to both stdout and pipe
91 if(childend == ends[0]) {
96 dup(childend); //by dupping childend, the returned fd will have close-on-exec turned off
97 execl("/bin/sh", "sh", "-c", command, (char *) NULL);
98 _exit(EXIT_FAILURE); //child should die here, (normally execl will take care of this but it can fail)
100 return fdopen(parentend, mode);
103 //remove backspaced chars, example: "dog^H^H^Hcat" becomes "cat"
104 //string has to end with \0 and it's length should fit in a int
106 static void remove_deleted_chars(char *string){
108 while(string[i] != 0){
109 if(string[i] == BACKSPACE){
111 strcpy( &(string[i-1]), &(string[i+1]) );
113 }else strcpy( &(string[i]), &(string[i+1]) ); //necessary for ^H's at the start of a string
118 static inline double get_barnum(char *buf)
130 if (sscanf(buf, "%lf", &barnum) == 0) {
131 NORM_ERR("reading exec value failed (perhaps it's not the "
135 if (barnum > 100.0 || barnum < 0.0) {
136 NORM_ERR("your exec value is not between 0 and 100, "
137 "therefore it will be ignored");
143 static inline void read_exec(const char *data, char *buf, const int size)
147 memset(buf, 0, size);
152 alarm(update_interval);
153 fp = pid_popen(data, "r", &childpid);
157 length = fread(buf, 1, size, fp);
160 if (length > 0 && buf[length - 1] == '\n') {
161 buf[length - 1] = '\0';
169 static void *threaded_exec(void *) __attribute__((noreturn));
171 static void *threaded_exec(void *arg)
174 struct text_object *obj = arg;
175 struct execi_data *ed = obj->data.opaque;
178 buff = malloc(text_buffer_size);
179 read_exec(ed->cmd, buff, text_buffer_size);
187 timed_thread_lock(ed->p_timed_thread);
191 timed_thread_unlock(ed->p_timed_thread);
192 if (timed_thread_test(ed->p_timed_thread, 0)) {
193 timed_thread_exit(ed->p_timed_thread);
199 /* check the execi fields and return true if the given interval has passed */
200 static int time_to_update(struct execi_data *ed)
204 if (current_update_time - ed->last_update >= ed->interval)
209 void scan_exec_arg(struct text_object *obj, const char *arg)
211 obj->data.s = strndup(arg ? arg : "", text_buffer_size);
214 void scan_pre_exec_arg(struct text_object *obj, const char *arg)
218 obj->type = OBJ_text;
219 read_exec(arg, buf, sizeof(buf));
220 obj->data.s = strndup(buf, text_buffer_size);
223 void scan_execi_arg(struct text_object *obj, const char *arg)
225 struct execi_data *ed;
228 ed = malloc(sizeof(struct execi_data));
229 memset(ed, 0, sizeof(struct execi_data));
231 if (sscanf(arg, "%f %n", &ed->interval, &n) <= 0) {
232 NORM_ERR("${execi* <interval> command}");
236 ed->cmd = strndup(arg + n, text_buffer_size);
237 obj->data.opaque = ed;
241 void scan_execgraph_arg(struct text_object *obj, const char *arg)
243 struct execi_data *ed;
246 ed = malloc(sizeof(struct execi_data));
247 memset(ed, 0, sizeof(struct execi_data));
249 buf = scan_graph(obj, arg, 100);
251 NORM_ERR("missing command argument to execgraph object");
255 obj->data.opaque = ed;
259 void print_exec(struct text_object *obj, char *p, int p_max_size)
261 read_exec(obj->data.s, p, p_max_size);
262 remove_deleted_chars(p);
265 void print_execp(struct text_object *obj, char *p, int p_max_size)
267 struct information *tmp_info;
268 struct text_object subroot;
271 buf = malloc(text_buffer_size);
272 memset(buf, 0, text_buffer_size);
274 read_exec(obj->data.s, buf, text_buffer_size);
276 tmp_info = malloc(sizeof(struct information));
277 memcpy(tmp_info, &info, sizeof(struct information));
278 parse_conky_vars(&subroot, buf, p, p_max_size, tmp_info);
280 free_text_objects(&subroot, 1);
285 void print_execi(struct text_object *obj, char *p, int p_max_size)
287 struct execi_data *ed = obj->data.opaque;
292 if (time_to_update(ed)) {
294 ed->buffer = malloc(text_buffer_size);
295 read_exec(ed->cmd, ed->buffer, text_buffer_size);
296 ed->last_update = current_update_time;
298 snprintf(p, p_max_size, "%s", ed->buffer);
301 void print_execpi(struct text_object *obj, char *p, int p_max_size)
303 struct execi_data *ed = obj->data.opaque;
304 struct text_object subroot;
305 struct information *tmp_info;
310 tmp_info = malloc(sizeof(struct information));
311 memcpy(tmp_info, &info, sizeof(struct information));
313 if (time_to_update(ed)) {
316 FILE *fp = pid_popen(ed->cmd, "r", &childpid);
319 ed->buffer = malloc(text_buffer_size);
321 length = fread(ed->buffer, 1, text_buffer_size, fp);
325 output[length] = '\0';
326 if (length > 0 && output[length - 1] == '\n') {
327 output[length - 1] = '\0';
330 ed->last_update = current_update_time;
332 parse_conky_vars(&subroot, ed->buffer, p, p_max_size, tmp_info);
333 free_text_objects(&subroot, 1);
337 void print_texeci(struct text_object *obj, char *p, int p_max_size)
339 struct execi_data *ed = obj->data.opaque;
344 if (!ed->p_timed_thread) {
345 ed->p_timed_thread = timed_thread_create(&threaded_exec,
346 (void *) obj, ed->interval * 1000000);
347 if (!ed->p_timed_thread) {
348 NORM_ERR("Error creating texeci timed thread");
351 * note that we don't register this thread with the
352 * timed_thread list, because we destroy it manually
354 if (timed_thread_run(ed->p_timed_thread)) {
355 NORM_ERR("Error running texeci timed thread");
358 timed_thread_lock(ed->p_timed_thread);
359 snprintf(p, p_max_size, "%s", ed->buffer);
360 timed_thread_unlock(ed->p_timed_thread);
364 void print_execgauge(struct text_object *obj, char *p, int p_max_size)
368 read_exec(obj->data.s, p, p_max_size);
369 barnum = get_barnum(p); /*using the same function*/
373 new_gauge(obj, p, p_max_size, round_to_int(barnum * 255.0));
378 void print_execgraph(struct text_object *obj, char *p, int p_max_size)
381 struct execi_data *ed = obj->data.opaque;
386 read_exec(ed->cmd, p, p_max_size);
387 barnum = get_barnum(p);
390 new_graph(obj, p, p_max_size, round_to_int(barnum));
394 void print_execigraph(struct text_object *obj, char *p, int p_max_size)
396 struct execi_data *ed = obj->data.opaque;
401 if (time_to_update(ed)) {
404 read_exec(ed->cmd, p, p_max_size);
405 barnum = get_barnum(p);
410 ed->last_update = current_update_time;
412 new_graph(obj, p, p_max_size, (int) (ed->barnum));
416 void print_execigauge(struct text_object *obj, char *p, int p_max_size)
418 struct execi_data *ed = obj->data.opaque;
423 if (time_to_update(ed)) {
426 read_exec(ed->cmd, p, p_max_size);
427 barnum = get_barnum(p);
430 ed->barnum = 255 * barnum / 100.0;
432 ed->last_update = current_update_time;
434 new_gauge(obj, p, p_max_size, round_to_int(ed->barnum));
437 void print_execbar(struct text_object *obj, char *p, int p_max_size)
440 read_exec(obj->data.s, p, p_max_size);
441 barnum = get_barnum(p);
445 new_bar(obj, p, p_max_size, round_to_int(barnum * 255.0));
449 void print_execibar(struct text_object *obj, char *p, int p_max_size)
451 struct execi_data *ed = obj->data.opaque;
457 if (time_to_update(ed)) {
458 read_exec(ed->cmd, p, p_max_size);
459 barnum = get_barnum(p);
464 ed->last_update = current_update_time;
466 new_bar(obj, p, p_max_size, round_to_int(ed->barnum * 2.55));
469 void free_exec(struct text_object *obj)
477 void free_execi(struct text_object *obj)
479 struct execi_data *ed = obj->data.opaque;
484 if (ed->p_timed_thread)
485 timed_thread_destroy(ed->p_timed_thread, &ed->p_timed_thread);
490 free(obj->data.opaque);
491 obj->data.opaque = NULL;