src/resolve/resolved-dnssd-bus.c

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