c73bd2578cf65683f6b421425c7e8c162afb4f33
3 * Linux Trace Toolkit Control
5 * Small program that controls LTT through libltt.
8 * Mathieu Desnoyers <mathieu.desnoyers@polymtl.ca>
15 #include <liblttctl/lttctl.h>
19 #include <sys/types.h>
27 /* Buffer for file copy : 4k seems optimal. */
42 static char *trace_name
= NULL
;
43 static char *mode_name
= NULL
;
44 static unsigned subbuf_size
= 0;
45 static unsigned n_subbufs
= 0;
46 static unsigned append_trace
= 0;
47 static enum trace_mode mode
= LTT_TRACE_NORMAL
;
48 static enum trace_ctl_op op
= CTL_OP_NONE
;
49 static char *channel_root
= NULL
;
50 static char *trace_root
= NULL
;
51 static char *num_threads
= "1";
53 static int sigchld_received
= 0;
55 void sigchld_handler(int signo
)
57 printf("signal %d received\n", signo
);
65 void show_arguments(void)
67 printf("Please use the following arguments :\n");
69 printf("-n name Name of the trace.\n");
70 printf("-b Create trace channels and start tracing (no daemon).\n");
71 printf("-c Create trace channels.\n");
72 printf("-m mode Normal or flight recorder mode.\n");
73 printf(" Mode values : normal (default) or flight.\n");
74 printf("-r Destroy trace channels.\n");
75 printf("-R Stop tracing and destroy trace channels.\n");
76 printf("-s Start tracing.\n");
77 //printf(" Note : will automatically create a normal trace if "
79 printf("-q Stop tracing.\n");
80 printf("-d Create trace, spawn a lttd daemon, start tracing.\n");
81 printf(" (optionnaly, you can set LTT_DAEMON\n");
82 printf(" and the LTT_FACILITIES env. vars.)\n");
83 printf("-t Trace root path. (ex. /root/traces/example_trace)\n");
84 printf("-l LTT channels root path. (ex. /mnt/relayfs/ltt)\n");
85 printf("-z Size of the subbuffers (will be rounded to next page size)\n");
86 printf("-x Number of subbuffers\n");
87 printf("-e Get XML facilities description\n");
88 printf("-a Append to trace\n");
89 printf("-N Number of lttd threads\n");
96 * Parses the command line arguments.
98 * Returns -1 if the arguments were correct, but doesn't ask for program
99 * continuation. Returns EINVAL if the arguments are incorrect, or 0 if OK.
101 int parse_arguments(int argc
, char **argv
)
107 if(strcmp(argv
[1], "-h") == 0) {
114 switch(argv
[argn
][0]) {
116 switch(argv
[argn
][1]) {
119 trace_name
= argv
[argn
+1];
122 printf("Specify a trace name after -n.\n");
129 op
= CTL_OP_CREATE_START
;
135 mode_name
= argv
[argn
+1];
137 if(strcmp(mode_name
, "normal") == 0)
138 mode
= LTT_TRACE_NORMAL
;
139 else if(strcmp(mode_name
, "flight") == 0)
140 mode
= LTT_TRACE_FLIGHT
;
142 printf("Invalid mode '%s'.\n", argv
[argn
]);
147 printf("Specify a mode after -m.\n");
156 op
= CTL_OP_STOP_DESTROY
;
166 subbuf_size
= (unsigned)atoi(argv
[argn
+1]);
169 printf("Specify a number of subbuffers after -z.\n");
176 n_subbufs
= (unsigned)atoi(argv
[argn
+1]);
179 printf("Specify a subbuffer size after -x.\n");
188 op
= CTL_OP_DESCRIPTION
;
192 trace_root
= argv
[argn
+1];
195 printf("Specify a trace root path after -t.\n");
202 channel_root
= argv
[argn
+1];
205 printf("Specify a channel root path after -l.\n");
215 num_threads
= argv
[argn
+1];
220 printf("Invalid argument '%s'.\n", argv
[argn
]);
226 printf("Invalid argument '%s'.\n", argv
[argn
]);
233 if(op
!= CTL_OP_DESCRIPTION
&& trace_name
== NULL
) {
234 printf("Please specify a trace name.\n");
239 if(op
== CTL_OP_NONE
) {
240 printf("Please specify an operation.\n");
245 if(op
== CTL_OP_DAEMON
) {
246 if(trace_root
== NULL
) {
247 printf("Please specify -t trace_root_path with the -d option.\n");
251 if(channel_root
== NULL
) {
252 printf("Please specify -l ltt_root_path with the -d option.\n");
258 if(op
== CTL_OP_DESCRIPTION
) {
259 if(trace_root
== NULL
) {
260 printf("Please specify -t trace_root_path with the -e option.\n");
271 printf("Linux Trace Toolkit Trace Control\n");
273 if(trace_name
!= NULL
) {
274 printf("Controlling trace : %s\n", trace_name
);
279 int create_eventdefs(void)
282 char eventdefs_path
[PATH_MAX
];
283 char eventdefs_file
[PATH_MAX
];
284 char facilities_file
[PATH_MAX
];
285 char read_buf
[BUF_SIZE
];
286 struct dirent
*entry
;
287 char *facilities_path
= getenv("LTT_FACILITIES");
288 if(facilities_path
== NULL
) facilities_path
=
289 PACKAGE_DATA_DIR
"/" PACKAGE
"/facilities";
291 ret
= mkdir(trace_root
, S_IRWXU
|S_IRWXG
|S_IRWXO
);
292 if(ret
== -1 && errno
!= EEXIST
) {
294 perror("Cannot create trace_root directory");
295 printf("trace_root is %s\n", trace_root
);
300 size_t trace_root_len
= strlen(trace_root
);
301 strncpy(eventdefs_path
, trace_root
, PATH_MAX
);
302 strncat(eventdefs_path
, "/eventdefs/", PATH_MAX
- trace_root_len
);
303 size_t eventdefs_path_len
= strlen(eventdefs_path
);
304 ret
= mkdir(eventdefs_path
, S_IRWXU
|S_IRWXG
|S_IRWXO
);
305 if(ret
== -1 && (!append_trace
|| errno
!= EEXIST
)) {
307 perror("Cannot create eventdefs directory");
312 DIR *facilities_dir
= opendir(facilities_path
);
314 if(facilities_dir
== NULL
) {
315 perror("Cannot open facilities directory");
320 while((entry
= readdir(facilities_dir
)) != NULL
) {
321 if(entry
->d_name
[0] == '.') continue;
323 printf("Appending facility file %s\n", entry
->d_name
);
324 strncpy(eventdefs_file
, eventdefs_path
, PATH_MAX
);
325 strncat(eventdefs_file
, entry
->d_name
, PATH_MAX
- eventdefs_path_len
);
326 /* Append to the file */
327 FILE *dest
= fopen(eventdefs_file
, "a");
329 perror("Cannot create eventdefs file");
332 strncpy(facilities_file
, facilities_path
, PATH_MAX
);
333 size_t facilities_dir_len
= strlen(facilities_path
);
334 strncat(facilities_file
, "/", PATH_MAX
- facilities_dir_len
);
335 strncat(facilities_file
, entry
->d_name
, PATH_MAX
- facilities_dir_len
-1);
336 FILE *src
= fopen(facilities_file
, "r");
339 perror("Cannot open eventdefs file for reading");
344 size_t read_size
, write_size
;
345 read_size
= fread(read_buf
, sizeof(char), BUF_SIZE
, src
);
348 perror("Cannot read eventdefs file");
351 write_size
= fwrite(read_buf
, sizeof(char), read_size
, dest
);
354 perror("Cannot write eventdefs file");
359 /* Add spacing between facilities */
360 fwrite("\n", 1, 1, dest
);
368 closedir(facilities_dir
);
376 int lttctl_daemon(struct lttctl_handle
*handle
, char *trace_name
)
378 char channel_path
[PATH_MAX
] = "";
381 char *lttd_path
= getenv("LTT_DAEMON");
382 struct sigaction act
;
384 if(lttd_path
== NULL
) lttd_path
=
385 PACKAGE_BIN_DIR
"/lttd";
387 strcat(channel_path
, channel_root
);
388 strcat(channel_path
, "/");
389 strcat(channel_path
, trace_name
);
392 ret
= lttctl_create_trace(handle
, trace_name
, mode
, subbuf_size
, n_subbufs
);
393 if(ret
!= 0) goto create_error
;
395 act
.sa_handler
= sigchld_handler
;
396 sigemptyset(&(act
.sa_mask
));
397 sigaddset(&(act
.sa_mask
), SIGCHLD
);
398 sigaction(SIGCHLD
, &act
, NULL
);
405 while(!(sigchld_received
)) pause();
407 waitpid(pid
, &status
, 0);
409 if(WIFEXITED(status
))
410 ret
= WEXITSTATUS(status
);
411 if(ret
) goto start_error
;
413 printf("Creating supplementary trace files\n");
414 ret
= create_eventdefs();
415 if(ret
) goto start_error
;
417 } else if(pid
== 0) {
421 ret
= execlp(lttd_path
, lttd_path
, "-t", trace_root
, "-c",
422 channel_path
, "-d", "-a", "-N", num_threads
, NULL
);
424 ret
= execlp(lttd_path
, lttd_path
, "-t", trace_root
, "-c",
425 channel_path
, "-d", "-N", num_threads
, NULL
);
428 perror("Error in executing the lttd daemon");
433 perror("Error in forking for lttd daemon");
436 ret
= lttctl_start(handle
, trace_name
);
437 if(ret
!= 0) goto start_error
;
443 printf("Trace start error\n");
444 ret
|= lttctl_destroy_trace(handle
, trace_name
);
449 int main(int argc
, char ** argv
)
452 struct lttctl_handle
*handle
;
454 ret
= parse_arguments(argc
, argv
);
456 if(ret
!= 0) show_arguments();
457 if(ret
== EINVAL
) return EINVAL
;
458 if(ret
== -1) return 0;
462 handle
= lttctl_create_handle();
464 if(handle
== NULL
) return -1;
467 case CTL_OP_CREATE_START
:
468 ret
= lttctl_create_trace(handle
, trace_name
, mode
, subbuf_size
,
471 ret
= lttctl_start(handle
, trace_name
);
474 ret
= lttctl_create_trace(handle
, trace_name
, mode
, subbuf_size
,
478 ret
= lttctl_destroy_trace(handle
, trace_name
);
480 case CTL_OP_STOP_DESTROY
:
481 ret
= lttctl_stop(handle
, trace_name
);
483 ret
= lttctl_destroy_trace(handle
, trace_name
);
486 ret
= lttctl_start(handle
, trace_name
);
489 ret
= lttctl_stop(handle
, trace_name
);
492 ret
= lttctl_daemon(handle
, trace_name
);
494 case CTL_OP_DESCRIPTION
:
495 ret
= create_eventdefs();
501 ret
|= lttctl_destroy_handle(handle
);
This page took 0.107353 seconds and 4 git commands to generate.