[babeltrace.git] / plugins / utils / trimmer / copy.c

/*
 * copy.c
 *
 * Babeltrace Copy Trace Structure
 *
 * Copyright 2017 Julien Desfossez <jdesfossez@efficios.com>
 *
 * Author: Julien Desfossez <jdesfossez@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/ctf-ir/event.h>
#include <babeltrace/ctf-ir/packet.h>
#include <babeltrace/ctf-ir/event-class.h>
#include <babeltrace/ctf-ir/stream.h>
#include <babeltrace/ctf-ir/stream-class.h>
#include <babeltrace/ctf-ir/clock-class.h>
#include <babeltrace/ctf-ir/fields.h>
#include <babeltrace/ctf-writer/stream-class.h>
#include <babeltrace/ctf-writer/stream.h>

#include <ctfcopytrace.h>
#include "iterator.h"

static
struct bt_ctf_packet *lookup_packet(struct trimmer_iterator *trim_it,
		struct bt_ctf_packet *packet)
{
	return (struct bt_ctf_packet *) g_hash_table_lookup(
			trim_it->packet_map,
			(gpointer) packet);
}

static
struct bt_ctf_packet *insert_new_packet(struct trimmer_iterator *trim_it,
		struct bt_ctf_packet *packet,
		struct bt_ctf_stream *stream)
{
	struct bt_ctf_packet *writer_packet = NULL;

	writer_packet = bt_ctf_packet_create(stream);
	if (!writer_packet) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		goto end;
	}
	g_hash_table_insert(trim_it->packet_map, (gpointer) packet, writer_packet);

end:
	return writer_packet;
}

BT_HIDDEN
enum bt_component_status update_packet_context_field(FILE *err,
		struct bt_ctf_packet *writer_packet,
		const char *name, int64_t value)
{
	enum bt_component_status ret;
	struct bt_ctf_field *packet_context = NULL, *writer_packet_context = NULL;
	struct bt_ctf_field_type *struct_type = NULL, *field_type = NULL;
	struct bt_ctf_field *field = NULL, *writer_field = NULL;
	int nr_fields, i, int_ret;

	packet_context = bt_ctf_packet_get_context(writer_packet);
	if (!packet_context) {
		fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
				__LINE__);
		goto error;
	}

	struct_type = bt_ctf_field_get_type(packet_context);
	if (!struct_type) {
		fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
				__LINE__);
		goto error;
	}

	writer_packet_context = bt_ctf_packet_get_context(writer_packet);
	if (!writer_packet_context) {
		fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
				__LINE__);
		goto error;
	}

	nr_fields = bt_ctf_field_type_structure_get_field_count(struct_type);
	for (i = 0; i < nr_fields; i++) {
		const char *field_name;

		field = bt_ctf_field_structure_get_field_by_index(
				packet_context, i);
		if (!field) {
			fprintf(err, "[error] %s in %s:%d\n", __func__,
					__FILE__, __LINE__);
			goto error;
		}
		if (bt_ctf_field_type_structure_get_field(struct_type,
					&field_name, &field_type, i) < 0) {
			fprintf(err, "[error] %s in %s:%d\n", __func__,
					__FILE__, __LINE__);
			goto error;
		}
		if (strcmp(field_name, name)) {
			BT_PUT(field_type);
			BT_PUT(field);
			continue;
		}
		if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
			fprintf(err, "[error] Unexpected packet context field type\n");
			goto error;
		}
		writer_field = bt_ctf_field_structure_get_field(writer_packet_context,
				field_name);
		if (!writer_field) {
			fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
					__LINE__);
			goto error;
		}

		int_ret = bt_ctf_field_unsigned_integer_set_value(writer_field, value);
		if (int_ret < 0) {
			fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
					__LINE__);
			goto error;
		}
		BT_PUT(writer_field);
		BT_PUT(field_type);
		BT_PUT(field);
	}

	ret = BT_COMPONENT_STATUS_OK;
	goto end;

error:
	bt_put(writer_field);
	bt_put(field_type);
	bt_put(field);
	ret = BT_COMPONENT_STATUS_ERROR;
end:
	bt_put(struct_type);
	bt_put(packet_context);
	return ret;
}

BT_HIDDEN
struct bt_ctf_packet *trimmer_new_packet(
		struct trimmer_iterator *trim_it,
		struct bt_ctf_packet *packet)
{
	struct bt_ctf_stream *stream = NULL;
	struct bt_ctf_packet *writer_packet = NULL;
	int int_ret;

	stream = bt_ctf_packet_get_stream(packet);
	if (!stream) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n",
				__func__, __FILE__, __LINE__);
		goto error;
	}

	/*
	 * If a packet was already opened, close it and remove it from
	 * the HT.
	 */
	writer_packet = lookup_packet(trim_it, packet);
	if (writer_packet) {
		g_hash_table_remove(trim_it->packet_map, packet);
		BT_PUT(writer_packet);
	}

	writer_packet = insert_new_packet(trim_it, packet, stream);
	if (!writer_packet) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n",
				__func__, __FILE__, __LINE__);
		goto error;
	}
	bt_get(writer_packet);

	int_ret = ctf_packet_copy_context(trim_it->err, packet,
			stream, writer_packet);
	if (int_ret < 0) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n",
				__func__, __FILE__, __LINE__);
		goto error;
	}

	goto end;

error:
	BT_PUT(writer_packet);
end:
	bt_put(stream);
	return writer_packet;
}

BT_HIDDEN
struct bt_ctf_packet *trimmer_close_packet(
		struct trimmer_iterator *trim_it,
		struct bt_ctf_packet *packet)
{
	struct bt_ctf_packet *writer_packet = NULL;

	writer_packet = lookup_packet(trim_it, packet);
	if (!writer_packet) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		goto end;
	}

	g_hash_table_remove(trim_it->packet_map, packet);

end:
	return writer_packet;
}

BT_HIDDEN
struct bt_ctf_event *trimmer_output_event(
		struct trimmer_iterator *trim_it,
		struct bt_ctf_event *event)
{
	struct bt_ctf_event_class *event_class = NULL;
	struct bt_ctf_event *writer_event = NULL;
	struct bt_ctf_packet *packet = NULL, *writer_packet = NULL;
	const char *event_name;
	int int_ret;

	event_class = bt_ctf_event_get_class(event);
	if (!event_class) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		goto error;
	}

	event_name = bt_ctf_event_class_get_name(event_class);
	if (!event_name) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		goto error;
	}

	writer_event = ctf_copy_event(trim_it->err, event, event_class, false);
	if (!writer_event) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		fprintf(trim_it->err, "[error] Failed to copy event %s\n",
				bt_ctf_event_class_get_name(event_class));
		goto error;
	}

	packet = bt_ctf_event_get_packet(event);
	if (!packet) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		goto error;
	}

	writer_packet = lookup_packet(trim_it, packet);
	if (!writer_packet) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		goto error;
	}
	bt_get(writer_packet);

	int_ret = bt_ctf_event_set_packet(writer_event, writer_packet);
	if (int_ret < 0) {
		fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
				__FILE__, __LINE__);
		fprintf(trim_it->err, "[error] Failed to append event %s\n",
				bt_ctf_event_class_get_name(event_class));
		goto error;
	}

	/* We keep the reference on the writer_event to create a notification. */
	goto end;

error:
	BT_PUT(writer_event);
end:
	bt_put(writer_packet);
	bt_put(packet);
	bt_put(event_class);
	return writer_event;
}
Commit	Line	Data
19ce87a4 JD	1	/*
	2	* copy.c
	3	*
	4	* Babeltrace Copy Trace Structure
	5	*
	6	* Copyright 2017 Julien Desfossez <jdesfossez@efficios.com>
	7	*
	8	* Author: Julien Desfossez <jdesfossez@efficios.com>
	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/ctf-ir/event.h>
	30	#include <babeltrace/ctf-ir/packet.h>
	31	#include <babeltrace/ctf-ir/event-class.h>
	32	#include <babeltrace/ctf-ir/stream.h>
	33	#include <babeltrace/ctf-ir/stream-class.h>
	34	#include <babeltrace/ctf-ir/clock-class.h>
	35	#include <babeltrace/ctf-ir/fields.h>
	36	#include <babeltrace/ctf-writer/stream-class.h>
	37	#include <babeltrace/ctf-writer/stream.h>
	38
	39	#include <ctfcopytrace.h>
	40	#include "iterator.h"
	41
	42	static
	43	struct bt_ctf_packet lookup_packet(struct trimmer_iterator trim_it,
	44	struct bt_ctf_packet *packet)
	45	{
	46	return (struct bt_ctf_packet *) g_hash_table_lookup(
	47	trim_it->packet_map,
	48	(gpointer) packet);
	49	}
	50
	51	static
	52	struct bt_ctf_packet insert_new_packet(struct trimmer_iterator trim_it,
	53	struct bt_ctf_packet *packet,
	54	struct bt_ctf_stream *stream)
	55	{
9ae49d3d	56	struct bt_ctf_packet *writer_packet = NULL;
19ce87a4 JD	57
	58	writer_packet = bt_ctf_packet_create(stream);
	59	if (!writer_packet) {
	60	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
	61	__FILE__, __LINE__);
	62	goto end;
	63	}
	64	g_hash_table_insert(trim_it->packet_map, (gpointer) packet, writer_packet);
	65
	66	end:
	67	return writer_packet;
	68	}
	69
	70	BT_HIDDEN
	71	enum bt_component_status update_packet_context_field(FILE *err,
	72	struct bt_ctf_packet *writer_packet,
	73	const char *name, int64_t value)
	74	{
	75	enum bt_component_status ret;
9ae49d3d JD	76	struct bt_ctf_field packet_context = NULL, writer_packet_context = NULL;
	77	struct bt_ctf_field_type struct_type = NULL, field_type = NULL;
	78	struct bt_ctf_field field = NULL, writer_field = NULL;
19ce87a4 JD	79	int nr_fields, i, int_ret;
	80
	81	packet_context = bt_ctf_packet_get_context(writer_packet);
	82	if (!packet_context) {
19ce87a4 JD	83	fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
19ce87a4 JD	84	__LINE__);
9ae49d3d	85	goto error;
19ce87a4 JD	86	}
	87
	88	struct_type = bt_ctf_field_get_type(packet_context);
	89	if (!struct_type) {
19ce87a4 JD	90	fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
19ce87a4 JD	91	__LINE__);
9ae49d3d	92	goto error;
19ce87a4 JD	93	}
	94
	95	writer_packet_context = bt_ctf_packet_get_context(writer_packet);
	96	if (!writer_packet_context) {
19ce87a4 JD	97	fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
19ce87a4 JD	98	__LINE__);
9ae49d3d	99	goto error;
19ce87a4 JD	100	}
	101
	102	nr_fields = bt_ctf_field_type_structure_get_field_count(struct_type);
	103	for (i = 0; i < nr_fields; i++) {
19ce87a4 JD	104	const char *field_name;
	105
	106	field = bt_ctf_field_structure_get_field_by_index(
	107	packet_context, i);
	108	if (!field) {
19ce87a4 JD	109	fprintf(err, "[error] %s in %s:%d\n", __func__,
19ce87a4 JD	110	__FILE__, __LINE__);
9ae49d3d	111	goto error;
19ce87a4 JD	112	}
	113	if (bt_ctf_field_type_structure_get_field(struct_type,
	114	&field_name, &field_type, i) < 0) {
19ce87a4 JD	115	fprintf(err, "[error] %s in %s:%d\n", __func__,
19ce87a4 JD	116	__FILE__, __LINE__);
9ae49d3d	117	goto error;
19ce87a4 JD	118	}
19ce87a4 JD	119	if (strcmp(field_name, name)) {
9ae49d3d JD	120	BT_PUT(field_type);
9ae49d3d JD	121	BT_PUT(field);
19ce87a4 JD	122	continue;
19ce87a4 JD	123	}
1487a16a	124	if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
19ce87a4	125	fprintf(err, "[error] Unexpected packet context field type\n");
9ae49d3d	126	goto error;
19ce87a4 JD	127	}
	128	writer_field = bt_ctf_field_structure_get_field(writer_packet_context,
	129	field_name);
	130	if (!writer_field) {
19ce87a4 JD	131	fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
19ce87a4 JD	132	__LINE__);
9ae49d3d	133	goto error;
19ce87a4 JD	134	}
	135
	136	int_ret = bt_ctf_field_unsigned_integer_set_value(writer_field, value);
19ce87a4	137	if (int_ret < 0) {
19ce87a4 JD	138	fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
19ce87a4 JD	139	__LINE__);
9ae49d3d	140	goto error;
19ce87a4	141	}
9ae49d3d JD	142	BT_PUT(writer_field);
	143	BT_PUT(field_type);
	144	BT_PUT(field);
19ce87a4 JD	145	}
	146
	147	ret = BT_COMPONENT_STATUS_OK;
9ae49d3d	148	goto end;
19ce87a4	149
9ae49d3d JD	150	error:
	151	bt_put(writer_field);
	152	bt_put(field_type);
	153	bt_put(field);
	154	ret = BT_COMPONENT_STATUS_ERROR;
	155	end:
19ce87a4	156	bt_put(struct_type);
19ce87a4	157	bt_put(packet_context);
19ce87a4 JD	158	return ret;
	159	}
	160
	161	BT_HIDDEN
	162	struct bt_ctf_packet *trimmer_new_packet(
	163	struct trimmer_iterator *trim_it,
	164	struct bt_ctf_packet *packet)
	165	{
9ae49d3d	166	struct bt_ctf_stream *stream = NULL;
19ce87a4 JD	167	struct bt_ctf_packet *writer_packet = NULL;
	168	int int_ret;
	169
	170	stream = bt_ctf_packet_get_stream(packet);
	171	if (!stream) {
	172	fprintf(trim_it->err, "[error] %s in %s:%d\n",
	173	__func__, __FILE__, __LINE__);
9ae49d3d	174	goto error;
19ce87a4 JD	175	}
	176
	177	/*
	178	* If a packet was already opened, close it and remove it from
	179	* the HT.
	180	*/
	181	writer_packet = lookup_packet(trim_it, packet);
	182	if (writer_packet) {
	183	g_hash_table_remove(trim_it->packet_map, packet);
9ae49d3d	184	BT_PUT(writer_packet);
19ce87a4 JD	185	}
	186
	187	writer_packet = insert_new_packet(trim_it, packet, stream);
	188	if (!writer_packet) {
	189	fprintf(trim_it->err, "[error] %s in %s:%d\n",
	190	__func__, __FILE__, __LINE__);
9ae49d3d	191	goto error;
19ce87a4 JD	192	}
	193	bt_get(writer_packet);
	194
9877e1aa JD	195	int_ret = ctf_packet_copy_context(trim_it->err, packet,
	196	stream, writer_packet);
	197	if (int_ret < 0) {
19ce87a4 JD	198	fprintf(trim_it->err, "[error] %s in %s:%d\n",
19ce87a4 JD	199	__func__, __FILE__, __LINE__);
9ae49d3d	200	goto error;
19ce87a4 JD	201	}
19ce87a4 JD	202
9ae49d3d JD	203	goto end;
	204
	205	error:
	206	BT_PUT(writer_packet);
	207	end:
19ce87a4	208	bt_put(stream);
19ce87a4 JD	209	return writer_packet;
	210	}
	211
	212	BT_HIDDEN
	213	struct bt_ctf_packet *trimmer_close_packet(
	214	struct trimmer_iterator *trim_it,
	215	struct bt_ctf_packet *packet)
	216	{
9ae49d3d	217	struct bt_ctf_packet *writer_packet = NULL;
19ce87a4 JD	218
	219	writer_packet = lookup_packet(trim_it, packet);
	220	if (!writer_packet) {
	221	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
	222	__FILE__, __LINE__);
	223	goto end;
	224	}
	225
	226	g_hash_table_remove(trim_it->packet_map, packet);
	227
	228	end:
	229	return writer_packet;
	230	}
	231
	232	BT_HIDDEN
	233	struct bt_ctf_event *trimmer_output_event(
	234	struct trimmer_iterator *trim_it,
	235	struct bt_ctf_event *event)
	236	{
9ae49d3d	237	struct bt_ctf_event_class *event_class = NULL;
19ce87a4	238	struct bt_ctf_event *writer_event = NULL;
9ae49d3d	239	struct bt_ctf_packet packet = NULL, writer_packet = NULL;
19ce87a4 JD	240	const char *event_name;
	241	int int_ret;
	242
	243	event_class = bt_ctf_event_get_class(event);
	244	if (!event_class) {
	245	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
	246	__FILE__, __LINE__);
9ae49d3d	247	goto error;
19ce87a4 JD	248	}
	249
	250	event_name = bt_ctf_event_class_get_name(event_class);
	251	if (!event_name) {
	252	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
	253	__FILE__, __LINE__);
9ae49d3d	254	goto error;
19ce87a4 JD	255	}
	256
	257	writer_event = ctf_copy_event(trim_it->err, event, event_class, false);
	258	if (!writer_event) {
	259	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
	260	__FILE__, __LINE__);
	261	fprintf(trim_it->err, "[error] Failed to copy event %s\n",
	262	bt_ctf_event_class_get_name(event_class));
9ae49d3d	263	goto error;
19ce87a4 JD	264	}
	265
	266	packet = bt_ctf_event_get_packet(event);
	267	if (!packet) {
19ce87a4 JD	268	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
19ce87a4 JD	269	__FILE__, __LINE__);
9ae49d3d	270	goto error;
19ce87a4 JD	271	}
	272
	273	writer_packet = lookup_packet(trim_it, packet);
	274	if (!writer_packet) {
19ce87a4 JD	275	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
19ce87a4 JD	276	__FILE__, __LINE__);
9ae49d3d	277	goto error;
19ce87a4 JD	278	}
	279	bt_get(writer_packet);
	280
	281	int_ret = bt_ctf_event_set_packet(writer_event, writer_packet);
	282	if (int_ret < 0) {
19ce87a4 JD	283	fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__,
	284	__FILE__, __LINE__);
	285	fprintf(trim_it->err, "[error] Failed to append event %s\n",
	286	bt_ctf_event_class_get_name(event_class));
9ae49d3d	287	goto error;
19ce87a4 JD	288	}
19ce87a4 JD	289
9ae49d3d JD	290	/* We keep the reference on the writer_event to create a notification. */
9ae49d3d JD	291	goto end;
19ce87a4	292
9ae49d3d JD	293	error:
	294	BT_PUT(writer_event);
	295	end:
19ce87a4	296	bt_put(writer_packet);
19ce87a4	297	bt_put(packet);
19ce87a4	298	bt_put(event_class);
19ce87a4 JD	299	return writer_event;
19ce87a4 JD	300	}