perf cpu_map: Add cpu_map__new_event function

Introducing the cpu_map__new_event function to create a struct cpu_map
object from a cpu_map event.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>
Tested-by: Kan Liang <kan.liang@intel.com>
Cc: David Ahern <dsahern@gmail.com>
Cc: Namhyung Kim <namhyung@kernel.org>
Cc: Peter Zijlstra <a.p.zijlstra@chello.nl>
Link: http://lkml.kernel.org/r/1445784728-21732-10-git-send-email-jolsa@kernel.org
Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

1 files changed, 21 insertions, 4 deletions

diff --git a/tools/perf/tests/cpumap.c b/tools/perf/tests/cpumap.c
index 715480558088..4cb6418a8ffc 100644
--- a/tools/perf/tests/cpumap.c
+++ b/tools/perf/tests/cpumap.c
@@ -6,12 +6,13 @@ static int process_event_mask(struct perf_tool *tool __maybe_unused,
                          struct perf_sample *sample __maybe_unused,
                          struct machine *machine __maybe_unused)
 {
-        struct cpu_map_event *map = &event->cpu_map;
+        struct cpu_map_event *map_event = &event->cpu_map;
         struct cpu_map_mask *mask;
         struct cpu_map_data *data;
+        struct cpu_map *map;
         int i;
 
-        data = &map->data;
+        data = &map_event->data;
 
         TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__MASK);
 
@@ -23,6 +24,14 @@ static int process_event_mask(struct perf_tool *tool __maybe_unused,
                 TEST_ASSERT_VAL("wrong cpu", test_bit(i, mask->mask));
         }
 
+        map = cpu_map__new_data(data);
+        TEST_ASSERT_VAL("wrong nr",  map->nr == 20);
+
+        for (i = 0; i < 20; i++) {
+                TEST_ASSERT_VAL("wrong cpu", map->map[i] == i);
+        }
+
+        cpu_map__put(map);
         return 0;
 }
 
@@ -31,11 +40,12 @@ static int process_event_cpus(struct perf_tool *tool __maybe_unused,
                          struct perf_sample *sample __maybe_unused,
                          struct machine *machine __maybe_unused)
 {
-        struct cpu_map_event *map = &event->cpu_map;
+        struct cpu_map_event *map_event = &event->cpu_map;
         struct cpu_map_entries *cpus;
         struct cpu_map_data *data;
+        struct cpu_map *map;
 
-        data = &map->data;
+        data = &map_event->data;
 
         TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__CPUS);
 
@@ -44,6 +54,13 @@ static int process_event_cpus(struct perf_tool *tool __maybe_unused,
         TEST_ASSERT_VAL("wrong nr",   cpus->nr == 2);
         TEST_ASSERT_VAL("wrong cpu",  cpus->cpu[0] == 1);
         TEST_ASSERT_VAL("wrong cpu",  cpus->cpu[1] == 256);
+
+        map = cpu_map__new_data(data);
+        TEST_ASSERT_VAL("wrong nr",  map->nr == 2);
+        TEST_ASSERT_VAL("wrong cpu", map->map[0] == 1);
+        TEST_ASSERT_VAL("wrong cpu", map->map[1] == 256);
+        TEST_ASSERT_VAL("wrong refcnt", atomic_read(&map->refcnt) == 1);
+        cpu_map__put(map);
         return 0;
 }