- if (class->type == BT_COMPONENT_CLASS_TYPE_SOURCE) {
- struct bt_component_class_source *class_src
- = container_of(class, struct bt_component_class_source,
- parent);
-
- BT_ASSERT(class_src->msg_iter_cls);
- bt_message_iterator_class_put_ref(class_src->msg_iter_cls);
- class_src->msg_iter_cls = NULL;
- } else if (class->type == BT_COMPONENT_CLASS_TYPE_FILTER) {
- struct bt_component_class_filter *class_flt
- = container_of(class, struct bt_component_class_filter,
- parent);
-
- BT_ASSERT(class_flt->msg_iter_cls);
- bt_message_iterator_class_put_ref(class_flt->msg_iter_cls);
- class_flt->msg_iter_cls = NULL;
+ if (bt_component_class_has_message_iterator_class(class)) {
+ struct bt_component_class_with_iterator_class *class_with_iter_class =
+ container_of(class, struct bt_component_class_with_iterator_class, parent);
+
+ BT_ASSERT(class_with_iter_class->msg_iter_cls);
+ bt_message_iterator_class_put_ref(class_with_iter_class->msg_iter_cls);
+ class_with_iter_class->msg_iter_cls = NULL;