fortify-headers

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

Add sendto() check

Diffstat:
M
include/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