87 lines
2.7 KiB
C++
87 lines
2.7 KiB
C++
|
/* Copyright (c) 2018, Google Inc.
|
||
|
|
*
|
||
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
||
|
|
* purpose with or without fee is hereby granted, provided that the above
|
||
|
|
* copyright notice and this permission notice appear in all copies.
|
||
|
|
*
|
||
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
||
|
|
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
|
||
|
|
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
|
||
|
|
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
|
||
|
|
|
||
|
|
#include "test_state.h"
|
||
|
|
|
||
|
|
#include <openssl/ssl.h>
|
||
|
|
|
||
|
|
#include "../../crypto/internal.h"
|
||
|
|
#include "../internal.h"
|
||
|
|
|
||
|
|
static CRYPTO_once_t g_once = CRYPTO_ONCE_INIT;
|
||
|
|
static int g_state_index = 0;
|
||
|
|
// Some code treats the zero time special, so initialize the clock to a
|
||
|
|
// non-zero time.
|
||
|
|
static timeval g_clock = { 1234, 1234 };
|
||
|
|
|
||
|
|
static void TestStateExFree(void *parent, void *ptr, CRYPTO_EX_DATA *ad,
|
||
|
|
int index, long argl, void *argp) {
|
||
|
|
delete ((TestState *)ptr);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void init_once() {
|
||
|
|
g_state_index = SSL_get_ex_new_index(0, NULL, NULL, NULL, TestStateExFree);
|
||
|
|
if (g_state_index < 0) {
|
||
|
|
abort();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
struct timeval *GetClock() {
|
||
|
|
CRYPTO_once(&g_once, init_once);
|
||
|
|
return &g_clock;
|
||
|
|
}
|
||
|
|
|
||
|
|
void AdvanceClock(unsigned seconds) {
|
||
|
|
CRYPTO_once(&g_once, init_once);
|
||
|
|
g_clock.tv_sec += seconds;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool SetTestState(SSL *ssl, std::unique_ptr<TestState> state) {
|
||
|
|
CRYPTO_once(&g_once, init_once);
|
||
|
|
// |SSL_set_ex_data| takes ownership of |state| only on success.
|
||
|
|
if (SSL_set_ex_data(ssl, g_state_index, state.get()) == 1) {
|
||
|
|
state.release();
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
TestState *GetTestState(const SSL *ssl) {
|
||
|
|
CRYPTO_once(&g_once, init_once);
|
||
|
|
return (TestState *)SSL_get_ex_data(ssl, g_state_index);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool MoveTestState(SSL *dest, SSL *src) {
|
||
|
|
TestState *state = GetTestState(src);
|
||
|
|
if (!SSL_set_ex_data(src, g_state_index, nullptr) ||
|
||
|
|
!SSL_set_ex_data(dest, g_state_index, state)) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ssl_ctx_add_session(SSL_SESSION *session, void *void_param) {
|
||
|
|
SSL_CTX *ctx = reinterpret_cast<SSL_CTX *>(void_param);
|
||
|
|
bssl::UniquePtr<SSL_SESSION> new_session = bssl::SSL_SESSION_dup(
|
||
|
|
session, SSL_SESSION_INCLUDE_NONAUTH | SSL_SESSION_INCLUDE_TICKET);
|
||
|
|
if (new_session != nullptr) {
|
||
|
|
SSL_CTX_add_session(ctx, new_session.get());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void CopySessions(SSL_CTX *dst, const SSL_CTX *src) {
|
||
|
|
lh_SSL_SESSION_doall_arg(src->sessions, ssl_ctx_add_session, dst);
|
||
|
|
}
|