X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fnotification-packet.h;h=6b237c3b2e93fa1eef2083022212652198e6d334;hb=094ff7c009937bb23c056333baffe734308a6b06;hp=38c4f6e6f1f3a5b75ac1580cff3bf4b8e5d3b7c8;hpb=50842bdc4c21f3de2b63e29cdac730af8b6dcca6;p=babeltrace.git

diff --git a/include/babeltrace/graph/notification-packet.h b/include/babeltrace/graph/notification-packet.h
index 38c4f6e6..6b237c3b 100644
--- a/include/babeltrace/graph/notification-packet.h
+++ b/include/babeltrace/graph/notification-packet.h
@@ -27,6 +27,9 @@
  * SOFTWARE.
  */
 
+/* For bt_get() */
+#include <babeltrace/ref.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -40,14 +43,26 @@ extern struct bt_notification *bt_notification_packet_begin_create(
 extern struct bt_notification *bt_notification_packet_end_create(
 		struct bt_packet *packet);
 
-/*** BT_NOTIFICATION_TYPE_PACKET_BEGIN ***/
-extern struct bt_packet *bt_notification_packet_begin_get_packet(
+extern struct bt_packet *bt_notification_packet_begin_borrow_packet(
 		struct bt_notification *notification);
 
-/*** BT_NOTIFICATION_TYPE_PACKET_END ***/
-extern struct bt_packet *bt_notification_packet_end_get_packet(
+static inline
+struct bt_packet *bt_notification_packet_begin_get_packet(
+		struct bt_notification *notification)
+{
+	return bt_get(bt_notification_packet_begin_borrow_packet(notification));
+}
+
+extern struct bt_packet *bt_notification_packet_end_borrow_packet(
 		struct bt_notification *notification);
 
+static inline
+struct bt_packet *bt_notification_packet_end_get_packet(
+		struct bt_notification *notification)
+{
+	return bt_get(bt_notification_packet_end_borrow_packet(notification));
+}
+
 #ifdef __cplusplus
 }
 #endif