aboutsummaryrefslogtreecommitdiff
path: root/src/zenmonitor-lib.c
diff options
context:
space:
mode:
authorAnthony Wang2021-04-01 13:40:13 -0500
committerAnthony Wang2021-04-01 13:40:13 -0500
commit3038c0d1a05de9a56da51d3a434a7045595a8be1 (patch)
tree27cd76125fe5405f719dd07e3832137ff5161c6f /src/zenmonitor-lib.c
parentf12d6bcb6a5794bd6289a89eccee61b65cedbbb1 (diff)
parent389c31401bd26ffb2ef5bd1a2b29b1d70693d649 (diff)
Merge branch 'zenmonitor-cli' of https://github.com/shunonymous/zenmonitor into shunonymous-zenmonitor-cli
Diffstat (limited to 'src/zenmonitor-lib.c')
-rw-r--r--src/zenmonitor-lib.c170
1 files changed, 170 insertions, 0 deletions
diff --git a/src/zenmonitor-lib.c b/src/zenmonitor-lib.c
new file mode 100644
index 0000000..a59680b
--- /dev/null
+++ b/src/zenmonitor-lib.c
@@ -0,0 +1,170 @@
+#include <assert.h>
+#include <cpuid.h>
+#include <strings.h>
+#include <time.h>
+
+#include "zenpower.h"
+#include "msr.h"
+#include "os.h"
+#include "zenmonitor.h"
+
+#define AMD_STRING "AuthenticAMD"
+#define ZEN_FAMILY 0x17
+#define ZEN3_FAMILY 0x19
+
+const guint SENSOR_DATA_STORE_NUM = 4096;
+
+// AMD PPR = https://www.amd.com/system/files/TechDocs/54945_PPR_Family_17h_Models_00h-0Fh.pdf
+
+gboolean check_zen() {
+ guint32 eax = 0, ebx = 0, ecx = 0, edx = 0, ext_family;
+ char vendor[13];
+
+ __get_cpuid(0, &eax, &ebx, &ecx, &edx);
+
+ memcpy(vendor, &ebx, 4);
+ memcpy(vendor+4, &edx, 4);
+ memcpy(vendor+8, &ecx, 4);
+ vendor[12] = 0;
+
+ if (strcmp(vendor, AMD_STRING) != 0){
+ return FALSE;
+ }
+
+ __get_cpuid(1, &eax, &ebx, &ecx, &edx);
+
+ ext_family = ((eax >> 8) & 0xF) + ((eax >> 20) & 0xFF);
+ if (ext_family != ZEN_FAMILY && ext_family != ZEN3_FAMILY){
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+gchar *cpu_model() {
+ guint32 eax = 0, ebx = 0, ecx = 0, edx = 0;
+ char model[48];
+
+ // AMD PPR: page 65-68 - CPUID_Fn80000002_EAX-CPUID_Fn80000004_EDX
+ __get_cpuid(0x80000002, &eax, &ebx, &ecx, &edx);
+ memcpy(model, &eax, 4);
+ memcpy(model+4, &ebx, 4);
+ memcpy(model+8, &ecx, 4);
+ memcpy(model+12, &edx, 4);
+
+ __get_cpuid(0x80000003, &eax, &ebx, &ecx, &edx);
+ memcpy(model+16, &eax, 4);
+ memcpy(model+20, &ebx, 4);
+ memcpy(model+24, &ecx, 4);
+ memcpy(model+28, &edx, 4);
+
+ __get_cpuid(0x80000004, &eax, &ebx, &ecx, &edx);
+ memcpy(model+32, &eax, 4);
+ memcpy(model+36, &ebx, 4);
+ memcpy(model+40, &ecx, 4);
+ memcpy(model+44, &edx, 4);
+
+ model[48] = 0;
+ return g_strdup(g_strchomp(model));
+}
+
+guint get_core_count() {
+ guint eax = 0, ebx = 0, ecx = 0, edx = 0;
+ guint logical_cpus, threads_per_code;
+
+ // AMD PPR: page 57 - CPUID_Fn00000001_EBX
+ __get_cpuid(1, &eax, &ebx, &ecx, &edx);
+ logical_cpus = (ebx >> 16) & 0xFF;
+
+ // AMD PPR: page 82 - CPUID_Fn8000001E_EBX
+ __get_cpuid(0x8000001E, &eax, &ebx, &ecx, &edx);
+ threads_per_code = ((ebx >> 8) & 0xF) + 1;
+
+ if (threads_per_code == 0)
+ return logical_cpus;
+
+ return logical_cpus / threads_per_code;
+}
+
+SensorDataStore *sensor_data_store_new()
+{
+ SensorDataStore *ret;
+
+ ret = g_new0(SensorDataStore, 1);
+ ret->labels = g_ptr_array_new();
+ ret->data = g_ptr_array_new();
+ ret->time = g_array_new(FALSE, TRUE, sizeof(struct timespec));
+
+ return ret;
+}
+
+void sensor_data_store_add_entry(SensorDataStore *store, gchar *entry)
+{
+ GArray *data;
+ data = g_array_new(TRUE, TRUE, sizeof(float));
+
+ g_ptr_array_add(store->labels, entry);
+ g_ptr_array_add(store->data, data);
+}
+
+void sensor_data_store_keep_time(SensorDataStore *store)
+{
+ struct timespec ts;
+ timespec_get(&ts, TIME_UTC);
+ g_array_append_val(store->time, ts);
+}
+
+gint sensor_data_store_drop_entry(SensorDataStore *store, gchar *entry)
+{
+ guint index = 0;
+ gboolean found = g_ptr_array_find(store->labels, entry, &index);
+ if(!found)
+ {
+ return 1;
+ }
+
+ g_ptr_array_remove_index(store->labels, index);
+ g_ptr_array_remove_index(store->data, index);
+
+ return 0;
+}
+
+gint sensor_data_store_add_data(SensorDataStore *store, gchar *entry, float value)
+{
+ guint index = 0;
+ gboolean found = g_ptr_array_find(store->labels, entry, &index);
+
+ if(!found)
+ {
+ return 1;
+ }
+
+ GArray *data = g_ptr_array_index(store->data, index);
+ g_array_append_val(data, value);
+
+ return 0;
+}
+
+void sensor_data_store_free(SensorDataStore *store)
+{
+ if(store)
+ {
+ g_array_free(store->time, TRUE);
+ g_ptr_array_free(store->labels, TRUE);
+ g_ptr_array_free(store->data, TRUE);
+ g_free(store);
+ }
+}
+
+SensorInit *sensor_init_new() {
+ return g_new0(SensorInit, 1);
+}
+
+void sensor_init_free(SensorInit *s) {
+ if (s) {
+ g_free(s->label);
+ g_free(s->hint);
+ g_free(s);
+ }
+}
+