riscv/hwprobe/which-cpus.c

22 	       "which-cpus: [-h] [<key=value> [<key=value> ...]]\n\n"
25 	       "   <key=value>, outputs the cpulist for cpus which all match the given set\n"
29 static void print_cpulist(cpu_set_t *cpus)
33 	if (!CPU_COUNT(cpus)) {
34 		printf("cpus: None\n");
38 	printf("cpus:");
39 	for (int i = 0, c = 0; i < CPU_COUNT(cpus); i++, c++) {
40 		if (start != end && !CPU_ISSET(c, cpus))
43 		while (!CPU_ISSET(c, cpus))
59 static void do_which_cpus(int argc, char **argv, cpu_set_t *cpus)
78 	rc = riscv_hwprobe(pairs, nr_pairs, sizeof(cpu_set_t), (unsigned long *)cpus, RISCV_HWPROBE_WHICH_CPUS);
80 	print_cpulist(cpus);
87 	cpu_set_t cpus_aff, cpus;
116 	CPU_ZERO(&cpus);
117 	rc = riscv_hwprobe(pairs, 1, 0, (unsigned long *)&cpus, RISCV_HWPROBE_WHICH_CPUS);
122 	ksft_test_result(rc == -EINVAL, "NULL cpus\n");
125 	CPU_ZERO(&cpus);
126 	rc = riscv_hwprobe(pairs, 1, sizeof(cpu_set_t), (unsigned long *)&cpus, RISCV_HWPROBE_WHICH_CPUS);
127 	ksft_test_result(rc == 0 && CPU_COUNT(&cpus) == 0, "unknown key\n");
131 	CPU_ZERO(&cpus);
132 	rc = riscv_hwprobe(pairs, 2, sizeof(cpu_set_t), (unsigned long *)&cpus, RISCV_HWPROBE_WHICH_CPUS);
137 	CPU_ZERO(&cpus);
138 	rc = riscv_hwprobe(pairs, 2, sizeof(cpu_set_t), (unsigned long *)&cpus, RISCV_HWPROBE_WHICH_CPUS);
139 	ksft_test_result(rc == 0 && CPU_COUNT(&cpus) == sysconf(_SC_NPROCESSORS_ONLN), "set all cpus\n");
143 	memcpy(&cpus, &cpus_aff, sizeof(cpu_set_t));
144 	rc = riscv_hwprobe(pairs, 2, sizeof(cpu_set_t), (unsigned long *)&cpus, RISCV_HWPROBE_WHICH_CPUS);
145 	ksft_test_result(rc == 0 && CPU_EQUAL(&cpus, &cpus_aff), "set all affinity cpus\n");
149 	memcpy(&cpus, &cpus_aff, sizeof(cpu_set_t));
150 	rc = riscv_hwprobe(pairs, 2, sizeof(cpu_set_t), (unsigned long *)&cpus, RISCV_HWPROBE_WHICH_CPUS);
151 	ksft_test_result(rc == 0 && CPU_COUNT(&cpus) == 0, "clear all cpus\n");