d05674535dfbd5d4c9ecf3fdc0b5cb807ee4842d
1 /* Copyright (C) 2009 Pierre-Marc Fournier
3 * This library is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU Lesser General Public
5 * License as published by the Free Software Foundation; either
6 * version 2.1 of the License, or (at your option) any later version.
8 * This library 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 GNU
11 * Lesser General Public License for more details.
13 * You should have received a copy of the GNU Lesser General Public
14 * License along with this library; if not, write to the Free Software
15 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
50 char *progname
= NULL
;
54 fprintf(stderr
, "usage: %s COMMAND PIDs...\n", progname
);
55 fprintf(stderr
, "\nControl the tracing of a process that supports LTTng Userspace Tracing.\n\
58 --start-trace\t\t\tStart tracing\n\
59 --stop-trace\t\t\tStop tracing\n\
60 --destroy-trace\t\t\tDestroy the trace\n\
61 --enable-marker \"CHANNEL/MARKER\"\tEnable a marker\n\
62 --disable-marker \"CHANNEL/MARKER\"\tDisable a marker\n\
63 --list-markers\t\t\tList the markers of the process, their\n\t\t\t\t\t state and format string\n\
68 int parse_opts_long(int argc
, char **argv
, struct ust_opts
*opts
)
74 opts
->regex_state
= -1;
78 static struct option long_options
[] = {
79 {"start-trace", 0, 0, 1000},
80 {"stop-trace", 0, 0, 1001},
81 {"destroy-trace", 0, 0, 1002},
82 {"list-markers", 0, 0, 1004},
83 {"print-markers", 0, 0, 1005},
85 {"enable-marker", 1, 0, 1007},
86 {"disable-marker", 1, 0, 1008},
87 {"start", 0, 0, 1009},
89 {"version", 0, 0, 1010},
90 {"online-pids", 0, 0, 1011},
94 c
= getopt_long(argc
, argv
, "h", long_options
, &option_index
);
100 printf("option %s", long_options
[option_index
].name
);
102 printf(" with arg %s", optarg
);
107 opts
->cmd
= START_TRACE
;
110 opts
->cmd
= STOP_TRACE
;
119 opts
->cmd
= LIST_MARKERS
;
122 opts
->cmd
= ENABLE_MARKER
;
123 opts
->regex
= strdup(optarg
);
126 opts
->cmd
= DISABLE_MARKER
;
127 opts
->regex
= strdup(optarg
);
130 opts
->cmd
= GET_ONLINE_PIDS
;
136 printf("Version 0.1\n");
139 /* unknown option or other error; error is
140 printed by getopt, just return */
146 if (argc
- optind
> 0 && opts
->cmd
!= GET_ONLINE_PIDS
) {
149 opts
->pids
= malloc((argc
-optind
+1) * sizeof(pid_t
));
151 for(i
=optind
; i
<argc
; i
++) {
152 /* don't take any chances, use a long long */
155 tmp
= strtoull(argv
[i
], &endptr
, 10);
156 if(*endptr
!= '\0') {
157 ERR("The pid \"%s\" is invalid.", argv
[i
]);
160 opts
->pids
[pididx
++] = (pid_t
) tmp
;
162 opts
->pids
[pididx
] = -1;
165 if (opts
->cmd
== ENABLE_MARKER
|| opts
->cmd
== DISABLE_MARKER
) {
166 opts
->regex_state
= regcomp(&opts
->preg
, opts
->regex
, 0);
167 if (opts
->regex_state
) {
168 ERR("invalid regular expression\n");
175 static void regex_change_m_state(struct ust_opts
*opts
, pid_t pid
)
177 struct marker_status
*cmsf
= NULL
;
179 int e
= (opts
->cmd
== ENABLE_MARKER
);
181 if (opts
->regex_state
!= 0) {
185 if (ustcmd_get_cmsf(&cmsf
, pid
)) {
186 fprintf(stderr
, "error while trying to get markers for PID "
187 "%u\n", (unsigned int) pid
);
190 while (cmsf
[i
].channel
!= NULL
) {
192 asprintf(&mc
, "%s/%s", cmsf
[i
].channel
, cmsf
[i
].marker
);
193 if (regexec(&opts
->preg
, mc
, 0, NULL
, 0) == 0) {
194 /* We got a match! */
195 if (ustcmd_set_marker_state(mc
,
196 e
? USTCMD_MS_ON
: USTCMD_MS_OFF
, pid
)) {
198 "error while trying to %sable marker"
199 "\"%s\" for PID %u\n",
200 e
? "en" : "dis", mc
,
203 printf("sucessfully %sabled marker "
204 "\"%s\" for PID %u\n",
205 e
? "en" : "dis", mc
,
212 ustcmd_free_cmsf(cmsf
);
215 int main(int argc
, char *argv
[])
219 struct ust_opts opts
;
224 fprintf(stderr
, "No operation specified.\n");
229 result
= parse_opts_long(argc
, argv
, &opts
);
231 fprintf(stderr
, "\n");
236 if(opts
.pids
== NULL
&& opts
.cmd
!= GET_ONLINE_PIDS
) {
237 fprintf(stderr
, "No pid specified.\n");
241 if(opts
.cmd
== UNKNOWN
) {
242 fprintf(stderr
, "No command specified.\n");
246 if (opts
.cmd
== GET_ONLINE_PIDS
) {
247 pid_t
*pp
= ustcmd_get_online_pids();
252 printf("%u\n", (unsigned int) pp
[i
]);
262 struct marker_status
*cmsf
= NULL
;
264 while(*pidit
!= -1) {
267 result
= ustcmd_start_trace(*pidit
);
269 ERR("error while trying to for trace with PID %u\n", (unsigned int) *pidit
);
272 //printf("sucessfully started trace for PID %u\n", (unsigned int) *pidit);
276 result
= ustcmd_stop_trace(*pidit
);
278 ERR("error while trying to stop trace for PID %u\n", (unsigned int) *pidit
);
281 //printf("sucessfully stopped trace for PID %u\n", (unsigned int) *pidit);
285 result
= ustcmd_setup_and_start(*pidit
);
287 ERR("error while trying to setup/start trace for PID %u\n", (unsigned int) *pidit
);
290 //printf("sucessfully setup/started trace for PID %u\n", (unsigned int) *pidit);
294 result
= ustcmd_destroy_trace(*pidit
);
296 ERR("error while trying to destroy trace with PID %u\n", (unsigned int) *pidit
);
299 //printf("sucessfully destroyed trace for PID %u\n", (unsigned int) *pidit);
304 if (ustcmd_get_cmsf(&cmsf
, *pidit
)) {
306 "error while trying to list markers for"
307 " PID %u\n", (unsigned int) *pidit
);
311 while (cmsf
[i
].channel
!= NULL
) {
312 printf("{PID: %u, channel/marker: %s/%s, "
313 "state: %u, fs: %s}\n",
314 (unsigned int) *pidit
,
321 ustcmd_free_cmsf(cmsf
);
326 regex_change_m_state(&opts
, *pidit
);
330 ERR("unknown command\n");
337 if (opts
.pids
!= NULL
) {
340 if (opts
.regex
!= NULL
) {
This page took 0.042051 seconds and 4 git commands to generate.