aboutsummaryrefslogtreecommitdiff
path: root/src/blocks/ram.c
blob: da56baa99996181569865bc1748b58c2dda3054d (plain)
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include <string.h>
#include <stdio.h>
#include <assert.h>
#include <stdlib.h>

#include "../block.h"

#include "../any_log.h"

typedef struct {
    block_text_t block;
    char *format;
} block_ram_t;

static const struct timespec block_ram_interval = {
    .tv_sec = 1,
    .tv_nsec = 0,
};

static void block_ram_update(block_t *block)
{
    const char *path = "/proc/meminfo";
    FILE *meminfo = fopen(path, "rb");

    if (meminfo == NULL) {
        log_value_debug("Failed to open meminfo",
                        "s:label", block->label,
                        "s:path", path,
                        "i:errno", errno);
        return;
    }

    uintmax_t total, unused, available, buffers, cached;
    int n = fscanf(meminfo,
                   "MemTotal: %ju kB\n"
                   "MemFree: %ju kB\n"
                   "MemAvailable: %ju kB\n"
                   "Buffers: %ju kB\n"
                   "Cached: %ju kB\n",
                   &total, &unused, &available, &buffers, &cached);
    fclose(meminfo);

    if (n != 5) {
        log_value_debug("Failed to read meminfo",
                        "s:label", block->label,
                        "i:errno", errno);
        return;
    }

    static const char *ram_formats[] = {
        "total",
        "free",
        "used",
        "free-percentage",
        "used-percentage",
        NULL,
    };

    char buffer[32][5] = { 0 };
    snprintf(buffer[0], 32, "%ld", total);
    snprintf(buffer[1], 32, "%ld", available);
    snprintf(buffer[2], 32, "%ld", total - available);
    snprintf(buffer[3], 32, "%ld", 100 * available / total);
    snprintf(buffer[4], 32, "%ld", 100 * (total - available) / total);

    const char *ram_values[] = {
        buffer[0],
        buffer[1],
        buffer[2],
        buffer[3],
        buffer[4],
        NULL,
    };

    block_ram_t *ram = (block_ram_t *)block;
    free(ram->block.text);

    ram->block.text = strformat(ram->format, '%', ram_formats, ram_values);
    assert(ram->block.text != NULL);
}

static block_t *block_ram_alloc(const block_scheme_t *scheme)
{
    block_t *block = calloc(1, sizeof(block_ram_t));
    block->type = BLOCK_TEXT;
    block->update_interval = block_ram_interval;
    block->update_fn = block_ram_update;
    return block;
}

static void block_ram_clean(block_t *block)
{
    block_ram_t *ram = (block_ram_t *)block;
    free(ram->format);
    free(ram->block.text);
}

static bool block_ram_validate(block_t *block, const block_scheme_t *scheme)
{
    block_ram_t *ram = (block_ram_t *)block;

    if (ram->block.text == NULL) {
        log_error("Block '%s' requires key '%s'", block->label, "text");
        return false;
    }

    if (strstr(ram->block.text, "%{") == NULL) {
        log_warn("Block '%s' does not use any ram variable", block->label);

        block->update_fn = NULL;
        log_debug("Disabled updates for block '%s'", block->label);
        return true;
    }

    ram->format = ram->block.text;
    ram->block.text = strcopy("?");
    return true;
}

const block_scheme_t block_ram_scheme = {
    .name = "ram",
    .entries = NULL,
    .alloc_fn = block_ram_alloc,
    .clean_fn = block_ram_clean,
    .validate_fn = block_ram_validate,
    .resolve_fn = NULL,
};