diff options
author | Dan Klein <danklei@cisco.com> | 2015-08-24 13:22:48 +0300 |
---|---|---|
committer | Dan Klein <danklei@cisco.com> | 2015-08-24 13:22:48 +0300 |
commit | dab741a80699f86e86c91718872a052cca9bbb25 (patch) | |
tree | 1959c4a2cea440170a5113dcb067796cb20ffb64 /external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp | |
parent | d3f26ece7d4383df0b22fe9c3cb3e695381ec737 (diff) |
Fixed dependencies of Control Plane to use external_lib sources
Diffstat (limited to 'external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp')
-rw-r--r-- | external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp | 478 |
1 files changed, 478 insertions, 0 deletions
diff --git a/external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp b/external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp new file mode 100644 index 00000000..a3b70436 --- /dev/null +++ b/external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp @@ -0,0 +1,478 @@ +/* + Copyright (c) 2007-2015 Contributors as noted in the AUTHORS file + + This file is part of libzmq, the ZeroMQ core engine in C++. + + libzmq is free software; you can redistribute it and/or modify it under + the terms of the GNU Lesser General Public License (LGPL) as published + by the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + As a special exception, the Contributors give you permission to link + this library with independent modules to produce an executable, + regardless of the license terms of these independent modules, and to + copy and distribute the resulting executable under terms of your choice, + provided that you also meet, for each linked independent module, the + terms and conditions of the license of that module. An independent + module is a module which is not derived from or based on this library. + If you modify this library, you must extend this exception to your + version of the library. + + libzmq is distributed in the hope that it will be useful, but WITHOUT + ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public + License for more details. + + You should have received a copy of the GNU Lesser General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. +*/ + +#include <new> +#include <string> + +#include "socks_connecter.hpp" +#include "stream_engine.hpp" +#include "platform.hpp" +#include "random.hpp" +#include "err.hpp" +#include "ip.hpp" +#include "tcp.hpp" +#include "address.hpp" +#include "tcp_address.hpp" +#include "session_base.hpp" +#include "socks.hpp" + +#ifdef ZMQ_HAVE_WINDOWS +#include "windows.hpp" +#else +#include <unistd.h> +#include <sys/types.h> +#include <sys/socket.h> +#endif + +zmq::socks_connecter_t::socks_connecter_t (class io_thread_t *io_thread_, + class session_base_t *session_, const options_t &options_, + address_t *addr_, address_t *proxy_addr_, bool delayed_start_) : + own_t (io_thread_, options_), + io_object_t (io_thread_), + addr (addr_), + proxy_addr (proxy_addr_), + status (unplugged), + s (retired_fd), + delayed_start (delayed_start_), + session (session_), + current_reconnect_ivl (options.reconnect_ivl) +{ + zmq_assert (addr); + zmq_assert (addr->protocol == "tcp"); + proxy_addr->to_string (endpoint); + socket = session->get_socket (); +} + +zmq::socks_connecter_t::~socks_connecter_t () +{ + zmq_assert (s == retired_fd); + delete proxy_addr; +} + +void zmq::socks_connecter_t::process_plug () +{ + if (delayed_start) + start_timer (); + else + initiate_connect (); +} + +void zmq::socks_connecter_t::process_term (int linger_) +{ + switch (status) { + case unplugged: + break; + case waiting_for_reconnect_time: + cancel_timer (reconnect_timer_id); + break; + case waiting_for_proxy_connection: + case sending_greeting: + case waiting_for_choice: + case sending_request: + case waiting_for_response: + rm_fd (handle); + if (s != retired_fd) + close (); + break; + } + + own_t::process_term (linger_); +} + +void zmq::socks_connecter_t::in_event () +{ + zmq_assert (status != unplugged + && status != waiting_for_reconnect_time); + + if (status == waiting_for_choice) { + const int rc = choice_decoder.input (s); + if (rc == 0 || rc == -1) + error (); + else + if (choice_decoder.message_ready ()) { + const socks_choice_t choice = choice_decoder.decode (); + const int rc = process_server_response (choice); + if (rc == -1) + error (); + else { + std::string hostname = ""; + uint16_t port = 0; + if (parse_address (addr->address, hostname, port) == -1) + error (); + else { + request_encoder.encode ( + socks_request_t (1, hostname, port)); + reset_pollin (handle); + set_pollout (handle); + status = sending_request; + } + } + } + } + else + if (status == waiting_for_response) { + const int rc = response_decoder.input (s); + if (rc == 0 || rc == -1) + error (); + else + if (response_decoder.message_ready ()) { + const socks_response_t response = response_decoder.decode (); + const int rc = process_server_response (response); + if (rc == -1) + error (); + else { + // Remember our fd for ZMQ_SRCFD in messages + socket->set_fd (s); + + // Create the engine object for this connection. + stream_engine_t *engine = new (std::nothrow) + stream_engine_t (s, options, endpoint); + alloc_assert (engine); + + // Attach the engine to the corresponding session object. + send_attach (session, engine); + + socket->event_connected (endpoint, s); + + rm_fd (handle); + s = -1; + status = unplugged; + + // Shut the connecter down. + terminate (); + } + } + } + else + error (); +} + +void zmq::socks_connecter_t::out_event () +{ + zmq_assert (status == waiting_for_proxy_connection + || status == sending_greeting + || status == sending_request); + + if (status == waiting_for_proxy_connection) { + const int rc = check_proxy_connection (); + if (rc == -1) + error (); + else { + greeting_encoder.encode ( + socks_greeting_t (socks_no_auth_required)); + status = sending_greeting; + } + } + else + if (status == sending_greeting) { + zmq_assert (greeting_encoder.has_pending_data ()); + const int rc = greeting_encoder.output (s); + if (rc == -1 || rc == 0) + error (); + else + if (!greeting_encoder.has_pending_data ()) { + reset_pollout (handle); + set_pollin (handle); + status = waiting_for_choice; + } + } + else { + zmq_assert (request_encoder.has_pending_data ()); + const int rc = request_encoder.output (s); + if (rc == -1 || rc == 0) + error (); + else + if (!request_encoder.has_pending_data ()) { + reset_pollout (handle); + set_pollin (handle); + status = waiting_for_response; + } + } +} + +void zmq::socks_connecter_t::initiate_connect () +{ + // Open the connecting socket. + const int rc = connect_to_proxy (); + + // Connect may succeed in synchronous manner. + if (rc == 0) { + handle = add_fd (s); + set_pollout (handle); + status = sending_greeting; + } + // Connection establishment may be delayed. Poll for its completion. + else + if (errno == EINPROGRESS) { + handle = add_fd (s); + set_pollout (handle); + status = waiting_for_proxy_connection; + socket->event_connect_delayed (endpoint, zmq_errno ()); + } + // Handle any other error condition by eventual reconnect. + else { + if (s != retired_fd) + close (); + start_timer (); + } +} + +int zmq::socks_connecter_t::process_server_response ( + const socks_choice_t &response) +{ + // We do not support any authentication method for now. + return response.method == 0? 0: -1; +} + +int zmq::socks_connecter_t::process_server_response ( + const socks_response_t &response) +{ + return response.response_code == 0? 0: -1; +} + +void zmq::socks_connecter_t::timer_event (int id_) +{ + zmq_assert (status == waiting_for_reconnect_time); + zmq_assert (id_ == reconnect_timer_id); + initiate_connect (); +} + +void zmq::socks_connecter_t::error () +{ + rm_fd (handle); + close (); + greeting_encoder.reset (); + choice_decoder.reset (); + request_encoder.reset (); + response_decoder.reset (); + start_timer (); +} + +void zmq::socks_connecter_t::start_timer () +{ + const int interval = get_new_reconnect_ivl (); + add_timer (interval, reconnect_timer_id); + status = waiting_for_reconnect_time; + socket->event_connect_retried (endpoint, interval); +} + +int zmq::socks_connecter_t::get_new_reconnect_ivl () +{ + // The new interval is the current interval + random value. + const int interval = current_reconnect_ivl + + generate_random () % options.reconnect_ivl; + + // Only change the current reconnect interval if the maximum reconnect + // interval was set and if it's larger than the reconnect interval. + if (options.reconnect_ivl_max > 0 && + options.reconnect_ivl_max > options.reconnect_ivl) + // Calculate the next interval + current_reconnect_ivl = + std::min (current_reconnect_ivl * 2, options.reconnect_ivl_max); + return interval; +} + +int zmq::socks_connecter_t::connect_to_proxy () +{ + zmq_assert (s == retired_fd); + + // Resolve the address + delete proxy_addr->resolved.tcp_addr; + proxy_addr->resolved.tcp_addr = new (std::nothrow) tcp_address_t (); + alloc_assert (proxy_addr->resolved.tcp_addr); + + int rc = proxy_addr->resolved.tcp_addr->resolve ( + proxy_addr->address.c_str (), false, options.ipv6); + if (rc != 0) { + delete proxy_addr->resolved.tcp_addr; + proxy_addr->resolved.tcp_addr = NULL; + return -1; + } + zmq_assert (proxy_addr->resolved.tcp_addr != NULL); + const tcp_address_t *tcp_addr = proxy_addr->resolved.tcp_addr; + + // Create the socket. + s = open_socket (tcp_addr->family (), SOCK_STREAM, IPPROTO_TCP); +#ifdef ZMQ_HAVE_WINDOWS + if (s == INVALID_SOCKET) + return -1; +#else + if (s == -1) + return -1; +#endif + + // On some systems, IPv4 mapping in IPv6 sockets is disabled by default. + // Switch it on in such cases. + if (tcp_addr->family () == AF_INET6) + enable_ipv4_mapping (s); + + // Set the IP Type-Of-Service priority for this socket + if (options.tos != 0) + set_ip_type_of_service (s, options.tos); + + // Set the socket to non-blocking mode so that we get async connect(). + unblock_socket (s); + + // Set the socket buffer limits for the underlying socket. + if (options.sndbuf != 0) + set_tcp_send_buffer (s, options.sndbuf); + if (options.rcvbuf != 0) + set_tcp_receive_buffer (s, options.rcvbuf); + + // Set the IP Type-Of-Service for the underlying socket + if (options.tos != 0) + set_ip_type_of_service (s, options.tos); + + // Set a source address for conversations + if (tcp_addr->has_src_addr ()) { + rc = ::bind (s, tcp_addr->src_addr (), tcp_addr->src_addrlen ()); + if (rc == -1) { + close (); + return -1; + } + } + + // Connect to the remote peer. + rc = ::connect (s, tcp_addr->addr (), tcp_addr->addrlen ()); + + // Connect was successfull immediately. + if (rc == 0) + return 0; + + // Translate error codes indicating asynchronous connect has been + // launched to a uniform EINPROGRESS. +#ifdef ZMQ_HAVE_WINDOWS + const int error_code = WSAGetLastError (); + if (error_code == WSAEINPROGRESS || error_code == WSAEWOULDBLOCK) + errno = EINPROGRESS; + else { + errno = wsa_error_to_errno (error_code); + close (); + } +#else + if (errno == EINTR) + errno = EINPROGRESS; +#endif + return -1; +} + +zmq::fd_t zmq::socks_connecter_t::check_proxy_connection () +{ + // Async connect has finished. Check whether an error occurred + int err = 0; +#ifdef ZMQ_HAVE_HPUX + int len = sizeof err; +#else + socklen_t len = sizeof err; +#endif + + const int rc = getsockopt (s, SOL_SOCKET, SO_ERROR, (char*) &err, &len); + + // Assert if the error was caused by 0MQ bug. + // Networking problems are OK. No need to assert. +#ifdef ZMQ_HAVE_WINDOWS + zmq_assert (rc == 0); + if (err != 0) { + wsa_assert (err == WSAECONNREFUSED + || err == WSAETIMEDOUT + || err == WSAECONNABORTED + || err == WSAEHOSTUNREACH + || err == WSAENETUNREACH + || err == WSAENETDOWN + || err == WSAEACCES + || err == WSAEINVAL + || err == WSAEADDRINUSE); + return -1; + } +#else + // Following code should handle both Berkeley-derived socket + // implementations and Solaris. + if (rc == -1) + err = errno; + if (err != 0) { + errno = err; + errno_assert ( + errno == ECONNREFUSED || + errno == ECONNRESET || + errno == ETIMEDOUT || + errno == EHOSTUNREACH || + errno == ENETUNREACH || + errno == ENETDOWN || + errno == EINVAL); + return -1; + } +#endif + + tune_tcp_socket (s); + tune_tcp_keepalives (s, options.tcp_keepalive, options.tcp_keepalive_cnt, + options.tcp_keepalive_idle, options.tcp_keepalive_intvl); + + return 0; +} + +void zmq::socks_connecter_t::close () +{ + zmq_assert (s != retired_fd); +#ifdef ZMQ_HAVE_WINDOWS + const int rc = closesocket (s); + wsa_assert (rc != SOCKET_ERROR); +#else + const int rc = ::close (s); + errno_assert (rc == 0); +#endif + socket->event_closed (endpoint, s); + s = retired_fd; +} + +int zmq::socks_connecter_t::parse_address ( + const std::string &address_, std::string &hostname_, uint16_t &port_) +{ + // Find the ':' at end that separates address from the port number. + const size_t idx = address_.rfind (':'); + if (idx == std::string::npos) { + errno = EINVAL; + return -1; + } + + // Extract hostname + if (idx < 2 || address_ [0] != '[' || address_ [idx - 1] != ']') + hostname_ = address_.substr (0, idx); + else + hostname_ = address_.substr (1, idx - 2); + + // Separate the hostname/port. + const std::string port_str = address_.substr (idx + 1); + // Parse the port number (0 is not a valid port). + port_ = (uint16_t) atoi (port_str.c_str ()); + if (port_ == 0) { + errno = EINVAL; + return -1; + } + return 0; +} |