Prevent core from doing DNS requests when UDP is disabled.

This commit is contained in:
irungentoo 2014-09-20 12:19:21 -04:00
parent 53ce4f393b
commit fcc6d43cf2
No known key found for this signature in database
GPG Key ID: 10349DC9BED89E98

View File

@ -808,33 +808,40 @@ uint64_t tox_file_data_remaining(const Tox *tox, int32_t friendnumber, uint8_t f
int tox_add_tcp_relay(Tox *tox, const char *address, uint16_t port, const uint8_t *public_key) int tox_add_tcp_relay(Tox *tox, const char *address, uint16_t port, const uint8_t *public_key)
{ {
Messenger *m = tox; Messenger *m = tox;
IP_Port ip_port_v64; IP_Port ip_port, ip_port_v4;
IP *ip_extra = NULL; if (!addr_parse_ip(address, &ip_port.ip)) {
IP_Port ip_port_v4; if (m->options.udp_disabled) /* Disable DNS when udp is disabled. */
ip_init(&ip_port_v64.ip, m->options.ipv6enabled); return 0;
if (m->options.ipv6enabled) { IP *ip_extra = NULL;
/* setup for getting BOTH: an IPv6 AND an IPv4 address */ ip_init(&ip_port.ip, m->options.ipv6enabled);
ip_port_v64.ip.family = AF_UNSPEC;
ip_reset(&ip_port_v4.ip); if (m->options.ipv6enabled) {
ip_extra = &ip_port_v4.ip; /* setup for getting BOTH: an IPv6 AND an IPv4 address */
ip_port.ip.family = AF_UNSPEC;
ip_reset(&ip_port_v4.ip);
ip_extra = &ip_port_v4.ip;
}
if (!addr_resolve(address, &ip_port.ip, ip_extra))
return 0;
} }
if (addr_resolve_or_parse_ip(address, &ip_port_v64.ip, ip_extra)) { ip_port.port = htons(port);
ip_port_v64.port = htons(port); add_tcp_relay(m->net_crypto, ip_port, public_key);
add_tcp_relay(m->net_crypto, ip_port_v64, public_key); onion_add_path_node(m->onion_c, ip_port, public_key); //TODO: move this
onion_add_path_node(m->onion_c, ip_port_v64, public_key); //TODO: move this return 1;
return 1;
} else {
return 0;
}
} }
int tox_bootstrap_from_address(Tox *tox, const char *address, uint16_t port, const uint8_t *public_key) int tox_bootstrap_from_address(Tox *tox, const char *address, uint16_t port, const uint8_t *public_key)
{ {
Messenger *m = tox; Messenger *m = tox;
tox_add_tcp_relay(tox, address, port, public_key); int ret = tox_add_tcp_relay(tox, address, port, public_key);
return DHT_bootstrap_from_address(m->dht, address, m->options.ipv6enabled, htons(port), public_key); if (m->options.udp_disabled) {
return ret;
} else { /* DHT only works on UDP. */
return DHT_bootstrap_from_address(m->dht, address, m->options.ipv6enabled, htons(port), public_key);
}
} }
/* return 0 if we are not connected to the DHT. /* return 0 if we are not connected to the DHT.