2 * Copyright (C) 2018 Jérémie Galarneau <jeremie.galarneau@efficios.com>
4 * SPDX-License-Identifier: GPL-2.0-only
14 #include <sys/types.h>
20 #include <sys/types.h>
24 #include <common/compat/directory-handle.h>
25 #include <common/compat/errno.h>
26 #include <common/error.h>
27 #include <common/fd-tracker/fd-tracker.h>
30 int lttng_opt_quiet
= 1;
31 int lttng_opt_verbose
;
34 /* Number of TAP tests in this file */
36 /* 3 for stdin, stdout, and stderr */
37 #define STDIO_FD_COUNT 3
38 #define TRACKER_FD_LIMIT 50
39 #define TMP_DIR_PATTERN "/tmp/fd-tracker-XXXXXX"
40 #define TEST_UNLINK_DIRECTORY_NAME "unlinked_files"
43 #define SELF_FD_DIR "/proc/self/fd"
45 /* Most Unices have /dev/fd */
46 #define SELF_FD_DIR "/dev/fd"
50 * Count of fds, beyond stdin, stderr, stdout that were open
51 * at the launch of the test. This allows the test to succeed when
52 * run by automake's test runner or valgrind which both open
53 * fds behind our back.
55 int unknown_fds_count
;
57 const char file_contents
[] = "Bacon ipsum dolor amet jerky drumstick sirloin "
58 "strip steak venison boudin filet mignon picanha doner shoulder. "
59 "Strip steak brisket alcatra, venison beef chuck cupim pastrami. "
60 "Landjaeger tri-tip salami leberkas ball tip, ham hock chuck sausage "
61 "flank jerky cupim. Pig bacon chuck pancetta andouille.";
64 void get_temporary_directories(char **_test_directory
, char **_unlink_directory
)
67 char tmp_path_pattern
[] = TMP_DIR_PATTERN
;
70 output_dir
= mkdtemp(tmp_path_pattern
);
72 diag("Failed to create temporary path of the form %s",
77 *_test_directory
= strdup(output_dir
);
78 LTTNG_ASSERT(*_test_directory
);
79 ret
= asprintf(_unlink_directory
, "%s/%s", output_dir
,
80 TEST_UNLINK_DIRECTORY_NAME
);
93 dir
= opendir(SELF_FD_DIR
);
95 perror("# Failed to enumerate " SELF_FD_DIR
" to count the number of used file descriptors");
100 while ((entry
= readdir(dir
)) != NULL
) {
101 if (!strcmp(entry
->d_name
, ".") || !strcmp(entry
->d_name
, "..")) {
106 /* Don't account for the file descriptor opened by opendir(). */
109 perror("# Failed to close test program's " SELF_FD_DIR
" directory file descriptor");
116 void check_fd_count(int expected_count
)
121 ok(count
== expected_count
, "Expected %d open file descriptors (%d are open)",
122 expected_count
, count
);
126 int noop_open(void *data
, int *fds
)
128 *fds
= *((int *) data
);
133 int noop_close(void *data
, int *fds
)
139 void track_std_fds(struct fd_tracker
*tracker
)
142 struct { int fd
; const char *name
; } files
[] = {
143 { .fd
= fileno(stdin
), .name
= "stdin" },
144 { .fd
= fileno(stdout
), .name
= "stdout" },
145 { .fd
= fileno(stderr
), .name
= "stderr" },
148 for (i
= 0; i
< sizeof(files
) / sizeof(*files
); i
++) {
151 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
152 &files
[i
].name
, 1, noop_open
, &files
[i
].fd
);
153 LTTNG_ASSERT(out_fd
== files
[i
].fd
);
155 ok(ret
== 0, "Track unsuspendable fd %d (%s)", files
[i
].fd
,
161 void untrack_std_fds(struct fd_tracker
*tracker
)
164 struct { int fd
; const char *name
; } files
[] = {
165 { .fd
= fileno(stdin
), .name
= "stdin" },
166 { .fd
= fileno(stdout
), .name
= "stdout" },
167 { .fd
= fileno(stderr
), .name
= "stderr" },
169 unsigned int fds_set_to_minus_1
= 0;
171 for (i
= 0; i
< sizeof(files
) / sizeof(*files
); i
++) {
172 int fd
= files
[i
].fd
;
173 int ret
= fd_tracker_close_unsuspendable_fd(tracker
,
174 &files
[i
].fd
, 1, noop_close
, NULL
);
176 ok(ret
== 0, "Untrack unsuspendable fd %d (%s)", fd
,
178 fds_set_to_minus_1
+= (files
[i
].fd
== -1);
183 * Basic test opening and closing three unsuspendable fds.
186 void test_unsuspendable_basic(void)
189 struct fd_tracker
*tracker
;
190 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
192 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
194 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
195 ok(tracker
, "Created an fd tracker with a limit of %d simulateously opened file descriptors",
201 track_std_fds(tracker
);
202 untrack_std_fds(tracker
);
204 fd_tracker_destroy(tracker
);
205 ret
= rmdir(test_directory
);
206 ok(ret
== 0, "Test directory is empty");
208 free(test_directory
);
209 free(unlinked_files_directory
);
213 int error_open(void *data
, int *fds
)
215 return *((int *) data
);
219 int error_close(void *data
, int *fds
)
221 return *((int *) data
);
225 * Validate that user callback return values are returned to the
226 * caller of the fd tracker.
229 void test_unsuspendable_cb_return(void)
231 int ret
, stdout_fd
= fileno(stdout
), out_fd
= 42;
232 struct fd_tracker
*tracker
;
233 int expected_error
= -ENETDOWN
;
234 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
236 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
238 tracker
= fd_tracker_create(test_directory
, TRACKER_FD_LIMIT
);
239 LTTNG_ASSERT(tracker
);
241 /* The error_open callback should fail and return 'expected_error'. */
242 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
243 NULL
, 1, error_open
, &expected_error
);
244 ok(ret
== expected_error
, "fd_tracker_open_unsuspendable_fd() forwards the user callback's error code");
245 ok(out_fd
== 42, "Output fd parameter is unaffected on error of fd_tracker_open_unsuspendable_fd()");
248 * Track a valid fd since we don't want the tracker to fail with an
249 * invalid fd error for this test.
251 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
252 NULL
, 1, noop_open
, &stdout_fd
);
253 ok(out_fd
== stdout_fd
, "fd_tracker_open_unsuspendable_fd() sets the output fd parameter to the newly-tracked fd's value");
256 ret
= fd_tracker_close_unsuspendable_fd(tracker
,
257 &stdout_fd
, 1, error_close
, &expected_error
);
258 ok(ret
== expected_error
, "fd_tracker_close_unsuspendable_fd() forwards the user callback's error code");
259 ret
= fd_tracker_close_unsuspendable_fd(tracker
,
260 &stdout_fd
, 1, noop_close
, &expected_error
);
263 fd_tracker_destroy(tracker
);
264 ret
= rmdir(test_directory
);
265 ok(ret
== 0, "Test directory is empty");
266 free(test_directory
);
267 free(unlinked_files_directory
);
271 * Validate that the tracker refuses to track two identical unsuspendable
275 void test_unsuspendable_duplicate(void)
277 int ret
, stdout_fd
= fileno(stdout
), out_fd
;
278 struct fd_tracker
*tracker
;
279 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
281 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
283 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
284 LTTNG_ASSERT(tracker
);
286 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
287 NULL
, 1, noop_open
, &stdout_fd
);
289 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
290 NULL
, 1, noop_open
, &stdout_fd
);
291 ok(ret
== -EEXIST
, "EEXIST reported on open of an already tracked file descriptor");
293 ret
= fd_tracker_close_unsuspendable_fd(tracker
,
294 &stdout_fd
, 1, noop_close
, NULL
);
297 fd_tracker_destroy(tracker
);
298 ret
= rmdir(test_directory
);
299 ok(ret
== 0, "Test directory is empty");
300 free(test_directory
);
301 free(unlinked_files_directory
);
305 int open_pipes(void *data
, int *out_fds
)
308 const unsigned int pipe_count
= TRACKER_FD_LIMIT
/ 2;
310 for (i
= 0; i
< pipe_count
; i
++) {
311 int ret
= pipe(&out_fds
[i
* 2]);
321 int close_pipes(void *data
, int *fds
)
326 for (i
= 0; i
< TRACKER_FD_LIMIT
; i
++) {
327 int ret
= close(pipes
[i
]);
337 * Validate that the tracker enforces the open file descriptor limit
338 * when unsuspendable file descriptors are being opened.
341 void test_unsuspendable_limit(void)
343 struct fd_tracker
*tracker
;
344 int ret
, stdout_fd
= fileno(stdout
), out_fd
;
345 int fds
[TRACKER_FD_LIMIT
];
346 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
348 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
350 /* This test assumes TRACKER_FD_LIMIT is a multiple of 2. */
351 LTTNG_ASSERT((TRACKER_FD_LIMIT
% 2 == 0) && TRACKER_FD_LIMIT
);
353 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
354 LTTNG_ASSERT(tracker
);
356 ret
= fd_tracker_open_unsuspendable_fd(tracker
, fds
,
357 NULL
, TRACKER_FD_LIMIT
, open_pipes
, NULL
);
358 ok(ret
== 0, "File descriptor tracker allowed the user to meet its limit with unsuspendable file descriptors (%d)",
361 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
362 NULL
, 1, noop_open
, &stdout_fd
);
363 ok(ret
== -EMFILE
, "EMFILE reported when exceeding the file descriptor limit while opening an unsuspendable fd");
365 ret
= fd_tracker_close_unsuspendable_fd(tracker
,
366 fds
, TRACKER_FD_LIMIT
, close_pipes
, NULL
);
369 fd_tracker_destroy(tracker
);
370 ret
= rmdir(test_directory
);
371 ok(ret
== 0, "Test directory is empty");
372 free(test_directory
);
373 free(unlinked_files_directory
);
377 * Validate that the tracker refuses to track two identical unsuspendable
381 void test_unsuspendable_close_untracked(void)
383 int ret
, stdout_fd
= fileno(stdout
), unknown_fds
[2], out_fd
;
384 struct fd_tracker
*tracker
;
385 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
387 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
389 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
394 ret
= pipe(unknown_fds
);
396 ret
= close(unknown_fds
[0]);
397 LTTNG_ASSERT(ret
== 0);
398 ret
= close(unknown_fds
[1]);
399 LTTNG_ASSERT(ret
== 0);
401 ret
= fd_tracker_open_unsuspendable_fd(tracker
, &out_fd
,
402 NULL
, 1, noop_open
, &stdout_fd
);
405 ret
= fd_tracker_close_unsuspendable_fd(tracker
,
406 unknown_fds
, 1, noop_close
, NULL
);
407 ok(ret
== -EINVAL
, "EINVAL reported on close of an untracked file descriptor");
409 ret
= fd_tracker_close_unsuspendable_fd(tracker
,
410 &stdout_fd
, 1, noop_close
, NULL
);
413 fd_tracker_destroy(tracker
);
414 ret
= rmdir(test_directory
);
415 ok(ret
== 0, "Test directory is empty");
417 free(test_directory
);
418 free(unlinked_files_directory
);
421 static int open_files(struct fd_tracker
*tracker
,
422 struct lttng_directory_handle
*directory
,
424 struct fs_handle
**handles
,
430 for (i
= 0; i
< count
; i
++) {
433 struct fs_handle
*handle
;
434 mode_t mode
= S_IWUSR
| S_IRUSR
;
436 p_ret
= asprintf(&file_path
, "file-%u", i
);
437 LTTNG_ASSERT(p_ret
>= 0);
438 file_paths
[i
] = file_path
;
440 handle
= fd_tracker_open_fs_handle(tracker
, directory
, file_path
,
441 O_RDWR
| O_CREAT
, &mode
);
451 static int open_same_file(struct fd_tracker
*tracker
,
452 struct lttng_directory_handle
*directory
,
455 struct fs_handle
**handles
)
460 for (i
= 0; i
< count
; i
++) {
461 struct fs_handle
*handle
;
462 mode_t mode
= S_IWUSR
| S_IRUSR
;
464 handle
= fd_tracker_open_fs_handle(tracker
, directory
, file
,
465 O_RDWR
| O_CREAT
, &mode
);
476 int cleanup_files(struct fd_tracker
*tracker
, const char *dir
,
477 unsigned int count
, struct fs_handle
**handles
,
483 for (i
= 0; i
< count
; i
++) {
484 char *file_path
= file_paths
[i
];
489 if (fs_handle_unlink(handles
[i
])) {
490 diag("Failed to unlink fs_handle to file %s", file_path
);
493 if (fs_handle_close(handles
[i
])) {
494 diag("Failed to close fs_handle to file %s", file_path
);
503 void test_suspendable_limit(void)
506 const int files_to_create
= TRACKER_FD_LIMIT
* 10;
507 struct fd_tracker
*tracker
;
508 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
509 char *output_files
[files_to_create
];
510 struct fs_handle
*handles
[files_to_create
];
511 struct lttng_directory_handle
*dir_handle
= NULL
;
512 int dir_handle_fd_count
;
514 memset(output_files
, 0, sizeof(output_files
));
515 memset(handles
, 0, sizeof(handles
));
517 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
519 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
524 dir_handle
= lttng_directory_handle_create(test_directory
);
525 LTTNG_ASSERT(dir_handle
);
526 dir_handle_fd_count
= !!lttng_directory_handle_uses_fd(dir_handle
);
528 ret
= open_files(tracker
, dir_handle
, files_to_create
, handles
,
530 ok(!ret
, "Created %d files with a limit of %d simultaneously-opened file descriptor",
531 files_to_create
, TRACKER_FD_LIMIT
);
532 check_fd_count(TRACKER_FD_LIMIT
+ STDIO_FD_COUNT
+ unknown_fds_count
+
533 dir_handle_fd_count
);
535 ret
= cleanup_files(tracker
, test_directory
, files_to_create
, handles
,
537 ok(!ret
, "Close all opened filesystem handles");
538 ret
= rmdir(test_directory
);
539 ok(ret
== 0, "Test directory is empty");
540 fd_tracker_destroy(tracker
);
541 lttng_directory_handle_put(dir_handle
);
543 free(test_directory
);
544 free(unlinked_files_directory
);
548 void test_mixed_limit(void)
551 const int files_to_create
= TRACKER_FD_LIMIT
;
552 struct fd_tracker
*tracker
;
553 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
554 char *output_files
[files_to_create
];
555 struct fs_handle
*handles
[files_to_create
];
556 struct lttng_directory_handle
*dir_handle
= NULL
;
557 int dir_handle_fd_count
;
559 memset(output_files
, 0, sizeof(output_files
));
560 memset(handles
, 0, sizeof(handles
));
562 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
564 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
569 dir_handle
= lttng_directory_handle_create(test_directory
);
570 LTTNG_ASSERT(dir_handle
);
571 dir_handle_fd_count
= !!lttng_directory_handle_uses_fd(dir_handle
);
573 ret
= open_files(tracker
, dir_handle
, files_to_create
, handles
,
575 ok(!ret
, "Created %d files with a limit of %d simultaneously-opened file descriptor",
576 files_to_create
, TRACKER_FD_LIMIT
);
577 diag("Check file descriptor count after opening %u files", files_to_create
);
578 check_fd_count(TRACKER_FD_LIMIT
+ STDIO_FD_COUNT
+ unknown_fds_count
+
579 dir_handle_fd_count
);
582 * Open unsuspendable fds (stdin, stdout, stderr) and verify that the fd
583 * cap is still respected.
585 diag("Check file descriptor count after adding %d unsuspendable fds",
587 track_std_fds(tracker
);
588 check_fd_count(TRACKER_FD_LIMIT
+ unknown_fds_count
+
589 dir_handle_fd_count
);
590 diag("Untrack unsuspendable file descriptors");
591 untrack_std_fds(tracker
);
592 check_fd_count(TRACKER_FD_LIMIT
+ unknown_fds_count
+
593 dir_handle_fd_count
);
595 ret
= cleanup_files(tracker
, test_directory
, files_to_create
, handles
,
597 ok(!ret
, "Close all opened filesystem handles");
598 ret
= rmdir(test_directory
);
599 ok(ret
== 0, "Test directory is empty");
600 fd_tracker_destroy(tracker
);
601 lttng_directory_handle_put(dir_handle
);
603 free(test_directory
);
604 free(unlinked_files_directory
);
608 * Open more files than allowed by the fd tracker's cap and write,
609 * byte-by-byte, and in round-robin, a string. The goal is to force
610 * the fd tracker to suspend and resume the fs_handles often and
611 * verify that the fd cap is always respected.
613 * The content of the files is also verified at the end.
616 void test_suspendable_restore(void)
619 const int files_to_create
= TRACKER_FD_LIMIT
* 10;
620 struct fd_tracker
*tracker
;
621 char *output_files
[files_to_create
];
622 struct fs_handle
*handles
[files_to_create
];
623 size_t content_index
;
625 bool write_success
= true;
626 bool fd_cap_respected
= true;
627 bool content_ok
= true;
628 struct lttng_directory_handle
*dir_handle
= NULL
;
629 int dir_handle_fd_count
;
630 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
632 memset(output_files
, 0, sizeof(output_files
));
633 memset(handles
, 0, sizeof(handles
));
635 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
637 tracker
= fd_tracker_create(unlinked_files_directory
, TRACKER_FD_LIMIT
);
642 dir_handle
= lttng_directory_handle_create(test_directory
);
643 LTTNG_ASSERT(dir_handle
);
644 dir_handle_fd_count
= !!lttng_directory_handle_uses_fd(dir_handle
);
646 ret
= open_files(tracker
, dir_handle
, files_to_create
, handles
,
648 ok(!ret
, "Created %d files with a limit of %d simultaneously-opened file descriptor",
649 files_to_create
, TRACKER_FD_LIMIT
);
650 diag("Check file descriptor count after opening %u files", files_to_create
);
651 check_fd_count(TRACKER_FD_LIMIT
+ STDIO_FD_COUNT
+ unknown_fds_count
+
652 dir_handle_fd_count
);
654 for (content_index
= 0; content_index
< sizeof(file_contents
); content_index
++) {
655 for (handle_index
= 0; handle_index
< files_to_create
; handle_index
++) {
657 struct fs_handle
*handle
= handles
[handle_index
];
658 const char *path
= output_files
[handle_index
];
660 fd
= fs_handle_get_fd(handle
);
662 write_success
= false;
663 diag("Failed to restore fs_handle to %s",
669 ret
= write(fd
, file_contents
+ content_index
, 1);
670 } while (ret
< 0 && errno
== EINTR
);
673 write_success
= false;
674 PERROR("write() to %s failed", path
);
678 if (fd_count() > (TRACKER_FD_LIMIT
+ STDIO_FD_COUNT
+
680 dir_handle_fd_count
)) {
681 fd_cap_respected
= false;
684 fs_handle_put_fd(handle
);
688 ok(write_success
, "Wrote reference string to %d files",
690 ok(fd_cap_respected
, "FD tracker enforced the file descriptor cap");
692 /* Validate the contents of the files. */
693 for (handle_index
= 0; handle_index
< files_to_create
; handle_index
++) {
695 const char *path
= output_files
[handle_index
];
696 char read_buf
[sizeof(file_contents
)];
698 size_t to_read
= sizeof(read_buf
);
701 fd
= lttng_directory_handle_open_file(
702 dir_handle
, path
, O_RDONLY
, 0);
703 LTTNG_ASSERT(fd
>= 0);
704 ret
= fstat(fd
, &fd_stat
);
706 if (fd_stat
.st_size
!= sizeof(file_contents
)) {
707 diag("Content size of file %s doesn't match, got %" PRId64
", expected %zu",
708 path
, (int64_t) fd_stat
.st_size
,
709 sizeof(file_contents
));
717 ret
= read(fd
, read_pos
, to_read
);
722 } while (to_read
&& (ret
< 0 && errno
== EINTR
));
725 PERROR("Failed to read file %s", path
);
730 if (strcmp(file_contents
, read_buf
)) {
732 diag("File content doesn't match the expectated string");
738 ok(content_ok
, "Files contain the expected content");
739 ret
= cleanup_files(tracker
, test_directory
, files_to_create
, handles
,
741 ok(!ret
, "Close all opened filesystem handles");
742 ret
= rmdir(test_directory
);
743 ok(ret
== 0, "Test directory is empty");
744 fd_tracker_destroy(tracker
);
745 lttng_directory_handle_put(dir_handle
);
747 free(test_directory
);
748 free(unlinked_files_directory
);
752 void test_unlink(void)
755 struct fd_tracker
*tracker
;
756 const int handles_to_open
= 2;
757 struct fs_handle
*handles
[handles_to_open
];
758 struct fs_handle
*new_handle
= NULL
;
760 struct lttng_directory_handle
*dir_handle
= NULL
;
761 const char file_name
[] = "my_file";
762 char *test_directory
= NULL
, *unlinked_files_directory
= NULL
;
763 char *unlinked_file_zero
= NULL
, *unlinked_file_one
= NULL
;
766 get_temporary_directories(&test_directory
, &unlinked_files_directory
);
767 ret
= asprintf(&unlinked_file_zero
, "%s/%u", unlinked_files_directory
,
769 LTTNG_ASSERT(ret
> 0);
770 ret
= asprintf(&unlinked_file_one
, "%s/%u", unlinked_files_directory
,
772 LTTNG_ASSERT(ret
> 0);
774 tracker
= fd_tracker_create(unlinked_files_directory
, 1);
779 dir_handle
= lttng_directory_handle_create(test_directory
);
780 LTTNG_ASSERT(dir_handle
);
782 /* Open two handles to the same file. */
783 ret
= open_same_file(tracker
, dir_handle
, file_name
, handles_to_open
,
785 ok(!ret
, "Successfully opened %i handles to %s/%s", handles_to_open
,
786 test_directory
, file_name
);
792 * Unlinking the first handle should cause the file to be renamed
795 ret
= fs_handle_unlink(handles
[0]);
796 ok(!ret
, "Successfully unlinked the first handle to %s/%s",
797 test_directory
, file_name
);
800 * The original file should no longer exist on the file system, and a
801 * new file named '0' should exist.
803 ok(lttng_directory_handle_stat(dir_handle
, file_name
, &statbuf
) == -1 &&
805 "%s no longer present on file system after unlink",
807 ok(lttng_directory_handle_stat(
808 dir_handle
, unlinked_file_zero
, &statbuf
) == 0,
809 "%s exists on file system after unlink",
813 * It should be possible to use the file descriptors of both handles.
814 * Since only one file descriptor can be opened at once, this should
815 * force the fd_tracker to suspend and restore the handles.
817 fd
= fs_handle_get_fd(handles
[0]);
818 ok(fd
>= 0, "Got fd from first handle");
820 fd
= fs_handle_get_fd(handles
[1]);
821 ok (fd
< 0, "fd tracker does not allow two fds to be used at once");
823 fs_handle_put_fd(handles
[0]);
824 fd
= fs_handle_get_fd(handles
[1]);
825 ok(fd
>= 0, "Got fd from second handle");
826 fs_handle_put_fd(handles
[1]);
828 /* The second unlink should fail with -ENOENT. */
829 ret
= fs_handle_unlink(handles
[1]);
831 "ENOENT is reported when attempting to unlink the second handle to %s/%s",
832 test_directory
, file_name
);
835 * Opening a new handle to 'my_file' should succeed.
837 ret
= open_same_file(tracker
, dir_handle
, file_name
, 1, &new_handle
);
838 ok(!ret
, "Successfully opened a new handle to previously unlinked file %s/%s",
839 test_directory
, file_name
);
840 LTTNG_ASSERT(new_handle
);
843 * Unlinking the new handle should cause the file to be renamed
844 * to '1' since '0' already exists.
846 ret
= fs_handle_unlink(new_handle
);
847 ok(!ret
, "Successfully unlinked the new handle handle to %s/%s",
848 test_directory
, file_name
);
849 ok(stat(unlinked_file_one
, &statbuf
) == 0,
850 "%s exists on file system after unlink",
853 ret
= fs_handle_close(handles
[0]);
854 ok(!ret
, "Successfully closed the first handle");
855 ret
= fs_handle_close(handles
[1]);
856 ok(!ret
, "Successfully closed the second handle");
857 ret
= fs_handle_close(new_handle
);
858 ok(!ret
, "Successfully closed the third handle");
860 ok(lttng_directory_handle_stat(dir_handle
, file_name
, &statbuf
) == -1 &&
862 "%s no longer present on file system after handle close",
864 ok(lttng_directory_handle_stat(
865 dir_handle
, unlinked_file_zero
, &statbuf
) == -1 &&
867 "%s no longer present on file system after handle close",
869 ok(lttng_directory_handle_stat(dir_handle
, unlinked_file_one
,
872 "%s no longer present on file system after handle close",
875 ret
= rmdir(test_directory
);
876 ok(ret
== 0, "Test directory is empty");
878 fd_tracker_destroy(tracker
);
879 free(test_directory
);
880 free(unlinked_files_directory
);
881 free(unlinked_file_zero
);
882 free(unlinked_file_one
);
883 lttng_directory_handle_put(dir_handle
);
886 int main(int argc
, char **argv
)
888 plan_tests(NUM_TESTS
);
889 diag("File descriptor tracker unit tests");
891 rcu_register_thread();
893 unknown_fds_count
= fd_count() - STDIO_FD_COUNT
;
894 LTTNG_ASSERT(unknown_fds_count
>= 0);
896 diag("Unsuspendable - basic");
897 test_unsuspendable_basic();
898 diag("Unsuspendable - callback return values");
899 test_unsuspendable_cb_return();
900 diag("Unsuspendable - duplicate file descriptors");
901 test_unsuspendable_duplicate();
902 diag("Unsuspendable - closing an untracked file descriptor");
903 test_unsuspendable_close_untracked();
904 diag("Unsuspendable - check that file descriptor limit is enforced");
905 test_unsuspendable_limit();
907 diag("Suspendable - check that file descriptor limit is enforced");
908 test_suspendable_limit();
909 diag("Suspendable - restoration test");
910 test_suspendable_restore();
912 diag("Mixed - check that file descriptor limit is enforced");
915 diag("Suspendable - Unlinking test");
919 rcu_unregister_thread();
920 return exit_status();