-/* 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
*
* Please see COPYING for details
*
* Copyright (c) 2004, Hannu Saransaari and Lauri Hakkarainen
- * Copyright (c) 2005-2007 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/>.
*
- * $Id$ */
+ */
#include "conky.h"
+#include "logging.h"
+#include "fs.h"
+#include "specials.h"
+#include "text_object.h"
+#include <ctype.h>
#include <unistd.h>
-#include <stdlib.h>
-#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/mount.h>
#endif
+#if !defined(HAVE_STRUCT_STATFS_F_FSTYPENAME) && !defined (__OpenBSD__) && !defined(__FreeBSD__)
+#include <mntent.h>
+#endif
+
#define MAX_FS_STATS 64
static struct fs_stat fs_stats_[MAX_FS_STATS];
static void update_fs_stat(struct fs_stat *fs);
-void update_fs_stats()
+void get_fs_type(const char *path, char *result);
+
+int update_fs_stats(void)
{
unsigned i;
+ static double last_fs_update = 0.0;
+
+ if (current_update_time - last_fs_update < 13)
+ return 0;
for (i = 0; i < MAX_FS_STATS; ++i) {
- if (fs_stats[i].path) {
+ if (fs_stats[i].set) {
update_fs_stat(&fs_stats[i]);
}
}
+ last_fs_update = current_update_time;
+ return 0;
}
-void clear_fs_stats()
+void clear_fs_stats(void)
{
unsigned i;
-
for (i = 0; i < MAX_FS_STATS; ++i) {
- if (fs_stats[i].path) {
- free(fs_stats[i].path);
- fs_stats[i].path = NULL;
- }
+ memset(&fs_stats[i], 0, sizeof(struct fs_stat));
}
}
/* lookup existing or get new */
for (i = 0; i < MAX_FS_STATS; ++i) {
- if (fs_stats[i].path) {
- if (strcmp(fs_stats[i].path, s) == 0) {
+ if (fs_stats[i].set) {
+ if (strncmp(fs_stats[i].path, s, DEFAULT_TEXT_BUFFER_SIZE) == 0) {
return &fs_stats[i];
}
} else {
}
/* new path */
if (!new) {
- ERR("too many fs stats");
+ NORM_ERR("too many fs stats");
return 0;
}
- new->path = strdup(s);
+ strncpy(new->path, s, DEFAULT_TEXT_BUFFER_SIZE);
+ new->set = 1;
update_fs_stat(new);
return new;
}
struct statfs s;
if (statfs(fs->path, &s) == 0) {
- fs->size = (long long) s.f_blocks * s.f_bsize;
+ fs->size = (long long)s.f_blocks * s.f_bsize;
/* bfree (root) or bavail (non-roots) ? */
- fs->avail = (long long) s.f_bavail * s.f_bsize;
- fs->free = (long long) s.f_bfree * s.f_bsize;
+ fs->avail = (long long)s.f_bavail * s.f_bsize;
+ fs->free = (long long)s.f_bfree * s.f_bsize;
+ get_fs_type(fs->path, fs->type);
} else {
fs->size = 0;
fs->avail = 0;
fs->free = 0;
- ERR("statfs '%s': %s", fs->path, strerror(errno));
+ strncpy(fs->type, "unknown", DEFAULT_TEXT_BUFFER_SIZE);
+ NORM_ERR("statfs '%s': %s", fs->path, strerror(errno));
+ }
+}
+
+void get_fs_type(const char *path, char *result)
+{
+
+#if defined(HAVE_STRUCT_STATFS_F_FSTYPENAME) || defined(__FreeBSD__) || defined (__OpenBSD__)
+
+ struct statfs s;
+ if (statfs(path, &s) == 0) {
+ strncpy(result, s.f_fstypename, DEFAULT_TEXT_BUFFER_SIZE);
+ } else {
+ NORM_ERR("statfs '%s': %s", path, strerror(errno));
+ }
+ return;
+
+#else /* HAVE_STRUCT_STATFS_F_FSTYPENAME */
+
+ struct mntent *me;
+ FILE *mtab = setmntent("/etc/mtab", "r");
+ char *search_path;
+ int match;
+ char *slash;
+
+ if (mtab == NULL) {
+ NORM_ERR("setmntent /etc/mtab: %s", strerror(errno));
+ strncpy(result, "unknown", DEFAULT_TEXT_BUFFER_SIZE);
+ return;
+ }
+
+ me = getmntent(mtab);
+
+ // find our path in the mtab
+ search_path = strdup(path);
+ do {
+ while ((match = strcmp(search_path, me->mnt_dir))
+ && getmntent(mtab));
+ if (!match)
+ break;
+ fseek(mtab, 0, SEEK_SET);
+ slash = strrchr(search_path, '/');
+ if (slash == NULL)
+ CRIT_ERR(NULL, NULL, "invalid path '%s'", path);
+ if (strlen(slash) == 1) /* trailing slash */
+ *(slash) = '\0';
+ else if (strlen(slash) > 1)
+ *(slash + 1) = '\0';
+ else
+ CRIT_ERR(NULL, NULL, "found a crack in the matrix!");
+ } while (strlen(search_path) > 0);
+ free(search_path);
+
+ endmntent(mtab);
+
+ if (me && !match) {
+ strncpy(result, me->mnt_type, DEFAULT_TEXT_BUFFER_SIZE);
+ return;
+ }
+#endif /* HAVE_STRUCT_STATFS_F_FSTYPENAME */
+
+ strncpy(result, "unknown", DEFAULT_TEXT_BUFFER_SIZE);
+
+}
+
+void init_fs_bar(struct text_object *obj, const char *arg)
+{
+ arg = scan_bar(obj, arg);
+ if (arg) {
+ while (isspace(*arg)) {
+ arg++;
+ }
+ if (*arg == '\0') {
+ arg = "/";
+ }
+ } else {
+ arg = "/";
}
+ obj->data.opaque = prepare_fs_stat(arg);
+}
+
+void print_fs_bar(struct text_object *obj, int be_free_bar, char *p, int p_max_size)
+{
+ double val = 1.0;
+ struct fs_stat *fs = obj->data.opaque;
+
+ if (!fs)
+ return;
+
+ if (fs->size)
+ val = (double)fs->avail / (double)fs->size;
+
+ if (!be_free_bar)
+ val = 1.0 - val;
+
+ new_bar(obj, p, p_max_size, (int)(255 * val));
+}
+
+void init_fs(struct text_object *obj, const char *arg)
+{
+ obj->data.opaque = prepare_fs_stat(arg ? arg : "/");
+}
+
+void print_fs_perc(struct text_object *obj, int be_free, char *p, int p_max_size)
+{
+ struct fs_stat *fs = obj->data.opaque;
+ int val = 100;
+
+ if (!fs)
+ return;
+
+ if (fs->size)
+ val = fs->avail * 100 / fs->size;
+
+ if (!be_free)
+ val = 100 - val;
+
+ percent_print(p, p_max_size, val);
+}
+
+#define HUMAN_PRINT_FS_GENERATOR(name, expr) \
+void print_fs_##name(struct text_object *obj, char *p, int p_max_size) \
+{ \
+ struct fs_stat *fs = obj->data.opaque; \
+ if (fs) \
+ human_readable(expr, p, p_max_size); \
+}
+
+HUMAN_PRINT_FS_GENERATOR(free, fs->avail)
+HUMAN_PRINT_FS_GENERATOR(size, fs->size)
+HUMAN_PRINT_FS_GENERATOR(used, fs->size - fs->free)
+
+void print_fs_type(struct text_object *obj, char *p, int p_max_size)
+{
+ struct fs_stat *fs = obj->data.opaque;
+
+ if (fs)
+ snprintf(p, p_max_size, "%s", fs->type);
}
projects / monky / blobdiff