2 * Copyright 2015-2016 The OpenSSL Project Authors. All Rights Reserved.
4 * Licensed under the OpenSSL license (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
12 #include <openssl/async.h>
13 #include <openssl/crypto.h>
14 #include <../apps/apps.h>
17 static ASYNC_JOB *currjob = NULL;
19 static int only_pause(void *args)
26 static int add_two(void *args)
35 static int save_current(void *args)
37 currjob = ASYNC_get_current_job();
43 #define MAGIC_WAIT_FD ((OSSL_ASYNC_FD)99)
44 static int waitfd(void *args)
47 ASYNC_WAIT_CTX *waitctx;
49 job = ASYNC_get_current_job();
52 waitctx = ASYNC_get_wait_ctx(job);
55 if(!ASYNC_WAIT_CTX_set_wait_fd(waitctx, waitctx, MAGIC_WAIT_FD, NULL, NULL))
59 if (!ASYNC_WAIT_CTX_clear_fd(waitctx, waitctx))
65 static int blockpause(void *args)
69 ASYNC_unblock_pause();
75 static int test_ASYNC_init_thread()
77 ASYNC_JOB *job1 = NULL, *job2 = NULL, *job3 = NULL;
78 int funcret1, funcret2, funcret3;
79 ASYNC_WAIT_CTX *waitctx = NULL;
81 if ( !ASYNC_init_thread(2, 0)
82 || (waitctx = ASYNC_WAIT_CTX_new()) == NULL
83 || ASYNC_start_job(&job1, waitctx, &funcret1, only_pause, NULL, 0)
85 || ASYNC_start_job(&job2, waitctx, &funcret2, only_pause, NULL, 0)
87 || ASYNC_start_job(&job3, waitctx, &funcret3, only_pause, NULL, 0)
89 || ASYNC_start_job(&job1, waitctx, &funcret1, only_pause, NULL, 0)
91 || ASYNC_start_job(&job3, waitctx, &funcret3, only_pause, NULL, 0)
93 || ASYNC_start_job(&job2, waitctx, &funcret2, only_pause, NULL, 0)
95 || ASYNC_start_job(&job3, waitctx, &funcret3, only_pause, NULL, 0)
100 fprintf(stderr, "test_ASYNC_init_thread() failed\n");
101 ASYNC_WAIT_CTX_free(waitctx);
102 ASYNC_cleanup_thread();
106 ASYNC_WAIT_CTX_free(waitctx);
107 ASYNC_cleanup_thread();
111 static int test_ASYNC_start_job()
113 ASYNC_JOB *job = NULL;
115 ASYNC_WAIT_CTX *waitctx = NULL;
119 if ( !ASYNC_init_thread(1, 0)
120 || (waitctx = ASYNC_WAIT_CTX_new()) == NULL
121 || ASYNC_start_job(&job, waitctx, &funcret, add_two, NULL, 0)
124 || ASYNC_start_job(&job, waitctx, &funcret, add_two, NULL, 0)
128 fprintf(stderr, "test_ASYNC_start_job() failed\n");
129 ASYNC_WAIT_CTX_free(waitctx);
130 ASYNC_cleanup_thread();
134 ASYNC_WAIT_CTX_free(waitctx);
135 ASYNC_cleanup_thread();
139 static int test_ASYNC_get_current_job()
141 ASYNC_JOB *job = NULL;
143 ASYNC_WAIT_CTX *waitctx = NULL;
147 if ( !ASYNC_init_thread(1, 0)
148 || (waitctx = ASYNC_WAIT_CTX_new()) == NULL
149 || ASYNC_start_job(&job, waitctx, &funcret, save_current, NULL, 0)
152 || ASYNC_start_job(&job, waitctx, &funcret, save_current, NULL, 0)
155 fprintf(stderr, "test_ASYNC_get_current_job() failed\n");
156 ASYNC_WAIT_CTX_free(waitctx);
157 ASYNC_cleanup_thread();
161 ASYNC_WAIT_CTX_free(waitctx);
162 ASYNC_cleanup_thread();
166 static int test_ASYNC_WAIT_CTX_get_all_fds()
168 ASYNC_JOB *job = NULL;
170 ASYNC_WAIT_CTX *waitctx = NULL;
171 OSSL_ASYNC_FD fd = OSSL_BAD_ASYNC_FD, delfd = OSSL_BAD_ASYNC_FD;
172 size_t numfds, numdelfds;
174 if ( !ASYNC_init_thread(1, 0)
175 || (waitctx = ASYNC_WAIT_CTX_new()) == NULL
176 /* On first run we're not expecting any wait fds */
177 || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0)
179 || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds)
181 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL,
185 /* On second run we're expecting one added fd */
186 || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0)
188 || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds)
190 || !ASYNC_WAIT_CTX_get_all_fds(waitctx, &fd, &numfds)
191 || fd != MAGIC_WAIT_FD
192 || (fd = OSSL_BAD_ASYNC_FD, 0) /* Assign to something else */
193 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL,
197 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, &fd, &numfds, NULL,
199 || fd != MAGIC_WAIT_FD
200 /* On final run we expect one deleted fd */
201 || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0)
203 || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds)
205 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL,
209 || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, &delfd,
211 || delfd != MAGIC_WAIT_FD
213 fprintf(stderr, "test_ASYNC_get_wait_fd() failed\n");
214 ASYNC_WAIT_CTX_free(waitctx);
215 ASYNC_cleanup_thread();
219 ASYNC_WAIT_CTX_free(waitctx);
220 ASYNC_cleanup_thread();
224 static int test_ASYNC_block_pause()
226 ASYNC_JOB *job = NULL;
228 ASYNC_WAIT_CTX *waitctx = NULL;
230 if ( !ASYNC_init_thread(1, 0)
231 || (waitctx = ASYNC_WAIT_CTX_new()) == NULL
232 || ASYNC_start_job(&job, waitctx, &funcret, blockpause, NULL, 0)
234 || ASYNC_start_job(&job, waitctx, &funcret, blockpause, NULL, 0)
237 fprintf(stderr, "test_ASYNC_block_pause() failed\n");
238 ASYNC_WAIT_CTX_free(waitctx);
239 ASYNC_cleanup_thread();
243 ASYNC_WAIT_CTX_free(waitctx);
244 ASYNC_cleanup_thread();
248 int main(int argc, char **argv)
250 if (!ASYNC_is_capable()) {
252 "OpenSSL build is not ASYNC capable - skipping async tests\n");
254 CRYPTO_set_mem_debug(1);
255 CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
257 if ( !test_ASYNC_init_thread()
258 || !test_ASYNC_start_job()
259 || !test_ASYNC_get_current_job()
260 || !test_ASYNC_WAIT_CTX_get_all_fds()
261 || !test_ASYNC_block_pause()) {