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

   1 /***
   2   This file is part of catta.
   3
   4   catta 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   catta 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 catta; 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 <catta/alternative.h>
  35 #include <catta/malloc.h>
  36 #include <catta/simple-watch.h>
  37 #include <catta/timeval.h>
  38
  39 #include <catta/core.h>
  40 #include <catta/log.h>
  41 #include <catta/lookup.h>
  42 #include <catta/publish.h>
  43
  44 static char *name = NULL;
  45 static CattaSEntryGroup *group = NULL;
  46 static int try = 0;
  47 static CattaServer *catta = NULL;
  48 static const CattaPoll *poll_api;
  49
  50 static void dump_line(const char *text, CATTA_GCC_UNUSED void* userdata) {
  51     printf("%s\n", text);
  52 }
  53
  54 static void dump_timeout_callback(CattaTimeout *timeout, CATTA_GCC_UNUSED void* userdata) {
  55     struct timeval tv;
  56
  57     catta_server_dump(catta, dump_line, NULL);
  58
  59     catta_elapse_time(&tv, 5000, 0);
  60     poll_api->timeout_update(timeout, &tv);
  61 }
  62
  63 static void entry_group_callback(CattaServer *s, CattaSEntryGroup *g, CattaEntryGroupState state, void* userdata);
  64
  65 static void create_service(const char *t) {
  66     char *n;
  67
  68     assert(t || name);
  69
  70     n = t ? catta_strdup(t) : catta_alternative_service_name(name);
  71     catta_free(name);
  72     name = n;
  73
  74     if (group)
  75         catta_s_entry_group_reset(group);
  76     else
  77         group = catta_s_entry_group_new(catta, entry_group_callback, NULL);
  78
  79     catta_server_add_service(catta, group, CATTA_IF_UNSPEC, CATTA_PROTO_UNSPEC, 0, name, "_http._tcp", NULL, NULL, 80, "foo", NULL);
  80     catta_s_entry_group_commit(group);
  81
  82     try++;
  83 }
  84
  85 static void rename_timeout_callback(CattaTimeout *timeout, CATTA_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     catta_elapse_time(&tv, 5000, 0);
  94     poll_api->timeout_update(timeout, &tv);
  95 }
  96
  97 static void entry_group_callback(CATTA_GCC_UNUSED CattaServer *s, CATTA_GCC_UNUSED CattaSEntryGroup *g, CattaEntryGroupState state, CATTA_GCC_UNUSED void* userdata) {
  98     if (state == CATTA_ENTRY_GROUP_COLLISION)
  99         create_service(NULL);
 100     else if (state == CATTA_ENTRY_GROUP_ESTABLISHED) {
 101         catta_log_debug("ESTABLISHED !!!!");
 102         try = 0;
 103     }
 104 }
 105
 106 static void server_callback(CattaServer *s, CattaServerState state, CATTA_GCC_UNUSED void* userdata) {
 107     catta_log_debug("server state: %i", state);
 108
 109     if (state == CATTA_SERVER_RUNNING) {
 110         catta_server_dump(catta, dump_line, NULL);
 111     } else if (state == CATTA_SERVER_COLLISION) {
 112         char *n;
 113
 114         n = catta_alternative_host_name(catta_server_get_host_name(s));
 115         catta_log_warn("Host name conflict, retrying with <%s>", n);
 116         catta_server_set_host_name(s, n);
 117         catta_free(n);
 118
 119     }
 120 }
 121
 122 int main(CATTA_GCC_UNUSED int argc, CATTA_GCC_UNUSED char *argv[]) {
 123     int error;
 124     CattaSimplePoll *simple_poll;
 125     struct timeval tv;
 126     struct CattaServerConfig config;
 127
 128     simple_poll = catta_simple_poll_new();
 129     poll_api = catta_simple_poll_get(simple_poll);
 130
 131     catta_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     catta = catta_server_new(poll_api, &config, server_callback, NULL, &error);
 137     catta_server_config_free(&config);
 138
 139     catta_elapse_time(&tv, 5000, 0);
 140     poll_api->timeout_new(poll_api, &tv, dump_timeout_callback, catta);
 141
 142     catta_elapse_time(&tv, 5000, 0);
 143     poll_api->timeout_new(poll_api, &tv, rename_timeout_callback, catta);
 144
 145     /* Evil, but the conformace test requires that*/
 146     create_service("gurke");
 147
 148
 149     catta_simple_poll_loop(simple_poll);
 150
 151     if (group)
 152         catta_s_entry_group_free(group);
 153     catta_server_free(catta);
 154
 155     catta_simple_poll_free(simple_poll);
 156
 157     return 0;
 158 }