feat&test: open_msr and test

libpmu/pmu.c

@@ -51,6 +51,14 @@ uint64_t pmu_event_to_hexcode(PMU_EVENT *event)
     return hexcode;
 }
 
+inline int pmu_open_msr(int core)
+{
+    core %= 512;
+    char msr_path[32];
+    sprintf(msr_path, "/dev/cpu/%d/msr", core);
+    return open(msr_path, O_RDWR);
+}
+
 inline void pmu_set_event(int core, int *msr_fd, uint64_t hexcode, size_t pmu_id)
 {
     core %= 512;
@@ -64,6 +72,15 @@ inline void pmu_set_event(int core, int *msr_fd, uint64_t hexcode, size_t pmu_id
     lseek(*msr_fd, 0x38F, SEEK_SET);
 }
 
+inline void pmu_set_msr_event(int msr_fd, uint64_t hexcode, size_t pmu_id)
+{
+    /* DISABLE ALL COUNTERS */
+    write_to_IA32_PERF_GLOBAL_CTRL(msr_fd, 0ull);
+
+    write_to_IA32_PERFEVTSELi(msr_fd, pmu_id, hexcode);
+    lseek(msr_fd, 0x38F, SEEK_SET);
+}
+
 inline void pmu_set_pmc(int msr_fd, size_t pmu_id, uint64_t val)
 {
     write_to_IA32_PMCi(msr_fd, pmu_id, val);

libpmu/pmu.h

@@ -30,8 +30,12 @@ typedef struct PMU_EVENT_STRUCT
 
 uint64_t pmu_event_to_hexcode(PMU_EVENT *event);
 
+int pmu_open_msr(int core);
+
 void pmu_set_event(int core, int *msr_fd, uint64_t hexcode, size_t pmu_id);
 
+void pmu_set_msr_event(int msr_fd, uint64_t hexcode, size_t pmu_id);
+
 void pmu_set_pmc(int msr_fd, size_t pmu_id, uint64_t val);
 
 void pmu_record_start(int msr_fd);

tests/Makefile

@@ -29,4 +29,10 @@ clean:
 	rm -rf *.out
 
 test:
-	make clean && make && taskset -c 0 sudo ./basic.c.out
+	make clean 
+
+	$(CC) $(LIBC) basic.c -o basic.c.out $(CFLAGS)
+	$(CC) $(LIBC) all.c -o all.c.out $(CFLAGS)
+
+	taskset -c 0 sudo ./basic.c.out
+	taskset -c 0 sudo ./all.c.out

tests/all.c

@@ -0,0 +1,48 @@
+#include <stdio.h>
+#include <stdint.h>
+#include <assert.h>
+#include <math.h>
+#include "cacheutils.h"
+#include "./../libpmu/pmu.h"
+
+int main(int argc, char const *argv[])
+{
+    PMU_EVENT pe0 = {
+        .event_code = 0xD1,
+        .umask = 0x08,
+        .user_mode = 1,
+        .os = 0,
+        .edge_detect = 0,
+        .pc = 0,
+        .int_enable = 0,
+        .enable_couters = 1,
+        .invert = 0,
+        .counter_mask = 0,
+    }; // MEM_LOAD_RETIRED.L1_MISS
+
+    uint64_t hexcode = pmu_event_to_hexcode(&pe0);
+
+    int msr_fd = pmu_open_msr(0), core = 0, pmu_id = 0;
+
+    pmu_set_event(core, &msr_fd, hexcode, pmu_id);
+
+    pmu_set_msr_event(msr_fd, hexcode, pmu_id);
+
+    pmu_set_pmc(msr_fd, pmu_id, 0);
+
+    pmu_record_start(msr_fd);
+
+    uint64_t pmu_val_msr = pmu_get_MSR_pmc(msr_fd, pmu_id);
+
+    uint64_t pmu_val = pmu_get_rdpmc(pmu_id);
+
+    uint64_t msrs_num = pmu_get_MSRs_num();
+
+    uint64_t pmc_bit_width = pmu_get_PMC_bit_width();
+
+    pmu_record_stop(msr_fd);
+
+    printf("pmu-utils all test function passed\n");
+
+    return 0;
+}

tests/basic.c

@@ -14,7 +14,7 @@ int main(int argc, char const *argv[])
     uint64_t pmc_bit_width = pmu_get_PMC_bit_width();
     printf("PMC_bit_width: \t%lu\n", pmc_bit_width);
 
-    int msr_fd, core = 0, pmu_id = 0;
+    int core = 0, pmu_id = 0;
     PMU_EVENT pe0 = {
         .event_code = 0xD1,
         .umask = 0x08,
@@ -31,7 +31,11 @@ int main(int argc, char const *argv[])
     uint64_t pe0_code = pmu_event_to_hexcode(&pe0);
     printf("Event Hexcode: \t0x%lx\n", pe0_code);
 
+    int msr_fd = pmu_open_msr(core);
+
     pmu_set_event(core, &msr_fd, pe0_code, pmu_id);
+    pmu_set_msr_event(msr_fd, pe0_code, pmu_id);
+
     pmu_set_pmc(msr_fd, pmu_id, 0);
     pmu_record_start(msr_fd);
     uint64_t start_pmu_value = pmu_get_MSR_pmc(msr_fd, pmu_id);