From: Sean Christopherson <[email protected]>

Use pin_self_to_cpu() and pin_task_to_cpu() to pin the vCPU thread and
the stealer thread to pCPU0 in the steal_time test. Eliminating the usage
of pthread_attr_setaffinity_np() in particular allows building the test
against non-glibc C libraries.

Reported-by: Aqib Faruqui <[email protected]>
Closes: https://lore.kernel.org/all/[email protected]/
Signed-off-by: Sean Christopherson <[email protected]>
Link: https://lore.kernel.org/all/[email protected]/
[[email protected]: resend, fix changelog typos; original series was never 
reposted]
Signed-off-by: Hisam Mehboob <[email protected]>
---
 tools/testing/selftests/kvm/steal_time.c | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/tools/testing/selftests/kvm/steal_time.c 
b/tools/testing/selftests/kvm/steal_time.c
index 76fcdd1fd3cb..19a97d6a3607 100644
--- a/tools/testing/selftests/kvm/steal_time.c
+++ b/tools/testing/selftests/kvm/steal_time.c
@@ -508,9 +508,7 @@ int main(int ac, char **av)
 {
        struct kvm_vcpu *vcpus[NR_VCPUS];
        struct kvm_vm *vm;
-       pthread_attr_t attr;
        pthread_t thread;
-       cpu_set_t cpuset;
        unsigned int gpages;
        long stolen_time;
        long run_delay;
@@ -520,11 +518,7 @@ int main(int ac, char **av)
        verbose = ac > 1 && (!strncmp(av[1], "-v", 3) || !strncmp(av[1], 
"--verbose", 10));
 
        /* Set CPU affinity so we can force preemption of the VCPU */
-       CPU_ZERO(&cpuset);
-       CPU_SET(0, &cpuset);
-       pthread_attr_init(&attr);
-       pthread_attr_setaffinity_np(&attr, sizeof(cpu_set_t), &cpuset);
-       pthread_setaffinity_np(pthread_self(), sizeof(cpu_set_t), &cpuset);
+       pin_self_to_cpu(0);
 
        /* Create a VM and an identity mapped memslot for the steal time 
structure */
        vm = vm_create_with_vcpus(NR_VCPUS, guest_code, vcpus);
@@ -558,7 +552,9 @@ int main(int ac, char **av)
 
                /* Steal time from the VCPU. The steal time thread has the same 
CPU affinity as the VCPUs. */
                run_delay = get_run_delay();
-               pthread_create(&thread, &attr, do_steal_time, NULL);
+               pthread_create(&thread, NULL, do_steal_time, NULL);
+               pin_task_to_cpu(thread, 0);
+
                do
                        sched_yield();
                while (get_run_delay() - run_delay < MIN_RUN_DELAY_NS);
-- 
2.51.0


Reply via email to