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         WARN ("No local service found for service id '%d'", service_id);
 188         return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 189     }
 190
 191     if (!msgport_service_unregister (service))
 192         return MESSAGEPORT_ERROR_IO_ERROR;
 193
 194     object_path = (const gchar *)g_hash_table_lookup (manager->local_services,
 195                                                       GINT_TO_POINTER(service_id));
 196     g_hash_table_remove (manager->local_services, GINT_TO_POINTER(service_id));
 197     g_hash_table_remove (manager->services, object_path);
 198
 199     return MESSAGEPORT_ERROR_NONE;
 200 }
 201
 202 messageport_error_e
 203 msgport_manager_check_remote_service (MsgPortManager *manager, const gchar *app_id, const gchar *port, gboolean is_trusted, guint *service_id_out)
 204 {
 205     GError *error = NULL;
 206     guint remote_service_id = 0;
 207
 208     if (service_id_out) *service_id_out = 0;
 209
 210     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 211     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 212     g_return_val_if_fail (app_id && port, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 213
 214     if (!app_id || !port) return MESSAGEPORT_ERROR_INVALID_PARAMETER;
 215
 216     msgport_dbus_glue_manager_call_check_for_remote_service_sync (manager->proxy,
 217             app_id, port, is_trusted, &remote_service_id, NULL, &error);
 218
 219     if (error) {
 220         WARN ("No service found for app_id %s, port name %s: %s", app_id, port, error->message);
 221         g_error_free (error);
 222         return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 223     }
 224     else {
 225         DBG ("Got service id %d for %s, %s", remote_service_id, app_id, port);
 226
 227         if (service_id_out)  *service_id_out = remote_service_id;
 228     }
 229
 230     return MESSAGEPORT_ERROR_NONE;
 231 }
 232
 233 messageport_error_e
 234 msgport_manager_get_service_name (MsgPortManager *manager, int service_id, gchar **name_out)
 235 {
 236     MsgPortService *service = NULL;
 237     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 238     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 239     g_return_val_if_fail (name_out && service_id, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 240
 241     service = _get_local_port (manager, service_id);
 242     if (!service) return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 243
 244     *name_out = g_strdup (msgport_service_name (service));
 245     DBG ("PORT NAME : %s", *name_out);
 246
 247     return MESSAGEPORT_ERROR_NONE;
 248 }
 249
 250 messageport_error_e
 251 msgport_manager_send_message (MsgPortManager *manager, const gchar *remote_app_id, const gchar *remote_port, gboolean is_trusted, GVariant *data)
 252 {
 253     guint service_id = 0;
 254     messageport_error_e res = MESSAGEPORT_ERROR_NONE;
 255     GError *error = NULL;
 256
 257     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 258     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 259     g_return_val_if_fail (remote_app_id && remote_port, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 260
 261     res = msgport_manager_check_remote_service (manager, remote_app_id, remote_port, is_trusted, &service_id);
 262     if (service_id == 0) return res;
 263
 264     msgport_dbus_glue_manager_call_send_message_sync (manager->proxy, service_id, data, NULL, &error);
 265
 266     if (error) {
 267         WARN ("Failed to send message to (%s:%s) : %s", remote_app_id, remote_port, error->message);
 268         g_error_free (error);
 269         res = MESSAGEPORT_ERROR_IO_ERROR;
 270     }
 271
 272     return res;
 273 }
 274
 275 messageport_error_e
 276 msgport_manager_send_bidirectional_message (MsgPortManager *manager, int local_port_id, const gchar *remote_app_id, const gchar *remote_port, gboolean is_trusted, GVariant *data)
 277 {
 278     MsgPortService *service = NULL;
 279     guint remote_service_id = 0;
 280     messageport_error_e res = 0;
 281
 282     g_return_val_if_fail (manager && MSGPORT_IS_MANAGER (manager), MESSAGEPORT_ERROR_IO_ERROR);
 283     g_return_val_if_fail (manager->proxy, MESSAGEPORT_ERROR_IO_ERROR);
 284     g_return_val_if_fail (local_port_id > 0 && remote_app_id && remote_port, MESSAGEPORT_ERROR_INVALID_PARAMETER);
 285
 286     service = _get_local_port (manager, local_port_id);
 287     if (!service) {
 288         WARN ("No local service found for service id '%d'", local_port_id);
 289         return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 290     }
 291     if ( (res = msgport_manager_check_remote_service (manager, remote_app_id, remote_port, is_trusted, &remote_service_id) != MESSAGEPORT_ERROR_NONE)) {
 292         WARN ("No remote port informatuon for %s:%s, error : %d", remote_app_id, remote_port, res);
 293         return MESSAGEPORT_ERROR_MESSAGEPORT_NOT_FOUND;
 294     }
 295
 296     DBG ("Sending message from local service '%p' to remote sercie id '%d'", service, remote_service_id);
 297     return msgport_service_send_message (service, remote_service_id, data);
 298 }
 299