| /* |
| * Copyright 2015-2018 The OpenSSL Project Authors. All Rights Reserved. |
| * |
| * Licensed under the OpenSSL license (the "License"). You may not use |
| * this file except in compliance with the License. You can obtain a copy |
| * in the file LICENSE in the source distribution or at |
| * https://www.openssl.org/source/license.html |
| */ |
| |
| #ifdef _WIN32 |
| # include <windows.h> |
| #endif |
| |
| #include <stdio.h> |
| #include <string.h> |
| #include <openssl/async.h> |
| #include <openssl/crypto.h> |
| |
| static int ctr = 0; |
| static ASYNC_JOB *currjob = NULL; |
| |
| static int only_pause(void *args) |
| { |
| ASYNC_pause_job(); |
| |
| return 1; |
| } |
| |
| static int add_two(void *args) |
| { |
| ctr++; |
| ASYNC_pause_job(); |
| ctr++; |
| |
| return 2; |
| } |
| |
| static int save_current(void *args) |
| { |
| currjob = ASYNC_get_current_job(); |
| ASYNC_pause_job(); |
| |
| return 1; |
| } |
| |
| #define MAGIC_WAIT_FD ((OSSL_ASYNC_FD)99) |
| static int waitfd(void *args) |
| { |
| ASYNC_JOB *job; |
| ASYNC_WAIT_CTX *waitctx; |
| job = ASYNC_get_current_job(); |
| if (job == NULL) |
| return 0; |
| waitctx = ASYNC_get_wait_ctx(job); |
| if (waitctx == NULL) |
| return 0; |
| |
| /* First case: no fd added or removed */ |
| ASYNC_pause_job(); |
| |
| /* Second case: one fd added */ |
| if (!ASYNC_WAIT_CTX_set_wait_fd(waitctx, waitctx, MAGIC_WAIT_FD, NULL, NULL)) |
| return 0; |
| ASYNC_pause_job(); |
| |
| /* Third case: all fd removed */ |
| if (!ASYNC_WAIT_CTX_clear_fd(waitctx, waitctx)) |
| return 0; |
| ASYNC_pause_job(); |
| |
| /* Last case: fd added and immediately removed */ |
| if (!ASYNC_WAIT_CTX_set_wait_fd(waitctx, waitctx, MAGIC_WAIT_FD, NULL, NULL)) |
| return 0; |
| if (!ASYNC_WAIT_CTX_clear_fd(waitctx, waitctx)) |
| return 0; |
| |
| return 1; |
| } |
| |
| static int blockpause(void *args) |
| { |
| ASYNC_block_pause(); |
| ASYNC_pause_job(); |
| ASYNC_unblock_pause(); |
| ASYNC_pause_job(); |
| |
| return 1; |
| } |
| |
| static int test_ASYNC_init_thread(void) |
| { |
| ASYNC_JOB *job1 = NULL, *job2 = NULL, *job3 = NULL; |
| int funcret1, funcret2, funcret3; |
| ASYNC_WAIT_CTX *waitctx = NULL; |
| |
| if ( !ASYNC_init_thread(2, 0) |
| || (waitctx = ASYNC_WAIT_CTX_new()) == NULL |
| || ASYNC_start_job(&job1, waitctx, &funcret1, only_pause, NULL, 0) |
| != ASYNC_PAUSE |
| || ASYNC_start_job(&job2, waitctx, &funcret2, only_pause, NULL, 0) |
| != ASYNC_PAUSE |
| || ASYNC_start_job(&job3, waitctx, &funcret3, only_pause, NULL, 0) |
| != ASYNC_NO_JOBS |
| || ASYNC_start_job(&job1, waitctx, &funcret1, only_pause, NULL, 0) |
| != ASYNC_FINISH |
| || ASYNC_start_job(&job3, waitctx, &funcret3, only_pause, NULL, 0) |
| != ASYNC_PAUSE |
| || ASYNC_start_job(&job2, waitctx, &funcret2, only_pause, NULL, 0) |
| != ASYNC_FINISH |
| || ASYNC_start_job(&job3, waitctx, &funcret3, only_pause, NULL, 0) |
| != ASYNC_FINISH |
| || funcret1 != 1 |
| || funcret2 != 1 |
| || funcret3 != 1) { |
| fprintf(stderr, "test_ASYNC_init_thread() failed\n"); |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 0; |
| } |
| |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 1; |
| } |
| |
| static int test_ASYNC_start_job(void) |
| { |
| ASYNC_JOB *job = NULL; |
| int funcret; |
| ASYNC_WAIT_CTX *waitctx = NULL; |
| |
| ctr = 0; |
| |
| if ( !ASYNC_init_thread(1, 0) |
| || (waitctx = ASYNC_WAIT_CTX_new()) == NULL |
| || ASYNC_start_job(&job, waitctx, &funcret, add_two, NULL, 0) |
| != ASYNC_PAUSE |
| || ctr != 1 |
| || ASYNC_start_job(&job, waitctx, &funcret, add_two, NULL, 0) |
| != ASYNC_FINISH |
| || ctr != 2 |
| || funcret != 2) { |
| fprintf(stderr, "test_ASYNC_start_job() failed\n"); |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 0; |
| } |
| |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 1; |
| } |
| |
| static int test_ASYNC_get_current_job(void) |
| { |
| ASYNC_JOB *job = NULL; |
| int funcret; |
| ASYNC_WAIT_CTX *waitctx = NULL; |
| |
| currjob = NULL; |
| |
| if ( !ASYNC_init_thread(1, 0) |
| || (waitctx = ASYNC_WAIT_CTX_new()) == NULL |
| || ASYNC_start_job(&job, waitctx, &funcret, save_current, NULL, 0) |
| != ASYNC_PAUSE |
| || currjob != job |
| || ASYNC_start_job(&job, waitctx, &funcret, save_current, NULL, 0) |
| != ASYNC_FINISH |
| || funcret != 1) { |
| fprintf(stderr, "test_ASYNC_get_current_job() failed\n"); |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 0; |
| } |
| |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 1; |
| } |
| |
| static int test_ASYNC_WAIT_CTX_get_all_fds(void) |
| { |
| ASYNC_JOB *job = NULL; |
| int funcret; |
| ASYNC_WAIT_CTX *waitctx = NULL; |
| OSSL_ASYNC_FD fd = OSSL_BAD_ASYNC_FD, delfd = OSSL_BAD_ASYNC_FD; |
| size_t numfds, numdelfds; |
| |
| if ( !ASYNC_init_thread(1, 0) |
| || (waitctx = ASYNC_WAIT_CTX_new()) == NULL |
| /* On first run we're not expecting any wait fds */ |
| || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0) |
| != ASYNC_PAUSE |
| || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds) |
| || numfds != 0 |
| || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL, |
| &numdelfds) |
| || numfds != 0 |
| || numdelfds != 0 |
| /* On second run we're expecting one added fd */ |
| || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0) |
| != ASYNC_PAUSE |
| || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds) |
| || numfds != 1 |
| || !ASYNC_WAIT_CTX_get_all_fds(waitctx, &fd, &numfds) |
| || fd != MAGIC_WAIT_FD |
| || (fd = OSSL_BAD_ASYNC_FD, 0) /* Assign to something else */ |
| || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL, |
| &numdelfds) |
| || numfds != 1 |
| || numdelfds != 0 |
| || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, &fd, &numfds, NULL, |
| &numdelfds) |
| || fd != MAGIC_WAIT_FD |
| /* On third run we expect one deleted fd */ |
| || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0) |
| != ASYNC_PAUSE |
| || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds) |
| || numfds != 0 |
| || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL, |
| &numdelfds) |
| || numfds != 0 |
| || numdelfds != 1 |
| || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, &delfd, |
| &numdelfds) |
| || delfd != MAGIC_WAIT_FD |
| /* On last run we are not expecting any wait fd */ |
| || ASYNC_start_job(&job, waitctx, &funcret, waitfd, NULL, 0) |
| != ASYNC_FINISH |
| || !ASYNC_WAIT_CTX_get_all_fds(waitctx, NULL, &numfds) |
| || numfds != 0 |
| || !ASYNC_WAIT_CTX_get_changed_fds(waitctx, NULL, &numfds, NULL, |
| &numdelfds) |
| || numfds != 0 |
| || numdelfds != 0 |
| || funcret != 1) { |
| fprintf(stderr, "test_ASYNC_get_wait_fd() failed\n"); |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 0; |
| } |
| |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 1; |
| } |
| |
| static int test_ASYNC_block_pause(void) |
| { |
| ASYNC_JOB *job = NULL; |
| int funcret; |
| ASYNC_WAIT_CTX *waitctx = NULL; |
| |
| if ( !ASYNC_init_thread(1, 0) |
| || (waitctx = ASYNC_WAIT_CTX_new()) == NULL |
| || ASYNC_start_job(&job, waitctx, &funcret, blockpause, NULL, 0) |
| != ASYNC_PAUSE |
| || ASYNC_start_job(&job, waitctx, &funcret, blockpause, NULL, 0) |
| != ASYNC_FINISH |
| || funcret != 1) { |
| fprintf(stderr, "test_ASYNC_block_pause() failed\n"); |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 0; |
| } |
| |
| ASYNC_WAIT_CTX_free(waitctx); |
| ASYNC_cleanup_thread(); |
| return 1; |
| } |
| |
| int main(int argc, char **argv) |
| { |
| if (!ASYNC_is_capable()) { |
| fprintf(stderr, |
| "OpenSSL build is not ASYNC capable - skipping async tests\n"); |
| } else { |
| CRYPTO_set_mem_debug(1); |
| CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON); |
| |
| if ( !test_ASYNC_init_thread() |
| || !test_ASYNC_start_job() |
| || !test_ASYNC_get_current_job() |
| || !test_ASYNC_WAIT_CTX_get_all_fds() |
| || !test_ASYNC_block_pause()) { |
| return 1; |
| } |
| } |
| printf("PASS\n"); |
| return 0; |
| } |