Add internal BT_ASSERT() and BT_ASSERT_PRE() helpers

[babeltrace.git] / include / babeltrace / graph / notification-packet-internal.h

diff --git a/include/babeltrace/graph/notification-packet-internal.h b/include/babeltrace/graph/notification-packet-internal.h
index 8d525c55f673d6b3dbfdf163853d164c1a5354ae..7164d55baf07349f09df45143118387005f8d3eb 100644 (file)
--- a/include/babeltrace/graph/notification-packet-internal.h
+++ b/include/babeltrace/graph/notification-packet-internal.h
@@ -27,17 +27,42 @@
  * SOFTWARE.
  */
 
+#include <babeltrace/compiler-internal.h>
 #include <babeltrace/ctf-ir/packet.h>
 #include <babeltrace/graph/notification-internal.h>
 
 struct bt_notification_packet_begin {
        struct bt_notification parent;
-       struct bt_ctf_packet *packet;
+       struct bt_packet *packet;
 };
 
 struct bt_notification_packet_end {
        struct bt_notification parent;
-       struct bt_ctf_packet *packet;
+       struct bt_packet *packet;
 };
 
+static inline
+struct bt_packet *bt_notification_packet_begin_borrow_packet(
+               struct bt_notification *notif)
+{
+       struct bt_notification_packet_begin *notif_packet_begin =
+               container_of(notif,
+                       struct bt_notification_packet_begin, parent);
+
+       assert(notif_packet_begin);
+       return notif_packet_begin->packet;
+}
+
+static inline
+struct bt_packet *bt_notification_packet_end_borrow_packet(
+               struct bt_notification *notif)
+{
+       struct bt_notification_packet_end *notif_packet_end =
+               container_of(notif,
+                       struct bt_notification_packet_end, parent);
+
+       assert(notif_packet_end);
+       return notif_packet_end->packet;
+}
+
 #endif /* BABELTRACE_COMPONENT_NOTIFICATION_PACKET_INTERNAL_H */