[lttng-tools.git] / tests / utils / testapp / gen-ust-events / gen-ust-events.c

/*
 * Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
 *
 * This library is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by the
 * Free Software Foundation; version 2.1 of the License.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 */

#define _LGPL_SOURCE
#include <assert.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdbool.h>
#include <signal.h>
#include <poll.h>
#include <errno.h>
#include "utils.h"
#include "signal-helper.h"

#define TRACEPOINT_DEFINE
#include "tp.h"

int main(int argc, char **argv)
{
	unsigned int i, netint;
	long values[] = { 1, 2, 3 };
	char text[10] = "test";
	double dbl = 2.0;
	float flt = 2222.0;
	int nr_iter = 100, ret = 0, first_event_file_created = 0;
	useconds_t nr_usec = 0;
	char *after_first_event_file_path = NULL;
	char *before_last_event_file_path = NULL;

	if (set_signal_handler()) {
		ret = -1;
		goto end;
	}

	if (argc >= 2) {
		/*
		 * If nr_iter is negative, do an infinite tracing loop.
		 */
		nr_iter = atoi(argv[1]);
	}

	if (argc >= 3) {
		/* By default, don't wait unless user specifies. */
		nr_usec = atoi(argv[2]);
	}

	if (argc >= 4) {
		after_first_event_file_path = argv[3];
	}

	if (argc >= 5) {
		before_last_event_file_path = argv[4];
	}

	for (i = 0; nr_iter < 0 || i < nr_iter; i++) {
		if (nr_iter >= 0 && i == nr_iter - 1) {
			/*
			 * Wait on synchronization before writing last
			 * event.
			 */
			if (before_last_event_file_path) {
				ret = wait_on_file(before_last_event_file_path);
				if (ret != 0) {
					goto end;
				}
			}
		}
		netint = htonl(i);
		tracepoint(tp, tptest, i, netint, values, text,
			strlen(text), dbl, flt);

		/*
		 * First loop we create the file if asked to indicate
		 * that at least one tracepoint has been hit.
		 */
		if (after_first_event_file_path && first_event_file_created == 0) {
			ret = create_file(after_first_event_file_path);

			if (ret != 0) {
				goto end;
			} else {
				first_event_file_created = 1;
			}
		}

		if (nr_usec) {
		        if (usleep_safe(nr_usec)) {
				ret = -1;
				goto end;
			}
		}
		if (should_quit) {
			break;
		}
	}

end:
	exit(!ret ? EXIT_SUCCESS : EXIT_FAILURE);
}
Commit	Line	Data
7e0cc23b CB	1	/*
	2	* Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
	3	*
	4	* This library is free software; you can redistribute it and/or modify it
	5	* under the terms of the GNU Lesser General Public License as published by the
	6	* Free Software Foundation; version 2.1 of the License.
	7	*
	8	* This library is distributed in the hope that it will be useful, but WITHOUT
	9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	10	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
	11	* for more details.
	12	*
	13	* You should have received a copy of the GNU Lesser General Public License
	14	* along with this library; if not, write to the Free Software Foundation,
	15	* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	16	*/
	17
70dac0d7	18	#define _LGPL_SOURCE
209b934f	19	#include <assert.h>
7e0cc23b	20	#include <arpa/inet.h>
209b934f	21	#include <fcntl.h>
7e0cc23b CB	22	#include <stdarg.h>
	23	#include <stdio.h>
	24	#include <stdlib.h>
	25	#include <string.h>
	26	#include <sys/mman.h>
	27	#include <sys/stat.h>
	28	#include <sys/types.h>
	29	#include <unistd.h>
0fc2834c	30	#include <stdbool.h>
5fcaccbc MD	31	#include <signal.h>
	32	#include <poll.h>
	33	#include <errno.h>
ae941114	34	#include "utils.h"
95983a02	35	#include "signal-helper.h"
7e0cc23b CB	36
	37	#define TRACEPOINT_DEFINE
	38	#include "tp.h"
	39
	40	int main(int argc, char **argv)
	41	{
0fc2834c	42	unsigned int i, netint;
7e0cc23b CB	43	long values[] = { 1, 2, 3 };
	44	char text[10] = "test";
	45	double dbl = 2.0;
	46	float flt = 2222.0;
b2047add	47	int nr_iter = 100, ret = 0, first_event_file_created = 0;
7e0cc23b	48	useconds_t nr_usec = 0;
5fcaccbc MD	49	char *after_first_event_file_path = NULL;
5fcaccbc MD	50	char *before_last_event_file_path = NULL;
7e0cc23b	51
95983a02 JG	52	if (set_signal_handler()) {
	53	ret = -1;
	54	goto end;
	55	}
	56
7e0cc23b	57	if (argc >= 2) {
0fc2834c MD	58	/*
	59	* If nr_iter is negative, do an infinite tracing loop.
	60	*/
7e0cc23b CB	61	nr_iter = atoi(argv[1]);
	62	}
	63
2c34b645	64	if (argc >= 3) {
7e0cc23b CB	65	/* By default, don't wait unless user specifies. */
	66	nr_usec = atoi(argv[2]);
	67	}
	68
2c34b645	69	if (argc >= 4) {
5fcaccbc MD	70	after_first_event_file_path = argv[3];
	71	}
	72
	73	if (argc >= 5) {
	74	before_last_event_file_path = argv[4];
209b934f DG	75	}
209b934f DG	76
0fc2834c	77	for (i = 0; nr_iter < 0 \|\| i < nr_iter; i++) {
5fcaccbc MD	78	if (nr_iter >= 0 && i == nr_iter - 1) {
	79	/*
	80	* Wait on synchronization before writing last
	81	* event.
	82	*/
b2047add FD	83	if (before_last_event_file_path) {
	84	ret = wait_on_file(before_last_event_file_path);
	85	if (ret != 0) {
	86	goto end;
	87	}
	88	}
5fcaccbc	89	}
7e0cc23b	90	netint = htonl(i);
5fcaccbc MD	91	tracepoint(tp, tptest, i, netint, values, text,
5fcaccbc MD	92	strlen(text), dbl, flt);
209b934f DG	93
209b934f DG	94	/*
5fcaccbc MD	95	* First loop we create the file if asked to indicate
5fcaccbc MD	96	* that at least one tracepoint has been hit.
209b934f	97	*/
b2047add FD	98	if (after_first_event_file_path && first_event_file_created == 0) {
	99	ret = create_file(after_first_event_file_path);
	100
	101	if (ret != 0) {
	102	goto end;
	103	} else {
	104	first_event_file_created = 1;
	105	}
	106	}
	107
b734f5f7	108	if (nr_usec) {
ae941114 JG	109	if (usleep_safe(nr_usec)) {
	110	ret = -1;
	111	goto end;
	112	}
b734f5f7	113	}
95983a02 JG	114	if (should_quit) {
	115	break;
	116	}
7e0cc23b CB	117	}
7e0cc23b CB	118
ae941114 JG	119	end:
ae941114 JG	120	exit(!ret ? EXIT_SUCCESS : EXIT_FAILURE);
7e0cc23b	121	}