fortify-headers/include/poll.h

#ifndef _FORTIFY_POLL_H
#define _FORTIFY_POLL_H

#include_next <poll.h>

#if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0

#ifdef __cplusplus
extern "C" {
#endif

#undef poll

__typeof__(poll) __poll_orig __asm__(__USER_LABEL_PREFIX__ "poll");
extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
int poll(struct pollfd *fds, nfds_t nfds, int timeout)
{
	__typeof__(sizeof 0) bos = __builtin_object_size(fds, 0);

	if (nfds > bos / sizeof(struct pollfd))
		__builtin_trap();
	return __poll_orig(fds, nfds, timeout);
}

#ifdef _GNU_SOURCE
#undef ppoll
__typeof__(ppoll) __ppoll_orig __asm__(__USER_LABEL_PREFIX__ "ppoll");
extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
int 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_orig(fds, nfds, timeout, mask);
}
#endif

#ifdef __cplusplus
}
#endif

#endif

#endif
Put include guards in the reserved namespace fortify-headers is considered part of the implementation. 2015-03-11 18:55:51 +00:00			`#ifndef _FORTIFY_POLL_H`
			`#define _FORTIFY_POLL_H`
Add poll() check Not sure if including stddef.h from poll.h is acceptable. 2015-02-28 12:10:41 +00:00
			`#include_next <poll.h>`

			`#if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0`

Restore C++ support 2015-03-13 23:09:15 +00:00			`#ifdef __cplusplus`
			`extern "C" {`
			`#endif`

Rework fortify implementation to use extern inline Overriding functions with macros is legal in C but a lot of software is not prepared for it. Use the extern inline method to achieve the same result. 2015-03-13 11:00:46 +00:00			`#undef poll`
Add ifdef guards for C++ code 2015-03-11 09:26:11 +00:00
Use __typeof__ to in part avoid replicating function types 2015-03-14 19:06:48 +00:00			`__typeof__(poll) __poll_orig __asm__(__USER_LABEL_PREFIX__ "poll");`
Add __artificial__ to aid in debugging 2015-03-14 09:38:11 +00:00			`extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))`
Rework fortify implementation to use extern inline Overriding functions with macros is legal in C but a lot of software is not prepared for it. Use the extern inline method to achieve the same result. 2015-03-13 11:00:46 +00:00			`int poll(struct pollfd *fds, nfds_t nfds, int timeout)`
Add poll() check Not sure if including stddef.h from poll.h is acceptable. 2015-02-28 12:10:41 +00:00			`{`
Use __typeof__(sizeof 0) trickery We cannot pull stddef.h and since fortify already relies on GCC features we can use the above typeof trick to get a usable size_t. 2015-02-28 16:01:26 +00:00			`__typeof__(sizeof 0) bos = __builtin_object_size(fds, 0);`
Add poll() check Not sure if including stddef.h from poll.h is acceptable. 2015-02-28 12:10:41 +00:00
Remove redundant checks We can never have an array of more than SIZE_MAX/2/sizeof(gid_t) gid_t's. 2015-02-28 20:49:00 +00:00			`if (nfds > bos / sizeof(struct pollfd))`
Add poll() check Not sure if including stddef.h from poll.h is acceptable. 2015-02-28 12:10:41 +00:00			`__builtin_trap();`
Rework fortify implementation to use extern inline Overriding functions with macros is legal in C but a lot of software is not prepared for it. Use the extern inline method to achieve the same result. 2015-03-13 11:00:46 +00:00			`return __poll_orig(fds, nfds, timeout);`
Add poll() check Not sure if including stddef.h from poll.h is acceptable. 2015-02-28 12:10:41 +00:00			`}`

Add ppoll() check 2015-03-03 14:28:25 +00:00			`#ifdef _GNU_SOURCE`
Rework fortify implementation to use extern inline Overriding functions with macros is legal in C but a lot of software is not prepared for it. Use the extern inline method to achieve the same result. 2015-03-13 11:00:46 +00:00			`#undef ppoll`
Use __typeof__ to in part avoid replicating function types 2015-03-14 19:06:48 +00:00			`__typeof__(ppoll) __ppoll_orig __asm__(__USER_LABEL_PREFIX__ "ppoll");`
Add __artificial__ to aid in debugging 2015-03-14 09:38:11 +00:00			`extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))`
Rework fortify implementation to use extern inline Overriding functions with macros is legal in C but a lot of software is not prepared for it. Use the extern inline method to achieve the same result. 2015-03-13 11:00:46 +00:00			`int ppoll(struct pollfd fds, nfds_t nfds, const struct timespec timeout, const sigset_t *mask)`
Add ppoll() check 2015-03-03 14:28:25 +00:00			`{`
			`__typeof__(sizeof 0) bos = __builtin_object_size(fds, 0);`

			`if (nfds > bos / sizeof(struct pollfd))`
			`__builtin_trap();`
Rework fortify implementation to use extern inline Overriding functions with macros is legal in C but a lot of software is not prepared for it. Use the extern inline method to achieve the same result. 2015-03-13 11:00:46 +00:00			`return __ppoll_orig(fds, nfds, timeout, mask);`
Add ppoll() check 2015-03-03 14:28:25 +00:00			`}`
			`#endif`

Restore C++ support 2015-03-13 23:09:15 +00:00			`#ifdef __cplusplus`
			`}`
Add ifdef guards for C++ code 2015-03-11 09:26:11 +00:00			`#endif`

Add poll() check Not sure if including stddef.h from poll.h is acceptable. 2015-02-28 12:10:41 +00:00			`#endif`

			`#endif`