summaryrefslogtreecommitdiff
path: root/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c
diff options
context:
space:
mode:
Diffstat (limited to 'sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c')
-rw-r--r--sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c280
1 files changed, 280 insertions, 0 deletions
diff --git a/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c b/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c
new file mode 100644
index 0000000000..d9677436d6
--- /dev/null
+++ b/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c
@@ -0,0 +1,280 @@
+/* Generic test for CPU affinity functions, multi-threaded variant.
+ Copyright (C) 2015-2016 Free Software Foundation, Inc.
+ This file is part of the GNU C Library.
+
+ The GNU C Library is free software; you can redistribute it and/or
+ modify it under the terms of the GNU Lesser General Public
+ License as published by the Free Software Foundation; either
+ version 2.1 of the License, or (at your option) any later version.
+
+ The GNU C Library is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public
+ License along with the GNU C Library; if not, see
+ <http://www.gnu.org/licenses/>. */
+
+/* Before including this file, a test has to declare the helper
+ getaffinity and setaffinity functions described in
+ tst-skeleton-affinity.c, which is included below. */
+
+#include <errno.h>
+#include <pthread.h>
+#include <stdbool.h>
+#include <stdlib.h>
+#include <sys/time.h>
+
+struct conf;
+static bool early_test (struct conf *);
+
+/* Arbitrary run time for each pass. */
+#define PASS_TIMEOUT 2
+
+/* There are two passes (one with sched_yield, one without), and we
+ double the timeout to be on the safe side. */
+#define TIMEOUT (2 * PASS_TIMEOUT * 2)
+
+#include "tst-skeleton-affinity.c"
+
+/* 0 if still running, 1 of stopping requested. */
+static int still_running;
+
+/* 0 if no scheduling failures, 1 if failures are encountered. */
+static int failed;
+
+static void *
+thread_burn_one_cpu (void *closure)
+{
+ int cpu = (uintptr_t) closure;
+ while (__atomic_load_n (&still_running, __ATOMIC_RELAXED) == 0)
+ {
+ int current = sched_getcpu ();
+ if (sched_getcpu () != cpu)
+ {
+ printf ("error: Pinned thread %d ran on impossible cpu %d\n",
+ cpu, current);
+ __atomic_store_n (&failed, 1, __ATOMIC_RELAXED);
+ /* Terminate early. */
+ __atomic_store_n (&still_running, 1, __ATOMIC_RELAXED);
+ }
+ }
+ return NULL;
+}
+
+struct burn_thread
+{
+ pthread_t self;
+ struct conf *conf;
+ cpu_set_t *initial_set;
+ cpu_set_t *seen_set;
+ int thread;
+};
+
+static void *
+thread_burn_any_cpu (void *closure)
+{
+ struct burn_thread *param = closure;
+
+ /* Schedule this thread around a bit to see if it lands on another
+ CPU. Run this for 2 seconds, once with sched_yield, once
+ without. */
+ for (int pass = 1; pass <= 2; ++pass)
+ {
+ time_t start = time (NULL);
+ while (time (NULL) - start <= PASS_TIMEOUT)
+ {
+ int cpu = sched_getcpu ();
+ if (cpu > param->conf->last_cpu
+ || !CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (param->conf->set_size),
+ param->initial_set))
+ {
+ printf ("error: Unpinned thread %d ran on impossible CPU %d\n",
+ param->thread, cpu);
+ __atomic_store_n (&failed, 1, __ATOMIC_RELAXED);
+ return NULL;
+ }
+ CPU_SET_S (cpu, CPU_ALLOC_SIZE (param->conf->set_size),
+ param->seen_set);
+ if (pass == 1)
+ sched_yield ();
+ }
+ }
+ return NULL;
+}
+
+static void
+stop_and_join_threads (struct conf *conf, cpu_set_t *set,
+ pthread_t *pinned_first, pthread_t *pinned_last,
+ struct burn_thread *other_first,
+ struct burn_thread *other_last)
+{
+ __atomic_store_n (&still_running, 1, __ATOMIC_RELAXED);
+ for (pthread_t *p = pinned_first; p < pinned_last; ++p)
+ {
+ int cpu = p - pinned_first;
+ if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), set))
+ continue;
+
+ int ret = pthread_join (*p, NULL);
+ if (ret != 0)
+ {
+ printf ("error: Failed to join thread %d: %s\n", cpu, strerror (ret));
+ fflush (stdout);
+ /* Cannot shut down cleanly with threads still running. */
+ abort ();
+ }
+ }
+
+ for (struct burn_thread *p = other_first; p < other_last; ++p)
+ {
+ int cpu = p - other_first;
+ if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), set))
+ continue;
+
+ int ret = pthread_join (p->self, NULL);
+ if (ret != 0)
+ {
+ printf ("error: Failed to join thread %d: %s\n", cpu, strerror (ret));
+ fflush (stdout);
+ /* Cannot shut down cleanly with threads still running. */
+ abort ();
+ }
+ }
+}
+
+/* Tries to check that the initial set of CPUs is complete and that
+ the main thread will not run on any other threads. */
+static bool
+early_test (struct conf *conf)
+{
+ pthread_t *pinned_threads
+ = calloc (conf->last_cpu + 1, sizeof (*pinned_threads));
+ struct burn_thread *other_threads
+ = calloc (conf->last_cpu + 1, sizeof (*other_threads));
+ cpu_set_t *initial_set = CPU_ALLOC (conf->set_size);
+ cpu_set_t *scratch_set = CPU_ALLOC (conf->set_size);
+
+ if (pinned_threads == NULL || other_threads == NULL
+ || initial_set == NULL || scratch_set == NULL)
+ {
+ puts ("error: Memory allocation failure");
+ return false;
+ }
+ if (getaffinity (CPU_ALLOC_SIZE (conf->set_size), initial_set) < 0)
+ {
+ printf ("error: pthread_getaffinity_np failed: %m\n");
+ return false;
+ }
+ for (int cpu = 0; cpu <= conf->last_cpu; ++cpu)
+ {
+ if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set))
+ continue;
+ other_threads[cpu].conf = conf;
+ other_threads[cpu].initial_set = initial_set;
+ other_threads[cpu].thread = cpu;
+ other_threads[cpu].seen_set = CPU_ALLOC (conf->set_size);
+ if (other_threads[cpu].seen_set == NULL)
+ {
+ puts ("error: Memory allocation failure");
+ return false;
+ }
+ CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size),
+ other_threads[cpu].seen_set);
+ }
+
+ pthread_attr_t attr;
+ int ret = pthread_attr_init (&attr);
+ if (ret != 0)
+ {
+ printf ("error: pthread_attr_init failed: %s\n", strerror (ret));
+ return false;
+ }
+
+ /* Spawn a thread pinned to each available CPU. */
+ for (int cpu = 0; cpu <= conf->last_cpu; ++cpu)
+ {
+ if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set))
+ continue;
+ CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set);
+ CPU_SET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), scratch_set);
+ ret = pthread_attr_setaffinity_np
+ (&attr, CPU_ALLOC_SIZE (conf->set_size), scratch_set);
+ if (ret != 0)
+ {
+ printf ("error: pthread_attr_setaffinity_np for CPU %d failed: %s\n",
+ cpu, strerror (ret));
+ stop_and_join_threads (conf, initial_set,
+ pinned_threads, pinned_threads + cpu,
+ NULL, NULL);
+ return false;
+ }
+ ret = pthread_create (pinned_threads + cpu, &attr,
+ thread_burn_one_cpu, (void *) (uintptr_t) cpu);
+ if (ret != 0)
+ {
+ printf ("error: pthread_create for CPU %d failed: %s\n",
+ cpu, strerror (ret));
+ stop_and_join_threads (conf, initial_set,
+ pinned_threads, pinned_threads + cpu,
+ NULL, NULL);
+ return false;
+ }
+ }
+
+ /* Spawn another set of threads running on all CPUs. */
+ for (int cpu = 0; cpu <= conf->last_cpu; ++cpu)
+ {
+ if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set))
+ continue;
+ ret = pthread_create (&other_threads[cpu].self, NULL,
+ thread_burn_any_cpu, other_threads + cpu);
+ if (ret != 0)
+ {
+ printf ("error: pthread_create for thread %d failed: %s\n",
+ cpu, strerror (ret));
+ stop_and_join_threads (conf, initial_set,
+ pinned_threads,
+ pinned_threads + conf->last_cpu + 1,
+ other_threads, other_threads + cpu);
+ return false;
+ }
+ }
+
+ /* Main thread. */
+ struct burn_thread main_thread;
+ main_thread.conf = conf;
+ main_thread.initial_set = initial_set;
+ main_thread.seen_set = scratch_set;
+ main_thread.thread = -1;
+ CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), main_thread.seen_set);
+ thread_burn_any_cpu (&main_thread);
+ stop_and_join_threads (conf, initial_set,
+ pinned_threads,
+ pinned_threads + conf->last_cpu + 1,
+ other_threads, other_threads + conf->last_cpu + 1);
+
+ printf ("info: Main thread ran on %d CPU(s) of %d available CPU(s)\n",
+ CPU_COUNT_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set),
+ CPU_COUNT_S (CPU_ALLOC_SIZE (conf->set_size), initial_set));
+ CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set);
+ for (int cpu = 0; cpu <= conf->last_cpu; ++cpu)
+ {
+ if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set))
+ continue;
+ CPU_OR_S (CPU_ALLOC_SIZE (conf->set_size),
+ scratch_set, scratch_set, other_threads[cpu].seen_set);
+ CPU_FREE (other_threads[cpu].seen_set);
+ }
+ printf ("info: Other threads ran on %d CPU(s)\n",
+ CPU_COUNT_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set));;
+
+
+ pthread_attr_destroy (&attr);
+ CPU_FREE (scratch_set);
+ CPU_FREE (initial_set);
+ free (pinned_threads);
+ free (other_threads);
+ return failed == 0;
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-05-01 16:18:15 +0000