mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-06 18:35:19 +00:00
Add unit test for OpenSSL manual initialization
This test checks whether Thrift leaves OpenSSL functionality available after the last TSSLSocketFactory is destroyed when manual initialization is set. It uses the EVP_get_digestbyname function as an example function that requires OpenSSL initialization to work properly. Signed-off-by: Alan Dunn <amdunn@gmail.com> Signed-off-by: Roger Meier <roger@apache.org>
This commit is contained in:
parent
c0ff5561fe
commit
bee7b73802
1
.gitignore
vendored
1
.gitignore
vendored
@ -89,6 +89,7 @@ test-driver
|
||||
/lib/cpp/test/TransportTest
|
||||
/lib/cpp/test/UnitTests
|
||||
/lib/cpp/test/ZlibTest
|
||||
/lib/cpp/test/OpenSSLManualInitTest
|
||||
/lib/cpp/test/concurrency_test
|
||||
/lib/cpp/test/link_test
|
||||
/lib/cpp/test/processor_test
|
||||
|
@ -64,7 +64,8 @@ check_PROGRAMS = \
|
||||
ZlibTest \
|
||||
TFileTransportTest \
|
||||
UnitTests \
|
||||
link_test
|
||||
link_test \
|
||||
OpenSSLManualInitTest
|
||||
# disable these test ... too strong
|
||||
# processor_test
|
||||
# concurrency_test
|
||||
@ -215,6 +216,14 @@ processor_test_LDADD = libprocessortest.la \
|
||||
$(BOOST_LDFLAGS) \
|
||||
-levent \
|
||||
-l:libboost_unit_test_framework.a
|
||||
|
||||
OpenSSLManualInitTest_SOURCES = \
|
||||
OpenSSLManualInitTest.cpp
|
||||
|
||||
OpenSSLManualInitTest_LDADD = \
|
||||
$(top_builddir)/lib/cpp/libthrift.la \
|
||||
-l:libboost_unit_test_framework.a
|
||||
|
||||
#
|
||||
# Common thrift code generation rules
|
||||
#
|
||||
|
77
lib/cpp/test/OpenSSLManualInitTest.cpp
Normal file
77
lib/cpp/test/OpenSSLManualInitTest.cpp
Normal file
@ -0,0 +1,77 @@
|
||||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing,
|
||||
* software distributed under the License is distributed on an
|
||||
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
* KIND, either express or implied. See the License for the
|
||||
* specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*/
|
||||
// To show that this test actually tests something, you can change
|
||||
// MANUAL_OPENSSL_INIT to 0 to cause automatic OpenSSL init/cleanup,
|
||||
// which will cause the test to fail
|
||||
#define MANUAL_OPENSSL_INIT 1
|
||||
|
||||
#include <boost/test/unit_test.hpp>
|
||||
|
||||
#include <openssl/evp.h>
|
||||
|
||||
#include <thrift/transport/TSSLSocket.h>
|
||||
|
||||
using namespace std;
|
||||
using namespace apache::thrift::transport;
|
||||
|
||||
void make_isolated_sslsocketfactory() {
|
||||
// Here we create an isolated TSSLSocketFactory to ensure the
|
||||
// constructor and destructor of TSSLSocketFactory get run. Thus
|
||||
// without manual initialization normally OpenSSL would be
|
||||
// uninitialized after this function.
|
||||
TSSLSocketFactory factory;
|
||||
}
|
||||
|
||||
void openssl_init() {
|
||||
#if MANUAL_OPENSSL_INIT
|
||||
TSSLSocketFactory::setManualOpenSSLInitialization(true);
|
||||
initializeOpenSSL();
|
||||
#endif
|
||||
}
|
||||
|
||||
void openssl_cleanup() {
|
||||
#if MANUAL_OPENSSL_INIT
|
||||
cleanupOpenSSL();
|
||||
#endif
|
||||
}
|
||||
|
||||
void test_openssl_availability() {
|
||||
// Check whether Thrift leaves OpenSSL functionality available after
|
||||
// the last TSSLSocketFactory is destroyed when manual
|
||||
// initialization is set
|
||||
openssl_init();
|
||||
make_isolated_sslsocketfactory();
|
||||
|
||||
// The following function is one that will fail if OpenSSL is
|
||||
// uninitialized. It might also fail on very old versions of
|
||||
// OpenSSL...
|
||||
const EVP_MD* md = EVP_get_digestbyname("SHA256");
|
||||
BOOST_CHECK(md != NULL);
|
||||
openssl_cleanup();
|
||||
}
|
||||
|
||||
boost::unit_test::test_suite* init_unit_test_suite(int argc, char* argv[]) {
|
||||
boost::unit_test::test_suite* suite =
|
||||
&boost::unit_test::framework::master_test_suite();
|
||||
suite->p_name.value = "OpenSSLManualInit";
|
||||
|
||||
suite->add(BOOST_TEST_CASE(test_openssl_availability));
|
||||
|
||||
return NULL;
|
||||
}
|
Loading…
Reference in New Issue
Block a user