thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-19 16:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optional secured communication between ClickHouse and Zookeeper

Browse Source
This commit is contained in:
Konstantin Lebedev 2020-04-06 22:27:57 +03:00
parent f48fdda678
commit b666f60af8
10 changed files with 97 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
programs/server/config.xml
View File

@ -37,7 +37,7 @@
<preferServerCiphers>true</preferServerCiphers>
</server>
<client> <!-- Used for connecting to https dictionary source -->
<client> <!-- Used for connecting to https dictionary source and secured Zookeeper communication -->
<loadDefaultCAFile>true</loadDefaultCAFile>
<cacheSessions>true</cacheSessions>
<disableProtocols>sslv2,sslv3</disableProtocols>

4
src/Common/ZooKeeper/CMakeLists.txt
View File

@ -7,6 +7,10 @@ add_library(clickhouse_common_zookeeper ${clickhouse_common_zookeeper_headers} $
target_link_libraries (clickhouse_common_zookeeper PUBLIC clickhouse_common_io common PRIVATE string_utils PUBLIC ${Poco_Util_LIBRARY})
target_include_directories(clickhouse_common_zookeeper PUBLIC ${DBMS_INCLUDE_DIR})
if (USE_POCO_NETSSL)
target_link_libraries (clickhouse_common_zookeeper PRIVATE ${Poco_NetSSL_LIBRARY} ${Poco_Crypto_LIBRARY})
endif()
if (ENABLE_TESTS)
add_subdirectory (tests)
endif ()

29
src/Common/ZooKeeper/ZooKeeper.cpp
View File

@ -59,30 +59,36 @@ void ZooKeeper::init(const std::string & implementation_, const std::string & ho
if (implementation == "zookeeper")
{
if (hosts.empty())
throw KeeperException("No addresses passed to ZooKeeper constructor.", Coordination::ZBADARGUMENTS);
throw KeeperException("No hosts passed to ZooKeeper constructor.", Coordination::ZBADARGUMENTS);
std::vector<std::string> addresses_strings;
splitInto<','>(addresses_strings, hosts);
Coordination::ZooKeeper::Addresses addresses;
addresses.reserve(addresses_strings.size());
std::vector<std::string> hosts_strings;
splitInto<','>(hosts_strings, hosts);
Coordination::ZooKeeper::Nodes nodes;
nodes.reserve(hosts_strings.size());
for (const auto & address_string : addresses_strings)
for (auto & host_string : hosts_strings)
{
try
{
addresses.emplace_back(address_string);
bool secure = bool(startsWith(host_string, "secure://"));
if (secure) {
host_string.erase(0, strlen("secure://"));
}
nodes.emplace_back(Coordination::ZooKeeper::Node{Poco::Net::SocketAddress{host_string}, secure});
}
catch (const Poco::Net::DNSException & e)
{
LOG_ERROR(log, "Cannot use ZooKeeper address " << address_string << ", reason: " << e.displayText());
LOG_ERROR(log, "Cannot use ZooKeeper host " << host_string << ", reason: " << e.displayText());
}
}
if (addresses.empty())
throw KeeperException("Cannot use any of provided ZooKeeper addresses", Coordination::ZBADARGUMENTS);
if (nodes.empty())
throw KeeperException("Cannot use any of provided ZooKeeper nodes", Coordination::ZBADARGUMENTS);
impl = std::make_unique<Coordination::ZooKeeper>(
addresses,
nodes,
chroot,
identity_.empty() ? "" : "digest",
identity_,
@ -130,6 +136,7 @@ struct ZooKeeperArgs
if (startsWith(key, "node"))
{
hosts_strings.push_back(
(config.getBool(config_name + "." + key + ".secure", false) ? "secure://" : "") +
config.getString(config_name + "." + key + ".host") + ":"
+ config.getString(config_name + "." + key + ".port", "2181")
);

4
src/Common/ZooKeeper/ZooKeeper.h
View File

@ -63,10 +63,14 @@ public:
<node>
<host>example1</host>
<port>2181</port>
<!-- Optional. Enables communication over SSL . -->
<secure>1</secure>
</node>
<node>
<host>example2</host>
<port>2181</port>
<!-- Optional. Enables communication over SSL . -->
<secure>1</secure>
</node>
<session_timeout_ms>30000</session_timeout_ms>
<operation_timeout_ms>10000</operation_timeout_ms>

53
src/Common/ZooKeeper/ZooKeeperImpl.cpp
View File

@ -11,6 +11,11 @@
#include <Poco/Exception.h>
#include <Poco/Net/NetException.h>
#include <Common/config.h>
#if USE_POCO_NETSSL
#include <Poco/Net/SecureStreamSocket.h>
#endif
#include <array>
@ -44,6 +49,13 @@ namespace CurrentMetrics
extern const Metric ZooKeeperWatch;
}
namespace DB
{
namespace ErrorCodes
{
extern const int SUPPORT_IS_DISABLED;
}
}
/** ZooKeeper wire protocol.
@ -817,7 +829,7 @@ ZooKeeper::~ZooKeeper()
ZooKeeper::ZooKeeper(
const Addresses & addresses,
const Nodes & nodes,
const String & root_path_,
const String & auth_scheme,
const String & auth_data,
@ -851,7 +863,7 @@ ZooKeeper::ZooKeeper(
default_acls.emplace_back(std::move(acl));
}
connect(addresses, connection_timeout);
connect(nodes, connection_timeout);
if (!auth_scheme.empty())
sendAuth(auth_scheme, auth_data);
@ -864,11 +876,11 @@ ZooKeeper::ZooKeeper(
void ZooKeeper::connect(
const Addresses & addresses,
const Nodes & nodes,
Poco::Timespan connection_timeout)
{
if (addresses.empty())
throw Exception("No addresses passed to ZooKeeper constructor", ZBADARGUMENTS);
if (nodes.empty())
throw Exception("No nodes passed to ZooKeeper constructor", ZBADARGUMENTS);
static constexpr size_t num_tries = 3;
bool connected = false;
@ -876,12 +888,25 @@ void ZooKeeper::connect(
WriteBufferFromOwnString fail_reasons;
for (size_t try_no = 0; try_no < num_tries; ++try_no)
{
for (const auto & address : addresses)
for (const auto & node : nodes)
{
try
{
socket = Poco::Net::StreamSocket(); /// Reset the state of previous attempt.
socket.connect(address, connection_timeout);
/// Reset the state of previous attempt.
if (node.secure)
{
#if USE_POCO_NETSSL
socket = Poco::Net::SecureStreamSocket();
#else
throw Exception{"Communication with ZooKeeper over SSL is disabled because poco library was built without NetSSL support.", ErrorCodes::SUPPORT_IS_DISABLED};
#endif
}
else
{
socket = Poco::Net::StreamSocket();
}
socket.connect(node.address, connection_timeout);
socket.setReceiveTimeout(operation_timeout);
socket.setSendTimeout(operation_timeout);
@ -915,7 +940,7 @@ void ZooKeeper::connect(
}
catch (...)
{
fail_reasons << "\n" << getCurrentExceptionMessage(false) << ", " << address.toString();
fail_reasons << "\n" << getCurrentExceptionMessage(false) << ", " << node.address.toString();
}
}
@ -926,15 +951,19 @@ void ZooKeeper::connect(
if (!connected)
{
WriteBufferFromOwnString message;
message << "All connection tries failed while connecting to ZooKeeper. Addresses: ";
message << "All connection tries failed while connecting to ZooKeeper. nodes: ";
bool first = true;
for (const auto & address : addresses)
for (const auto & node : nodes)
{
if (first)
first = false;
else
message << ", ";
message << address.toString();
if (node.secure)
message << "secure://";
message << node.address.toString();
}
message << fail_reasons.str() << "\n";

13
src/Common/ZooKeeper/ZooKeeperImpl.h
View File

@ -93,17 +93,22 @@ struct ZooKeeperRequest;
class ZooKeeper : public IKeeper
{
public:
using Addresses = std::vector<Poco::Net::SocketAddress>;
struct Node {
Poco::Net::SocketAddress address;
bool secure;
};
using Nodes = std::vector<Node>;
using XID = int32_t;
using OpNum = int32_t;
/** Connection to addresses is performed in order. If you want, shuffle them manually.
/** Connection to nodes is performed in order. If you want, shuffle them manually.
* Operation timeout couldn't be greater than session timeout.
* Operation timeout applies independently for network read, network write, waiting for events and synchronization.
*/
ZooKeeper(
const Addresses & addresses,
const Nodes & nodes,
const String & root_path,
const String & auth_scheme,
const String & auth_data,
@ -213,7 +218,7 @@ private:
ThreadFromGlobalPool receive_thread;
void connect(
const Addresses & addresses,
const Nodes & node,
Poco::Timespan connection_timeout);
void sendHandshake();

25
src/Common/ZooKeeper/tests/zkutil_test_commands_new_lib.cpp
View File

@ -1,6 +1,7 @@
#include <Poco/ConsoleChannel.h>
#include <Poco/Logger.h>
#include <Poco/Event.h>
#include <Common/StringUtils/StringUtils.h>
#include <Common/ZooKeeper/ZooKeeperImpl.h>
#include <Common/typeid_cast.h>
#include <iostream>
@ -23,15 +24,23 @@ try
Poco::Logger::root().setChannel(channel);
Poco::Logger::root().setLevel("trace");
std::string addresses_arg = argv[1];
std::vector<std::string> addresses_strings;
splitInto<','>(addresses_strings, addresses_arg);
ZooKeeper::Addresses addresses;
addresses.reserve(addresses_strings.size());
for (const auto & address_string : addresses_strings)
addresses.emplace_back(address_string);
std::string hosts_arg = argv[1];
std::vector<std::string> hosts_strings;
splitInto<','>(hosts_strings, hosts_arg);
ZooKeeper::Nodes nodes;
nodes.reserve(hosts_strings.size());
for (auto & host_string : hosts_strings) {
bool secure = bool(startsWith(host_string, "secure://"));
ZooKeeper zk(addresses, {}, {}, {}, {5, 0}, {0, 50000}, {0, 50000});
if (secure) {
host_string.erase(0, strlen("secure://"));
}
nodes.emplace_back(ZooKeeper::Node{Poco::Net::SocketAddress{host_string},secure});
}
ZooKeeper zk(nodes, {}, {}, {}, {5, 0}, {0, 50000}, {0, 50000});
Poco::Event event(true);

2
src/Common/ZooKeeper/tests/zookeeper_impl.cpp
View File

@ -5,7 +5,7 @@
int main()
try
{
Coordination::ZooKeeper zookeeper({Poco::Net::SocketAddress{"localhost:2181"}}, "", "", "", {30, 0}, {0, 50000}, {0, 50000});
Coordination::ZooKeeper zookeeper({Coordination::ZooKeeper::Node{Poco::Net::SocketAddress{"localhost:2181"}, false}}, "", "", "", {30, 0}, {0, 50000}, {0, 50000});
zookeeper.create("/test", "hello", false, false, {}, [](const Coordination::CreateResponse & response)
{

2
tests/integration/test_config_corresponding_root/configs/config.xml
View File

@ -37,7 +37,7 @@
<preferServerCiphers>true</preferServerCiphers>
</server>
<client> <!-- Used for connecting to https dictionary source -->
<client> <!-- Used for connecting to https dictionary source and secured Zookeeper communication -->
<loadDefaultCAFile>true</loadDefaultCAFile>
<cacheSessions>true</cacheSessions>
<disableProtocols>sslv2,sslv3</disableProtocols>

2
tests/server-test.xml
View File

@ -31,7 +31,7 @@
<preferServerCiphers>true</preferServerCiphers>
</server>
<client> <!-- Used for connecting to https dictionary source -->
<client> <!-- Used for connecting to https dictionary source and secured Zookeeper communication -->
<loadDefaultCAFile>true</loadDefaultCAFile>
<cacheSessions>true</cacheSessions>
<disableProtocols>sslv2,sslv3</disableProtocols>