Add poll() check

Not sure if including stddef.h from poll.h is acceptable.

1 files changed, 25 insertions, 0 deletions

diff --git a/include/poll.h b/include/poll.h
new file mode 100644
index 0000000..54fc4ab
--- /dev/null
+++ b/include/poll.h
@@ -0,0 +1,25 @@
+#ifndef FORTIFY_POLL_H_
+#define FORTIFY_POLL_H_
+
+#include_next <stddef.h>
+#include_next <poll.h>
+
+#if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0
+
+static inline __attribute__ ((always_inline))
+int
+__fortify_poll(struct pollfd *fds, nfds_t nfds, int timeout)
+{
+        size_t bos = __builtin_object_size(fds, 0);
+
+        if (bos != -1 && nfds > bos / sizeof(struct pollfd))
+                __builtin_trap();
+        return poll(fds, nfds, timeout);
+}
+
+#undef poll
+#define poll(fds, nfds, timeout) __fortify_poll(fds, nfds, timeout)
+
+#endif
+
+#endif