1 /*
2 * Copyright (C) 2017 The Android Open Source Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #ifndef _POLL_H_
30 #error "Never include this file directly; instead, include <poll.h>"
31 #endif
32
33 int __poll_chk(struct pollfd*, nfds_t, int, size_t) __INTRODUCED_IN(23);
34 int __ppoll_chk(struct pollfd*, nfds_t, const struct timespec*, const sigset_t*, size_t)
35 __INTRODUCED_IN(23);
36
37 #if defined(__BIONIC_FORTIFY)
38 #if __ANDROID_API__ >= __ANDROID_API_M__
39 #if defined(__clang__)
40 __BIONIC_FORTIFY_INLINE
poll(struct pollfd * const fds __pass_object_size,nfds_t fd_count,int timeout)41 int poll(struct pollfd* const fds __pass_object_size, nfds_t fd_count, int timeout)
42 __overloadable
43 __clang_error_if(__bos(fds) != __BIONIC_FORTIFY_UNKNOWN_SIZE &&
44 __bos(fds) < sizeof(*fds) * fd_count,
45 "in call to 'poll', fd_count is larger than the given buffer") {
46 size_t bos_fds = __bos(fds);
47
48 if (bos_fds == __BIONIC_FORTIFY_UNKNOWN_SIZE) {
49 return __call_bypassing_fortify(poll)(fds, fd_count, timeout);
50 }
51 return __poll_chk(fds, fd_count, timeout, bos_fds);
52 }
53
54 __BIONIC_FORTIFY_INLINE
ppoll(struct pollfd * const fds __pass_object_size,nfds_t fd_count,const struct timespec * timeout,const sigset_t * mask)55 int ppoll(struct pollfd* const fds __pass_object_size, nfds_t fd_count, const struct timespec* timeout, const sigset_t* mask)
56 __overloadable
57 __clang_error_if(__bos(fds) != __BIONIC_FORTIFY_UNKNOWN_SIZE &&
58 __bos(fds) < sizeof(*fds) * fd_count,
59 "in call to 'ppoll', fd_count is larger than the given buffer") {
60 size_t bos_fds = __bos(fds);
61
62 if (bos_fds == __BIONIC_FORTIFY_UNKNOWN_SIZE) {
63 return __call_bypassing_fortify(ppoll)(fds, fd_count, timeout, mask);
64 }
65 return __ppoll_chk(fds, fd_count, timeout, mask, bos_fds);
66 }
67 #else /* defined(__clang__) */
68 int __poll_real(struct pollfd*, nfds_t, int) __RENAME(poll);
69 __errordecl(__poll_too_small_error, "poll: pollfd array smaller than fd count");
70
71 int __ppoll_real(struct pollfd*, nfds_t, const struct timespec*, const sigset_t*) __RENAME(ppoll)
72 __INTRODUCED_IN(21);
73 __errordecl(__ppoll_too_small_error, "ppoll: pollfd array smaller than fd count");
74
75 __BIONIC_FORTIFY_INLINE
poll(struct pollfd * fds,nfds_t fd_count,int timeout)76 int poll(struct pollfd* fds, nfds_t fd_count, int timeout) {
77 if (__bos(fds) != __BIONIC_FORTIFY_UNKNOWN_SIZE) {
78 if (!__builtin_constant_p(fd_count)) {
79 return __poll_chk(fds, fd_count, timeout, __bos(fds));
80 } else if (__bos(fds) / sizeof(*fds) < fd_count) {
81 __poll_too_small_error();
82 }
83 }
84 return __poll_real(fds, fd_count, timeout);
85 }
86
87 __BIONIC_FORTIFY_INLINE
ppoll(struct pollfd * fds,nfds_t fd_count,const struct timespec * timeout,const sigset_t * mask)88 int ppoll(struct pollfd* fds, nfds_t fd_count, const struct timespec* timeout,
89 const sigset_t* mask) {
90 if (__bos(fds) != __BIONIC_FORTIFY_UNKNOWN_SIZE) {
91 if (!__builtin_constant_p(fd_count)) {
92 return __ppoll_chk(fds, fd_count, timeout, mask, __bos(fds));
93 } else if (__bos(fds) / sizeof(*fds) < fd_count) {
94 __ppoll_too_small_error();
95 }
96 }
97 return __ppoll_real(fds, fd_count, timeout, mask);
98 }
99
100 #endif /* defined(__clang__) */
101 #endif /* __ANDROID_API__ >= __ANDROID_API_M__ */
102 #endif /* defined(__BIONIC_FORTIFY) */
103