src/resolve/resolved-dnssd-bus.c

   1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
   2
   3 #include "alloc-util.h"
   4 #include "bus-polkit.h"
   5 #include "missing_capability.h"
   6 #include "resolved-dnssd-bus.h"
   7 #include "resolved-dnssd.h"
   8 #include "resolved-link.h"
   9 #include "resolved-manager.h"
  10 #include "strv.h"
  11 #include "user-util.h"
  12
  13 int bus_dnssd_method_unregister(sd_bus_message *message, void *userdata, sd_bus_error *error) {
  14         DnssdService *s = ASSERT_PTR(userdata);
  15         Manager *m;
  16         Link *l;
  17         int r;
  18
  19         assert(message);
  20
  21         m = s->manager;
  22
  23         r = bus_verify_polkit_async_full(
  24                         message,
  25                         "org.freedesktop.resolve1.unregister-service",
  26                         /* details= */ NULL,
  27                         /* interactive= */ false,
  28                         /* good_user= */ s->originator,
  29                         &m->polkit_registry,
  30                         error);
  31         if (r < 0)
  32                 return r;
  33         if (r == 0)
  34                 return 1; /* Polkit will call us back */
  35
  36         HASHMAP_FOREACH(l, m->links) {
  37                 if (l->mdns_ipv4_scope) {
  38                         r = dns_scope_announce(l->mdns_ipv4_scope, true);
  39                         if (r < 0)
  40                                 log_warning_errno(r, "Failed to send goodbye messages in IPv4 scope: %m");
  41
  42                         dns_zone_remove_rr(&l->mdns_ipv4_scope->zone, s->ptr_rr);
  43                         dns_zone_remove_rr(&l->mdns_ipv4_scope->zone, s->srv_rr);
  44                         LIST_FOREACH(items, txt_data, s->txt_data_items)
  45                                 dns_zone_remove_rr(&l->mdns_ipv4_scope->zone, txt_data->rr);
  46                 }
  47
  48                 if (l->mdns_ipv6_scope) {
  49                         r = dns_scope_announce(l->mdns_ipv6_scope, true);
  50                         if (r < 0)
  51                                 log_warning_errno(r, "Failed to send goodbye messages in IPv6 scope: %m");
  52
  53                         dns_zone_remove_rr(&l->mdns_ipv6_scope->zone, s->ptr_rr);
  54                         dns_zone_remove_rr(&l->mdns_ipv6_scope->zone, s->srv_rr);
  55                         LIST_FOREACH(items, txt_data, s->txt_data_items)
  56                                 dns_zone_remove_rr(&l->mdns_ipv6_scope->zone, txt_data->rr);
  57                 }
  58         }
  59
  60         dnssd_service_free(s);
  61
  62         manager_refresh_rrs(m);
  63
  64         return sd_bus_reply_method_return(message, NULL);
  65 }
  66
  67 static int dnssd_object_find(sd_bus *bus, const char *path, const char *interface, void *userdata, void **found, sd_bus_error *error) {
  68         _cleanup_free_ char *name = NULL;
  69         Manager *m = ASSERT_PTR(userdata);
  70         DnssdService *service;
  71         int r;
  72
  73         assert(bus);
  74         assert(path);
  75         assert(interface);
  76         assert(found);
  77
  78         r = sd_bus_path_decode(path, "/org/freedesktop/resolve1/dnssd", &name);
  79         if (r <= 0)
  80                 return 0;
  81
  82         service = hashmap_get(m->dnssd_services, name);
  83         if (!service)
  84                 return 0;
  85
  86         *found = service;
  87         return 1;
  88 }
  89
  90 static int dnssd_node_enumerator(sd_bus *bus, const char *path, void *userdata, char ***nodes, sd_bus_error *error) {
  91         _cleanup_strv_free_ char **l = NULL;
  92         Manager *m = ASSERT_PTR(userdata);
  93         DnssdService *service;
  94         unsigned c = 0;
  95         int r;
  96
  97         assert(bus);
  98         assert(path);
  99         assert(nodes);
 100
 101         l = new0(char*, hashmap_size(m->dnssd_services) + 1);
 102         if (!l)
 103                 return -ENOMEM;
 104
 105         HASHMAP_FOREACH(service, m->dnssd_services) {
 106                 char *p;
 107
 108                 r = sd_bus_path_encode("/org/freedesktop/resolve1/dnssd", service->name, &p);
 109                 if (r < 0)
 110                         return r;
 111
 112                 l[c++] = p;
 113         }
 114
 115         l[c] = NULL;
 116         *nodes = TAKE_PTR(l);
 117
 118         return 1;
 119 }
 120
 121 static const sd_bus_vtable dnssd_vtable[] = {
 122         SD_BUS_VTABLE_START(0),
 123
 124         SD_BUS_METHOD("Unregister", NULL, NULL, bus_dnssd_method_unregister, SD_BUS_VTABLE_UNPRIVILEGED),
 125         SD_BUS_SIGNAL("Conflicted", NULL, 0),
 126
 127         SD_BUS_VTABLE_END
 128 };
 129
 130 const BusObjectImplementation dnssd_object = {
 131         "/org/freedesktop/resolve1/dnssd",
 132         "org.freedesktop.resolve1.DnssdService",
 133         .fallback_vtables = BUS_FALLBACK_VTABLES({dnssd_vtable, dnssd_object_find}),
 134         .node_enumerator = dnssd_node_enumerator,
 135 };