X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Futils%2Ftestapp%2Fgen-ust-events%2Fgen-ust-events.c;h=df1e58e417fa5f464b50a01f52fc2e3a97755881;hp=2f4dc275e1878a1dd89c791cd388da7b6fe82e5b;hb=6f9f5cd0134460ad499ce753ff4886440b0b8a0f;hpb=70dac0d716db5de1ede7448ab4092741d20bf248 diff --git a/tests/utils/testapp/gen-ust-events/gen-ust-events.c b/tests/utils/testapp/gen-ust-events/gen-ust-events.c index 2f4dc275e..df1e58e41 100644 --- a/tests/utils/testapp/gen-ust-events/gen-ust-events.c +++ b/tests/utils/testapp/gen-ust-events/gen-ust-events.c @@ -1,21 +1,12 @@ /* - * Copyright (C) - 2012 David Goulet + * Copyright (C) 2012 David Goulet * - * 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. + * SPDX-License-Identifier: LGPL-2.1-only * - * 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 #include #include #include @@ -27,63 +18,179 @@ #include #include #include +#include +#include +#include +#include +#include "utils.h" +#include "signal-helper.h" #define TRACEPOINT_DEFINE #include "tp.h" -void create_file(const char *path) +static struct option long_options[] = { - int ret; - - assert(path); - - ret = creat(path, S_IRWXU); - if (ret < 0) { - fprintf(stderr, "Failed to create file %s\n", path); - return; - } - - (void) close(ret); -} + /* These options set a flag. */ + {"iter", required_argument, 0, 'i'}, + {"wait", required_argument, 0, 'w'}, + {"sync-after-first-event", required_argument, 0, 'a'}, + {"sync-before-last-event", required_argument, 0, 'b'}, + {"sync-before-last-event-touch", required_argument, 0, 'c'}, + {"sync-before-exit", required_argument, 0, 'd'}, + {"sync-before-exit-touch", required_argument, 0, 'e'}, + {0, 0, 0, 0} +}; int main(int argc, char **argv) { - int i, netint; + unsigned int i, netint; + int option_index; + int option; long values[] = { 1, 2, 3 }; char text[10] = "test"; + char escape[10] = "\\*"; double dbl = 2.0; float flt = 2222.0; - unsigned int nr_iter = 100; + uint32_t net_values[] = { 1, 2, 3 }; + int nr_iter = 100, ret = 0, first_event_file_created = 0; useconds_t nr_usec = 0; - char *tmp_file_path = NULL; + char *after_first_event_file_path = NULL; + char *before_last_event_file_path = NULL; + /* + * Touch a file to indicate that all events except one were + * generated. + */ + char *before_last_event_file_path_touch = NULL; + /* Touch file when we are exiting */ + char *before_exit_file_path_touch = NULL; + /* Wait on file before exiting */ + char *before_exit_file_path = NULL; - if (argc >= 2) { - nr_iter = atoi(argv[1]); + for (i = 0; i < 3; i++) { + net_values[i] = htonl(net_values[i]); } - if (argc >= 3) { - /* By default, don't wait unless user specifies. */ - nr_usec = atoi(argv[2]); + while ((option = getopt_long(argc, argv, "i:w:a:b:c:d:", + long_options, &option_index)) != -1) { + switch (option) { + case 'a': + after_first_event_file_path = strdup(optarg); + break; + case 'b': + before_last_event_file_path = strdup(optarg); + break; + case 'c': + before_last_event_file_path_touch = strdup(optarg); + break; + case 'd': + before_exit_file_path = strdup(optarg); + break; + case 'e': + before_exit_file_path_touch = strdup(optarg); + break; + case 'i': + nr_iter = atoi(optarg); + break; + case 'w': + nr_usec = atoi(optarg); + break; + case '?': + /* getopt_long already printed an error message. */ + default: + ret = -1; + goto end; + } + } + + if (optind != argc) { + fprintf(stderr, "Error: takes long options only.\n"); + + /* + * Aborting the test program for now because callers typically don't check + * the test program return value, and the transition from positional + * arguments to getopt causes hangs when caller scripts are not updated. + * An abort is easier to diagnose and fix. This is a temporary solution: + * we should eventually ensure that all scripts test and report the test + * app return values. + */ + abort(); + + ret = -1; + goto end; } - if (argc >= 4) { - tmp_file_path = argv[3]; + + if (set_signal_handler()) { + ret = -1; + goto end; } - for (i = 0; i < nr_iter; i++) { + for (i = 0; nr_iter < 0 || i < nr_iter; i++) { + if (nr_iter >= 0 && i == nr_iter - 1) { + if (before_last_event_file_path_touch) { + ret = create_file(before_last_event_file_path_touch); + if (ret != 0) { + goto end; + } + } + + /* + * 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); + tracepoint(tp, tptest, i, netint, values, text, + strlen(text), escape, net_values, dbl, flt); /* - * First loop we create the file if asked to indicate that at least one - * tracepoint has been hit. + * First loop we create the file if asked to indicate + * that at least one tracepoint has been hit. */ - if (i == 0 && tmp_file_path) { - create_file(tmp_file_path); + 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; } - usleep(nr_usec); } - return 0; + if (before_exit_file_path_touch) { + ret = create_file(before_exit_file_path_touch); + if (ret != 0) { + goto end; + } + } + if (before_exit_file_path) { + ret = wait_on_file(before_exit_file_path); + if (ret != 0) { + goto end; + } + } +end: + free(after_first_event_file_path); + free(before_last_event_file_path); + free(before_last_event_file_path_touch); + free(before_exit_file_path); + free(before_exit_file_path_touch); + exit(!ret ? EXIT_SUCCESS : EXIT_FAILURE); }