2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License as published by the Free
6 * Software Foundation; only version 2 of the License.
8 * This program is distributed in the hope that it will be useful, but WITHOUT
9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
13 * You should have received a copy of the GNU General Public License along with
14 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
15 * Place - Suite 330, Boston, MA 02111-1307, USA.
23 #include <urcu/list.h>
25 #include <lttng-sessiond-comm.h>
29 #include "hashtable.h"
34 * Add kernel context to an event of a specific channel.
36 static int add_kctx_to_event(struct lttng_kernel_context
*kctx
,
37 struct ltt_kernel_channel
*kchan
, char *event_name
)
40 struct ltt_kernel_event
*kevent
;
42 DBG("Add kernel context to event %s", event_name
);
44 kevent
= trace_kernel_get_event_by_name(event_name
, kchan
);
46 ret
= kernel_add_event_context(kevent
, kctx
);
60 * Add kernel context to all channel.
62 * If event_name is specified, add context to event instead.
64 static int add_kctx_all_channels(struct ltt_kernel_session
*ksession
,
65 struct lttng_kernel_context
*kctx
, char *event_name
)
67 int ret
, no_event
= 0, found
= 0;
68 struct ltt_kernel_channel
*kchan
;
70 if (strlen(event_name
) == 0) {
74 DBG("Adding kernel context to all channels (event: %s)", event_name
);
76 /* Go over all channels */
77 cds_list_for_each_entry(kchan
, &ksession
->channel_list
.head
, list
) {
79 ret
= kernel_add_channel_context(kchan
, kctx
);
81 ret
= LTTCOMM_KERN_CONTEXT_FAIL
;
85 ret
= add_kctx_to_event(kctx
, kchan
, event_name
);
87 ret
= LTTCOMM_KERN_CONTEXT_FAIL
;
89 } else if (ret
== 1) {
90 /* Event found and context added */
97 if (!found
&& !no_event
) {
98 ret
= LTTCOMM_NO_EVENT
;
109 * Add kernel context to a specific channel.
111 * If event_name is specified, add context to that event.
113 static int add_kctx_to_channel(struct lttng_kernel_context
*kctx
,
114 struct ltt_kernel_channel
*kchan
, char *event_name
)
116 int ret
, no_event
= 0, found
= 0;
118 if (strlen(event_name
) == 0) {
122 DBG("Add kernel context to channel '%s', event '%s'",
123 kchan
->channel
->name
, event_name
);
126 ret
= kernel_add_channel_context(kchan
, kctx
);
128 ret
= LTTCOMM_KERN_CONTEXT_FAIL
;
132 ret
= add_kctx_to_event(kctx
, kchan
, event_name
);
134 ret
= LTTCOMM_KERN_CONTEXT_FAIL
;
136 } else if (ret
== 1) {
137 /* Event found and context added */
142 if (!found
&& !no_event
) {
143 ret
= LTTCOMM_NO_EVENT
;
154 * Add kernel context to tracer.
156 int context_kernel_add(struct ltt_kernel_session
*ksession
,
157 struct lttng_event_context
*ctx
, char *event_name
,
161 struct ltt_kernel_channel
*kchan
;
162 struct lttng_kernel_context kctx
;
164 /* Setup kernel context structure */
166 kctx
.u
.perf_counter
.type
= ctx
->u
.perf_counter
.type
;
167 kctx
.u
.perf_counter
.config
= ctx
->u
.perf_counter
.config
;
168 strncpy(kctx
.u
.perf_counter
.name
, ctx
->u
.perf_counter
.name
,
169 LTTNG_SYMBOL_NAME_LEN
);
170 kctx
.u
.perf_counter
.name
[LTTNG_SYMBOL_NAME_LEN
- 1] = '\0';
172 if (strlen(channel_name
) == 0) {
173 ret
= add_kctx_all_channels(ksession
, &kctx
, event_name
);
174 if (ret
!= LTTCOMM_OK
) {
178 /* Get kernel channel */
179 kchan
= trace_kernel_get_channel_by_name(channel_name
, ksession
);
181 ret
= LTTCOMM_KERN_CHAN_NOT_FOUND
;
185 ret
= add_kctx_to_channel(&kctx
, kchan
, event_name
);
186 if (ret
!= LTTCOMM_OK
) {
198 * Add UST context to tracer.
200 int context_ust_add(struct ltt_ust_session
*usess
, int domain
,
201 struct lttng_event_context
*ctx
, char *event_name
,
204 int ret
= LTTCOMM_OK
, have_event
= 0;
205 struct cds_lfht_iter iter
, uiter
;
206 struct cds_lfht
*chan_ht
;
207 struct ltt_ust_channel
*uchan
= NULL
;
208 struct ltt_ust_event
*uevent
= NULL
;
209 struct ltt_ust_context
*uctx
;
212 case LTTNG_DOMAIN_UST
:
213 chan_ht
= usess
->domain_global
.channels
;
215 case LTTNG_DOMAIN_UST_EXEC_NAME
:
216 case LTTNG_DOMAIN_UST_PID
:
217 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
219 ret
= LTTCOMM_NOT_IMPLEMENTED
;
223 if (strlen(event_name
) != 0) {
227 /* Get UST channel if defined */
228 if (strlen(channel_name
) != 0) {
229 uchan
= trace_ust_find_channel_by_name(chan_ht
, channel_name
);
231 ret
= LTTCOMM_UST_CHAN_NOT_FOUND
;
236 /* If UST channel specified and event name, get UST event ref */
237 if (uchan
&& have_event
) {
238 uevent
= trace_ust_find_event_by_name(uchan
->events
, event_name
);
239 if (uevent
== NULL
) {
240 ret
= LTTCOMM_UST_EVENT_NOT_FOUND
;
245 /* Create ltt UST context */
246 uctx
= trace_ust_create_context(ctx
);
252 /* At this point, we have 4 possibilities */
254 if (uchan
&& uevent
) { /* Add ctx to event in channel */
255 ret
= ust_app_add_ctx_event_glb(usess
, uchan
, uevent
, uctx
);
256 } else if (uchan
&& !have_event
) { /* Add ctx to channel */
257 ret
= ust_app_add_ctx_channel_glb(usess
, uchan
, uctx
);
258 } else if (!uchan
&& have_event
) { /* Add ctx to event */
259 cds_lfht_for_each_entry(chan_ht
, &iter
, uchan
, node
) {
260 uevent
= trace_ust_find_event_by_name(uchan
->events
, event_name
);
261 if (uevent
!= NULL
) {
262 ret
= ust_app_add_ctx_event_glb(usess
, uchan
, uevent
, uctx
);
264 * LTTng UST does not allowed the same event to be registered
265 * multiple time in different or the same channel. So, if we
266 * found our event in the first place, no need to continue.
271 ret
= LTTCOMM_UST_EVENT_NOT_FOUND
;
273 } else if (!uchan
&& !have_event
) { /* Add ctx to all events, all channels */
274 cds_lfht_for_each_entry(chan_ht
, &iter
, uchan
, node
) {
275 ret
= ust_app_add_ctx_channel_glb(usess
, uchan
, uctx
);
277 /* Add failed so uctx was not added. We can keep it. */
280 cds_lfht_for_each_entry(uchan
->events
, &uiter
, uevent
, node
) {
281 ret
= ust_app_add_ctx_event_glb(usess
, uchan
, uevent
, uctx
);
283 /* Add failed so uctx was not added. We can keep it. */
286 /* Create ltt UST context */
287 uctx
= trace_ust_create_context(ctx
);
293 /* Create ltt UST context */
294 uctx
= trace_ust_create_context(ctx
);
305 ret
= LTTCOMM_UST_CONTEXT_EXIST
;