atmos
/
fortify-headers
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add recvfrom() checks

This commit is contained in:
sin 2015-02-04 14:58:32 +00:00
parent cc262554a1
commit 91c0c1270f
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
include/sys/socket.h
View File

@ -23,8 +23,25 @@ __fortify_recv(int sockfd, void *buf, size_t n, int flags)
return recv(sockfd, buf, n, flags);
}
__errordecl(__recvfrom_error, "recvfrom: buffer overflow detected");
static inline __attribute__ ((always_inline))
ssize_t
__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);
if (__builtin_constant_p(n) && n > bos)
__recvfrom_error();
if (n > bos)
__builtin_trap();
return recvfrom(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)
#endif