@@ -40,23 +40,24 @@ static int Usage(const char *program) {
return 1;
}
struct Async State {
AsyncState() : cert_ready (false) {}
struct Test State {
TestState() : cert_ready(false), early_callback_called (false) {}
ScopedEVP_PKEY channel_id;
bool cert_ready;
ScopedSSL_SESSION session;
ScopedSSL_SESSION pending_session;
bool early_callback_called;
};
static void Async ExFree(void *parent, void *ptr, CRYPTO_EX_DATA *ad, int index,
long argl, void *argp) {
delete ((Async State *)ptr);
static void TestState ExFree(void *parent, void *ptr, CRYPTO_EX_DATA *ad,
int index, long argl, void *argp) {
delete ((Test State *)ptr);
}
static int g_config_index = 0;
static int g_clock_index = 0;
static int g_async _index = 0;
static int g_state _index = 0;
static bool SetConfigPtr(SSL *ssl, const TestConfig *config) {
return SSL_set_ex_data(ssl, g_config_index, (void *)config) == 1;
@@ -74,16 +75,16 @@ static OPENSSL_timeval *GetClockPtr(SSL *ssl) {
return (OPENSSL_timeval *)SSL_get_ex_data(ssl, g_clock_index);
}
static bool SetAsyncState(SSL *ssl, std::unique_ptr<Async State> async) {
if (SSL_set_ex_data(ssl, g_async _index, (void *)async.get()) == 1) {
static bool SetTestState(SSL *ssl, std::unique_ptr<Test State> async) {
if (SSL_set_ex_data(ssl, g_state _index, (void *)async.get()) == 1) {
async.release();
return true;
}
return false;
}
static AsyncState *GetAsync State(SSL *ssl) {
return (AsyncState *)SSL_get_ex_data(ssl, g_async _index);
static TestState *GetTest State(SSL *ssl) {
return (TestState *)SSL_get_ex_data(ssl, g_state _index);
}
static ScopedEVP_PKEY LoadPrivateKey(const std::string &file) {
@@ -110,12 +111,9 @@ static bool InstallCertificate(SSL *ssl) {
return true;
}
static int g_early_callback_called = 0;
static int SelectCertificateCallback(const struct ssl_early_callback_ctx *ctx) {
g_early_callback_called = 1;
const TestConfig *config = GetConfigPtr(ctx->ssl);
GetTestState(ctx->ssl)->early_callback_called = true;
if (config->expected_server_name.empty()) {
return 1;
@@ -274,11 +272,11 @@ static void CurrentTimeCallback(SSL *ssl, OPENSSL_timeval *out_clock) {
}
static void ChannelIdCallback(SSL *ssl, EVP_PKEY **out_pkey) {
*out_pkey = GetAsync State(ssl)->channel_id.release();
*out_pkey = GetTest State(ssl)->channel_id.release();
}
static int CertCallback(SSL *ssl, void *arg) {
if (!GetAsync State(ssl)->cert_ready) {
if (!GetTest State(ssl)->cert_ready) {
return -1;
}
if (!InstallCertificate(ssl)) {
@@ -289,7 +287,7 @@ static int CertCallback(SSL *ssl, void *arg) {
static SSL_SESSION *GetSessionCallback(SSL *ssl, uint8_t *data, int len,
int *copy) {
AsyncState *async_state = GetAsync State(ssl);
TestState *async_state = GetTest State(ssl);
if (async_state->session) {
*copy = 0;
return async_state->session.release();
@@ -395,15 +393,15 @@ static int RetryAsync(SSL *ssl, int ret, BIO *async,
AsyncBioAllowWrite(async, 1);
return 1;
case SSL_ERROR_WANT_CHANNEL_ID_LOOKUP:
GetAsync State(ssl)->channel_id =
GetTest State(ssl)->channel_id =
LoadPrivateKey(GetConfigPtr(ssl)->send_channel_id);
return 1;
case SSL_ERROR_WANT_X509_LOOKUP:
GetAsync State(ssl)->cert_ready = true;
GetTest State(ssl)->cert_ready = true;
return 1;
case SSL_ERROR_PENDING_SESSION:
GetAsync State(ssl)->session =
std::move(GetAsync State(ssl)->pending_session);
GetTest State(ssl)->session =
std::move(GetTest State(ssl)->pending_session);
return 1;
default:
return 0;
@@ -413,8 +411,6 @@ static int RetryAsync(SSL *ssl, int ret, BIO *async,
static int DoExchange(ScopedSSL_SESSION *out_session, SSL_CTX *ssl_ctx,
const TestConfig *config, bool is_resume,
int fd, SSL_SESSION *session) {
g_early_callback_called = 0;
OPENSSL_timeval clock = {0}, clock_delta = {0};
ScopedSSL ssl(SSL_new(ssl_ctx));
if (!ssl) {
@@ -424,7 +420,7 @@ static int DoExchange(ScopedSSL_SESSION *out_session, SSL_CTX *ssl_ctx,
if (!SetConfigPtr(ssl.get(), config) ||
!SetClockPtr(ssl.get(), &clock) |
!SetAsyncState(ssl.get(), std::unique_ptr<AsyncState>(new Async State))) {
!SetTestState(ssl.get(), std::unique_ptr<TestState>(new Test State))) {
BIO_print_errors_fp(stdout);
return 1;
}
@@ -564,7 +560,7 @@ static int DoExchange(ScopedSSL_SESSION *out_session, SSL_CTX *ssl_ctx,
} else if (config->async) {
// The internal session cache is disabled, so install the session
// manually.
GetAsync State(ssl.get())->pending_session.reset(
GetTest State(ssl.get())->pending_session.reset(
SSL_SESSION_up_ref(session));
}
}
@@ -605,7 +601,7 @@ static int DoExchange(ScopedSSL_SESSION *out_session, SSL_CTX *ssl_ctx,
return 2;
}
if (!g_ early_callback_called) {
if (!GetTestState(ssl.get())-> early_callback_called) {
fprintf(stderr, "early callback not called\n");
return 2;
}
@@ -823,8 +819,8 @@ int main(int argc, char **argv) {
}
g_config_index = SSL_get_ex_new_index(0, NULL, NULL, NULL, NULL);
g_clock_index = SSL_get_ex_new_index(0, NULL, NULL, NULL, NULL);
g_async_index = SSL_get_ex_new_index(0, NULL, NULL, NULL, Async ExFree);
if (g_config_index < 0 || g_clock_index < 0 || g_async _index < 0) {
g_state_index = SSL_get_ex_new_index(0, NULL, NULL, NULL, TestState ExFree);
if (g_config_index < 0 || g_clock_index < 0 || g_state _index < 0) {
return 1;
}