2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
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.
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.
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.
23 #include <urcu/list.h>
25 #include <common/error.h>
26 #include <common/sessiond-comm/sessiond-comm.h>
31 #include "trace-ust.h"
34 * Add kernel context to all channel.
36 static int add_kctx_all_channels(struct ltt_kernel_session
*ksession
,
37 struct lttng_kernel_context
*kctx
)
40 struct ltt_kernel_channel
*kchan
;
45 DBG("Adding kernel context to all channels");
47 /* Go over all channels */
48 cds_list_for_each_entry(kchan
, &ksession
->channel_list
.head
, list
) {
49 ret
= kernel_add_channel_context(kchan
, kctx
);
51 ret
= LTTNG_ERR_KERN_CONTEXT_FAIL
;
63 * Add kernel context to a specific channel.
65 static int add_kctx_to_channel(struct lttng_kernel_context
*kctx
,
66 struct ltt_kernel_channel
*kchan
)
73 DBG("Add kernel context to channel '%s'", kchan
->channel
->name
);
75 ret
= kernel_add_channel_context(kchan
, kctx
);
77 ret
= LTTNG_ERR_KERN_CONTEXT_FAIL
;
88 * Add UST context to channel.
90 static int add_uctx_to_channel(struct ltt_ust_session
*usess
, int domain
,
91 struct ltt_ust_channel
*uchan
, struct lttng_event_context
*ctx
)
94 struct ltt_ust_context
*uctx
;
95 struct lttng_ht_iter iter
;
96 struct lttng_ht_node_ulong
*uctx_node
;
102 /* Create ltt UST context */
103 uctx
= trace_ust_create_context(ctx
);
110 case LTTNG_DOMAIN_UST
:
111 ret
= ust_app_add_ctx_channel_glb(usess
, uchan
, uctx
);
123 /* Lookup context before adding it */
124 lttng_ht_lookup(uchan
->ctx
, (void *)((unsigned long)uctx
->ctx
.ctx
), &iter
);
125 uctx_node
= lttng_ht_iter_get_node_ulong(&iter
);
126 if (uctx_node
!= NULL
) {
132 /* Add ltt UST context node to ltt UST channel */
133 lttng_ht_add_unique_ulong(uchan
->ctx
, &uctx
->node
);
136 DBG("Context UST %d added to channel %s", uctx
->ctx
.ctx
, uchan
->name
);
146 * Add kernel context to tracer.
148 int context_kernel_add(struct ltt_kernel_session
*ksession
,
149 struct lttng_event_context
*ctx
, char *channel_name
)
152 struct ltt_kernel_channel
*kchan
;
153 struct lttng_kernel_context kctx
;
157 assert(channel_name
);
159 /* Setup kernel context structure */
161 case LTTNG_EVENT_CONTEXT_PID
:
162 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_PID
;
164 case LTTNG_EVENT_CONTEXT_PERF_COUNTER
:
165 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_PERF_COUNTER
;
167 case LTTNG_EVENT_CONTEXT_PROCNAME
:
168 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_PROCNAME
;
170 case LTTNG_EVENT_CONTEXT_PRIO
:
171 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_PRIO
;
173 case LTTNG_EVENT_CONTEXT_NICE
:
174 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_NICE
;
176 case LTTNG_EVENT_CONTEXT_VPID
:
177 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_VPID
;
179 case LTTNG_EVENT_CONTEXT_TID
:
180 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_TID
;
182 case LTTNG_EVENT_CONTEXT_VTID
:
183 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_VTID
;
185 case LTTNG_EVENT_CONTEXT_PPID
:
186 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_PPID
;
188 case LTTNG_EVENT_CONTEXT_VPPID
:
189 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_VPPID
;
191 case LTTNG_EVENT_CONTEXT_HOSTNAME
:
192 kctx
.ctx
= LTTNG_KERNEL_CONTEXT_HOSTNAME
;
195 return LTTNG_ERR_KERN_CONTEXT_FAIL
;
198 kctx
.u
.perf_counter
.type
= ctx
->u
.perf_counter
.type
;
199 kctx
.u
.perf_counter
.config
= ctx
->u
.perf_counter
.config
;
200 strncpy(kctx
.u
.perf_counter
.name
, ctx
->u
.perf_counter
.name
,
201 LTTNG_SYMBOL_NAME_LEN
);
202 kctx
.u
.perf_counter
.name
[LTTNG_SYMBOL_NAME_LEN
- 1] = '\0';
204 if (*channel_name
== '\0') {
205 ret
= add_kctx_all_channels(ksession
, &kctx
);
206 if (ret
!= LTTNG_OK
) {
210 /* Get kernel channel */
211 kchan
= trace_kernel_get_channel_by_name(channel_name
, ksession
);
213 ret
= LTTNG_ERR_KERN_CHAN_NOT_FOUND
;
217 ret
= add_kctx_to_channel(&kctx
, kchan
);
218 if (ret
!= LTTNG_OK
) {
230 * Add UST context to tracer.
232 int context_ust_add(struct ltt_ust_session
*usess
, int domain
,
233 struct lttng_event_context
*ctx
, char *channel_name
)
236 struct lttng_ht_iter iter
;
237 struct lttng_ht
*chan_ht
;
238 struct ltt_ust_channel
*uchan
= NULL
;
242 assert(channel_name
);
247 * Define which channel's hashtable to use from the domain or quit if
251 case LTTNG_DOMAIN_UST
:
252 chan_ht
= usess
->domain_global
.channels
;
255 case LTTNG_DOMAIN_UST_EXEC_NAME
:
256 case LTTNG_DOMAIN_UST_PID
:
257 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
264 /* Get UST channel if defined */
265 if (channel_name
!= '\0') {
266 uchan
= trace_ust_find_channel_by_name(chan_ht
, channel_name
);
268 ret
= LTTNG_ERR_UST_CHAN_NOT_FOUND
;
274 /* Add ctx to channel */
275 ret
= add_uctx_to_channel(usess
, domain
, uchan
, ctx
);
278 /* Add ctx all events, all channels */
279 cds_lfht_for_each_entry(chan_ht
->ht
, &iter
.iter
, uchan
, node
.node
) {
280 ret
= add_uctx_to_channel(usess
, domain
, uchan
, ctx
);
282 ERR("Context failed for channel %s", uchan
->name
);
291 ret
= LTTNG_ERR_UST_CONTEXT_EXIST
;
294 ret
= LTTNG_ERR_FATAL
;
297 ret
= LTTNG_ERR_UST_CONTEXT_INVAL
;
300 ret
= LTTNG_ERR_UNKNOWN_DOMAIN
;