3party/libzmq
0
0
Fork 0
mirror of https://github.com/zeromq/libzmq.git synced 2025-03-11 00:16:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Problem: complexity of term_endpoint

Browse Source 
Solution: extract resolve_tcp_addr function
This commit is contained in:
Simon Giesecke 2018-08-15 13:11:57 +02:00
parent a2d736c14f
commit cdc6c66f9b
2 changed files with 37 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
src/socket_base.cpp
View File

@ -968,6 +968,33 @@ int zmq::socket_base_t::connect (const char *addr_)
return 0; return 0;
} }
std::string zmq::socket_base_t::resolve_tcp_addr (std::string endpoint_address_,
const char *tcp_address_)
{
// The resolved last_endpoint is used as a key in the endpoints map.
// The address passed by the user might not match in the TCP case due to
// IPv4-in-IPv6 mapping (EG: tcp://[::ffff:127.0.0.1]:9999), so try to
// resolve before giving up. Given at this stage we don't know whether a
// socket is connected or bound, try with both.
if (_endpoints.find (endpoint_address_) == _endpoints.end ()) {
tcp_address_t *tcp_addr = new (std::nothrow) tcp_address_t ();
alloc_assert (tcp_addr);
int rc = tcp_addr->resolve (tcp_address_, false, options.ipv6);
if (rc == 0) {
tcp_addr->to_string (endpoint_address_);
if (_endpoints.find (endpoint_address_) == _endpoints.end ()) {
rc = tcp_addr->resolve (tcp_address_, true, options.ipv6);
if (rc == 0) {
tcp_addr->to_string (endpoint_address_);
}
}
}
LIBZMQ_DELETE (tcp_addr);
}
return endpoint_address_;
}
void zmq::socket_base_t::add_endpoint (const char *addr_, void zmq::socket_base_t::add_endpoint (const char *addr_,
own_t *endpoint_, own_t *endpoint_,
pipe_t *pipe_) pipe_t *pipe_)
@ -996,7 +1023,7 @@ int zmq::socket_base_t::term_endpoint (const char *addr_)
// Process pending commands, if any, since there could be pending unprocessed process_own()'s // Process pending commands, if any, since there could be pending unprocessed process_own()'s
// (from launch_child() for example) we're asked to terminate now. // (from launch_child() for example) we're asked to terminate now.
int rc = process_commands (0, false); const int rc = process_commands (0, false);
if (unlikely (rc != 0)) { if (unlikely (rc != 0)) {
return -1; return -1;
} }
@ -1008,41 +1035,19 @@ int zmq::socket_base_t::term_endpoint (const char *addr_)
return -1; return -1;
} }
// Disconnect an inproc socket
if (protocol == "inproc") {
const std::string addr_str = std::string (addr_); const std::string addr_str = std::string (addr_);
// Disconnect an inproc socket
if (protocol == "inproc") {
return unregister_endpoint (addr_str, this) == 0 return unregister_endpoint (addr_str, this) == 0
? 0 ? 0
: _inprocs.erase_pipes (addr_str); : _inprocs.erase_pipes (addr_str);
} }
std::string resolved_addr = addr_; const std::string resolved_addr =
protocol == protocol_name::tcp
// The resolved last_endpoint is used as a key in the endpoints map. ? resolve_tcp_addr (addr_str, address.c_str ())
// The address passed by the user might not match in the TCP case due to : addr_str;
// IPv4-in-IPv6 mapping (EG: tcp://[::ffff:127.0.0.1]:9999), so try to
// resolve before giving up. Given at this stage we don't know whether a
// socket is connected or bound, try with both.
if (protocol == protocol_name::tcp) {
if (_endpoints.find (resolved_addr) == _endpoints.end ()) {
tcp_address_t *tcp_addr = new (std::nothrow) tcp_address_t ();
alloc_assert (tcp_addr);
rc = tcp_addr->resolve (address.c_str (), false, options.ipv6);
if (rc == 0) {
tcp_addr->to_string (resolved_addr);
if (_endpoints.find (resolved_addr) == _endpoints.end ()) {
rc =
tcp_addr->resolve (address.c_str (), true, options.ipv6);
if (rc == 0) {
tcp_addr->to_string (resolved_addr);
}
}
}
LIBZMQ_DELETE (tcp_addr);
}
}
// Find the endpoints range (if any) corresponding to the addr_ string. // Find the endpoints range (if any) corresponding to the addr_ string.
const std::pair<endpoints_t::iterator, endpoints_t::iterator> range = const std::pair<endpoints_t::iterator, endpoints_t::iterator> range =

3
src/socket_base.hpp
View File

@ -259,6 +259,9 @@ class socket_base_t : public own_t,
void update_pipe_options (int option_); void update_pipe_options (int option_);
std::string resolve_tcp_addr (std::string endpoint_address_,
const char *tcp_address_);
// Socket's mailbox object. // Socket's mailbox object.
i_mailbox *_mailbox; i_mailbox *_mailbox;