src/resolve/resolved-dnssd-bus.c

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