16c872901SJiri Olsa #include "tests.h" 26c872901SJiri Olsa #include "cpumap.h" 36c872901SJiri Olsa 46c872901SJiri Olsa static int process_event_mask(struct perf_tool *tool __maybe_unused, 56c872901SJiri Olsa union perf_event *event, 66c872901SJiri Olsa struct perf_sample *sample __maybe_unused, 76c872901SJiri Olsa struct machine *machine __maybe_unused) 86c872901SJiri Olsa { 9f77b57adSJiri Olsa struct cpu_map_event *map_event = &event->cpu_map; 106c872901SJiri Olsa struct cpu_map_mask *mask; 116c872901SJiri Olsa struct cpu_map_data *data; 12f77b57adSJiri Olsa struct cpu_map *map; 136c872901SJiri Olsa int i; 146c872901SJiri Olsa 15f77b57adSJiri Olsa data = &map_event->data; 166c872901SJiri Olsa 176c872901SJiri Olsa TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__MASK); 186c872901SJiri Olsa 196c872901SJiri Olsa mask = (struct cpu_map_mask *)data->data; 206c872901SJiri Olsa 216c872901SJiri Olsa TEST_ASSERT_VAL("wrong nr", mask->nr == 1); 226c872901SJiri Olsa 236c872901SJiri Olsa for (i = 0; i < 20; i++) { 246c872901SJiri Olsa TEST_ASSERT_VAL("wrong cpu", test_bit(i, mask->mask)); 256c872901SJiri Olsa } 266c872901SJiri Olsa 27f77b57adSJiri Olsa map = cpu_map__new_data(data); 28f77b57adSJiri Olsa TEST_ASSERT_VAL("wrong nr", map->nr == 20); 29f77b57adSJiri Olsa 30f77b57adSJiri Olsa for (i = 0; i < 20; i++) { 31f77b57adSJiri Olsa TEST_ASSERT_VAL("wrong cpu", map->map[i] == i); 32f77b57adSJiri Olsa } 33f77b57adSJiri Olsa 34f77b57adSJiri Olsa cpu_map__put(map); 356c872901SJiri Olsa return 0; 366c872901SJiri Olsa } 376c872901SJiri Olsa 386c872901SJiri Olsa static int process_event_cpus(struct perf_tool *tool __maybe_unused, 396c872901SJiri Olsa union perf_event *event, 406c872901SJiri Olsa struct perf_sample *sample __maybe_unused, 416c872901SJiri Olsa struct machine *machine __maybe_unused) 426c872901SJiri Olsa { 43f77b57adSJiri Olsa struct cpu_map_event *map_event = &event->cpu_map; 446c872901SJiri Olsa struct cpu_map_entries *cpus; 456c872901SJiri Olsa struct cpu_map_data *data; 46f77b57adSJiri Olsa struct cpu_map *map; 476c872901SJiri Olsa 48f77b57adSJiri Olsa data = &map_event->data; 496c872901SJiri Olsa 506c872901SJiri Olsa TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__CPUS); 516c872901SJiri Olsa 526c872901SJiri Olsa cpus = (struct cpu_map_entries *)data->data; 536c872901SJiri Olsa 546c872901SJiri Olsa TEST_ASSERT_VAL("wrong nr", cpus->nr == 2); 556c872901SJiri Olsa TEST_ASSERT_VAL("wrong cpu", cpus->cpu[0] == 1); 566c872901SJiri Olsa TEST_ASSERT_VAL("wrong cpu", cpus->cpu[1] == 256); 57f77b57adSJiri Olsa 58f77b57adSJiri Olsa map = cpu_map__new_data(data); 59f77b57adSJiri Olsa TEST_ASSERT_VAL("wrong nr", map->nr == 2); 60f77b57adSJiri Olsa TEST_ASSERT_VAL("wrong cpu", map->map[0] == 1); 61f77b57adSJiri Olsa TEST_ASSERT_VAL("wrong cpu", map->map[1] == 256); 62f77b57adSJiri Olsa TEST_ASSERT_VAL("wrong refcnt", atomic_read(&map->refcnt) == 1); 63f77b57adSJiri Olsa cpu_map__put(map); 646c872901SJiri Olsa return 0; 656c872901SJiri Olsa } 666c872901SJiri Olsa 676c872901SJiri Olsa 686c872901SJiri Olsa int test__cpu_map_synthesize(int subtest __maybe_unused) 696c872901SJiri Olsa { 706c872901SJiri Olsa struct cpu_map *cpus; 716c872901SJiri Olsa 726c872901SJiri Olsa /* This one is better stores in mask. */ 736c872901SJiri Olsa cpus = cpu_map__new("0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19"); 746c872901SJiri Olsa 756c872901SJiri Olsa TEST_ASSERT_VAL("failed to synthesize map", 766c872901SJiri Olsa !perf_event__synthesize_cpu_map(NULL, cpus, process_event_mask, NULL)); 776c872901SJiri Olsa 786c872901SJiri Olsa cpu_map__put(cpus); 796c872901SJiri Olsa 806c872901SJiri Olsa /* This one is better stores in cpu values. */ 816c872901SJiri Olsa cpus = cpu_map__new("1,256"); 826c872901SJiri Olsa 836c872901SJiri Olsa TEST_ASSERT_VAL("failed to synthesize map", 846c872901SJiri Olsa !perf_event__synthesize_cpu_map(NULL, cpus, process_event_cpus, NULL)); 856c872901SJiri Olsa 866c872901SJiri Olsa cpu_map__put(cpus); 876c872901SJiri Olsa return 0; 886c872901SJiri Olsa } 89*a24020e6SJiri Olsa 90*a24020e6SJiri Olsa static int cpu_map_print(const char *str) 91*a24020e6SJiri Olsa { 92*a24020e6SJiri Olsa struct cpu_map *map = cpu_map__new(str); 93*a24020e6SJiri Olsa char buf[100]; 94*a24020e6SJiri Olsa 95*a24020e6SJiri Olsa if (!map) 96*a24020e6SJiri Olsa return -1; 97*a24020e6SJiri Olsa 98*a24020e6SJiri Olsa cpu_map__snprint(map, buf, sizeof(buf)); 99*a24020e6SJiri Olsa return !strcmp(buf, str); 100*a24020e6SJiri Olsa } 101*a24020e6SJiri Olsa 102*a24020e6SJiri Olsa int test__cpu_map_print(int subtest __maybe_unused) 103*a24020e6SJiri Olsa { 104*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1")); 105*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,5")); 106*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,3,5,7,9,11,13,15,17,19,21-40")); 107*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("2-5")); 108*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,3-6,8-10,24,35-37")); 109*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,3-6,8-10,24,35-37")); 110*a24020e6SJiri Olsa TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1-10,12-20,22-30,32-40")); 111*a24020e6SJiri Olsa return 0; 112*a24020e6SJiri Olsa } 113