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.
26 #include <lttng-share.h>
27 #include <lttng-sessiond-comm.h>
28 #include <lttng/lttng-consumer.h>
30 #include "hashtable.h"
31 #include "ust-consumer.h"
34 * Send all stream fds of UST channel to the consumer.
36 static int send_channel_streams(int sock
,
37 struct ust_app_channel
*uchan
)
40 struct lttcomm_consumer_msg lum
;
41 struct ltt_ust_stream
*stream
, *tmp
;
43 DBG("Sending streams of channel %s to UST consumer", uchan
->name
);
46 lum
.cmd_type
= LTTNG_CONSUMER_ADD_CHANNEL
;
49 * We need to keep shm_fd open while we transfer the stream file
50 * descriptors to make sure this key stays unique within the
51 * session daemon. We can free the channel shm_fd without
52 * problem after we finished sending stream fds for that
55 lum
.u
.channel
.channel_key
= uchan
->obj
->shm_fd
;
56 lum
.u
.channel
.max_sb_size
= uchan
->attr
.subbuf_size
;
57 lum
.u
.channel
.mmap_len
= uchan
->obj
->memory_map_size
;
58 DBG("Sending channel %d to consumer", lum
.u
.channel
.channel_key
);
59 ret
= lttcomm_send_unix_sock(sock
, &lum
, sizeof(lum
));
61 perror("send consumer channel");
64 fd
= uchan
->obj
->shm_fd
;
65 ret
= lttcomm_send_fds_unix_sock(sock
, &fd
, 1);
67 perror("send consumer channel ancillary data");
71 cds_list_for_each_entry_safe(stream
, tmp
, &uchan
->streams
.head
, list
) {
74 if (!stream
->obj
->shm_fd
) {
77 lum
.cmd_type
= LTTNG_CONSUMER_ADD_STREAM
;
78 lum
.u
.stream
.channel_key
= uchan
->obj
->shm_fd
;
79 lum
.u
.stream
.stream_key
= stream
->obj
->shm_fd
;
80 lum
.u
.stream
.state
= LTTNG_CONSUMER_ACTIVE_STREAM
;
82 * FIXME Hack alert! we force MMAP for now. Mixup
83 * between EVENT and UST enums elsewhere.
85 lum
.u
.stream
.output
= DEFAULT_UST_CHANNEL_OUTPUT
;
86 lum
.u
.stream
.mmap_len
= stream
->obj
->memory_map_size
;
87 strncpy(lum
.u
.stream
.path_name
, stream
->pathname
, PATH_MAX
- 1);
88 lum
.u
.stream
.path_name
[PATH_MAX
- 1] = '\0';
89 DBG("Sending stream %d to consumer", lum
.u
.stream
.stream_key
);
90 ret
= lttcomm_send_unix_sock(sock
, &lum
, sizeof(lum
));
92 perror("send consumer stream");
96 fds
[0] = stream
->obj
->shm_fd
;
97 fds
[1] = stream
->obj
->wait_fd
;
98 ret
= lttcomm_send_fds_unix_sock(sock
, fds
, 2);
100 perror("send consumer stream ancillary data");
105 * We release the stream object here, as we have passed
106 * it to the consumer.
108 /* Ensure we don't let the app know (sock = -1). */
109 ustctl_release_object(-1, stream
->obj
);
110 cds_list_del(&stream
->list
);
113 /* Ensure we don't let the app know (sock = -1). */
114 ustctl_release_object(-1, uchan
->obj
);
116 DBG("consumer channel streams sent");
125 * Send all stream fds of the UST session to the consumer.
127 int ust_consumer_send_session(int consumer_fd
, struct ust_app_session
*usess
)
130 int sock
= consumer_fd
;
131 struct cds_lfht_iter iter
;
132 struct cds_lfht_node
*node
;
133 struct lttcomm_consumer_msg lum
;
134 struct ust_app_channel
*uchan
;
136 DBG("Sending metadata stream fd");
138 if (usess
->metadata
->obj
->shm_fd
!= 0) {
142 /* Send metadata channel fd */
143 lum
.cmd_type
= LTTNG_CONSUMER_ADD_CHANNEL
;
144 lum
.u
.channel
.channel_key
= usess
->metadata
->obj
->shm_fd
;
145 lum
.u
.channel
.max_sb_size
= usess
->metadata
->attr
.subbuf_size
;
146 lum
.u
.channel
.mmap_len
= usess
->metadata
->obj
->memory_map_size
;
147 DBG("Sending metadata channel %d to consumer", lum
.u
.channel
.channel_key
);
148 ret
= lttcomm_send_unix_sock(sock
, &lum
, sizeof(lum
));
150 perror("send consumer channel");
153 fd
= usess
->metadata
->obj
->shm_fd
;
154 ret
= lttcomm_send_fds_unix_sock(sock
, &fd
, 1);
156 perror("send consumer metadata channel");
160 /* Send metadata stream fd */
161 lum
.cmd_type
= LTTNG_CONSUMER_ADD_STREAM
;
162 lum
.u
.stream
.channel_key
= usess
->metadata
->obj
->shm_fd
;
163 lum
.u
.stream
.stream_key
= usess
->metadata
->stream_obj
->shm_fd
;
164 lum
.u
.stream
.state
= LTTNG_CONSUMER_ACTIVE_STREAM
;
165 lum
.u
.stream
.output
= DEFAULT_UST_CHANNEL_OUTPUT
;
166 lum
.u
.stream
.mmap_len
= usess
->metadata
->stream_obj
->memory_map_size
;
167 strncpy(lum
.u
.stream
.path_name
, usess
->metadata
->pathname
, PATH_MAX
- 1);
168 lum
.u
.stream
.path_name
[PATH_MAX
- 1] = '\0';
169 DBG("Sending metadata stream %d to consumer", lum
.u
.stream
.stream_key
);
170 ret
= lttcomm_send_unix_sock(sock
, &lum
, sizeof(lum
));
172 perror("send consumer metadata stream");
175 fds
[0] = usess
->metadata
->stream_obj
->shm_fd
;
176 fds
[1] = usess
->metadata
->stream_obj
->wait_fd
;
177 ret
= lttcomm_send_fds_unix_sock(sock
, fds
, 2);
179 perror("send consumer stream");
182 /* Metadata fds passed to consumer, release them. */
183 /* Ensure we don't let the app know (sock = -1). */
184 ustctl_release_object(-1, usess
->metadata
->stream_obj
);
185 ustctl_release_object(-1, usess
->metadata
->obj
);
188 /* Send each channel fd streams of session */
190 hashtable_get_first(usess
->channels
, &iter
);
191 while ((node
= hashtable_iter_get_node(&iter
)) != NULL
) {
192 uchan
= caa_container_of(node
, struct ust_app_channel
, node
);
194 ret
= send_channel_streams(sock
, uchan
);
199 hashtable_get_next(usess
->channels
, &iter
);
203 DBG("consumer fds (metadata and channel streams) sent");