Commit | Line | Data |
---|---|---|
54012638 DG |
1 | /* |
2 | * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca> | |
3 | * | |
d14d33bf AM |
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. | |
54012638 DG |
7 | * |
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. | |
12 | * | |
d14d33bf AM |
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. | |
54012638 DG |
16 | */ |
17 | ||
18 | #define _GNU_SOURCE | |
19 | #include <stdio.h> | |
20 | #include <stdlib.h> | |
21 | #include <string.h> | |
c363b55d | 22 | #include <unistd.h> |
54012638 | 23 | |
990570ed DG |
24 | #include <common/common.h> |
25 | #include <common/defaults.h> | |
1e307fab | 26 | |
00e2e675 | 27 | #include "consumer.h" |
62499ad6 | 28 | #include "trace-kernel.h" |
54012638 | 29 | |
19e70852 | 30 | /* |
050349bb | 31 | * Find the channel name for the given kernel session. |
19e70852 | 32 | */ |
62499ad6 | 33 | struct ltt_kernel_channel *trace_kernel_get_channel_by_name( |
19e70852 DG |
34 | char *name, struct ltt_kernel_session *session) |
35 | { | |
36 | struct ltt_kernel_channel *chan; | |
37 | ||
38 | if (session == NULL) { | |
39 | ERR("Undefine session"); | |
40 | goto error; | |
41 | } | |
42 | ||
54d01ffb DG |
43 | DBG("Trying to find channel %s", name); |
44 | ||
19e70852 DG |
45 | cds_list_for_each_entry(chan, &session->channel_list.head, list) { |
46 | if (strcmp(name, chan->channel->name) == 0) { | |
47 | DBG("Found channel by name %s", name); | |
48 | return chan; | |
49 | } | |
50 | } | |
51 | ||
52 | error: | |
53 | return NULL; | |
54 | } | |
55 | ||
56 | /* | |
050349bb | 57 | * Find the event name for the given channel. |
19e70852 | 58 | */ |
62499ad6 | 59 | struct ltt_kernel_event *trace_kernel_get_event_by_name( |
19e70852 DG |
60 | char *name, struct ltt_kernel_channel *channel) |
61 | { | |
62 | struct ltt_kernel_event *ev; | |
63 | ||
64 | if (channel == NULL) { | |
65 | ERR("Undefine channel"); | |
66 | goto error; | |
67 | } | |
68 | ||
69 | cds_list_for_each_entry(ev, &channel->events_list.head, list) { | |
70 | if (strcmp(name, ev->event->name) == 0) { | |
71 | DBG("Found event by name %s for channel %s", name, | |
72 | channel->channel->name); | |
73 | return ev; | |
74 | } | |
75 | } | |
76 | ||
77 | error: | |
78 | return NULL; | |
79 | } | |
80 | ||
54012638 | 81 | /* |
050349bb | 82 | * Allocate and initialize a kernel session data structure. |
54012638 | 83 | * |
050349bb | 84 | * Return pointer to structure or NULL. |
54012638 | 85 | */ |
f9815039 | 86 | struct ltt_kernel_session *trace_kernel_create_session(char *path) |
54012638 | 87 | { |
f9815039 | 88 | int ret; |
a4b92340 | 89 | struct ltt_kernel_session *lks = NULL; |
54012638 DG |
90 | |
91 | /* Allocate a new ltt kernel session */ | |
ba7f0ae5 | 92 | lks = zmalloc(sizeof(struct ltt_kernel_session)); |
54012638 | 93 | if (lks == NULL) { |
df0f840b | 94 | PERROR("create kernel session zmalloc"); |
a4b92340 | 95 | goto alloc_error; |
54012638 DG |
96 | } |
97 | ||
98 | /* Init data structure */ | |
03550b58 MD |
99 | lks->fd = -1; |
100 | lks->metadata_stream_fd = -1; | |
54012638 DG |
101 | lks->channel_count = 0; |
102 | lks->stream_count_global = 0; | |
103 | lks->metadata = NULL; | |
104 | CDS_INIT_LIST_HEAD(&lks->channel_list.head); | |
105 | ||
00e2e675 DG |
106 | lks->consumer = consumer_create_output(CONSUMER_DST_LOCAL); |
107 | if (lks->consumer == NULL) { | |
108 | goto error; | |
109 | } | |
110 | ||
111 | /* | |
112 | * The tmp_consumer stays NULL until a set_consumer_uri command is | |
113 | * executed. At this point, the consumer should be nullify until an | |
114 | * enable_consumer command. This assignment is symbolic since we've zmalloc | |
115 | * the struct. | |
116 | */ | |
117 | lks->tmp_consumer = NULL; | |
118 | ||
a4b92340 DG |
119 | if (path && strlen(path) > 0) { |
120 | /* Use the default consumer output which is the tracing session path. */ | |
121 | ret = snprintf(lks->consumer->dst.trace_path, PATH_MAX, | |
122 | "%s" DEFAULT_KERNEL_TRACE_DIR, path); | |
123 | if (ret < 0) { | |
124 | PERROR("snprintf consumer trace path"); | |
125 | goto error; | |
126 | } | |
00e2e675 | 127 | |
a4b92340 DG |
128 | /* Set session path */ |
129 | ret = asprintf(&lks->trace_path, "%s" DEFAULT_KERNEL_TRACE_DIR, path); | |
130 | if (ret < 0) { | |
131 | PERROR("asprintf kernel traces path"); | |
132 | goto error; | |
133 | } | |
f9815039 DG |
134 | } |
135 | ||
54012638 DG |
136 | return lks; |
137 | ||
138 | error: | |
a4b92340 DG |
139 | free(lks); |
140 | ||
141 | alloc_error: | |
54012638 DG |
142 | return NULL; |
143 | } | |
144 | ||
145 | /* | |
050349bb | 146 | * Allocate and initialize a kernel channel data structure. |
54012638 | 147 | * |
050349bb | 148 | * Return pointer to structure or NULL. |
54012638 | 149 | */ |
00e2e675 DG |
150 | struct ltt_kernel_channel *trace_kernel_create_channel( |
151 | struct lttng_channel *chan, char *path) | |
54012638 | 152 | { |
54012638 | 153 | struct ltt_kernel_channel *lkc; |
54012638 | 154 | |
ba7f0ae5 | 155 | lkc = zmalloc(sizeof(struct ltt_kernel_channel)); |
f3ed775e | 156 | if (lkc == NULL) { |
df0f840b | 157 | PERROR("ltt_kernel_channel zmalloc"); |
54012638 DG |
158 | goto error; |
159 | } | |
160 | ||
ba7f0ae5 | 161 | lkc->channel = zmalloc(sizeof(struct lttng_channel)); |
f3ed775e | 162 | if (lkc->channel == NULL) { |
df0f840b | 163 | PERROR("lttng_channel zmalloc"); |
f3ed775e DG |
164 | goto error; |
165 | } | |
166 | memcpy(lkc->channel, chan, sizeof(struct lttng_channel)); | |
54012638 | 167 | |
03550b58 | 168 | lkc->fd = -1; |
54012638 | 169 | lkc->stream_count = 0; |
cbbbb275 | 170 | lkc->event_count = 0; |
d36b8583 | 171 | lkc->enabled = 1; |
e75cda3a | 172 | lkc->ctx = NULL; |
54012638 DG |
173 | /* Init linked list */ |
174 | CDS_INIT_LIST_HEAD(&lkc->events_list.head); | |
175 | CDS_INIT_LIST_HEAD(&lkc->stream_list.head); | |
54012638 DG |
176 | |
177 | return lkc; | |
178 | ||
179 | error: | |
180 | return NULL; | |
181 | } | |
182 | ||
183 | /* | |
050349bb | 184 | * Allocate and initialize a kernel event. Set name and event type. |
54012638 | 185 | * |
050349bb | 186 | * Return pointer to structure or NULL. |
54012638 | 187 | */ |
62499ad6 | 188 | struct ltt_kernel_event *trace_kernel_create_event(struct lttng_event *ev) |
54012638 DG |
189 | { |
190 | struct ltt_kernel_event *lke; | |
191 | struct lttng_kernel_event *attr; | |
192 | ||
ba7f0ae5 DG |
193 | lke = zmalloc(sizeof(struct ltt_kernel_event)); |
194 | attr = zmalloc(sizeof(struct lttng_kernel_event)); | |
54012638 | 195 | if (lke == NULL || attr == NULL) { |
df0f840b | 196 | PERROR("kernel event zmalloc"); |
54012638 DG |
197 | goto error; |
198 | } | |
199 | ||
f3ed775e | 200 | switch (ev->type) { |
7d29a247 | 201 | case LTTNG_EVENT_PROBE: |
e6ddca71 | 202 | attr->instrumentation = LTTNG_KERNEL_KPROBE; |
7d29a247 DG |
203 | attr->u.kprobe.addr = ev->attr.probe.addr; |
204 | attr->u.kprobe.offset = ev->attr.probe.offset; | |
f3ed775e | 205 | strncpy(attr->u.kprobe.symbol_name, |
dbbb3ec5 DG |
206 | ev->attr.probe.symbol_name, LTTNG_KERNEL_SYM_NAME_LEN); |
207 | attr->u.kprobe.symbol_name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0'; | |
f3ed775e DG |
208 | break; |
209 | case LTTNG_EVENT_FUNCTION: | |
8f0d098b MD |
210 | attr->instrumentation = LTTNG_KERNEL_KRETPROBE; |
211 | attr->u.kretprobe.addr = ev->attr.probe.addr; | |
212 | attr->u.kretprobe.offset = ev->attr.probe.offset; | |
213 | attr->u.kretprobe.offset = ev->attr.probe.offset; | |
214 | strncpy(attr->u.kretprobe.symbol_name, | |
dbbb3ec5 DG |
215 | ev->attr.probe.symbol_name, LTTNG_KERNEL_SYM_NAME_LEN); |
216 | attr->u.kretprobe.symbol_name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0'; | |
8f0d098b MD |
217 | break; |
218 | case LTTNG_EVENT_FUNCTION_ENTRY: | |
f3ed775e DG |
219 | attr->instrumentation = LTTNG_KERNEL_FUNCTION; |
220 | strncpy(attr->u.ftrace.symbol_name, | |
dbbb3ec5 DG |
221 | ev->attr.ftrace.symbol_name, LTTNG_KERNEL_SYM_NAME_LEN); |
222 | attr->u.ftrace.symbol_name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0'; | |
f3ed775e | 223 | break; |
e6ddca71 DG |
224 | case LTTNG_EVENT_TRACEPOINT: |
225 | attr->instrumentation = LTTNG_KERNEL_TRACEPOINT; | |
f3ed775e | 226 | break; |
a54bd42d MD |
227 | case LTTNG_EVENT_SYSCALL: |
228 | attr->instrumentation = LTTNG_KERNEL_SYSCALL; | |
0133c199 | 229 | break; |
7a3d1328 MD |
230 | case LTTNG_EVENT_ALL: |
231 | attr->instrumentation = LTTNG_KERNEL_ALL; | |
232 | break; | |
f3ed775e DG |
233 | default: |
234 | ERR("Unknown kernel instrumentation type (%d)", ev->type); | |
235 | goto error; | |
236 | } | |
237 | ||
238 | /* Copy event name */ | |
dbbb3ec5 DG |
239 | strncpy(attr->name, ev->name, LTTNG_KERNEL_SYM_NAME_LEN); |
240 | attr->name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0'; | |
f3ed775e | 241 | |
54012638 | 242 | /* Setting up a kernel event */ |
03550b58 | 243 | lke->fd = -1; |
54012638 | 244 | lke->event = attr; |
d36b8583 | 245 | lke->enabled = 1; |
e75cda3a | 246 | lke->ctx = NULL; |
54012638 DG |
247 | |
248 | return lke; | |
249 | ||
250 | error: | |
a2c0da86 MD |
251 | free(lke); |
252 | free(attr); | |
54012638 DG |
253 | return NULL; |
254 | } | |
255 | ||
256 | /* | |
050349bb | 257 | * Allocate and initialize a kernel metadata. |
54012638 | 258 | * |
050349bb | 259 | * Return pointer to structure or NULL. |
54012638 | 260 | */ |
a4b92340 | 261 | struct ltt_kernel_metadata *trace_kernel_create_metadata(void) |
54012638 | 262 | { |
54012638 | 263 | struct ltt_kernel_metadata *lkm; |
f3ed775e | 264 | struct lttng_channel *chan; |
54012638 | 265 | |
ba7f0ae5 DG |
266 | lkm = zmalloc(sizeof(struct ltt_kernel_metadata)); |
267 | chan = zmalloc(sizeof(struct lttng_channel)); | |
f3ed775e | 268 | if (lkm == NULL || chan == NULL) { |
df0f840b | 269 | PERROR("kernel metadata zmalloc"); |
54012638 DG |
270 | goto error; |
271 | } | |
272 | ||
273 | /* Set default attributes */ | |
f3ed775e | 274 | chan->attr.overwrite = DEFAULT_CHANNEL_OVERWRITE; |
b389abbe MD |
275 | chan->attr.subbuf_size = DEFAULT_METADATA_SUBBUF_SIZE; |
276 | chan->attr.num_subbuf = DEFAULT_METADATA_SUBBUF_NUM; | |
f3ed775e DG |
277 | chan->attr.switch_timer_interval = DEFAULT_CHANNEL_SWITCH_TIMER; |
278 | chan->attr.read_timer_interval = DEFAULT_CHANNEL_READ_TIMER; | |
7d452e12 | 279 | chan->attr.output = DEFAULT_KERNEL_CHANNEL_OUTPUT; |
54012638 DG |
280 | |
281 | /* Init metadata */ | |
03550b58 | 282 | lkm->fd = -1; |
f3ed775e | 283 | lkm->conf = chan; |
54012638 DG |
284 | |
285 | return lkm; | |
286 | ||
287 | error: | |
a2c0da86 MD |
288 | free(lkm); |
289 | free(chan); | |
54012638 DG |
290 | return NULL; |
291 | } | |
292 | ||
293 | /* | |
050349bb DG |
294 | * Allocate and initialize a kernel stream. The stream is set to ACTIVE_FD by |
295 | * default. | |
54012638 | 296 | * |
050349bb | 297 | * Return pointer to structure or NULL. |
54012638 | 298 | */ |
00e2e675 DG |
299 | struct ltt_kernel_stream *trace_kernel_create_stream(const char *name, |
300 | unsigned int count) | |
54012638 | 301 | { |
00e2e675 | 302 | int ret; |
54012638 DG |
303 | struct ltt_kernel_stream *lks; |
304 | ||
ba7f0ae5 | 305 | lks = zmalloc(sizeof(struct ltt_kernel_stream)); |
54012638 | 306 | if (lks == NULL) { |
df0f840b | 307 | PERROR("kernel stream zmalloc"); |
54012638 DG |
308 | goto error; |
309 | } | |
310 | ||
00e2e675 DG |
311 | /* Set name */ |
312 | ret = snprintf(lks->name, sizeof(lks->name), "%s_%d", name, count); | |
313 | if (ret < 0) { | |
314 | PERROR("snprintf stream name"); | |
315 | goto error; | |
316 | } | |
317 | lks->name[sizeof(lks->name) - 1] = '\0'; | |
318 | ||
54012638 | 319 | /* Init stream */ |
03550b58 | 320 | lks->fd = -1; |
54012638 DG |
321 | lks->state = 0; |
322 | ||
323 | return lks; | |
324 | ||
325 | error: | |
326 | return NULL; | |
327 | } | |
c363b55d | 328 | |
050349bb DG |
329 | /* |
330 | * Cleanup kernel stream structure. | |
331 | */ | |
62499ad6 | 332 | void trace_kernel_destroy_stream(struct ltt_kernel_stream *stream) |
c363b55d | 333 | { |
799e2c4f MD |
334 | int ret; |
335 | ||
33a2b854 | 336 | DBG("[trace] Closing stream fd %d", stream->fd); |
c363b55d | 337 | /* Close kernel fd */ |
03550b58 MD |
338 | if (stream->fd >= 0) { |
339 | ret = close(stream->fd); | |
340 | if (ret) { | |
341 | PERROR("close"); | |
342 | } | |
799e2c4f | 343 | } |
c363b55d DG |
344 | /* Remove from stream list */ |
345 | cds_list_del(&stream->list); | |
f9815039 | 346 | |
c363b55d DG |
347 | free(stream); |
348 | } | |
349 | ||
050349bb DG |
350 | /* |
351 | * Cleanup kernel event structure. | |
352 | */ | |
62499ad6 | 353 | void trace_kernel_destroy_event(struct ltt_kernel_event *event) |
c363b55d | 354 | { |
799e2c4f MD |
355 | int ret; |
356 | ||
87eb4ab8 MD |
357 | if (event->fd >= 0) { |
358 | DBG("[trace] Closing event fd %d", event->fd); | |
359 | /* Close kernel fd */ | |
799e2c4f MD |
360 | ret = close(event->fd); |
361 | if (ret) { | |
362 | PERROR("close"); | |
363 | } | |
87eb4ab8 MD |
364 | } else { |
365 | DBG("[trace] Tearing down event (no associated fd)"); | |
366 | } | |
c363b55d DG |
367 | |
368 | /* Remove from event list */ | |
369 | cds_list_del(&event->list); | |
f9815039 DG |
370 | |
371 | free(event->event); | |
372 | free(event->ctx); | |
c363b55d DG |
373 | free(event); |
374 | } | |
375 | ||
050349bb DG |
376 | /* |
377 | * Cleanup kernel channel structure. | |
378 | */ | |
62499ad6 | 379 | void trace_kernel_destroy_channel(struct ltt_kernel_channel *channel) |
c363b55d | 380 | { |
af9737e9 DG |
381 | struct ltt_kernel_stream *stream, *stmp; |
382 | struct ltt_kernel_event *event, *etmp; | |
799e2c4f | 383 | int ret; |
c363b55d | 384 | |
33a2b854 | 385 | DBG("[trace] Closing channel fd %d", channel->fd); |
c363b55d | 386 | /* Close kernel fd */ |
03550b58 MD |
387 | if (channel->fd >= 0) { |
388 | ret = close(channel->fd); | |
389 | if (ret) { | |
390 | PERROR("close"); | |
391 | } | |
799e2c4f | 392 | } |
c363b55d DG |
393 | |
394 | /* For each stream in the channel list */ | |
af9737e9 | 395 | cds_list_for_each_entry_safe(stream, stmp, &channel->stream_list.head, list) { |
62499ad6 | 396 | trace_kernel_destroy_stream(stream); |
c363b55d DG |
397 | } |
398 | ||
399 | /* For each event in the channel list */ | |
af9737e9 | 400 | cds_list_for_each_entry_safe(event, etmp, &channel->events_list.head, list) { |
62499ad6 | 401 | trace_kernel_destroy_event(event); |
c363b55d DG |
402 | } |
403 | ||
404 | /* Remove from channel list */ | |
405 | cds_list_del(&channel->list); | |
f9815039 | 406 | |
f9815039 DG |
407 | free(channel->channel); |
408 | free(channel->ctx); | |
c363b55d DG |
409 | free(channel); |
410 | } | |
411 | ||
050349bb DG |
412 | /* |
413 | * Cleanup kernel metadata structure. | |
414 | */ | |
62499ad6 | 415 | void trace_kernel_destroy_metadata(struct ltt_kernel_metadata *metadata) |
c363b55d | 416 | { |
799e2c4f MD |
417 | int ret; |
418 | ||
33a2b854 | 419 | DBG("[trace] Closing metadata fd %d", metadata->fd); |
c363b55d | 420 | /* Close kernel fd */ |
03550b58 MD |
421 | if (metadata->fd >= 0) { |
422 | ret = close(metadata->fd); | |
423 | if (ret) { | |
424 | PERROR("close"); | |
425 | } | |
799e2c4f | 426 | } |
c363b55d | 427 | |
f9815039 | 428 | free(metadata->conf); |
c363b55d DG |
429 | free(metadata); |
430 | } | |
431 | ||
050349bb | 432 | /* |
62499ad6 | 433 | * Cleanup kernel session structure |
050349bb | 434 | */ |
62499ad6 | 435 | void trace_kernel_destroy_session(struct ltt_kernel_session *session) |
c363b55d | 436 | { |
af9737e9 | 437 | struct ltt_kernel_channel *channel, *ctmp; |
799e2c4f | 438 | int ret; |
c363b55d | 439 | |
33a2b854 | 440 | DBG("[trace] Closing session fd %d", session->fd); |
c363b55d | 441 | /* Close kernel fds */ |
03550b58 MD |
442 | if (session->fd >= 0) { |
443 | ret = close(session->fd); | |
444 | if (ret) { | |
445 | PERROR("close"); | |
446 | } | |
799e2c4f | 447 | } |
f9815039 | 448 | |
03550b58 | 449 | if (session->metadata_stream_fd >= 0) { |
70dc1c34 | 450 | DBG("[trace] Closing metadata stream fd %d", session->metadata_stream_fd); |
799e2c4f MD |
451 | ret = close(session->metadata_stream_fd); |
452 | if (ret) { | |
453 | PERROR("close"); | |
454 | } | |
70dc1c34 | 455 | } |
c363b55d | 456 | |
d36b8583 | 457 | if (session->metadata != NULL) { |
62499ad6 | 458 | trace_kernel_destroy_metadata(session->metadata); |
d36b8583 | 459 | } |
c363b55d | 460 | |
af9737e9 | 461 | cds_list_for_each_entry_safe(channel, ctmp, &session->channel_list.head, list) { |
62499ad6 | 462 | trace_kernel_destroy_channel(channel); |
c363b55d DG |
463 | } |
464 | ||
00e2e675 DG |
465 | /* Wipe consumer output object */ |
466 | consumer_destroy_output(session->consumer); | |
467 | consumer_destroy_output(session->tmp_consumer); | |
468 | ||
f9815039 | 469 | free(session->trace_path); |
c363b55d DG |
470 | free(session); |
471 | } |