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
|
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <mach/mach.h>
#include <stdbool.h>
#include <time.h>
#define MAX_TOPPROC_LEN 28
static const char TOPPROC[] = { "/bin/ps -Aceo pid,pcpu,comm -r" };
static const char FILTER_PATTERN[] = { "com.apple." };
struct cpu {
host_t host;
mach_msg_type_number_t count;
host_cpu_load_info_data_t load;
host_cpu_load_info_data_t prev_load;
bool has_prev_load;
char command[256];
};
static inline void cpu_init(struct cpu* cpu) {
cpu->host = mach_host_self();
cpu->count = HOST_CPU_LOAD_INFO_COUNT;
cpu->has_prev_load = false;
snprintf(cpu->command, 100, "");
}
static inline void cpu_update(struct cpu* cpu) {
kern_return_t error = host_statistics(cpu->host,
HOST_CPU_LOAD_INFO,
(host_info_t)&cpu->load,
&cpu->count );
if (error != KERN_SUCCESS) {
printf("Error: Could not read cpu host statistics.\n");
return;
}
if (cpu->has_prev_load) {
uint32_t delta_user = cpu->load.cpu_ticks[CPU_STATE_USER]
- cpu->prev_load.cpu_ticks[CPU_STATE_USER];
uint32_t delta_system = cpu->load.cpu_ticks[CPU_STATE_SYSTEM]
- cpu->prev_load.cpu_ticks[CPU_STATE_SYSTEM];
uint32_t delta_idle = cpu->load.cpu_ticks[CPU_STATE_IDLE]
- cpu->prev_load.cpu_ticks[CPU_STATE_IDLE];
double user_perc = (double)delta_user / (double)(delta_system
+ delta_user
+ delta_idle);
double sys_perc = (double)delta_system / (double)(delta_system
+ delta_user
+ delta_idle);
double total_perc = user_perc + sys_perc;
FILE* file;
char line[1024];
file = popen(TOPPROC, "r");
if (!file) {
printf("Error: TOPPROC command errored out...\n" );
return;
}
fgets(line, sizeof(line), file);
fgets(line, sizeof(line), file);
char* start = strstr(line, FILTER_PATTERN);
char topproc[MAX_TOPPROC_LEN + 4];
uint32_t caret = 0;
for (int i = 0; i < sizeof(line); i++) {
if (start && i == start - line) {
i+=9;
continue;
}
if (caret >= MAX_TOPPROC_LEN && caret <= MAX_TOPPROC_LEN + 2) {
topproc[caret++] = '.';
continue;
}
if (caret > MAX_TOPPROC_LEN + 2) break;
topproc[caret++] = line[i];
if (line[i] == '\0') break;
}
topproc[MAX_TOPPROC_LEN + 3] = '\0';
pclose(file);
char color[16];
if (total_perc >= .7) {
snprintf(color, 16, "%s", getenv("RED"));
} else if (total_perc >= .3) {
snprintf(color, 16, "%s", getenv("ORANGE"));
} else if (total_perc >= .1) {
snprintf(color, 16, "%s", getenv("YELLOW"));
} else {
snprintf(color, 16, "%s", getenv("LABEL_COLOR"));
}
snprintf(cpu->command, 256, "--push cpu.sys %.2f "
"--push cpu.user %.2f "
"--set cpu.top label='%s' "
"--set cpu.percent label=%.0f%% label.color=%s ",
sys_perc,
user_perc,
topproc,
total_perc*100.,
color );
}
else {
snprintf(cpu->command, 256, "");
}
cpu->prev_load = cpu->load;
cpu->has_prev_load = true;
}
|
