2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
3 * Copyright (C) 2016 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License, version 2 only,
7 * as published by the Free Software Foundation.
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.
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
26 #include <sys/types.h>
30 #include <urcu/list.h>
32 #include <common/mi-lttng.h>
34 #include "../command.h"
36 static char *opt_channel_name
;
37 static char *opt_session_name
;
38 static int opt_kernel
;
39 static int opt_userspace
;
42 static char *opt_type
;
44 #ifdef LTTNG_EMBED_HELP
45 static const char help_msg
[] =
46 #include <lttng-add-context.1.h>
60 static struct lttng_handle
*handle
;
61 static struct mi_writer
*writer
;
64 * Taken from the LTTng ABI
68 CONTEXT_PERF_COUNTER
= 1, /* Backward compat. */
77 CONTEXT_PTHREAD_ID
= 10,
78 CONTEXT_HOSTNAME
= 11,
80 CONTEXT_PERF_CPU_COUNTER
= 13,
81 CONTEXT_PERF_THREAD_COUNTER
= 14,
82 CONTEXT_APP_CONTEXT
= 15,
83 CONTEXT_INTERRUPTIBLE
= 16,
84 CONTEXT_PREEMPTIBLE
= 17,
85 CONTEXT_NEED_RESCHEDULE
= 18,
86 CONTEXT_MIGRATABLE
= 19,
87 CONTEXT_CALLSTACK_KERNEL
= 20,
88 CONTEXT_CALLSTACK_USER
= 21,
92 * Taken from the Perf ABI (all enum perf_*)
95 PERF_TYPE_HARDWARE
= 0,
96 PERF_TYPE_SOFTWARE
= 1,
97 PERF_TYPE_HW_CACHE
= 3,
101 enum perf_count_hard
{
102 PERF_COUNT_HW_CPU_CYCLES
= 0,
103 PERF_COUNT_HW_INSTRUCTIONS
= 1,
104 PERF_COUNT_HW_CACHE_REFERENCES
= 2,
105 PERF_COUNT_HW_CACHE_MISSES
= 3,
106 PERF_COUNT_HW_BRANCH_INSTRUCTIONS
= 4,
107 PERF_COUNT_HW_BRANCH_MISSES
= 5,
108 PERF_COUNT_HW_BUS_CYCLES
= 6,
109 PERF_COUNT_HW_STALLED_CYCLES_FRONTEND
= 7,
110 PERF_COUNT_HW_STALLED_CYCLES_BACKEND
= 8,
113 enum perf_count_soft
{
114 PERF_COUNT_SW_CPU_CLOCK
= 0,
115 PERF_COUNT_SW_TASK_CLOCK
= 1,
116 PERF_COUNT_SW_PAGE_FAULTS
= 2,
117 PERF_COUNT_SW_CONTEXT_SWITCHES
= 3,
118 PERF_COUNT_SW_CPU_MIGRATIONS
= 4,
119 PERF_COUNT_SW_PAGE_FAULTS_MIN
= 5,
120 PERF_COUNT_SW_PAGE_FAULTS_MAJ
= 6,
121 PERF_COUNT_SW_ALIGNMENT_FAULTS
= 7,
122 PERF_COUNT_SW_EMULATION_FAULTS
= 8,
126 * Generalized hardware cache events:
128 * { L1-D, L1-I, LLC, ITLB, DTLB, BPU } x
129 * { read, write, prefetch } x
130 * { accesses, misses }
132 enum perf_hw_cache_id
{
133 PERF_COUNT_HW_CACHE_L1D
= 0,
134 PERF_COUNT_HW_CACHE_L1I
= 1,
135 PERF_COUNT_HW_CACHE_LL
= 2,
136 PERF_COUNT_HW_CACHE_DTLB
= 3,
137 PERF_COUNT_HW_CACHE_ITLB
= 4,
138 PERF_COUNT_HW_CACHE_BPU
= 5,
140 PERF_COUNT_HW_CACHE_MAX
, /* non-ABI */
143 enum perf_hw_cache_op_id
{
144 PERF_COUNT_HW_CACHE_OP_READ
= 0,
145 PERF_COUNT_HW_CACHE_OP_WRITE
= 1,
146 PERF_COUNT_HW_CACHE_OP_PREFETCH
= 2,
148 PERF_COUNT_HW_CACHE_OP_MAX
, /* non-ABI */
151 enum perf_hw_cache_op_result_id
{
152 PERF_COUNT_HW_CACHE_RESULT_ACCESS
= 0,
153 PERF_COUNT_HW_CACHE_RESULT_MISS
= 1,
155 PERF_COUNT_HW_CACHE_RESULT_MAX
, /* non-ABI */
158 static struct poptOption long_options
[] = {
159 /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
160 {"help", 'h', POPT_ARG_NONE
, 0, OPT_HELP
, 0, 0},
161 {"session", 's', POPT_ARG_STRING
, &opt_session_name
, 0, 0, 0},
162 {"channel", 'c', POPT_ARG_STRING
, &opt_channel_name
, 0, 0, 0},
163 {"kernel", 'k', POPT_ARG_VAL
, &opt_kernel
, 1, 0, 0},
164 {"userspace", 'u', POPT_ARG_NONE
, 0, OPT_USERSPACE
, 0, 0},
165 {"jul", 'j', POPT_ARG_NONE
, 0, OPT_JUL
, 0, 0},
166 {"log4j", 'l', POPT_ARG_NONE
, 0, OPT_LOG4J
, 0, 0},
167 {"type", 't', POPT_ARG_STRING
, &opt_type
, OPT_TYPE
, 0, 0},
168 {"list", 0, POPT_ARG_NONE
, NULL
, OPT_LIST
, NULL
, NULL
},
169 {"list-options", 0, POPT_ARG_NONE
, NULL
, OPT_LIST_OPTIONS
, NULL
, NULL
},
170 {0, 0, 0, 0, 0, 0, 0}
176 #define PERF_HW(optstr, name, type, hide) \
178 optstr, type, hide, \
179 .u.perf = { PERF_TYPE_HARDWARE, PERF_COUNT_HW_##name, },\
182 #define PERF_SW(optstr, name, type, hide) \
184 optstr, type, hide, \
185 .u.perf = { PERF_TYPE_SOFTWARE, PERF_COUNT_SW_##name, },\
188 #define _PERF_HW_CACHE(optstr, name, type, op, result, hide) \
190 optstr, type, hide, \
192 PERF_TYPE_HW_CACHE, \
193 (uint64_t) PERF_COUNT_HW_CACHE_##name \
194 | ((uint64_t) PERF_COUNT_HW_CACHE_OP_##op << 8) \
195 | ((uint64_t) PERF_COUNT_HW_CACHE_RESULT_##result << 16), \
199 #define PERF_HW_CACHE(optstr, name, type, hide) \
200 _PERF_HW_CACHE(optstr "-loads", name, type, \
201 READ, ACCESS, hide), \
202 _PERF_HW_CACHE(optstr "-load-misses", name, type, \
204 _PERF_HW_CACHE(optstr "-stores", name, type, \
205 WRITE, ACCESS, hide), \
206 _PERF_HW_CACHE(optstr "-store-misses", name, type, \
207 WRITE, MISS, hide), \
208 _PERF_HW_CACHE(optstr "-prefetches", name, type, \
209 PREFETCH, ACCESS, hide), \
210 _PERF_HW_CACHE(optstr "-prefetch-misses", name, type, \
211 PREFETCH, MISS, hide)
214 const struct ctx_opts
{
216 enum context_type ctx_type
;
217 int hide_help
; /* Hide from --help */
229 { "pid", CONTEXT_PID
},
230 { "procname", CONTEXT_PROCNAME
},
231 { "prio", CONTEXT_PRIO
},
232 { "nice", CONTEXT_NICE
},
233 { "vpid", CONTEXT_VPID
},
234 { "tid", CONTEXT_TID
},
235 { "pthread_id", CONTEXT_PTHREAD_ID
},
236 { "vtid", CONTEXT_VTID
},
237 { "ppid", CONTEXT_PPID
},
238 { "vppid", CONTEXT_VPPID
},
239 { "hostname", CONTEXT_HOSTNAME
},
240 { "ip", CONTEXT_IP
},
241 { "interruptible", CONTEXT_INTERRUPTIBLE
},
242 { "preemptible", CONTEXT_PREEMPTIBLE
},
243 { "need_reschedule", CONTEXT_NEED_RESCHEDULE
},
244 { "migratable", CONTEXT_MIGRATABLE
},
245 { "callstack-kernel", CONTEXT_CALLSTACK_KERNEL
},
246 #if HAVE_MODULES_USERSPACE_CALLSTACK_CONTEXT
247 { "callstack-user", CONTEXT_CALLSTACK_USER
},
252 /* Perf per-CPU counters */
253 PERF_HW("perf:cpu:cpu-cycles", CPU_CYCLES
,
254 CONTEXT_PERF_CPU_COUNTER
, 0),
255 PERF_HW("perf:cpu:cycles", CPU_CYCLES
,
256 CONTEXT_PERF_CPU_COUNTER
, 0),
257 PERF_HW("perf:cpu:stalled-cycles-frontend", STALLED_CYCLES_FRONTEND
,
258 CONTEXT_PERF_CPU_COUNTER
, 0),
259 PERF_HW("perf:cpu:idle-cycles-frontend", STALLED_CYCLES_FRONTEND
,
260 CONTEXT_PERF_CPU_COUNTER
, 0),
261 PERF_HW("perf:cpu:stalled-cycles-backend", STALLED_CYCLES_BACKEND
,
262 CONTEXT_PERF_CPU_COUNTER
, 0),
263 PERF_HW("perf:cpu:idle-cycles-backend", STALLED_CYCLES_BACKEND
,
264 CONTEXT_PERF_CPU_COUNTER
, 0),
265 PERF_HW("perf:cpu:instructions", INSTRUCTIONS
,
266 CONTEXT_PERF_CPU_COUNTER
, 0),
267 PERF_HW("perf:cpu:cache-references", CACHE_REFERENCES
,
268 CONTEXT_PERF_CPU_COUNTER
, 0),
269 PERF_HW("perf:cpu:cache-misses", CACHE_MISSES
,
270 CONTEXT_PERF_CPU_COUNTER
, 0),
271 PERF_HW("perf:cpu:branch-instructions", BRANCH_INSTRUCTIONS
,
272 CONTEXT_PERF_CPU_COUNTER
, 0),
273 PERF_HW("perf:cpu:branches", BRANCH_INSTRUCTIONS
,
274 CONTEXT_PERF_CPU_COUNTER
, 0),
275 PERF_HW("perf:cpu:branch-misses", BRANCH_MISSES
,
276 CONTEXT_PERF_CPU_COUNTER
, 0),
277 PERF_HW("perf:cpu:bus-cycles", BUS_CYCLES
,
278 CONTEXT_PERF_CPU_COUNTER
, 0),
280 PERF_HW_CACHE("perf:cpu:L1-dcache", L1D
,
281 CONTEXT_PERF_CPU_COUNTER
, 0),
282 PERF_HW_CACHE("perf:cpu:L1-icache", L1I
,
283 CONTEXT_PERF_CPU_COUNTER
, 0),
284 PERF_HW_CACHE("perf:cpu:LLC", LL
,
285 CONTEXT_PERF_CPU_COUNTER
, 0),
286 PERF_HW_CACHE("perf:cpu:dTLB", DTLB
,
287 CONTEXT_PERF_CPU_COUNTER
, 0),
288 _PERF_HW_CACHE("perf:cpu:iTLB-loads", ITLB
,
289 CONTEXT_PERF_CPU_COUNTER
, READ
, ACCESS
, 0),
290 _PERF_HW_CACHE("perf:cpu:iTLB-load-misses", ITLB
,
291 CONTEXT_PERF_CPU_COUNTER
, READ
, MISS
, 0),
292 _PERF_HW_CACHE("perf:cpu:branch-loads", BPU
,
293 CONTEXT_PERF_CPU_COUNTER
, READ
, ACCESS
, 0),
294 _PERF_HW_CACHE("perf:cpu:branch-load-misses", BPU
,
295 CONTEXT_PERF_CPU_COUNTER
, READ
, MISS
, 0),
297 PERF_SW("perf:cpu:cpu-clock", CPU_CLOCK
,
298 CONTEXT_PERF_CPU_COUNTER
, 0),
299 PERF_SW("perf:cpu:task-clock", TASK_CLOCK
,
300 CONTEXT_PERF_CPU_COUNTER
, 0),
301 PERF_SW("perf:cpu:page-fault", PAGE_FAULTS
,
302 CONTEXT_PERF_CPU_COUNTER
, 0),
303 PERF_SW("perf:cpu:faults", PAGE_FAULTS
,
304 CONTEXT_PERF_CPU_COUNTER
, 0),
305 PERF_SW("perf:cpu:major-faults", PAGE_FAULTS_MAJ
,
306 CONTEXT_PERF_CPU_COUNTER
, 0),
307 PERF_SW("perf:cpu:minor-faults", PAGE_FAULTS_MIN
,
308 CONTEXT_PERF_CPU_COUNTER
, 0),
309 PERF_SW("perf:cpu:context-switches", CONTEXT_SWITCHES
,
310 CONTEXT_PERF_CPU_COUNTER
, 0),
311 PERF_SW("perf:cpu:cs", CONTEXT_SWITCHES
,
312 CONTEXT_PERF_CPU_COUNTER
, 0),
313 PERF_SW("perf:cpu:cpu-migrations", CPU_MIGRATIONS
,
314 CONTEXT_PERF_CPU_COUNTER
, 0),
315 PERF_SW("perf:cpu:migrations", CPU_MIGRATIONS
,
316 CONTEXT_PERF_CPU_COUNTER
, 0),
317 PERF_SW("perf:cpu:alignment-faults", ALIGNMENT_FAULTS
,
318 CONTEXT_PERF_CPU_COUNTER
, 0),
319 PERF_SW("perf:cpu:emulation-faults", EMULATION_FAULTS
,
320 CONTEXT_PERF_CPU_COUNTER
, 0),
322 /* Perf per-thread counters */
323 PERF_HW("perf:thread:cpu-cycles", CPU_CYCLES
,
324 CONTEXT_PERF_THREAD_COUNTER
, 0),
325 PERF_HW("perf:thread:cycles", CPU_CYCLES
,
326 CONTEXT_PERF_THREAD_COUNTER
, 0),
327 PERF_HW("perf:thread:stalled-cycles-frontend", STALLED_CYCLES_FRONTEND
,
328 CONTEXT_PERF_THREAD_COUNTER
, 0),
329 PERF_HW("perf:thread:idle-cycles-frontend", STALLED_CYCLES_FRONTEND
,
330 CONTEXT_PERF_THREAD_COUNTER
, 0),
331 PERF_HW("perf:thread:stalled-cycles-backend", STALLED_CYCLES_BACKEND
,
332 CONTEXT_PERF_THREAD_COUNTER
, 0),
333 PERF_HW("perf:thread:idle-cycles-backend", STALLED_CYCLES_BACKEND
,
334 CONTEXT_PERF_THREAD_COUNTER
, 0),
335 PERF_HW("perf:thread:instructions", INSTRUCTIONS
,
336 CONTEXT_PERF_THREAD_COUNTER
, 0),
337 PERF_HW("perf:thread:cache-references", CACHE_REFERENCES
,
338 CONTEXT_PERF_THREAD_COUNTER
, 0),
339 PERF_HW("perf:thread:cache-misses", CACHE_MISSES
,
340 CONTEXT_PERF_THREAD_COUNTER
, 0),
341 PERF_HW("perf:thread:branch-instructions", BRANCH_INSTRUCTIONS
,
342 CONTEXT_PERF_THREAD_COUNTER
, 0),
343 PERF_HW("perf:thread:branches", BRANCH_INSTRUCTIONS
,
344 CONTEXT_PERF_THREAD_COUNTER
, 0),
345 PERF_HW("perf:thread:branch-misses", BRANCH_MISSES
,
346 CONTEXT_PERF_THREAD_COUNTER
, 0),
347 PERF_HW("perf:thread:bus-cycles", BUS_CYCLES
,
348 CONTEXT_PERF_THREAD_COUNTER
, 0),
350 PERF_HW_CACHE("perf:thread:L1-dcache", L1D
,
351 CONTEXT_PERF_THREAD_COUNTER
, 0),
352 PERF_HW_CACHE("perf:thread:L1-icache", L1I
,
353 CONTEXT_PERF_THREAD_COUNTER
, 0),
354 PERF_HW_CACHE("perf:thread:LLC", LL
,
355 CONTEXT_PERF_THREAD_COUNTER
, 0),
356 PERF_HW_CACHE("perf:thread:dTLB", DTLB
,
357 CONTEXT_PERF_THREAD_COUNTER
, 0),
358 _PERF_HW_CACHE("perf:thread:iTLB-loads", ITLB
,
359 CONTEXT_PERF_THREAD_COUNTER
, READ
, ACCESS
, 0),
360 _PERF_HW_CACHE("perf:thread:iTLB-load-misses", ITLB
,
361 CONTEXT_PERF_THREAD_COUNTER
, READ
, MISS
, 0),
362 _PERF_HW_CACHE("perf:thread:branch-loads", BPU
,
363 CONTEXT_PERF_THREAD_COUNTER
, READ
, ACCESS
, 0),
364 _PERF_HW_CACHE("perf:thread:branch-load-misses", BPU
,
365 CONTEXT_PERF_THREAD_COUNTER
, READ
, MISS
, 0),
367 PERF_SW("perf:thread:cpu-clock", CPU_CLOCK
,
368 CONTEXT_PERF_THREAD_COUNTER
, 0),
369 PERF_SW("perf:thread:task-clock", TASK_CLOCK
,
370 CONTEXT_PERF_THREAD_COUNTER
, 0),
371 PERF_SW("perf:thread:page-fault", PAGE_FAULTS
,
372 CONTEXT_PERF_THREAD_COUNTER
, 0),
373 PERF_SW("perf:thread:faults", PAGE_FAULTS
,
374 CONTEXT_PERF_THREAD_COUNTER
, 0),
375 PERF_SW("perf:thread:major-faults", PAGE_FAULTS_MAJ
,
376 CONTEXT_PERF_THREAD_COUNTER
, 0),
377 PERF_SW("perf:thread:minor-faults", PAGE_FAULTS_MIN
,
378 CONTEXT_PERF_THREAD_COUNTER
, 0),
379 PERF_SW("perf:thread:context-switches", CONTEXT_SWITCHES
,
380 CONTEXT_PERF_THREAD_COUNTER
, 0),
381 PERF_SW("perf:thread:cs", CONTEXT_SWITCHES
,
382 CONTEXT_PERF_THREAD_COUNTER
, 0),
383 PERF_SW("perf:thread:cpu-migrations", CPU_MIGRATIONS
,
384 CONTEXT_PERF_THREAD_COUNTER
, 0),
385 PERF_SW("perf:thread:migrations", CPU_MIGRATIONS
,
386 CONTEXT_PERF_THREAD_COUNTER
, 0),
387 PERF_SW("perf:thread:alignment-faults", ALIGNMENT_FAULTS
,
388 CONTEXT_PERF_THREAD_COUNTER
, 0),
389 PERF_SW("perf:thread:emulation-faults", EMULATION_FAULTS
,
390 CONTEXT_PERF_THREAD_COUNTER
, 0),
393 * Perf per-CPU counters, backward compatibilty for names.
394 * Hidden from help listing.
396 PERF_HW("perf:cpu-cycles", CPU_CYCLES
,
397 CONTEXT_PERF_COUNTER
, 1),
398 PERF_HW("perf:cycles", CPU_CYCLES
,
399 CONTEXT_PERF_COUNTER
, 1),
400 PERF_HW("perf:stalled-cycles-frontend", STALLED_CYCLES_FRONTEND
,
401 CONTEXT_PERF_COUNTER
, 1),
402 PERF_HW("perf:idle-cycles-frontend", STALLED_CYCLES_FRONTEND
,
403 CONTEXT_PERF_COUNTER
, 1),
404 PERF_HW("perf:stalled-cycles-backend", STALLED_CYCLES_BACKEND
,
405 CONTEXT_PERF_COUNTER
, 1),
406 PERF_HW("perf:idle-cycles-backend", STALLED_CYCLES_BACKEND
,
407 CONTEXT_PERF_COUNTER
, 1),
408 PERF_HW("perf:instructions", INSTRUCTIONS
,
409 CONTEXT_PERF_COUNTER
, 1),
410 PERF_HW("perf:cache-references", CACHE_REFERENCES
,
411 CONTEXT_PERF_COUNTER
, 1),
412 PERF_HW("perf:cache-misses", CACHE_MISSES
,
413 CONTEXT_PERF_COUNTER
, 1),
414 PERF_HW("perf:branch-instructions", BRANCH_INSTRUCTIONS
,
415 CONTEXT_PERF_COUNTER
, 1),
416 PERF_HW("perf:branches", BRANCH_INSTRUCTIONS
,
417 CONTEXT_PERF_COUNTER
, 1),
418 PERF_HW("perf:branch-misses", BRANCH_MISSES
,
419 CONTEXT_PERF_COUNTER
, 1),
420 PERF_HW("perf:bus-cycles", BUS_CYCLES
,
421 CONTEXT_PERF_COUNTER
, 1),
423 PERF_HW_CACHE("perf:L1-dcache", L1D
,
424 CONTEXT_PERF_COUNTER
, 1),
425 PERF_HW_CACHE("perf:L1-icache", L1I
,
426 CONTEXT_PERF_COUNTER
, 1),
427 PERF_HW_CACHE("perf:LLC", LL
,
428 CONTEXT_PERF_COUNTER
, 1),
429 PERF_HW_CACHE("perf:dTLB", DTLB
,
430 CONTEXT_PERF_COUNTER
, 1),
431 _PERF_HW_CACHE("perf:iTLB-loads", ITLB
,
432 CONTEXT_PERF_COUNTER
, READ
, ACCESS
, 1),
433 _PERF_HW_CACHE("perf:iTLB-load-misses", ITLB
,
434 CONTEXT_PERF_COUNTER
, READ
, MISS
, 1),
435 _PERF_HW_CACHE("perf:branch-loads", BPU
,
436 CONTEXT_PERF_COUNTER
, READ
, ACCESS
, 1),
437 _PERF_HW_CACHE("perf:branch-load-misses", BPU
,
438 CONTEXT_PERF_COUNTER
, READ
, MISS
, 1),
440 PERF_SW("perf:cpu-clock", CPU_CLOCK
,
441 CONTEXT_PERF_COUNTER
, 1),
442 PERF_SW("perf:task-clock", TASK_CLOCK
,
443 CONTEXT_PERF_COUNTER
, 1),
444 PERF_SW("perf:page-fault", PAGE_FAULTS
,
445 CONTEXT_PERF_COUNTER
, 1),
446 PERF_SW("perf:faults", PAGE_FAULTS
,
447 CONTEXT_PERF_COUNTER
, 1),
448 PERF_SW("perf:major-faults", PAGE_FAULTS_MAJ
,
449 CONTEXT_PERF_COUNTER
, 1),
450 PERF_SW("perf:minor-faults", PAGE_FAULTS_MIN
,
451 CONTEXT_PERF_COUNTER
, 1),
452 PERF_SW("perf:context-switches", CONTEXT_SWITCHES
,
453 CONTEXT_PERF_COUNTER
, 1),
454 PERF_SW("perf:cs", CONTEXT_SWITCHES
,
455 CONTEXT_PERF_COUNTER
, 1),
456 PERF_SW("perf:cpu-migrations", CPU_MIGRATIONS
,
457 CONTEXT_PERF_COUNTER
, 1),
458 PERF_SW("perf:migrations", CPU_MIGRATIONS
,
459 CONTEXT_PERF_COUNTER
, 1),
460 PERF_SW("perf:alignment-faults", ALIGNMENT_FAULTS
,
461 CONTEXT_PERF_COUNTER
, 1),
462 PERF_SW("perf:emulation-faults", EMULATION_FAULTS
,
463 CONTEXT_PERF_COUNTER
, 1),
465 { NULL
, -1 }, /* Closure */
469 #undef _PERF_HW_CACHE
474 * Context type for command line option parsing.
477 struct ctx_opts
*opt
;
478 struct cds_list_head list
;
482 * List of context type. Use to enable multiple context on a single command
485 struct ctx_type_list
{
486 struct cds_list_head head
;
488 .head
= CDS_LIST_HEAD_INIT(ctx_type_list
.head
),
494 * Find context numerical value from string.
496 * Return -1 if not found.
498 static int find_ctx_type_idx(const char *opt
)
502 while (ctx_opts
[i
].symbol
!= NULL
) {
503 if (strcmp(opt
, ctx_opts
[i
].symbol
) == 0) {
516 enum lttng_domain_type
get_domain(void)
519 return LTTNG_DOMAIN_KERNEL
;
520 } else if (opt_userspace
) {
521 return LTTNG_DOMAIN_UST
;
522 } else if (opt_jul
) {
523 return LTTNG_DOMAIN_JUL
;
524 } else if (opt_log4j
) {
525 return LTTNG_DOMAIN_LOG4J
;
542 ret
= fileno(stdout
);
544 PERROR("Unable to retrive fileno of stdout");
549 writer
= mi_lttng_writer_create(ret
, lttng_opt_mi
);
555 /* Open command element */
556 ret
= mi_lttng_writer_command_open(writer
,
557 mi_lttng_element_command_add_context
);
563 /* Open output element */
564 ret
= mi_lttng_writer_open_element(writer
,
565 mi_lttng_element_command_output
);
575 int mi_close(enum cmd_error_code success
)
584 /* Close output element */
585 ret
= mi_lttng_writer_close_element(writer
);
592 ret
= mi_lttng_writer_write_element_bool(writer
,
593 mi_lttng_element_command_success
, !success
);
599 /* Command element close */
600 ret
= mi_lttng_writer_command_close(writer
);
610 void populate_context(struct lttng_event_context
*context
,
611 const struct ctx_opts
*opt
)
615 context
->ctx
= (enum lttng_event_context_type
) opt
->ctx_type
;
616 switch (context
->ctx
) {
617 case LTTNG_EVENT_CONTEXT_PERF_COUNTER
:
618 case LTTNG_EVENT_CONTEXT_PERF_CPU_COUNTER
:
619 case LTTNG_EVENT_CONTEXT_PERF_THREAD_COUNTER
:
620 context
->u
.perf_counter
.type
= opt
->u
.perf
.type
;
621 context
->u
.perf_counter
.config
= opt
->u
.perf
.config
;
622 strncpy(context
->u
.perf_counter
.name
, opt
->symbol
,
623 LTTNG_SYMBOL_NAME_LEN
);
624 context
->u
.perf_counter
.name
[LTTNG_SYMBOL_NAME_LEN
- 1] = '\0';
625 /* Replace : and - by _ */
626 while ((ptr
= strchr(context
->u
.perf_counter
.name
, '-')) != NULL
) {
629 while ((ptr
= strchr(context
->u
.perf_counter
.name
, ':')) != NULL
) {
633 case LTTNG_EVENT_CONTEXT_APP_CONTEXT
:
634 context
->u
.app_ctx
.provider_name
=
635 opt
->u
.app_ctx
.provider_name
;
636 context
->u
.app_ctx
.ctx_name
=
637 opt
->u
.app_ctx
.ctx_name
;
645 * Pretty print context type.
648 int print_ctx_type(void)
654 struct lttng_event_context context
;
656 memset(&context
, 0, sizeof(context
));
665 /* Open a contexts element */
666 ret
= mi_lttng_writer_open_element(writer
, config_element_contexts
);
673 while (ctx_opts
[i
].symbol
!= NULL
) {
674 if (!ctx_opts
[i
].hide_help
) {
676 populate_context(&context
, &ctx_opts
[i
]);
677 ret
= mi_lttng_context(writer
, &context
, 1);
683 ret
= mi_lttng_writer_write_element_string(
685 mi_lttng_element_context_symbol
,
692 ret
= mi_lttng_writer_close_element(writer
);
698 fprintf(ofp
, "%s\n", ctx_opts
[i
].symbol
);
705 /* Close contexts element */
706 ret
= mi_lttng_writer_close_element(writer
);
721 * Add context to channel or event.
723 static int add_context(char *session_name
)
725 int ret
= CMD_SUCCESS
, warn
= 0, success
= 0;
726 struct lttng_event_context context
;
727 struct lttng_domain dom
;
728 struct ctx_type
*type
;
730 memset(&context
, 0, sizeof(context
));
731 memset(&dom
, 0, sizeof(dom
));
733 dom
.type
= get_domain();
734 handle
= lttng_create_handle(session_name
, &dom
);
735 if (handle
== NULL
) {
741 /* Open a contexts element */
742 ret
= mi_lttng_writer_open_element(writer
, config_element_contexts
);
748 /* Iterate over all the context types given */
749 cds_list_for_each_entry(type
, &ctx_type_list
.head
, list
) {
750 DBG("Adding context...");
752 populate_context(&context
, type
->opt
);
755 /* We leave context open the update the success of the command */
756 ret
= mi_lttng_context(writer
, &context
, 1);
762 ret
= mi_lttng_writer_write_element_string(writer
,
763 mi_lttng_element_context_symbol
,
771 ret
= lttng_add_context(handle
, &context
, NULL
, opt_channel_name
);
773 ERR("%s: %s", type
->opt
->symbol
, lttng_strerror(ret
));
777 if (opt_channel_name
) {
778 MSG("%s context %s added to channel %s",
779 get_domain_str(dom
.type
), type
->opt
->symbol
,
782 MSG("%s context %s added to all channels",
783 get_domain_str(dom
.type
), type
->opt
->symbol
);
789 /* Is the single operation a success ? */
790 ret
= mi_lttng_writer_write_element_bool(writer
,
791 mi_lttng_element_success
, success
);
797 /* Close the context element */
798 ret
= mi_lttng_writer_close_element(writer
);
807 /* Close contexts element */
808 ret
= mi_lttng_writer_close_element(writer
);
817 lttng_destroy_handle(handle
);
820 * This means that at least one add_context failed and tells the user to
821 * look on stderr for error(s).
830 void destroy_ctx_type(struct ctx_type
*type
)
836 free(type
->opt
->symbol
);
843 struct ctx_type
*create_ctx_type(void)
845 struct ctx_type
*type
= zmalloc(sizeof(*type
));
848 PERROR("malloc ctx_type");
852 type
->opt
= zmalloc(sizeof(*type
->opt
));
854 PERROR("malloc ctx_type options");
855 destroy_ctx_type(type
);
864 int find_ctx_type_perf_raw(const char *ctx
, struct ctx_type
*type
)
868 char *tmp_list
, *cur_list
;
870 cur_list
= tmp_list
= strdup(ctx
);
872 PERROR("strdup temp list");
877 /* Looking for "perf:[cpu|thread]:raw:<mask>:<name>". */
881 next
= strtok(cur_list
, ":");
888 if (strncmp(next
, "perf", 4) != 0) {
894 if (strncmp(next
, "cpu", 3) == 0) {
895 type
->opt
->ctx_type
= CONTEXT_PERF_CPU_COUNTER
;
896 } else if (strncmp(next
, "thread", 4) == 0) {
897 type
->opt
->ctx_type
= CONTEXT_PERF_THREAD_COUNTER
;
904 if (strncmp(next
, "raw", 3) != 0) {
913 if (strlen(next
) < 2 || next
[0] != 'r') {
914 ERR("Wrong perf raw mask format: expected rNNN");
919 type
->opt
->u
.perf
.config
= strtoll(next
+ 1, &endptr
, 16);
920 if (errno
!= 0 || !endptr
|| *endptr
) {
921 ERR("Wrong perf raw mask format: expected rNNN");
931 ERR("Too many ':' in perf raw format");
939 ERR("Invalid perf counter specifier, expected a specifier of "
940 "the form perf:cpu:raw:rNNN:<name> or "
941 "perf:thread:raw:rNNN:<name>");
955 struct ctx_type
*get_context_type(const char *ctx
)
958 struct ctx_type
*type
= NULL
;
959 const char app_ctx_prefix
[] = "$app.";
960 char *provider_name
= NULL
, *ctx_name
= NULL
;
961 size_t i
, len
, colon_pos
= 0, provider_name_len
, ctx_name_len
;
967 type
= create_ctx_type();
972 /* Check if ctx matches a known static context. */
973 opt_index
= find_ctx_type_idx(ctx
);
974 if (opt_index
>= 0) {
975 *type
->opt
= ctx_opts
[opt_index
];
976 type
->opt
->symbol
= strdup(ctx_opts
[opt_index
].symbol
);
980 /* Check if ctx is a raw perf context. */
981 ret
= find_ctx_type_perf_raw(ctx
, type
);
983 type
->opt
->u
.perf
.type
= PERF_TYPE_RAW
;
984 type
->opt
->symbol
= strdup(ctx
);
985 if (!type
->opt
->symbol
) {
986 PERROR("Copy perf field name");
993 * No match found against static contexts; check if it is an app
997 if (len
<= sizeof(app_ctx_prefix
) - 1) {
1001 /* String starts with $app. */
1002 if (strncmp(ctx
, app_ctx_prefix
, sizeof(app_ctx_prefix
) - 1)) {
1006 /* Validate that the ':' separator is present. */
1007 for (i
= sizeof(app_ctx_prefix
); i
< len
; i
++) {
1008 const char c
= ctx
[i
];
1017 * No colon found or no ctx name ("$app.provider:") or no provider name
1018 * given ("$app.:..."), which is invalid.
1020 if (!colon_pos
|| colon_pos
== len
||
1021 colon_pos
== sizeof(app_ctx_prefix
)) {
1022 ERR("Invalid application context provided: no provider or context name provided.");
1026 provider_name_len
= colon_pos
- sizeof(app_ctx_prefix
) + 2;
1027 provider_name
= zmalloc(provider_name_len
);
1028 if (!provider_name
) {
1029 PERROR("malloc provider_name");
1032 strncpy(provider_name
, ctx
+ sizeof(app_ctx_prefix
) - 1,
1033 provider_name_len
- 1);
1034 type
->opt
->u
.app_ctx
.provider_name
= provider_name
;
1036 ctx_name_len
= len
- colon_pos
;
1037 ctx_name
= zmalloc(ctx_name_len
);
1039 PERROR("malloc ctx_name");
1042 strncpy(ctx_name
, ctx
+ colon_pos
+ 1, ctx_name_len
- 1);
1043 type
->opt
->u
.app_ctx
.ctx_name
= ctx_name
;
1044 type
->opt
->ctx_type
= CONTEXT_APP_CONTEXT
;
1045 type
->opt
->symbol
= strdup(ctx
);
1049 free(provider_name
);
1051 destroy_ctx_type(type
);
1056 * Add context to channel or event.
1058 int cmd_add_context(int argc
, const char **argv
)
1060 int opt
, ret
= CMD_SUCCESS
, command_ret
= CMD_SUCCESS
;
1061 static poptContext pc
;
1062 struct ctx_type
*type
, *tmptype
;
1063 char *session_name
= NULL
;
1064 const char *leftover
= NULL
;
1071 pc
= poptGetContext(NULL
, argc
, argv
, long_options
, 0);
1072 poptReadDefaultConfig(pc
, 0);
1074 while ((opt
= poptGetNextOpt(pc
)) != -1) {
1080 ret
= print_ctx_type();
1084 type
= get_context_type(opt_type
);
1086 ERR("Unknown context type %s", opt_type
);
1090 cds_list_add_tail(&type
->list
, &ctx_type_list
.head
);
1102 case OPT_LIST_OPTIONS
:
1103 list_cmd_options(stdout
, long_options
);
1106 ret
= CMD_UNDEFINED
;
1111 leftover
= poptGetArg(pc
);
1113 ERR("Unknown argument: %s", leftover
);
1118 ret
= print_missing_or_multiple_domains(opt_kernel
+ opt_userspace
+
1119 opt_jul
+ opt_log4j
);
1126 ERR("Missing mandatory -t TYPE");
1131 if (!opt_session_name
) {
1132 session_name
= get_session_name();
1133 if (session_name
== NULL
) {
1138 session_name
= opt_session_name
;
1146 command_ret
= add_context(session_name
);
1147 ret
= mi_close(command_ret
);
1153 if (!opt_session_name
) {
1158 if (writer
&& mi_lttng_writer_destroy(writer
)) {
1159 /* Preserve original error code */
1160 ret
= ret
? ret
: LTTNG_ERR_MI_IO_FAIL
;
1163 /* Cleanup allocated memory */
1164 cds_list_for_each_entry_safe(type
, tmptype
, &ctx_type_list
.head
, list
) {
1165 destroy_ctx_type(type
);
1168 /* Overwrite ret if an error occurred during add_context() */
1169 ret
= command_ret
? command_ret
: ret
;
1171 poptFreeContext(pc
);