src/resolve/resolved-dnssd-bus.c

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