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 _SYS_SOCKET_H_
30 #error "Never include this file directly; instead, include <sys/socket.h>"
31 #endif
32
33 extern ssize_t __sendto_chk(int, const void*, size_t, size_t, int, const struct sockaddr*,
34 socklen_t) __INTRODUCED_IN(26);
35 ssize_t __recvfrom_chk(int, void*, size_t, size_t, int, struct sockaddr*,
36 socklen_t*) __INTRODUCED_IN(21);
37
38 #if defined(__BIONIC_FORTIFY)
39
40 #define __recvfrom_bad_size "'recvfrom' called with size bigger than buffer"
41 #define __sendto_bad_size "'sendto' called with size bigger than buffer"
42 #if defined(__clang__)
43 #if __ANDROID_API__ >= __ANDROID_API_N__
44 __BIONIC_FORTIFY_INLINE
recvfrom(int fd,void * const buf __pass_object_size0,size_t len,int flags,struct sockaddr * src_addr,socklen_t * addr_len)45 ssize_t recvfrom(int fd, void* const buf __pass_object_size0, size_t len, int flags, struct sockaddr* src_addr, socklen_t* addr_len)
46 __overloadable
47 __clang_error_if(__bos(buf) != __BIONIC_FORTIFY_UNKNOWN_SIZE && __bos(buf) < len,
48 __recvfrom_bad_size) {
49 size_t bos = __bos0(buf);
50
51 if (bos == __BIONIC_FORTIFY_UNKNOWN_SIZE) {
52 return __call_bypassing_fortify(recvfrom)(fd, buf, len, flags, src_addr, addr_len);
53 }
54 return __recvfrom_chk(fd, buf, len, bos, flags, src_addr, addr_len);
55 }
56 #endif /* __ANDROID_API__ >= __ANDROID_API_N__ */
57
58 #if __ANDROID_API__ >= __ANDROID_API_N_MR1__
59 __BIONIC_FORTIFY_INLINE
sendto(int fd,const void * const buf __pass_object_size0,size_t len,int flags,const struct sockaddr * dest_addr,socklen_t addr_len)60 ssize_t sendto(int fd, const void* const buf __pass_object_size0, size_t len, int flags, const struct sockaddr* dest_addr, socklen_t addr_len)
61 __overloadable
62 __clang_error_if(__bos0(buf) != __BIONIC_FORTIFY_UNKNOWN_SIZE && __bos0(buf) < len,
63 __sendto_bad_size) {
64 size_t bos = __bos0(buf);
65
66 if (bos == __BIONIC_FORTIFY_UNKNOWN_SIZE) {
67 return __call_bypassing_fortify(sendto)(fd, buf, len, flags, dest_addr, addr_len);
68 }
69 return __sendto_chk(fd, buf, len, bos, flags, dest_addr, addr_len);
70 }
71 #endif /* __ANDROID_API__ >= __ANDROID_API_N_MR1__ */
72
73 __BIONIC_FORTIFY_INLINE
recv(int socket,void * const buf __pass_object_size0,size_t len,int flags)74 ssize_t recv(int socket, void* const buf __pass_object_size0, size_t len, int flags)
75 __overloadable
76 __clang_error_if(__bos0(buf) != __BIONIC_FORTIFY_UNKNOWN_SIZE && __bos0(buf) < len,
77 "'recv' called with size bigger than buffer") {
78 return recvfrom(socket, buf, len, flags, NULL, 0);
79 }
80
81 __BIONIC_FORTIFY_INLINE
send(int socket,const void * const buf __pass_object_size0,size_t len,int flags)82 ssize_t send(int socket, const void* const buf __pass_object_size0, size_t len, int flags)
83 __overloadable
84 __clang_error_if(__bos0(buf) != __BIONIC_FORTIFY_UNKNOWN_SIZE && __bos0(buf) < len,
85 "'send' called with size bigger than buffer") {
86 return sendto(socket, buf, len, flags, NULL, 0);
87 }
88
89 #else /* defined(__clang__) */
90 ssize_t __recvfrom_real(int, void*, size_t, int, struct sockaddr*, socklen_t*) __RENAME(recvfrom);
91 __errordecl(__recvfrom_error, __recvfrom_bad_size);
92
93 extern ssize_t __sendto_real(int, const void*, size_t, int, const struct sockaddr*, socklen_t)
94 __RENAME(sendto);
95 __errordecl(__sendto_error, __sendto_bad_size);
96
97 #if __ANDROID_API__ >= __ANDROID_API_N__
98 __BIONIC_FORTIFY_INLINE
recvfrom(int fd,void * buf,size_t len,int flags,struct sockaddr * src_addr,socklen_t * addr_len)99 ssize_t recvfrom(int fd, void* buf, size_t len, int flags,
100 struct sockaddr* src_addr, socklen_t* addr_len) {
101 size_t bos = __bos0(buf);
102
103 if (bos == __BIONIC_FORTIFY_UNKNOWN_SIZE) {
104 return __recvfrom_real(fd, buf, len, flags, src_addr, addr_len);
105 }
106
107 if (__builtin_constant_p(len) && (len <= bos)) {
108 return __recvfrom_real(fd, buf, len, flags, src_addr, addr_len);
109 }
110
111 if (__builtin_constant_p(len) && (len > bos)) {
112 __recvfrom_error();
113 }
114
115 return __recvfrom_chk(fd, buf, len, bos, flags, src_addr, addr_len);
116 }
117 #endif /* __ANDROID_API__ >= __ANDROID_API_N__ */
118
119 #if __ANDROID_API__ >= __ANDROID_API_N_MR1__
120 __BIONIC_FORTIFY_INLINE
sendto(int fd,const void * buf,size_t len,int flags,const struct sockaddr * dest_addr,socklen_t addr_len)121 ssize_t sendto(int fd, const void* buf, size_t len, int flags,
122 const struct sockaddr* dest_addr, socklen_t addr_len) {
123 size_t bos = __bos0(buf);
124
125 if (bos == __BIONIC_FORTIFY_UNKNOWN_SIZE) {
126 return __sendto_real(fd, buf, len, flags, dest_addr, addr_len);
127 }
128
129 if (__builtin_constant_p(len) && (len <= bos)) {
130 return __sendto_real(fd, buf, len, flags, dest_addr, addr_len);
131 }
132
133 if (__builtin_constant_p(len) && (len > bos)) {
134 __sendto_error();
135 }
136
137 return __sendto_chk(fd, buf, len, bos, flags, dest_addr, addr_len);
138 }
139 #endif /* __ANDROID_API__ >= __ANDROID_API_N_MR1__ */
140
141 __BIONIC_FORTIFY_INLINE
recv(int socket,void * buf,size_t len,int flags)142 ssize_t recv(int socket, void* buf, size_t len, int flags) {
143 return recvfrom(socket, buf, len, flags, NULL, 0);
144 }
145
146 __BIONIC_FORTIFY_INLINE
send(int socket,const void * buf,size_t len,int flags)147 ssize_t send(int socket, const void* buf, size_t len, int flags) {
148 return sendto(socket, buf, len, flags, NULL, 0);
149 }
150 #endif /* defined(__clang__) */
151
152 #undef __recvfrom_bad_size
153 #undef __sendto_bad_size
154 #endif /* __BIONIC_FORTIFY */
155