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