fortify-headers

standalone fortify-source implementation
git clone git://git.2f30.org/fortify-headers
Log | Files | Refs | README | LICENSE

commit 91db6f58998609a44fdd119d7faf0010ad28848c
parent 958ead636c14f10fce3954712f6e70d98db952b8
Author: sin <sin@2f30.org>
Date:   Sat, 28 Feb 2015 16:56:21 +0000

Add sendto() check

Diffstat:
Minclude/sys/socket.h | 17++++++++++++++++-
1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/include/sys/socket.h b/include/sys/socket.h @@ -18,7 +18,8 @@ __fortify_recv(int sockfd, void *buf, size_t n, int flags) static inline __attribute__ ((always_inline)) ssize_t -__fortify_recvfrom(int sockfd, void *buf, size_t n, int flags, struct sockaddr *sa, socklen_t *salen) +__fortify_recvfrom(int sockfd, void *buf, size_t n, int flags, + struct sockaddr *sa, socklen_t *salen) { size_t bos = __builtin_object_size(buf, 0); @@ -38,12 +39,26 @@ __fortify_send(int sockfd, const void *buf, size_t n, int flags) return send(sockfd, buf, n, flags); } +static inline __attribute__ ((always_inline)) +ssize_t +__fortify_sendto(int sockfd, const void *buf, size_t n, int flags, + const struct sockaddr *sa, socklen_t salen) +{ + size_t bos = __builtin_object_size(buf, 0); + + if (n > bos) + __builtin_trap(); + return sendto(sockfd, buf, n, flags, sa, salen); +} + #undef recv #define recv(sockfd, buf, n, flags) __fortify_recv(sockfd, buf, n, flags) #undef recvfrom #define recvfrom(sockfd, buf, n, flags, sa, salen) __fortify_recvfrom(sockfd, buf, n, flags, sa, salen) #undef send #define send(sockfd, buf, n, flags) __fortify_send(sockfd, buf, n, flags) +#undef sendto +#define sendto(sockfd, buf, n, flags, sa, salen) __fortify_sendto(sockfd, buf, n, flags, sa, salen) #endif