mirror of
https://github.com/zeromq/libzmq.git
synced 2025-03-10 16:06:09 +00:00
Problem: socks_connecter_t duplicates code around opening and configuring a TCP socket
Solution: use tcp_open_socket function
This commit is contained in:
parent
68d520ef68
commit
2f7a450294
@ -215,49 +215,26 @@ int zmq::socks_connecter_t::connect_to_proxy ()
|
|||||||
zmq_assert (_s == retired_fd);
|
zmq_assert (_s == retired_fd);
|
||||||
|
|
||||||
// Resolve the address
|
// Resolve the address
|
||||||
LIBZMQ_DELETE (_proxy_addr->resolved.tcp_addr);
|
if (_addr->resolved.tcp_addr != NULL) {
|
||||||
_proxy_addr->resolved.tcp_addr = new (std::nothrow) tcp_address_t ();
|
LIBZMQ_DELETE (_addr->resolved.tcp_addr);
|
||||||
alloc_assert (_proxy_addr->resolved.tcp_addr);
|
}
|
||||||
|
|
||||||
int rc = _proxy_addr->resolved.tcp_addr->resolve (
|
_addr->resolved.tcp_addr = new (std::nothrow) tcp_address_t ();
|
||||||
_proxy_addr->address.c_str (), false, options.ipv6);
|
alloc_assert (_addr->resolved.tcp_addr);
|
||||||
if (rc != 0) {
|
// Automatic fallback to ipv4 is disabled here since this was the existing
|
||||||
LIBZMQ_DELETE (_proxy_addr->resolved.tcp_addr);
|
// behaviour, however I don't see a real reason for this. Maybe this can
|
||||||
|
// be changed to true (and then the parameter can be removed entirely).
|
||||||
|
_s = tcp_open_socket (_addr->address.c_str (), options, false,
|
||||||
|
_addr->resolved.tcp_addr);
|
||||||
|
if (_s == retired_fd) {
|
||||||
|
LIBZMQ_DELETE (_addr->resolved.tcp_addr);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
zmq_assert (_proxy_addr->resolved.tcp_addr != NULL);
|
zmq_assert (_addr->resolved.tcp_addr != NULL);
|
||||||
const tcp_address_t *tcp_addr = _proxy_addr->resolved.tcp_addr;
|
|
||||||
|
|
||||||
// Create the socket.
|
const tcp_address_t *const tcp_addr = _addr->resolved.tcp_addr;
|
||||||
_s = open_socket (tcp_addr->family (), SOCK_STREAM, IPPROTO_TCP);
|
|
||||||
if (_s == retired_fd)
|
|
||||||
return -1;
|
|
||||||
|
|
||||||
// On some systems, IPv4 mapping in IPv6 sockets is disabled by default.
|
int rc;
|
||||||
// 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);
|
|
||||||
|
|
||||||
// Bind the socket to a device if applicable
|
|
||||||
if (!options.bound_device.empty ())
|
|
||||||
bind_to_device (_s, options.bound_device);
|
|
||||||
|
|
||||||
// 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
|
// Set a source address for conversations
|
||||||
if (tcp_addr->has_src_addr ()) {
|
if (tcp_addr->has_src_addr ()) {
|
||||||
|
@ -385,6 +385,7 @@ void zmq::tcp_tune_loopback_fast_path (const fd_t socket_)
|
|||||||
|
|
||||||
zmq::fd_t zmq::tcp_open_socket (const char *address_,
|
zmq::fd_t zmq::tcp_open_socket (const char *address_,
|
||||||
const zmq::options_t &options_,
|
const zmq::options_t &options_,
|
||||||
|
bool fallback_to_ipv4_,
|
||||||
zmq::tcp_address_t *out_tcp_addr_)
|
zmq::tcp_address_t *out_tcp_addr_)
|
||||||
{
|
{
|
||||||
// Convert the textual address into address structure.
|
// Convert the textual address into address structure.
|
||||||
@ -396,8 +397,9 @@ zmq::fd_t zmq::tcp_open_socket (const char *address_,
|
|||||||
fd_t s = open_socket (out_tcp_addr_->family (), SOCK_STREAM, IPPROTO_TCP);
|
fd_t s = open_socket (out_tcp_addr_->family (), SOCK_STREAM, IPPROTO_TCP);
|
||||||
|
|
||||||
// IPv6 address family not supported, try automatic downgrade to IPv4.
|
// IPv6 address family not supported, try automatic downgrade to IPv4.
|
||||||
if (s == retired_fd && out_tcp_addr_->family () == AF_INET6
|
if (s == retired_fd && fallback_to_ipv4_
|
||||||
&& errno == EAFNOSUPPORT && options_.ipv6) {
|
&& out_tcp_addr_->family () == AF_INET6 && errno == EAFNOSUPPORT
|
||||||
|
&& options_.ipv6) {
|
||||||
rc = out_tcp_addr_->resolve (address_, false, false);
|
rc = out_tcp_addr_->resolve (address_, false, false);
|
||||||
if (rc != 0) {
|
if (rc != 0) {
|
||||||
return retired_fd;
|
return retired_fd;
|
||||||
|
@ -79,6 +79,7 @@ void tcp_tune_loopback_fast_path (const fd_t socket_);
|
|||||||
// errno is set to an error code describing the cause of the error.
|
// errno is set to an error code describing the cause of the error.
|
||||||
fd_t tcp_open_socket (const char *address_,
|
fd_t tcp_open_socket (const char *address_,
|
||||||
const options_t &options_,
|
const options_t &options_,
|
||||||
|
bool fallback_to_ipv4_,
|
||||||
tcp_address_t *out_tcp_addr_);
|
tcp_address_t *out_tcp_addr_);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -174,7 +174,7 @@ int zmq::tcp_connecter_t::open ()
|
|||||||
|
|
||||||
_addr->resolved.tcp_addr = new (std::nothrow) tcp_address_t ();
|
_addr->resolved.tcp_addr = new (std::nothrow) tcp_address_t ();
|
||||||
alloc_assert (_addr->resolved.tcp_addr);
|
alloc_assert (_addr->resolved.tcp_addr);
|
||||||
_s = tcp_open_socket (_addr->address.c_str (), options,
|
_s = tcp_open_socket (_addr->address.c_str (), options, true,
|
||||||
_addr->resolved.tcp_addr);
|
_addr->resolved.tcp_addr);
|
||||||
if (_s == retired_fd) {
|
if (_s == retired_fd) {
|
||||||
LIBZMQ_DELETE (_addr->resolved.tcp_addr);
|
LIBZMQ_DELETE (_addr->resolved.tcp_addr);
|
||||||
|
@ -103,7 +103,7 @@ zmq::tcp_listener_t::get_socket_name (zmq::fd_t fd_,
|
|||||||
|
|
||||||
int zmq::tcp_listener_t::create_socket (const char *addr_)
|
int zmq::tcp_listener_t::create_socket (const char *addr_)
|
||||||
{
|
{
|
||||||
_s = tcp_open_socket (addr_, options, &_address);
|
_s = tcp_open_socket (addr_, options, true, &_address);
|
||||||
if (_s == retired_fd) {
|
if (_s == retired_fd) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user