X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=avahi-daemon%2Fstatic-services.c;h=4c3491f025fdf7b9436328f632ca48753144e22d;hb=7a5b2f69af7d36d6cd4153142f125fa011784e03;hp=4e97299d99329ad853d8b68723bb1d5d38692fa7;hpb=219241ab5223b705ade51485ea9227b0a6089c0d;p=catta diff --git a/avahi-daemon/static-services.c b/avahi-daemon/static-services.c index 4e97299..4c3491f 100644 --- a/avahi-daemon/static-services.c +++ b/avahi-daemon/static-services.c @@ -1,18 +1,16 @@ -/* $Id$ */ - /*** This file is part of avahi. - + avahi is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. - + avahi is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. - + You should have received a copy of the GNU Lesser General Public License along with avahi; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 @@ -30,69 +28,84 @@ #include #include #include +#include -#include +#ifdef USE_EXPAT_H #include +#endif /* USE_EXPAT_H */ -#include +#ifdef USE_BSDXML_H +#include +#endif /* USE_BSDXML_H */ + +#include +#include +#include +#include +#include #include +#include #include "main.h" +#include "static-services.h" typedef struct StaticService StaticService; typedef struct StaticServiceGroup StaticServiceGroup; struct StaticService { StaticServiceGroup *group; - - gchar *type; - gchar *domain_name; - gchar *host_name; - guint16 port; + + char *type; + char *domain_name; + char *host_name; + uint16_t port; + int protocol; + + AvahiStringList *subtypes; AvahiStringList *txt_records; - + AVAHI_LLIST_FIELDS(StaticService, services); }; struct StaticServiceGroup { - gchar *filename; + char *filename; time_t mtime; - gchar *name, *chosen_name; - gboolean replace_wildcards; + char *name, *chosen_name; + int replace_wildcards; - AvahiEntryGroup *entry_group; + AvahiSEntryGroup *entry_group; AVAHI_LLIST_HEAD(StaticService, services); AVAHI_LLIST_FIELDS(StaticServiceGroup, groups); }; static AVAHI_LLIST_HEAD(StaticServiceGroup, groups) = NULL; -static gchar *replacestr(const gchar *pattern, const gchar *a, const gchar *b) { - gchar *r = NULL, *e, *n; +static char *replacestr(const char *pattern, const char *a, const char *b) { + char *r = NULL, *e, *n; while ((e = strstr(pattern, a))) { - gchar *k; + char *k; - k = g_strndup(pattern, e - pattern); + k = avahi_strndup(pattern, e - pattern); if (r) - n = g_strconcat(r, k, b, NULL); + n = avahi_strdup_printf("%s%s%s", r, k, b); else - n = g_strconcat(k, b, NULL); + n = avahi_strdup_printf("%s%s", k, b); - g_free(k); - g_free(r); + avahi_free(k); + avahi_free(r); r = n; pattern = e + strlen(a); } if (!r) - return g_strdup(pattern); + return avahi_strdup(pattern); - n = g_strconcat(r, pattern, NULL); - g_free(r); + n = avahi_strdup_printf("%s%s", r, pattern); + avahi_free(r); return n; } @@ -102,30 +115,32 @@ static void remove_static_service_group_from_server(StaticServiceGroup *g); static StaticService *static_service_new(StaticServiceGroup *group) { StaticService *s; - - g_assert(group); - s = g_new(StaticService, 1); + + assert(group); + s = avahi_new(StaticService, 1); s->group = group; s->type = s->host_name = s->domain_name = NULL; s->port = 0; + s->protocol = AVAHI_PROTO_UNSPEC; s->txt_records = NULL; + s->subtypes = NULL; AVAHI_LLIST_PREPEND(StaticService, services, group->services, s); return s; } -static StaticServiceGroup *static_service_group_new(gchar *filename) { +static StaticServiceGroup *static_service_group_new(char *filename) { StaticServiceGroup *g; - g_assert(filename); + assert(filename); - g = g_new(StaticServiceGroup, 1); - g->filename = g_strdup(filename); + g = avahi_new(StaticServiceGroup, 1); + g->filename = avahi_strdup(filename); g->mtime = 0; g->name = g->chosen_name = NULL; - g->replace_wildcards = FALSE; + g->replace_wildcards = 0; g->entry_group = NULL; AVAHI_LLIST_HEAD_INIT(StaticService, g->services); @@ -135,100 +150,149 @@ static StaticServiceGroup *static_service_group_new(gchar *filename) { } static void static_service_free(StaticService *s) { - g_assert(s); - + assert(s); + AVAHI_LLIST_REMOVE(StaticService, services, s->group->services, s); - g_free(s->type); - g_free(s->host_name); - g_free(s->domain_name); + avahi_free(s->type); + avahi_free(s->host_name); + avahi_free(s->domain_name); avahi_string_list_free(s->txt_records); - - g_free(s); + avahi_string_list_free(s->subtypes); + + avahi_free(s); } static void static_service_group_free(StaticServiceGroup *g) { - g_assert(g); + assert(g); - remove_static_service_group_from_server(g); + if (g->entry_group) + avahi_s_entry_group_free(g->entry_group); while (g->services) static_service_free(g->services); AVAHI_LLIST_REMOVE(StaticServiceGroup, groups, groups, g); - g_free(g->filename); - g_free(g->name); - g_free(g->chosen_name); - g_free(g); + avahi_free(g->filename); + avahi_free(g->name); + avahi_free(g->chosen_name); + avahi_free(g); } -static void entry_group_callback(AvahiServer *s, AvahiEntryGroup *eg, AvahiEntryGroupState state, gpointer userdata) { +static void entry_group_callback(AvahiServer *s, AVAHI_GCC_UNUSED AvahiSEntryGroup *eg, AvahiEntryGroupState state, void* userdata) { StaticServiceGroup *g = userdata; - - g_assert(s); - g_assert(g); - - if (state == AVAHI_ENTRY_GROUP_COLLISION) { - gchar *n; - remove_static_service_group_from_server(g); + assert(s); + assert(g); - n = avahi_alternative_service_name(g->chosen_name); - g_free(g->chosen_name); - g->chosen_name = n; + switch (state) { - avahi_log_notice("Service name conflict for \"%s\" (%s), retrying with \"%s\".", g->name, g->filename, g->chosen_name); + case AVAHI_ENTRY_GROUP_COLLISION: { + char *n; - add_static_service_group_to_server(g); - } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED) - avahi_log_info("Service \"%s\" (%s) successfully established.", g->chosen_name, g->filename); + remove_static_service_group_from_server(g); + + n = avahi_alternative_service_name(g->chosen_name); + avahi_free(g->chosen_name); + g->chosen_name = n; + + avahi_log_notice("Service name conflict for \"%s\" (%s), retrying with \"%s\".", g->name, g->filename, g->chosen_name); + + add_static_service_group_to_server(g); + break; + } + + case AVAHI_ENTRY_GROUP_ESTABLISHED: + avahi_log_info("Service \"%s\" (%s) successfully established.", g->chosen_name, g->filename); + break; + + case AVAHI_ENTRY_GROUP_FAILURE: + avahi_log_warn("Failed to publish service \"%s\" (%s): %s", g->chosen_name, g->filename, avahi_strerror(avahi_server_errno(s))); + remove_static_service_group_from_server(g); + break; + + case AVAHI_ENTRY_GROUP_UNCOMMITED: + case AVAHI_ENTRY_GROUP_REGISTERING: + ; + } } static void add_static_service_group_to_server(StaticServiceGroup *g) { StaticService *s; - g_assert(g); - - if (g->entry_group) + assert(g); + + if (g->entry_group && !avahi_s_entry_group_is_empty(g->entry_group)) + /* This service group is already registered in the server */ return; - - if (g->chosen_name) - g_free(g->chosen_name); - - if (g->replace_wildcards) - g->chosen_name = replacestr(g->name, "%h", avahi_server_get_host_name(avahi_server)); - else - g->chosen_name = g_strdup(g->name); - - g->entry_group = avahi_entry_group_new(avahi_server, entry_group_callback, g); - + + if (!g->chosen_name || (g->replace_wildcards && strstr(g->name, "%h"))) { + + avahi_free(g->chosen_name); + + if (g->replace_wildcards) { + char label[AVAHI_LABEL_MAX]; + const char *p; + + p = avahi_server_get_host_name(avahi_server); + avahi_unescape_label(&p, label, sizeof(label)); + + g->chosen_name = replacestr(g->name, "%h", label); + } else + g->chosen_name = avahi_strdup(g->name); + + } + + if (!g->entry_group) + g->entry_group = avahi_s_entry_group_new(avahi_server, entry_group_callback, g); + + assert(avahi_s_entry_group_is_empty(g->entry_group)); + for (s = g->services; s; s = s->services_next) { + AvahiStringList *i; if (avahi_server_add_service_strlst( avahi_server, g->entry_group, - -1, AF_UNSPEC, - s->type, g->chosen_name, - s->domain_name, s->host_name, s->port, - avahi_string_list_copy(s->txt_records)) < 0) { - avahi_log_error("Failed to add service '%s' of type '%s', ignoring service group (%s)", g->chosen_name, s->type, g->filename); + AVAHI_IF_UNSPEC, s->protocol, + 0, + g->chosen_name, s->type, s->domain_name, + s->host_name, s->port, + s->txt_records) < 0) { + avahi_log_error("Failed to add service '%s' of type '%s', ignoring service group (%s): %s", + g->chosen_name, s->type, g->filename, + avahi_strerror(avahi_server_errno(avahi_server))); remove_static_service_group_from_server(g); return; } + + for (i = s->subtypes; i; i = i->next) { + + if (avahi_server_add_service_subtype( + avahi_server, + g->entry_group, + AVAHI_IF_UNSPEC, s->protocol, + 0, + g->chosen_name, s->type, s->domain_name, + (char*) i->text) < 0) { + + avahi_log_error("Failed to add subtype '%s' for service '%s' of type '%s', ignoring subtype (%s): %s", + i->text, g->chosen_name, s->type, g->filename, + avahi_strerror(avahi_server_errno(avahi_server))); + } + } } - avahi_entry_group_commit(g->entry_group); + avahi_s_entry_group_commit(g->entry_group); } static void remove_static_service_group_from_server(StaticServiceGroup *g) { - g_assert(g); + assert(g); - if (g->entry_group) { - avahi_entry_group_free(g->entry_group); - g->entry_group = NULL; - } + if (g->entry_group) + avahi_s_entry_group_reset(g->entry_group); } typedef enum { @@ -237,6 +301,7 @@ typedef enum { XML_TAG_NAME, XML_TAG_SERVICE, XML_TAG_TYPE, + XML_TAG_SUBTYPE, XML_TAG_DOMAIN_NAME, XML_TAG_HOST_NAME, XML_TAG_PORT, @@ -247,14 +312,18 @@ struct xml_userdata { StaticServiceGroup *group; StaticService *service; xml_tag_name current_tag; - gboolean failed; - gchar *buf; + int failed; + char *buf; }; +#ifndef XMLCALL +#define XMLCALL +#endif + static void XMLCALL xml_start(void *data, const char *el, const char *attr[]) { struct xml_userdata *u = data; - - g_assert(u); + + assert(u); if (u->failed) return; @@ -269,90 +338,115 @@ static void XMLCALL xml_start(void *data, const char *el, const char *attr[]) { u->current_tag = XML_TAG_NAME; if (attr[0]) { - if (strcmp(attr[0], "replace-wildcards") == 0) + if (strcmp(attr[0], "replace-wildcards") == 0) u->group->replace_wildcards = strcmp(attr[1], "yes") == 0; else goto invalid_attr; + + if (attr[2]) + goto invalid_attr; } - if (attr[2]) - goto invalid_attr; - - - } else if (u->current_tag == XML_TAG_SERVICE_GROUP && strcmp(el, "service") == 0) { - if (attr[0]) - goto invalid_attr; + u->current_tag = XML_TAG_SERVICE; - g_assert(!u->service); + assert(!u->service); u->service = static_service_new(u->group); - u->current_tag = XML_TAG_SERVICE; + if (attr[0]) { + if (strcmp(attr[0], "protocol") == 0) { + AvahiProtocol protocol; + + if (strcmp(attr[1], "ipv4") == 0) { + protocol = AVAHI_PROTO_INET; + } else if (strcmp(attr[1], "ipv6") == 0) { + protocol = AVAHI_PROTO_INET6; + } else if (strcmp(attr[1], "any") == 0) { + protocol = AVAHI_PROTO_UNSPEC; + } else { + avahi_log_error("%s: parse failure: invalid protocol specification \"%s\".", u->group->filename, attr[1]); + u->failed = 1; + return; + } + + u->service->protocol = protocol; + } else + goto invalid_attr; + + if (attr[2]) + goto invalid_attr; + } + } else if (u->current_tag == XML_TAG_SERVICE && strcmp(el, "type") == 0) { if (attr[0]) goto invalid_attr; u->current_tag = XML_TAG_TYPE; + } else if (u->current_tag == XML_TAG_SERVICE && strcmp(el, "subtype") == 0) { + if (attr[0]) + goto invalid_attr; + + u->current_tag = XML_TAG_SUBTYPE; } else if (u->current_tag == XML_TAG_SERVICE && strcmp(el, "domain-name") == 0) { if (attr[0]) goto invalid_attr; - + u->current_tag = XML_TAG_DOMAIN_NAME; } else if (u->current_tag == XML_TAG_SERVICE && strcmp(el, "host-name") == 0) { if (attr[0]) goto invalid_attr; - + u->current_tag = XML_TAG_HOST_NAME; } else if (u->current_tag == XML_TAG_SERVICE && strcmp(el, "port") == 0) { if (attr[0]) goto invalid_attr; - + u->current_tag = XML_TAG_PORT; } else if (u->current_tag == XML_TAG_SERVICE && strcmp(el, "txt-record") == 0) { if (attr[0]) goto invalid_attr; - + u->current_tag = XML_TAG_TXT_RECORD; } else { avahi_log_error("%s: parse failure: didn't expect element <%s>.", u->group->filename, el); - u->failed = TRUE; + u->failed = 1; } return; invalid_attr: avahi_log_error("%s: parse failure: invalid attribute for element <%s>.", u->group->filename, el); - u->failed = TRUE; + u->failed = 1; return; } - -static void XMLCALL xml_end(void *data, const char *el) { + +static void XMLCALL xml_end(void *data, AVAHI_GCC_UNUSED const char *el) { struct xml_userdata *u = data; - g_assert(u); + assert(u); if (u->failed) return; - + switch (u->current_tag) { case XML_TAG_SERVICE_GROUP: if (!u->group->name || !u->group->services) { avahi_log_error("%s: parse failure: service group incomplete.", u->group->filename); - u->failed = TRUE; + u->failed = 1; return; } - + u->current_tag = XML_TAG_INVALID; break; case XML_TAG_SERVICE: - if (u->service->port == 0 || !u->service->type) { + if (!u->service->type) { avahi_log_error("%s: parse failure: service incomplete.", u->group->filename); - u->failed = TRUE; + u->failed = 1; return; } - + u->service = NULL; u->current_tag = XML_TAG_SERVICE_GROUP; break; @@ -363,30 +457,38 @@ static void XMLCALL xml_end(void *data, const char *el) { case XML_TAG_PORT: { int p; - g_assert(u->service); - + assert(u->service); + p = u->buf ? atoi(u->buf) : 0; - if (p <= 0 || p > 0xFFFF) { + if (p < 0 || p > 0xFFFF) { avahi_log_error("%s: parse failure: invalid port specification \"%s\".", u->group->filename, u->buf); - u->failed = TRUE; + u->failed = 1; return; } - u->service->port = (guint16) p; + u->service->port = (uint16_t) p; u->current_tag = XML_TAG_SERVICE; break; } case XML_TAG_TXT_RECORD: { - g_assert(u->service); - + assert(u->service); + u->service->txt_records = avahi_string_list_add(u->service->txt_records, u->buf ? u->buf : ""); u->current_tag = XML_TAG_SERVICE; break; } - + + case XML_TAG_SUBTYPE: { + assert(u->service); + + u->service->subtypes = avahi_string_list_add(u->service->subtypes, u->buf ? u->buf : ""); + u->current_tag = XML_TAG_SERVICE; + break; + } + case XML_TAG_TYPE: case XML_TAG_DOMAIN_NAME: case XML_TAG_HOST_NAME: @@ -397,31 +499,32 @@ static void XMLCALL xml_end(void *data, const char *el) { ; } - g_free(u->buf); + avahi_free(u->buf); u->buf = NULL; } -static gchar *append_cdata(gchar *t, const gchar *n, int length) { - gchar *r, *k; - +static char *append_cdata(char *t, const char *n, int length) { + char *r, *k; + if (!length) return t; - k = g_strndup(n, length); + + k = avahi_strndup(n, length); if (t) { - r = g_strconcat(t, k, NULL); - g_free(k); - g_free(t); + r = avahi_strdup_printf("%s%s", t, k); + avahi_free(k); + avahi_free(t); } else r = k; - + return r; } static void XMLCALL xml_cdata(void *data, const XML_Char *s, int len) { struct xml_userdata *u = data; - g_assert(u); + assert(u); if (u->failed) return; @@ -430,24 +533,26 @@ static void XMLCALL xml_cdata(void *data, const XML_Char *s, int len) { case XML_TAG_NAME: u->group->name = append_cdata(u->group->name, s, len); break; - + case XML_TAG_TYPE: - g_assert(u->service); + assert(u->service); u->service->type = append_cdata(u->service->type, s, len); break; case XML_TAG_DOMAIN_NAME: - g_assert(u->service); + assert(u->service); u->service->domain_name = append_cdata(u->service->domain_name, s, len); break; case XML_TAG_HOST_NAME: - g_assert(u->service); + assert(u->service); u->service->host_name = append_cdata(u->service->host_name, s, len); break; case XML_TAG_PORT: case XML_TAG_TXT_RECORD: + case XML_TAG_SUBTYPE: + assert(u->service); u->buf = append_cdata(u->buf, s, len); break; @@ -458,32 +563,32 @@ static void XMLCALL xml_cdata(void *data, const XML_Char *s, int len) { } } -static gint static_service_group_load(StaticServiceGroup *g) { +static int static_service_group_load(StaticServiceGroup *g) { XML_Parser parser = NULL; - gint fd = -1; + int fd = -1; struct xml_userdata u; - gint r = -1; + int r = -1; struct stat st; ssize_t n; - g_assert(g); + assert(g); u.buf = NULL; u.group = g; u.service = NULL; u.current_tag = XML_TAG_INVALID; - u.failed = FALSE; + u.failed = 0; /* Cleanup old data in this service group, if available */ remove_static_service_group_from_server(g); while (g->services) static_service_free(g->services); - g_free(g->name); - g_free(g->chosen_name); + avahi_free(g->name); + avahi_free(g->chosen_name); g->name = g->chosen_name = NULL; - g->replace_wildcards = FALSE; - + g->replace_wildcards = 0; + if (!(parser = XML_ParserCreate(NULL))) { avahi_log_error("XML_ParserCreate() failed."); goto finish; @@ -500,7 +605,7 @@ static gint static_service_group_load(StaticServiceGroup *g) { } g->mtime = st.st_mtime; - + XML_SetUserData(parser, &u); XML_SetElementHandler(parser, xml_start, xml_end); @@ -522,7 +627,7 @@ static gint static_service_group_load(StaticServiceGroup *g) { } if (!XML_ParseBuffer(parser, n, n == 0)) { - avahi_log_error("XML_ParseBuffer() failed at line %d: %s.\n", XML_GetCurrentLineNumber(parser), XML_ErrorString(XML_GetErrorCode(parser))); + avahi_log_error("XML_ParseBuffer() failed at line %d: %s.\n", (int) XML_GetCurrentLineNumber(parser), XML_ErrorString(XML_GetErrorCode(parser))); goto finish; } @@ -539,21 +644,21 @@ finish: if (parser) XML_ParserFree(parser); - g_free(u.buf); + avahi_free(u.buf); return r; } -static void load_file(gchar *n) { +static void load_file(char *n) { StaticServiceGroup *g; - g_assert(n); + assert(n); for (g = groups; g; g = g->groups_next) if (strcmp(g->filename, n) == 0) return; - avahi_log_info("Loading service file %s", n); - + avahi_log_info("Loading service file %s.", n); + g = static_service_group_new(n); if (static_service_group_load(g) < 0) { avahi_log_error("Failed to load service group file %s, ignoring.", g->filename); @@ -561,10 +666,11 @@ static void load_file(gchar *n) { } } -void static_service_load(void) { +void static_service_load(int in_chroot) { StaticServiceGroup *g, *n; glob_t globbuf; - gchar **p; + int globret; + char **p; for (g = groups; g; g = n) { struct stat st; @@ -577,11 +683,11 @@ void static_service_load(void) { avahi_log_info("Service group file %s vanished, removing services.", g->filename); else avahi_log_warn("Failed to stat() file %s, ignoring: %s", g->filename, strerror(errno)); - + static_service_group_free(g); } else if (st.st_mtime != g->mtime) { avahi_log_info("Service group file %s changed, reloading.", g->filename); - + if (static_service_group_load(g) < 0) { avahi_log_warn("Failed to load service group file %s, removing service.", g->filename); static_service_group_free(g); @@ -590,18 +696,35 @@ void static_service_load(void) { } memset(&globbuf, 0, sizeof(globbuf)); - if (glob(AVAHI_SERVICE_DIRECTORY "/*.service", GLOB_ERR, NULL, &globbuf) != 0) - avahi_log_error("Failed to read service directory."); + + if ((globret = glob(in_chroot ? "/services/*.service" : AVAHI_SERVICE_DIR "/*.service", GLOB_ERR, NULL, &globbuf)) != 0) + + switch (globret) { +#ifdef GLOB_NOSPACE + case GLOB_NOSPACE: + avahi_log_error("Not enough memory to read service directory "AVAHI_SERVICE_DIR"."); + break; +#endif +#ifdef GLOB_NOMATCH + case GLOB_NOMATCH: + avahi_log_info("No service file found in "AVAHI_SERVICE_DIR"."); + break; +#endif + default: + avahi_log_error("Failed to read "AVAHI_SERVICE_DIR"."); + break; + } + else { for (p = globbuf.gl_pathv; *p; p++) load_file(*p); - + globfree(&globbuf); } } void static_service_free_all(void) { - + while (groups) static_service_group_free(groups); }