git.meshlink.io Git - catta/blob - avahi/conformance-test.c

   1 /***
   2   This file is part of avahi.
   3
   4   avahi is free software; you can redistribute it and/or modify it
   5   under the terms of the GNU Lesser General Public License as
   6   published by the Free Software Foundation; either version 2.1 of the
   7   License, or (at your option) any later version.
   8
   9   avahi is distributed in the hope that it will be useful, but WITHOUT
  10   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  12   Public License for more details.
  13
  14   You should have received a copy of the GNU Lesser General Public
  15   License along with avahi; if not, write to the Free Software
  16   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  17   USA.
  18 ***/
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <stdlib.h>
  25 #include <unistd.h>
  26 #include <stdio.h>
  27 #include <assert.h>
  28
  29 #include <sys/types.h>
  30 #include <sys/socket.h>
  31 #include <netinet/in.h>
  32 #include <arpa/inet.h>
  33
  34 #include <avahi/alternative.h>
  35 #include <avahi/malloc.h>
  36 #include <avahi/simple-watch.h>
  37 #include <avahi/timeval.h>
  38
  39 #include <avahi/core.h>
  40 #include <avahi/log.h>
  41 #include <avahi/lookup.h>
  42 #include <avahi/publish.h>
  43
  44 static char *name = NULL;
  45 static AvahiSEntryGroup *group = NULL;
  46 static int try = 0;
  47 static AvahiServer *avahi = NULL;
  48 static const AvahiPoll *poll_api;
  49
  50 static void dump_line(const char *text, AVAHI_GCC_UNUSED void* userdata) {
  51     printf("%s\n", text);
  52 }
  53
  54 static void dump_timeout_callback(AvahiTimeout *timeout, AVAHI_GCC_UNUSED void* userdata) {
  55     struct timeval tv;
  56
  57     avahi_server_dump(avahi, dump_line, NULL);
  58
  59     avahi_elapse_time(&tv, 5000, 0);
  60     poll_api->timeout_update(timeout, &tv);
  61 }
  62
  63 static void entry_group_callback(AvahiServer *s, AvahiSEntryGroup *g, AvahiEntryGroupState state, void* userdata);
  64
  65 static void create_service(const char *t) {
  66     char *n;
  67
  68     assert(t || name);
  69
  70     n = t ? avahi_strdup(t) : avahi_alternative_service_name(name);
  71     avahi_free(name);
  72     name = n;
  73
  74     if (group)
  75         avahi_s_entry_group_reset(group);
  76     else
  77         group = avahi_s_entry_group_new(avahi, entry_group_callback, NULL);
  78
  79     avahi_server_add_service(avahi, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, name, "_http._tcp", NULL, NULL, 80, "foo", NULL);
  80     avahi_s_entry_group_commit(group);
  81
  82     try++;
  83 }
  84
  85 static void rename_timeout_callback(AvahiTimeout *timeout, AVAHI_GCC_UNUSED void *userdata) {
  86     struct timeval tv;
  87
  88     if (access("flag", F_OK) == 0) {
  89         create_service("New - Bonjour Service Name");
  90         return;
  91     }
  92
  93     avahi_elapse_time(&tv, 5000, 0);
  94     poll_api->timeout_update(timeout, &tv);
  95 }
  96
  97 static void entry_group_callback(AVAHI_GCC_UNUSED AvahiServer *s, AVAHI_GCC_UNUSED AvahiSEntryGroup *g, AvahiEntryGroupState state, AVAHI_GCC_UNUSED void* userdata) {
  98     if (state == AVAHI_ENTRY_GROUP_COLLISION)
  99         create_service(NULL);
 100     else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
 101         avahi_log_debug("ESTABLISHED !!!!");
 102         try = 0;
 103     }
 104 }
 105
 106 static void server_callback(AvahiServer *s, AvahiServerState state, AVAHI_GCC_UNUSED void* userdata) {
 107     avahi_log_debug("server state: %i", state);
 108
 109     if (state == AVAHI_SERVER_RUNNING) {
 110         avahi_server_dump(avahi, dump_line, NULL);
 111     } else if (state == AVAHI_SERVER_COLLISION) {
 112         char *n;
 113
 114         n = avahi_alternative_host_name(avahi_server_get_host_name(s));
 115         avahi_log_warn("Host name conflict, retrying with <%s>", n);
 116         avahi_server_set_host_name(s, n);
 117         avahi_free(n);
 118
 119     }
 120 }
 121
 122 int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char *argv[]) {
 123     int error;
 124     AvahiSimplePoll *simple_poll;
 125     struct timeval tv;
 126     struct AvahiServerConfig config;
 127
 128     simple_poll = avahi_simple_poll_new();
 129     poll_api = avahi_simple_poll_get(simple_poll);
 130
 131     avahi_server_config_init(&config);
 132     config.publish_workstation = 0;
 133     config.use_ipv6 = 0;
 134     config.publish_domain = 0;
 135     config.publish_hinfo = 0;
 136     avahi = avahi_server_new(poll_api, &config, server_callback, NULL, &error);
 137     avahi_server_config_free(&config);
 138
 139     avahi_elapse_time(&tv, 5000, 0);
 140     poll_api->timeout_new(poll_api, &tv, dump_timeout_callback, avahi);
 141
 142     avahi_elapse_time(&tv, 5000, 0);
 143     poll_api->timeout_new(poll_api, &tv, rename_timeout_callback, avahi);
 144
 145     /* Evil, but the conformace test requires that*/
 146     create_service("gurke");
 147
 148
 149     avahi_simple_poll_loop(simple_poll);
 150
 151     if (group)
 152         avahi_s_entry_group_free(group);
 153     avahi_server_free(avahi);
 154
 155     avahi_simple_poll_free(simple_poll);
 156
 157     return 0;
 158 }