Component creation

[babeltrace.git] / plugins / source.c

diff --git a/plugins/source.c b/plugins/source.c
index 094cab24cb2c1ac39de06b3635ffd263190f9bcf..6e76206afb6e00c26627e044d44df098aba08eaf 100644 (file)
--- a/plugins/source.c
+++ b/plugins/source.c
@@ -26,6 +26,7 @@
  * SOFTWARE.
  */
 
+#include <babeltrace/ref.h>
 #include <babeltrace/compiler.h>
 #include <babeltrace/plugin/source-internal.h>
 #include <babeltrace/plugin/component-internal.h>
@@ -46,27 +47,28 @@ void bt_component_source_destroy(struct bt_component *component)
 }
 
 BT_HIDDEN
-struct bt_component *bt_component_source_create(const char *name)
+struct bt_component *bt_component_source_create(
+               struct bt_component_class *class, const char *name)
 {
        struct bt_component_source *source = NULL;
        enum bt_component_status ret;
+
        source = g_new0(struct bt_component_source, 1);
        if (!source) {
                goto end;
        }
 
-       ret = bt_component_init(&source->parent, name,
-               BT_COMPONENT_TYPE_SOURCE, bt_component_source_destroy);
+       ret = bt_component_init(&source->parent, class, name,
+               bt_component_source_destroy);
        if (ret != BT_COMPONENT_STATUS_OK) {
-               g_free(source);
-               source = NULL;
+               BT_PUT(source);
                goto end;
        }
 end:
        return source ? &source->parent : NULL;
 }
 
-struct bt_notification_iterator *bt_plugin_source_create_iterator(
+struct bt_notification_iterator *bt_component_source_create_iterator(
                struct bt_component *component)
 {
        enum bt_component_status ret_component;
@@ -101,6 +103,6 @@ struct bt_notification_iterator *bt_plugin_source_create_iterator(
 end:
        return iterator;
 error:
-       bt_notification_iterator_put(iterator);
-       return NULL;
+       BT_PUT(iterator);
+       return iterator;
 }