Loading...
--- xnu/xnu-12377.121.6/tests/poll.c
+++ /dev/null
@@ -1,184 +0,0 @@
-#ifdef T_NAMESPACE
-#undef T_NAMESPACE
-#endif
-#include <darwintest.h>
-
-#include <dispatch/dispatch.h>
-#include <fcntl.h>
-#include <mach/mach.h>
-#include <poll.h>
-#include <stdint.h>
-#include <unistd.h>
-#include <libproc.h>
-
-T_GLOBAL_META(T_META_NAMESPACE("xnu.poll"),
- T_META_RUN_CONCURRENTLY(true));
-
-#define SLEEP_TIME_SECS 1
-#define POLL_TIMEOUT_MS 1800
-static_assert(POLL_TIMEOUT_MS > (SLEEP_TIME_SECS * 1000),
- "poll timeout should be longer than sleep time");
-
-/*
- * This matches the behavior of other UNIXes, but is under-specified in POSIX.
- *
- * See <rdar://problem/28372390>.
- */
-T_DECL(sleep_with_no_fds,
- "poll() called with no fds provided should act like sleep", T_META_TAG_VM_PREFERRED)
-{
- uint64_t begin_time, sleep_time, poll_time;
- struct pollfd pfd = { .fd = 0, .events = 0, .revents = 0 };
-
- begin_time = mach_absolute_time();
- sleep(SLEEP_TIME_SECS);
- sleep_time = mach_absolute_time() - begin_time;
- T_LOG("sleep(%d) ~= %llu mach absolute time units", SLEEP_TIME_SECS, sleep_time);
-
- begin_time = mach_absolute_time();
- T_ASSERT_POSIX_SUCCESS(poll(&pfd, 0, POLL_TIMEOUT_MS),
- "poll() with 0 events and timeout %d ms", POLL_TIMEOUT_MS);
- poll_time = mach_absolute_time() - begin_time;
-
- T_EXPECT_GT(poll_time, sleep_time,
- "poll(... %d) should wait longer than sleep(1)", POLL_TIMEOUT_MS);
-}
-
-#define LAUNCHD_PATH "/sbin/launchd"
-#define PIPE_DIR_TIMEOUT_SECS 1
-
-/*
- * See <rdar://problem/28539155>.
- */
-T_DECL(directories,
- "poll() with directories should return an error", T_META_TAG_VM_PREFERRED)
-{
- int file, dir, pipes[2];
- struct pollfd pfd[] = {
- { .events = POLLIN },
- { .events = POLLIN },
- { .events = POLLIN },
- };
-
- file = open(LAUNCHD_PATH, O_RDONLY | O_NONBLOCK);
- T_QUIET; T_ASSERT_POSIX_SUCCESS(file, "open(%s)", LAUNCHD_PATH);
- dir = open(".", O_RDONLY | O_NONBLOCK);
- T_QUIET; T_ASSERT_POSIX_SUCCESS(dir, "open(\".\")");
- T_QUIET; T_ASSERT_POSIX_SUCCESS(pipe(pipes), NULL);
-
- /* just directory */
- pfd[0].fd = dir;
- T_EXPECT_POSIX_SUCCESS(poll(pfd, 1, -1), "poll() with a directory");
- T_QUIET; T_EXPECT_TRUE(pfd[0].revents & POLLNVAL,
- "directory should be an invalid event");
-
- /* file and directory */
- pfd[0].fd = file; pfd[0].revents = 0;
- pfd[1].fd = dir; pfd[1].revents = 0;
- T_EXPECT_POSIX_SUCCESS(poll(pfd, 2, -1),
- "poll() with a file and directory");
- T_QUIET; T_EXPECT_TRUE(pfd[0].revents & POLLIN, "file should be readable");
- T_QUIET; T_EXPECT_TRUE(pfd[1].revents & POLLNVAL,
- "directory should be an invalid event");
-
- /* directory and file */
- pfd[0].fd = dir; pfd[0].revents = 0;
- pfd[1].fd = file; pfd[1].revents = 0;
- T_EXPECT_POSIX_SUCCESS(poll(pfd, 2, -1),
- "poll() with a directory and a file");
- T_QUIET; T_EXPECT_TRUE(pfd[0].revents & POLLNVAL,
- "directory should be an invalid event");
- T_QUIET; T_EXPECT_TRUE(pfd[1].revents & POLLIN, "file should be readable");
-
- /* file and pipe */
- pfd[0].fd = file; pfd[0].revents = 0;
- pfd[1].fd = pipes[0]; pfd[0].revents = 0;
- T_EXPECT_POSIX_SUCCESS(poll(pfd, 2, -1),
- "poll() with a file and pipe");
- T_QUIET; T_EXPECT_TRUE(pfd[0].revents & POLLIN, "file should be readable");
- T_QUIET; T_EXPECT_FALSE(pfd[1].revents & POLLIN,
- "pipe should not be readable");
-
- /* file, directory, and pipe */
- pfd[0].fd = file; pfd[0].revents = 0;
- pfd[1].fd = dir; pfd[1].revents = 0;
- pfd[2].fd = pipes[0]; pfd[2].revents = 0;
- T_EXPECT_POSIX_SUCCESS(poll(pfd, 3, -1),
- "poll() with a file, directory, and pipe");
- T_QUIET; T_EXPECT_TRUE(pfd[0].revents & POLLIN, "file should be readable");
- T_QUIET; T_EXPECT_TRUE(pfd[1].revents & POLLNVAL,
- "directory should be an invalid event");
- T_QUIET; T_EXPECT_FALSE(pfd[2].revents & POLLIN, "pipe should not be readable");
-
- /* directory and pipe */
- __block bool timed_out = true;
- pfd[0].fd = dir; pfd[0].revents = 0;
- pfd[1].fd = pipes[0]; pfd[1].revents = 0;
- dispatch_after(dispatch_time(DISPATCH_TIME_NOW,
- PIPE_DIR_TIMEOUT_SECS * NSEC_PER_SEC),
- dispatch_get_global_queue(QOS_CLASS_DEFAULT, 0), ^{
- T_ASSERT_FALSE(timed_out, "poll timed out after %d seconds",
- PIPE_DIR_TIMEOUT_SECS);
- });
-
- T_EXPECT_POSIX_SUCCESS(poll(pfd, 3, -1),
- "poll() with a directory and pipe");
- timed_out = false;
-
- T_QUIET; T_EXPECT_TRUE(pfd[0].revents & POLLNVAL,
- "directory should be an invalid event");
- T_QUIET; T_EXPECT_FALSE(pfd[1].revents & POLLIN, "pipe should not be readable");
-}
-
-#define PRIVATE
-#include <libproc.h>
-
-static void *
-leak_thread(void *ptr)
-{
- T_LOG("Trying to find kevent kernel pointer...\n");
-
- unsigned char *buffer = (unsigned char*) malloc(16392 * 8);
-
- while (1) {
- memset(buffer, 0, 16392 * 8);
-
- // Dump the kevent udatas for self
- // PT: Note that this is exposed by libproc but isn't declared in the header,
- // hence the forward declaration here. It seems fine to expose this in the header,
- // but I'm unsure of the implications.
- int proc_list_uptrs(int pid, uint64_t *buf, uint32_t bufsz);
- int ret = proc_list_uptrs(getpid(), buffer, 16392 * 8);
-
- if (ret > 0) {
- T_LOG("udata pointers returned: %d\n", ret);
- uint64_t *ptrs = (uint64_t*) buffer;
- for (int i = 0; i < ret; i++) {
- T_EXPECT_EQ(ptrs[i] & 0xffffff0000000000, 0, "kptr? -> 0x%llx\n", ptrs[i]);
- }
- break;
- }
- }
-
- free(buffer);
- return NULL;
-}
-
-T_DECL(poll_dont_leak_kernel_pointers, "poll and proc_pidinfo should not leak kernel pointers", T_META_TAG_VM_PREFERRED)
-{
- pthread_t thr;
- pthread_create(&thr, NULL, *leak_thread, (void *) NULL);
-
- struct pollfd fds[2] = {};
- fds[0].fd = STDERR_FILENO;
- fds[0].events = POLLERR | POLLHUP;
- fds[0].revents = POLLERR;
- fds[1].fd = STDOUT_FILENO;
- fds[1].events = POLLERR | POLLHUP;
- fds[1].revents = POLLERR;
-
- //int poll_nocancel(struct pollfd *fds, u_int nfds, int timeout)
- poll(fds, 2, 5000);
-
- pthread_join(thr, NULL);
-}