b428f1add176567f5972126581b0dac3fb25c2df
[lttng-tools.git] / tests / regression / tools / live / live_test.c
1 /*
2 * Copyright (C) 2013 Julien Desfossez <jdesfossez@efficios.com>
3 *
4 * SPDX-License-Identifier: GPL-2.0-only
5 *
6 */
7
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <string.h>
11 #include <unistd.h>
12 #include <common/compat/time.h>
13 #include <sys/types.h>
14 #include <inttypes.h>
15 #include <stdlib.h>
16 #include <sys/socket.h>
17 #include <netinet/in.h>
18 #include <netdb.h>
19 #include <fcntl.h>
20 #include <sys/mman.h>
21 #include <sys/stat.h>
22
23 #include <tap/tap.h>
24 #include <lttng/lttng.h>
25
26 #include <urcu/list.h>
27 #include <common/common.h>
28
29 #include <bin/lttng-relayd/lttng-viewer-abi.h>
30 #include <common/index/ctf-index.h>
31
32 #include <common/compat/errno.h>
33 #include <common/compat/endian.h>
34
35 #define SESSION1 "test1"
36 #define RELAYD_URL "net://localhost"
37 #define LIVE_TIMER 2000000
38
39 /* Number of TAP tests in this file */
40 #define NUM_TESTS 11
41 #define mmap_size 524288
42
43 #ifdef HAVE_LIBLTTNG_UST_CTL
44 #include <lttng/ust-sigbus.h>
45 DEFINE_LTTNG_UST_SIGBUS_STATE();
46 #endif
47
48 static int control_sock;
49 struct live_session *session;
50
51 static int first_packet_offset;
52 static int first_packet_len;
53 static int first_packet_stream_id = -1;
54
55 struct viewer_stream {
56 uint64_t id;
57 uint64_t ctf_trace_id;
58 void *mmap_base;
59 int fd;
60 int metadata_flag;
61 int first_read;
62 char path[PATH_MAX];
63 };
64
65 struct live_session {
66 struct viewer_stream *streams;
67 uint64_t live_timer_interval;
68 uint64_t stream_count;
69 };
70
71 static
72 ssize_t lttng_live_recv(int fd, void *buf, size_t len)
73 {
74 ssize_t ret;
75 size_t copied = 0, to_copy = len;
76
77 do {
78 ret = recv(fd, buf + copied, to_copy, 0);
79 if (ret > 0) {
80 LTTNG_ASSERT(ret <= to_copy);
81 copied += ret;
82 to_copy -= ret;
83 }
84 } while ((ret > 0 && to_copy > 0)
85 || (ret < 0 && errno == EINTR));
86 if (ret > 0)
87 ret = copied;
88 /* ret = 0 means orderly shutdown, ret < 0 is error. */
89 return ret;
90 }
91
92 static
93 ssize_t lttng_live_send(int fd, const void *buf, size_t len)
94 {
95 ssize_t ret;
96
97 do {
98 ret = send(fd, buf, len, MSG_NOSIGNAL);
99 } while (ret < 0 && errno == EINTR);
100 return ret;
101 }
102
103 static
104 int connect_viewer(const char *hostname)
105 {
106 struct hostent *host;
107 struct sockaddr_in server_addr;
108 int ret;
109
110 host = gethostbyname(hostname);
111 if (!host) {
112 ret = -1;
113 goto end;
114 }
115
116 if ((control_sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
117 PERROR("Socket");
118 ret = -1;
119 goto end;
120 }
121
122 server_addr.sin_family = AF_INET;
123 server_addr.sin_port = htons(5344);
124 server_addr.sin_addr = *((struct in_addr *) host->h_addr);
125 bzero(&(server_addr.sin_zero), 8);
126
127 if (connect(control_sock, (struct sockaddr *) &server_addr,
128 sizeof(struct sockaddr)) == -1) {
129 PERROR("Connect");
130 ret = -1;
131 goto end;
132 }
133
134 server_addr.sin_family = AF_INET;
135 server_addr.sin_port = htons(5345);
136 server_addr.sin_addr = *((struct in_addr *) host->h_addr);
137 bzero(&(server_addr.sin_zero), 8);
138
139 ret = 0;
140
141 end:
142 return ret;
143 }
144
145 static
146 int establish_connection(void)
147 {
148 struct lttng_viewer_cmd cmd;
149 struct lttng_viewer_connect connect;
150 ssize_t ret_len;
151
152 cmd.cmd = htobe32(LTTNG_VIEWER_CONNECT);
153 cmd.data_size = htobe64(sizeof(connect));
154 cmd.cmd_version = htobe32(0);
155
156 memset(&connect, 0, sizeof(connect));
157 connect.major = htobe32(VERSION_MAJOR);
158 connect.minor = htobe32(VERSION_MINOR);
159 connect.type = htobe32(LTTNG_VIEWER_CLIENT_COMMAND);
160
161 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
162 if (ret_len < 0) {
163 diag("Error sending cmd");
164 goto error;
165 }
166 ret_len = lttng_live_send(control_sock, &connect, sizeof(connect));
167 if (ret_len < 0) {
168 diag("Error sending version");
169 goto error;
170 }
171
172 ret_len = lttng_live_recv(control_sock, &connect, sizeof(connect));
173 if (ret_len == 0) {
174 diag("[error] Remote side has closed connection");
175 goto error;
176 }
177 if (ret_len < 0) {
178 diag("Error receiving version");
179 goto error;
180 }
181 return 0;
182
183 error:
184 return -1;
185 }
186
187 /*
188 * Returns the number of sessions, should be 1 during the unit test.
189 */
190 static
191 int list_sessions(uint64_t *session_id)
192 {
193 struct lttng_viewer_cmd cmd;
194 struct lttng_viewer_list_sessions list;
195 struct lttng_viewer_session lsession;
196 int i;
197 ssize_t ret_len;
198 int first_session = 0;
199
200 cmd.cmd = htobe32(LTTNG_VIEWER_LIST_SESSIONS);
201 cmd.data_size = htobe64(0);
202 cmd.cmd_version = htobe32(0);
203
204 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
205 if (ret_len < 0) {
206 diag("Error sending cmd");
207 goto error;
208 }
209
210 ret_len = lttng_live_recv(control_sock, &list, sizeof(list));
211 if (ret_len == 0) {
212 diag("[error] Remote side has closed connection");
213 goto error;
214 }
215 if (ret_len < 0) {
216 diag("Error receiving session list");
217 goto error;
218 }
219
220 for (i = 0; i < be32toh(list.sessions_count); i++) {
221 ret_len = lttng_live_recv(control_sock, &lsession, sizeof(lsession));
222 if (ret_len < 0) {
223 diag("Error receiving session");
224 goto error;
225 }
226 if (lsession.streams > 0 && first_session <= 0) {
227 first_session = be64toh(lsession.id);
228 *session_id = first_session;
229 }
230 }
231
232 return be32toh(list.sessions_count);
233
234 error:
235 return -1;
236 }
237
238 static
239 int create_viewer_session(void)
240 {
241 struct lttng_viewer_cmd cmd;
242 struct lttng_viewer_create_session_response resp;
243 ssize_t ret_len;
244
245 cmd.cmd = htobe32(LTTNG_VIEWER_CREATE_SESSION);
246 cmd.data_size = htobe64(0);
247 cmd.cmd_version = htobe32(0);
248
249 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
250 if (ret_len < 0) {
251 diag("[error] Error sending cmd");
252 goto error;
253 }
254 LTTNG_ASSERT(ret_len == sizeof(cmd));
255
256 ret_len = lttng_live_recv(control_sock, &resp, sizeof(resp));
257 if (ret_len == 0) {
258 diag("[error] Remote side has closed connection");
259 goto error;
260 }
261 if (ret_len < 0) {
262 diag("[error] Error receiving create session reply");
263 goto error;
264 }
265 LTTNG_ASSERT(ret_len == sizeof(resp));
266
267 if (be32toh(resp.status) != LTTNG_VIEWER_CREATE_SESSION_OK) {
268 diag("[error] Error creating viewer session");
269 goto error;
270 }
271 return 0;
272
273 error:
274 return -1;
275 }
276
277 static
278 int attach_session(uint64_t id)
279 {
280 struct lttng_viewer_cmd cmd;
281 struct lttng_viewer_attach_session_request rq;
282 struct lttng_viewer_attach_session_response rp;
283 struct lttng_viewer_stream stream;
284 int i;
285 ssize_t ret_len;
286
287 session = zmalloc(sizeof(struct live_session));
288 if (!session) {
289 goto error;
290 }
291
292 cmd.cmd = htobe32(LTTNG_VIEWER_ATTACH_SESSION);
293 cmd.data_size = htobe64(sizeof(rq));
294 cmd.cmd_version = htobe32(0);
295
296 memset(&rq, 0, sizeof(rq));
297 rq.session_id = htobe64(id);
298 rq.seek = htobe32(LTTNG_VIEWER_SEEK_BEGINNING);
299
300 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
301 if (ret_len < 0) {
302 diag("Error sending cmd LTTNG_VIEWER_ATTACH_SESSION");
303 goto error;
304 }
305 ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
306 if (ret_len < 0) {
307 diag("Error sending attach request");
308 goto error;
309 }
310
311 ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
312 if (ret_len == 0) {
313 diag("[error] Remote side has closed connection");
314 goto error;
315 }
316 if (ret_len < 0) {
317 diag("Error receiving attach response");
318 goto error;
319 }
320 if (be32toh(rp.status) != LTTNG_VIEWER_ATTACH_OK) {
321 goto error;
322 }
323
324 session->stream_count = be32toh(rp.streams_count);
325 if (session->stream_count == 0) {
326 diag("Got session stream count == 0");
327 goto error;
328 }
329 session->streams = zmalloc(session->stream_count *
330 sizeof(struct viewer_stream));
331 if (!session->streams) {
332 goto error;
333 }
334
335 for (i = 0; i < be32toh(rp.streams_count); i++) {
336 ret_len = lttng_live_recv(control_sock, &stream, sizeof(stream));
337 if (ret_len == 0) {
338 diag("[error] Remote side has closed connection");
339 goto error;
340 }
341 if (ret_len < 0) {
342 diag("Error receiving stream");
343 goto error;
344 }
345 session->streams[i].id = be64toh(stream.id);
346
347 session->streams[i].ctf_trace_id = be64toh(stream.ctf_trace_id);
348 session->streams[i].first_read = 1;
349 session->streams[i].mmap_base = mmap(NULL, mmap_size,
350 PROT_READ | PROT_WRITE,
351 MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
352 if (session->streams[i].mmap_base == MAP_FAILED) {
353 diag("mmap error");
354 goto error;
355 }
356
357 if (be32toh(stream.metadata_flag)) {
358 session->streams[i].metadata_flag = 1;
359 }
360 }
361 return session->stream_count;
362
363 error:
364 return -1;
365 }
366
367 static
368 int get_metadata(void)
369 {
370 struct lttng_viewer_cmd cmd;
371 struct lttng_viewer_get_metadata rq;
372 struct lttng_viewer_metadata_packet rp;
373 ssize_t ret_len;
374 int ret;
375 uint64_t i;
376 char *data = NULL;
377 uint64_t len = 0;
378 int metadata_stream_id = -1;
379
380 cmd.cmd = htobe32(LTTNG_VIEWER_GET_METADATA);
381 cmd.data_size = htobe64(sizeof(rq));
382 cmd.cmd_version = htobe32(0);
383
384 for (i = 0; i < session->stream_count; i++) {
385 if (session->streams[i].metadata_flag) {
386 metadata_stream_id = i;
387 break;
388 }
389 }
390
391 if (metadata_stream_id < 0) {
392 diag("No metadata stream found");
393 goto error;
394 }
395
396 rq.stream_id = htobe64(session->streams[metadata_stream_id].id);
397
398 retry:
399 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
400 if (ret_len < 0) {
401 diag("Error sending cmd");
402 goto error;
403 }
404 ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
405 if (ret_len < 0) {
406 diag("Error sending get_metadata request");
407 goto error;
408 }
409 ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
410 if (ret_len == 0) {
411 diag("[error] Remote side has closed connection");
412 goto error;
413 }
414 if (ret_len < 0) {
415 diag("Error receiving metadata response");
416 goto error;
417 }
418 switch (be32toh(rp.status)) {
419 case LTTNG_VIEWER_METADATA_OK:
420 break;
421 case LTTNG_VIEWER_NO_NEW_METADATA:
422 diag("Got LTTNG_VIEWER_NO_NEW_METADATA:");
423 usleep(50);
424 goto retry;
425 case LTTNG_VIEWER_METADATA_ERR:
426 diag("Got LTTNG_VIEWER_METADATA_ERR:");
427 goto error;
428 default:
429 diag("Got unknown status during LTTNG_VIEWER_GET_METADATA");
430 goto error;
431 }
432
433 len = be64toh(rp.len);
434 if (len <= 0) {
435 goto error;
436 }
437
438 data = zmalloc(len);
439 if (!data) {
440 PERROR("relay data zmalloc");
441 goto error;
442 }
443 ret_len = lttng_live_recv(control_sock, data, len);
444 if (ret_len == 0) {
445 diag("[error] Remote side has closed connection");
446 goto error_free_data;
447 }
448 if (ret_len < 0) {
449 diag("Error receiving trace packet");
450 goto error_free_data;
451 }
452 free(data);
453 ret = len;
454
455 return ret;
456
457 error_free_data:
458 free(data);
459 error:
460 return -1;
461 }
462
463 static
464 int get_next_index(void)
465 {
466 struct lttng_viewer_cmd cmd;
467 struct lttng_viewer_get_next_index rq;
468 struct lttng_viewer_index rp;
469 ssize_t ret_len;
470 int id;
471
472 cmd.cmd = htobe32(LTTNG_VIEWER_GET_NEXT_INDEX);
473 cmd.data_size = htobe64(sizeof(rq));
474 cmd.cmd_version = htobe32(0);
475
476 for (id = 0; id < session->stream_count; id++) {
477 if (session->streams[id].metadata_flag) {
478 continue;
479 }
480 memset(&rq, 0, sizeof(rq));
481 rq.stream_id = htobe64(session->streams[id].id);
482
483 retry:
484 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
485 if (ret_len < 0) {
486 diag("Error sending cmd");
487 goto error;
488 }
489 ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
490 if (ret_len < 0) {
491 diag("Error sending get_next_index request");
492 goto error;
493 }
494 ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
495 if (ret_len == 0) {
496 diag("[error] Remote side has closed connection");
497 goto error;
498 }
499 if (ret_len < 0) {
500 diag("Error receiving index response");
501 goto error;
502 }
503
504 rp.flags = be32toh(rp.flags);
505
506 switch (be32toh(rp.status)) {
507 case LTTNG_VIEWER_INDEX_INACTIVE:
508 /* Skip this stream. */
509 diag("Got LTTNG_VIEWER_INDEX_INACTIVE");
510 continue;
511 case LTTNG_VIEWER_INDEX_OK:
512 break;
513 case LTTNG_VIEWER_INDEX_RETRY:
514 sleep(1);
515 goto retry;
516 case LTTNG_VIEWER_INDEX_HUP:
517 diag("Got LTTNG_VIEWER_INDEX_HUP");
518 session->streams[id].id = -1ULL;
519 session->streams[id].fd = -1;
520 goto error;
521 case LTTNG_VIEWER_INDEX_ERR:
522 diag("Got LTTNG_VIEWER_INDEX_ERR");
523 goto error;
524 default:
525 diag("Unknown reply status during LTTNG_VIEWER_GET_NEXT_INDEX (%d)", be32toh(rp.status));
526 goto error;
527 }
528 if (first_packet_stream_id < 0) {
529 /*
530 * Initialize the first packet stream id. That is,
531 * the first active stream encoutered.
532 */
533 first_packet_offset = be64toh(rp.offset);
534 first_packet_len = be64toh(rp.packet_size) / CHAR_BIT;
535 first_packet_stream_id = id;
536 diag("Got first packet index with offset %d and len %d",
537 first_packet_offset, first_packet_len);
538 }
539 }
540 return 0;
541
542 error:
543 return -1;
544 }
545
546 static
547 int get_data_packet(int id, uint64_t offset,
548 uint64_t len)
549 {
550 struct lttng_viewer_cmd cmd;
551 struct lttng_viewer_get_packet rq;
552 struct lttng_viewer_trace_packet rp;
553 ssize_t ret_len;
554
555 cmd.cmd = htobe32(LTTNG_VIEWER_GET_PACKET);
556 cmd.data_size = htobe64(sizeof(rq));
557 cmd.cmd_version = htobe32(0);
558
559 memset(&rq, 0, sizeof(rq));
560 rq.stream_id = htobe64(session->streams[id].id);
561 /* Already in big endian. */
562 rq.offset = offset;
563 rq.len = htobe32(len);
564
565 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
566 if (ret_len < 0) {
567 diag("Error sending cmd");
568 goto error;
569 }
570 ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
571 if (ret_len < 0) {
572 diag("Error sending get_data_packet request");
573 goto error;
574 }
575 ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
576 if (ret_len == 0) {
577 diag("[error] Remote side has closed connection");
578 goto error;
579 }
580 if (ret_len < 0) {
581 diag("Error receiving data response");
582 goto error;
583 }
584 rp.flags = be32toh(rp.flags);
585
586 switch (be32toh(rp.status)) {
587 case LTTNG_VIEWER_GET_PACKET_OK:
588 len = be32toh(rp.len);
589 if (len == 0) {
590 diag("Got LTTNG_VIEWER_GET_PACKET_OK, but len == 0");
591 goto error;
592 }
593 break;
594 case LTTNG_VIEWER_GET_PACKET_RETRY:
595 diag("Got LTTNG_VIEWER_GET_PACKET_RETRY:");
596 goto error;
597 case LTTNG_VIEWER_GET_PACKET_ERR:
598 if (rp.flags & LTTNG_VIEWER_FLAG_NEW_METADATA) {
599 diag("Got LTTNG_VIEWER_GET_PACKET_ERR with NEW_METADATA flag");
600 goto end;
601 }
602 diag("Got LTTNG_VIEWER_GET_PACKET_ERR:");
603 goto error;
604 default:
605 diag("Got unknown status code during LTTNG_VIEWER_GET_PACKET");
606 goto error;
607 }
608
609 if (len > mmap_size) {
610 diag("mmap_size not big enough");
611 goto error;
612 }
613
614 ret_len = lttng_live_recv(control_sock, session->streams[id].mmap_base, len);
615 if (ret_len == 0) {
616 diag("[error] Remote side has closed connection");
617 goto error;
618 }
619 if (ret_len < 0) {
620 diag("Error receiving trace packet");
621 goto error;
622 }
623 end:
624 return 0;
625 error:
626 return -1;
627 }
628
629 static
630 int detach_viewer_session(uint64_t id)
631 {
632 struct lttng_viewer_cmd cmd;
633 struct lttng_viewer_detach_session_response resp;
634 struct lttng_viewer_detach_session_request rq;
635 int ret;
636 ssize_t ret_len;
637
638 cmd.cmd = htobe32(LTTNG_VIEWER_DETACH_SESSION);
639 cmd.data_size = htobe64(sizeof(rq));
640 cmd.cmd_version = htobe32(0);
641
642 memset(&rq, 0, sizeof(rq));
643 rq.session_id = htobe64(id);
644
645 ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
646 if (ret_len < 0) {
647 fprintf(stderr, "[error] Error sending cmd\n");
648 ret = ret_len;
649 goto error;
650 }
651
652 ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
653 if (ret_len < 0) {
654 fprintf(stderr, "Error sending attach request\n");
655 ret = ret_len;
656 goto error;
657 }
658
659 ret_len = lttng_live_recv(control_sock, &resp, sizeof(resp));
660 if (ret_len < 0) {
661 fprintf(stderr, "[error] Error receiving detach session reply\n");
662 ret = ret_len;
663 goto error;
664 }
665
666 if (be32toh(resp.status) != LTTNG_VIEWER_DETACH_SESSION_OK) {
667 fprintf(stderr, "[error] Error detaching viewer session\n");
668 ret = -1;
669 goto error;
670 }
671 ret = 0;
672
673 error:
674 return ret;
675 }
676
677 int main(int argc, char **argv)
678 {
679 int ret;
680 uint64_t session_id;
681
682 plan_tests(NUM_TESTS);
683
684 diag("Live unit tests");
685
686 ret = connect_viewer("localhost");
687 ok(ret == 0, "Connect viewer to relayd");
688
689 ret = establish_connection();
690 ok(ret == 0, "Established connection and version check with %d.%d",
691 VERSION_MAJOR, VERSION_MINOR);
692
693 ret = list_sessions(&session_id);
694 ok(ret > 0, "List sessions : %d session(s)", ret);
695 if (ret < 0) {
696 goto end;
697 }
698
699 ret = create_viewer_session();
700 ok(ret == 0, "Create viewer session");
701
702 ret = attach_session(session_id);
703 ok(ret > 0, "Attach to session, %d stream(s) received", ret);
704
705 ret = get_metadata();
706 ok(ret > 0, "Get metadata, received %d bytes", ret);
707
708 ret = get_next_index();
709 ok(ret == 0, "Get one index per stream");
710
711 ret = get_data_packet(first_packet_stream_id, first_packet_offset,
712 first_packet_len);
713 ok(ret == 0,
714 "Get one data packet for stream %d, offset %d, len %d",
715 first_packet_stream_id, first_packet_offset,
716 first_packet_len);
717
718 ret = detach_viewer_session(session_id);
719 ok(ret == 0, "Detach viewer session");
720
721 ret = list_sessions(&session_id);
722 ok(ret > 0, "List sessions : %d session(s)", ret);
723
724 ret = attach_session(session_id);
725 ok(ret > 0, "Attach to session, %d streams received", ret);
726 end:
727 return exit_status();
728 }
This page took 0.043426 seconds and 4 git commands to generate.