Fix wrong returned valu on UST start trace
[lttng-tools.git] / lttng-sessiond / ust-app.c
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
6 * as published by the Free Software Foundation; only version 2
7 * of the License.
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 <errno.h>
21 #include <pthread.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <unistd.h>
26
27 #include <lttngerr.h>
28 #include <lttng-share.h>
29
30 #include "hashtable.h"
31 #include "ust-app.h"
32 #include "../hashtable/hash.h"
33 #include "ust-ctl.h"
34 #include "ust-consumer.h"
35
36 /*
37 * Delete a traceable application structure from the global list.
38 */
39 static void delete_ust_app(struct ust_app *lta)
40 {
41 int ret;
42 struct cds_lfht_node *node;
43 struct cds_lfht_iter iter;
44
45 rcu_read_lock();
46
47 free(lta->sessions);
48 close(lta->key.sock);
49
50 /* Remove from apps hash table */
51 node = hashtable_lookup(ust_app_ht,
52 (void *) ((unsigned long) lta->key.pid), sizeof(void *), &iter);
53 if (node == NULL) {
54 ERR("UST app pid %d not found in hash table", lta->key.pid);
55 } else {
56 ret = hashtable_del(ust_app_ht, &iter);
57 if (ret) {
58 ERR("UST app unable to delete app %d from hash table",
59 lta->key.pid);
60 } else {
61 DBG2("UST app pid %d deleted", lta->key.pid);
62 }
63 }
64
65 /* Remove from key hash table */
66 node = hashtable_lookup(ust_app_sock_key_map,
67 (void *) ((unsigned long) lta->key.sock), sizeof(void *), &iter);
68 if (node == NULL) {
69 ERR("UST app key %d not found in key hash table", lta->key.sock);
70 } else {
71 ret = hashtable_del(ust_app_sock_key_map, &iter);
72 if (ret) {
73 ERR("UST app unable to delete app sock %d from key hash table",
74 lta->key.sock);
75 } else {
76 DBG2("UST app pair sock %d key %d deleted",
77 lta->key.sock, lta->key.pid);
78 }
79 }
80
81 free(lta);
82
83 rcu_read_unlock();
84 }
85
86 /*
87 * URCU intermediate call to delete an UST app.
88 */
89 static void delete_ust_app_rcu(struct rcu_head *head)
90 {
91 struct cds_lfht_node *node =
92 caa_container_of(head, struct cds_lfht_node, head);
93 struct ust_app *app =
94 caa_container_of(node, struct ust_app, node);
95
96 delete_ust_app(app);
97 }
98
99 /*
100 * Find an ust_app using the sock and return it.
101 */
102 static struct ust_app *find_app_by_sock(int sock)
103 {
104 struct cds_lfht_node *node;
105 struct ust_app_key *key;
106 struct cds_lfht_iter iter;
107
108 rcu_read_lock();
109
110 node = hashtable_lookup(ust_app_sock_key_map,
111 (void *)((unsigned long) sock), sizeof(void *), &iter);
112 if (node == NULL) {
113 DBG2("UST app find by sock %d key not found", sock);
114 rcu_read_unlock();
115 goto error;
116 }
117
118 key = caa_container_of(node, struct ust_app_key, node);
119
120 node = hashtable_lookup(ust_app_ht,
121 (void *)((unsigned long) key->pid), sizeof(void *), &iter);
122 if (node == NULL) {
123 DBG2("UST app find by sock %d not found", sock);
124 rcu_read_unlock();
125 goto error;
126 }
127 rcu_read_unlock();
128
129 return caa_container_of(node, struct ust_app, node);
130
131 error:
132 return NULL;
133 }
134
135 /*
136 * Return pointer to traceable apps list.
137 */
138 struct cds_lfht *ust_app_get_ht(void)
139 {
140 return ust_app_ht;
141 }
142
143 /*
144 * Return ust app pointer or NULL if not found.
145 */
146 struct ust_app *ust_app_find_by_pid(pid_t pid)
147 {
148 struct cds_lfht_node *node;
149 struct cds_lfht_iter iter;
150
151 rcu_read_lock();
152 node = hashtable_lookup(ust_app_ht,
153 (void *)((unsigned long) pid), sizeof(void *), &iter);
154 if (node == NULL) {
155 rcu_read_unlock();
156 DBG2("UST app no found with pid %d", pid);
157 goto error;
158 }
159 rcu_read_unlock();
160
161 DBG2("Found UST app by pid %d", pid);
162
163 return caa_container_of(node, struct ust_app, node);
164
165 error:
166 return NULL;
167 }
168
169 /*
170 * Using pid and uid (of the app), allocate a new ust_app struct and
171 * add it to the global traceable app list.
172 *
173 * On success, return 0, else return malloc ENOMEM.
174 */
175 int ust_app_register(struct ust_register_msg *msg, int sock)
176 {
177 struct ust_app *lta;
178
179 lta = malloc(sizeof(struct ust_app));
180 if (lta == NULL) {
181 PERROR("malloc");
182 return -ENOMEM;
183 }
184
185 lta->uid = msg->uid;
186 lta->gid = msg->gid;
187 lta->key.pid = msg->pid;
188 lta->ppid = msg->ppid;
189 lta->v_major = msg->major;
190 lta->v_minor = msg->minor;
191 lta->key.sock = sock;
192 strncpy(lta->name, msg->name, sizeof(lta->name));
193 lta->name[16] = '\0';
194 hashtable_node_init(&lta->node, (void *)((unsigned long)lta->key.pid),
195 sizeof(void *));
196
197 /* Session hashtable */
198 lta->sessions = hashtable_new(0);
199
200 /* Set sock key map */
201 hashtable_node_init(&lta->key.node, (void *)((unsigned long)lta->key.sock),
202 sizeof(void *));
203
204 rcu_read_lock();
205 hashtable_add_unique(ust_app_ht, &lta->node);
206 hashtable_add_unique(ust_app_sock_key_map, &lta->key.node);
207 rcu_read_unlock();
208
209 DBG("App registered with pid:%d ppid:%d uid:%d gid:%d sock:%d name:%s"
210 " (version %d.%d)", lta->key.pid, lta->ppid, lta->uid, lta->gid,
211 lta->key.sock, lta->name, lta->v_major, lta->v_minor);
212
213 return 0;
214 }
215
216 /*
217 * Unregister app by removing it from the global traceable app list and freeing
218 * the data struct.
219 *
220 * The socket is already closed at this point so no close to sock.
221 */
222 void ust_app_unregister(int sock)
223 {
224 struct ust_app *lta;
225
226 DBG2("UST app unregistering sock %d", sock);
227
228 lta = find_app_by_sock(sock);
229 if (lta) {
230 DBG("PID %d unregistering with sock %d", lta->key.pid, sock);
231 /* FIXME: Better use a call_rcu here ? */
232 delete_ust_app(lta);
233 }
234 }
235
236 /*
237 * Return traceable_app_count
238 */
239 unsigned long ust_app_list_count(void)
240 {
241 unsigned long count;
242
243 rcu_read_lock();
244 count = hashtable_get_count(ust_app_ht);
245 rcu_read_unlock();
246
247 return count;
248 }
249
250 /*
251 * Free and clean all traceable apps of the global list.
252 */
253 void ust_app_clean_list(void)
254 {
255 int ret;
256 struct cds_lfht_node *node;
257 struct cds_lfht_iter iter;
258
259 DBG2("UST app clean hash table");
260
261 rcu_read_lock();
262
263 hashtable_get_first(ust_app_ht, &iter);
264 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
265 ret = hashtable_del(ust_app_ht, &iter);
266 if (!ret) {
267 call_rcu(&node->head, delete_ust_app_rcu);
268 }
269 hashtable_get_next(ust_app_ht, &iter);
270 }
271
272 rcu_read_unlock();
273 }
274
275 /*
276 * Init UST app hash table.
277 */
278 void ust_app_ht_alloc(void)
279 {
280 ust_app_ht = hashtable_new(0);
281 ust_app_sock_key_map = hashtable_new(0);
282 }
283
284 /*
285 * Alloc new UST app session.
286 */
287 static struct ust_app_session *alloc_app_session(void)
288 {
289 struct ust_app_session *ua_sess;
290
291 ua_sess = zmalloc(sizeof(struct ust_app_session));
292 if (ua_sess == NULL) {
293 PERROR("malloc");
294 goto error;
295 }
296
297 ua_sess->enabled = 0;
298 ua_sess->handle = -1;
299 ua_sess->channels = hashtable_new_str(0);
300 ua_sess->metadata = NULL;
301 ua_sess->obj = NULL;
302
303 return ua_sess;
304
305 error:
306 return NULL;
307 }
308
309 static struct ust_app_channel *alloc_app_channel(char *name)
310 {
311 struct ust_app_channel *ua_chan;
312
313 ua_chan = zmalloc(sizeof(struct ust_app_channel));
314 if (ua_chan == NULL) {
315 PERROR("malloc");
316 goto error;
317 }
318
319 strncpy(ua_chan->name, name, sizeof(ua_chan->name));
320 ua_chan->name[sizeof(ua_chan->name) - 1] = '\0';
321 ua_chan->enabled = 0;
322 ua_chan->handle = -1;
323 ua_chan->obj = NULL;
324 ua_chan->ctx = hashtable_new(0);
325 CDS_INIT_LIST_HEAD(&ua_chan->streams.head);
326 ua_chan->events = hashtable_new_str(0);
327 hashtable_node_init(&ua_chan->node, (void *) ua_chan->name,
328 strlen(ua_chan->name));
329
330 DBG3("UST app channel %s allocated", ua_chan->name);
331
332 return ua_chan;
333
334 error:
335 return NULL;
336 }
337
338 static struct ust_app_event *alloc_app_event(char *name)
339 {
340 struct ust_app_event *ua_event;
341
342 ua_event = zmalloc(sizeof(struct ust_app_event));
343 if (ua_event == NULL) {
344 PERROR("malloc");
345 goto error;
346 }
347
348 strncpy(ua_event->name, name, sizeof(ua_event->name));
349 ua_event->name[sizeof(ua_event->name) - 1] = '\0';
350 ua_event->ctx = hashtable_new(0);
351 hashtable_node_init(&ua_event->node, (void *) ua_event->name,
352 strlen(ua_event->name));
353
354 DBG3("UST app event %s allocated", ua_event->name);
355
356 return ua_event;
357
358 error:
359 return NULL;
360 }
361
362 static void shallow_copy_event(struct ust_app_event *ua_event,
363 struct ltt_ust_event *uevent)
364 {
365 strncpy(ua_event->name, uevent->attr.name, sizeof(ua_event->name));
366 ua_event->name[sizeof(ua_event->name) - 1] = '\0';
367
368 /* TODO: support copy context */
369 }
370
371 static void shallow_copy_channel(struct ust_app_channel *ua_chan,
372 struct ltt_ust_channel *uchan)
373 {
374 struct cds_lfht_iter iter;
375 struct cds_lfht_node *node, *ua_event_node;
376 struct ltt_ust_event *uevent;
377 struct ust_app_event *ua_event;
378
379 DBG2("Shallow copy of UST app channel %s", ua_chan->name);
380
381 strncpy(ua_chan->name, uchan->name, sizeof(ua_chan->name));
382 ua_chan->name[sizeof(ua_chan->name) - 1] = '\0';
383
384 /* TODO: support copy context */
385
386 hashtable_get_first(uchan->events, &iter);
387 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
388 uevent = caa_container_of(node, struct ltt_ust_event, node);
389
390 ua_event_node = hashtable_lookup(ua_chan->events,
391 (void *) uevent->attr.name, strlen(uevent->attr.name), &iter);
392 if (ua_event_node == NULL) {
393 DBG2("UST event %s not found on shallow copy channel",
394 uevent->attr.name);
395 ua_event = alloc_app_event(uevent->attr.name);
396 if (ua_event == NULL) {
397 continue;
398 }
399 hashtable_add_unique(ua_chan->events, &ua_event->node);
400 } else {
401 ua_event = caa_container_of(node, struct ust_app_event, node);
402 }
403
404 shallow_copy_event(ua_event, uevent);
405
406 /* Get next UST events */
407 hashtable_get_next(uchan->events, &iter);
408 }
409
410 DBG3("Shallow copy channel done");
411 }
412
413 static void shallow_copy_session(struct ust_app_session *ua_sess,
414 struct ltt_ust_session *usess)
415 {
416 struct cds_lfht_node *node, *ua_chan_node;
417 struct cds_lfht_iter iter;
418 struct ltt_ust_channel *uchan;
419 struct ust_app_channel *ua_chan;
420
421 DBG2("Shallow copy of session handle");
422
423 ua_sess->uid = usess->uid;
424
425 /* TODO: support all UST domain */
426
427 /* Iterate over all channels in global domain. */
428 hashtable_get_first(usess->domain_global.channels, &iter);
429 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
430 uchan = caa_container_of(node, struct ltt_ust_channel, node);
431
432 ua_chan_node = hashtable_lookup(ua_sess->channels,
433 (void *) uchan->name, strlen(uchan->name), &iter);
434 if (ua_chan_node == NULL) {
435 DBG2("Channel %s not found on shallow session copy, creating it",
436 uchan->name);
437 ua_chan = alloc_app_channel(uchan->name);
438 if (ua_chan == NULL) {
439 /* malloc failed... continuing */
440 continue;
441 }
442 hashtable_add_unique(ua_sess->channels, &ua_chan->node);
443 } else {
444 ua_chan = caa_container_of(node, struct ust_app_channel, node);
445 }
446
447 shallow_copy_channel(ua_chan, uchan);
448
449 /* Next item in hash table */
450 hashtable_get_next(usess->domain_global.channels, &iter);
451 }
452 }
453
454 static struct ust_app_session *lookup_session_by_app(
455 struct ltt_ust_session *usess, struct ust_app *app)
456 {
457 struct cds_lfht_iter iter;
458 struct cds_lfht_node *node;
459
460 /* Get right UST app session from app */
461 node = hashtable_lookup(app->sessions,
462 (void *) ((unsigned long) usess->uid),
463 sizeof(void *), &iter);
464 if (node == NULL) {
465 goto error;
466 }
467
468 return caa_container_of(node, struct ust_app_session, node);
469
470 error:
471 return NULL;
472 }
473
474 int ust_app_add_channel(struct ltt_ust_session *usess,
475 struct ltt_ust_channel *uchan)
476 {
477 int ret = 0;
478 struct cds_lfht_iter iter;
479 struct cds_lfht_node *node, *ua_chan_node;
480 struct ust_app *app;
481 struct ust_app_session *ua_sess;
482 struct ust_app_channel *ua_chan;
483
484 DBG2("UST app adding channel %s to global domain for session uid %d",
485 uchan->name, usess->uid);
486
487 rcu_read_lock();
488 hashtable_get_first(ust_app_ht, &iter);
489 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
490 app = caa_container_of(node, struct ust_app, node);
491
492 ua_sess = lookup_session_by_app(usess, app);
493 if (ua_sess == NULL) {
494 DBG2("UST app pid: %d session uid %d not found, creating one",
495 app->key.pid, usess->uid);
496 ua_sess = alloc_app_session();
497 if (ua_sess == NULL) {
498 /* Only malloc can failed so something is really wrong */
499 goto next;
500 }
501 shallow_copy_session(ua_sess, usess);
502 }
503
504 if (ua_sess->handle == -1) {
505 ret = ustctl_create_session(app->key.sock);
506 if (ret < 0) {
507 DBG("Error creating session for app pid %d, sock %d",
508 app->key.pid, app->key.sock);
509 /* TODO: free() ua_sess */
510 goto next;
511 }
512
513 DBG2("UST app ustctl create session handle %d", ret);
514 ua_sess->handle = ret;
515
516 /* Add ust app session to app's HT */
517 hashtable_node_init(&ua_sess->node,
518 (void *)((unsigned long) ua_sess->uid), sizeof(void *));
519 hashtable_add_unique(app->sessions, &ua_sess->node);
520 }
521
522 /* Lookup channel in the ust app session */
523 ua_chan_node = hashtable_lookup(ua_sess->channels,
524 (void *) uchan->name, strlen(uchan->name), &iter);
525 if (ua_chan_node == NULL) {
526 ERR("Channel suppose to be present with the above shallow "
527 "session copy. Continuing...");
528 goto next;
529 }
530
531 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
532
533 /* TODO: remove cast and use lttng-ust-abi.h */
534 ret = ustctl_create_channel(app->key.sock, ua_sess->handle,
535 (struct lttng_ust_channel_attr *)&uchan->attr, &ua_chan->obj);
536 if (ret < 0) {
537 DBG("Error creating channel %s for app (pid: %d, sock: %d) "
538 "and session handle %d with ret %d",
539 uchan->name, app->key.pid, app->key.sock,
540 ua_sess->handle, ret);
541 goto next;
542 }
543
544 ua_chan->handle = ua_chan->obj->handle;
545 ua_chan->attr.shm_fd = ua_chan->obj->shm_fd;
546 ua_chan->attr.wait_fd = ua_chan->obj->wait_fd;
547 ua_chan->attr.memory_map_size = ua_chan->obj->memory_map_size;
548
549 DBG2("Channel %s UST create successfully for pid:%d and sock:%d",
550 uchan->name, app->key.pid, app->key.sock);
551
552 next:
553 /* Next applications */
554 hashtable_get_next(ust_app_ht, &iter);
555 }
556 rcu_read_unlock();
557
558 return ret;
559 }
560
561 int ust_app_add_event(struct ltt_ust_session *usess,
562 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
563 {
564 int ret = 0;
565 struct cds_lfht_iter iter;
566 struct cds_lfht_node *node, *ua_chan_node, *ua_event_node;
567 struct ust_app *app;
568 struct ust_app_session *ua_sess;
569 struct ust_app_channel *ua_chan;
570 struct ust_app_event *ua_event;
571 struct lttng_ust_event ltt_uevent;
572 struct lttng_ust_object_data *obj_event;
573
574 DBG2("UST app adding event %s to global domain for session uid %d",
575 uevent->attr.name, usess->uid);
576
577 rcu_read_lock();
578 hashtable_get_first(ust_app_ht, &iter);
579 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
580 app = caa_container_of(node, struct ust_app, node);
581
582 ua_sess = lookup_session_by_app(usess, app);
583 if (ua_sess == NULL) {
584 DBG2("UST app (pid: %d, sock: %d) session not found, creating one",
585 app->key.pid, app->key.sock);
586 ua_sess = alloc_app_session();
587 if (ua_sess == NULL) {
588 /* Only malloc can failed so something is really wrong */
589 goto next;
590 }
591 shallow_copy_session(ua_sess, usess);
592 }
593
594 if (ua_sess->handle == -1) {
595 ret = ustctl_create_session(app->key.sock);
596 if (ret < 0) {
597 DBG("Error creating session for app pid %d, sock %d",
598 app->key.pid, app->key.sock);
599 /* TODO: free() ua_sess */
600 goto next;
601 }
602
603 DBG2("UST app ustctl create session handle %d", ret);
604 ua_sess->handle = ret;
605 /* Add ust app session to app's HT */
606 hashtable_node_init(&ua_sess->node,
607 (void *)((unsigned long) ua_sess->uid), sizeof(void *));
608 hashtable_add_unique(app->sessions, &ua_sess->node);
609 }
610
611 /* Lookup channel in the ust app session */
612 ua_chan_node = hashtable_lookup(ua_sess->channels,
613 (void *) uchan->name, strlen(uchan->name), &iter);
614 if (ua_chan_node == NULL) {
615 ERR("Channel suppose to be present with the above shallow "
616 "session copy. Continuing...");
617 goto next;
618 }
619
620 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
621
622 /* Prepare lttng ust event */
623 strncpy(ltt_uevent.name, uevent->attr.name, sizeof(ltt_uevent.name));
624 ltt_uevent.name[sizeof(ltt_uevent.name) - 1] = '\0';
625 /* TODO: adjust to other instrumentation types */
626 ltt_uevent.instrumentation = LTTNG_UST_TRACEPOINT;
627
628 /* Get event node */
629 ua_event_node = hashtable_lookup(ua_chan->events,
630 (void *) uevent->attr.name, strlen(uevent->attr.name), &iter);
631 if (ua_event_node == NULL) {
632 DBG2("UST app event %s not found, creating one", uevent->attr.name);
633 /* Does not exist so create one */
634 ua_event = alloc_app_event(uevent->attr.name);
635 if (ua_event == NULL) {
636 /* Only malloc can failed so something is really wrong */
637 goto next;
638 }
639
640 shallow_copy_event(ua_event, uevent);
641
642 /* Create UST event on tracer */
643 ret = ustctl_create_event(app->key.sock, &ltt_uevent, ua_chan->obj,
644 &obj_event);
645 if (ret < 0) {
646 ERR("Error ustctl create event %s for app pid: %d with ret %d",
647 uevent->attr.name, app->key.pid, ret);
648 /* TODO: free() ua_event and obj_event */
649 goto next;
650 }
651 ua_event->obj = obj_event;
652 ua_event->handle = obj_event->handle;
653 ua_event->enabled = 1;
654 } else {
655 ua_event = caa_container_of(ua_event_node,
656 struct ust_app_event, node);
657
658 if (ua_event->enabled == 0) {
659 ret = ustctl_enable(app->key.sock, ua_event->obj);
660 if (ret < 0) {
661 ERR("Error ustctl enable event %s for app "
662 "pid: %d with ret %d", uevent->attr.name,
663 app->key.pid, ret);
664 goto next;
665 }
666 ua_event->enabled = 1;
667 }
668 }
669
670 hashtable_add_unique(ua_chan->events, &ua_event->node);
671
672 DBG2("Event %s UST create successfully for pid:%d", uevent->attr.name,
673 app->key.pid);
674
675 next:
676 /* Next applications */
677 hashtable_get_next(ust_app_ht, &iter);
678 }
679 rcu_read_unlock();
680
681 return ret;
682 }
683
684 int ust_app_start_trace(struct ltt_ust_session *usess)
685 {
686 int ret = 0;
687 struct cds_lfht_iter iter;
688 struct cds_lfht_node *node, *ua_chan_node;
689 struct ust_app *app;
690 struct ust_app_session *ua_sess;
691 struct ust_app_channel *ua_chan;
692 struct lttng_ust_channel_attr uattr;
693 struct ltt_ust_channel *uchan;
694
695 DBG("Starting all UST traces");
696
697 rcu_read_lock();
698 hashtable_get_first(ust_app_ht, &iter);
699 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
700 app = caa_container_of(node, struct ust_app, node);
701
702 ua_sess = lookup_session_by_app(usess, app);
703 if (ua_sess == NULL) {
704 /* Only malloc can failed so something is really wrong */
705 goto next;
706 }
707
708 if (ua_sess->metadata == NULL) {
709 /* Allocate UST metadata */
710 ua_sess->metadata = trace_ust_create_metadata(usess->pathname);
711 if (ua_sess->metadata == NULL) {
712 ERR("UST app session %d creating metadata failed",
713 ua_sess->handle);
714 goto next;
715 }
716
717 uattr.overwrite = ua_sess->metadata->attr.overwrite;
718 uattr.subbuf_size = ua_sess->metadata->attr.subbuf_size;
719 uattr.num_subbuf = ua_sess->metadata->attr.num_subbuf;
720 uattr.switch_timer_interval =
721 ua_sess->metadata->attr.switch_timer_interval;
722 uattr.read_timer_interval =
723 ua_sess->metadata->attr.read_timer_interval;
724 uattr.output = ua_sess->metadata->attr.output;
725
726 /* UST tracer metadata creation */
727 ret = ustctl_open_metadata(app->key.sock, ua_sess->handle, &uattr,
728 &ua_sess->metadata->obj);
729 if (ret < 0) {
730 ERR("UST app open metadata failed for app pid:%d",
731 app->key.pid);
732 goto next;
733 }
734
735 DBG2("UST metadata opened for app pid %d", app->key.pid);
736 }
737
738 /* Open UST metadata stream */
739 if (ua_sess->metadata->stream_obj == NULL) {
740 ret = ustctl_create_stream(app->key.sock, ua_sess->metadata->obj,
741 &ua_sess->metadata->stream_obj);
742 if (ret < 0) {
743 ERR("UST create metadata stream failed");
744 goto next;
745 }
746
747 ret = snprintf(ua_sess->metadata->pathname, PATH_MAX, "%s/%s",
748 usess->pathname, "metadata");
749 if (ret < 0) {
750 PERROR("asprintf UST create stream");
751 goto next;
752 }
753
754 DBG2("UST metadata stream object created for app pid %d",
755 app->key.pid);
756 }
757
758 /* For each channel */
759 hashtable_get_first(usess->domain_global.channels, &iter);
760 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
761 uchan = caa_container_of(node, struct ltt_ust_channel, node);
762
763 /* Lookup channel in the ust app session */
764 ua_chan_node = hashtable_lookup(ua_sess->channels,
765 (void *) uchan->name, strlen(uchan->name), &iter);
766 if (ua_chan_node == NULL) {
767 ERR("Channel suppose to be present with the above shallow "
768 "session copy. Continuing...");
769 goto next;
770 }
771
772 ua_chan = caa_container_of(ua_chan_node,
773 struct ust_app_channel, node);
774
775 for (;;) {
776 struct lttng_ust_object_data *obj;
777 struct ltt_ust_stream *ustream;
778
779 ret = ustctl_create_stream(app->key.sock, ua_chan->obj,
780 &obj);
781 if (ret < 0) {
782 /* Got all streams */
783 goto next_chan;
784 }
785
786 ustream = malloc(sizeof(*ustream));
787 if (ustream == NULL) {
788 goto next_chan;
789 }
790 memset(ustream, 0, sizeof(struct ltt_ust_stream));
791 ustream->obj = obj;
792 ustream->handle = ustream->obj->handle;
793 /* Order is important */
794 cds_list_add_tail(&ustream->list, &ua_chan->streams.head);
795 ret = snprintf(ustream->pathname, PATH_MAX, "%s/%s_%u",
796 uchan->pathname, uchan->name,
797 ua_chan->streams.count++);
798 if (ret < 0) {
799 PERROR("asprintf UST create stream");
800 goto next_chan;
801 }
802 }
803
804 next_chan:
805 /* Next applications */
806 hashtable_get_next(ua_sess->channels, &iter);
807 }
808
809 /* Setup UST consumer socket and send fds to it */
810 ret = ust_consumer_send_session(usess->consumer_fd, ua_sess);
811 if (ret < 0) {
812 goto next;
813 }
814
815 /* This start the UST tracing */
816 ret = ustctl_start_session(app->key.sock, ua_sess->handle);
817 if (ret < 0) {
818 ERR("Error starting tracing for app pid: %d", app->key.pid);
819 goto next;
820 }
821
822 /* Quiescent wait after starting trace */
823 ustctl_wait_quiescent(app->key.sock);
824 next:
825 /* Next applications */
826 hashtable_get_next(ust_app_ht, &iter);
827 }
828 rcu_read_unlock();
829
830 return 0;
831 }
832
833 void ust_app_global_update(struct ltt_ust_session *usess, int sock)
834 {
835 int ret = 0;
836 int session_existed = 1;
837 struct cds_lfht_iter iter;
838 struct cds_lfht_node *node, *ua_chan_node;
839 struct ust_app *app;
840 struct ust_app_session *ua_sess;
841 struct ust_app_channel *ua_chan;
842 struct ust_app_event *ua_event;
843 struct lttng_ust_event ltt_uevent;
844 struct ltt_ust_channel *uchan;
845 struct lttng_ust_object_data *obj_event;
846
847 DBG2("UST app global update for app sock %d for session uid %d", sock,
848 usess->uid);
849
850 rcu_read_lock();
851 app = find_app_by_sock(sock);
852 if (app == NULL) {
853 ERR("Failed to update app sock %d", sock);
854 goto error;
855 }
856
857 ua_sess = lookup_session_by_app(usess, app);
858 if (ua_sess == NULL) {
859 DBG2("UST app pid: %d session uid %d not found, creating one",
860 app->key.pid, usess->uid);
861 ua_sess = alloc_app_session();
862 if (ua_sess == NULL) {
863 /* Only malloc can failed so something is really wrong */
864 goto error;
865 }
866 shallow_copy_session(ua_sess, usess);
867 session_existed= 0;
868 }
869
870 if (ua_sess->handle == -1) {
871 ret = ustctl_create_session(app->key.sock);
872 if (ret < 0) {
873 DBG("Error creating session for app pid %d, sock %d",
874 app->key.pid, app->key.sock);
875 /* TODO: free() ua_sess */
876 goto error;
877 }
878
879 DBG2("UST app ustctl create session handle %d", ret);
880 ua_sess->handle = ret;
881
882 /* Add ust app session to app's HT */
883 hashtable_node_init(&ua_sess->node,
884 (void *)((unsigned long) ua_sess->uid), sizeof(void *));
885 hashtable_add_unique(app->sessions, &ua_sess->node);
886 }
887
888 if (session_existed) {
889 goto error;
890 }
891
892 /* Iterate over all channels */
893 hashtable_get_first(usess->domain_global.channels, &iter);
894 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
895 uchan = caa_container_of(node, struct ltt_ust_channel, node);
896
897 /* Lookup channel in the ust app session */
898 ua_chan_node = hashtable_lookup(ua_sess->channels,
899 (void *) uchan->name, strlen(uchan->name), &iter);
900 if (ua_chan_node == NULL) {
901 ERR("UST app channel not found for uchan %s", uchan->name);
902 goto next_chan;
903 }
904
905 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
906
907 /* TODO: remove cast and use lttng-ust-abi.h */
908 ret = ustctl_create_channel(app->key.sock, ua_sess->handle,
909 (struct lttng_ust_channel_attr *)&uchan->attr, &ua_chan->obj);
910 if (ret < 0) {
911 DBG("Error creating channel %s for app (pid: %d, sock: %d) "
912 "and session handle %d with ret %d",
913 uchan->name, app->key.pid, app->key.sock,
914 ua_sess->handle, ret);
915 goto next_chan;
916 }
917
918 ua_chan->handle = ua_chan->obj->handle;
919 ua_chan->attr.shm_fd = ua_chan->obj->shm_fd;
920 ua_chan->attr.wait_fd = ua_chan->obj->wait_fd;
921 ua_chan->attr.memory_map_size = ua_chan->obj->memory_map_size;
922
923 DBG2("Channel %s UST create successfully for pid:%d and sock:%d",
924 uchan->name, app->key.pid, app->key.sock);
925
926 /* For each event(s) of that channel */
927 hashtable_get_first(ua_chan->events, &iter);
928 while ((node = hashtable_iter_get_node(&iter)) != NULL) {
929 ua_event = caa_container_of(node, struct ust_app_event, node);
930
931 /* Prepare lttng ust event */
932 memset(&ltt_uevent, 0, sizeof(ltt_uevent));
933 strncpy(ltt_uevent.name, ua_event->name, sizeof(ltt_uevent.name));
934 ltt_uevent.name[sizeof(ltt_uevent.name) - 1] = '\0';
935
936 /* TODO: adjust to other instrumentation types */
937 ltt_uevent.instrumentation = LTTNG_UST_TRACEPOINT;
938
939 /* Create UST event on tracer */
940 ret = ustctl_create_event(app->key.sock, &ltt_uevent, ua_chan->obj,
941 &obj_event);
942 if (ret < 0) {
943 ERR("Error ustctl create event %s for app pid: %d with ret %d",
944 ua_event->name, app->key.pid, ret);
945 /* TODO: free() ua_event and obj_event */
946 goto next_event;
947 }
948
949 ua_event->obj = obj_event;
950 ua_event->handle = obj_event->handle;
951 ua_event->enabled = 1;
952
953 DBG2("Event %s UST create successfully for pid:%d",
954 ua_event->name, app->key.pid);
955
956 next_event:
957 hashtable_get_next(ua_chan->events, &iter);
958 }
959
960 next_chan:
961 /* Next applications */
962 hashtable_get_next(ua_sess->channels, &iter);
963 }
964
965 error:
966 rcu_read_unlock();
967 return;
968 }
This page took 0.076999 seconds and 4 git commands to generate.