528 					   int core_id, u32 vcpu_id)  in ne_set_enclave_threads_per_core()  argument
532 	if (core_id < 0 && vcpu_id == 0) {  in ne_set_enclave_threads_per_core()
541 				    "CPU %d is not in NE CPU pool\n", vcpu_id);  in ne_set_enclave_threads_per_core()
573 static int ne_get_cpu_from_cpu_pool(struct ne_enclave *ne_enclave, u32 *vcpu_id)  in ne_get_cpu_from_cpu_pool()  argument
588 				*vcpu_id = cpu;  in ne_get_cpu_from_cpu_pool()
601 	rc = ne_set_enclave_threads_per_core(ne_enclave, core_id, *vcpu_id);  in ne_get_cpu_from_cpu_pool()
605 	*vcpu_id = cpumask_any(ne_enclave->threads_per_core[core_id]);  in ne_get_cpu_from_cpu_pool()
626 static int ne_get_vcpu_core_from_cpu_pool(u32 vcpu_id)  in ne_get_vcpu_core_from_cpu_pool()  argument
632 		if (cpumask_test_cpu(vcpu_id, ne_cpu_pool.avail_threads_per_core[i])) {  in ne_get_vcpu_core_from_cpu_pool()
652 static int ne_check_cpu_in_cpu_pool(struct ne_enclave *ne_enclave, u32 vcpu_id)  in ne_check_cpu_in_cpu_pool()  argument
658 	if (ne_donated_cpu(ne_enclave, vcpu_id)) {  in ne_check_cpu_in_cpu_pool()
660 				    "CPU %d already used\n", vcpu_id);  in ne_check_cpu_in_cpu_pool()
670 		if (cpumask_test_cpu(vcpu_id, ne_enclave->threads_per_core[i]))  in ne_check_cpu_in_cpu_pool()
680 	core_id = ne_get_vcpu_core_from_cpu_pool(vcpu_id);  in ne_check_cpu_in_cpu_pool()
682 	rc = ne_set_enclave_threads_per_core(ne_enclave, core_id, vcpu_id);  in ne_check_cpu_in_cpu_pool()
706 static int ne_add_vcpu_ioctl(struct ne_enclave *ne_enclave, u32 vcpu_id)  in ne_add_vcpu_ioctl()  argument
717 	slot_add_vcpu_req.vcpu_id = vcpu_id;  in ne_add_vcpu_ioctl()
729 	cpumask_set_cpu(vcpu_id, ne_enclave->vcpu_ids);  in ne_add_vcpu_ioctl()
1152 		u32 vcpu_id = 0;  in ne_enclave_ioctl()  local
1154 		if (copy_from_user(&vcpu_id, (void __user *)arg, sizeof(vcpu_id)))  in ne_enclave_ioctl()
1168 		if (vcpu_id >= (ne_enclave->nr_parent_vm_cores *  in ne_enclave_ioctl()
1178 		if (!vcpu_id) {  in ne_enclave_ioctl()
1180 			rc = ne_get_cpu_from_cpu_pool(ne_enclave, &vcpu_id);  in ne_enclave_ioctl()
1192 			rc = ne_check_cpu_in_cpu_pool(ne_enclave, vcpu_id);  in ne_enclave_ioctl()
1196 						    vcpu_id, rc);  in ne_enclave_ioctl()
1204 		rc = ne_add_vcpu_ioctl(ne_enclave, vcpu_id);  in ne_enclave_ioctl()
1213 		if (copy_to_user((void __user *)arg, &vcpu_id, sizeof(vcpu_id)))  in ne_enclave_ioctl()