summaryrefslogtreecommitdiffstats
path: root/external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp
diff options
context:
space:
mode:
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.cpp478
1 files changed, 0 insertions, 478 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
deleted file mode 100644
index a3b70436..00000000
--- a/external_libs/python/pyzmq-14.7.0/bundled/zeromq/src/socks_connecter.cpp
+++ /dev/null
@@ -1,478 +0,0 @@
-/*
- 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;
-}