Implement local thread pools
Implement the ASYNC_JOB as a local thread pool. Remove the API support for global pools. Reviewed-by: Rich Salz <rsalz@openssl.org>
This commit is contained in:
parent
f4da39d200
commit
0ff2b9ac0b
@ -1661,7 +1661,7 @@ int s_server_main(int argc, char *argv[])
|
|||||||
|
|
||||||
if (async) {
|
if (async) {
|
||||||
SSL_CTX_set_mode(ctx, SSL_MODE_ASYNC);
|
SSL_CTX_set_mode(ctx, SSL_MODE_ASYNC);
|
||||||
ASYNC_init_pool(0, 0, 0);
|
ASYNC_init_pool(0, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifndef OPENSSL_NO_SRTP
|
#ifndef OPENSSL_NO_SRTP
|
||||||
|
@ -65,6 +65,10 @@ __thread ASYNC_CTX *sysvctx;
|
|||||||
|
|
||||||
#define STACKSIZE 32768
|
#define STACKSIZE 32768
|
||||||
|
|
||||||
|
__thread size_t pool_max_size = 0;
|
||||||
|
__thread size_t pool_curr_size = 0;
|
||||||
|
__thread STACK_OF(ASYNC_JOB) *pool = NULL;
|
||||||
|
|
||||||
int ASYNC_FIBRE_init(ASYNC_FIBRE *fibre)
|
int ASYNC_FIBRE_init(ASYNC_FIBRE *fibre)
|
||||||
{
|
{
|
||||||
void *stack = NULL;
|
void *stack = NULL;
|
||||||
@ -111,4 +115,42 @@ int async_read1(int fd, void *buf)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
STACK_OF(ASYNC_JOB) *async_get_pool(void)
|
||||||
|
{
|
||||||
|
return pool;
|
||||||
|
}
|
||||||
|
|
||||||
|
void async_set_pool(STACK_OF(ASYNC_JOB) *poolin, size_t curr_size,
|
||||||
|
size_t max_size)
|
||||||
|
{
|
||||||
|
pool = poolin;
|
||||||
|
pool_curr_size = curr_size;
|
||||||
|
pool_max_size = max_size;
|
||||||
|
}
|
||||||
|
|
||||||
|
void async_increment_pool_size(void)
|
||||||
|
{
|
||||||
|
pool_curr_size++;
|
||||||
|
}
|
||||||
|
|
||||||
|
void async_release_job_to_pool(ASYNC_JOB *job)
|
||||||
|
{
|
||||||
|
sk_ASYNC_JOB_push(pool, job);
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t async_pool_max_size(void)
|
||||||
|
{
|
||||||
|
return pool_max_size;
|
||||||
|
}
|
||||||
|
|
||||||
|
void async_release_pool(void)
|
||||||
|
{
|
||||||
|
sk_ASYNC_JOB_free(pool);
|
||||||
|
}
|
||||||
|
|
||||||
|
int async_pool_can_grow(void)
|
||||||
|
{
|
||||||
|
return (pool_max_size == 0) || (pool_curr_size < pool_max_size);
|
||||||
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
@ -103,6 +103,5 @@ int async_pipe(int *pipefds);
|
|||||||
int async_write1(int fd, const void *buf);
|
int async_write1(int fd, const void *buf);
|
||||||
int async_read1(int fd, void *buf);
|
int async_read1(int fd, void *buf);
|
||||||
|
|
||||||
|
|
||||||
# endif
|
# endif
|
||||||
#endif
|
#endif
|
||||||
|
@ -51,7 +51,6 @@
|
|||||||
* ====================================================================
|
* ====================================================================
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <openssl/crypto.h>
|
|
||||||
#include <openssl/async.h>
|
#include <openssl/async.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include "async_locl.h"
|
#include "async_locl.h"
|
||||||
@ -61,13 +60,6 @@
|
|||||||
#define ASYNC_JOB_PAUSED 2
|
#define ASYNC_JOB_PAUSED 2
|
||||||
#define ASYNC_JOB_STOPPING 3
|
#define ASYNC_JOB_STOPPING 3
|
||||||
|
|
||||||
static size_t pool_max_size = 0;
|
|
||||||
static size_t curr_size = 0;
|
|
||||||
|
|
||||||
DECLARE_STACK_OF(ASYNC_JOB)
|
|
||||||
static STACK_OF(ASYNC_JOB) *pool = NULL;
|
|
||||||
|
|
||||||
|
|
||||||
static ASYNC_CTX *ASYNC_CTX_new(void)
|
static ASYNC_CTX *ASYNC_CTX_new(void)
|
||||||
{
|
{
|
||||||
ASYNC_CTX *nctx = NULL;
|
ASYNC_CTX *nctx = NULL;
|
||||||
@ -139,27 +131,29 @@ static void ASYNC_JOB_free(ASYNC_JOB *job)
|
|||||||
|
|
||||||
static ASYNC_JOB *async_get_pool_job(void) {
|
static ASYNC_JOB *async_get_pool_job(void) {
|
||||||
ASYNC_JOB *job;
|
ASYNC_JOB *job;
|
||||||
|
STACK_OF(ASYNC_JOB) *pool;
|
||||||
|
|
||||||
|
pool = async_get_pool();
|
||||||
if (pool == NULL) {
|
if (pool == NULL) {
|
||||||
/*
|
/*
|
||||||
* Pool has not been initialised, so init with the defaults, i.e.
|
* Pool has not been initialised, so init with the defaults, i.e.
|
||||||
* global pool, with no max size and no pre-created jobs
|
* global pool, with no max size and no pre-created jobs
|
||||||
*/
|
*/
|
||||||
if (ASYNC_init_pool(0, 0, 0) == 0)
|
if (ASYNC_init_pool(0, 0) == 0)
|
||||||
return NULL;
|
return NULL;
|
||||||
|
pool = async_get_pool();
|
||||||
}
|
}
|
||||||
|
|
||||||
job = sk_ASYNC_JOB_pop(pool);
|
job = sk_ASYNC_JOB_pop(pool);
|
||||||
if (job == NULL) {
|
if (job == NULL) {
|
||||||
/* Pool is empty */
|
/* Pool is empty */
|
||||||
if (pool_max_size && curr_size >= pool_max_size) {
|
if (!async_pool_can_grow())
|
||||||
/* Pool is at max size. We cannot continue */
|
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
|
||||||
job = ASYNC_JOB_new();
|
job = ASYNC_JOB_new();
|
||||||
if (job) {
|
if (job) {
|
||||||
ASYNC_FIBRE_makecontext(&job->fibrectx);
|
ASYNC_FIBRE_makecontext(&job->fibrectx);
|
||||||
curr_size++;
|
async_increment_pool_size();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return job;
|
return job;
|
||||||
@ -170,7 +164,7 @@ static void async_release_job(ASYNC_JOB *job) {
|
|||||||
OPENSSL_free(job->funcargs);
|
OPENSSL_free(job->funcargs);
|
||||||
job->funcargs = NULL;
|
job->funcargs = NULL;
|
||||||
/* Ignore error return */
|
/* Ignore error return */
|
||||||
sk_ASYNC_JOB_push(pool, job);
|
async_release_job_to_pool(job);
|
||||||
}
|
}
|
||||||
|
|
||||||
void ASYNC_start_func(void)
|
void ASYNC_start_func(void)
|
||||||
@ -301,14 +295,14 @@ int ASYNC_in_job(void)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int ASYNC_init_pool(unsigned int local, size_t max_size, size_t init_size)
|
int ASYNC_init_pool(size_t max_size, size_t init_size)
|
||||||
{
|
{
|
||||||
if (local != 0) {
|
STACK_OF(ASYNC_JOB) *pool;
|
||||||
/* We only support a global pool so far */
|
size_t curr_size = 0;
|
||||||
return 0;
|
|
||||||
}
|
if (init_size > max_size)
|
||||||
|
return 0;
|
||||||
|
|
||||||
pool_max_size = max_size;
|
|
||||||
pool = sk_ASYNC_JOB_new_null();
|
pool = sk_ASYNC_JOB_new_null();
|
||||||
if (pool == NULL) {
|
if (pool == NULL) {
|
||||||
return 0;
|
return 0;
|
||||||
@ -332,18 +326,24 @@ int ASYNC_init_pool(unsigned int local, size_t max_size, size_t init_size)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async_set_pool(pool, curr_size, max_size);
|
||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void ASYNC_free_pool(void)
|
void ASYNC_free_pool(void)
|
||||||
{
|
{
|
||||||
ASYNC_JOB *job;
|
ASYNC_JOB *job;
|
||||||
|
STACK_OF(ASYNC_JOB) *pool;
|
||||||
|
|
||||||
|
pool = async_get_pool();
|
||||||
|
if (pool == NULL)
|
||||||
|
return;
|
||||||
do {
|
do {
|
||||||
job = sk_ASYNC_JOB_pop(pool);
|
job = sk_ASYNC_JOB_pop(pool);
|
||||||
ASYNC_JOB_free(job);
|
ASYNC_JOB_free(job);
|
||||||
} while (job);
|
} while (job);
|
||||||
sk_ASYNC_JOB_free(pool);
|
async_release_pool();
|
||||||
}
|
}
|
||||||
|
|
||||||
ASYNC_JOB *ASYNC_get_current_job(void)
|
ASYNC_JOB *ASYNC_get_current_job(void)
|
||||||
|
@ -52,6 +52,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include <openssl/async.h>
|
#include <openssl/async.h>
|
||||||
|
#include <openssl/crypto.h>
|
||||||
|
|
||||||
typedef struct async_ctx_st ASYNC_CTX;
|
typedef struct async_ctx_st ASYNC_CTX;
|
||||||
|
|
||||||
@ -75,4 +76,14 @@ struct async_job_st {
|
|||||||
int wake_fd;
|
int wake_fd;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
DECLARE_STACK_OF(ASYNC_JOB)
|
||||||
|
|
||||||
void ASYNC_start_func(void);
|
void ASYNC_start_func(void);
|
||||||
|
STACK_OF(ASYNC_JOB) *async_get_pool(void);
|
||||||
|
void async_set_pool(STACK_OF(ASYNC_JOB) *poolin, size_t curr_size,
|
||||||
|
size_t max_size);
|
||||||
|
void async_increment_pool_size(void);
|
||||||
|
void async_release_job_to_pool(ASYNC_JOB *job);
|
||||||
|
size_t async_pool_max_size(void);
|
||||||
|
void async_release_pool(void);
|
||||||
|
int async_pool_can_grow(void);
|
||||||
|
@ -67,7 +67,7 @@ typedef struct async_job_st ASYNC_JOB;
|
|||||||
#define ASYNC_PAUSE 2
|
#define ASYNC_PAUSE 2
|
||||||
#define ASYNC_FINISH 3
|
#define ASYNC_FINISH 3
|
||||||
|
|
||||||
int ASYNC_init_pool(unsigned int local, size_t max_size, size_t init_size);
|
int ASYNC_init_pool(size_t max_size, size_t init_size);
|
||||||
void ASYNC_free_pool(void);
|
void ASYNC_free_pool(void);
|
||||||
|
|
||||||
int ASYNC_start_job(ASYNC_JOB **job, int *ret, int (*func)(void *),
|
int ASYNC_start_job(ASYNC_JOB **job, int *ret, int (*func)(void *),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user