git.meshlink.io Git - catta/blob - avahi-daemon/static-hosts.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of avahi.
   5
   6   avahi is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as
   8   published by the Free Software Foundation; either version 2.1 of the
   9   License, or (at your option) any later version.
  10
  11   avahi is distributed in the hope that it will be useful, but WITHOUT
  12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  14   Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with avahi; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <string.h>
  27 #include <errno.h>
  28 #include <stdio.h>
  29
  30 #include <avahi-common/llist.h>
  31 #include <avahi-common/malloc.h>
  32 #include <avahi-common/error.h>
  33 #include <avahi-core/log.h>
  34 #include <avahi-core/publish.h>
  35
  36 #include "main.h"
  37 #include "static-hosts.h"
  38
  39 typedef struct StaticHost StaticHost;
  40
  41 struct StaticHost {
  42     AvahiSEntryGroup *group;
  43
  44     char *host, *ip;
  45
  46     AVAHI_LLIST_FIELDS(StaticHost, hosts);
  47 };
  48
  49 static AVAHI_LLIST_HEAD(StaticHost, hosts) = NULL;
  50
  51 static void add_static_host_to_server(StaticHost *h);
  52 static void remove_static_host_from_server(StaticHost *h);
  53
  54 static void entry_group_callback(AvahiServer *s, AVAHI_GCC_UNUSED AvahiSEntryGroup *eg, AvahiEntryGroupState state, void* userdata) {
  55     StaticHost *h;
  56
  57     assert(s);
  58     assert(eg);
  59
  60     h = userdata;
  61
  62     switch (state) {
  63
  64         case AVAHI_ENTRY_GROUP_COLLISION:
  65             avahi_log_error("Host name conflict for \"%s\", not established.", h->host);
  66             break;
  67
  68         case AVAHI_ENTRY_GROUP_ESTABLISHED:
  69             avahi_log_notice ("Static host name \"%s\" successfully established.", h->host);
  70             break;
  71
  72         case AVAHI_ENTRY_GROUP_FAILURE:
  73             avahi_log_notice ("Failed to establish static host name \"%s\": %s.", h->host, avahi_strerror (avahi_server_errno (s)));
  74             break;
  75
  76         case AVAHI_ENTRY_GROUP_UNCOMMITED:
  77         case AVAHI_ENTRY_GROUP_REGISTERING:
  78             ;
  79     }
  80 }
  81
  82 static StaticHost *static_host_new(void) {
  83     StaticHost *s;
  84
  85     s = avahi_new(StaticHost, 1);
  86
  87     s->group = NULL;
  88     s->host = NULL;
  89     s->ip = NULL;
  90
  91     AVAHI_LLIST_PREPEND(StaticHost, hosts, hosts, s);
  92
  93     return s;
  94 }
  95
  96 static void static_host_free(StaticHost *s) {
  97     assert(s);
  98
  99     AVAHI_LLIST_REMOVE(StaticHost, hosts, hosts, s);
 100
 101     if (s->group)
 102         avahi_s_entry_group_free (s->group);
 103
 104     avahi_free(s->host);
 105     avahi_free(s->ip);
 106
 107     avahi_free(s);
 108 }
 109
 110 static void add_static_host_to_server(StaticHost *h)
 111 {
 112     AvahiAddress a;
 113     AvahiProtocol p;
 114     int err;
 115     const AvahiServerConfig *config;
 116
 117     if (!h->group)
 118         if (!(h->group = avahi_s_entry_group_new (avahi_server, entry_group_callback, h))) {
 119             avahi_log_error("avahi_s_entry_group_new() failed: %s", avahi_strerror(avahi_server_errno(avahi_server)));
 120             return;
 121         }
 122
 123     if (!avahi_address_parse (h->ip, AVAHI_PROTO_UNSPEC, &a)) {
 124         avahi_log_error("Static host name %s: avahi_address_parse failed", h->host);
 125         return;
 126     }
 127
 128     config = avahi_server_get_config(avahi_server);
 129
 130     p = (a.proto == AVAHI_PROTO_INET && config->publish_a_on_ipv6) ||
 131         (a.proto == AVAHI_PROTO_INET6 && config->publish_aaaa_on_ipv4) ? AVAHI_PROTO_UNSPEC : a.proto;
 132
 133     if ((err = avahi_server_add_address(avahi_server, h->group, AVAHI_IF_UNSPEC, p, 0, h->host, &a)) < 0) {
 134         avahi_log_error ("Static host name %s: avahi_server_add_address failure: %s", h->host, avahi_strerror(err));
 135         return;
 136     }
 137
 138     avahi_s_entry_group_commit (h->group);
 139 }
 140
 141 static void remove_static_host_from_server(StaticHost *h)
 142 {
 143     if (h->group)
 144         avahi_s_entry_group_reset (h->group);
 145 }
 146
 147 void static_hosts_add_to_server(void) {
 148     StaticHost *h;
 149
 150     for (h = hosts; h; h = h->hosts_next) {
 151         add_static_host_to_server(h);
 152     }
 153 }
 154
 155 void static_hosts_remove_from_server(void) {
 156     StaticHost *h;
 157
 158     for (h = hosts; h; h = h->hosts_next) {
 159         remove_static_host_from_server(h);
 160     }
 161 }
 162
 163 void static_hosts_load(int in_chroot) {
 164     FILE *f;
 165     unsigned int line = 0;
 166     const char *filename = (in_chroot ? "/hosts" : AVAHI_CONFIG_DIR "/hosts");
 167
 168     if (!(f = fopen(filename, "r")))
 169     {
 170         if (errno != ENOENT)
 171             avahi_log_error ("Failed to open static hosts file: %s", strerror (errno));
 172         return;
 173     }
 174
 175     while (!feof(f)) {
 176         unsigned int len;
 177         char ln[256], *s;
 178         char *host, *ip;
 179         StaticHost *h;
 180
 181         if (!fgets(ln, sizeof (ln), f))
 182             break;
 183
 184         line++;
 185
 186         /* Find the start of the line, ignore whitespace */
 187         s = ln + strspn(ln, " \t");
 188         /* Set the end of the string to NULL */
 189         s[strcspn(s, "#\r\n")] = 0;
 190
 191         /* Ignore blank lines */
 192         if (*s == 0)
 193             continue;
 194
 195         /* Read the first string (ip) up to the next whitespace */
 196         len = strcspn(s, " \t");
 197         ip = avahi_strndup(s, len);
 198
 199         /* Skip past it */
 200         s += len;
 201
 202         /* Find the next token */
 203         s += strspn(s, " \t");
 204         len = strcspn(s, " \t");
 205         host = avahi_strndup(s, len);
 206
 207         if (*host == 0)
 208         {
 209             avahi_log_error("%s:%d: Error, unexpected end of line!", filename, line);
 210             avahi_free(host);
 211             avahi_free(ip);
 212             break;
 213         }
 214
 215         /* Skip over the host */
 216         s += len;
 217
 218         /* Skip past any more spaces */
 219         s += strspn(s, " \t");
 220
 221         /* Anything left? */
 222         if (*s != 0) {
 223             avahi_log_error ("%s:%d: Junk on the end of the line!", filename, line);
 224             avahi_free(host);
 225             avahi_free(ip);
 226             break;
 227         }
 228
 229         h = static_host_new();
 230         h->host = host;
 231         h->ip = ip;
 232     }
 233
 234     fclose(f);
 235 }
 236
 237 void static_hosts_free_all (void)
 238 {
 239     while(hosts)
 240         static_host_free(hosts);
 241 }