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

Add vsprintf() check

This commit is contained in:
sin 2015-03-01 12:10:40 +00:00
parent 66b34fbe59
commit 488c3c7183
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
include/stdio.h
View File

@ -16,6 +16,26 @@ __fortify_fgets(char *s, int n, FILE *fp)
return fgets(s, n, fp);
}
static inline
__attribute__ ((always_inline))
__attribute__ ((__format__ (printf, 2, 0)))
__attribute__ ((__nonnull__ (2)))
int
__fortify_vsprintf(char *s, const char *fmt, __builtin_va_list ap)
{
size_t bos = __builtin_object_size(s, 0);
int r;
if (bos != -1) {
r = vsnprintf(s, bos, fmt, ap);
if (r == -1 || (size_t)r >= bos)
__builtin_trap();
} else {
r = vsprintf(s, fmt, ap);
}
return r;
}
static inline
__attribute__ ((always_inline))
__attribute__ ((__format__ (printf, 3, 0)))
@ -32,6 +52,8 @@ __fortify_vsnprintf(char *s, size_t n, const char *fmt, __builtin_va_list ap)
#undef fgets
#define fgets(s, n, fp) __fortify_fgets(s, n, fp)
#undef vsprintf
#define vsprintf(s, fmt, ap) __fortify_vsprintf(s, fmt, ap)
#undef vsnprintf
#define vsnprintf(s, n, fmt, ap) __fortify_vsnprintf(s, n, fmt, ap)