lib/msgport-manager.c

   1 #include "msgport-manager.h"
   2 #include "msgport-service.h"
   3 #include "message-port.h" /* MESSAGEPORT_ERROR */
   4 #include "common/dbus-manager-glue.h"
   5 #include "common/log.h"
   6 #include "config.h" /* MESSAGEPORT_BUS_ADDRESS */
   7 #include <gio/gio.h>
   8
   9 struct _MsgPortManager
  10 {
  11     GObject parent;
  12
  13     MsgPortDbusGlueManager *proxy;
  14     GHashTable *services; /* {gchar*:MsgPortService*} */
  15     GHashTable *local_services; /* {gint: gchar *} */
  16     GHashTable *remote_services; /* {gint: gchar *} */
  17 };
  18
  19 G_DEFINE_TYPE (MsgPortManager, msgport_manager, G_TYPE_OBJECT)
  20
  21 static void
  22 _unregister_service_cb (int service_id, const gchar *object_path, MsgPortManager *manager)
  23 {
  24     MsgPortService *service = g_hash_table_lookup (manager->services, object_path);
  25
  26     if (service) msgport_service_unregister (service);
  27 }
  28
  29 static void
  30 _finalize (GObject *self)
  31 {
  32     MsgPortManager *manager = MSGPORT_MANAGER (self);
  33
  34     if (manager->local_services) {
  35         g_hash_table_unref (manager->local_services);
  36         manager->local_services = NULL;
  37     }
  38
  39     if (manager->remote_services) {
  40         g_hash_table_unref (manager->remote_services);
  41         manager->remote_services = NULL;
  42     }
  43
  44     G_OBJECT_CLASS (msgport_manager_parent_class)->finalize (self);
  45 }
  46
  47 static void
  48 _dispose (GObject *self)
  49 {
  50     MsgPortManager *manager = MSGPORT_MANAGER (self);
  51
  52     g_hash_table_foreach (manager->local_services, (GHFunc)_unregister_service_cb, manager);
  53
  54     if (manager->services) {
  55         g_hash_table_unref (manager->services);
  56         manager->services = NULL;
  57     }
  58
  59     g_clear_object (&manager->proxy);
  60
  61     G_OBJECT_CLASS (msgport_manager_parent_class)->dispose (self);
  62 }
  63
  64 static void
  65 msgport_manager_class_init (MsgPortManagerClass *klass)
  66 {
  67     GObjectClass *g_klass = G_OBJECT_CLASS (klass);
  68
  69     g_klass->finalize = _finalize;
  70     g_klass->dispose = _dispose;
  71 }
  72
  73 static void
  74 msgport_manager_init (MsgPortManager *manager)
  75 {
  76     GError          *error = NULL;
  77     GDBusConnection *connection = NULL;
  78     gchar           *bus_address = g_strdup_printf (MESSAGEPORT_BUS_ADDRESS);
  79
  80     manager->services = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_object_unref);
  81     manager->local_services = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, NULL);
  82     manager->remote_services = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, NULL);
  83
  84     connection = g_dbus_connection_new_for_address_sync (bus_address,
  85             G_DBUS_CONNECTION_FLAGS_AUTHENTICATION_CLIENT, NULL, NULL, &error);
  86     if (error) {
  87         WARN ("Fail to connect messageport server at address %s: %s", bus_address, error->message);
  88         g_error_free (error);
  89     }
  90     else {
  91         manager->proxy = msgport_dbus_glue_manager_proxy_new_sync (
  92             connection, G_DBUS_PROXY_FLAGS_DO_NOT_LOAD_PROPERTIES, NULL, "/", NULL, &error);
  93         if (error) {
  94             WARN ("Fail to get manager proxy : %s", error->message);
  95             g_error_free (error);
  96         }
  97     }
  98 }
  99
 100 MsgPortManager * msgport_manager_new ()
 101 {
 102     return g_object_new (MSGPORT_TYPE_MANAGER, NULL);
 103 }
 104
 105 static MsgPortManager *__manager;
 106
 107 MsgPortManager * msgport_get_manager ()
 108 {
 109     if (!__manager) {
 110         __manager = msgport_manager_new ();
 111     }
 112
 113     return __manager;
 114 }
 115
 116 static int
 117 _create_and_cache_service (MsgPortManager *manager, gchar *object_path, messageport_message_cb cb)
 118 {
 119     int id;
 120     MsgPortService *service = msgport_service_new (
 121             g_dbus_proxy_get_connection (G_DBUS_PROXY(manager->proxy)),
 122             object_path, cb);
 123     if (!service) {
 124         return MESSAGEPORT_ERROR_IO_ERROR;
 125     }
 126
 127     id = msgport_service_id (service);
 128
 129     g_hash_table_insert (manager->services, object_path, service);
 130     g_hash_table_insert (manager->local_services, GINT_TO_POINTER (id), object_path);
 131
 132     return id;
 133 }
 134
 135 static MsgPortService *
 136 _get_local_port (MsgPortManager *manager, int service_id)
 137 {
 138     const gchar *object_path = NULL;
 139     MsgPortService *service = NULL;
 140
 141     object_path = g_hash_table_lookup (manager->local_services, GINT_TO_POINTER(service_id));
 142     if (!object_path) return NULL;
 143
 144     service = MSGPORT_SERVICE (g_hash_table_lookup (manager->services, object_path));
 145     if (!service) {
 146         g_hash_table_remove (manager->local_services, GINT_TO_POINTER (service_id));
 147         return NULL;
 148     }
 149
 150     return service;
 151 }
 152
 153 messageport_error_e
 154 msgport_manager_register_service (MsgPortManager *manager, const gchar *port_name, gboolean is_trusted, messageport_message_cb message_cb, int *service_id)
 155 {
 156     GError *error = NULL;
 157     gchar *object_path = NULL;
 158
 159     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 160     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 161     g_return_val_if_fail (service_id && port_name && message_cb, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 162
 163     msgport_dbus_glue_manager_call_register_service_sync (manager->proxy,
 164             port_name, is_trusted, &object_path, NULL, &error);
 165
 166     if (error) {
 167         WARN ("unable to register service (%s): %s", port_name, error->message);
 168         g_error_free (error);
 169         return MESSAGEPORT_ERROR_IO_ERROR;
 170     }
 171
 172     *service_id = _create_and_cache_service (manager, object_path, message_cb);
 173
 174     return MESSAGEPORT_ERROR_NONE;
 175 }
 176
 177 messageport_error_e
 178 msgport_manager_unregister_servcie (MsgPortManager *manager, int service_id)
 179 {
 180     const gchar *object_path = NULL;
 181     MsgPortService *service = NULL;
 182     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), FALSE);
 183     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 184
 185     service = _get_local_port (manager, service_id);
 186     if (!service) {
 187         return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 188     }
 189
 190     if (!msgport_service_unregister (service))
 191         return MESSAGEPORT_ERROR_IO_ERROR;
 192
 193     object_path = (const gchar *)g_hash_table_lookup (manager->local_services,
 194                                                       GINT_TO_POINTER(service_id));
 195     g_hash_table_remove (manager->local_services, GINT_TO_POINTER(service_id));
 196     g_hash_table_remove (manager->services, object_path);
 197
 198     return MESSAGEPORT_ERROR_NONE;
 199 }
 200
 201 messageport_error_e
 202 msgport_manager_check_remote_service (MsgPortManager *manager, const gchar *app_id, const gchar *port, gboolean is_trusted, guint *service_id_out)
 203 {
 204     GError *error = NULL;
 205     guint remote_service_id = 0;
 206
 207     if (service_id_out) *service_id_out = 0;
 208
 209     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 210     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 211     g_return_val_if_fail (app_id && port, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 212
 213     if (!app_id || !port) return MESSAGEPORT_ERROR_INVALID_PARAMETER;
 214
 215     msgport_dbus_glue_manager_call_check_for_remote_service_sync (manager->proxy,
 216             app_id, port, is_trusted, &remote_service_id, NULL, &error);
 217
 218     if (error) {
 219         WARN ("No service found for app_id %s, port name %s: %s", app_id, port, error->message);
 220         g_error_free (error);
 221         return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 222     }
 223     else {
 224         DBG ("Got service id %d for %s, %s", remote_service_id, app_id, port);
 225
 226         if (service_id_out)  *service_id_out = remote_service_id;
 227     }
 228
 229     return MESSAGEPORT_ERROR_NONE;
 230 }
 231
 232 messageport_error_e
 233 msgport_manager_get_service_name (MsgPortManager *manager, int service_id, gchar **name_out)
 234 {
 235     MsgPortService *service = NULL;
 236     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 237     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 238     g_return_val_if_fail (name_out && service_id, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 239
 240     service = _get_local_port (manager, service_id);
 241     if (!service) return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 242
 243     *name_out = g_strdup (msgport_service_name (service));
 244     DBG ("PORT NAME : %s", *name_out);
 245
 246     return MESSAGEPORT_ERROR_NONE;
 247 }
 248
 249 messageport_error_e
 250 msgport_manager_send_message (MsgPortManager *manager, const gchar *remote_app_id, const gchar *remote_port, gboolean is_trusted, GVariant *data)
 251 {
 252     guint service_id = 0;
 253     messageport_error_e res = MESSAGEPORT_ERROR_NONE;
 254     GError *error = NULL;
 255
 256     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 257     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 258     g_return_val_if_fail (remote_app_id && remote_port, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 259
 260     res = msgport_manager_check_remote_service (manager, remote_app_id, remote_port, is_trusted, &service_id);
 261     if (service_id == 0) return res;
 262
 263     msgport_dbus_glue_manager_call_send_message_sync (manager->proxy, service_id, data, NULL, &error);
 264
 265     if (error) {
 266         WARN ("Failed to send message to (%s:%s) : %s", remote_app_id, remote_port, error->message);
 267         g_error_free (error);
 268         res = MESSAGEPORT_ERROR_IO_ERROR;
 269     }
 270
 271     return res;
 272 }
 273