Add bt_component_class_freeze()

[babeltrace.git] / lib / component / component.c

diff --git a/lib/component/component.c b/lib/component/component.c
index d82ee0d8fd78d284ef7624fe357689e76b61ff2c..c188261c7554bcca85088e132359bf45d19c952f 100644 (file)
--- a/lib/component/component.c
+++ b/lib/component/component.c
@@ -29,10 +29,10 @@
 #include <babeltrace/component/component.h>
 #include <babeltrace/component/component-internal.h>
 #include <babeltrace/component/component-class-internal.h>
-#include <babeltrace/component/source-internal.h>
-#include <babeltrace/component/filter-internal.h>
+#include <babeltrace/component/component-source-internal.h>
+#include <babeltrace/component/component-filter-internal.h>
+#include <babeltrace/component/component-sink-internal.h>
 #include <babeltrace/component/notification/iterator-internal.h>
-#include <babeltrace/component/sink-internal.h>
 #include <babeltrace/babeltrace-internal.h>
 #include <babeltrace/compiler.h>
 #include <babeltrace/ref.h>
@@ -225,6 +225,8 @@ struct bt_component *bt_component_create(
                BT_PUT(component);
                goto end;
        }
+
+       bt_component_class_freeze(component->class);
 end:
        return component;
 }