1 /************************************************************************** 2 * 3 * Copyright 2009-2010 VMware, Inc. 4 * All Rights Reserved. 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining a 7 * copy of this software and associated documentation files (the 8 * "Software"), to deal in the Software without restriction, including 9 * without limitation the rights to use, copy, modify, merge, publish, 10 * distribute, sub license, and/or sell copies of the Software, and to 11 * permit persons to whom the Software is furnished to do so, subject to 12 * the following conditions: 13 * 14 * The above copyright notice and this permission notice (including the 15 * next paragraph) shall be included in all copies or substantial portions 16 * of the Software. 17 * 18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. 21 * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR 22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, 23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE 24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 25 * 26 **************************************************************************/ 27 28 29 /* 30 * Test case for pipe_barrier. 31 * 32 * The test succeeds if no thread exits before all the other threads reach 33 * the barrier. 34 */ 35 36 37 #include <stdio.h> 38 #include <stdlib.h> 39 40 #include "os/os_thread.h" 41 #include "os/os_time.h" 42 #include "util/u_atomic.h" 43 44 45 #define NUM_THREADS 10 46 47 static int verbosity = 0; 48 49 static pipe_thread threads[NUM_THREADS]; 50 static pipe_barrier barrier; 51 static int thread_ids[NUM_THREADS]; 52 53 static volatile int waiting = 0; 54 static volatile int proceeded = 0; 55 56 57 #define LOG(fmt, ...) \ 58 if (verbosity > 0) { \ 59 fprintf(stdout, fmt, ##__VA_ARGS__); \ 60 } 61 62 #define CHECK(_cond) \ 63 if (!(_cond)) { \ 64 fprintf(stderr, "%s:%u: `%s` failed\n", __FILE__, __LINE__, #_cond); \ 65 _exit(EXIT_FAILURE); \ 66 } 67 68 69 static PIPE_THREAD_ROUTINE(thread_function, thread_data) 70 { 71 int thread_id = *((int *) thread_data); 72 73 LOG("thread %d starting\n", thread_id); 74 os_time_sleep(thread_id * 100 * 1000); 75 LOG("thread %d before barrier\n", thread_id); 76 77 CHECK(p_atomic_read(&proceeded) == 0); 78 p_atomic_inc(&waiting); 79 80 pipe_barrier_wait(&barrier); 81 82 CHECK(p_atomic_read(&waiting) == NUM_THREADS); 83 84 p_atomic_inc(&proceeded); 85 86 LOG("thread %d exiting\n", thread_id); 87 88 return 0; 89 } 90 91 92 int main(int argc, char *argv[]) 93 { 94 int i; 95 96 for (i = 1; i < argc; ++i) { 97 const char *arg = argv[i]; 98 if (strcmp(arg, "-v") == 0) { 99 ++verbosity; 100 } else { 101 fprintf(stderr, "error: unrecognized option `%s`\n", arg); 102 exit(EXIT_FAILURE); 103 } 104 } 105 106 // Disable buffering 107 setbuf(stdout, NULL); 108 109 LOG("pipe_barrier_test starting\n"); 110 111 pipe_barrier_init(&barrier, NUM_THREADS); 112 113 for (i = 0; i < NUM_THREADS; i++) { 114 thread_ids[i] = i; 115 threads[i] = pipe_thread_create(thread_function, (void *) &thread_ids[i]); 116 } 117 118 for (i = 0; i < NUM_THREADS; i++ ) { 119 pipe_thread_wait(threads[i]); 120 } 121 122 CHECK(p_atomic_read(&proceeded) == NUM_THREADS); 123 124 pipe_barrier_destroy(&barrier); 125 126 LOG("pipe_barrier_test exiting\n"); 127 128 return 0; 129 } 130