lib: make public reference count functions have strict types

[babeltrace.git] / lib / graph / notification / notification.c

diff --git a/lib/graph/notification/notification.c b/lib/graph/notification/notification.c
index 2b8c12d77ffcbacd2a503af35dbbc654885ab5ab..7a2d859b073fef1e47645e6ff46f070af2d6379f 100644 (file)
--- a/lib/graph/notification/notification.c
+++ b/lib/graph/notification/notification.c
@@ -1,8 +1,7 @@
 /*
+ * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
  * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
- * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
- *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to deal
  * in the Software without restriction, including without limitation the rights
@@ -25,11 +24,11 @@
 #define BT_LOG_TAG "NOTIF"
 #include <babeltrace/lib-logging-internal.h>
 
-#include <babeltrace/graph/private-notification.h>
-#include <babeltrace/graph/notification-internal.h>
-#include <babeltrace/graph/graph-internal.h>
 #include <babeltrace/assert-internal.h>
 #include <babeltrace/assert-pre-internal.h>
+#include <babeltrace/graph/notification-const.h>
+#include <babeltrace/graph/notification-internal.h>
+#include <babeltrace/graph/graph-internal.h>
 
 BT_ASSERT_PRE_FUNC
 static inline void _init_seq_num(struct bt_notification *notification)
@@ -49,7 +48,8 @@ void bt_notification_init(struct bt_notification *notification,
                bt_object_release_func release,
                struct bt_graph *graph)
 {
-       BT_ASSERT(type >= 0 && type < BT_NOTIFICATION_TYPE_NR);
+       BT_ASSERT(type >= 0 &&
+               type <= BT_NOTIFICATION_TYPE_PACKET_END);
        notification->type = type;
        init_seq_num(notification);
        bt_object_init_shared(&notification->base, release);
@@ -61,7 +61,7 @@ void bt_notification_init(struct bt_notification *notification,
 }
 
 enum bt_notification_type bt_notification_get_type(
-               struct bt_notification *notification)
+               const struct bt_notification *notification)
 {
        BT_ASSERT_PRE_NON_NULL(notification, "Notification");
        return notification->type;
@@ -73,3 +73,13 @@ void bt_notification_unlink_graph(struct bt_notification *notif)
        BT_ASSERT(notif);
        notif->graph = NULL;
 }
+
+void bt_notification_get_ref(const struct bt_notification *notification)
+{
+       bt_object_get_ref(notification);
+}
+
+void bt_notification_put_ref(const struct bt_notification *notification)
+{
+       bt_object_put_ref(notification);
+}