1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#include <string.h>
#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <sys/statvfs.h>
#include <sys/stat.h>
#include "scheme.h"
#include "../any_log.h"
typedef struct {
block_text_t block;
char *format;
char *path;
} block_fs_t;
static void block_fs_update(block_t *block)
{
block_fs_t *fs = (block_fs_t *)block;
struct statvfs sbuf;
assert(statvfs(fs->path, &sbuf) == 0);
static const char *fs_formats[] = {
"total",
"free",
"used",
"free-percentage",
"used-percentage",
NULL,
};
char buffer[5][32] = { 0 };
snprintf(buffer[0], 32, "%lu", sbuf.f_bsize * (uint64_t)sbuf.f_blocks);
snprintf(buffer[1], 32, "%lu", sbuf.f_bsize * (uint64_t)sbuf.f_bavail);
snprintf(buffer[2], 32, "%lu", sbuf.f_bsize * (sbuf.f_blocks - sbuf.f_bavail));
snprintf(buffer[3], 32, "%lu", (100 * sbuf.f_bavail) / sbuf.f_blocks);
snprintf(buffer[4], 32, "%lu", (100 * (sbuf.f_blocks - sbuf.f_bavail)) / sbuf.f_blocks);
const char *fs_values[] = {
buffer[0], buffer[1], buffer[2], buffer[3], buffer[4], NULL,
};
free(fs->block.text);
fs->block.text = strformat(fs->format, '%', fs_formats, fs_values);
assert(fs->block.text != NULL);
}
static void block_fs_clean(block_t *block)
{
block_fs_t *fs = (block_fs_t *)block;
free(fs->format);
free(fs->path);
}
static bool block_fs_validate(block_t *block, const block_scheme_t *scheme)
{
block_fs_t *fs = (block_fs_t *)block;
if (fs->block.text == NULL) {
log_error("Block '%s' requires key '%s'", block->label, "text");
return false;
}
if (strstr(fs->block.text, "%{") == NULL) {
log_warn("Block '%s' does not use any fs variable", block->label);
block->update_fn = NULL;
log_debug("Disabled updates for block '%s'", block->label);
return true;
}
struct stat sbuf;
if (fs->path == NULL || stat(fs->path, &sbuf) < 0) {
log_error("Block '%s' was given an invalid path '%s'", block->label, fs->path);
return false;
}
fs->format = fs->block.text;
fs->block.text = NULL;
return true;
}
static const config_entry_t block_fs_entries[] = {
{ "path", CONFIG_STRING, NULL, offsetof(block_fs_t, path) },
{ 0 },
};
const block_scheme_t block_fs_scheme = {
.name = "fs",
.block = {
.type = BLOCK_TEXT,
.update_interval = {
.tv_sec = 20,
.tv_nsec = 0,
},
.update_fn = block_fs_update,
.clean_fn = block_fs_clean,
},
.size = sizeof(block_fs_t),
.entries = block_fs_entries,
.validate = block_fs_validate,
};
|
