Add ppoll() check

1 files changed, 19 insertions, 0 deletions

diff --git a/include/poll.h b/include/poll.h
index 4eb2713..8810bca 100644
--- a/include/poll.h
+++ b/include/poll.h
@@ -16,9 +16,28 @@ __fortify_poll(struct pollfd *fds, nfds_t nfds, int timeout)
         return poll(fds, nfds, timeout);
 }
 
+#ifdef _GNU_SOURCE
+int
+__fortify_ppoll(struct pollfd *fds, nfds_t nfds,
+                const struct timespec *timeout, const sigset_t *mask)
+{
+        __typeof__(sizeof 0) bos = __builtin_object_size(fds, 0);
+
+        if (nfds > bos / sizeof(struct pollfd))
+                __builtin_trap();
+        return ppoll(fds, nfds, timeout, mask);
+}
+
+#endif
+
 #undef poll
 #define poll(fds, nfds, timeout) __fortify_poll(fds, nfds, timeout)
 
+#ifdef _GNU_SOURCE
+#undef ppoll
+#define ppoll(fds, nfds, timeout, mask) __fortify_ppoll(fds, nfds, timeout, mask)
+#endif
+
 #endif
 
 #endif