Commit | Line | Data |
---|---|---|
d65106b1 DG |
1 | /* |
2 | * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca> | |
e9492550 | 3 | * Copyright (C) 2016 - Jérémie Galarneau <jeremie.galarneau@efficios.com> |
d65106b1 | 4 | * |
d14d33bf AM |
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. | |
d65106b1 DG |
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 | * | |
d14d33bf AM |
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. | |
d65106b1 DG |
17 | */ |
18 | ||
6c1c0768 | 19 | #define _LGPL_SOURCE |
90b9a268 | 20 | #include <ctype.h> |
d65106b1 DG |
21 | #include <popt.h> |
22 | #include <stdio.h> | |
23 | #include <stdlib.h> | |
24 | #include <string.h> | |
25 | #include <sys/stat.h> | |
26 | #include <sys/types.h> | |
27 | #include <unistd.h> | |
3ecec76a | 28 | #include <assert.h> |
d65106b1 | 29 | |
3301e740 DG |
30 | #include <urcu/list.h> |
31 | ||
89b72577 JRJ |
32 | #include <common/mi-lttng.h> |
33 | ||
c399183f | 34 | #include "../command.h" |
d65106b1 | 35 | |
d65106b1 | 36 | static char *opt_channel_name; |
5440dc42 | 37 | static char *opt_session_name; |
55cc08a6 | 38 | static int opt_kernel; |
d65106b1 | 39 | static int opt_userspace; |
e9492550 JG |
40 | static int opt_jul; |
41 | static int opt_log4j; | |
d78d6610 | 42 | static char *opt_type; |
89b72577 | 43 | |
4fc83d94 PP |
44 | #ifdef LTTNG_EMBED_HELP |
45 | static const char help_msg[] = | |
46 | #include <lttng-add-context.1.h> | |
47 | ; | |
48 | #endif | |
49 | ||
d65106b1 DG |
50 | enum { |
51 | OPT_HELP = 1, | |
52 | OPT_TYPE, | |
eeac7d46 | 53 | OPT_USERSPACE, |
e9492550 JG |
54 | OPT_JUL, |
55 | OPT_LOG4J, | |
679b4943 | 56 | OPT_LIST_OPTIONS, |
83cea427 | 57 | OPT_LIST, |
d65106b1 DG |
58 | }; |
59 | ||
cd80958d | 60 | static struct lttng_handle *handle; |
89b72577 | 61 | static struct mi_writer *writer; |
cd80958d | 62 | |
90b9a268 DG |
63 | /* |
64 | * Taken from the LTTng ABI | |
65 | */ | |
66 | enum context_type { | |
67 | CONTEXT_PID = 0, | |
aa3514e9 | 68 | CONTEXT_PERF_COUNTER = 1, /* Backward compat. */ |
95da1297 | 69 | CONTEXT_PROCNAME = 2, |
90b9a268 DG |
70 | CONTEXT_PRIO = 3, |
71 | CONTEXT_NICE = 4, | |
72 | CONTEXT_VPID = 5, | |
73 | CONTEXT_TID = 6, | |
74 | CONTEXT_VTID = 7, | |
75 | CONTEXT_PPID = 8, | |
76 | CONTEXT_VPPID = 9, | |
9197c5c4 | 77 | CONTEXT_PTHREAD_ID = 10, |
54773d68 | 78 | CONTEXT_HOSTNAME = 11, |
7c612c2e | 79 | CONTEXT_IP = 12, |
aa3514e9 MD |
80 | CONTEXT_PERF_CPU_COUNTER = 13, |
81 | CONTEXT_PERF_THREAD_COUNTER = 14, | |
e9492550 | 82 | CONTEXT_APP_CONTEXT = 15, |
1ae5e83e JD |
83 | CONTEXT_INTERRUPTIBLE = 16, |
84 | CONTEXT_PREEMPTIBLE = 17, | |
85 | CONTEXT_NEED_RESCHEDULE = 18, | |
86 | CONTEXT_MIGRATABLE = 19, | |
3301e740 DG |
87 | }; |
88 | ||
90b9a268 DG |
89 | /* |
90 | * Taken from the Perf ABI (all enum perf_*) | |
91 | */ | |
92 | enum perf_type { | |
93 | PERF_TYPE_HARDWARE = 0, | |
94 | PERF_TYPE_SOFTWARE = 1, | |
b13d56d7 | 95 | PERF_TYPE_HW_CACHE = 3, |
4fe444da | 96 | PERF_TYPE_RAW = 4, |
3301e740 DG |
97 | }; |
98 | ||
90b9a268 | 99 | enum perf_count_hard { |
b13d56d7 MD |
100 | PERF_COUNT_HW_CPU_CYCLES = 0, |
101 | PERF_COUNT_HW_INSTRUCTIONS = 1, | |
102 | PERF_COUNT_HW_CACHE_REFERENCES = 2, | |
103 | PERF_COUNT_HW_CACHE_MISSES = 3, | |
104 | PERF_COUNT_HW_BRANCH_INSTRUCTIONS = 4, | |
105 | PERF_COUNT_HW_BRANCH_MISSES = 5, | |
106 | PERF_COUNT_HW_BUS_CYCLES = 6, | |
107 | PERF_COUNT_HW_STALLED_CYCLES_FRONTEND = 7, | |
108 | PERF_COUNT_HW_STALLED_CYCLES_BACKEND = 8, | |
90b9a268 DG |
109 | }; |
110 | ||
111 | enum perf_count_soft { | |
112 | PERF_COUNT_SW_CPU_CLOCK = 0, | |
113 | PERF_COUNT_SW_TASK_CLOCK = 1, | |
114 | PERF_COUNT_SW_PAGE_FAULTS = 2, | |
115 | PERF_COUNT_SW_CONTEXT_SWITCHES = 3, | |
116 | PERF_COUNT_SW_CPU_MIGRATIONS = 4, | |
117 | PERF_COUNT_SW_PAGE_FAULTS_MIN = 5, | |
118 | PERF_COUNT_SW_PAGE_FAULTS_MAJ = 6, | |
119 | PERF_COUNT_SW_ALIGNMENT_FAULTS = 7, | |
120 | PERF_COUNT_SW_EMULATION_FAULTS = 8, | |
3301e740 DG |
121 | }; |
122 | ||
b13d56d7 MD |
123 | /* |
124 | * Generalized hardware cache events: | |
125 | * | |
126 | * { L1-D, L1-I, LLC, ITLB, DTLB, BPU } x | |
127 | * { read, write, prefetch } x | |
128 | * { accesses, misses } | |
129 | */ | |
130 | enum perf_hw_cache_id { | |
131 | PERF_COUNT_HW_CACHE_L1D = 0, | |
132 | PERF_COUNT_HW_CACHE_L1I = 1, | |
133 | PERF_COUNT_HW_CACHE_LL = 2, | |
134 | PERF_COUNT_HW_CACHE_DTLB = 3, | |
135 | PERF_COUNT_HW_CACHE_ITLB = 4, | |
136 | PERF_COUNT_HW_CACHE_BPU = 5, | |
137 | ||
138 | PERF_COUNT_HW_CACHE_MAX, /* non-ABI */ | |
139 | }; | |
140 | ||
141 | enum perf_hw_cache_op_id { | |
142 | PERF_COUNT_HW_CACHE_OP_READ = 0, | |
143 | PERF_COUNT_HW_CACHE_OP_WRITE = 1, | |
144 | PERF_COUNT_HW_CACHE_OP_PREFETCH = 2, | |
145 | ||
146 | PERF_COUNT_HW_CACHE_OP_MAX, /* non-ABI */ | |
147 | }; | |
148 | ||
149 | enum perf_hw_cache_op_result_id { | |
150 | PERF_COUNT_HW_CACHE_RESULT_ACCESS = 0, | |
151 | PERF_COUNT_HW_CACHE_RESULT_MISS = 1, | |
152 | ||
153 | PERF_COUNT_HW_CACHE_RESULT_MAX, /* non-ABI */ | |
154 | }; | |
155 | ||
d65106b1 DG |
156 | static struct poptOption long_options[] = { |
157 | /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */ | |
158 | {"help", 'h', POPT_ARG_NONE, 0, OPT_HELP, 0, 0}, | |
5440dc42 | 159 | {"session", 's', POPT_ARG_STRING, &opt_session_name, 0, 0, 0}, |
d65106b1 | 160 | {"channel", 'c', POPT_ARG_STRING, &opt_channel_name, 0, 0, 0}, |
d65106b1 | 161 | {"kernel", 'k', POPT_ARG_VAL, &opt_kernel, 1, 0, 0}, |
d78d6610 | 162 | {"userspace", 'u', POPT_ARG_NONE, 0, OPT_USERSPACE, 0, 0}, |
e9492550 JG |
163 | {"jul", 'j', POPT_ARG_NONE, 0, OPT_JUL, 0, 0}, |
164 | {"log4j", 'l', POPT_ARG_NONE, 0, OPT_LOG4J, 0, 0}, | |
6caa2bcc | 165 | {"type", 't', POPT_ARG_STRING, &opt_type, OPT_TYPE, 0, 0}, |
83cea427 | 166 | {"list", 0, POPT_ARG_NONE, NULL, OPT_LIST, NULL, NULL}, |
679b4943 | 167 | {"list-options", 0, POPT_ARG_NONE, NULL, OPT_LIST_OPTIONS, NULL, NULL}, |
d65106b1 DG |
168 | {0, 0, 0, 0, 0, 0, 0} |
169 | }; | |
170 | ||
90b9a268 | 171 | /* |
b13d56d7 | 172 | * Context options |
90b9a268 | 173 | */ |
aa3514e9 | 174 | #define PERF_HW(optstr, name, type, hide) \ |
b13d56d7 | 175 | { \ |
aa3514e9 | 176 | optstr, type, hide, \ |
b13d56d7 MD |
177 | .u.perf = { PERF_TYPE_HARDWARE, PERF_COUNT_HW_##name, },\ |
178 | } | |
90b9a268 | 179 | |
aa3514e9 | 180 | #define PERF_SW(optstr, name, type, hide) \ |
b13d56d7 | 181 | { \ |
aa3514e9 | 182 | optstr, type, hide, \ |
b13d56d7 MD |
183 | .u.perf = { PERF_TYPE_SOFTWARE, PERF_COUNT_SW_##name, },\ |
184 | } | |
90b9a268 | 185 | |
aa3514e9 | 186 | #define _PERF_HW_CACHE(optstr, name, type, op, result, hide) \ |
b13d56d7 | 187 | { \ |
aa3514e9 | 188 | optstr, type, hide, \ |
b13d56d7 MD |
189 | .u.perf = { \ |
190 | PERF_TYPE_HW_CACHE, \ | |
191 | (uint64_t) PERF_COUNT_HW_CACHE_##name \ | |
18829107 MD |
192 | | ((uint64_t) PERF_COUNT_HW_CACHE_OP_##op << 8) \ |
193 | | ((uint64_t) PERF_COUNT_HW_CACHE_RESULT_##result << 16), \ | |
b13d56d7 MD |
194 | }, \ |
195 | } | |
196 | ||
aa3514e9 MD |
197 | #define PERF_HW_CACHE(optstr, name, type, hide) \ |
198 | _PERF_HW_CACHE(optstr "-loads", name, type, \ | |
199 | READ, ACCESS, hide), \ | |
200 | _PERF_HW_CACHE(optstr "-load-misses", name, type, \ | |
201 | READ, MISS, hide), \ | |
202 | _PERF_HW_CACHE(optstr "-stores", name, type, \ | |
203 | WRITE, ACCESS, hide), \ | |
204 | _PERF_HW_CACHE(optstr "-store-misses", name, type, \ | |
205 | WRITE, MISS, hide), \ | |
206 | _PERF_HW_CACHE(optstr "-prefetches", name, type, \ | |
207 | PREFETCH, ACCESS, hide), \ | |
208 | _PERF_HW_CACHE(optstr "-prefetch-misses", name, type, \ | |
209 | PREFETCH, MISS, hide) | |
b13d56d7 MD |
210 | |
211 | static | |
212 | const struct ctx_opts { | |
90b9a268 | 213 | char *symbol; |
b13d56d7 | 214 | enum context_type ctx_type; |
aa3514e9 | 215 | int hide_help; /* Hide from --help */ |
b13d56d7 MD |
216 | union { |
217 | struct { | |
218 | uint32_t type; | |
219 | uint64_t config; | |
220 | } perf; | |
e9492550 JG |
221 | struct { |
222 | char *provider_name; | |
223 | char *ctx_name; | |
224 | } app_ctx; | |
b13d56d7 MD |
225 | } u; |
226 | } ctx_opts[] = { | |
227 | { "pid", CONTEXT_PID }, | |
95da1297 | 228 | { "procname", CONTEXT_PROCNAME }, |
b13d56d7 MD |
229 | { "prio", CONTEXT_PRIO }, |
230 | { "nice", CONTEXT_NICE }, | |
231 | { "vpid", CONTEXT_VPID }, | |
232 | { "tid", CONTEXT_TID }, | |
9197c5c4 | 233 | { "pthread_id", CONTEXT_PTHREAD_ID }, |
b13d56d7 MD |
234 | { "vtid", CONTEXT_VTID }, |
235 | { "ppid", CONTEXT_PPID }, | |
236 | { "vppid", CONTEXT_VPPID }, | |
54773d68 | 237 | { "hostname", CONTEXT_HOSTNAME }, |
7c612c2e | 238 | { "ip", CONTEXT_IP }, |
1ae5e83e JD |
239 | { "interruptible", CONTEXT_INTERRUPTIBLE }, |
240 | { "preemptible", CONTEXT_PREEMPTIBLE }, | |
241 | { "need_reschedule", CONTEXT_NEED_RESCHEDULE }, | |
242 | { "migratable", CONTEXT_MIGRATABLE }, | |
aa3514e9 | 243 | |
b13d56d7 | 244 | /* Perf options */ |
aa3514e9 MD |
245 | |
246 | /* Perf per-CPU counters */ | |
247 | PERF_HW("perf:cpu:cpu-cycles", CPU_CYCLES, | |
248 | CONTEXT_PERF_CPU_COUNTER, 0), | |
249 | PERF_HW("perf:cpu:cycles", CPU_CYCLES, | |
250 | CONTEXT_PERF_CPU_COUNTER, 0), | |
251 | PERF_HW("perf:cpu:stalled-cycles-frontend", STALLED_CYCLES_FRONTEND, | |
252 | CONTEXT_PERF_CPU_COUNTER, 0), | |
253 | PERF_HW("perf:cpu:idle-cycles-frontend", STALLED_CYCLES_FRONTEND, | |
254 | CONTEXT_PERF_CPU_COUNTER, 0), | |
255 | PERF_HW("perf:cpu:stalled-cycles-backend", STALLED_CYCLES_BACKEND, | |
256 | CONTEXT_PERF_CPU_COUNTER, 0), | |
257 | PERF_HW("perf:cpu:idle-cycles-backend", STALLED_CYCLES_BACKEND, | |
258 | CONTEXT_PERF_CPU_COUNTER, 0), | |
259 | PERF_HW("perf:cpu:instructions", INSTRUCTIONS, | |
260 | CONTEXT_PERF_CPU_COUNTER, 0), | |
261 | PERF_HW("perf:cpu:cache-references", CACHE_REFERENCES, | |
262 | CONTEXT_PERF_CPU_COUNTER, 0), | |
263 | PERF_HW("perf:cpu:cache-misses", CACHE_MISSES, | |
264 | CONTEXT_PERF_CPU_COUNTER, 0), | |
265 | PERF_HW("perf:cpu:branch-instructions", BRANCH_INSTRUCTIONS, | |
266 | CONTEXT_PERF_CPU_COUNTER, 0), | |
267 | PERF_HW("perf:cpu:branches", BRANCH_INSTRUCTIONS, | |
268 | CONTEXT_PERF_CPU_COUNTER, 0), | |
269 | PERF_HW("perf:cpu:branch-misses", BRANCH_MISSES, | |
270 | CONTEXT_PERF_CPU_COUNTER, 0), | |
271 | PERF_HW("perf:cpu:bus-cycles", BUS_CYCLES, | |
272 | CONTEXT_PERF_CPU_COUNTER, 0), | |
273 | ||
274 | PERF_HW_CACHE("perf:cpu:L1-dcache", L1D, | |
275 | CONTEXT_PERF_CPU_COUNTER, 0), | |
276 | PERF_HW_CACHE("perf:cpu:L1-icache", L1I, | |
277 | CONTEXT_PERF_CPU_COUNTER, 0), | |
278 | PERF_HW_CACHE("perf:cpu:LLC", LL, | |
279 | CONTEXT_PERF_CPU_COUNTER, 0), | |
280 | PERF_HW_CACHE("perf:cpu:dTLB", DTLB, | |
281 | CONTEXT_PERF_CPU_COUNTER, 0), | |
282 | _PERF_HW_CACHE("perf:cpu:iTLB-loads", ITLB, | |
283 | CONTEXT_PERF_CPU_COUNTER, READ, ACCESS, 0), | |
284 | _PERF_HW_CACHE("perf:cpu:iTLB-load-misses", ITLB, | |
285 | CONTEXT_PERF_CPU_COUNTER, READ, MISS, 0), | |
286 | _PERF_HW_CACHE("perf:cpu:branch-loads", BPU, | |
287 | CONTEXT_PERF_CPU_COUNTER, READ, ACCESS, 0), | |
288 | _PERF_HW_CACHE("perf:cpu:branch-load-misses", BPU, | |
289 | CONTEXT_PERF_CPU_COUNTER, READ, MISS, 0), | |
290 | ||
291 | PERF_SW("perf:cpu:cpu-clock", CPU_CLOCK, | |
292 | CONTEXT_PERF_CPU_COUNTER, 0), | |
293 | PERF_SW("perf:cpu:task-clock", TASK_CLOCK, | |
294 | CONTEXT_PERF_CPU_COUNTER, 0), | |
295 | PERF_SW("perf:cpu:page-fault", PAGE_FAULTS, | |
296 | CONTEXT_PERF_CPU_COUNTER, 0), | |
297 | PERF_SW("perf:cpu:faults", PAGE_FAULTS, | |
298 | CONTEXT_PERF_CPU_COUNTER, 0), | |
299 | PERF_SW("perf:cpu:major-faults", PAGE_FAULTS_MAJ, | |
300 | CONTEXT_PERF_CPU_COUNTER, 0), | |
301 | PERF_SW("perf:cpu:minor-faults", PAGE_FAULTS_MIN, | |
302 | CONTEXT_PERF_CPU_COUNTER, 0), | |
303 | PERF_SW("perf:cpu:context-switches", CONTEXT_SWITCHES, | |
304 | CONTEXT_PERF_CPU_COUNTER, 0), | |
305 | PERF_SW("perf:cpu:cs", CONTEXT_SWITCHES, | |
306 | CONTEXT_PERF_CPU_COUNTER, 0), | |
307 | PERF_SW("perf:cpu:cpu-migrations", CPU_MIGRATIONS, | |
308 | CONTEXT_PERF_CPU_COUNTER, 0), | |
309 | PERF_SW("perf:cpu:migrations", CPU_MIGRATIONS, | |
310 | CONTEXT_PERF_CPU_COUNTER, 0), | |
311 | PERF_SW("perf:cpu:alignment-faults", ALIGNMENT_FAULTS, | |
312 | CONTEXT_PERF_CPU_COUNTER, 0), | |
313 | PERF_SW("perf:cpu:emulation-faults", EMULATION_FAULTS, | |
314 | CONTEXT_PERF_CPU_COUNTER, 0), | |
315 | ||
316 | /* Perf per-thread counters */ | |
317 | PERF_HW("perf:thread:cpu-cycles", CPU_CYCLES, | |
318 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
319 | PERF_HW("perf:thread:cycles", CPU_CYCLES, | |
320 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
321 | PERF_HW("perf:thread:stalled-cycles-frontend", STALLED_CYCLES_FRONTEND, | |
322 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
323 | PERF_HW("perf:thread:idle-cycles-frontend", STALLED_CYCLES_FRONTEND, | |
324 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
325 | PERF_HW("perf:thread:stalled-cycles-backend", STALLED_CYCLES_BACKEND, | |
326 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
327 | PERF_HW("perf:thread:idle-cycles-backend", STALLED_CYCLES_BACKEND, | |
328 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
329 | PERF_HW("perf:thread:instructions", INSTRUCTIONS, | |
330 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
331 | PERF_HW("perf:thread:cache-references", CACHE_REFERENCES, | |
332 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
333 | PERF_HW("perf:thread:cache-misses", CACHE_MISSES, | |
334 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
335 | PERF_HW("perf:thread:branch-instructions", BRANCH_INSTRUCTIONS, | |
336 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
337 | PERF_HW("perf:thread:branches", BRANCH_INSTRUCTIONS, | |
338 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
339 | PERF_HW("perf:thread:branch-misses", BRANCH_MISSES, | |
340 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
341 | PERF_HW("perf:thread:bus-cycles", BUS_CYCLES, | |
342 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
343 | ||
344 | PERF_HW_CACHE("perf:thread:L1-dcache", L1D, | |
345 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
346 | PERF_HW_CACHE("perf:thread:L1-icache", L1I, | |
347 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
348 | PERF_HW_CACHE("perf:thread:LLC", LL, | |
349 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
350 | PERF_HW_CACHE("perf:thread:dTLB", DTLB, | |
351 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
352 | _PERF_HW_CACHE("perf:thread:iTLB-loads", ITLB, | |
353 | CONTEXT_PERF_THREAD_COUNTER, READ, ACCESS, 0), | |
354 | _PERF_HW_CACHE("perf:thread:iTLB-load-misses", ITLB, | |
355 | CONTEXT_PERF_THREAD_COUNTER, READ, MISS, 0), | |
356 | _PERF_HW_CACHE("perf:thread:branch-loads", BPU, | |
357 | CONTEXT_PERF_THREAD_COUNTER, READ, ACCESS, 0), | |
358 | _PERF_HW_CACHE("perf:thread:branch-load-misses", BPU, | |
359 | CONTEXT_PERF_THREAD_COUNTER, READ, MISS, 0), | |
360 | ||
361 | PERF_SW("perf:thread:cpu-clock", CPU_CLOCK, | |
362 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
363 | PERF_SW("perf:thread:task-clock", TASK_CLOCK, | |
364 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
365 | PERF_SW("perf:thread:page-fault", PAGE_FAULTS, | |
366 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
367 | PERF_SW("perf:thread:faults", PAGE_FAULTS, | |
368 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
369 | PERF_SW("perf:thread:major-faults", PAGE_FAULTS_MAJ, | |
370 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
371 | PERF_SW("perf:thread:minor-faults", PAGE_FAULTS_MIN, | |
372 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
373 | PERF_SW("perf:thread:context-switches", CONTEXT_SWITCHES, | |
374 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
375 | PERF_SW("perf:thread:cs", CONTEXT_SWITCHES, | |
376 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
377 | PERF_SW("perf:thread:cpu-migrations", CPU_MIGRATIONS, | |
378 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
379 | PERF_SW("perf:thread:migrations", CPU_MIGRATIONS, | |
380 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
381 | PERF_SW("perf:thread:alignment-faults", ALIGNMENT_FAULTS, | |
382 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
383 | PERF_SW("perf:thread:emulation-faults", EMULATION_FAULTS, | |
384 | CONTEXT_PERF_THREAD_COUNTER, 0), | |
385 | ||
386 | /* | |
387 | * Perf per-CPU counters, backward compatibilty for names. | |
388 | * Hidden from help listing. | |
389 | */ | |
390 | PERF_HW("perf:cpu-cycles", CPU_CYCLES, | |
391 | CONTEXT_PERF_COUNTER, 1), | |
392 | PERF_HW("perf:cycles", CPU_CYCLES, | |
393 | CONTEXT_PERF_COUNTER, 1), | |
394 | PERF_HW("perf:stalled-cycles-frontend", STALLED_CYCLES_FRONTEND, | |
395 | CONTEXT_PERF_COUNTER, 1), | |
396 | PERF_HW("perf:idle-cycles-frontend", STALLED_CYCLES_FRONTEND, | |
397 | CONTEXT_PERF_COUNTER, 1), | |
398 | PERF_HW("perf:stalled-cycles-backend", STALLED_CYCLES_BACKEND, | |
399 | CONTEXT_PERF_COUNTER, 1), | |
400 | PERF_HW("perf:idle-cycles-backend", STALLED_CYCLES_BACKEND, | |
401 | CONTEXT_PERF_COUNTER, 1), | |
402 | PERF_HW("perf:instructions", INSTRUCTIONS, | |
403 | CONTEXT_PERF_COUNTER, 1), | |
404 | PERF_HW("perf:cache-references", CACHE_REFERENCES, | |
405 | CONTEXT_PERF_COUNTER, 1), | |
406 | PERF_HW("perf:cache-misses", CACHE_MISSES, | |
407 | CONTEXT_PERF_COUNTER, 1), | |
408 | PERF_HW("perf:branch-instructions", BRANCH_INSTRUCTIONS, | |
409 | CONTEXT_PERF_COUNTER, 1), | |
410 | PERF_HW("perf:branches", BRANCH_INSTRUCTIONS, | |
411 | CONTEXT_PERF_COUNTER, 1), | |
412 | PERF_HW("perf:branch-misses", BRANCH_MISSES, | |
413 | CONTEXT_PERF_COUNTER, 1), | |
414 | PERF_HW("perf:bus-cycles", BUS_CYCLES, | |
415 | CONTEXT_PERF_COUNTER, 1), | |
416 | ||
417 | PERF_HW_CACHE("perf:L1-dcache", L1D, | |
418 | CONTEXT_PERF_COUNTER, 1), | |
419 | PERF_HW_CACHE("perf:L1-icache", L1I, | |
420 | CONTEXT_PERF_COUNTER, 1), | |
421 | PERF_HW_CACHE("perf:LLC", LL, | |
422 | CONTEXT_PERF_COUNTER, 1), | |
423 | PERF_HW_CACHE("perf:dTLB", DTLB, | |
424 | CONTEXT_PERF_COUNTER, 1), | |
425 | _PERF_HW_CACHE("perf:iTLB-loads", ITLB, | |
426 | CONTEXT_PERF_COUNTER, READ, ACCESS, 1), | |
427 | _PERF_HW_CACHE("perf:iTLB-load-misses", ITLB, | |
428 | CONTEXT_PERF_COUNTER, READ, MISS, 1), | |
429 | _PERF_HW_CACHE("perf:branch-loads", BPU, | |
430 | CONTEXT_PERF_COUNTER, READ, ACCESS, 1), | |
431 | _PERF_HW_CACHE("perf:branch-load-misses", BPU, | |
432 | CONTEXT_PERF_COUNTER, READ, MISS, 1), | |
433 | ||
434 | PERF_SW("perf:cpu-clock", CPU_CLOCK, | |
435 | CONTEXT_PERF_COUNTER, 1), | |
436 | PERF_SW("perf:task-clock", TASK_CLOCK, | |
437 | CONTEXT_PERF_COUNTER, 1), | |
438 | PERF_SW("perf:page-fault", PAGE_FAULTS, | |
439 | CONTEXT_PERF_COUNTER, 1), | |
440 | PERF_SW("perf:faults", PAGE_FAULTS, | |
441 | CONTEXT_PERF_COUNTER, 1), | |
442 | PERF_SW("perf:major-faults", PAGE_FAULTS_MAJ, | |
443 | CONTEXT_PERF_COUNTER, 1), | |
444 | PERF_SW("perf:minor-faults", PAGE_FAULTS_MIN, | |
445 | CONTEXT_PERF_COUNTER, 1), | |
446 | PERF_SW("perf:context-switches", CONTEXT_SWITCHES, | |
447 | CONTEXT_PERF_COUNTER, 1), | |
448 | PERF_SW("perf:cs", CONTEXT_SWITCHES, | |
449 | CONTEXT_PERF_COUNTER, 1), | |
450 | PERF_SW("perf:cpu-migrations", CPU_MIGRATIONS, | |
451 | CONTEXT_PERF_COUNTER, 1), | |
452 | PERF_SW("perf:migrations", CPU_MIGRATIONS, | |
453 | CONTEXT_PERF_COUNTER, 1), | |
454 | PERF_SW("perf:alignment-faults", ALIGNMENT_FAULTS, | |
455 | CONTEXT_PERF_COUNTER, 1), | |
456 | PERF_SW("perf:emulation-faults", EMULATION_FAULTS, | |
457 | CONTEXT_PERF_COUNTER, 1), | |
458 | ||
b13d56d7 | 459 | { NULL, -1 }, /* Closure */ |
90b9a268 DG |
460 | }; |
461 | ||
aa3514e9 MD |
462 | #undef PERF_HW_CACHE |
463 | #undef _PERF_HW_CACHE | |
b13d56d7 MD |
464 | #undef PERF_SW |
465 | #undef PERF_HW | |
466 | ||
90b9a268 | 467 | /* |
b13d56d7 | 468 | * Context type for command line option parsing. |
90b9a268 | 469 | */ |
b13d56d7 | 470 | struct ctx_type { |
e9492550 | 471 | struct ctx_opts *opt; |
b13d56d7 | 472 | struct cds_list_head list; |
90b9a268 DG |
473 | }; |
474 | ||
475 | /* | |
476 | * List of context type. Use to enable multiple context on a single command | |
477 | * line entry. | |
478 | */ | |
479 | struct ctx_type_list { | |
480 | struct cds_list_head head; | |
481 | } ctx_type_list = { | |
482 | .head = CDS_LIST_HEAD_INIT(ctx_type_list.head), | |
483 | }; | |
484 | ||
90b9a268 | 485 | |
90b9a268 | 486 | |
90b9a268 DG |
487 | /* |
488 | * Find context numerical value from string. | |
e9492550 JG |
489 | * |
490 | * Return -1 if not found. | |
90b9a268 DG |
491 | */ |
492 | static int find_ctx_type_idx(const char *opt) | |
493 | { | |
e9492550 | 494 | int ret, i = 0; |
90b9a268 DG |
495 | |
496 | while (ctx_opts[i].symbol != NULL) { | |
497 | if (strcmp(opt, ctx_opts[i].symbol) == 0) { | |
498 | ret = i; | |
499 | goto end; | |
500 | } | |
501 | i++; | |
502 | } | |
503 | ||
e9492550 | 504 | ret = -1; |
90b9a268 DG |
505 | end: |
506 | return ret; | |
507 | } | |
508 | ||
e9492550 JG |
509 | static |
510 | enum lttng_domain_type get_domain(void) | |
511 | { | |
512 | if (opt_kernel) { | |
513 | return LTTNG_DOMAIN_KERNEL; | |
514 | } else if (opt_userspace) { | |
515 | return LTTNG_DOMAIN_UST; | |
516 | } else if (opt_jul) { | |
517 | return LTTNG_DOMAIN_JUL; | |
518 | } else if (opt_log4j) { | |
519 | return LTTNG_DOMAIN_LOG4J; | |
520 | } else { | |
521 | assert(0); | |
522 | } | |
523 | } | |
524 | ||
59deec0c JR |
525 | static |
526 | int mi_open(void) | |
527 | { | |
528 | int ret; | |
529 | ||
530 | /* MI check */ | |
531 | if (!lttng_opt_mi) { | |
532 | ret = 0; | |
533 | goto end; | |
534 | } | |
535 | ||
536 | ret = fileno(stdout); | |
537 | if (ret < 0) { | |
538 | PERROR("Unable to retrive fileno of stdout"); | |
539 | ret = CMD_ERROR; | |
540 | goto end; | |
541 | } | |
542 | ||
543 | writer = mi_lttng_writer_create(ret, lttng_opt_mi); | |
544 | if (!writer) { | |
545 | ret = CMD_ERROR; | |
546 | goto end; | |
547 | } | |
548 | ||
549 | /* Open command element */ | |
550 | ret = mi_lttng_writer_command_open(writer, | |
551 | mi_lttng_element_command_add_context); | |
552 | if (ret) { | |
553 | ret = CMD_ERROR; | |
554 | goto end; | |
555 | } | |
556 | ||
557 | /* Open output element */ | |
558 | ret = mi_lttng_writer_open_element(writer, | |
559 | mi_lttng_element_command_output); | |
560 | if (ret) { | |
561 | ret = CMD_ERROR; | |
562 | goto end; | |
563 | } | |
564 | end: | |
565 | return ret; | |
566 | } | |
567 | ||
568 | static | |
569 | int mi_close(enum cmd_error_code success) | |
570 | { | |
571 | int ret; | |
572 | ||
573 | /* MI closing */ | |
574 | if (!lttng_opt_mi) { | |
575 | ret = 0; | |
576 | goto end; | |
577 | } | |
578 | /* Close output element */ | |
579 | ret = mi_lttng_writer_close_element(writer); | |
580 | if (ret) { | |
581 | ret = CMD_ERROR; | |
582 | goto end; | |
583 | } | |
584 | ||
585 | /* Success ? */ | |
586 | ret = mi_lttng_writer_write_element_bool(writer, | |
587 | mi_lttng_element_command_success, !success); | |
588 | if (ret) { | |
589 | ret = CMD_ERROR; | |
590 | goto end; | |
591 | } | |
592 | ||
593 | /* Command element close */ | |
594 | ret = mi_lttng_writer_command_close(writer); | |
595 | if (ret) { | |
596 | ret = CMD_ERROR; | |
597 | goto end; | |
598 | } | |
599 | end: | |
600 | return ret; | |
601 | } | |
602 | ||
603 | static | |
604 | void populate_context(struct lttng_event_context *context, | |
605 | const struct ctx_opts *opt) | |
606 | { | |
607 | char *ptr; | |
608 | ||
609 | context->ctx = (enum lttng_event_context_type) opt->ctx_type; | |
610 | switch (context->ctx) { | |
611 | case LTTNG_EVENT_CONTEXT_PERF_COUNTER: | |
612 | case LTTNG_EVENT_CONTEXT_PERF_CPU_COUNTER: | |
613 | case LTTNG_EVENT_CONTEXT_PERF_THREAD_COUNTER: | |
614 | context->u.perf_counter.type = opt->u.perf.type; | |
615 | context->u.perf_counter.config = opt->u.perf.config; | |
616 | strncpy(context->u.perf_counter.name, opt->symbol, | |
617 | LTTNG_SYMBOL_NAME_LEN); | |
618 | context->u.perf_counter.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0'; | |
619 | /* Replace : and - by _ */ | |
620 | while ((ptr = strchr(context->u.perf_counter.name, '-')) != NULL) { | |
621 | *ptr = '_'; | |
622 | } | |
623 | while ((ptr = strchr(context->u.perf_counter.name, ':')) != NULL) { | |
624 | *ptr = '_'; | |
625 | } | |
626 | break; | |
627 | case LTTNG_EVENT_CONTEXT_APP_CONTEXT: | |
628 | context->u.app_ctx.provider_name = | |
629 | opt->u.app_ctx.provider_name; | |
630 | context->u.app_ctx.ctx_name = | |
631 | opt->u.app_ctx.ctx_name; | |
632 | break; | |
633 | default: | |
634 | break; | |
635 | } | |
636 | } | |
637 | ||
638 | /* | |
639 | * Pretty print context type. | |
640 | */ | |
641 | static | |
642 | int print_ctx_type(void) | |
643 | { | |
644 | ||
645 | FILE *ofp = stdout; | |
646 | int i = 0; | |
647 | int ret; | |
648 | struct lttng_event_context context; | |
649 | ||
650 | memset(&context, 0, sizeof(context)); | |
651 | ||
652 | ret = mi_open(); | |
653 | if (ret) { | |
654 | ret = CMD_ERROR; | |
655 | goto end; | |
656 | } | |
657 | ||
658 | if (lttng_opt_mi) { | |
659 | /* Open a contexts element */ | |
660 | ret = mi_lttng_writer_open_element(writer, config_element_contexts); | |
661 | if (ret) { | |
662 | ret = CMD_ERROR; | |
663 | goto end; | |
664 | } | |
665 | } | |
666 | ||
667 | while (ctx_opts[i].symbol != NULL) { | |
668 | if (!ctx_opts[i].hide_help) { | |
669 | if (lttng_opt_mi) { | |
670 | populate_context(&context, &ctx_opts[i]); | |
671 | ret = mi_lttng_context(writer, &context, 1); | |
672 | if (ret) { | |
673 | ret = CMD_ERROR; | |
674 | goto end; | |
675 | } | |
676 | ||
677 | ret = mi_lttng_writer_write_element_string( | |
678 | writer, | |
679 | mi_lttng_element_context_symbol, | |
680 | ctx_opts[i].symbol); | |
681 | if (ret) { | |
682 | ret = CMD_ERROR; | |
683 | goto end; | |
684 | } | |
685 | ||
686 | ret = mi_lttng_writer_close_element(writer); | |
687 | if (ret) { | |
688 | ret = CMD_ERROR; | |
689 | goto end; | |
690 | } | |
691 | } else { | |
692 | fprintf(ofp, "%s\n", ctx_opts[i].symbol); | |
693 | } | |
694 | } | |
695 | i++; | |
696 | } | |
697 | ||
698 | if (lttng_opt_mi) { | |
699 | /* Close contexts element */ | |
700 | ret = mi_lttng_writer_close_element(writer); | |
701 | if (ret) { | |
702 | goto end; | |
703 | } | |
704 | } | |
705 | ||
706 | end: | |
707 | ret = mi_close(ret); | |
708 | if (ret) { | |
709 | ret = CMD_ERROR; | |
710 | } | |
711 | return ret; | |
712 | } | |
713 | ||
90b9a268 DG |
714 | /* |
715 | * Add context to channel or event. | |
d65106b1 | 716 | */ |
cd80958d | 717 | static int add_context(char *session_name) |
d65106b1 | 718 | { |
89b72577 | 719 | int ret = CMD_SUCCESS, warn = 0, success = 0; |
7d29a247 DG |
720 | struct lttng_event_context context; |
721 | struct lttng_domain dom; | |
3301e740 | 722 | struct ctx_type *type; |
d65106b1 | 723 | |
441c16a7 MD |
724 | memset(&context, 0, sizeof(context)); |
725 | memset(&dom, 0, sizeof(dom)); | |
726 | ||
e9492550 | 727 | dom.type = get_domain(); |
cd80958d DG |
728 | handle = lttng_create_handle(session_name, &dom); |
729 | if (handle == NULL) { | |
af87c45a | 730 | ret = CMD_ERROR; |
cd80958d DG |
731 | goto error; |
732 | } | |
733 | ||
89b72577 JRJ |
734 | if (lttng_opt_mi) { |
735 | /* Open a contexts element */ | |
736 | ret = mi_lttng_writer_open_element(writer, config_element_contexts); | |
737 | if (ret) { | |
738 | goto error; | |
739 | } | |
740 | } | |
741 | ||
5eb00805 | 742 | /* Iterate over all the context types given */ |
3301e740 | 743 | cds_list_for_each_entry(type, &ctx_type_list.head, list) { |
55cc08a6 DG |
744 | DBG("Adding context..."); |
745 | ||
59deec0c JR |
746 | populate_context(&context, type->opt); |
747 | ||
89b72577 JRJ |
748 | if (lttng_opt_mi) { |
749 | /* We leave context open the update the success of the command */ | |
750 | ret = mi_lttng_context(writer, &context, 1); | |
751 | if (ret) { | |
752 | ret = CMD_ERROR; | |
753 | goto error; | |
754 | } | |
59deec0c JR |
755 | |
756 | ret = mi_lttng_writer_write_element_string(writer, | |
757 | mi_lttng_element_context_symbol, | |
758 | type->opt->symbol); | |
759 | if (ret) { | |
760 | ret = CMD_ERROR; | |
761 | goto error; | |
762 | } | |
89b72577 JRJ |
763 | } |
764 | ||
601d5acf | 765 | ret = lttng_add_context(handle, &context, NULL, opt_channel_name); |
55cc08a6 | 766 | if (ret < 0) { |
b58471ff | 767 | ERR("%s: %s", type->opt->symbol, lttng_strerror(ret)); |
d16c1a4c | 768 | warn = 1; |
89b72577 | 769 | success = 0; |
d65106b1 | 770 | } else { |
601d5acf | 771 | if (opt_channel_name) { |
b58471ff | 772 | MSG("%s context %s added to channel %s", |
0df4a6dd | 773 | get_domain_str(dom.type), type->opt->symbol, |
b58471ff | 774 | opt_channel_name); |
b58471ff DG |
775 | } else { |
776 | MSG("%s context %s added to all channels", | |
62a7b8ed | 777 | get_domain_str(dom.type), type->opt->symbol); |
b58471ff | 778 | } |
89b72577 JRJ |
779 | success = 1; |
780 | } | |
781 | ||
782 | if (lttng_opt_mi) { | |
783 | /* Is the single operation a success ? */ | |
784 | ret = mi_lttng_writer_write_element_bool(writer, | |
785 | mi_lttng_element_success, success); | |
786 | if (ret) { | |
787 | ret = CMD_ERROR; | |
788 | goto error; | |
789 | } | |
790 | ||
791 | /* Close the context element */ | |
792 | ret = mi_lttng_writer_close_element(writer); | |
793 | if (ret) { | |
794 | ret = CMD_ERROR; | |
795 | goto error; | |
796 | } | |
797 | } | |
798 | } | |
799 | ||
800 | if (lttng_opt_mi) { | |
801 | /* Close contexts element */ | |
802 | ret = mi_lttng_writer_close_element(writer); | |
803 | if (ret) { | |
804 | goto error; | |
d65106b1 | 805 | } |
d65106b1 DG |
806 | } |
807 | ||
af87c45a DG |
808 | ret = CMD_SUCCESS; |
809 | ||
d65106b1 | 810 | error: |
cd80958d DG |
811 | lttng_destroy_handle(handle); |
812 | ||
d16c1a4c DG |
813 | /* |
814 | * This means that at least one add_context failed and tells the user to | |
815 | * look on stderr for error(s). | |
816 | */ | |
89b72577 | 817 | if (!ret && warn) { |
d16c1a4c DG |
818 | ret = CMD_WARNING; |
819 | } | |
d65106b1 DG |
820 | return ret; |
821 | } | |
822 | ||
e9492550 JG |
823 | static |
824 | void destroy_ctx_type(struct ctx_type *type) | |
825 | { | |
826 | if (!type) { | |
827 | return; | |
828 | } | |
1e19c0f6 JR |
829 | if (type->opt) { |
830 | free(type->opt->symbol); | |
831 | } | |
e9492550 JG |
832 | free(type->opt); |
833 | free(type); | |
834 | } | |
835 | ||
836 | static | |
837 | struct ctx_type *create_ctx_type(void) | |
838 | { | |
839 | struct ctx_type *type = zmalloc(sizeof(*type)); | |
840 | ||
841 | if (!type) { | |
842 | PERROR("malloc ctx_type"); | |
843 | goto end; | |
844 | } | |
845 | ||
846 | type->opt = zmalloc(sizeof(*type->opt)); | |
847 | if (!type->opt) { | |
848 | PERROR("malloc ctx_type options"); | |
849 | destroy_ctx_type(type); | |
8e32b63b | 850 | type = NULL; |
e9492550 JG |
851 | goto end; |
852 | } | |
853 | end: | |
854 | return type; | |
855 | } | |
856 | ||
4fe444da JD |
857 | static |
858 | int find_ctx_type_perf_raw(const char *ctx, struct ctx_type *type) | |
859 | { | |
860 | int ret; | |
861 | int field_pos = 0; | |
862 | char *tmp_list, *cur_list; | |
863 | ||
864 | cur_list = tmp_list = strdup(ctx); | |
865 | if (!tmp_list) { | |
866 | PERROR("strdup temp list"); | |
867 | ret = -ENOMEM; | |
868 | goto end; | |
869 | } | |
870 | ||
871 | /* Looking for "perf:[cpu|thread]:raw:<mask>:<name>". */ | |
872 | for (;;) { | |
873 | char *next; | |
874 | ||
875 | next = strtok(cur_list, ":"); | |
876 | if (!next) { | |
877 | break; | |
878 | } | |
879 | cur_list = NULL; | |
880 | switch (field_pos) { | |
881 | case 0: | |
882 | if (strncmp(next, "perf", 4) != 0) { | |
883 | ret = -1; | |
884 | goto end; | |
885 | } | |
886 | break; | |
887 | case 1: | |
888 | if (strncmp(next, "cpu", 3) == 0) { | |
889 | type->opt->ctx_type = CONTEXT_PERF_CPU_COUNTER; | |
890 | } else if (strncmp(next, "thread", 4) == 0) { | |
891 | type->opt->ctx_type = CONTEXT_PERF_THREAD_COUNTER; | |
892 | } else { | |
893 | ret = -1; | |
894 | goto end; | |
895 | } | |
896 | break; | |
897 | case 2: | |
898 | if (strncmp(next, "raw", 3) != 0) { | |
899 | ret = -1; | |
900 | goto end; | |
901 | } | |
902 | break; | |
903 | case 3: | |
904 | { | |
905 | char *endptr; | |
906 | ||
907 | if (strlen(next) < 2 || next[0] != 'r') { | |
908 | ERR("Wrong perf raw mask format: expected rNNN"); | |
909 | ret = -1; | |
910 | goto end; | |
911 | } | |
912 | errno = 0; | |
913 | type->opt->u.perf.config = strtoll(next + 1, &endptr, 16); | |
914 | if (errno != 0 || !endptr || *endptr) { | |
915 | ERR("Wrong perf raw mask format: expected rNNN"); | |
916 | ret = -1; | |
917 | goto end; | |
918 | } | |
919 | break; | |
920 | } | |
921 | case 4: | |
922 | /* name */ | |
923 | break; | |
924 | case 5: | |
925 | ERR("Too many ':' in perf raw format"); | |
926 | ret = -1; | |
927 | goto end; | |
928 | }; | |
929 | field_pos++; | |
930 | } | |
931 | ||
932 | if (field_pos < 5) { | |
933 | ERR("Invalid perf counter specifier, expected a specifier of " | |
934 | "the form perf:cpu:raw:rNNN:<name> or " | |
935 | "perf:thread:raw:rNNN:<name>"); | |
936 | ret = -1; | |
937 | goto end; | |
938 | } | |
939 | ||
940 | ret = 0; | |
941 | goto end; | |
942 | ||
943 | end: | |
944 | free(tmp_list); | |
945 | return ret; | |
946 | } | |
947 | ||
e9492550 JG |
948 | static |
949 | struct ctx_type *get_context_type(const char *ctx) | |
950 | { | |
4fe444da | 951 | int opt_index, ret; |
e9492550 JG |
952 | struct ctx_type *type = NULL; |
953 | const char app_ctx_prefix[] = "$app."; | |
954 | char *provider_name = NULL, *ctx_name = NULL; | |
955 | size_t i, len, colon_pos = 0, provider_name_len, ctx_name_len; | |
956 | ||
957 | if (!ctx) { | |
958 | goto not_found; | |
959 | } | |
960 | ||
961 | type = create_ctx_type(); | |
962 | if (!type) { | |
963 | goto not_found; | |
964 | } | |
965 | ||
966 | /* Check if ctx matches a known static context. */ | |
967 | opt_index = find_ctx_type_idx(ctx); | |
968 | if (opt_index >= 0) { | |
969 | *type->opt = ctx_opts[opt_index]; | |
970 | type->opt->symbol = strdup(ctx_opts[opt_index].symbol); | |
971 | goto found; | |
972 | } | |
973 | ||
4fe444da JD |
974 | /* Check if ctx is a raw perf context. */ |
975 | ret = find_ctx_type_perf_raw(ctx, type); | |
976 | if (ret == 0) { | |
977 | type->opt->u.perf.type = PERF_TYPE_RAW; | |
978 | type->opt->symbol = strdup(ctx); | |
979 | if (!type->opt->symbol) { | |
980 | PERROR("Copy perf field name"); | |
981 | goto not_found; | |
982 | } | |
983 | goto found; | |
984 | } | |
985 | ||
e9492550 JG |
986 | /* |
987 | * No match found against static contexts; check if it is an app | |
988 | * context. | |
989 | */ | |
990 | len = strlen(ctx); | |
991 | if (len <= sizeof(app_ctx_prefix) - 1) { | |
992 | goto not_found; | |
993 | } | |
994 | ||
995 | /* String starts with $app. */ | |
996 | if (strncmp(ctx, app_ctx_prefix, sizeof(app_ctx_prefix) - 1)) { | |
997 | goto not_found; | |
998 | } | |
999 | ||
1000 | /* Validate that the ':' separator is present. */ | |
1001 | for (i = sizeof(app_ctx_prefix); i < len; i++) { | |
1002 | const char c = ctx[i]; | |
1003 | ||
1004 | if (c == ':') { | |
1005 | colon_pos = i; | |
1006 | break; | |
1007 | } | |
1008 | } | |
1009 | ||
1010 | /* | |
1011 | * No colon found or no ctx name ("$app.provider:") or no provider name | |
1012 | * given ("$app.:..."), which is invalid. | |
1013 | */ | |
1014 | if (!colon_pos || colon_pos == len || | |
1015 | colon_pos == sizeof(app_ctx_prefix)) { | |
1016 | ERR("Invalid application context provided: no provider or context name provided."); | |
1017 | goto not_found; | |
1018 | } | |
1019 | ||
1020 | provider_name_len = colon_pos - sizeof(app_ctx_prefix) + 2; | |
1021 | provider_name = zmalloc(provider_name_len); | |
1022 | if (!provider_name) { | |
1023 | PERROR("malloc provider_name"); | |
1024 | goto not_found; | |
1025 | } | |
1026 | strncpy(provider_name, ctx + sizeof(app_ctx_prefix) - 1, | |
1027 | provider_name_len - 1); | |
1028 | type->opt->u.app_ctx.provider_name = provider_name; | |
1029 | ||
1030 | ctx_name_len = len - colon_pos; | |
1031 | ctx_name = zmalloc(ctx_name_len); | |
1032 | if (!ctx_name) { | |
1033 | PERROR("malloc ctx_name"); | |
1034 | goto not_found; | |
1035 | } | |
1036 | strncpy(ctx_name, ctx + colon_pos + 1, ctx_name_len - 1); | |
1037 | type->opt->u.app_ctx.ctx_name = ctx_name; | |
1038 | type->opt->ctx_type = CONTEXT_APP_CONTEXT; | |
1039 | type->opt->symbol = strdup(ctx); | |
1040 | found: | |
1041 | return type; | |
1042 | not_found: | |
1043 | free(provider_name); | |
1044 | free(ctx_name); | |
1045 | destroy_ctx_type(type); | |
1046 | return NULL; | |
1047 | } | |
1048 | ||
d65106b1 | 1049 | /* |
5eb00805 | 1050 | * Add context to channel or event. |
d65106b1 DG |
1051 | */ |
1052 | int cmd_add_context(int argc, const char **argv) | |
1053 | { | |
e9492550 | 1054 | int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS; |
d65106b1 | 1055 | static poptContext pc; |
b13d56d7 | 1056 | struct ctx_type *type, *tmptype; |
cd80958d | 1057 | char *session_name = NULL; |
68c7f6e5 | 1058 | const char *leftover = NULL; |
d65106b1 | 1059 | |
636167b6 | 1060 | if (argc < 2) { |
5eb00805 | 1061 | ret = CMD_ERROR; |
636167b6 DG |
1062 | goto end; |
1063 | } | |
1064 | ||
d65106b1 DG |
1065 | pc = poptGetContext(NULL, argc, argv, long_options, 0); |
1066 | poptReadDefaultConfig(pc, 0); | |
1067 | ||
1068 | while ((opt = poptGetNextOpt(pc)) != -1) { | |
1069 | switch (opt) { | |
1070 | case OPT_HELP: | |
4ba92f18 | 1071 | SHOW_HELP(); |
d65106b1 | 1072 | goto end; |
83cea427 | 1073 | case OPT_LIST: |
59deec0c | 1074 | ret = print_ctx_type(); |
83cea427 | 1075 | goto end; |
d65106b1 | 1076 | case OPT_TYPE: |
e9492550 JG |
1077 | { |
1078 | type = get_context_type(opt_type); | |
1079 | if (!type) { | |
6caa2bcc | 1080 | ERR("Unknown context type %s", opt_type); |
5eb00805 TD |
1081 | ret = CMD_FATAL; |
1082 | goto end; | |
1083 | } | |
e9492550 | 1084 | cds_list_add_tail(&type->list, &ctx_type_list.head); |
d65106b1 | 1085 | break; |
e9492550 | 1086 | } |
eeac7d46 MD |
1087 | case OPT_USERSPACE: |
1088 | opt_userspace = 1; | |
eeac7d46 | 1089 | break; |
e9492550 JG |
1090 | case OPT_JUL: |
1091 | opt_jul = 1; | |
1092 | break; | |
1093 | case OPT_LOG4J: | |
1094 | opt_log4j = 1; | |
1095 | break; | |
679b4943 SM |
1096 | case OPT_LIST_OPTIONS: |
1097 | list_cmd_options(stdout, long_options); | |
679b4943 | 1098 | goto end; |
d65106b1 | 1099 | default: |
d65106b1 DG |
1100 | ret = CMD_UNDEFINED; |
1101 | goto end; | |
1102 | } | |
1103 | } | |
1104 | ||
68c7f6e5 JD |
1105 | leftover = poptGetArg(pc); |
1106 | if (leftover) { | |
1107 | ERR("Unknown argument: %s", leftover); | |
1108 | ret = CMD_ERROR; | |
1109 | goto end; | |
1110 | } | |
1111 | ||
e9492550 JG |
1112 | ret = print_missing_or_multiple_domains(opt_kernel + opt_userspace + |
1113 | opt_jul + opt_log4j); | |
3ecec76a PP |
1114 | if (ret) { |
1115 | ret = CMD_ERROR; | |
1116 | goto end; | |
1117 | } | |
1118 | ||
ae856491 DG |
1119 | if (!opt_type) { |
1120 | ERR("Missing mandatory -t TYPE"); | |
ae856491 DG |
1121 | ret = CMD_ERROR; |
1122 | goto end; | |
1123 | } | |
1124 | ||
cd80958d DG |
1125 | if (!opt_session_name) { |
1126 | session_name = get_session_name(); | |
1127 | if (session_name == NULL) { | |
5eb00805 | 1128 | ret = CMD_ERROR; |
cd80958d DG |
1129 | goto end; |
1130 | } | |
1131 | } else { | |
1132 | session_name = opt_session_name; | |
1133 | } | |
1134 | ||
59deec0c JR |
1135 | ret = mi_open(); |
1136 | if (ret) { | |
1137 | goto end; | |
89b72577 JRJ |
1138 | } |
1139 | ||
1140 | command_ret = add_context(session_name); | |
59deec0c JR |
1141 | ret = mi_close(command_ret); |
1142 | if (ret) { | |
1143 | goto end; | |
89b72577 JRJ |
1144 | } |
1145 | ||
1146 | end: | |
1256f150 DT |
1147 | if (!opt_session_name) { |
1148 | free(session_name); | |
1149 | } | |
1150 | ||
89b72577 JRJ |
1151 | /* Mi clean-up */ |
1152 | if (writer && mi_lttng_writer_destroy(writer)) { | |
1153 | /* Preserve original error code */ | |
1154 | ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL; | |
1155 | } | |
1156 | ||
3301e740 | 1157 | /* Cleanup allocated memory */ |
b13d56d7 | 1158 | cds_list_for_each_entry_safe(type, tmptype, &ctx_type_list.head, list) { |
e9492550 | 1159 | destroy_ctx_type(type); |
3301e740 DG |
1160 | } |
1161 | ||
89b72577 JRJ |
1162 | /* Overwrite ret if an error occurred during add_context() */ |
1163 | ret = command_ret ? command_ret : ret; | |
1164 | ||
ca1c3607 | 1165 | poptFreeContext(pc); |
d65106b1 DG |
1166 | return ret; |
1167 | } |