2 * Copyright 2015-2018 The OpenSSL Project Authors. All Rights Reserved.
4 * Licensed under the Apache License 2.0 (the "License"). You may not use
5 * this file except in compliance with the License. You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
16 #include <openssl/async.h>
17 #include <openssl/crypto.h>
20 static ASYNC_JOB
*currjob
= NULL
;
22 static int only_pause(void *args
)
29 static int add_two(void *args
)
38 static int save_current(void *args
)
40 currjob
= ASYNC_get_current_job();
46 #define MAGIC_WAIT_FD ((OSSL_ASYNC_FD)99)
47 static int waitfd(void *args
)
50 ASYNC_WAIT_CTX
*waitctx
;
51 job
= ASYNC_get_current_job();
54 waitctx
= ASYNC_get_wait_ctx(job
);
58 /* First case: no fd added or removed */
61 /* Second case: one fd added */
62 if (!ASYNC_WAIT_CTX_set_wait_fd(waitctx
, waitctx
, MAGIC_WAIT_FD
, NULL
, NULL
))
66 /* Third case: all fd removed */
67 if (!ASYNC_WAIT_CTX_clear_fd(waitctx
, waitctx
))
71 /* Last case: fd added and immediately removed */
72 if (!ASYNC_WAIT_CTX_set_wait_fd(waitctx
, waitctx
, MAGIC_WAIT_FD
, NULL
, NULL
))
74 if (!ASYNC_WAIT_CTX_clear_fd(waitctx
, waitctx
))
80 static int blockpause(void *args
)
84 ASYNC_unblock_pause();
90 static int test_ASYNC_init_thread(void)
92 ASYNC_JOB
*job1
= NULL
, *job2
= NULL
, *job3
= NULL
;
93 int funcret1
, funcret2
, funcret3
;
94 ASYNC_WAIT_CTX
*waitctx
= NULL
;
96 if ( !ASYNC_init_thread(2, 0)
97 || (waitctx
= ASYNC_WAIT_CTX_new()) == NULL
98 || ASYNC_start_job(&job1
, waitctx
, &funcret1
, only_pause
, NULL
, 0)
100 || ASYNC_start_job(&job2
, waitctx
, &funcret2
, only_pause
, NULL
, 0)
102 || ASYNC_start_job(&job3
, waitctx
, &funcret3
, only_pause
, NULL
, 0)
104 || ASYNC_start_job(&job1
, waitctx
, &funcret1
, only_pause
, NULL
, 0)
106 || ASYNC_start_job(&job3
, waitctx
, &funcret3
, only_pause
, NULL
, 0)
108 || ASYNC_start_job(&job2
, waitctx
, &funcret2
, only_pause
, NULL
, 0)
110 || ASYNC_start_job(&job3
, waitctx
, &funcret3
, only_pause
, NULL
, 0)
115 fprintf(stderr
, "test_ASYNC_init_thread() failed\n");
116 ASYNC_WAIT_CTX_free(waitctx
);
117 ASYNC_cleanup_thread();
121 ASYNC_WAIT_CTX_free(waitctx
);
122 ASYNC_cleanup_thread();
126 static int test_callback(void *arg
)
128 printf("callback test pass\n");
132 static int test_ASYNC_callback_status(void)
134 ASYNC_WAIT_CTX
*waitctx
= NULL
;
136 ASYNC_callback_fn get_callback
;
140 if ( !ASYNC_init_thread(1, 0)
141 || (waitctx
= ASYNC_WAIT_CTX_new()) == NULL
142 || ASYNC_WAIT_CTX_set_callback(waitctx
, test_callback
, (void*)&set_arg
)
144 || ASYNC_WAIT_CTX_get_callback(waitctx
, &get_callback
, &get_arg
)
146 || test_callback
!= get_callback
147 || get_arg
!= (void*)&set_arg
148 || (*get_callback
)(get_arg
) != 1
149 || ASYNC_WAIT_CTX_set_status(waitctx
, set_status
) != 1
150 || set_status
!= ASYNC_WAIT_CTX_get_status(waitctx
)) {
151 fprintf(stderr
, "test_ASYNC_callback_status() failed\n");
152 ASYNC_WAIT_CTX_free(waitctx
);
153 ASYNC_cleanup_thread();
157 ASYNC_WAIT_CTX_free(waitctx
);
158 ASYNC_cleanup_thread();
163 static int test_ASYNC_start_job(void)
165 ASYNC_JOB
*job
= NULL
;
167 ASYNC_WAIT_CTX
*waitctx
= NULL
;
171 if ( !ASYNC_init_thread(1, 0)
172 || (waitctx
= ASYNC_WAIT_CTX_new()) == NULL
173 || ASYNC_start_job(&job
, waitctx
, &funcret
, add_two
, NULL
, 0)
176 || ASYNC_start_job(&job
, waitctx
, &funcret
, add_two
, NULL
, 0)
180 fprintf(stderr
, "test_ASYNC_start_job() failed\n");
181 ASYNC_WAIT_CTX_free(waitctx
);
182 ASYNC_cleanup_thread();
186 ASYNC_WAIT_CTX_free(waitctx
);
187 ASYNC_cleanup_thread();
191 static int test_ASYNC_get_current_job(void)
193 ASYNC_JOB
*job
= NULL
;
195 ASYNC_WAIT_CTX
*waitctx
= NULL
;
199 if ( !ASYNC_init_thread(1, 0)
200 || (waitctx
= ASYNC_WAIT_CTX_new()) == NULL
201 || ASYNC_start_job(&job
, waitctx
, &funcret
, save_current
, NULL
, 0)
204 || ASYNC_start_job(&job
, waitctx
, &funcret
, save_current
, NULL
, 0)
207 fprintf(stderr
, "test_ASYNC_get_current_job() failed\n");
208 ASYNC_WAIT_CTX_free(waitctx
);
209 ASYNC_cleanup_thread();
213 ASYNC_WAIT_CTX_free(waitctx
);
214 ASYNC_cleanup_thread();
218 static int test_ASYNC_WAIT_CTX_get_all_fds(void)
220 ASYNC_JOB
*job
= NULL
;
222 ASYNC_WAIT_CTX
*waitctx
= NULL
;
223 OSSL_ASYNC_FD fd
= OSSL_BAD_ASYNC_FD
, delfd
= OSSL_BAD_ASYNC_FD
;
224 size_t numfds
, numdelfds
;
226 if ( !ASYNC_init_thread(1, 0)
227 || (waitctx
= ASYNC_WAIT_CTX_new()) == NULL
228 /* On first run we're not expecting any wait fds */
229 || ASYNC_start_job(&job
, waitctx
, &funcret
, waitfd
, NULL
, 0)
231 || !ASYNC_WAIT_CTX_get_all_fds(waitctx
, NULL
, &numfds
)
233 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx
, NULL
, &numfds
, NULL
,
237 /* On second run we're expecting one added fd */
238 || ASYNC_start_job(&job
, waitctx
, &funcret
, waitfd
, NULL
, 0)
240 || !ASYNC_WAIT_CTX_get_all_fds(waitctx
, NULL
, &numfds
)
242 || !ASYNC_WAIT_CTX_get_all_fds(waitctx
, &fd
, &numfds
)
243 || fd
!= MAGIC_WAIT_FD
244 || (fd
= OSSL_BAD_ASYNC_FD
, 0) /* Assign to something else */
245 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx
, NULL
, &numfds
, NULL
,
249 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx
, &fd
, &numfds
, NULL
,
251 || fd
!= MAGIC_WAIT_FD
252 /* On third run we expect one deleted fd */
253 || ASYNC_start_job(&job
, waitctx
, &funcret
, waitfd
, NULL
, 0)
255 || !ASYNC_WAIT_CTX_get_all_fds(waitctx
, NULL
, &numfds
)
257 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx
, NULL
, &numfds
, NULL
,
261 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx
, NULL
, &numfds
, &delfd
,
263 || delfd
!= MAGIC_WAIT_FD
264 /* On last run we are not expecting any wait fd */
265 || ASYNC_start_job(&job
, waitctx
, &funcret
, waitfd
, NULL
, 0)
267 || !ASYNC_WAIT_CTX_get_all_fds(waitctx
, NULL
, &numfds
)
269 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx
, NULL
, &numfds
, NULL
,
274 fprintf(stderr
, "test_ASYNC_get_wait_fd() failed\n");
275 ASYNC_WAIT_CTX_free(waitctx
);
276 ASYNC_cleanup_thread();
280 ASYNC_WAIT_CTX_free(waitctx
);
281 ASYNC_cleanup_thread();
285 static int test_ASYNC_block_pause(void)
287 ASYNC_JOB
*job
= NULL
;
289 ASYNC_WAIT_CTX
*waitctx
= NULL
;
291 if ( !ASYNC_init_thread(1, 0)
292 || (waitctx
= ASYNC_WAIT_CTX_new()) == NULL
293 || ASYNC_start_job(&job
, waitctx
, &funcret
, blockpause
, NULL
, 0)
295 || ASYNC_start_job(&job
, waitctx
, &funcret
, blockpause
, NULL
, 0)
298 fprintf(stderr
, "test_ASYNC_block_pause() failed\n");
299 ASYNC_WAIT_CTX_free(waitctx
);
300 ASYNC_cleanup_thread();
304 ASYNC_WAIT_CTX_free(waitctx
);
305 ASYNC_cleanup_thread();
309 int main(int argc
, char **argv
)
311 if (!ASYNC_is_capable()) {
313 "OpenSSL build is not ASYNC capable - skipping async tests\n");
315 CRYPTO_set_mem_debug(1);
316 CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON
);
318 if ( !test_ASYNC_init_thread()
319 || !test_ASYNC_callback_status()
320 || !test_ASYNC_start_job()
321 || !test_ASYNC_get_current_job()
322 || !test_ASYNC_WAIT_CTX_get_all_fds()
323 || !test_ASYNC_block_pause()) {