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 __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)41 ssize_t recvfrom(int fd, void* const buf __pass_object_size0, size_t len, int flags, struct sockaddr* src_addr, socklen_t* addr_len)
42     __overloadable
43     __clang_error_if(__bos_unevaluated_lt(__bos0(buf), len),
44                      "'recvfrom' called with size bigger than buffer") {
45 #if __ANDROID_API__ >= 24 && __BIONIC_FORTIFY_RUNTIME_CHECKS_ENABLED
46   size_t bos = __bos0(buf);
47 
48   if (!__bos_trivially_ge(bos, len)) {
49     return __recvfrom_chk(fd, buf, len, bos, flags, src_addr, addr_len);
50   }
51 #endif
52   return __call_bypassing_fortify(recvfrom)(fd, buf, len, flags, src_addr, addr_len);
53 }
54 
55 __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)56 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)
57     __overloadable
58     __clang_error_if(__bos_unevaluated_lt(__bos0(buf), len),
59                      "'sendto' called with size bigger than buffer") {
60 #if __ANDROID_API__ >= 26 && __BIONIC_FORTIFY_RUNTIME_CHECKS_ENABLED
61   size_t bos = __bos0(buf);
62 
63   if (!__bos_trivially_ge(bos, len)) {
64     return __sendto_chk(fd, buf, len, bos, flags, dest_addr, addr_len);
65   }
66 #endif
67   return __call_bypassing_fortify(sendto)(fd, buf, len, flags, dest_addr, addr_len);
68 }
69 
70 __BIONIC_FORTIFY_INLINE
recv(int socket,void * const buf __pass_object_size0,size_t len,int flags)71 ssize_t recv(int socket, void* const buf __pass_object_size0, size_t len, int flags)
72     __overloadable
73     __clang_error_if(__bos_unevaluated_lt(__bos0(buf), len),
74                      "'recv' called with size bigger than buffer") {
75   return recvfrom(socket, buf, len, flags, NULL, 0);
76 }
77 
78 __BIONIC_FORTIFY_INLINE
send(int socket,const void * const buf __pass_object_size0,size_t len,int flags)79 ssize_t send(int socket, const void* const buf __pass_object_size0, size_t len, int flags)
80     __overloadable
81     __clang_error_if(__bos_unevaluated_lt(__bos0(buf), len),
82                      "'send' called with size bigger than buffer") {
83   return sendto(socket, buf, len, flags, NULL, 0);
84 }
85 
86 #endif /* __BIONIC_FORTIFY */
87