ThriftClientConnection Class Reference

#include <ThriftClient.h>

Collaboration diagram for ThriftClientConnection:

Public Member Functions
	ThriftClientConnection (const std::string &server_host, const int port, const ThriftConnectionType conn_type, bool skip_host_verify, const std::string &ca_cert_name, const std::string &trust_cert_file)
	ThriftClientConnection (const std::string &server_host, const int port, const ThriftConnectionType conn_type, bool skip_host_verify, std::shared_ptr< TSSLSocketFactory > factory)
	ThriftClientConnection ()
std::shared_ptr< TTransport >	open_buffered_client_transport (const std::string &server_host, const int port, const std::string &ca_cert_name, const bool with_timeout=false, const bool with_keepalive=true, const unsigned connect_timeout=0, const unsigned recv_timeount=0, const unsigned send_timeout=0)
std::shared_ptr< TTransport >	open_http_client_transport (const std::string &server_host, const int port, const std::string &trust_cert_file_, bool use_https, bool skip_verify)
std::shared_ptr< TProtocol >	get_protocol ()
virtual	~ThriftClientConnection ()

Private Attributes
std::string	server_host_
int	port_
ThriftConnectionType	conn_type_
bool	skip_host_verify_
std::string	ca_cert_name_
std::string	trust_cert_file_
bool	using_X509_store_ = false
std::shared_ptr < TSSLSocketFactory >	factory_

Detailed Description

Definition at line 31 of file ThriftClient.h.

Constructor & Destructor Documentation

ThriftClientConnection::ThriftClientConnection ( const std::string &  server_host, const int  port, const ThriftConnectionType  conn_type, bool  skip_host_verify, const std::string &  ca_cert_name, const std::string &  trust_cert_file  )

inline

Definition at line 33 of file ThriftClient.h.

      : server_host_(server_host)
      , port_(port)
      , conn_type_(conn_type)
      , skip_host_verify_(skip_host_verify)
      , ca_cert_name_(ca_cert_name)
      , trust_cert_file_(trust_cert_file){};

ThriftClientConnection::ThriftClientConnection	(	const std::string &	server_host,
		const int	port,
		const ThriftConnectionType	conn_type,
		bool	skip_host_verify,
		std::shared_ptr< TSSLSocketFactory >	factory
	)

Definition at line 171 of file ThriftClient.cpp.

References BINARY_SSL, conn_type_, factory_, HTTPS, skip_host_verify_, and using_X509_store_.

    : server_host_(server_host)
    , port_(port)
    , conn_type_(conn_type)
    , skip_host_verify_(skip_host_verify)
    , trust_cert_file_("") {
  if (factory && (conn_type_ == ThriftConnectionType::BINARY_SSL ||
                  conn_type_ == ThriftConnectionType::HTTPS)) {
    using_X509_store_ = true;
    factory_ = factory;
    factory_->ciphers("ALL:!ADH:!LOW:!EXP:!MD5:@STRENGTH");
    if (skip_host_verify_) {
      factory_->access(
          std::shared_ptr<InsecureAccessManager>(new InsecureAccessManager()));
    }
  }
}

ThriftClientConnection::ThriftClientConnection ( )

inline

Definition at line 52 of file ThriftClient.h.

{};

ThriftClientConnection::~ThriftClientConnection ( )

virtual

Definition at line 170 of file ThriftClient.cpp.

{}

Member Function Documentation

std::shared_ptr< TProtocol > ThriftClientConnection::get_protocol

(

)

Definition at line 193 of file ThriftClient.cpp.

References ca_cert_name_, conn_type_, HTTP, HTTPS, open_buffered_client_transport(), open_http_client_transport(), port_, server_host_, skip_host_verify_, and to_string().

Referenced by RowToColumnLoader::createConnection().

                                                              {
  std::shared_ptr<apache::thrift::transport::TTransport> mytransport;
  if (conn_type_ == ThriftConnectionType::HTTP ||
      conn_type_ == ThriftConnectionType::HTTPS) {
    mytransport = open_http_client_transport(server_host_,
                                             port_,
                                             ca_cert_name_,
                                             conn_type_ == ThriftConnectionType::HTTPS,
                                             skip_host_verify_);

  } else {
    mytransport = open_buffered_client_transport(server_host_, port_, ca_cert_name_);
  }

  try {
    mytransport->open();
  } catch (const apache::thrift::TException& e) {
    throw apache::thrift::TException(std::string(e.what()) + ": host " + server_host_ +
                                     ", port " + std::to_string(port_));
  }
  if (conn_type_ == ThriftConnectionType::HTTP ||
      conn_type_ == ThriftConnectionType::HTTPS) {
    return std::shared_ptr<TProtocol>(new TJSONProtocol(mytransport));
  } else {
    return std::shared_ptr<TProtocol>(new TBinaryProtocol(mytransport));
  }
}

Here is the call graph for this function:

Here is the caller graph for this function:

std::shared_ptr< TTransport > ThriftClientConnection::open_buffered_client_transport	(	const std::string &	server_host,
		const int	port,
		const std::string &	ca_cert_name,
		const bool	with_timeout = false,
		const bool	with_keepalive = true,
		const unsigned	connect_timeout = 0,
		const unsigned	recv_timeount = 0,
		const unsigned	send_timeout = 0
	)

Definition at line 221 of file ThriftClient.cpp.

References shared::default_tconfig(), factory_, and using_X509_store_.

Referenced by get_protocol().

                           {
  std::shared_ptr<TTransport> transport;

  if (!factory_ && !ca_cert_name.empty()) {
    // need to build a factory once for ssl conection
    factory_ =
        std::shared_ptr<TSSLSocketFactory>(new TSSLSocketFactory(SSLProtocol::SSLTLS));
    factory_->ciphers("ALL:!ADH:!LOW:!EXP:!MD5:@STRENGTH");
    factory_->loadTrustedCertificates(ca_cert_name.c_str());
    factory_->authenticate(false);
    factory_->access(std::shared_ptr<InsecureAccessManager>(new InsecureAccessManager()));
  }
  if (!using_X509_store_ && ca_cert_name.empty()) {
#ifdef HAVE_THRIFT_MESSAGE_LIMIT
    const auto socket =
        std::make_shared<TSocket>(server_host, port, shared::default_tconfig());
#else
    const auto socket = std::make_shared<TSocket>(server_host, port);
#endif
    if (with_timeout) {
      socket->setKeepAlive(with_keepalive);
      socket->setConnTimeout(connect_timeout);
      socket->setRecvTimeout(recv_timeout);
      socket->setSendTimeout(send_timeout);
#ifdef __APPLE__
      socket->setLinger(false, 0);
#endif
    }
#ifdef HAVE_THRIFT_MESSAGE_LIMIT
    transport = std::make_shared<TBufferedTransport>(socket, shared::default_tconfig());
#else
    transport = std::make_shared<TBufferedTransport>(socket);
#endif
  } else {
    std::shared_ptr<TSocket> secure_socket = factory_->createSocket(server_host, port);
    if (with_timeout) {
      secure_socket->setKeepAlive(with_keepalive);
      secure_socket->setConnTimeout(connect_timeout);
      secure_socket->setRecvTimeout(recv_timeout);
      secure_socket->setSendTimeout(send_timeout);
#ifdef __APPLE__
      secure_socket->setLinger(false, 0);
#endif
    }
#ifdef HAVE_THRIFT_MESSAGE_LIMIT
    transport = std::shared_ptr<TTransport>(
        new TBufferedTransport(secure_socket, shared::default_tconfig()));
#else
    transport = std::shared_ptr<TTransport>(new TBufferedTransport(secure_socket));
#endif
  }

  return transport;
}

Here is the call graph for this function:

Here is the caller graph for this function:

std::shared_ptr< TTransport > ThriftClientConnection::open_http_client_transport	(	const std::string &	server_host,
		const int	port,
		const std::string &	trust_cert_file_,
		bool	use_https,
		bool	skip_verify
	)

Definition at line 284 of file ThriftClient.cpp.

References check_standard_ca(), factory_, trust_cert_file_, and using_X509_store_.

Referenced by get_protocol().

                      {
  trust_cert_file_ = trust_cert_fileX;
  check_standard_ca(trust_cert_file_);

  if (!factory_) {
    factory_ =
        std::shared_ptr<TSSLSocketFactory>(new TSSLSocketFactory(SSLProtocol::SSLTLS));
  }
  std::shared_ptr<TTransport> transport;
  std::shared_ptr<TTransport> socket;
  if (use_https) {
    if (skip_verify) {
      factory_->authenticate(false);
      factory_->access(
          std::shared_ptr<InsecureAccessManager>(new InsecureAccessManager()));
    }
    if (!using_X509_store_) {
      factory_->loadTrustedCertificates(trust_cert_file_.c_str());
    }
    socket = factory_->createSocket(server_host, port);
    // transport = std::shared_ptr<TTransport>(new THttpClient(socket,
    // server_host,
    // "/"));
    transport =
        std::shared_ptr<TTransport>(new ProxyTHttpClient(socket, server_host, "/"));
  } else {
    transport = std::shared_ptr<TTransport>(new ProxyTHttpClient(server_host, port, "/"));
  }
  return transport;
}

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

std::string ThriftClientConnection::ca_cert_name_

private

Definition at line 79 of file ThriftClient.h.

Referenced by get_protocol().

ThriftConnectionType ThriftClientConnection::conn_type_

private

Definition at line 77 of file ThriftClient.h.

Referenced by get_protocol(), and ThriftClientConnection().

std::shared_ptr<TSSLSocketFactory> ThriftClientConnection::factory_

private

Definition at line 82 of file ThriftClient.h.

Referenced by open_buffered_client_transport(), open_http_client_transport(), and ThriftClientConnection().

int ThriftClientConnection::port_

private

Definition at line 76 of file ThriftClient.h.

Referenced by get_protocol().

std::string ThriftClientConnection::server_host_

private

Definition at line 75 of file ThriftClient.h.

Referenced by get_protocol().

bool ThriftClientConnection::skip_host_verify_

private

Definition at line 78 of file ThriftClient.h.

Referenced by get_protocol(), and ThriftClientConnection().

std::string ThriftClientConnection::trust_cert_file_

private

Definition at line 80 of file ThriftClient.h.

Referenced by open_http_client_transport().

bool ThriftClientConnection::using_X509_store_ = false

private

Definition at line 81 of file ThriftClient.h.

Referenced by open_buffered_client_transport(), open_http_client_transport(), and ThriftClientConnection().

---

The documentation for this class was generated from the following files:

• /home/jenkins-slave/workspace/core-os-doxygen/Shared/ThriftClient.h
• /home/jenkins-slave/workspace/core-os-doxygen/Shared/ThriftClient.cpp