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:
Matt Caswell 2015-09-16 17:01:58 +01:00
parent f4da39d200
commit 0ff2b9ac0b
6 changed files with 76 additions and 24 deletions

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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)

View File

@ -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);

View File

@ -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 *),