1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2024, Klara, Inc.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
16 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
17 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
19 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25 * POSSIBILITY OF SUCH DAMAGE.
26 */
27 #ifndef _SSP_SOCKET_H_
28 #define _SSP_SOCKET_H_
29
30 #include <ssp/ssp.h>
31
32 #if __SSP_FORTIFY_LEVEL > 0
33
34 #include <sys/_null.h>
35
36 __BEGIN_DECLS
37
38 __ssp_inline void
__ssp_check_msghdr(struct msghdr * hdr)39 __ssp_check_msghdr(struct msghdr *hdr)
40 {
41 if (__ssp_bos(hdr->msg_name) < hdr->msg_namelen)
42 __chk_fail();
43
44 __ssp_check_iovec(hdr->msg_iov, hdr->msg_iovlen);
45
46 if (__ssp_bos(hdr->msg_control) < hdr->msg_controllen)
47 __chk_fail();
48 }
49
50 __ssp_redirect_raw_impl(int, getpeername, getpeername,
51 (int fdes, struct sockaddr *__restrict name, socklen_t *__restrict namelen))
52 {
53 size_t namesz = __ssp_bos(name);
54
55 if (namesz != (size_t)-1 && namesz < *namelen)
56 __chk_fail();
57
58 return (__ssp_real(getpeername)(fdes, name, namelen));
59 }
60
61 __ssp_redirect_raw_impl(int, getsockname, getsockname,
62 (int fdes, struct sockaddr *__restrict name,
63 socklen_t *__restrict namelen))
64 {
65 size_t namesz = __ssp_bos(name);
66
67 if (namesz != (size_t)-1 && namesz < *namelen)
68 __chk_fail();
69
70 return (__ssp_real(getsockname)(fdes, name, namelen));
71 }
72
73 __ssp_redirect(ssize_t, recv, (int __sock, void *__buf, size_t __len,
74 int __flags), (__sock, __buf, __len, __flags));
75
76 __ssp_redirect_raw_impl(ssize_t, recvfrom, recvfrom,
77 (int s, void *buf, size_t len, int flags,
78 struct sockaddr *__restrict from,
79 socklen_t *__restrict fromlen))
80 {
81 if (__ssp_bos(buf) < len)
82 __chk_fail();
83 if (from != NULL && __ssp_bos(from) < *fromlen)
84 __chk_fail();
85
86 return (__ssp_real(recvfrom)(s, buf, len, flags, from, fromlen));
87 }
88
89 __ssp_redirect_raw_impl(ssize_t, recvmsg, recvmsg,
90 (int s, struct msghdr *hdr, int flags))
91 {
92 __ssp_check_msghdr(hdr);
93 return (__ssp_real(recvmsg)(s, hdr, flags));
94 }
95
96 #if __BSD_VISIBLE
97 struct timespec;
98
99 __ssp_redirect_raw_impl(ssize_t, recvmmsg, recvmmsg,
100 (int s, struct mmsghdr *__restrict hdrvec, size_t vlen, int flags,
101 const struct timespec *__restrict timeout))
102 {
103 const size_t vecsz = __ssp_bos(hdrvec);
104 size_t i;
105
106 if (vecsz != (size_t)-1 && vecsz / sizeof(*hdrvec) < vlen)
107 __chk_fail();
108
109 for (i = 0; i < vlen; i++) {
110 __ssp_check_msghdr(&hdrvec[i].msg_hdr);
111 }
112
113 return (__ssp_real(recvmmsg)(s, hdrvec, vlen, flags, timeout));
114 }
115 #endif
116
117 __END_DECLS
118
119 #endif /* __SSP_FORTIFY_LEVEL > 0 */
120 #endif /* _SSP_SOCKET_H_ */
121