2 Copyright © 2017 Dmitry Rozhkov
5 #include "conf-files.h"
6 #include "conf-parser.h"
7 #include "resolved-dnssd.h"
8 #include "resolved-dns-rr.h"
9 #include "resolved-manager.h"
10 #include "specifier.h"
13 const char* const dnssd_service_dirs
[] = {
16 "/usr/lib/systemd/dnssd",
23 DnssdTxtData
*dnssd_txtdata_free(DnssdTxtData
*txt_data
) {
27 dns_resource_record_unref(txt_data
->rr
);
28 dns_txt_item_free_all(txt_data
->txt
);
30 return mfree(txt_data
);
33 DnssdTxtData
*dnssd_txtdata_free_all(DnssdTxtData
*txt_data
) {
39 next
= txt_data
->items_next
;
41 dnssd_txtdata_free(txt_data
);
43 return dnssd_txtdata_free_all(next
);
46 DnssdService
*dnssd_service_free(DnssdService
*service
) {
51 hashmap_remove(service
->manager
->dnssd_services
, service
->name
);
53 dns_resource_record_unref(service
->ptr_rr
);
54 dns_resource_record_unref(service
->srv_rr
);
56 dnssd_txtdata_free_all(service
->txt_data_items
);
58 free(service
->filename
);
61 free(service
->name_template
);
63 return mfree(service
);
66 static int dnssd_service_load(Manager
*manager
, const char *filename
) {
67 _cleanup_(dnssd_service_freep
) DnssdService
*service
= NULL
;
68 _cleanup_(dnssd_txtdata_freep
) DnssdTxtData
*txt_data
= NULL
;
70 const char *dropin_dirname
;
76 service
= new0(DnssdService
, 1);
80 service
->filename
= strdup(filename
);
81 if (!service
->filename
)
84 service
->name
= strdup(basename(filename
));
88 d
= endswith(service
->name
, ".dnssd");
92 assert(streq(d
, ".dnssd"));
96 dropin_dirname
= strjoina(service
->name
, ".dnssd.d");
98 r
= config_parse_many(filename
, dnssd_service_dirs
, dropin_dirname
,
100 config_item_perf_lookup
, resolved_dnssd_gperf_lookup
,
105 if (!service
->name_template
) {
106 log_error("%s doesn't define service instance name", service
->name
);
110 if (!service
->type
) {
111 log_error("%s doesn't define service type", service
->name
);
115 if (LIST_IS_EMPTY(service
->txt_data_items
)) {
116 txt_data
= new0(DnssdTxtData
, 1);
120 r
= dns_txt_item_new_empty(&txt_data
->txt
);
124 LIST_PREPEND(items
, service
->txt_data_items
, txt_data
);
128 r
= hashmap_ensure_allocated(&manager
->dnssd_services
, &string_hash_ops
);
132 r
= hashmap_put(manager
->dnssd_services
, service
->name
, service
);
136 service
->manager
= manager
;
138 r
= dnssd_update_rrs(service
);
147 static int specifier_dnssd_host_name(char specifier
, void *data
, void *userdata
, char **ret
) {
148 DnssdService
*s
= (DnssdService
*) userdata
;
153 assert(s
->manager
->llmnr_hostname
);
155 n
= strdup(s
->manager
->llmnr_hostname
);
163 int dnssd_render_instance_name(DnssdService
*s
, char **ret_name
) {
164 static const Specifier specifier_table
[] = {
165 { 'b', specifier_boot_id
, NULL
},
166 { 'H', specifier_dnssd_host_name
, NULL
},
167 { 'm', specifier_machine_id
, NULL
},
168 { 'v', specifier_kernel_release
, NULL
},
171 _cleanup_free_
char *name
= NULL
;
175 assert(s
->name_template
);
177 r
= specifier_printf(s
->name_template
, specifier_table
, s
, &name
);
179 return log_debug_errno(r
, "Failed to replace specifiers: %m");
181 if (!dns_service_name_is_valid(name
)) {
182 log_debug("Service instance name '%s' is invalid.", name
);
186 *ret_name
= TAKE_PTR(name
);
191 int dnssd_load(Manager
*manager
) {
192 _cleanup_strv_free_
char **files
= NULL
;
198 if (manager
->mdns_support
!= RESOLVE_SUPPORT_YES
)
201 r
= conf_files_list_strv(&files
, ".dnssd", NULL
, 0, dnssd_service_dirs
);
203 return log_error_errno(r
, "Failed to enumerate .dnssd files: %m");
205 STRV_FOREACH_BACKWARDS(f
, files
) {
206 r
= dnssd_service_load(manager
, *f
);
208 log_warning_errno(r
, "Failed to load '%s': %m", *f
);;
214 int dnssd_update_rrs(DnssdService
*s
) {
215 _cleanup_free_
char *n
= NULL
;
216 _cleanup_free_
char *service_name
= NULL
;
217 _cleanup_free_
char *full_name
= NULL
;
218 DnssdTxtData
*txt_data
;
222 assert(s
->txt_data_items
);
225 s
->ptr_rr
= dns_resource_record_unref(s
->ptr_rr
);
226 s
->srv_rr
= dns_resource_record_unref(s
->srv_rr
);
227 LIST_FOREACH(items
, txt_data
, s
->txt_data_items
)
228 txt_data
->rr
= dns_resource_record_unref(txt_data
->rr
);
230 r
= dnssd_render_instance_name(s
, &n
);
234 r
= dns_name_concat(s
->type
, "local", &service_name
);
237 r
= dns_name_concat(n
, service_name
, &full_name
);
241 LIST_FOREACH(items
, txt_data
, s
->txt_data_items
) {
242 txt_data
->rr
= dns_resource_record_new_full(DNS_CLASS_IN
, DNS_TYPE_TXT
,
247 txt_data
->rr
->ttl
= MDNS_DEFAULT_TTL
;
248 txt_data
->rr
->txt
.items
= dns_txt_item_copy(txt_data
->txt
);
249 if (!txt_data
->rr
->txt
.items
)
253 s
->ptr_rr
= dns_resource_record_new_full(DNS_CLASS_IN
, DNS_TYPE_PTR
,
258 s
->ptr_rr
->ttl
= MDNS_DEFAULT_TTL
;
259 s
->ptr_rr
->ptr
.name
= strdup(full_name
);
260 if (!s
->ptr_rr
->ptr
.name
)
263 s
->srv_rr
= dns_resource_record_new_full(DNS_CLASS_IN
, DNS_TYPE_SRV
,
268 s
->srv_rr
->ttl
= MDNS_DEFAULT_TTL
;
269 s
->srv_rr
->srv
.priority
= s
->priority
;
270 s
->srv_rr
->srv
.weight
= s
->weight
;
271 s
->srv_rr
->srv
.port
= s
->port
;
272 s
->srv_rr
->srv
.name
= strdup(s
->manager
->mdns_hostname
);
273 if (!s
->srv_rr
->srv
.name
)
279 LIST_FOREACH(items
, txt_data
, s
->txt_data_items
)
280 txt_data
->rr
= dns_resource_record_unref(txt_data
->rr
);
281 s
->ptr_rr
= dns_resource_record_unref(s
->ptr_rr
);
282 s
->srv_rr
= dns_resource_record_unref(s
->srv_rr
);
286 int dnssd_txt_item_new_from_string(const char *key
, const char *value
, DnsTxtItem
**ret_item
) {
290 length
= strlen(key
);
293 length
+= strlen(value
) + 1; /* length of value plus '=' */
295 i
= malloc0(offsetof(DnsTxtItem
, data
) + length
+ 1); /* for safety reasons we add an extra NUL byte */
299 memcpy(i
->data
, key
, strlen(key
));
300 if (!isempty(value
)) {
301 memcpy(i
->data
+ strlen(key
), "=", 1);
302 memcpy(i
->data
+ strlen(key
) + 1, value
, strlen(value
));
306 *ret_item
= TAKE_PTR(i
);
311 int dnssd_txt_item_new_from_data(const char *key
, const void *data
, const size_t size
, DnsTxtItem
**ret_item
) {
315 length
= strlen(key
);
318 length
+= size
+ 1; /* size of date plus '=' */
320 i
= malloc0(offsetof(DnsTxtItem
, data
) + length
+ 1); /* for safety reasons we add an extra NUL byte */
324 memcpy(i
->data
, key
, strlen(key
));
326 memcpy(i
->data
+ strlen(key
), "=", 1);
327 memcpy(i
->data
+ strlen(key
) + 1, data
, size
);
331 *ret_item
= TAKE_PTR(i
);
336 void dnssd_signal_conflict(Manager
*manager
, const char *name
) {
341 HASHMAP_FOREACH(s
, manager
->dnssd_services
, i
) {
345 if (dns_name_equal(dns_resource_key_name(s
->srv_rr
->key
), name
)) {
346 _cleanup_free_
char *path
= NULL
;
350 r
= sd_bus_path_encode("/org/freedesktop/resolve1/dnssd", s
->name
, &path
);
352 log_error_errno(r
, "Can't get D-BUS object path: %m");
356 r
= sd_bus_emit_signal(manager
->bus
,
358 "org.freedesktop.resolve1.DnssdService",
362 log_error_errno(r
, "Cannot emit signal: %m");