libublksrv: use sched_setaffinity() instead of pthread function

pthread_setaffinity_np() is actually a wrapper around sched_setaffinity(). For our usecase we are trying to set affinity only of the current thread, so we can use sched_setaffinity() as a drop in replacement. This makes code more portable (< glibc2.3.4, bionic) and concise while keeping the functionality same. Signed-off-by: Sandeep Dhavale <[email protected]>
ublk-org · Nov 14, 2024 · adc4b7f · adc4b7f
1 parent 4ba6a89
commit adc4b7f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/lib/ublksrv.c b/lib/ublksrv.c
@@ -408,11 +408,8 @@ static void ublksrv_set_sched_affinity(struct _ublksrv_dev *dev,
 	const struct ublksrv_ctrl_dev *cdev = dev->ctrl_dev;
 	unsigned dev_id = cdev->dev_info.dev_id;
 	cpu_set_t *cpuset = ublksrv_get_queue_affinity(cdev, q_id);
-	pthread_t thread = pthread_self();
-	int ret;
 
-	ret = pthread_setaffinity_np(thread, sizeof(cpu_set_t), cpuset);
-	if (ret)
+	if (sched_setaffinity(0, sizeof(cpu_set_t), cpuset) < 0)
 		ublk_err("ublk dev %u queue %u set affinity failed",
 				dev_id, q_id);
 }