git.meshlink.io Git - catta/blob - avahi-daemon/main.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 <avahi-core/core.h>
  27
  28 #define DBUS_API_SUBJECT_TO_CHANGE
  29 #include <dbus/dbus.h>
  30 #include <dbus/dbus-glib-lowlevel.h>
  31
  32 #include "simple-protocol.h"
  33
  34 #define DBUS_SERVICE_AVAHI "org.freedesktop.Avahi"
  35
  36 static DBusHandlerResult
  37 do_register (DBusConnection *conn, DBusMessage *message)
  38 {
  39     DBusError error;
  40     char *s;
  41
  42     dbus_error_init (&error);
  43
  44     dbus_message_get_args (message, &error,
  45                            DBUS_TYPE_STRING, &s,
  46                            DBUS_TYPE_INVALID);
  47
  48     if (dbus_error_is_set (&error))
  49     {
  50         g_warning ("Error parsing register attempt");
  51         dbus_error_free (&error);
  52
  53         return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
  54     }
  55
  56     g_message ("Register received from: %s", s);
  57
  58     return DBUS_HANDLER_RESULT_HANDLED;
  59 }
  60
  61 static DBusHandlerResult
  62 signal_filter (DBusConnection *conn, DBusMessage *message, void *user_data)
  63 {
  64     GMainLoop *loop = user_data;
  65     DBusError error;
  66
  67     dbus_error_init (&error);
  68
  69     g_message ("dbus: interface=%s, path=%s, member=%s",
  70                dbus_message_get_interface (message),
  71                dbus_message_get_path (message),
  72                dbus_message_get_member (message));
  73
  74     if (dbus_message_is_signal (message,
  75                                 DBUS_INTERFACE_ORG_FREEDESKTOP_LOCAL,
  76                                 "Disconnected"))
  77     {
  78         /* No, we shouldn't quit, but until we get somewhere
  79          * usefull such that we can restore our state, we will */
  80         g_warning ("Disconnnected from d-bus");
  81
  82         g_main_loop_quit (loop);
  83         return DBUS_HANDLER_RESULT_HANDLED;
  84     } else if (dbus_message_is_method_call (message, DBUS_SERVICE_AVAHI,
  85                                             "Register"))
  86     {
  87         return do_register (conn, message);
  88     } else if (dbus_message_is_signal (message,
  89                                        DBUS_INTERFACE_ORG_FREEDESKTOP_DBUS,
  90                                        "ServiceAcquired"))
  91     {
  92         char *name;
  93
  94         dbus_message_get_args (message, &error,
  95                                DBUS_TYPE_STRING, &name,
  96                                DBUS_TYPE_INVALID);
  97
  98         if (dbus_error_is_set (&error))
  99         {
 100             g_warning ("Error parsing NameAcquired message");
 101             dbus_error_free (&error);
 102
 103             return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 104         }
 105
 106         g_message ("dbus: ServiceAcquired (%s)", name);
 107
 108         return DBUS_HANDLER_RESULT_HANDLED;
 109     }
 110
 111     g_message ("dbus: missed event");
 112
 113     return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 114 }
 115
 116 int main(int argc, char *argv[]) {
 117     GMainLoop *loop = NULL;
 118     DBusConnection *bus = NULL;
 119     DBusError error;
 120     gint r = -1;
 121
 122     loop = g_main_loop_new(NULL, FALSE);
 123
 124     dbus_error_init (&error);
 125
 126     bus = dbus_bus_get (DBUS_BUS_SYSTEM, &error);
 127
 128     if (bus == NULL)
 129     {
 130         g_warning ("dbus_bus_get(): %s", error.message);
 131         dbus_error_free (&error);
 132
 133         goto finish;
 134     }
 135
 136     dbus_connection_setup_with_g_main (bus, NULL);
 137     dbus_connection_set_exit_on_disconnect (bus, FALSE);
 138
 139     dbus_bus_acquire_service (bus, DBUS_SERVICE_AVAHI, 0, &error);
 140
 141     if (dbus_error_is_set (&error))
 142     {
 143         g_warning ("dbus_error_is_set (): %s", error.message);
 144         dbus_error_free (&error);
 145
 146         goto finish;
 147     }
 148
 149     dbus_connection_add_filter (bus, signal_filter, loop, NULL);
 150     dbus_bus_add_match (bus,
 151                         "type='method_call',interface='org.freedesktop.Avahi'",
 152                         &error);
 153
 154     if (dbus_error_is_set (&error))
 155     {
 156         g_warning ("dbus_bus_add_match (): %s", error.message);
 157         dbus_error_free (&error);
 158
 159         goto finish;
 160     }
 161
 162     if (simple_protocol_setup(NULL) < 0)
 163         goto finish;
 164
 165     g_main_loop_run(loop);
 166
 167     r = 0;
 168
 169 finish:
 170
 171     simple_protocol_shutdown();
 172
 173     if (bus) {
 174         dbus_connection_disconnect(bus);
 175         dbus_connection_unref(bus);
 176     }
 177
 178     if (loop)
 179         g_main_loop_unref(loop);
 180
 181     return r;
 182 }