From e2a76a9502415f2bbbc83d634afb4991da2ea960 Mon Sep 17 00:00:00 2001 From: sin Date: Wed, 28 Jan 2015 16:54:48 +0000 Subject: Add vsnprintf() checks --- include/stdio.h | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 include/stdio.h (limited to 'include/stdio.h') diff --git a/include/stdio.h b/include/stdio.h new file mode 100644 index 0000000..122e678 --- /dev/null +++ b/include/stdio.h @@ -0,0 +1,28 @@ +#ifndef FORTIFY_STDIO_H_ +#define FORTIFY_STDIO_H_ + +#include_next + +#if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0 + +static inline +__attribute__ ((always_inline)) +__attribute__ ((__format__ (printf, 3, 0))) +__attribute__ ((__nonnull__ (3))) +int __fortify_vsnprintf(char *__restrict s, size_t n, const char *__restrict fmt, __va_list ap) +{ + size_t bos = __builtin_object_size(s, 0); + + if (bos == (size_t)-1) + return vsnprintf(s, n, fmt, ap); + if (__builtin_constant_p(n) && n > bos) + __builtin_trap(); + return vsnprintf(s, n, fmt, ap); +} + +#undef vsnprintf +#define vsnprintf(s, n, fmt, ap) __fortify_vsnprintf(s, n, fmt, ap) + +#endif + +#endif -- cgit v1.3