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

#ifndef BABELTRACE_COMPONENT_NOTIFICATION_ITERATOR_INTERNAL_H
#define BABELTRACE_COMPONENT_NOTIFICATION_ITERATOR_INTERNAL_H

/*
 * BabelTrace - Notification Iterator Internal
 *
 * Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 * Copyright 2017 Philippe Proulx <pproulx@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
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <babeltrace/babeltrace-internal.h>
#include <babeltrace/object-internal.h>
#include <babeltrace/ref-internal.h>
#include <babeltrace/graph/connection.h>
#include <babeltrace/graph/notification.h>
#include <babeltrace/graph/notification-iterator.h>
#include <babeltrace/graph/private-connection-private-notification-iterator.h>
#include <babeltrace/types.h>
#include <babeltrace/assert-internal.h>
#include <stdbool.h>

struct bt_port;
struct bt_graph;

enum bt_notification_iterator_type {
	BT_NOTIFICATION_ITERATOR_TYPE_PRIVATE_CONNECTION,
	BT_NOTIFICATION_ITERATOR_TYPE_OUTPUT_PORT,
};

enum bt_private_connection_notification_iterator_notif_type {
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_EVENT =			(1U << 0),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_INACTIVITY =		(1U << 1),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_BEGIN =		(1U << 2),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_END =		(1U << 3),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_BEGIN =		(1U << 4),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_END =		(1U << 5),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_DISCARDED_EVENTS =	(1U << 6),
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_DISCARDED_PACKETS =	(1U << 7),
};

enum bt_private_connection_notification_iterator_state {
	/* Iterator is not initialized. */
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_NON_INITIALIZED,

	/* Iterator is active, not at the end yet, and not finalized. */
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ACTIVE,

	/*
	 * Iterator is ended, not finalized yet: the "next" method
	 * returns BT_NOTIFICATION_ITERATOR_STATUS_END.
	 */
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ENDED,

	/*
	 * Iterator is finalized, but not at the end yet. This means
	 * that the "next" method can still return queued notifications
	 * before returning the BT_NOTIFICATION_ITERATOR_STATUS_CANCELED
	 * status.
	 */
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED,

	/*
	 * Iterator is finalized and ended: the "next" method always
	 * returns BT_NOTIFICATION_ITERATOR_STATUS_CANCELED.
	 */
	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED,
};

struct bt_notification_iterator {
	struct bt_object base;
	enum bt_notification_iterator_type type;
	struct bt_notification *current_notification; /* owned by this */
};

struct bt_notification_iterator_private_connection {
	struct bt_notification_iterator base;
	struct bt_component *upstream_component; /* Weak */
	struct bt_port *upstream_port; /* Weak */
	struct bt_connection *connection; /* Weak */

	/*
	 * This hash table keeps the state of a stream as viewed by
	 * this notification iterator. This is used to, in developer
	 * mode:
	 *
	 * * Automatically enqueue "stream begin", "packet begin",
	 *   "packet end", and "stream end" notifications depending
	 *   on the stream's state and on the next notification returned
	 *   by the upstream component.
	 *
	 * * Make sure that, once the notification iterator has seen a
	 *   "stream end" notification for a given stream, no other
	 *   notifications which refer to this stream can be delivered
	 *   by this iterator.
	 *
	 * The key (struct bt_stream *) is not owned by this. The
	 * value is an allocated state structure.
	 */
	GHashTable *stream_states;

	enum bt_private_connection_notification_iterator_state state;
	void *user_data;
};

struct bt_notification_iterator_output_port {
	struct bt_notification_iterator base;
	struct bt_graph *graph; /* Owned by this */
	struct bt_component *colander; /* Owned by this */
	struct bt_port *output_port; /* Owned by this */
};

static inline
struct bt_notification *bt_notification_iterator_borrow_current_notification(
		struct bt_notification_iterator *iterator)
{
	BT_ASSERT(iterator);
	return iterator->current_notification;
}

static inline
void bt_notification_iterator_replace_current_notification(
		struct bt_notification_iterator *iterator,
		struct bt_notification *notification)
{
	BT_ASSERT(iterator);
	bt_put(iterator->current_notification);
	iterator->current_notification = bt_get(notification);
}

static inline
struct bt_notification_iterator_private_connection *
bt_private_connection_notification_iterator_borrow_from_private(
		struct bt_private_connection_private_notification_iterator *private_notification_iterator)
{
	return (void *) private_notification_iterator;
}

static inline
struct bt_private_connection_private_notification_iterator *
bt_private_connection_private_notification_iterator_from_notification_iterator(
		struct bt_notification_iterator_private_connection *iterator)
{
	return (void *) iterator;
}

BT_HIDDEN
enum bt_connection_status bt_private_connection_notification_iterator_create(
		struct bt_component *upstream_comp,
		struct bt_port *upstream_port,
		struct bt_connection *connection,
		struct bt_notification_iterator_private_connection **iterator);

BT_HIDDEN
void bt_private_connection_notification_iterator_finalize(
		struct bt_notification_iterator_private_connection *iterator);

BT_HIDDEN
void bt_private_connection_notification_iterator_set_connection(
		struct bt_notification_iterator_private_connection *iterator,
		struct bt_connection *connection);

static inline
const char *bt_notification_iterator_status_string(
		enum bt_notification_iterator_status status)
{
	switch (status) {
	case BT_NOTIFICATION_ITERATOR_STATUS_CANCELED:
		return "BT_NOTIFICATION_ITERATOR_STATUS_CANCELED";
	case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
		return "BT_NOTIFICATION_ITERATOR_STATUS_AGAIN";
	case BT_NOTIFICATION_ITERATOR_STATUS_END:
		return "BT_NOTIFICATION_ITERATOR_STATUS_END";
	case BT_NOTIFICATION_ITERATOR_STATUS_OK:
		return "BT_NOTIFICATION_ITERATOR_STATUS_OK";
	case BT_NOTIFICATION_ITERATOR_STATUS_INVALID:
		return "BT_NOTIFICATION_ITERATOR_STATUS_INVALID";
	case BT_NOTIFICATION_ITERATOR_STATUS_ERROR:
		return "BT_NOTIFICATION_ITERATOR_STATUS_ERROR";
	case BT_NOTIFICATION_ITERATOR_STATUS_NOMEM:
		return "BT_NOTIFICATION_ITERATOR_STATUS_NOMEM";
	case BT_NOTIFICATION_ITERATOR_STATUS_UNSUPPORTED:
		return "BT_NOTIFICATION_ITERATOR_STATUS_UNSUPPORTED";
	default:
		return "(unknown)";
	}
};

static inline
const char *bt_private_connection_notification_iterator_state_string(
		enum bt_private_connection_notification_iterator_state state)
{
	switch (state) {
	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ACTIVE:
		return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ACTIVE";
	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ENDED:
		return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ENDED";
	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED:
		return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED";
	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED:
		return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED";
	default:
		return "(unknown)";
	}
};

#endif /* BABELTRACE_COMPONENT_NOTIFICATION_ITERATOR_INTERNAL_H */
Commit	Line	Data
33b34c43 PP	1	#ifndef BABELTRACE_COMPONENT_NOTIFICATION_ITERATOR_INTERNAL_H
33b34c43 PP	2	#define BABELTRACE_COMPONENT_NOTIFICATION_ITERATOR_INTERNAL_H
47e5a032 JG	3
	4	/*
	5	* BabelTrace - Notification Iterator Internal
	6	*
	7	* Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com>
3230ee6b	8	* Copyright 2017 Philippe Proulx <pproulx@efficios.com>
47e5a032 JG	9	*
	10	* Permission is hereby granted, free of charge, to any person obtaining a copy
	11	* of this software and associated documentation files (the "Software"), to deal
	12	* in the Software without restriction, including without limitation the rights
	13	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	14	* copies of the Software, and to permit persons to whom the Software is
	15	* furnished to do so, subject to the following conditions:
	16	*
	17	* The above copyright notice and this permission notice shall be included in
	18	* all copies or substantial portions of the Software.
	19	*
	20	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	21	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	22	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	23	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	24	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	25	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	26	* SOFTWARE.
	27	*/
	28
	29	#include <babeltrace/babeltrace-internal.h>
890882ef	30	#include <babeltrace/object-internal.h>
fb2dcc52	31	#include <babeltrace/ref-internal.h>
bd14d768	32	#include <babeltrace/graph/connection.h>
b2e0c907 PP	33	#include <babeltrace/graph/notification.h>
b2e0c907 PP	34	#include <babeltrace/graph/notification-iterator.h>
90157d89	35	#include <babeltrace/graph/private-connection-private-notification-iterator.h>
c55a9f58	36	#include <babeltrace/types.h>
f6ccaed9	37	#include <babeltrace/assert-internal.h>
088d4023	38	#include <stdbool.h>
47e5a032	39
3230ee6b	40	struct bt_port;
8ed535b5	41	struct bt_graph;
3230ee6b	42
90157d89 PP	43	enum bt_notification_iterator_type {
90157d89 PP	44	BT_NOTIFICATION_ITERATOR_TYPE_PRIVATE_CONNECTION,
8ed535b5	45	BT_NOTIFICATION_ITERATOR_TYPE_OUTPUT_PORT,
fa054faf PP	46	};
fa054faf PP	47
90157d89 PP	48	enum bt_private_connection_notification_iterator_notif_type {
	49	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_EVENT = (1U << 0),
	50	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_INACTIVITY = (1U << 1),
	51	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_BEGIN = (1U << 2),
	52	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_END = (1U << 3),
	53	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_BEGIN = (1U << 4),
	54	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_END = (1U << 5),
	55	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_DISCARDED_EVENTS = (1U << 6),
	56	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_NOTIF_TYPE_DISCARDED_PACKETS = (1U << 7),
	57	};
	58
	59	enum bt_private_connection_notification_iterator_state {
088d4023	60	/* Iterator is not initialized. */
90157d89	61	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_NON_INITIALIZED,
088d4023	62
bd14d768	63	/* Iterator is active, not at the end yet, and not finalized. */
90157d89	64	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ACTIVE,
bd14d768 PP	65
	66	/*
	67	* Iterator is ended, not finalized yet: the "next" method
	68	* returns BT_NOTIFICATION_ITERATOR_STATUS_END.
	69	*/
90157d89	70	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ENDED,
bd14d768 PP	71
	72	/*
	73	* Iterator is finalized, but not at the end yet. This means
	74	* that the "next" method can still return queued notifications
	75	* before returning the BT_NOTIFICATION_ITERATOR_STATUS_CANCELED
	76	* status.
	77	*/
90157d89	78	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED,
bd14d768 PP	79
	80	/*
	81	* Iterator is finalized and ended: the "next" method always
	82	* returns BT_NOTIFICATION_ITERATOR_STATUS_CANCELED.
	83	*/
90157d89	84	BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED,
bd14d768 PP	85	};
bd14d768 PP	86
47e5a032	87	struct bt_notification_iterator {
b8a06801	88	struct bt_object base;
90157d89	89	enum bt_notification_iterator_type type;
8ed535b5	90	struct bt_notification current_notification; / owned by this */
90157d89 PP	91	};
	92
	93	struct bt_notification_iterator_private_connection {
	94	struct bt_notification_iterator base;
bd14d768 PP	95	struct bt_component upstream_component; / Weak */
	96	struct bt_port upstream_port; / Weak */
	97	struct bt_connection connection; / Weak */
3230ee6b PP	98
	99	/*
	100	* This hash table keeps the state of a stream as viewed by
f42867e2 PP	101	* this notification iterator. This is used to, in developer
f42867e2 PP	102	* mode:
3230ee6b PP	103	*
	104	* * Automatically enqueue "stream begin", "packet begin",
	105	* "packet end", and "stream end" notifications depending
	106	* on the stream's state and on the next notification returned
	107	* by the upstream component.
	108	*
f42867e2 PP	109	* * Make sure that, once the notification iterator has seen a
	110	* "stream end" notification for a given stream, no other
	111	* notifications which refer to this stream can be delivered
	112	* by this iterator.
3230ee6b	113	*
50842bdc	114	* The key (struct bt_stream *) is not owned by this. The
3230ee6b PP	115	* value is an allocated state structure.
	116	*/
	117	GHashTable *stream_states;
	118
90157d89	119	enum bt_private_connection_notification_iterator_state state;
47e5a032 JG	120	void *user_data;
	121	};
	122
8ed535b5 PP	123	struct bt_notification_iterator_output_port {
	124	struct bt_notification_iterator base;
	125	struct bt_graph graph; / Owned by this */
	126	struct bt_component colander; / Owned by this */
	127	struct bt_port output_port; / Owned by this */
	128	};
	129
	130	static inline
	131	struct bt_notification *bt_notification_iterator_borrow_current_notification(
	132	struct bt_notification_iterator *iterator)
	133	{
f6ccaed9	134	BT_ASSERT(iterator);
8ed535b5 PP	135	return iterator->current_notification;
	136	}
	137
	138	static inline
	139	void bt_notification_iterator_replace_current_notification(
	140	struct bt_notification_iterator *iterator,
	141	struct bt_notification *notification)
	142	{
f6ccaed9	143	BT_ASSERT(iterator);
8ed535b5 PP	144	bt_put(iterator->current_notification);
	145	iterator->current_notification = bt_get(notification);
	146	}
	147
890882ef	148	static inline
90157d89	149	struct bt_notification_iterator_private_connection *
6d137876	150	bt_private_connection_notification_iterator_borrow_from_private(
90157d89	151	struct bt_private_connection_private_notification_iterator *private_notification_iterator)
890882ef PP	152	{
	153	return (void *) private_notification_iterator;
	154	}
	155
	156	static inline
90157d89 PP	157	struct bt_private_connection_private_notification_iterator *
	158	bt_private_connection_private_notification_iterator_from_notification_iterator(
	159	struct bt_notification_iterator_private_connection *iterator)
890882ef	160	{
90157d89	161	return (void *) iterator;
890882ef PP	162	}
890882ef PP	163
47e5a032	164	BT_HIDDEN
90157d89	165	enum bt_connection_status bt_private_connection_notification_iterator_create(
73d5c1ad	166	struct bt_component *upstream_comp,
fa054faf	167	struct bt_port *upstream_port,
73d5c1ad	168	struct bt_connection *connection,
90157d89	169	struct bt_notification_iterator_private_connection **iterator);
47e5a032	170
bd14d768	171	BT_HIDDEN
90157d89 PP	172	void bt_private_connection_notification_iterator_finalize(
90157d89 PP	173	struct bt_notification_iterator_private_connection *iterator);
bd14d768 PP	174
bd14d768 PP	175	BT_HIDDEN
90157d89 PP	176	void bt_private_connection_notification_iterator_set_connection(
90157d89 PP	177	struct bt_notification_iterator_private_connection *iterator,
bd14d768 PP	178	struct bt_connection *connection);
bd14d768 PP	179
a36bfb16 PP	180	static inline
	181	const char *bt_notification_iterator_status_string(
	182	enum bt_notification_iterator_status status)
	183	{
	184	switch (status) {
	185	case BT_NOTIFICATION_ITERATOR_STATUS_CANCELED:
	186	return "BT_NOTIFICATION_ITERATOR_STATUS_CANCELED";
	187	case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
	188	return "BT_NOTIFICATION_ITERATOR_STATUS_AGAIN";
	189	case BT_NOTIFICATION_ITERATOR_STATUS_END:
	190	return "BT_NOTIFICATION_ITERATOR_STATUS_END";
	191	case BT_NOTIFICATION_ITERATOR_STATUS_OK:
	192	return "BT_NOTIFICATION_ITERATOR_STATUS_OK";
	193	case BT_NOTIFICATION_ITERATOR_STATUS_INVALID:
	194	return "BT_NOTIFICATION_ITERATOR_STATUS_INVALID";
	195	case BT_NOTIFICATION_ITERATOR_STATUS_ERROR:
	196	return "BT_NOTIFICATION_ITERATOR_STATUS_ERROR";
	197	case BT_NOTIFICATION_ITERATOR_STATUS_NOMEM:
	198	return "BT_NOTIFICATION_ITERATOR_STATUS_NOMEM";
	199	case BT_NOTIFICATION_ITERATOR_STATUS_UNSUPPORTED:
	200	return "BT_NOTIFICATION_ITERATOR_STATUS_UNSUPPORTED";
	201	default:
	202	return "(unknown)";
	203	}
	204	};
	205
8f9d7550	206	static inline
90157d89 PP	207	const char *bt_private_connection_notification_iterator_state_string(
90157d89 PP	208	enum bt_private_connection_notification_iterator_state state)
8f9d7550 PP	209	{
8f9d7550 PP	210	switch (state) {
90157d89 PP	211	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ACTIVE:
	212	return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ACTIVE";
	213	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ENDED:
	214	return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_ENDED";
	215	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED:
	216	return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED";
	217	case BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED:
	218	return "BT_PRIVATE_CONNECTION_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED";
8f9d7550 PP	219	default:
	220	return "(unknown)";
	221	}
	222	};
	223
33b34c43	224	#endif /* BABELTRACE_COMPONENT_NOTIFICATION_ITERATOR_INTERNAL_H */