tests/rcutorture: Factor out thread registration

[urcu.git] / tests / regression / rcutorture.h

diff --git a/tests/regression/rcutorture.h b/tests/regression/rcutorture.h
index 49dbff689e9e9da835aa2186a6ec3a711192e1b9..e20d82eeb15df7873f74e8b795b16f8d111110ec 100644 (file)
--- a/tests/regression/rcutorture.h
+++ b/tests/regression/rcutorture.h
@@ -1,3 +1,7 @@
+// SPDX-FileCopyrightText: 2008 Paul E. McKenney, IBM Corporation.
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
 /*
  * rcutorture.h: simple user-level performance/stress test of RCU.
  *
@@ -43,22 +47,6 @@
  * line lists the number of readers observing progressively more stale
  * data.  A correct RCU implementation will have all but the first two
  * numbers non-zero.
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- *
- * Copyright (c) 2008 Paul E. McKenney, IBM Corporation.
  */
 
 /*
@@ -348,6 +336,7 @@ void *rcu_read_stress_test(void *arg __attribute__((unused)))
 
 static pthread_mutex_t call_rcu_test_mutex = PTHREAD_MUTEX_INITIALIZER;
 static pthread_cond_t call_rcu_test_cond = PTHREAD_COND_INITIALIZER;
+static bool call_rcu_wait;
 
 static
 void rcu_update_stress_test_rcu(struct rcu_head *head __attribute__((unused)))
@@ -368,6 +357,7 @@ void rcu_update_stress_test_rcu(struct rcu_head *head __attribute__((unused)))
                        strerror(errno));
                abort();
        }
+       call_rcu_wait = false;
        ret = pthread_mutex_unlock(&call_rcu_test_mutex);
        if (ret) {
                errno = ret;
@@ -401,8 +391,14 @@ void *rcu_update_stress_test(void *arg __attribute__((unused)))
        struct rcu_head rh;
        enum writer_state writer_state = WRITER_STATE_SYNC_RCU;
 
+       rcu_register_thread();
+
+       /* Offline for poll. */
+       put_thread_offline();
        while (goflag == GOFLAG_INIT)
                (void) poll(NULL, 0, 1);
+       put_thread_online();
+
        while (goflag == GOFLAG_RUN) {
                i = rcu_stress_idx + 1;
                if (i >= RCU_STRESS_PIPE_LEN)
@@ -432,9 +428,8 @@ void *rcu_update_stress_test(void *arg __attribute__((unused)))
                                        strerror(errno));
                                abort();
                        }
-                       rcu_register_thread();
                        call_rcu(&rh, rcu_update_stress_test_rcu);
-                       rcu_unregister_thread();
+
                        /*
                         * Our MacOS X test machine with the following
                         * config:
@@ -449,8 +444,14 @@ void *rcu_update_stress_test(void *arg __attribute__((unused)))
                         * immediately after call_rcu (call_rcu needs
                         * us to be registered RCU readers).
                         */
-                       ret = pthread_cond_wait(&call_rcu_test_cond,
-                                       &call_rcu_test_mutex);
+                       call_rcu_wait = true;
+                       /* Offline for pthread_cond_wait. */
+                       put_thread_offline();
+                       do {
+                               ret = pthread_cond_wait(&call_rcu_test_cond,
+                                                       &call_rcu_test_mutex);
+                       } while (call_rcu_wait);
+                       put_thread_online();
                        if (ret) {
                                errno = ret;
                                diag("pthread_cond_signal: %s",
@@ -470,11 +471,13 @@ void *rcu_update_stress_test(void *arg __attribute__((unused)))
                {
                        struct urcu_gp_poll_state poll_state;
 
-                       rcu_register_thread();
                        poll_state = start_poll_synchronize_rcu();
-                       rcu_unregister_thread();
+
+                       /* Offline for poll. */
+                       put_thread_offline();
                        while (!poll_state_synchronize_rcu(poll_state))
                                (void) poll(NULL, 0, 1);        /* Wait for 1ms */
+                       put_thread_online();
                        break;
                }
                }
@@ -482,6 +485,8 @@ void *rcu_update_stress_test(void *arg __attribute__((unused)))
                advance_writer_state(&writer_state);
        }
 
+       rcu_unregister_thread();
+
        return NULL;
 }
 
@@ -572,7 +577,7 @@ int stresstest(int nreaders)
  */
 
 static
-void usage(char *argv[]) __attribute__((noreturn));
+void usage(char *argv[]) __attribute__((__noreturn__));
 
 static
 void usage(char *argv[])