Document libbabeltrace2's C API

[babeltrace.git] / src / lib / graph / message / message.c

diff --git a/src/lib/graph/message/message.c b/src/lib/graph/message/message.c
index 285bf8f8367e88d46ab17748d70766ab3aee9ca5..98d2eab24950e9a7269a93cd9b506231d973b062 100644 (file)
--- a/src/lib/graph/message/message.c
+++ b/src/lib/graph/message/message.c
@@ -27,7 +27,7 @@
 #include "common/assert.h"
 #include "lib/assert-pre.h"
 #include "lib/assert-post.h"
-#include <babeltrace2/graph/message-const.h>
+#include <babeltrace2/graph/message.h>
 #include "lib/graph/message/message.h"
 #include "lib/graph/graph.h"
 
@@ -37,7 +37,6 @@ void bt_message_init(struct bt_message *message,
                bt_object_release_func release,
                struct bt_graph *graph)
 {
-       BT_ASSERT(type >= 0 && type <= BT_MESSAGE_TYPE_DISCARDED_PACKETS);
        message->type = type;
        bt_object_init_shared(&message->base, release);
        message->graph = graph;
@@ -50,7 +49,7 @@ void bt_message_init(struct bt_message *message,
 enum bt_message_type bt_message_get_type(
                const struct bt_message *message)
 {
-       BT_ASSERT_PRE_NON_NULL(message, "Message");
+       BT_ASSERT_PRE_DEV_NON_NULL(message, "Message");
        return message->type;
 }