Xinqi Bao's Git

ram: Check for theoretical division by zero
[slstatus.git] / components / ram.c
1 /* See LICENSE file for copyright and license details. */
2 #include <stdio.h>
3
4 #include "../util.h"
5
6 #if defined(__linux__)
7 #include <inttypes.h>
8
9 const char *
10 ram_free(void)
11 {
12 uint64_t free;
13
14 if (pscanf("/proc/meminfo",
15 "MemTotal: %" PRIu64 " kB\n"
16 "MemFree: %" PRIu64 " kB\n"
17 "MemAvailable: %" PRIu64 " kB\n",
18 &free, &free, &free) != 3) {
19 return NULL;
20 }
21
22 return fmt_human(free * 1024, 1024);
23 }
24
25 const char *
26 ram_perc(void)
27 {
28 uint64_t total, free, buffers, cached;
29
30 if (pscanf("/proc/meminfo",
31 "MemTotal: %" PRIu64 " kB\n"
32 "MemFree: %" PRIu64 " kB\n"
33 "MemAvailable: %" PRIu64 " kB\nBuffers: %ld kB\n"
34 "Cached: %" PRIu64 " kB\n",
35 &total, &free, &buffers, &buffers, &cached) != 5) {
36 return NULL;
37 }
38
39 if (total == 0) {
40 return NULL;
41 }
42
43 return bprintf("%d", 100 * ((total - free) -
44 (buffers + cached)) / total);
45 }
46
47 const char *
48 ram_total(void)
49 {
50 uint64_t total;
51
52 if (pscanf("/proc/meminfo", "MemTotal: %" PRIu64 " kB\n",
53 &total) != 1) {
54 return NULL;
55 }
56
57 return fmt_human(total * 1024, 1024);
58 }
59
60 const char *
61 ram_used(void)
62 {
63 uint64_t total, free, buffers, cached;
64
65 if (pscanf("/proc/meminfo",
66 "MemTotal: %" PRIu64 " kB\n"
67 "MemFree: %" PRIu64 " kB\n"
68 "MemAvailable: %" PRIu64 " kB\nBuffers: %" PRIu64 " kB\n"
69 "Cached: %" PRIu64 " kB\n",
70 &total, &free, &buffers, &buffers, &cached) != 5) {
71 return NULL;
72 }
73
74 return fmt_human((total - free - buffers - cached) * 1024,
75 1024);
76 }
77 #elif defined(__OpenBSD__)
78 #include <stdlib.h>
79 #include <sys/sysctl.h>
80 #include <sys/types.h>
81 #include <unistd.h>
82
83 #define LOG1024 10
84 #define pagetok(size, pageshift) (size_t)(size << (pageshift - LOG1024))
85
86 inline int
87 load_uvmexp(struct uvmexp *uvmexp)
88 {
89 int uvmexp_mib[] = {CTL_VM, VM_UVMEXP};
90 size_t size;
91
92 size = sizeof(*uvmexp);
93
94 if (sysctl(uvmexp_mib, 2, uvmexp, &size, NULL, 0) >= 0) {
95 return 1;
96 }
97
98 return 0;
99 }
100
101 const char *
102 ram_free(void)
103 {
104 struct uvmexp uvmexp;
105 int free_pages;
106
107 if (load_uvmexp(&uvmexp)) {
108 free_pages = uvmexp.npages - uvmexp.active;
109 return fmt_human(pagetok(free_pages, uvmexp.pageshift) *
110 1024, 1024);
111 }
112
113 return NULL;
114 }
115
116 const char *
117 ram_perc(void)
118 {
119 struct uvmexp uvmexp;
120 int percent;
121
122 if (load_uvmexp(&uvmexp)) {
123 percent = uvmexp.active * 100 / uvmexp.npages;
124 return bprintf("%d", percent);
125 }
126
127 return NULL;
128 }
129
130 const char *
131 ram_total(void)
132 {
133 struct uvmexp uvmexp;
134
135 if (load_uvmexp(&uvmexp)) {
136 return fmt_human(pagetok(uvmexp.npages,
137 uvmexp.pageshift) * 1024,
138 1024);
139 }
140
141 return NULL;
142 }
143
144 const char *
145 ram_used(void)
146 {
147 struct uvmexp uvmexp;
148
149 if (load_uvmexp(&uvmexp)) {
150 return fmt_human(pagetok(uvmexp.active,
151 uvmexp.pageshift) * 1024,
152 1024);
153 }
154
155 return NULL;
156 }
157 #endif