/src/openssl36/crypto/async/async.c

Source
/*
 * Copyright 2015-2025 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the Apache License 2.0 (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
 */

/*
 * Without this we start getting longjmp crashes because it thinks we're jumping
 * up the stack when in fact we are jumping to an entirely different stack. The
 * cost of this is not having certain buffer overrun/underrun checks etc for
 * this source file :-(
 */
#undef _FORTIFY_SOURCE

/* This must be the first #include file */
#include "async_local.h"
#include "internal/threads_common.h"

#include <openssl/err.h>
#include "crypto/cryptlib.h"
#include <string.h>

#define ASYNC_JOB_RUNNING 0
#define ASYNC_JOB_PAUSING 1
#define ASYNC_JOB_PAUSED 2
#define ASYNC_JOB_STOPPING 3

static void async_delete_thread_state(void *arg);

static async_ctx *async_ctx_new(void)
{
    async_ctx *nctx;

    if (!ossl_init_thread_start(NULL, NULL, async_delete_thread_state))
        return NULL;

    nctx = OPENSSL_malloc(sizeof(*nctx));
    if (nctx == NULL)
        goto err;

    async_fibre_init_dispatcher(&nctx->dispatcher);
    nctx->currjob = NULL;
    nctx->blocked = 0;
    if (!CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_CTX_KEY,
            CRYPTO_THREAD_NO_CONTEXT, nctx))
        goto err;

    return nctx;
err:
    OPENSSL_free(nctx);

    return NULL;
}

async_ctx *async_get_ctx(void)
{
    return (async_ctx *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_CTX_KEY,
        CRYPTO_THREAD_NO_CONTEXT);
}

static int async_ctx_free(void)
{
    async_ctx *ctx;

    ctx = async_get_ctx();

    if (!CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_CTX_KEY,
            CRYPTO_THREAD_NO_CONTEXT, NULL))
        return 0;

    OPENSSL_free(ctx);

    return 1;
}

static ASYNC_JOB *async_job_new(void)
{
    ASYNC_JOB *job = NULL;

    job = OPENSSL_zalloc(sizeof(*job));
    if (job == NULL)
        return NULL;

    job->status = ASYNC_JOB_RUNNING;

    return job;
}

static void async_job_free(ASYNC_JOB *job)
{
    if (job != NULL) {
        OPENSSL_free(job->funcargs);
        async_fibre_free(&job->fibrectx);
        OPENSSL_free(job);
    }
}

static ASYNC_JOB *async_get_pool_job(void)
{
    ASYNC_JOB *job;
    async_pool *pool;

    pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
        CRYPTO_THREAD_NO_CONTEXT);
    if (pool == NULL) {
        /*
         * Pool has not been initialised, so init with the defaults, i.e.
         * no max size and no pre-created jobs
         */
        if (ASYNC_init_thread(0, 0) == 0)
            return NULL;
        pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
            CRYPTO_THREAD_NO_CONTEXT);
    }

    job = sk_ASYNC_JOB_pop(pool->jobs);
    if (job == NULL) {
        /* Pool is empty */
        if ((pool->max_size != 0) && (pool->curr_size >= pool->max_size))
            return NULL;

        job = async_job_new();
        if (job != NULL) {
            if (!async_fibre_makecontext(&job->fibrectx)) {
                async_job_free(job);
                return NULL;
            }
            pool->curr_size++;
        }
    }
    return job;
}

static void async_release_job(ASYNC_JOB *job)
{
    async_pool *pool;

    pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
        CRYPTO_THREAD_NO_CONTEXT);
    if (pool == NULL) {
        ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
        return;
    }
    OPENSSL_free(job->funcargs);
    job->funcargs = NULL;
    sk_ASYNC_JOB_push(pool->jobs, job);
}

void async_start_func(void)
{
    ASYNC_JOB *job;
    async_ctx *ctx = async_get_ctx();

    if (ctx == NULL) {
        ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
        return;
    }
    while (1) {
        /* Run the job */
        job = ctx->currjob;
        job->ret = job->func(job->funcargs);

        /* Stop the job */
        job->status = ASYNC_JOB_STOPPING;
        if (!async_fibre_swapcontext(&job->fibrectx,
                &ctx->dispatcher, 1)) {
            /*
             * Should not happen. Getting here will close the thread...can't do
             * much about it
             */
            ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
        }
    }
}

int ASYNC_start_job(ASYNC_JOB **job, ASYNC_WAIT_CTX *wctx, int *ret,
    int (*func)(void *), void *args, size_t size)
{
    async_ctx *ctx;
    OSSL_LIB_CTX *libctx;

    if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
        return ASYNC_ERR;

    ctx = async_get_ctx();
    if (ctx == NULL)
        ctx = async_ctx_new();
    if (ctx == NULL)
        return ASYNC_ERR;

    if (*job != NULL)
        ctx->currjob = *job;

    for (;;) {
        if (ctx->currjob != NULL) {
            if (ctx->currjob->status == ASYNC_JOB_STOPPING) {
                *ret = ctx->currjob->ret;
                ctx->currjob->waitctx = NULL;
                async_release_job(ctx->currjob);
                ctx->currjob = NULL;
                *job = NULL;
                return ASYNC_FINISH;
            }

            if (ctx->currjob->status == ASYNC_JOB_PAUSING) {
                *job = ctx->currjob;
                ctx->currjob->status = ASYNC_JOB_PAUSED;
                ctx->currjob = NULL;
                return ASYNC_PAUSE;
            }

            if (ctx->currjob->status == ASYNC_JOB_PAUSED) {
                if (*job == NULL)
                    return ASYNC_ERR;
                ctx->currjob = *job;

                /*
                 * Restore the default libctx to what it was the last time the
                 * fibre ran
                 */
                libctx = OSSL_LIB_CTX_set0_default(ctx->currjob->libctx);
                if (libctx == NULL) {
                    /* Failed to set the default context */
                    ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
                    goto err;
                }
                /* Resume previous job */
                if (!async_fibre_swapcontext(&ctx->dispatcher,
                        &ctx->currjob->fibrectx, 1)) {
                    ctx->currjob->libctx = OSSL_LIB_CTX_set0_default(libctx);
                    ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
                    goto err;
                }
                /*
                 * In case the fibre changed the default libctx we set it back
                 * again to what it was originally, and remember what it had
                 * been changed to.
                 */
                ctx->currjob->libctx = OSSL_LIB_CTX_set0_default(libctx);
                continue;
            }

            /* Should not happen */
            ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
            async_release_job(ctx->currjob);
            ctx->currjob = NULL;
            *job = NULL;
            return ASYNC_ERR;
        }

        /* Start a new job */
        if ((ctx->currjob = async_get_pool_job()) == NULL)
            return ASYNC_NO_JOBS;

        /* Check for size > 0 to avoid malloc(0) */
        if (args != NULL && size > 0) {
            ctx->currjob->funcargs = OPENSSL_malloc(size);
            if (ctx->currjob->funcargs == NULL) {
                async_release_job(ctx->currjob);
                ctx->currjob = NULL;
                return ASYNC_ERR;
            }
            memcpy(ctx->currjob->funcargs, args, size);
        } else {
            ctx->currjob->funcargs = NULL;
        }

        ctx->currjob->func = func;
        ctx->currjob->waitctx = wctx;
        libctx = ossl_lib_ctx_get_concrete(NULL);
        if (!async_fibre_swapcontext(&ctx->dispatcher,
                &ctx->currjob->fibrectx, 1)) {
            ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
            goto err;
        }
        /*
         * In case the fibre changed the default libctx we set it back again
         * to what it was, and remember what it had been changed to.
         */
        ctx->currjob->libctx = OSSL_LIB_CTX_set0_default(libctx);
    }

err:
    async_release_job(ctx->currjob);
    ctx->currjob = NULL;
    *job = NULL;
    return ASYNC_ERR;
}

int ASYNC_pause_job(void)
{
    ASYNC_JOB *job;
    async_ctx *ctx = async_get_ctx();

    if (ctx == NULL
        || ctx->currjob == NULL
        || ctx->blocked) {
        /*
         * Could be we've deliberately not been started within a job so this is
         * counted as success.
         */
        return 1;
    }

    job = ctx->currjob;
    job->status = ASYNC_JOB_PAUSING;

    if (!async_fibre_swapcontext(&job->fibrectx,
            &ctx->dispatcher, 1)) {
        ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
        return 0;
    }
    /* Reset counts of added and deleted fds */
    async_wait_ctx_reset_counts(job->waitctx);

    return 1;
}

static void async_empty_pool(async_pool *pool)
{
    ASYNC_JOB *job;

    if (pool == NULL || pool->jobs == NULL)
        return;

    do {
        job = sk_ASYNC_JOB_pop(pool->jobs);
        async_job_free(job);
    } while (job);
}

int async_init(void)
{
    return async_local_init();
}

void async_deinit(void)
{
    async_local_deinit();
}

int ASYNC_init_thread(size_t max_size, size_t init_size)
{
    async_pool *pool;
    size_t curr_size = 0;

    if (init_size > max_size || max_size > INT_MAX) {
        ERR_raise(ERR_LIB_ASYNC, ASYNC_R_INVALID_POOL_SIZE);
        return 0;
    }

    if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
        return 0;

    if (!ossl_init_thread_start(NULL, NULL, async_delete_thread_state))
        return 0;

    pool = OPENSSL_zalloc(sizeof(*pool));
    if (pool == NULL)
        return 0;

    pool->jobs = sk_ASYNC_JOB_new_reserve(NULL, (int)init_size);
    if (pool->jobs == NULL) {
        ERR_raise(ERR_LIB_ASYNC, ERR_R_CRYPTO_LIB);
        OPENSSL_free(pool);
        return 0;
    }

    pool->max_size = max_size;

    /* Pre-create jobs as required */
    while (init_size--) {
        ASYNC_JOB *job;
        job = async_job_new();
        if (job == NULL || !async_fibre_makecontext(&job->fibrectx)) {
            /*
             * Not actually fatal because we already created the pool, just
             * skip creation of any more jobs
             */
            async_job_free(job);
            break;
        }
        job->funcargs = NULL;
        sk_ASYNC_JOB_push(pool->jobs, job); /* Cannot fail due to reserve */
        curr_size++;
    }
    pool->curr_size = curr_size;
    if (!CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
            CRYPTO_THREAD_NO_CONTEXT, pool)) {
        ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SET_POOL);
        goto err;
    }

    return 1;
err:
    async_empty_pool(pool);
    sk_ASYNC_JOB_free(pool->jobs);
    OPENSSL_free(pool);
    return 0;
}

static void async_delete_thread_state(void *arg)
{
    async_pool *pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
        CRYPTO_THREAD_NO_CONTEXT);

    if (pool != NULL) {
        async_empty_pool(pool);
        sk_ASYNC_JOB_free(pool->jobs);
        OPENSSL_free(pool);
        CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
            CRYPTO_THREAD_NO_CONTEXT, NULL);
    }
    async_local_cleanup();
    async_ctx_free();
}

void ASYNC_cleanup_thread(void)
{
    if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
        return;

    async_delete_thread_state(NULL);
}

ASYNC_JOB *ASYNC_get_current_job(void)
{
    async_ctx *ctx;

    if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
        return NULL;

    ctx = async_get_ctx();
    if (ctx == NULL)
        return NULL;

    return ctx->currjob;
}

ASYNC_WAIT_CTX *ASYNC_get_wait_ctx(ASYNC_JOB *job)
{
    return job->waitctx;
}

void ASYNC_block_pause(void)
{
    async_ctx *ctx;

    if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
        return;

    ctx = async_get_ctx();
    if (ctx == NULL || ctx->currjob == NULL) {
        /*
         * We're not in a job anyway so ignore this
         */
        return;
    }
    ctx->blocked++;
}

void ASYNC_unblock_pause(void)
{
    async_ctx *ctx;

    if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
        return;

    ctx = async_get_ctx();
    if (ctx == NULL || ctx->currjob == NULL) {
        /*
         * We're not in a job anyway so ignore this
         */
        return;
    }
    if (ctx->blocked > 0)
        ctx->blocked--;
}

Coverage Report

Created: 2025-12-31 06:58

Line	Count	Source
1		/*
2		* Copyright 2015-2025 The OpenSSL Project Authors. All Rights Reserved.
3		*
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
8		*/
9
10		/*
11		* Without this we start getting longjmp crashes because it thinks we're jumping
12		* up the stack when in fact we are jumping to an entirely different stack. The
13		* cost of this is not having certain buffer overrun/underrun checks etc for
14		* this source file :-(
15		*/
16		#undef _FORTIFY_SOURCE
17
18		/* This must be the first #include file */
19		#include "async_local.h"
20		#include "internal/threads_common.h"
21
22		#include <openssl/err.h>
23		#include "crypto/cryptlib.h"
24		#include <string.h>
25
26	0	#define ASYNC_JOB_RUNNING 0
27	0	#define ASYNC_JOB_PAUSING 1
28	0	#define ASYNC_JOB_PAUSED 2
29	0	#define ASYNC_JOB_STOPPING 3
30
31		static void async_delete_thread_state(void *arg);
32
33		static async_ctx *async_ctx_new(void)
34	0	{
35	0	async_ctx *nctx;
36
37	0	if (!ossl_init_thread_start(NULL, NULL, async_delete_thread_state))
38	0	return NULL;
39
40	0	nctx = OPENSSL_malloc(sizeof(*nctx));
41	0	if (nctx == NULL)
42	0	goto err;
43
44	0	async_fibre_init_dispatcher(&nctx->dispatcher);
45	0	nctx->currjob = NULL;
46	0	nctx->blocked = 0;
47	0	if (!CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_CTX_KEY,
48	0	CRYPTO_THREAD_NO_CONTEXT, nctx))
49	0	goto err;
50
51	0	return nctx;
52	0	err:
53	0	OPENSSL_free(nctx);
54
55	0	return NULL;
56	0	}
57
58		async_ctx *async_get_ctx(void)
59	0	{
60	0	return (async_ctx *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_CTX_KEY,
61	0	CRYPTO_THREAD_NO_CONTEXT);
62	0	}
63
64		static int async_ctx_free(void)
65	0	{
66	0	async_ctx *ctx;
67
68	0	ctx = async_get_ctx();
69
70	0	if (!CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_CTX_KEY,
71	0	CRYPTO_THREAD_NO_CONTEXT, NULL))
72	0	return 0;
73
74	0	OPENSSL_free(ctx);
75
76	0	return 1;
77	0	}
78
79		static ASYNC_JOB *async_job_new(void)
80	0	{
81	0	ASYNC_JOB *job = NULL;
82
83	0	job = OPENSSL_zalloc(sizeof(*job));
84	0	if (job == NULL)
85	0	return NULL;
86
87	0	job->status = ASYNC_JOB_RUNNING;
88
89	0	return job;
90	0	}
91
92		static void async_job_free(ASYNC_JOB *job)
93	0	{
94	0	if (job != NULL) {
95	0	OPENSSL_free(job->funcargs);
96	0	async_fibre_free(&job->fibrectx);
97	0	OPENSSL_free(job);
98	0	}
99	0	}
100
101		static ASYNC_JOB *async_get_pool_job(void)
102	0	{
103	0	ASYNC_JOB *job;
104	0	async_pool *pool;
105
106	0	pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
107	0	CRYPTO_THREAD_NO_CONTEXT);
108	0	if (pool == NULL) {
109		/*
110		* Pool has not been initialised, so init with the defaults, i.e.
111		* no max size and no pre-created jobs
112		*/
113	0	if (ASYNC_init_thread(0, 0) == 0)
114	0	return NULL;
115	0	pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
116	0	CRYPTO_THREAD_NO_CONTEXT);
117	0	}
118
119	0	job = sk_ASYNC_JOB_pop(pool->jobs);
120	0	if (job == NULL) {
121		/* Pool is empty */
122	0	if ((pool->max_size != 0) && (pool->curr_size >= pool->max_size))
123	0	return NULL;
124
125	0	job = async_job_new();
126	0	if (job != NULL) {
127	0	if (!async_fibre_makecontext(&job->fibrectx)) {
128	0	async_job_free(job);
129	0	return NULL;
130	0	}
131	0	pool->curr_size++;
132	0	}
133	0	}
134	0	return job;
135	0	}
136
137		static void async_release_job(ASYNC_JOB *job)
138	0	{
139	0	async_pool *pool;
140
141	0	pool = (async_pool *)CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
142	0	CRYPTO_THREAD_NO_CONTEXT);
143	0	if (pool == NULL) {
144	0	ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
145	0	return;
146	0	}
147	0	OPENSSL_free(job->funcargs);
148	0	job->funcargs = NULL;
149	0	sk_ASYNC_JOB_push(pool->jobs, job);
150	0	}
151
152		void async_start_func(void)
153	0	{
154	0	ASYNC_JOB *job;
155	0	async_ctx *ctx = async_get_ctx();
156
157	0	if (ctx == NULL) {
158	0	ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
159	0	return;
160	0	}
161	0	while (1) {
162		/* Run the job */
163	0	job = ctx->currjob;
164	0	job->ret = job->func(job->funcargs);
165
166		/* Stop the job */
167	0	job->status = ASYNC_JOB_STOPPING;
168	0	if (!async_fibre_swapcontext(&job->fibrectx,
169	0	&ctx->dispatcher, 1)) {
170		/*
171		* Should not happen. Getting here will close the thread...can't do
172		* much about it
173		*/
174	0	ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
175	0	}
176	0	}
177	0	}
178
179		int ASYNC_start_job(ASYNC_JOB *job, ASYNC_WAIT_CTX wctx, int *ret,
180		int (func)(void ), void *args, size_t size)
181	0	{
182	0	async_ctx *ctx;
183	0	OSSL_LIB_CTX *libctx;
184
185	0	if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
186	0	return ASYNC_ERR;
187
188	0	ctx = async_get_ctx();
189	0	if (ctx == NULL)
190	0	ctx = async_ctx_new();
191	0	if (ctx == NULL)
192	0	return ASYNC_ERR;
193
194	0	if (*job != NULL)
195	0	ctx->currjob = *job;
196
197	0	for (;;) {
198	0	if (ctx->currjob != NULL) {
199	0	if (ctx->currjob->status == ASYNC_JOB_STOPPING) {
200	0	*ret = ctx->currjob->ret;
201	0	ctx->currjob->waitctx = NULL;
202	0	async_release_job(ctx->currjob);
203	0	ctx->currjob = NULL;
204	0	*job = NULL;
205	0	return ASYNC_FINISH;
206	0	}
207
208	0	if (ctx->currjob->status == ASYNC_JOB_PAUSING) {
209	0	*job = ctx->currjob;
210	0	ctx->currjob->status = ASYNC_JOB_PAUSED;
211	0	ctx->currjob = NULL;
212	0	return ASYNC_PAUSE;
213	0	}
214
215	0	if (ctx->currjob->status == ASYNC_JOB_PAUSED) {
216	0	if (*job == NULL)
217	0	return ASYNC_ERR;
218	0	ctx->currjob = *job;
219
220		/*
221		* Restore the default libctx to what it was the last time the
222		* fibre ran
223		*/
224	0	libctx = OSSL_LIB_CTX_set0_default(ctx->currjob->libctx);
225	0	if (libctx == NULL) {
226		/* Failed to set the default context */
227	0	ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
228	0	goto err;
229	0	}
230		/* Resume previous job */
231	0	if (!async_fibre_swapcontext(&ctx->dispatcher,
232	0	&ctx->currjob->fibrectx, 1)) {
233	0	ctx->currjob->libctx = OSSL_LIB_CTX_set0_default(libctx);
234	0	ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
235	0	goto err;
236	0	}
237		/*
238		* In case the fibre changed the default libctx we set it back
239		* again to what it was originally, and remember what it had
240		* been changed to.
241		*/
242	0	ctx->currjob->libctx = OSSL_LIB_CTX_set0_default(libctx);
243	0	continue;
244	0	}
245
246		/* Should not happen */
247	0	ERR_raise(ERR_LIB_ASYNC, ERR_R_INTERNAL_ERROR);
248	0	async_release_job(ctx->currjob);
249	0	ctx->currjob = NULL;
250	0	*job = NULL;
251	0	return ASYNC_ERR;
252	0	}
253
254		/* Start a new job */
255	0	if ((ctx->currjob = async_get_pool_job()) == NULL)
256	0	return ASYNC_NO_JOBS;
257
258		/* Check for size > 0 to avoid malloc(0) */
259	0	if (args != NULL && size > 0) {
260	0	ctx->currjob->funcargs = OPENSSL_malloc(size);
261	0	if (ctx->currjob->funcargs == NULL) {
262	0	async_release_job(ctx->currjob);
263	0	ctx->currjob = NULL;
264	0	return ASYNC_ERR;
265	0	}
266	0	memcpy(ctx->currjob->funcargs, args, size);
267	0	} else {
268	0	ctx->currjob->funcargs = NULL;
269	0	}
270
271	0	ctx->currjob->func = func;
272	0	ctx->currjob->waitctx = wctx;
273	0	libctx = ossl_lib_ctx_get_concrete(NULL);
274	0	if (!async_fibre_swapcontext(&ctx->dispatcher,
275	0	&ctx->currjob->fibrectx, 1)) {
276	0	ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
277	0	goto err;
278	0	}
279		/*
280		* In case the fibre changed the default libctx we set it back again
281		* to what it was, and remember what it had been changed to.
282		*/
283	0	ctx->currjob->libctx = OSSL_LIB_CTX_set0_default(libctx);
284	0	}
285
286	0	err:
287	0	async_release_job(ctx->currjob);
288	0	ctx->currjob = NULL;
289	0	*job = NULL;
290	0	return ASYNC_ERR;
291	0	}
292
293		int ASYNC_pause_job(void)
294	0	{
295	0	ASYNC_JOB *job;
296	0	async_ctx *ctx = async_get_ctx();
297
298	0	if (ctx == NULL
299	0	\|\| ctx->currjob == NULL
300	0	\|\| ctx->blocked) {
301		/*
302		* Could be we've deliberately not been started within a job so this is
303		* counted as success.
304		*/
305	0	return 1;
306	0	}
307
308	0	job = ctx->currjob;
309	0	job->status = ASYNC_JOB_PAUSING;
310
311	0	if (!async_fibre_swapcontext(&job->fibrectx,
312	0	&ctx->dispatcher, 1)) {
313	0	ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SWAP_CONTEXT);
314	0	return 0;
315	0	}
316		/* Reset counts of added and deleted fds */
317	0	async_wait_ctx_reset_counts(job->waitctx);
318
319	0	return 1;
320	0	}
321
322		static void async_empty_pool(async_pool *pool)
323	0	{
324	0	ASYNC_JOB *job;
325
326	0	if (pool == NULL \|\| pool->jobs == NULL)
327	0	return;
328
329	0	do {
330	0	job = sk_ASYNC_JOB_pop(pool->jobs);
331	0	async_job_free(job);
332	0	} while (job);
333	0	}
334
335		int async_init(void)
336	36	{
337	36	return async_local_init();
338	36	}
339
340		void async_deinit(void)
341	72	{
342	72	async_local_deinit();
343	72	}
344
345		int ASYNC_init_thread(size_t max_size, size_t init_size)
346	0	{
347	0	async_pool *pool;
348	0	size_t curr_size = 0;
349
350	0	if (init_size > max_size \|\| max_size > INT_MAX) {
351	0	ERR_raise(ERR_LIB_ASYNC, ASYNC_R_INVALID_POOL_SIZE);
352	0	return 0;
353	0	}
354
355	0	if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
356	0	return 0;
357
358	0	if (!ossl_init_thread_start(NULL, NULL, async_delete_thread_state))
359	0	return 0;
360
361	0	pool = OPENSSL_zalloc(sizeof(*pool));
362	0	if (pool == NULL)
363	0	return 0;
364
365	0	pool->jobs = sk_ASYNC_JOB_new_reserve(NULL, (int)init_size);
366	0	if (pool->jobs == NULL) {
367	0	ERR_raise(ERR_LIB_ASYNC, ERR_R_CRYPTO_LIB);
368	0	OPENSSL_free(pool);
369	0	return 0;
370	0	}
371
372	0	pool->max_size = max_size;
373
374		/* Pre-create jobs as required */
375	0	while (init_size--) {
376	0	ASYNC_JOB *job;
377	0	job = async_job_new();
378	0	if (job == NULL \|\| !async_fibre_makecontext(&job->fibrectx)) {
379		/*
380		* Not actually fatal because we already created the pool, just
381		* skip creation of any more jobs
382		*/
383	0	async_job_free(job);
384	0	break;
385	0	}
386	0	job->funcargs = NULL;
387	0	sk_ASYNC_JOB_push(pool->jobs, job); /* Cannot fail due to reserve */
388	0	curr_size++;
389	0	}
390	0	pool->curr_size = curr_size;
391	0	if (!CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
392	0	CRYPTO_THREAD_NO_CONTEXT, pool)) {
393	0	ERR_raise(ERR_LIB_ASYNC, ASYNC_R_FAILED_TO_SET_POOL);
394	0	goto err;
395	0	}
396
397	0	return 1;
398	0	err:
399	0	async_empty_pool(pool);
400	0	sk_ASYNC_JOB_free(pool->jobs);
401	0	OPENSSL_free(pool);
402	0	return 0;
403	0	}
404
405		static void async_delete_thread_state(void *arg)
406	0	{
407	0	async_pool pool = (async_pool )CRYPTO_THREAD_get_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
408	0	CRYPTO_THREAD_NO_CONTEXT);
409
410	0	if (pool != NULL) {
411	0	async_empty_pool(pool);
412	0	sk_ASYNC_JOB_free(pool->jobs);
413	0	OPENSSL_free(pool);
414	0	CRYPTO_THREAD_set_local_ex(CRYPTO_THREAD_LOCAL_ASYNC_POOL_KEY,
415	0	CRYPTO_THREAD_NO_CONTEXT, NULL);
416	0	}
417	0	async_local_cleanup();
418	0	async_ctx_free();
419	0	}
420
421		void ASYNC_cleanup_thread(void)
422	0	{
423	0	if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
424	0	return;
425
426	0	async_delete_thread_state(NULL);
427	0	}
428
429		ASYNC_JOB *ASYNC_get_current_job(void)
430	0	{
431	0	async_ctx *ctx;
432
433	0	if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
434	0	return NULL;
435
436	0	ctx = async_get_ctx();
437	0	if (ctx == NULL)
438	0	return NULL;
439
440	0	return ctx->currjob;
441	0	}
442
443		ASYNC_WAIT_CTX ASYNC_get_wait_ctx(ASYNC_JOB job)
444	0	{
445	0	return job->waitctx;
446	0	}
447
448		void ASYNC_block_pause(void)
449	0	{
450	0	async_ctx *ctx;
451
452	0	if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
453	0	return;
454
455	0	ctx = async_get_ctx();
456	0	if (ctx == NULL \|\| ctx->currjob == NULL) {
457		/*
458		* We're not in a job anyway so ignore this
459		*/
460	0	return;
461	0	}
462	0	ctx->blocked++;
463	0	}
464
465		void ASYNC_unblock_pause(void)
466	0	{
467	0	async_ctx *ctx;
468
469	0	if (!OPENSSL_init_crypto(OPENSSL_INIT_ASYNC, NULL))
470	0	return;
471
472	0	ctx = async_get_ctx();
473	0	if (ctx == NULL \|\| ctx->currjob == NULL) {
474		/*
475		* We're not in a job anyway so ignore this
476		*/
477	0	return;
478	0	}
479	0	if (ctx->blocked > 0)
480	0	ctx->blocked--;
481	0	}