mirror of
https://github.com/matrix-construct/construct
synced 2024-08-26 19:35:17 +02:00
318 lines
7 KiB
C++
318 lines
7 KiB
C++
/* authd/dns.c - authd DNS functions
|
|
* Copyright (c) 2016 William Pitcock <nenolod@dereferenced.org>
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice is present in all copies.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
|
|
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
|
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
|
|
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "authd.h"
|
|
#include "dns.h"
|
|
#include "notice.h"
|
|
#include "res.h"
|
|
|
|
using namespace ircd::defaults;
|
|
|
|
static void handle_lookup_ip_reply(void *data, struct DNSReply *reply);
|
|
static void handle_lookup_hostname_reply(void *data, struct DNSReply *reply);
|
|
|
|
uint64_t query_count = 0;
|
|
|
|
/* A bit different from ircd... you just get a dns_query object.
|
|
*
|
|
* It gets freed whenever the res code gets back to us.
|
|
*/
|
|
struct dns_query *
|
|
lookup_ip(const char *host, int aftype, DNSCB callback, void *data)
|
|
{
|
|
struct dns_query *query = (dns_query *)rb_malloc(sizeof(struct dns_query));
|
|
int g_type;
|
|
|
|
if(aftype == AF_INET)
|
|
{
|
|
query->type = QUERY_A;
|
|
g_type = T_A;
|
|
}
|
|
#ifdef RB_IPV6
|
|
else if(aftype == AF_INET6)
|
|
{
|
|
query->type = QUERY_AAAA;
|
|
g_type = T_AAAA;
|
|
}
|
|
#endif
|
|
else
|
|
{
|
|
rb_free(query);
|
|
return NULL;
|
|
}
|
|
|
|
query->id = query_count++;
|
|
query->callback = callback;
|
|
query->data = data;
|
|
|
|
query->query.ptr = query;
|
|
query->query.callback = handle_lookup_ip_reply;
|
|
|
|
gethost_byname_type(host, &query->query, g_type);
|
|
|
|
return query;
|
|
}
|
|
|
|
/* See lookup_ip's comment */
|
|
struct dns_query *
|
|
lookup_hostname(const char *ip, DNSCB callback, void *data)
|
|
{
|
|
struct dns_query *query = (dns_query *)rb_malloc(sizeof(struct dns_query));
|
|
int aftype;
|
|
|
|
if(!rb_inet_pton_sock(ip, (struct sockaddr *)&query->addr))
|
|
{
|
|
rb_free(query);
|
|
return NULL;
|
|
}
|
|
|
|
aftype = GET_SS_FAMILY(&query->addr);
|
|
|
|
if(aftype == AF_INET)
|
|
query->type = QUERY_PTR_A;
|
|
#ifdef RB_IPV6
|
|
else if(aftype == AF_INET6)
|
|
query->type = QUERY_PTR_AAAA;
|
|
#endif
|
|
else
|
|
{
|
|
rb_free(query);
|
|
return NULL;
|
|
}
|
|
|
|
query->id = query_count++;
|
|
query->callback = callback;
|
|
query->data = data;
|
|
|
|
query->query.ptr = query;
|
|
query->query.callback = handle_lookup_hostname_reply;
|
|
|
|
gethost_byaddr(&query->addr, &query->query);
|
|
|
|
return query;
|
|
}
|
|
|
|
/* Cancel a pending query */
|
|
void
|
|
cancel_query(struct dns_query *query)
|
|
{
|
|
query->callback = NULL;
|
|
query->data = NULL;
|
|
}
|
|
|
|
/* Callback from gethost_byname_type */
|
|
static void
|
|
handle_lookup_ip_reply(void *data, struct DNSReply *reply)
|
|
{
|
|
struct dns_query *query = (dns_query *)data;
|
|
char ip[HOSTIPLEN] = "*";
|
|
|
|
if(query == NULL)
|
|
{
|
|
/* Shouldn't happen */
|
|
warn_opers(L_CRIT, "DNS: handle_lookup_ip_reply: query == NULL!");
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
|
|
if(reply == NULL)
|
|
goto end;
|
|
|
|
switch(query->type)
|
|
{
|
|
case QUERY_A:
|
|
if(GET_SS_FAMILY(&reply->addr) == AF_INET)
|
|
rb_inet_ntop_sock((struct sockaddr *)&reply->addr, ip, sizeof(ip));
|
|
break;
|
|
#ifdef RB_IPV6
|
|
case QUERY_AAAA:
|
|
if(GET_SS_FAMILY(&reply->addr) == AF_INET6)
|
|
{
|
|
rb_inet_ntop_sock((struct sockaddr *)&reply->addr, ip, sizeof(ip));
|
|
if(ip[0] == ':')
|
|
{
|
|
memmove(&ip[1], ip, strlen(ip));
|
|
ip[0] = '0';
|
|
}
|
|
}
|
|
break;
|
|
#endif
|
|
default:
|
|
warn_opers(L_CRIT, "DNS: handle_lookup_ip_reply: unknown query type %d",
|
|
query->type);
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
|
|
end:
|
|
if(query->callback)
|
|
query->callback(ip, ip[0] != '*', query->type, query->data);
|
|
|
|
rb_free(query);
|
|
}
|
|
|
|
/* Callback from gethost_byaddr */
|
|
static void
|
|
handle_lookup_hostname_reply(void *data, struct DNSReply *reply)
|
|
{
|
|
struct dns_query *query = (dns_query *)data;
|
|
char *hostname = NULL;
|
|
|
|
if(query == NULL)
|
|
{
|
|
/* Shouldn't happen */
|
|
warn_opers(L_CRIT, "DNS: handle_lookup_hostname_reply: query == NULL!");
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
|
|
if(reply == NULL)
|
|
goto end;
|
|
|
|
if(query->type == QUERY_PTR_A)
|
|
{
|
|
struct sockaddr_in *ip, *ip_fwd;
|
|
ip = (struct sockaddr_in *) &query->addr;
|
|
ip_fwd = (struct sockaddr_in *) &reply->addr;
|
|
|
|
if(ip->sin_addr.s_addr == ip_fwd->sin_addr.s_addr)
|
|
hostname = reply->h_name;
|
|
}
|
|
#ifdef RB_IPV6
|
|
else if(query->type == QUERY_PTR_AAAA)
|
|
{
|
|
struct sockaddr_in6 *ip, *ip_fwd;
|
|
ip = (struct sockaddr_in6 *) &query->addr;
|
|
ip_fwd = (struct sockaddr_in6 *) &reply->addr;
|
|
|
|
if(memcmp(&ip->sin6_addr, &ip_fwd->sin6_addr, sizeof(struct in6_addr)) == 0)
|
|
hostname = reply->h_name;
|
|
}
|
|
#endif
|
|
else
|
|
{
|
|
/* Shouldn't happen */
|
|
warn_opers(L_CRIT, "DNS: handle_lookup_hostname_reply: unknown query type %d",
|
|
query->type);
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
end:
|
|
if(query->callback)
|
|
query->callback(hostname, hostname != NULL, query->type, query->data);
|
|
|
|
rb_free(query);
|
|
}
|
|
|
|
static void
|
|
submit_dns_answer(const char *reply, bool status, query_type type, void *data)
|
|
{
|
|
char *id = (char *)data;
|
|
|
|
if(!id || type == QUERY_INVALID)
|
|
{
|
|
warn_opers(L_CRIT, "DNS: submit_dns_answer gave us a bad query");
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
|
|
if(reply == NULL || status == false)
|
|
{
|
|
rb_helper_write(authd_helper, "E %s E %c *", id, type);
|
|
rb_free(id);
|
|
return;
|
|
}
|
|
|
|
rb_helper_write(authd_helper, "E %s O %c %s", id, type, reply);
|
|
rb_free(id);
|
|
}
|
|
|
|
void
|
|
handle_resolve_dns(int parc, char *parv[])
|
|
{
|
|
char *id = rb_strdup(parv[1]);
|
|
char qtype = *parv[2];
|
|
char *record = parv[3];
|
|
int aftype = AF_INET;
|
|
|
|
switch(qtype)
|
|
{
|
|
#ifdef RB_IPV6
|
|
case '6':
|
|
aftype = AF_INET6;
|
|
#endif
|
|
case '4':
|
|
if(!lookup_ip(record, aftype, submit_dns_answer, id))
|
|
submit_dns_answer(NULL, false, query_type(qtype), NULL);
|
|
break;
|
|
#ifdef RB_IPV6
|
|
case 'S':
|
|
#endif
|
|
case 'R':
|
|
if(!lookup_hostname(record, submit_dns_answer, id))
|
|
submit_dns_answer(NULL, false, query_type(qtype), NULL);
|
|
break;
|
|
default:
|
|
warn_opers(L_CRIT, "DNS: handle_resolve_dns got an unknown query: %c", qtype);
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
}
|
|
|
|
void
|
|
enumerate_nameservers(uint32_t rid, const char letter)
|
|
{
|
|
char buf[(HOSTIPLEN + 1) * IRCD_MAXNS];
|
|
size_t s = 0;
|
|
|
|
if (!irc_nscount)
|
|
{
|
|
/* Shouldn't happen */
|
|
warn_opers(L_CRIT, "DNS: no name servers!");
|
|
stats_error(rid, letter, "NONAMESERVERS");
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
|
|
for(int i = 0; i < irc_nscount; i++)
|
|
{
|
|
char addr[HOSTIPLEN];
|
|
size_t addrlen;
|
|
|
|
rb_inet_ntop_sock((struct sockaddr *)&irc_nsaddr_list[i], addr, sizeof(addr));
|
|
|
|
if (!addr[0])
|
|
{
|
|
/* Shouldn't happen */
|
|
warn_opers(L_CRIT, "DNS: bad nameserver!");
|
|
stats_error(rid, letter, "INVALIDNAMESERVER");
|
|
exit(EX_DNS_ERROR);
|
|
}
|
|
|
|
addrlen = strlen(addr) + 1;
|
|
(void)snprintf(&buf[s], sizeof(buf) - s, "%s ", addr);
|
|
s += addrlen;
|
|
}
|
|
|
|
if(s > 0)
|
|
buf[--s] = '\0';
|
|
|
|
stats_result(rid, letter, "%s", buf);
|
|
}
|
|
|
|
void
|
|
reload_nameservers(const char letter)
|
|
{
|
|
/* Not a whole lot to it */
|
|
restart_resolver();
|
|
}
|