1 /* Checking macros for poll functions.
2    Copyright (C) 2012-2021 Free Software Foundation, Inc.
3    This file is part of the GNU C Library.
4 
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9 
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14 
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, see
17    <https://www.gnu.org/licenses/>.  */
18 
19 #ifndef _SYS_POLL_H
20 # error "Never include <bits/poll2.h> directly; use <sys/poll.h> instead."
21 #endif
22 
23 
24 __BEGIN_DECLS
25 
26 extern int __REDIRECT (__poll_alias, (struct pollfd *__fds, nfds_t __nfds,
27 				      int __timeout), poll);
28 extern int __poll_chk (struct pollfd *__fds, nfds_t __nfds, int __timeout,
29 		       __SIZE_TYPE__ __fdslen)
30     __attr_access ((__write_only__, 1, 2));
31 extern int __REDIRECT (__poll_chk_warn, (struct pollfd *__fds, nfds_t __nfds,
32 					 int __timeout, __SIZE_TYPE__ __fdslen),
33 		       __poll_chk)
34   __warnattr ("poll called with fds buffer too small file nfds entries");
35 
36 __fortify_function __attr_access ((__write_only__, 1, 2)) int
poll(struct pollfd * __fds,nfds_t __nfds,int __timeout)37 poll (struct pollfd *__fds, nfds_t __nfds, int __timeout)
38 {
39   if (__glibc_objsize (__fds) != (__SIZE_TYPE__) -1)
40     {
41       if (! __builtin_constant_p (__nfds))
42 	return __poll_chk (__fds, __nfds, __timeout, __glibc_objsize (__fds));
43       else if (__glibc_objsize (__fds) / sizeof (*__fds) < __nfds)
44 	return __poll_chk_warn (__fds, __nfds, __timeout,
45 				__glibc_objsize (__fds));
46     }
47 
48   return __poll_alias (__fds, __nfds, __timeout);
49 }
50 
51 
52 #ifdef __USE_GNU
53 extern int __REDIRECT (__ppoll_alias, (struct pollfd *__fds, nfds_t __nfds,
54 				       const struct timespec *__timeout,
55 				       const __sigset_t *__ss), ppoll);
56 extern int __ppoll_chk (struct pollfd *__fds, nfds_t __nfds,
57 			const struct timespec *__timeout,
58 			const __sigset_t *__ss, __SIZE_TYPE__ __fdslen)
59     __attr_access ((__write_only__, 1, 2));
60 extern int __REDIRECT (__ppoll_chk_warn, (struct pollfd *__fds, nfds_t __nfds,
61 					  const struct timespec *__timeout,
62 					  const __sigset_t *__ss,
63 					  __SIZE_TYPE__ __fdslen),
64 		       __ppoll_chk)
65   __warnattr ("ppoll called with fds buffer too small file nfds entries");
66 
67 __fortify_function __attr_access ((__write_only__, 1, 2)) int
ppoll(struct pollfd * __fds,nfds_t __nfds,const struct timespec * __timeout,const __sigset_t * __ss)68 ppoll (struct pollfd *__fds, nfds_t __nfds, const struct timespec *__timeout,
69        const __sigset_t *__ss)
70 {
71   if (__glibc_objsize (__fds) != (__SIZE_TYPE__) -1)
72     {
73       if (! __builtin_constant_p (__nfds))
74 	return __ppoll_chk (__fds, __nfds, __timeout, __ss,
75 			    __glibc_objsize (__fds));
76       else if (__glibc_objsize (__fds) / sizeof (*__fds) < __nfds)
77 	return __ppoll_chk_warn (__fds, __nfds, __timeout, __ss,
78 				 __glibc_objsize (__fds));
79     }
80 
81   return __ppoll_alias (__fds, __nfds, __timeout, __ss);
82 }
83 #endif
84 
85 __END_DECLS