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