Add kretprobe support (new --function implementation)
[lttng-tools.git] / ltt-sessiond / trace.c
CommitLineData
54012638
DG
1/*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
82a3637f
DG
6 * as published by the Free Software Foundation; only version 2
7 * of the License.
54012638
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 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17 */
18
19#define _GNU_SOURCE
20#include <stdio.h>
21#include <stdlib.h>
22#include <string.h>
c363b55d 23#include <unistd.h>
54012638
DG
24#include <urcu/list.h>
25
33a2b854 26#include "lttngerr.h"
54012638
DG
27#include "trace.h"
28
19e70852
DG
29/*
30 * get_kernel_channel_by_name
31 *
32 * Find the channel name for the given kernel session.
33 */
34struct ltt_kernel_channel *get_kernel_channel_by_name(
35 char *name, struct ltt_kernel_session *session)
36{
37 struct ltt_kernel_channel *chan;
38
39 if (session == NULL) {
40 ERR("Undefine session");
41 goto error;
42 }
43
44 cds_list_for_each_entry(chan, &session->channel_list.head, list) {
45 if (strcmp(name, chan->channel->name) == 0) {
46 DBG("Found channel by name %s", name);
47 return chan;
48 }
49 }
50
51error:
52 return NULL;
53}
54
55/*
56 * get_kernel_event_by_name
57 *
58 * Find the event name for the given channel.
59 */
60struct ltt_kernel_event *get_kernel_event_by_name(
61 char *name, struct ltt_kernel_channel *channel)
62{
63 struct ltt_kernel_event *ev;
64
65 if (channel == NULL) {
66 ERR("Undefine channel");
67 goto error;
68 }
69
70 cds_list_for_each_entry(ev, &channel->events_list.head, list) {
71 if (strcmp(name, ev->event->name) == 0) {
72 DBG("Found event by name %s for channel %s", name,
73 channel->channel->name);
74 return ev;
75 }
76 }
77
78error:
79 return NULL;
80}
81
54012638
DG
82/*
83 * trace_create_kernel_session
84 *
85 * Allocate and initialize a kernel session data structure.
86 *
87 * Return pointer to structure or NULL.
88 */
89struct ltt_kernel_session *trace_create_kernel_session(void)
90{
91 struct ltt_kernel_session *lks;
92
93 /* Allocate a new ltt kernel session */
94 lks = malloc(sizeof(struct ltt_kernel_session));
95 if (lks == NULL) {
96 perror("create kernel session malloc");
97 goto error;
98 }
99
100 /* Init data structure */
101 lks->fd = 0;
102 lks->metadata_stream_fd = 0;
103 lks->channel_count = 0;
104 lks->stream_count_global = 0;
105 lks->metadata = NULL;
106 CDS_INIT_LIST_HEAD(&lks->channel_list.head);
107
108 return lks;
109
110error:
111 return NULL;
112}
113
114/*
115 * trace_create_kernel_channel
116 *
117 * Allocate and initialize a kernel channel data structure.
118 *
119 * Return pointer to structure or NULL.
120 */
b082db07 121struct ltt_kernel_channel *trace_create_kernel_channel(struct lttng_channel *chan, char *path)
54012638
DG
122{
123 int ret;
124 struct ltt_kernel_channel *lkc;
54012638
DG
125
126 lkc = malloc(sizeof(struct ltt_kernel_channel));
f3ed775e
DG
127 if (lkc == NULL) {
128 perror("ltt_kernel_channel malloc");
54012638
DG
129 goto error;
130 }
131
f3ed775e
DG
132 lkc->channel = malloc(sizeof(struct lttng_channel));
133 if (lkc->channel == NULL) {
134 perror("lttng_channel malloc");
135 goto error;
136 }
137 memcpy(lkc->channel, chan, sizeof(struct lttng_channel));
54012638
DG
138
139 lkc->fd = 0;
140 lkc->stream_count = 0;
cbbbb275 141 lkc->event_count = 0;
d36b8583 142 lkc->enabled = 1;
e75cda3a 143 lkc->ctx = NULL;
54012638
DG
144 /* Init linked list */
145 CDS_INIT_LIST_HEAD(&lkc->events_list.head);
146 CDS_INIT_LIST_HEAD(&lkc->stream_list.head);
147 /* Set default trace output path */
b082db07 148 ret = asprintf(&lkc->pathname, "%s", path);
54012638
DG
149 if (ret < 0) {
150 perror("asprintf kernel create channel");
151 goto error;
152 }
153
154 return lkc;
155
156error:
157 return NULL;
158}
159
160/*
161 * trace_create_kernel_event
162 *
163 * Allocate and initialize a kernel event. Set name and event type.
164 *
165 * Return pointer to structure or NULL.
166 */
f3ed775e 167struct ltt_kernel_event *trace_create_kernel_event(struct lttng_event *ev)
54012638
DG
168{
169 struct ltt_kernel_event *lke;
170 struct lttng_kernel_event *attr;
171
172 lke = malloc(sizeof(struct ltt_kernel_event));
173 attr = malloc(sizeof(struct lttng_kernel_event));
174 if (lke == NULL || attr == NULL) {
175 perror("kernel event malloc");
176 goto error;
177 }
178
f3ed775e 179 switch (ev->type) {
7d29a247 180 case LTTNG_EVENT_PROBE:
e6ddca71 181 attr->instrumentation = LTTNG_KERNEL_KPROBE;
7d29a247
DG
182 attr->u.kprobe.addr = ev->attr.probe.addr;
183 attr->u.kprobe.offset = ev->attr.probe.offset;
f3ed775e 184 strncpy(attr->u.kprobe.symbol_name,
7d29a247 185 ev->attr.probe.symbol_name, LTTNG_SYM_NAME_LEN);
f3ed775e
DG
186 break;
187 case LTTNG_EVENT_FUNCTION:
8f0d098b
MD
188 attr->instrumentation = LTTNG_KERNEL_KRETPROBE;
189 attr->u.kretprobe.addr = ev->attr.probe.addr;
190 attr->u.kretprobe.offset = ev->attr.probe.offset;
191 attr->u.kretprobe.offset = ev->attr.probe.offset;
192 strncpy(attr->u.kretprobe.symbol_name,
193 ev->attr.probe.symbol_name, LTTNG_SYM_NAME_LEN);
194 break;
195 case LTTNG_EVENT_FUNCTION_ENTRY:
f3ed775e
DG
196 attr->instrumentation = LTTNG_KERNEL_FUNCTION;
197 strncpy(attr->u.ftrace.symbol_name,
198 ev->attr.ftrace.symbol_name, LTTNG_SYM_NAME_LEN);
199 break;
e6ddca71
DG
200 case LTTNG_EVENT_TRACEPOINT:
201 attr->instrumentation = LTTNG_KERNEL_TRACEPOINT;
f3ed775e
DG
202 break;
203 default:
204 ERR("Unknown kernel instrumentation type (%d)", ev->type);
205 goto error;
206 }
207
208 /* Copy event name */
209 strncpy(attr->name, ev->name, LTTNG_SYM_NAME_LEN);
210
54012638
DG
211 /* Setting up a kernel event */
212 lke->fd = 0;
213 lke->event = attr;
d36b8583 214 lke->enabled = 1;
e75cda3a 215 lke->ctx = NULL;
54012638
DG
216
217 return lke;
218
219error:
220 return NULL;
221}
222
223/*
224 * trace_create_kernel_metadata
225 *
226 * Allocate and initialize a kernel metadata.
227 *
228 * Return pointer to structure or NULL.
229 */
58a97671 230struct ltt_kernel_metadata *trace_create_kernel_metadata(char *path)
54012638
DG
231{
232 int ret;
233 struct ltt_kernel_metadata *lkm;
f3ed775e 234 struct lttng_channel *chan;
54012638
DG
235
236 lkm = malloc(sizeof(struct ltt_kernel_metadata));
f3ed775e
DG
237 chan = malloc(sizeof(struct lttng_channel));
238 if (lkm == NULL || chan == NULL) {
54012638
DG
239 perror("kernel metadata malloc");
240 goto error;
241 }
242
243 /* Set default attributes */
f3ed775e
DG
244 chan->attr.overwrite = DEFAULT_CHANNEL_OVERWRITE;
245 chan->attr.subbuf_size = DEFAULT_CHANNEL_SUBBUF_SIZE;
246 chan->attr.num_subbuf = DEFAULT_CHANNEL_SUBBUF_NUM;
247 chan->attr.switch_timer_interval = DEFAULT_CHANNEL_SWITCH_TIMER;
248 chan->attr.read_timer_interval = DEFAULT_CHANNEL_READ_TIMER;
7d452e12 249 chan->attr.output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
54012638
DG
250
251 /* Init metadata */
252 lkm->fd = 0;
f3ed775e 253 lkm->conf = chan;
54012638 254 /* Set default metadata path */
58a97671 255 ret = asprintf(&lkm->pathname, "%s/metadata", path);
54012638
DG
256 if (ret < 0) {
257 perror("asprintf kernel metadata");
258 goto error;
259 }
260
261 return lkm;
262
263error:
264 return NULL;
265}
266
267/*
268 * trace_create_kernel_stream
269 *
270 * Allocate and initialize a kernel stream. The stream is set to ACTIVE_FD by
271 * default.
272 *
273 * Return pointer to structure or NULL.
274 */
275struct ltt_kernel_stream *trace_create_kernel_stream(void)
276{
277 struct ltt_kernel_stream *lks;
278
279 lks = malloc(sizeof(struct ltt_kernel_stream));
280 if (lks == NULL) {
281 perror("kernel stream malloc");
282 goto error;
283 }
284
285 /* Init stream */
286 lks->fd = 0;
287 lks->pathname = NULL;
288 lks->state = 0;
289
290 return lks;
291
292error:
293 return NULL;
294}
c363b55d
DG
295
296void trace_destroy_kernel_stream(struct ltt_kernel_stream *stream)
297{
33a2b854 298 DBG("[trace] Closing stream fd %d", stream->fd);
c363b55d
DG
299 /* Close kernel fd */
300 close(stream->fd);
301 free(stream->pathname);
302
303 /* Remove from stream list */
304 cds_list_del(&stream->list);
305 free(stream);
306}
307
308void trace_destroy_kernel_event(struct ltt_kernel_event *event)
309{
33a2b854 310 DBG("[trace] Closing event fd %d", event->fd);
c363b55d
DG
311 /* Close kernel fd */
312 close(event->fd);
313 /* Free attributes */
314 free(event->event);
315
316 /* Remove from event list */
317 cds_list_del(&event->list);
318 free(event);
319}
320
321void trace_destroy_kernel_channel(struct ltt_kernel_channel *channel)
322{
af9737e9
DG
323 struct ltt_kernel_stream *stream, *stmp;
324 struct ltt_kernel_event *event, *etmp;
c363b55d 325
33a2b854 326 DBG("[trace] Closing channel fd %d", channel->fd);
c363b55d
DG
327 /* Close kernel fd */
328 close(channel->fd);
329 free(channel->pathname);
330 /* Free attributes structure */
331 free(channel->channel);
332
333 /* For each stream in the channel list */
af9737e9 334 cds_list_for_each_entry_safe(stream, stmp, &channel->stream_list.head, list) {
c363b55d
DG
335 trace_destroy_kernel_stream(stream);
336 }
337
338 /* For each event in the channel list */
af9737e9 339 cds_list_for_each_entry_safe(event, etmp, &channel->events_list.head, list) {
c363b55d
DG
340 trace_destroy_kernel_event(event);
341 }
342
343 /* Remove from channel list */
344 cds_list_del(&channel->list);
345 free(channel);
346}
347
348void trace_destroy_kernel_metadata(struct ltt_kernel_metadata *metadata)
349{
33a2b854 350 DBG("[trace] Closing metadata fd %d", metadata->fd);
c363b55d
DG
351 /* Close kernel fd */
352 close(metadata->fd);
353 /* Free attributes */
354 free(metadata->conf);
355
356 free(metadata);
357}
358
359void trace_destroy_kernel_session(struct ltt_kernel_session *session)
360{
af9737e9 361 struct ltt_kernel_channel *channel, *ctmp;
c363b55d 362
33a2b854 363 DBG("[trace] Closing session fd %d", session->fd);
c363b55d
DG
364 /* Close kernel fds */
365 close(session->fd);
70dc1c34
DG
366 if (session->metadata_stream_fd != 0) {
367 DBG("[trace] Closing metadata stream fd %d", session->metadata_stream_fd);
368 close(session->metadata_stream_fd);
369 }
c363b55d 370
d36b8583
DG
371 if (session->metadata != NULL) {
372 trace_destroy_kernel_metadata(session->metadata);
373 }
c363b55d 374
af9737e9 375 cds_list_for_each_entry_safe(channel, ctmp, &session->channel_list.head, list) {
c363b55d
DG
376 trace_destroy_kernel_channel(channel);
377 }
378
379 free(session);
380}
This page took 0.037942 seconds and 4 git commands to generate.