X-Git-Url: https://git.efficios.com/?a=blobdiff_plain;f=lib%2Fgraph%2Fnotification%2Fnotification.c;h=8d045fa5feceb3f0012f689ee53e8f2c37816628;hb=f6ccaed94e575af57fe6bf38154771bee4871a2a;hp=692f740cb95bb402d85453654f7d5ac15663b6db;hpb=464ebc311d460b29f681703aea0aa00eef9e6475;p=babeltrace.git

diff --git a/lib/graph/notification/notification.c b/lib/graph/notification/notification.c
index 692f740c..8d045fa5 100644
--- a/lib/graph/notification/notification.c
+++ b/lib/graph/notification/notification.c
@@ -24,14 +24,19 @@
  * SOFTWARE.
  */
 
+#define BT_LOG_TAG "NOTIF"
+#include <babeltrace/lib-logging-internal.h>
+
 #include <babeltrace/graph/notification-internal.h>
+#include <babeltrace/assert-internal.h>
+#include <babeltrace/assert-pre-internal.h>
 
 BT_HIDDEN
 void bt_notification_init(struct bt_notification *notification,
 		enum bt_notification_type type,
 		bt_object_release_func release)
 {
-	assert(type > BT_NOTIFICATION_TYPE_ALL &&
+	BT_ASSERT(type > BT_NOTIFICATION_TYPE_ALL &&
 			type < BT_NOTIFICATION_TYPE_NR);
 	notification->type = type;
 	bt_object_init(&notification->base, release);
@@ -40,5 +45,6 @@ void bt_notification_init(struct bt_notification *notification,
 enum bt_notification_type bt_notification_get_type(
 		struct bt_notification *notification)
 {
-	return notification ? notification->type : BT_NOTIFICATION_TYPE_UNKNOWN;
+	BT_ASSERT_PRE_NON_NULL(notification, "Notification");
+	return notification->type;
 }