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