fortify-headers

standalone fortify-source implementation
git clone git://git.2f30.org/fortify-headers
Log | Files | Refs | README | LICENSE

poll.h (1891B)


      1 /*
      2  * Copyright (C) 2015-2016 Dimitris Papastamos <sin@2f30.org>
      3  * Copyright (C) 2022 q66 <q66@chimera-linux.org>
      4  *
      5  * Permission to use, copy, modify, and/or distribute this software for any
      6  * purpose with or without fee is hereby granted.
      7  *
      8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
      9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
     10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
     11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
     12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
     13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
     14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
     15  */
     16 
     17 #ifndef _FORTIFY_POLL_H
     18 #define _FORTIFY_POLL_H
     19 
     20 #if !defined(__cplusplus) && !defined(__clang__)
     21 __extension__
     22 #endif
     23 #include_next <poll.h>
     24 
     25 #if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0
     26 #include "fortify-headers.h"
     27 
     28 #ifdef __cplusplus
     29 extern "C" {
     30 #endif
     31 
     32 #undef poll
     33 
     34 #if __has_builtin(__builtin_poll)
     35 __diagnose_as_builtin(__builtin_poll, 1, 2, 3)
     36 #endif
     37 _FORTIFY_FN(poll) int poll(struct pollfd * _FORTIFY_POS0 __f, nfds_t __n, int __s)
     38 {
     39 	__fh_size_t __b = __fh_bos(__f, 0);
     40 
     41 	if (__n > __b / sizeof(struct pollfd))
     42 		__builtin_trap();
     43 	return __orig_poll(__f, __n, __s);
     44 }
     45 
     46 #if defined(_GNU_SOURCE) && !_REDIR_TIME64
     47 #undef ppoll
     48 #if __has_builtin(__builtin_ppoll)
     49 __diagnose_as_builtin(__builtin_ppoll, 1, 2, 3, 4)
     50 #endif
     51 _FORTIFY_FN(ppoll) int ppoll(struct pollfd * _FORTIFY_POS0 __f, nfds_t __n,
     52                              const struct timespec *__s, const sigset_t *__m)
     53 {
     54 	__fh_size_t __b = __fh_bos(__f, 0);
     55 
     56 	if (__n > __b / sizeof(struct pollfd))
     57 		__builtin_trap();
     58 	return __orig_ppoll(__f, __n, __s, __m);
     59 }
     60 #endif
     61 
     62 #ifdef __cplusplus
     63 }
     64 #endif
     65 
     66 #endif
     67 
     68 #endif