X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fself-component-class-source.h;h=f78b4f78701369ebe178b36c52a80f3e47a2c037;hb=7e69c04f557e46fb50e5a888a3bc3451afbd3b8e;hp=0cdbc8668724e8cd0851242ccd832aa612a46df0;hpb=707b7d3548cc010d582f7faed43630aed26508fd;p=babeltrace.git

diff --git a/include/babeltrace/graph/self-component-class-source.h b/include/babeltrace/graph/self-component-class-source.h
index 0cdbc866..f78b4f78 100644
--- a/include/babeltrace/graph/self-component-class-source.h
+++ b/include/babeltrace/graph/self-component-class-source.h
@@ -2,6 +2,7 @@
 #define BABELTRACE_GRAPH_SELF_COMPONENT_CLASS_SOURCE_H
 
 /*
+ * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
  * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
@@ -23,19 +24,23 @@
  * SOFTWARE.
  */
 
+/*
+ * For bt_component_class_source, bt_self_component_class_source,
+ * __BT_UPCAST_CONST
+ */
+#include <babeltrace/types.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-struct bt_component_class_source;
-struct bt_self_component_class_source;
-
 static inline
-struct bt_component_class_source *
+const bt_component_class_source *
 bt_self_component_class_source_as_component_class_source(
-		struct bt_self_component_class_source *self_comp_cls_source)
+		bt_self_component_class_source *self_comp_cls_source)
 {
-	return (void *) self_comp_cls_source;
+	return __BT_UPCAST_CONST(bt_component_class_source,
+		self_comp_cls_source);
 }
 
 #ifdef __cplusplus