2 * Session rotation example control application
4 * Copyright 2017, Julien Desfossez <jdesfossez@efficios.com>
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
27 * gcc -o rotate-client rotate-client-example.c -llttng-ctl
29 * Run with the following command to rotate the session every second and
30 * compress the chunk, until ctrl-c:
31 * ./rotate-client mysession 1 -1 ./rotate-client-compress.sh
34 #include <lttng/lttng.h>
41 #define DEFAULT_DATA_AVAILABILITY_WAIT_TIME 200000 /* usec */
43 static volatile int quit
= 0;
45 static void sighandler(int signal
__attribute__((unused
)))
47 printf("Signal caught, exiting\n");
51 static int setup_session(const char *session_name
, const char *path
)
54 struct lttng_domain dom
;
55 struct lttng_event ev
;
56 struct lttng_handle
*chan_handle
= NULL
;
58 printf("Creating session %s\n", session_name
);
59 ret
= lttng_create_session(session_name
, path
);
61 fprintf(stderr
, "Failed to create session, ret = %d\n", ret
);
65 dom
.type
= LTTNG_DOMAIN_KERNEL
;
66 dom
.buf_type
= LTTNG_BUFFER_GLOBAL
;
68 chan_handle
= lttng_create_handle(session_name
, &dom
);
69 if (chan_handle
== NULL
) {
74 memset(&ev
, 0, sizeof(ev
));
75 ev
.type
= LTTNG_EVENT_SYSCALL
;
77 ev
.loglevel_type
= LTTNG_EVENT_LOGLEVEL_ALL
;
79 ret
= lttng_enable_event_with_exclusions(
80 chan_handle
, &ev
, "mychan", NULL
, 0, NULL
);
82 fprintf(stderr
, "Failed to enable events (ret = %i)\n", ret
);
85 printf("Enabled all system call kernel events\n");
87 ret
= lttng_start_tracing(session_name
);
89 fprintf(stderr
, "Failed to start tracing\n");
96 lttng_destroy_handle(chan_handle
);
100 static int cleanup_session(const char *session_name
)
104 printf("Stopping session %s", session_name
);
105 ret
= lttng_stop_tracing_no_wait(session_name
);
107 fprintf(stderr
, "Failed to stop tracing\n");
113 ret
= lttng_data_pending(session_name
);
115 /* Return the data available call error. */
120 * Data sleep time before retrying (in usec). Don't sleep if the
121 * call returned value indicates availability.
124 usleep(DEFAULT_DATA_AVAILABILITY_WAIT_TIME
);
131 printf("Destroying session %s\n", session_name
);
132 ret
= lttng_destroy_session(session_name
);
134 fprintf(stderr
, "Failed to destroy the session\n");
144 static int rotate_session(const char *session_name
, const char *ext_program
)
147 struct lttng_rotation_handle
*handle
= NULL
;
148 enum lttng_rotation_status rotation_status
;
149 enum lttng_rotation_state rotation_state
= LTTNG_ROTATION_STATE_ONGOING
;
152 printf("Rotating the output files of session %s", session_name
);
154 ret
= lttng_rotate_session(session_name
, NULL
, &handle
);
156 fprintf(stderr
, "Failed to rotate session, %s\n",
157 lttng_strerror(ret
));
164 rotation_status
= lttng_rotation_handle_get_state(
165 handle
, &rotation_state
);
166 if (rotation_status
!= LTTNG_ROTATION_STATUS_OK
) {
168 fprintf(stderr
, "Failed to get the current rotation's state\n");
173 * Data sleep time before retrying (in usec). Don't
174 * sleep if the call returned value indicates
177 if (rotation_state
== LTTNG_ROTATION_STATE_ONGOING
) {
178 usleep(DEFAULT_DATA_AVAILABILITY_WAIT_TIME
);
182 } while (rotation_state
== LTTNG_ROTATION_STATE_ONGOING
);
185 switch (rotation_state
) {
186 case LTTNG_ROTATION_STATE_COMPLETED
:
188 const struct lttng_trace_archive_location
*location
;
189 const char *absolute_path
;
190 enum lttng_trace_archive_location_status location_status
;
192 rotation_status
= lttng_rotation_handle_get_archive_location(
194 if (rotation_status
!= LTTNG_ROTATION_STATUS_OK
) {
195 fprintf(stderr
, "Failed to retrieve the rotation's completed chunk archive location\n");
201 lttng_trace_archive_location_local_get_absolute_path(
202 location
, &absolute_path
);
203 if (location_status
!= LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK
) {
204 fprintf(stderr
, "Failed to get absolute path of completed chunk archive");
209 printf("Output files of session %s rotated to %s\n",
210 session_name
, absolute_path
);
211 ret
= snprintf(cmd
, PATH_MAX
, "%s %s", ext_program
,
214 fprintf(stderr
, "Failed to prepare command string\n");
220 case LTTNG_ROTATION_STATE_EXPIRED
:
221 printf("Output files of session %s rotated, but the handle expired\n",
225 case LTTNG_ROTATION_STATE_ERROR
:
226 fprintf(stderr
, "An error occurred with the rotation of session %s\n",
230 case LTTNG_ROTATION_STATE_ONGOING
:
233 case LTTNG_ROTATION_STATE_NO_ROTATION
:
234 fprintf(stderr
, "No rotation was performed on rotation request for session %s\n",
241 lttng_rotation_handle_destroy(handle
);
245 static int cleanup_dir(const char *path
)
250 ret
= snprintf(cmd
, PATH_MAX
, "rm -rf %s", path
);
252 fprintf(stderr
, "Failed to prepare rm -rf command string\n");
261 static void usage(const char *prog_name
)
263 fprintf(stderr
, "Usage: %s <session-name> <delay-sec> <nr-rotate> <program>\n",
265 fprintf(stderr
, " <session-name>: the name of the session you want to create\n");
266 fprintf(stderr
, " <delay-sec>: the delay in seconds between each rotation\n");
268 " <nr-rotate>: the number of rotation you want to perform, "
269 "-1 for infinite until ctrl-c\n");
271 " <program>: program to run on each chunk, it must be "
272 "executable, and expect a trace folder as only argument\n");
273 fprintf(stderr
, "\nThe trace folder is deleted when this program completes.\n");
276 int main(int argc
, char **argv
)
279 char tmppath
[] = "/tmp/lttng-rotate-XXXXXX";
280 char *session_name
, *path
, *ext_program
;
289 session_name
= argv
[1];
290 delay
= atoi(argv
[2]);
292 ext_program
= argv
[4];
295 fprintf(stderr
, "delay-sec must be a positive values\n");
300 if (signal(SIGINT
, sighandler
) == SIG_ERR
) {
302 perror("signal handler");
306 path
= mkdtemp(tmppath
);
308 fprintf(stderr
, "Failed to create temporary path\n");
311 printf("Output directory: %s\n", path
);
313 ret
= setup_session(session_name
, path
);
315 goto end_cleanup_dir
;
319 unsigned int sleep_time
;
322 for (i
= 0; i
< nr
; i
++) {
323 ret
= rotate_session(session_name
, ext_program
);
328 while (sleep_time
> 0) {
329 sleep_time
= sleep(sleep_time
);
337 ret
= rotate_session(session_name
, ext_program
);
346 ret
= cleanup_session(session_name
);
351 ret
= cleanup_dir(path
);