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
33extern ssize_t __sendto_chk(int, const void*, size_t, size_t, int, const struct sockaddr*,
34        socklen_t) __INTRODUCED_IN(26);
35ssize_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
45ssize_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
60ssize_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
74ssize_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
82ssize_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__) */
90ssize_t __recvfrom_real(int, void*, size_t, int, struct sockaddr*, socklen_t*) __RENAME(recvfrom);
91__errordecl(__recvfrom_error, __recvfrom_bad_size);
92
93extern 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
99ssize_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
121ssize_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
142ssize_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
147ssize_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