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
28// zig patch: ssp/socket.h header was added in FreeBSD 15
29#if __FreeBSD_version < 1500500
30 #error "ssp/socket.h did not exist before FreeBSD 15"
31#endif /* error for FreeBSD before 15 */
32
33#ifndef _SSP_SOCKET_H_
34#define _SSP_SOCKET_H_
35
36#include <ssp/ssp.h>
37
38#if __SSP_FORTIFY_LEVEL > 0
39
40#include <sys/_null.h>
41
42__BEGIN_DECLS
43
44__ssp_inline void
45__ssp_check_msghdr(struct msghdr *hdr)
46{
47 if (__ssp_bos(hdr->msg_name) < hdr->msg_namelen)
48 __chk_fail();
49
50 __ssp_check_iovec(hdr->msg_iov, hdr->msg_iovlen);
51
52 if (__ssp_bos(hdr->msg_control) < hdr->msg_controllen)
53 __chk_fail();
54}
55
56__ssp_redirect_raw_impl(int, getpeername, getpeername,
57 (int fdes, struct sockaddr *__restrict name, socklen_t *__restrict namelen))
58{
59 size_t namesz = __ssp_bos(name);
60
61 if (namesz != (size_t)-1 && namesz < *namelen)
62 __chk_fail();
63
64 return (__ssp_real(getpeername)(fdes, name, namelen));
65}
66
67__ssp_redirect_raw_impl(int, getsockname, getsockname,
68 (int fdes, struct sockaddr *__restrict name,
69 socklen_t *__restrict namelen))
70{
71 size_t namesz = __ssp_bos(name);
72
73 if (namesz != (size_t)-1 && namesz < *namelen)
74 __chk_fail();
75
76 return (__ssp_real(getsockname)(fdes, name, namelen));
77}
78
79__ssp_redirect(ssize_t, recv, (int __sock, void *__buf, size_t __len,
80 int __flags), (__sock, __buf, __len, __flags));
81
82__ssp_redirect_raw_impl(ssize_t, recvfrom, recvfrom,
83 (int s, void *buf, size_t len, int flags,
84 struct sockaddr *__restrict from,
85 socklen_t *__restrict fromlen))
86{
87 if (__ssp_bos(buf) < len)
88 __chk_fail();
89 if (from != NULL && __ssp_bos(from) < *fromlen)
90 __chk_fail();
91
92 return (__ssp_real(recvfrom)(s, buf, len, flags, from, fromlen));
93}
94
95__ssp_redirect_raw_impl(ssize_t, recvmsg, recvmsg,
96 (int s, struct msghdr *hdr, int flags))
97{
98 __ssp_check_msghdr(hdr);
99 return (__ssp_real(recvmsg)(s, hdr, flags));
100}
101
102#if __BSD_VISIBLE
103struct timespec;
104
105__ssp_redirect_raw_impl(ssize_t, recvmmsg, recvmmsg,
106 (int s, struct mmsghdr *__restrict hdrvec, size_t vlen, int flags,
107 const struct timespec *__restrict timeout))
108{
109 const size_t vecsz = __ssp_bos(hdrvec);
110 size_t i;
111
112 if (vecsz != (size_t)-1 && vecsz / sizeof(*hdrvec) < vlen)
113 __chk_fail();
114
115 for (i = 0; i < vlen; i++) {
116 __ssp_check_msghdr(&hdrvec[i].msg_hdr);
117 }
118
119 return (__ssp_real(recvmmsg)(s, hdrvec, vlen, flags, timeout));
120}
121#endif
122
123__END_DECLS
124
125#endif /* __SSP_FORTIFY_LEVEL > 0 */
126#endif /* _SSP_SOCKET_H_ */