1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
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;
}
|