2 * Copyright (C) 2011 - David Goulet <dgoulet@efficios.com>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License, version 2 only,
6 * as published by the Free Software Foundation.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24 #include <sys/types.h>
27 #include "../command.h"
29 static char *opt_session_name
;
30 static char *opt_viewer
;
31 static char *opt_trace_path
;
32 static const char *babeltrace_bin
= CONFIG_BABELTRACE_BIN
;
33 //static const char *lttv_gui_bin = CONFIG_LTTV_GUI_BIN;
35 #ifdef LTTNG_EMBED_HELP
36 static const char help_msg
[] =
37 #include <lttng-view.1.h>
46 static struct poptOption long_options
[] = {
47 /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
48 {"help", 'h', POPT_ARG_NONE
, 0, OPT_HELP
, 0, 0},
49 {"list-options", 0, POPT_ARG_NONE
, NULL
, OPT_LIST_OPTIONS
, NULL
, NULL
},
50 {"viewer", 'e', POPT_ARG_STRING
, &opt_viewer
, 0, 0, 0},
51 {"trace-path", 't', POPT_ARG_STRING
, &opt_trace_path
, 0, 0, 0},
56 * This is needed for each viewer since we are using execvp().
58 static const char *babeltrace_opts
[] = { "babeltrace" };
59 //static const char *lttv_gui_opts[] = { "lttv-gui", "-t", };
62 * Type is also use as the index in the viewers array. So please, make sure
63 * your enum value is in the right order in the array below.
66 VIEWER_BABELTRACE
= 0,
68 VIEWER_USER_DEFINED
= 2,
72 * NOTE: "lttv" is a shell command and it's not working for exec() family
73 * functions so we might think of removing this wrapper or using bash.
75 static struct viewers
{
76 const char *exec_name
;
77 enum viewer_type type
;
79 { "babeltrace", VIEWER_BABELTRACE
},
80 { "lttv-gui", VIEWER_LTTV_GUI
},
81 { NULL
, VIEWER_USER_DEFINED
},
84 /* Is the session we are trying to view is in live mode. */
85 static int session_live_mode
;
87 static struct viewers
*parse_options(void)
89 if (opt_viewer
== NULL
) {
90 /* Default is babeltrace */
91 return &(viewers
[VIEWER_BABELTRACE
]);
95 * This means that if -e, --viewers is used, we just override everything
96 * with it. For supported viewers like lttv, we could simply detect if "-t"
97 * is passed and if not, add the trace directory to it.
99 return &(viewers
[VIEWER_USER_DEFINED
]);
103 * Alloc an array of string pointer from a simple string having all options
104 * seperated by spaces. Also adds the trace path to the arguments.
106 * The returning pointer is ready to be passed to execvp().
108 static char **alloc_argv_from_user_opts(char *opts
, const char *trace_path
)
110 int i
= 0, ignore_space
= 0;
111 unsigned int num_opts
= 1;
112 char **argv
, *token
= opts
;
114 /* Count number of arguments. */
117 /* Use to ignore consecutive spaces */
126 } while (*token
!= '\0');
128 /* Add two here for the NULL terminating element and trace path */
129 argv
= zmalloc(sizeof(char *) * (num_opts
+ 2));
134 token
= strtok(opts
, " ");
135 while (token
!= NULL
) {
136 argv
[i
] = strdup(token
);
137 if (argv
[i
] == NULL
) {
140 token
= strtok(NULL
, " ");
144 argv
[num_opts
] = (char *) trace_path
;
145 argv
[num_opts
+ 1] = NULL
;
151 for (i
= 0; i
< num_opts
+ 2; i
++) {
161 * Alloc an array of string pointer from an array of strings. It also adds
162 * the trace path to the argv.
164 * The returning pointer is ready to be passed to execvp().
166 static char **alloc_argv_from_local_opts(const char **opts
, size_t opts_len
,
167 const char *trace_path
)
170 size_t size
, mem_len
;
173 /* Add one for the NULL terminating element. */
174 mem_len
= opts_len
+ 1;
175 if (session_live_mode
) {
176 /* Add 3 option for the live mode being "-i lttng-live URL". */
179 /* Add option for the trace path. */
183 size
= sizeof(char *) * mem_len
;
185 /* Add two here for the trace_path and the NULL terminating element. */
186 argv
= zmalloc(size
);
191 memcpy(argv
, opts
, size
);
193 if (session_live_mode
) {
194 argv
[opts_len
] = "-i";
195 argv
[opts_len
+ 1] = "lttng-live";
196 argv
[opts_len
+ 2] = (char *) trace_path
;
197 argv
[opts_len
+ 3] = NULL
;
199 argv
[opts_len
] = (char *) trace_path
;
200 argv
[opts_len
+ 1] = NULL
;
208 * Spawn viewer with the trace directory path.
210 static int spawn_viewer(const char *trace_path
)
214 const char *viewer_bin
= NULL
;
215 struct viewers
*viewer
;
218 /* Check for --viewer options */
219 viewer
= parse_options();
220 if (viewer
== NULL
) {
225 switch (viewer
->type
) {
226 case VIEWER_BABELTRACE
:
227 if (stat(babeltrace_bin
, &status
) == 0) {
228 viewer_bin
= babeltrace_bin
;
230 viewer_bin
= viewer
->exec_name
;
232 argv
= alloc_argv_from_local_opts(babeltrace_opts
,
233 ARRAY_SIZE(babeltrace_opts
), trace_path
);
235 case VIEWER_USER_DEFINED
:
236 argv
= alloc_argv_from_user_opts(opt_viewer
, trace_path
);
238 viewer_bin
= argv
[0];
242 viewer_bin
= viewers
[VIEWER_BABELTRACE
].exec_name
;
243 argv
= alloc_argv_from_local_opts(babeltrace_opts
,
244 ARRAY_SIZE(babeltrace_opts
), trace_path
);
248 if (argv
== NULL
|| !viewer_bin
) {
253 DBG("Using %s viewer", viewer_bin
);
255 ret
= execvp(viewer_bin
, argv
);
257 if (errno
== ENOENT
) {
258 ERR("%s not found on the system", viewer_bin
);
260 PERROR("exec: %s", viewer_bin
);
272 * Build the live path we need for the lttng live view.
274 static char *build_live_path(char *session_name
)
278 char hostname
[HOST_NAME_MAX
];
280 ret
= gethostname(hostname
, sizeof(hostname
));
282 PERROR("gethostname");
286 ret
= asprintf(&path
, "net://localhost/host/%s/%s", hostname
,
289 PERROR("asprintf live path");
298 * Exec viewer if found and use session name path.
300 static int view_trace(void)
303 char *session_name
, *trace_path
= NULL
;
304 struct lttng_session
*sessions
= NULL
;
307 * Safety net. If lttng is suid at some point for *any* useless reasons,
308 * this prevent any bad execution of binaries.
311 if (getuid() != geteuid()) {
312 ERR("UID does not match effective UID.");
315 } else if (getgid() != getegid()) {
316 ERR("GID does not match effective GID.");
322 /* User define trace path override the session name */
323 if (opt_trace_path
) {
325 } else if(opt_session_name
== NULL
) {
326 session_name
= get_session_name();
327 if (session_name
== NULL
) {
332 session_name
= opt_session_name
;
335 DBG("Viewing trace for session %s", session_name
);
338 int i
, count
, found
= 0;
340 /* Getting all sessions */
341 count
= lttng_list_sessions(&sessions
);
343 ERR("Unable to list sessions. Session name %s not found.",
345 MSG("Is there a session daemon running?");
350 /* Find our session listed by the session daemon */
351 for (i
= 0; i
< count
; i
++) {
352 if (strncmp(sessions
[i
].name
, session_name
, NAME_MAX
) == 0) {
359 MSG("Session name %s not found", session_name
);
364 session_live_mode
= sessions
[i
].live_timer_interval
;
366 DBG("Session live mode set to %d", session_live_mode
);
368 if (sessions
[i
].enabled
&& !session_live_mode
) {
369 WARN("Session %s is running. Please stop it before reading it.",
375 /* If the timer interval is set we are in live mode. */
376 if (session_live_mode
) {
377 trace_path
= build_live_path(session_name
);
383 /* Get file system session path. */
384 trace_path
= sessions
[i
].path
;
387 trace_path
= opt_trace_path
;
390 MSG("Trace directory: %s\n", trace_path
);
392 ret
= spawn_viewer(trace_path
);
394 /* Don't set ret so lttng can interpret the sessiond error. */
399 if (session_live_mode
) {
404 if (opt_session_name
== NULL
) {
412 * The 'view <options>' first level command
414 int cmd_view(int argc
, const char **argv
)
416 int opt
, ret
= CMD_SUCCESS
;
417 static poptContext pc
;
419 pc
= poptGetContext(NULL
, argc
, argv
, long_options
, 0);
420 poptReadDefaultConfig(pc
, 0);
423 WARN("mi does not apply to view command");
426 while ((opt
= poptGetNextOpt(pc
)) != -1) {
431 case OPT_LIST_OPTIONS
:
432 list_cmd_options(stdout
, long_options
);
440 opt_session_name
= (char*) poptGetArg(pc
);
This page took 0.041408 seconds and 4 git commands to generate.