X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fresolve-host%2Fresolve-host.c;h=4b46cdfa16cd7ff312be7863a8d456d7d66688fd;hp=72ab24c70eb9dbb33dbefb7c37e54d2167bf0286;hb=de292aa1dd1942e151ff034fcb88504a86742f97;hpb=02dd6e189a6d2b7f3884ad4cdb3d8c85e009c565 diff --git a/src/resolve-host/resolve-host.c b/src/resolve-host/resolve-host.c index 72ab24c70..4b46cdfa1 100644 --- a/src/resolve-host/resolve-host.c +++ b/src/resolve-host/resolve-host.c @@ -31,22 +31,27 @@ #include "af-list.h" #include "build.h" +#include "resolved-dns-packet.h" + #define DNS_CALL_TIMEOUT_USEC (45*USEC_PER_SEC) static int arg_family = AF_UNSPEC; static int arg_ifindex = 0; +static int arg_type = 0; +static uint16_t arg_class = 0; +static bool arg_legend = true; -static int resolve_host(sd_bus *bus, const char *name, int _family, int _ifindex) { +static int resolve_host(sd_bus *bus, const char *name) { _cleanup_bus_message_unref_ sd_bus_message *req = NULL, *reply = NULL; _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; + const char *canonical = NULL; unsigned c = 0; int r; assert(name); - log_debug("Resolving %s (family %s)", - name, af_to_name(_family)); + log_debug("Resolving %s (family %s, ifindex %i).", name, af_to_name(arg_family) ?: "*", arg_ifindex); r = sd_bus_message_new_method_call( bus, @@ -62,7 +67,7 @@ static int resolve_host(sd_bus *bus, const char *name, int _family, int _ifindex if (r < 0) return bus_log_create_error(r); - r = sd_bus_message_append(req, "si", name, AF_UNSPEC); + r = sd_bus_message_append(req, "si", name, arg_family); if (r < 0) return bus_log_create_error(r); @@ -99,10 +104,8 @@ static int resolve_host(sd_bus *bus, const char *name, int _family, int _ifindex if (r < 0) return bus_log_parse_error(r); - if ((_family != AF_UNSPEC && family != _family) || - !IN_SET(family, AF_INET, AF_INET6)) { - log_debug("%s: skipping entry with family %hu (%s)", - name, family, af_to_name(family) ?: "unknown"); + if (!IN_SET(family, AF_INET, AF_INET6)) { + log_debug("%s: skipping entry with family %d (%s)", name, family, af_to_name(family) ?: "unknown"); continue; } @@ -128,7 +131,7 @@ static int resolve_host(sd_bus *bus, const char *name, int _family, int _ifindex } } - if (_ifindex > 0 && ifindex > 0 && ifindex != _ifindex) { + if (arg_ifindex > 0 && ifindex > 0 && ifindex != arg_ifindex) { log_debug("%s: skipping entry with ifindex %i (%s)", name, ifindex, ifname); continue; @@ -140,56 +143,313 @@ static int resolve_host(sd_bus *bus, const char *name, int _family, int _ifindex continue; } - log_info("%*s%s %s%s%.*s", - (int) strlen(name), c == 0 ? name : "", c == 0 ? ":" : " ", - pretty, - *ifname ? "%" : "", (int) sizeof(ifname), *ifname ? ifname: ""); + printf("%*s%s %s%s%s\n", + (int) strlen(name), c == 0 ? name : "", c == 0 ? ":" : " ", + pretty, + isempty(ifname) ? "" : "%", ifname); c++; } + if (r < 0) + return bus_log_parse_error(r); + + r = sd_bus_message_exit_container(reply); + if (r < 0) + return bus_log_parse_error(r); + + r = sd_bus_message_read(reply, "s", &canonical); + if (r < 0) + return bus_log_parse_error(r); + + if (!streq(name, canonical)) { + printf("%*s%s (%s)\n", + (int) strlen(name), c == 0 ? name : "", c == 0 ? ":" : " ", + canonical); + } if (c == 0) { log_error("%s: no addresses found", name); - return -ENONET; + return -ESRCH; } + return 0; +} + +static int resolve_address(sd_bus *bus, int family, const union in_addr_union *address, int ifindex) { + _cleanup_bus_message_unref_ sd_bus_message *req = NULL, *reply = NULL; + _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; + _cleanup_free_ char *pretty = NULL; + char ifname[IF_NAMESIZE] = ""; + unsigned c = 0; + const char *n; + int r; + + assert(bus); + assert(IN_SET(family, AF_INET, AF_INET6)); + assert(address); + + r = in_addr_to_string(family, address, &pretty); + if (r < 0) + return log_oom(); + + if (ifindex > 0) { + char *t; + + t = if_indextoname(ifindex, ifname); + if (!t) { + log_error("Failed to resolve interface name for index %i", ifindex); + return -errno; + } + } + + log_debug("Resolving %s%s%s.", pretty, isempty(ifname) ? "" : "%", ifname); + + r = sd_bus_message_new_method_call( + bus, + &req, + "org.freedesktop.resolve1", + "/org/freedesktop/resolve1", + "org.freedesktop.resolve1.Manager", + "ResolveAddress"); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_message_set_auto_start(req, false); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_message_append(req, "i", family); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_message_append_array(req, 'y', address, FAMILY_ADDRESS_SIZE(family)); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_message_append(req, "i", ifindex); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_call(bus, req, DNS_CALL_TIMEOUT_USEC, &error, &reply); + if (r < 0) { + log_error("%s: resolve call failed: %s", pretty, bus_error_message(&error, r)); + return r; + } + + r = sd_bus_message_enter_container(reply, 'a', "s"); + if (r < 0) + return bus_log_create_error(r); + + while ((r = sd_bus_message_read(reply, "s", &n)) > 0) { + + printf("%*s%s%s%s %s\n", + (int) strlen(pretty), c == 0 ? pretty : "", + isempty(ifname) ? "" : "%", ifname, + c == 0 ? ":" : " ", + n); + + c++; + } + if (r < 0) + return bus_log_parse_error(r); + r = sd_bus_message_exit_container(reply); if (r < 0) return bus_log_parse_error(r); + if (c == 0) { + log_error("%s: no names found", pretty); + return -ESRCH; + } + + return 0; +} + +static int parse_address(const char *s, int *family, union in_addr_union *address, int *ifindex) { + const char *percent, *a; + int ifi = 0; + int r; + + percent = strchr(s, '%'); + if (percent) { + r = safe_atoi(percent+1, &ifi); + if (r < 0 || ifi <= 0) { + ifi = if_nametoindex(percent+1); + if (ifi <= 0) + return -EINVAL; + } + + a = strndupa(s, percent - s); + } else + a = s; + + r = in_addr_from_string_auto(a, family, address); + if (r < 0) + return r; + + *ifindex = ifi; return 0; } +static int resolve_record(sd_bus *bus, const char *name) { + + _cleanup_bus_message_unref_ sd_bus_message *req = NULL, *reply = NULL; + _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; + unsigned n = 0; + int r; + + assert(name); + + log_debug("Resolving %s %s %s.", name, dns_class_to_string(arg_class), dns_type_to_string(arg_type)); + + r = sd_bus_message_new_method_call( + bus, + &req, + "org.freedesktop.resolve1", + "/org/freedesktop/resolve1", + "org.freedesktop.resolve1.Manager", + "ResolveRecord"); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_message_set_auto_start(req, false); + if (r < 0) + return bus_log_create_error(r); + + assert((uint16_t) arg_type == arg_type); + r = sd_bus_message_append(req, "sqq", name, arg_class, arg_type); + if (r < 0) + return bus_log_create_error(r); + + r = sd_bus_call(bus, req, DNS_CALL_TIMEOUT_USEC, &error, &reply); + if (r < 0) { + log_error("%s: resolve call failed: %s", name, bus_error_message(&error, r)); + return r; + } + + r = sd_bus_message_enter_container(reply, 'a', "(qqay)"); + if (r < 0) + return bus_log_parse_error(r); + + while ((r = sd_bus_message_enter_container(reply, 'r', "qqay")) > 0) { + _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr = NULL; + _cleanup_(dns_packet_unrefp) DnsPacket *p = NULL; + _cleanup_free_ char *s = NULL; + uint16_t c, t; + const void *d; + size_t l; + + r = sd_bus_message_read(reply, "qq", &c, &t); + if (r < 0) + return bus_log_parse_error(r); + + r = sd_bus_message_read_array(reply, 'y', &d, &l); + if (r < 0) + return bus_log_parse_error(r); + + r = sd_bus_message_exit_container(reply); + if (r < 0) + return bus_log_parse_error(r); + + r = dns_packet_new(&p, DNS_PROTOCOL_DNS, 0); + if (r < 0) + return log_oom(); + + r = dns_packet_append_blob(p, d, l, NULL); + if (r < 0) + return log_oom(); + + r = dns_packet_read_rr(p, &rr, NULL); + if (r < 0) { + log_error("Failed to parse RR."); + return r; + } + + r = dns_resource_record_to_string(rr, &s); + if (r < 0) { + log_error("Failed to format RR."); + return r; + } + + printf("%s\n", s); + n++; + } + if (r < 0) + return bus_log_parse_error(r); + + r = sd_bus_message_exit_container(reply); + if (r < 0) + return bus_log_parse_error(r); + + if (n == 0) { + log_error("%s: no records found", name); + return -ESRCH; + } + + return 0; +} + +static void help_dns_types(void) { + int i; + const char *t; + + if (arg_legend) + puts("Known dns types:"); + for (i = 0; i < _DNS_TYPE_MAX; i++) { + t = dns_type_to_string(i); + if (t) + puts(t); + } +} + +static void help_dns_classes(void) { + int i; + const char *t; + + if (arg_legend) + puts("Known dns classes:"); + for (i = 0; i < _DNS_CLASS_MAX; i++) { + t = dns_class_to_string(i); + if (t) + puts(t); + } +} + static void help(void) { printf("%s [OPTIONS...]\n\n" "Resolve IPv4 or IPv6 addresses.\n\n" - "Options:\n" - " -4 Resolve IPv4 addresses\n" - " -6 Resolve IPv6 addresses\n" - " -i INTERFACE Filter by interface\n" - " -h --help Show this help and exit\n" - " --version Print version string and exit\n" - , program_invocation_short_name - ); + " -h --help Show this help\n" + " --version Show package version\n" + " -4 Resolve IPv4 addresses\n" + " -6 Resolve IPv6 addresses\n" + " -i INTERFACE Filter by interface\n" + " -t --type=TYPE Query RR with DNS type\n" + " -c --class=CLASS Query RR with DNS class\n" + " --no-legend Do not print column headers\n" + , program_invocation_short_name); } static int parse_argv(int argc, char *argv[]) { enum { ARG_VERSION = 0x100, + ARG_NO_LEGEND, }; static const struct option options[] = { { "help", no_argument, NULL, 'h' }, { "version", no_argument, NULL, ARG_VERSION }, + { "type", no_argument, NULL, 't' }, + { "class", no_argument, NULL, 'c' }, + { "no-legend", no_argument, NULL, ARG_NO_LEGEND }, {} }; - int c; + int c, r; assert(argc >= 0); assert(argv); - while ((c = getopt_long(argc, argv, "h46i:", options, NULL)) >= 0) + while ((c = getopt_long(argc, argv, "h46i:t:c:", options, NULL)) >= 0) switch(c) { case 'h': @@ -213,8 +473,41 @@ static int parse_argv(int argc, char *argv[]) { arg_ifindex = if_nametoindex(optarg); if (arg_ifindex <= 0) { log_error("Unknown interfaces %s: %m", optarg); - return -EINVAL; + return -errno; + } + break; + + case 't': + if (streq(optarg, "help")) { + help_dns_types(); + return 0; + } + + arg_type = dns_type_from_string(optarg); + if (arg_type < 0) { + log_error("Failed to parse RR record type %s", optarg); + return r; + } + assert(arg_type > 0 && (uint16_t) arg_type == arg_type); + + break; + + case 'c': + if (streq(optarg, "help")) { + help_dns_classes(); + return 0; + } + + r = dns_class_from_string(optarg, &arg_class); + if (r < 0) { + log_error("Failed to parse RR record class %s", optarg); + return r; } + + break; + + case ARG_NO_LEGEND: + arg_legend = false; break; case '?': @@ -224,6 +517,14 @@ static int parse_argv(int argc, char *argv[]) { assert_not_reached("Unhandled option"); } + if (arg_type == 0 && arg_class != 0) { + log_error("--class= may only be used in conjunction with --type="); + return -EINVAL; + } + + if (arg_type != 0 && arg_class == 0) + arg_class = DNS_CLASS_IN; + return 1 /* work to do */; } @@ -236,22 +537,40 @@ int main(int argc, char **argv) { r = parse_argv(argc, argv); if (r <= 0) - goto end; + goto finish; + + if (optind >= argc) { + log_error("No arguments passed"); + r = -EINVAL; + goto finish; + } r = sd_bus_open_system(&bus); if (r < 0) { log_error("sd_bus_open_system: %s", strerror(-r)); - goto end; + goto finish; } while (argv[optind]) { - int k; + int family, ifindex, k; + union in_addr_union a; + + if (arg_type != 0) + k = resolve_record(bus, argv[optind]); + else { + k = parse_address(argv[optind], &family, &a, &ifindex); + if (k >= 0) + k = resolve_address(bus, family, &a, ifindex); + else + k = resolve_host(bus, argv[optind]); + } - k = resolve_host(bus, argv[optind++], arg_family, arg_ifindex); if (r == 0) r = k; + + optind++; } - end: +finish: return r == 0 ? EXIT_SUCCESS : EXIT_FAILURE; }