Signed-off-by: Bryce Harrington <[email protected]>
---
tests/os-wrappers-test.c | 45 +++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 45 insertions(+)
diff --git a/tests/os-wrappers-test.c b/tests/os-wrappers-test.c
index b636b62..b838b93 100644
--- a/tests/os-wrappers-test.c
+++ b/tests/os-wrappers-test.c
@@ -162,6 +162,51 @@ TEST(os_wrappers_socket_cloexec_fallback)
do_os_wrappers_socket_cloexec(1);
}
+
+static void
+do_os_wrappers_set_cloexec_or_close(int n)
+{
+ int fd;
+ int nr_fds;
+ long flags;
+
+ nr_fds = count_open_fds();
+
+ /* Create a basic socket */
+ fd = socket(PF_LOCAL, SOCK_STREAM, 0);
+ assert(fd > 0);
+
+ /* Verify we can set it to close on exec */
+ assert(wl_os_set_cloexec_or_close(fd) == fd);
+
+ /* Verify the fd is now cloexec */
+ flags = fcntl(fd, F_GETFD);
+ assert(flags & FD_CLOEXEC);
+
+ /*
+ * Must have 2 calls if falling back, but must also allow
+ * falling back without a forced fallback.
+ */
+ assert(wrapped_calls_fcntl > n);
+
+ exec_fd_leak_check(nr_fds);
+}
+
+TEST(os_wrappers_set_cloexec_or_close)
+{
+ /* normal case */
+ init_fallbacks(0);
+ do_os_wrappers_set_cloexec_or_close(0);
+}
+
+TEST(os_wrappers_set_cloexec_or_close_fallback)
+{
+ /* forced fallback */
+ init_fallbacks(1);
+ do_os_wrappers_set_cloexec_or_close(1);
+}
+
+
static void
do_os_wrappers_dupfd_cloexec(int n)
{
--
1.9.1
_______________________________________________
wayland-devel mailing list
[email protected]
http://lists.freedesktop.org/mailman/listinfo/wayland-devel
