plugin: add a base class to handle modem tracking

3 files changed, 182 insertions, 1 deletions

diff --git a/src/Makefile.am b/src/Makefile.am
index e1557341..9de57200 100644
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -44,7 +44,9 @@ modem_manager_SOURCES = \
 	mm-options.c \
 	mm-options.h \
 	mm-plugin.c \
-	mm-plugin.h
+	mm-plugin.h \
+	mm-plugin-base.c \
+	mm-plugin-base.h
 
 mm-manager-glue.h: $(top_srcdir)/introspection/mm-manager.xml
 	dbus-binding-tool --prefix=mm_manager --mode=glib-server --output=$@ $<
diff --git a/src/mm-plugin-base.c b/src/mm-plugin-base.c
new file mode 100644
index 00000000..9d3799d5
--- /dev/null
+++ b/src/mm-plugin-base.c
@@ -0,0 +1,127 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/*
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details:
+ *
+ * Copyright (C) 2008 - 2009 Novell, Inc.
+ * Copyright (C) 2009 Red Hat, Inc.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+
+#include "mm-plugin-base.h"
+
+G_DEFINE_TYPE (MMPluginBase, mm_plugin_base, G_TYPE_OBJECT)
+
+#define MM_PLUGIN_BASE_GET_PRIVATE(o) (G_TYPE_INSTANCE_GET_PRIVATE ((o), MM_TYPE_PLUGIN_BASE, MMPluginBasePrivate))
+
+typedef struct {
+    GHashTable *modems;
+} MMPluginBasePrivate;
+
+
+typedef struct {
+    char *key;
+    gpointer modem;
+} FindInfo;
+
+static void
+find_modem (gpointer key, gpointer data, gpointer user_data)
+{
+    FindInfo *info = user_data;
+
+    if (!info->key && data == info->modem)
+        info->key = g_strdup ((const char *) key);
+}
+
+static void
+modem_destroyed (gpointer data, GObject *modem)
+{
+    MMPluginBase *self = MM_PLUGIN_BASE (data);
+    MMPluginBasePrivate *priv = MM_PLUGIN_BASE_GET_PRIVATE (self);
+    FindInfo info = { NULL, modem };
+
+    g_hash_table_foreach (priv->modems, find_modem, &info);
+    if (info.key)
+        g_hash_table_remove (priv->modems, info.key);
+    g_free (info.key);
+}
+
+gboolean
+mm_plugin_base_add_modem (MMPluginBase *self,
+                          MMModem *modem)
+{
+    MMPluginBasePrivate *priv;
+    const char *device;
+
+    g_return_val_if_fail (self != NULL, FALSE);
+    g_return_val_if_fail (MM_IS_PLUGIN_BASE (self), FALSE);
+    g_return_val_if_fail (modem != NULL, FALSE);
+    g_return_val_if_fail (MM_IS_MODEM (modem), FALSE);
+
+    priv = MM_PLUGIN_BASE_GET_PRIVATE (self);
+
+    device = mm_modem_get_device (modem);
+    if (g_hash_table_lookup (priv->modems, device))
+        return FALSE;
+
+    g_object_weak_ref (G_OBJECT (modem), modem_destroyed, self);
+    g_hash_table_insert (priv->modems, g_strdup (device), modem);
+    return TRUE;
+}
+
+MMModem *
+mm_plugin_base_find_modem (MMPluginBase *self,
+                           const char *master_device)
+{
+    MMPluginBasePrivate *priv;
+
+    g_return_val_if_fail (self != NULL, NULL);
+    g_return_val_if_fail (MM_IS_PLUGIN_BASE (self), NULL);
+    g_return_val_if_fail (master_device != NULL, NULL);
+    g_return_val_if_fail (strlen (master_device) > 0, NULL);
+
+    priv = MM_PLUGIN_BASE_GET_PRIVATE (self);
+    return g_hash_table_lookup (priv->modems, master_device);
+}
+
+/*****************************************************************************/
+
+static void
+mm_plugin_base_init (MMPluginBase *self)
+{
+    MMPluginBasePrivate *priv = MM_PLUGIN_BASE_GET_PRIVATE (self);
+
+    priv->modems = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, NULL);
+}
+
+static void
+finalize (GObject *object)
+{
+    MMPluginBasePrivate *priv = MM_PLUGIN_BASE_GET_PRIVATE (object);
+
+    g_hash_table_destroy (priv->modems);
+
+    G_OBJECT_CLASS (mm_plugin_base_parent_class)->finalize (object);
+}
+
+static void
+mm_plugin_base_class_init (MMPluginBaseClass *klass)
+{
+    GObjectClass *object_class = G_OBJECT_CLASS (klass);
+
+    g_type_class_add_private (object_class, sizeof (MMPluginBasePrivate));
+
+    /* Virtual methods */
+    object_class->finalize = finalize;
+}
diff --git a/src/mm-plugin-base.h b/src/mm-plugin-base.h
new file mode 100644
index 00000000..0a83d671
--- /dev/null
+++ b/src/mm-plugin-base.h
@@ -0,0 +1,52 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/*
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details:
+ *
+ * Copyright (C) 2009 Red Hat, Inc.
+ */
+
+#ifndef MM_PLUGIN_BASE_H
+#define MM_PLUGIN_BASE_H
+
+#include <glib.h>
+#include <glib/gtypes.h>
+#include <glib-object.h>
+
+#include "mm-modem.h"
+
+#define MM_TYPE_PLUGIN_BASE            (mm_plugin_base_get_type ())
+#define MM_PLUGIN_BASE(obj)            (G_TYPE_CHECK_INSTANCE_CAST ((obj), MM_TYPE_PLUGIN_BASE, MMPluginBase))
+#define MM_PLUGIN_BASE_CLASS(klass)    (G_TYPE_CHECK_CLASS_CAST ((klass),  MM_TYPE_PLUGIN_BASE, MMPluginBaseClass))
+#define MM_IS_PLUGIN_BASE(obj)         (G_TYPE_CHECK_INSTANCE_TYPE ((obj), MM_TYPE_PLUGIN_BASE))
+#define MM_IS_PLUBIN_BASE_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE ((klass),  MM_TYPE_PLUGIN_BASE))
+#define MM_PLUGIN_BASE_GET_CLASS(obj)  (G_TYPE_INSTANCE_GET_CLASS ((obj),  MM_TYPE_PLUGIN_BASE, MMPluginBaseClass))
+
+typedef struct _MMPluginBase MMPluginBase;
+typedef struct _MMPluginBaseClass MMPluginBaseClass;
+
+struct _MMPluginBase {
+    GObject parent;
+};
+
+struct _MMPluginBaseClass {
+    GObjectClass parent;
+};
+
+GType mm_plugin_base_get_type (void);
+
+gboolean mm_plugin_base_add_modem (MMPluginBase *self,
+                                   MMModem *modem);
+
+MMModem *mm_plugin_base_find_modem (MMPluginBase *self,
+                                    const char *master_device);
+
+#endif /* MM_PLUGIN_BASE_H */
+