d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen#include "bus-errors.h"

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen#include "bus-util.h"

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen#include "resolved.h"

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen#include "resolved-dns-domain.h"

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen

b5efdb8af40ea759a1ea584c1bc44ecc81dd00ceLennart Poetteringstatic int reply_query_state(DnsQuery *q) {

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering _cleanup_free_ char *ip = NULL;

6bedfcbb2970e06a4d3280c8fb62083d252ede73Lennart Poettering const char *name;

a8fbdf5424be099ba1b2b1ec261c02b8759d6b0cThomas Hindoe Paaboel Andersen int r;

4349cd7c1d153c4ffa23cf1cff1644e0afa9bcf0Lennart Poettering

6bedfcbb2970e06a4d3280c8fb62083d252ede73Lennart Poettering if (q->request_hostname)

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers name = q->request_hostname;

07630cea1f3a845c09309f197ac7c4f11edd3b62Lennart Poettering else {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = in_addr_to_string(q->request_family, &q->request_address, &ip);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers name = ip;

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers }

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers switch (q->state) {

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers case DNS_TRANSACTION_NO_SERVERS:

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers return sd_bus_reply_method_errorf(q->request, BUS_ERROR_NO_NAME_SERVERS, "No appropriate name servers or networks for name found");

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers case DNS_TRANSACTION_TIMEOUT:

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers return sd_bus_reply_method_errorf(q->request, SD_BUS_ERROR_TIMEOUT, "Query timed out");

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers case DNS_TRANSACTION_ATTEMPTS_MAX_REACHED:

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers return sd_bus_reply_method_errorf(q->request, SD_BUS_ERROR_TIMEOUT, "All attempts to contact name servers or networks failed");

0b6b6787e3f0ae8906ce0212bd629edbe931b73dKay Sievers

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek case DNS_TRANSACTION_RESOURCES:

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return sd_bus_reply_method_errorf(q->request, BUS_ERROR_NO_RESOURCES, "Not enough resources");

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek case DNS_TRANSACTION_INVALID_REPLY:

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return sd_bus_reply_method_errorf(q->request, BUS_ERROR_INVALID_REPLY, "Received invalid reply");

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek case DNS_TRANSACTION_FAILURE: {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (q->answer_rcode == DNS_RCODE_NXDOMAIN)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek sd_bus_error_setf(&error, _BUS_ERROR_DNS "NXDOMAIN", "'%s' not found", name);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek else {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek const char *rc, *n;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek char p[3]; /* the rcode is 4 bits long */

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek rc = dns_rcode_to_string(q->answer_rcode);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (!rc) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek sprintf(p, "%i", q->answer_rcode);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek rc = p;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek n = strappenda(_BUS_ERROR_DNS, rc);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek sd_bus_error_setf(&error, n, "Could not resolve '%s', server or network returned error %s", name, rc);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return sd_bus_reply_method_error(q->request, &error);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek case DNS_TRANSACTION_NULL:

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek case DNS_TRANSACTION_PENDING:

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek case DNS_TRANSACTION_SUCCESS:

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek default:

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek assert_not_reached("Impossible state");

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek}

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmekstatic int append_address(sd_bus_message *reply, DnsResourceRecord *rr, int ifindex) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek int r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek assert(reply);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek assert(rr);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_open_container(reply, 'r', "iayi");

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (rr->key->type == DNS_TYPE_A) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_append(reply, "i", AF_INET);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_append_array(reply, 'y', &rr->a.in_addr, sizeof(struct in_addr));

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek } else if (rr->key->type == DNS_TYPE_AAAA) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_append(reply, "i", AF_INET6);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_append_array(reply, 'y', &rr->aaaa.in6_addr, sizeof(struct in6_addr));

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek } else

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return -EAFNOSUPPORT;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_append(reply, "i", ifindex);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_close_container(reply);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return r;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek return 0;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek}

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmekstatic void bus_method_resolve_hostname_complete(DnsQuery *q) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *cname = NULL, *canonical = NULL;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek _cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek unsigned added = 0, i;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek int r, ifindex;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek assert(q);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (q->state != DNS_TRANSACTION_SUCCESS) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = reply_query_state(q);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek goto finish;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_new_method_return(q->request, &reply);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek goto finish;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_message_open_container(reply, 'a', "(iayi)");

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek goto finish;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek ifindex = q->answer_ifindex;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (q->answer) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek answer = dns_answer_ref(q->answer);

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak for (i = 0; i < answer->n_rrs; i++) {

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak r = dns_question_matches_rr(q->question, answer->rrs[i]);

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (r < 0)

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak goto finish;

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (r == 0) {

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak /* Hmm, if this is not an address record,

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak r = dns_question_matches_cname(q->question, answer->rrs[i]);

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (r < 0)

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak goto finish;

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (r > 0)

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak cname = dns_resource_record_ref(answer->rrs[i]);

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak continue;

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak }

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak r = append_address(reply, answer->rrs[i], ifindex);

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (r < 0)

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak goto finish;

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (!canonical)

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak canonical = dns_resource_record_ref(answer->rrs[i]);

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak added ++;

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak }

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak }

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (added <= 0) {

3519d230c8bafe834b2dac26ace49fcfba139823Karel Zak if (!cname) {

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_reply_method_errorf(q->request, BUS_ERROR_NO_SUCH_RR, "'%s' does not have any RR of requested type", q->request_hostname);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek goto finish;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek /* This has a cname? Then update the query with the

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = dns_query_cname_redirect(q, cname->cname.name);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r < 0) {

a75f4e2a02e287294b21ae9e5b1f28b2f8faea39Zbigniew Jędrzejewski-Szmek if (r == -ELOOP)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek r = sd_bus_reply_method_errorf(q->request, BUS_ERROR_CNAME_LOOP, "CNAME loop on '%s'", q->request_hostname);

a75f4e2a02e287294b21ae9e5b1f28b2f8faea39Zbigniew Jędrzejewski-Szmek else

a75f4e2a02e287294b21ae9e5b1f28b2f8faea39Zbigniew Jędrzejewski-Szmek r = sd_bus_reply_method_errno(q->request, -r, NULL);

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek goto finish;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek }

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek /* Before we restart the query, let's see if any of

c5e04d51277994cca29234b33a6b8fc90a183cf3Zbigniew Jędrzejewski-Szmek for (i = 0; i < answer->n_rrs; i++) {

c5e04d51277994cca29234b33a6b8fc90a183cf3Zbigniew Jędrzejewski-Szmek r = dns_question_matches_rr(q->question, answer->rrs[i]);

a75f4e2a02e287294b21ae9e5b1f28b2f8faea39Zbigniew Jędrzejewski-Szmek if (r < 0)

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek goto finish;

d15d0333be6a1ca7fdd99a1881d967b6be8f387aZbigniew Jędrzejewski-Szmek if (r == 0)

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering continue;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = append_address(reply, answer->rrs[i], ifindex);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (r < 0)

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering goto finish;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (!canonical)

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering canonical = dns_resource_record_ref(answer->rrs[i]);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering added++;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering }

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering /* If we didn't find anything, then let's restart the

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (added <= 0) {

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = dns_query_go(q);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (r == -ESRCH) {

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = sd_bus_reply_method_errorf(q->request, BUS_ERROR_NO_NAME_SERVERS, "No appropriate name servers or networks for name found");

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering goto finish;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering }

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (r < 0) {

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = sd_bus_reply_method_errno(q->request, -r, NULL);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering goto finish;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering }

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering return;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering }

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering }

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = sd_bus_message_close_container(reply);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (r < 0)

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering goto finish;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering /* Return the precise spelling and uppercasing reported by the server */

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering assert(canonical);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = sd_bus_message_append(reply, "s", DNS_RESOURCE_KEY_NAME(canonical->key));

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (r < 0)

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering goto finish;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering r = sd_bus_send(q->manager->bus, reply, NULL);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poetteringfinish:

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering if (r < 0) {

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering log_error("Failed to send hostname reply: %s", strerror(-r));

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering sd_bus_reply_method_errno(q->request, -r, NULL);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering }

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering dns_query_free(q);

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering}

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poetteringstatic int bus_method_resolve_hostname(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering _cleanup_(dns_question_unrefp) DnsQuestion *question = NULL;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering Manager *m = userdata;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering const char *hostname;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering int family;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering DnsQuery *q;

6550203eb471595e41e27f46e5d0a00a4c0e47bbLennart Poettering int r;

assert(bus);

assert(message);

assert(m);

r = sd_bus_message_read(message, "si", &hostname, &family);

if (r < 0)

return r;

if (!IN_SET(family, AF_INET, AF_INET6, AF_UNSPEC))

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Unknown address family %i", family);

r = dns_name_normalize(hostname, NULL);

if (r < 0)

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid hostname '%s'", hostname);

question = dns_question_new(family == AF_UNSPEC ? 2 : 1);

if (!question)

return -ENOMEM;

if (family != AF_INET6) {

_cleanup_(dns_resource_key_unrefp) DnsResourceKey *key = NULL;

key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, hostname);

if (!key)

return -ENOMEM;

r = dns_question_add(question, key);

if (r < 0)

return r;

}

if (family != AF_INET) {

_cleanup_(dns_resource_key_unrefp) DnsResourceKey *key = NULL;

key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_AAAA, hostname);

if (!key)

return -ENOMEM;

r = dns_question_add(question, key);

if (r < 0)

return r;

}

r = dns_query_new(m, &q, question);

if (r < 0)

return r;

q->request = sd_bus_message_ref(message);

q->request_family = family;

q->request_hostname = hostname;

q->complete = bus_method_resolve_hostname_complete;

r = dns_query_go(q);

if (r < 0) {

dns_query_free(q);

if (r == -ESRCH)

sd_bus_error_setf(error, BUS_ERROR_NO_NAME_SERVERS, "No appropriate name servers or networks for name found");

return r;

}

return 1;

}

static void bus_method_resolve_address_complete(DnsQuery *q) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL;

unsigned added = 0, i;

int r;

assert(q);

if (q->state != DNS_TRANSACTION_SUCCESS) {

r = reply_query_state(q);

goto finish;

}

r = sd_bus_message_new_method_return(q->request, &reply);

if (r < 0)

goto finish;

r = sd_bus_message_open_container(reply, 'a', "s");

if (r < 0)

goto finish;

if (q->answer) {

answer = dns_answer_ref(q->answer);

for (i = 0; i < answer->n_rrs; i++) {

r = dns_question_matches_rr(q->question, answer->rrs[i]);

if (r < 0)

goto finish;

if (r == 0)

continue;

r = sd_bus_message_append(reply, "s", answer->rrs[i]->ptr.name);

if (r < 0)

goto finish;

added ++;

}

}

if (added <= 0) {

_cleanup_free_ char *ip = NULL;

in_addr_to_string(q->request_family, &q->request_address, &ip);

r = sd_bus_reply_method_errorf(q->request, BUS_ERROR_NO_SUCH_RR, "Address '%s' does not have any RR of requested type", ip);

goto finish;

}

r = sd_bus_message_close_container(reply);

if (r < 0)

goto finish;

r = sd_bus_send(q->manager->bus, reply, NULL);

finish:

if (r < 0) {

log_error("Failed to send address reply: %s", strerror(-r));

sd_bus_reply_method_errno(q->request, -r, NULL);

}

dns_query_free(q);

}

static int bus_method_resolve_address(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {

_cleanup_(dns_resource_key_unrefp) DnsResourceKey *key = NULL;

_cleanup_(dns_question_unrefp) DnsQuestion *question = NULL;

_cleanup_free_ char *reverse = NULL;

Manager *m = userdata;

int family, ifindex;

const void *d;

DnsQuery *q;

size_t sz;

int r;

assert(bus);

assert(message);

assert(m);

r = sd_bus_message_read(message, "i", &family);

if (r < 0)

return r;

if (!IN_SET(family, AF_INET, AF_INET6))

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Unknown address family %i", family);

r = sd_bus_message_read_array(message, 'y', &d, &sz);

if (r < 0)

return r;

if (sz != FAMILY_ADDRESS_SIZE(family))

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid address size");

r = sd_bus_message_read(message, "i", &ifindex);

if (r < 0)

return r;

if (ifindex < 0)

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid interface index");

r = dns_name_reverse(family, d, &reverse);

if (r < 0)

return r;

question = dns_question_new(1);

if (!question)

return -ENOMEM;

key = dns_resource_key_new_consume(DNS_CLASS_IN, DNS_TYPE_PTR, reverse);

if (!key)

return -ENOMEM;

reverse = NULL;

r = dns_question_add(question, key);

if (r < 0)

return r;

r = dns_query_new(m, &q, question);

if (r < 0)

return r;

q->request = sd_bus_message_ref(message);

q->request_family = family;

memcpy(&q->request_address, d, sz);

q->complete = bus_method_resolve_address_complete;

r = dns_query_go(q);

if (r < 0) {

dns_query_free(q);

if (r == -ESRCH)

sd_bus_error_setf(error, BUS_ERROR_NO_NAME_SERVERS, "No appropriate name servers or networks for name found");

return r;

}

return 1;

}

static void bus_method_resolve_record_complete(DnsQuery *q) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL;

unsigned added = 0, i;

int r;

assert(q);

if (q->state != DNS_TRANSACTION_SUCCESS) {

r = reply_query_state(q);

goto finish;

}

r = sd_bus_message_new_method_return(q->request, &reply);

if (r < 0)

goto finish;

r = sd_bus_message_open_container(reply, 'a', "(qqay)");

if (r < 0)

goto finish;

if (q->answer) {

answer = dns_answer_ref(q->answer);

for (i = 0; i < answer->n_rrs; i++) {

_cleanup_(dns_packet_unrefp) DnsPacket *p = NULL;

size_t start;

r = dns_question_matches_rr(q->question, answer->rrs[i]);

if (r < 0)

goto finish;

if (r == 0)

continue;

r = dns_packet_new(&p, DNS_PROTOCOL_DNS, 0);

if (r < 0)

goto finish;

r = dns_packet_append_rr(p, answer->rrs[i], &start);

if (r < 0)

goto finish;

r = sd_bus_message_open_container(reply, 'r', "qqay");

if (r < 0)

goto finish;

r = sd_bus_message_append(reply, "qq", answer->rrs[i]->key->class, answer->rrs[i]->key->type);

if (r < 0)

goto finish;

r = sd_bus_message_append_array(reply, 'y', DNS_PACKET_DATA(p) + start, p->size - start);

if (r < 0)

goto finish;

r = sd_bus_message_close_container(reply);

if (r < 0)

goto finish;

added ++;

}

}

if (added <= 0) {

r = sd_bus_reply_method_errorf(q->request, BUS_ERROR_NO_SUCH_RR, "Name '%s' does not have any RR of the requested type", q->request_hostname);

goto finish;

}

r = sd_bus_message_close_container(reply);

if (r < 0)

goto finish;

r = sd_bus_send(q->manager->bus, reply, NULL);

finish:

if (r < 0) {

log_error("Failed to send record reply: %s", strerror(-r));

sd_bus_reply_method_errno(q->request, -r, NULL);

}

dns_query_free(q);

}

static int bus_method_resolve_record(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {

_cleanup_(dns_resource_key_unrefp) DnsResourceKey *key = NULL;

_cleanup_(dns_question_unrefp) DnsQuestion *question = NULL;

_cleanup_free_ char *reverse = NULL;

Manager *m = userdata;

DnsQuery *q;

int r;

uint16_t class, type;

const char *name;

assert(bus);

assert(message);

assert(m);

r = sd_bus_message_read(message, "sqq", &name, &class, &type);

if (r < 0)

return r;

r = dns_name_normalize(name, NULL);

if (r < 0)

return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid name '%s'", name);

question = dns_question_new(1);

if (!question)

return -ENOMEM;

key = dns_resource_key_new(class, type, name);

if (!key)

return -ENOMEM;

r = dns_question_add(question, key);

if (r < 0)

return r;

r = dns_query_new(m, &q, question);

if (r < 0)

return r;

q->request = sd_bus_message_ref(message);

q->request_hostname = name;

q->complete = bus_method_resolve_record_complete;

r = dns_query_go(q);

if (r < 0) {

dns_query_free(q);

if (r == -ESRCH)

sd_bus_error_setf(error, BUS_ERROR_NO_NAME_SERVERS, "No appropriate name servers or networks for name found");

return r;

}

return 1;

}

static const sd_bus_vtable resolve_vtable[] = {

SD_BUS_VTABLE_START(0),

SD_BUS_METHOD("ResolveHostname", "si", "a(iayi)s", bus_method_resolve_hostname, SD_BUS_VTABLE_UNPRIVILEGED),

SD_BUS_METHOD("ResolveAddress", "iayi", "as", bus_method_resolve_address, SD_BUS_VTABLE_UNPRIVILEGED),

SD_BUS_METHOD("ResolveRecord", "sqq", "a(qqay)", bus_method_resolve_record, SD_BUS_VTABLE_UNPRIVILEGED),

SD_BUS_VTABLE_END,

};

static int on_bus_retry(sd_event_source *s, usec_t usec, void *userdata) {

Manager *m = userdata;

assert(s);

assert(m);

m->bus_retry_event_source = sd_event_source_unref(m->bus_retry_event_source);

manager_connect_bus(m);

return 0;

}

int manager_connect_bus(Manager *m) {

int r;

assert(m);

if (m->bus)

return 0;

r = sd_bus_default_system(&m->bus);

if (r < 0) {

/* We failed to connect? Yuck, we must be in early

log_debug("Failed to connect to bus, trying again in 5s: %s", strerror(-r));

r = sd_event_add_time(m->event, &m->bus_retry_event_source, CLOCK_MONOTONIC, now(CLOCK_MONOTONIC) + 5*USEC_PER_SEC, 0, on_bus_retry, m);

if (r < 0) {

log_error("Failed to install bus reconnect time event: %s", strerror(-r));

return r;

}

return 0;

}

r = sd_bus_add_object_vtable(m->bus, NULL, "/org/freedesktop/resolve1", "org.freedesktop.resolve1.Manager", resolve_vtable, m);

if (r < 0) {

log_error("Failed to register object: %s", strerror(-r));

return r;

}

r = sd_bus_request_name(m->bus, "org.freedesktop.resolve1", 0);

if (r < 0) {

log_error("Failed to register name: %s", strerror(-r));

return r;

}

r = sd_bus_attach_event(m->bus, m->event, 0);

if (r < 0) {

log_error("Failed to attach bus to event loop: %s", strerror(-r));

return r;

}

return 0;

}