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

Add write() check

This commit is contained in:
sin 2015-02-24 19:34:37 +00:00
parent f05c09fac8
commit eb7c60712c
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
include/unistd.h
View File

@ -60,6 +60,17 @@ __fortify_read(int fd, void *buf, size_t n)
return read(fd, buf, n);
}
static inline __attribute__ ((always_inline))
ssize_t
__fortify_write(int fd, const void *buf, size_t n)
{
size_t bos = __builtin_object_size(buf, 0);
if (n > bos)
__builtin_trap();
return write(fd, buf, n);
}
#undef confstr
#define confstr(name, buf, len) __fortify_confstr(name, buf, len)
#undef getcwd
@ -70,6 +81,8 @@ __fortify_read(int fd, void *buf, size_t n)
#define pread(fd, buf, n, offset) __fortify_pread(fd, buf, n, offset)
#undef read
#define read(fd, buf, n) __fortify_read(fd, buf, n)
#undef write
#define write(fd, buf, n) __fortify_write(fd, buf, n)
#endif