diff options
Diffstat (limited to 'tests/test-progs/pthread/src/test_pthread_create_seq.cpp')
-rw-r--r-- | tests/test-progs/pthread/src/test_pthread_create_seq.cpp | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/tests/test-progs/pthread/src/test_pthread_create_seq.cpp b/tests/test-progs/pthread/src/test_pthread_create_seq.cpp new file mode 100644 index 000000000..fdffd7957 --- /dev/null +++ b/tests/test-progs/pthread/src/test_pthread_create_seq.cpp @@ -0,0 +1,116 @@ +/* + * Copyright (c) 2018, Cornell University + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or + * without modification, are permitted provided that the following + * conditions are met: + * + * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * Redistributions in binary form must reproduce the above + * copyright notice, this list of conditions and the following + * disclaimer in the documentation and/or other materials provided + * with the distribution. + * + * Neither the name of Cornell University nor the names of its + * contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND + * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, + * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF + * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED + * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN + * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + * + * Authors: Tuan Ta + */ + +#include <pthread.h> + +#include <cstdlib> +#include <iostream> + +//------------------------------------------------------------------------ +// Create n threads and run them one after another +// Each child thread writes its thread id to an output array +//------------------------------------------------------------------------ + +#define MAX_N_WORKER_THREADS 10 + +typedef struct +{ + int tid; + int* output; +} ThreadArg; + +void* func( void* args ) +{ + ThreadArg* my_args = ( ThreadArg* ) args; + + // write tid to this thread's output + (*my_args->output) = my_args->tid; + + return nullptr; +} + +int main( int argc, const char* argv[] ) +{ + int n_worker_threads = 0; + + // allocate all threads + pthread_t* threads = new pthread_t[MAX_N_WORKER_THREADS]; + ThreadArg* t_args = new ThreadArg[MAX_N_WORKER_THREADS]; + + // create an output array for all threads + int* outputs = new int[MAX_N_WORKER_THREADS]; + int ret; + + // try to spawn as many worker threads as possible + for ( int tid = 0; tid < MAX_N_WORKER_THREADS; ++tid ) { + + // set up thread args + t_args[tid].tid = tid; + t_args[tid].output = outputs + tid; + + // spawn thread + ret = pthread_create( threads + tid, nullptr, func, &t_args[tid] ); + if (ret != 0 ) { + break; + } + + n_worker_threads++; + + // wait for the thread to join before moving on + pthread_join( threads[tid], nullptr ); + } + + // verify + bool passed = true; + for ( int i = 0; i < n_worker_threads; ++i ) { + if ( outputs[i] != i ) { + passed = false; + break; + } + } + + // clean up + delete[] threads; + delete[] t_args; + delete[] outputs; + + // failed if outputs are not correct or no worker thread was spawned + if (!passed || n_worker_threads < 1) + return EXIT_FAILURE; + + return EXIT_SUCCESS; +} |