1 files changed, 24 insertions, 0 deletions

diff --git a/tools/perf/lib/cpumap.c b/tools/perf/lib/cpumap.c
index 80d587ab95aa..f3cfb4c71106 100644
--- a/tools/perf/lib/cpumap.c
+++ b/tools/perf/lib/cpumap.c
@@ -3,6 +3,8 @@
 #include <stdlib.h>
 #include <linux/refcount.h>
 #include <internal/cpumap.h>
+#include <asm/bug.h>
+#include <stdio.h>
 
 struct perf_cpu_map *perf_cpu_map__dummy_new(void)
 {
@@ -16,3 +18,25 @@ struct perf_cpu_map *perf_cpu_map__dummy_new(void)
 
         return cpus;
 }
+
+static void cpu_map__delete(struct perf_cpu_map *map)
+{
+        if (map) {
+                WARN_ONCE(refcount_read(&map->refcnt) != 0,
+                          "cpu_map refcnt unbalanced\n");
+                free(map);
+        }
+}
+
+struct perf_cpu_map *perf_cpu_map__get(struct perf_cpu_map *map)
+{
+        if (map)
+                refcount_inc(&map->refcnt);
+        return map;
+}
+
+void perf_cpu_map__put(struct perf_cpu_map *map)
+{
+        if (map && refcount_dec_and_test(&map->refcnt))
+                cpu_map__delete(map);
+}