| 1 | /* |
| 2 | * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca> |
| 3 | * |
| 4 | * This program is free software; you can redistribute it and/or |
| 5 | * modify it under the terms of the GNU General Public License |
| 6 | * as published by the Free Software Foundation; either version 2 |
| 7 | * of the License, or (at your option) any later version. |
| 8 | * |
| 9 | * This program is distributed in the hope that it will be useful, |
| 10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 | * GNU General Public License for more details. |
| 13 | * |
| 14 | * You should have received a copy of the GNU General Public License |
| 15 | * along with this program; if not, write to the Free Software |
| 16 | * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
| 17 | */ |
| 18 | |
| 19 | #include <popt.h> |
| 20 | #include <stdlib.h> |
| 21 | |
| 22 | #include "options.h" |
| 23 | |
| 24 | /* Option variables */ |
| 25 | char *opt_event_list; |
| 26 | char *opt_tracing_group; |
| 27 | char *opt_sessiond_path; |
| 28 | char *opt_session_name; |
| 29 | char *opt_trace_name; |
| 30 | int opt_destroy_trace; |
| 31 | int opt_create_session; |
| 32 | int opt_destroy_session; |
| 33 | int opt_trace_kernel = 0; |
| 34 | int opt_quiet = 0; |
| 35 | int opt_verbose = 0; |
| 36 | int opt_list_apps = 0; |
| 37 | int opt_no_sessiond = 0; |
| 38 | int opt_list_session = 0; |
| 39 | int opt_list_traces = 0; |
| 40 | int opt_create_trace; |
| 41 | int opt_start_trace = 0; |
| 42 | int opt_stop_trace = 0; |
| 43 | int opt_enable_event; |
| 44 | int opt_disable_event; |
| 45 | int opt_kern_create_channel; |
| 46 | pid_t opt_trace_pid = 0; |
| 47 | |
| 48 | enum { |
| 49 | OPT_HELP = 1, |
| 50 | OPT_ENABLE_EVENT, |
| 51 | OPT_DISABLE_EVENT, |
| 52 | OPT_CREATE_SESSION, |
| 53 | OPT_CREATE_TRACE, |
| 54 | OPT_DESTROY_SESSION, |
| 55 | OPT_DESTROY_TRACE, |
| 56 | OPT_START_TRACE, |
| 57 | OPT_STOP_TRACE, |
| 58 | }; |
| 59 | |
| 60 | static struct poptOption long_options[] = { |
| 61 | /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */ |
| 62 | {"create-session", 'c', POPT_ARG_STRING, 0, OPT_CREATE_SESSION, 0, 0}, |
| 63 | {"create-trace", 'C', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, 0, OPT_CREATE_TRACE, 0, 0}, |
| 64 | {"destroy-trace", 'D', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, 0, OPT_DESTROY_TRACE, 0, 0}, |
| 65 | {"destroy-session", 'd', POPT_ARG_STRING, 0, OPT_DESTROY_SESSION, 0, 0}, |
| 66 | {"disable-event", 0, POPT_ARG_STRING, 0, OPT_DISABLE_EVENT, 0, 0}, |
| 67 | {"enable-event", 'e', POPT_ARG_STRING, 0, OPT_ENABLE_EVENT, 0, 0}, |
| 68 | {"group", 0, POPT_ARG_STRING, &opt_tracing_group, 0, 0, 0}, |
| 69 | {"help", 'h', POPT_ARG_NONE, 0, OPT_HELP, 0, 0}, |
| 70 | {"kernel", 'k', POPT_ARG_VAL, &opt_trace_kernel, 1, 0, 0}, |
| 71 | {"kern-create-channel",0, POPT_ARG_VAL, &opt_kern_create_channel, 1, 0, 0}, |
| 72 | {"list-apps", 'L', POPT_ARG_VAL, &opt_list_apps, 1, 0, 0}, |
| 73 | {"list-sessions", 'l', POPT_ARG_VAL, &opt_list_session, 1, 0, 0}, |
| 74 | {"list-traces", 't', POPT_ARG_VAL, &opt_list_traces, 1, 0, 0}, |
| 75 | {"no-kernel", 0, POPT_ARG_VAL, &opt_trace_kernel, 0, 0, 0}, |
| 76 | {"no-sessiond", 0, POPT_ARG_VAL, &opt_no_sessiond, 1, 0, 0}, |
| 77 | {"pid", 'p', POPT_ARG_INT, &opt_trace_pid, 0, 0, 0}, |
| 78 | {"quiet", 'q', POPT_ARG_VAL, &opt_quiet, 1, 0, 0}, |
| 79 | {"session", 's', POPT_ARG_STRING, &opt_session_name, 0, 0, 0}, |
| 80 | {"sessiond-path", 0, POPT_ARG_STRING, &opt_sessiond_path, 0, 0, 0}, |
| 81 | {"start", 0, POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, 0, OPT_START_TRACE, 0, 0}, |
| 82 | {"stop", 0, POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, 0, OPT_STOP_TRACE, 0, 0}, |
| 83 | {"verbose", 'v', POPT_ARG_VAL, &opt_verbose, 1, 0, 0}, |
| 84 | {0, 0, 0, 0, 0, 0, 0} |
| 85 | }; |
| 86 | |
| 87 | |
| 88 | /* |
| 89 | * usage |
| 90 | */ |
| 91 | static void usage(FILE *ofp) |
| 92 | { |
| 93 | fprintf(ofp, "LTTng Trace Control " VERSION"\n\n"); |
| 94 | fprintf(ofp, "usage : lttng [OPTION]\n"); |
| 95 | fprintf(ofp, "\n"); |
| 96 | fprintf(ofp, "Options:\n"); |
| 97 | fprintf(ofp, " -v, --verbose Verbose mode\n"); |
| 98 | fprintf(ofp, " -q, --quiet Quiet mode\n"); |
| 99 | fprintf(ofp, " --help Show help\n"); |
| 100 | fprintf(ofp, " --group NAME Unix tracing group name. (default: tracing)\n"); |
| 101 | fprintf(ofp, " --no-sessiond Don't spawn a session daemon\n"); |
| 102 | fprintf(ofp, " --sessiond-path Session daemon full path\n"); |
| 103 | fprintf(ofp, " -L, --list-apps List traceable user-space applications\n"); |
| 104 | fprintf(ofp, "\n"); |
| 105 | fprintf(ofp, "Session options:\n"); |
| 106 | fprintf(ofp, " -c, --create-session NAME Create a new session\n"); |
| 107 | fprintf(ofp, " -l, --list-sessions List all available sessions by name\n"); |
| 108 | fprintf(ofp, " -s, --session UUID Specify tracing session using UUID\n"); |
| 109 | fprintf(ofp, " -d, --destroy-session NAME Destroy the session specified by NAME\n"); |
| 110 | fprintf(ofp, " -t, --list-traces List session's traces. Use -s to specify the session\n"); |
| 111 | fprintf(ofp, "\n"); |
| 112 | fprintf(ofp, "Tracing options:\n"); |
| 113 | fprintf(ofp, " -p, --pid PID Specify action on user-space tracer for PID\n"); |
| 114 | fprintf(ofp, " -k, --kernel Specify action on kernel tracer\n"); |
| 115 | fprintf(ofp, " -e, --enable-event LIST Enable tracing event (support marker and tracepoint)\n"); |
| 116 | fprintf(ofp, " --disable-event LIST Disable tracing event (support marker and tracepoint)\n"); |
| 117 | fprintf(ofp, " -C, --create-trace Create a trace. Allocate and setup a trace\n"); |
| 118 | fprintf(ofp, " -D, --destroy-trace [NAME] Destroy a trace. Use NAME to identify user-space trace\n"); |
| 119 | fprintf(ofp, " --start [NAME] Start tracing. Use NAME to identify user-space trace\n"); |
| 120 | fprintf(ofp, " --stop [NAME] Stop tracing. Use NAME to identify user-space trace\n"); |
| 121 | fprintf(ofp, "\n"); |
| 122 | fprintf(ofp, "Kernel tracing options:\n"); |
| 123 | fprintf(ofp, " --kern-create-channel Create a kernel channel\n"); |
| 124 | fprintf(ofp, "\n"); |
| 125 | fprintf(ofp, "User-space tracing options:\n"); |
| 126 | fprintf(ofp, "\n"); |
| 127 | fprintf(ofp, "Please see the lttng(1) man page for full documentation.\n"); |
| 128 | fprintf(ofp, "See http://lttng.org for updates, bug reports and news.\n"); |
| 129 | } |
| 130 | |
| 131 | /* |
| 132 | * parse_args |
| 133 | * |
| 134 | * Parse command line arguments. |
| 135 | * Return 0 if OK, else -1 |
| 136 | */ |
| 137 | int parse_args(int argc, const char **argv) |
| 138 | { |
| 139 | static poptContext pc; |
| 140 | int opt; |
| 141 | |
| 142 | /* If no options, fail */ |
| 143 | if (argc < 2) { |
| 144 | return -1; |
| 145 | } |
| 146 | |
| 147 | pc = poptGetContext(NULL, argc, argv, long_options, 0); |
| 148 | poptReadDefaultConfig(pc, 0); |
| 149 | |
| 150 | while ((opt = poptGetNextOpt(pc)) != -1) { |
| 151 | switch (opt) { |
| 152 | case OPT_HELP: |
| 153 | usage(stderr); |
| 154 | clean_exit(EXIT_SUCCESS); |
| 155 | break; |
| 156 | case OPT_CREATE_SESSION: |
| 157 | opt_create_session = 1; |
| 158 | opt_session_name = poptGetOptArg(pc); |
| 159 | break; |
| 160 | case OPT_DESTROY_SESSION: |
| 161 | opt_destroy_session = 1; |
| 162 | opt_session_name = poptGetOptArg(pc); |
| 163 | break; |
| 164 | case OPT_ENABLE_EVENT: |
| 165 | opt_enable_event = 1; |
| 166 | opt_event_list = poptGetOptArg(pc); |
| 167 | break; |
| 168 | case OPT_DESTROY_TRACE: |
| 169 | opt_destroy_trace = 1; |
| 170 | opt_trace_name = poptGetOptArg(pc); |
| 171 | break; |
| 172 | case OPT_START_TRACE: |
| 173 | opt_start_trace = 1; |
| 174 | opt_trace_name = poptGetOptArg(pc); |
| 175 | break; |
| 176 | case OPT_STOP_TRACE: |
| 177 | opt_stop_trace = 1; |
| 178 | opt_trace_name = poptGetOptArg(pc); |
| 179 | break; |
| 180 | case OPT_CREATE_TRACE: |
| 181 | opt_create_trace = 1; |
| 182 | opt_trace_name = poptGetOptArg(pc); |
| 183 | break; |
| 184 | default: |
| 185 | usage(stderr); |
| 186 | clean_exit(EXIT_FAILURE); |
| 187 | break; |
| 188 | } |
| 189 | } |
| 190 | |
| 191 | if (pc) { |
| 192 | poptFreeContext(pc); |
| 193 | } |
| 194 | |
| 195 | return 0; |
| 196 | } |