X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Fbindings%2Fpython%2Fbt2%2Fbt2%2Fnative_bt_message_iterator.i.h;h=deb149cfafd1d132eb655e472aebb222da0d6f74;hb=98b15851a941e7342b8bb19e265cdc3a40fabfb8;hp=f687dc9a480c01ff9eed498a432a812ba17d8420;hpb=6c373cc905e907ecbad698fee38db1d47a981b14;p=babeltrace.git

diff --git a/src/bindings/python/bt2/bt2/native_bt_message_iterator.i.h b/src/bindings/python/bt2/bt2/native_bt_message_iterator.i.h
index f687dc9a..deb149cf 100644
--- a/src/bindings/python/bt2/bt2/native_bt_message_iterator.i.h
+++ b/src/bindings/python/bt2/bt2/native_bt_message_iterator.i.h
@@ -22,15 +22,53 @@
  * THE SOFTWARE.
  */
 
+bt_self_component_port_input_message_iterator_create_from_message_iterator_status
+bt_bt2_self_component_port_input_message_iterator_create_from_message_iterator(
+		bt_self_message_iterator *self_msg_iter,
+		bt_self_component_port_input *input_port,
+		bt_self_component_port_input_message_iterator **message_iterator)
+{
+	bt_self_component_port_input_message_iterator_create_from_message_iterator_status
+		status;
+
+	status = bt_self_component_port_input_message_iterator_create_from_message_iterator(
+		self_msg_iter, input_port, message_iterator);
+
+	if (status != BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_CREATE_FROM_MESSAGE_ITERATOR_STATUS_OK) {
+		*message_iterator = NULL;
+	}
+
+	return status;
+}
+
+bt_self_component_port_input_message_iterator_create_from_sink_component_status
+bt_bt2_self_component_port_input_message_iterator_create_from_sink_component(
+		bt_self_component_sink *self_comp,
+		bt_self_component_port_input *input_port,
+		bt_self_component_port_input_message_iterator **message_iterator)
+{
+	bt_self_component_port_input_message_iterator_create_from_sink_component_status
+		status;
+
+	status = bt_self_component_port_input_message_iterator_create_from_sink_component(
+		self_comp, input_port, message_iterator);
+
+	if (status != BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK) {
+		*message_iterator = NULL;
+	}
+
+	return status;
+}
+
 static PyObject *bt_bt2_get_user_component_from_user_msg_iter(
 		bt_self_message_iterator *self_message_iterator)
 {
 	bt_self_component *self_component = bt_self_message_iterator_borrow_component(self_message_iterator);
 	PyObject *py_comp;
 
-	BT_ASSERT(self_component);
+	BT_ASSERT_DBG(self_component);
 	py_comp = bt_self_component_get_data(self_component);
-	BT_ASSERT(py_comp);
+	BT_ASSERT_DBG(py_comp);
 
 	/* Return new reference */
 	Py_INCREF(py_comp);