Add fread() check

1 files changed, 15 insertions, 0 deletions

diff --git a/include/stdio.h b/include/stdio.h
index 8486804..d55df07 100644
--- a/include/stdio.h
+++ b/include/stdio.h
@@ -16,6 +16,19 @@ __fortify_fgets(char *s, int n, FILE *fp)
         return fgets(s, n, fp);
 }
 
+static inline __attribute__ ((always_inline))
+size_t
+__fortify_fread(void *dst, size_t n, size_t nmemb, FILE *fp)
+{
+        size_t bos = __builtin_object_size(dst, 0);
+
+        if (n != 0 && (n * nmemb) / n != nmemb)
+                __builtin_trap();
+        if (n * nmemb > bos)
+                __builtin_trap();
+        return fread(dst, n, nmemb, fp);
+}
+
 static inline
 __attribute__ ((always_inline))
 __attribute__ ((__format__ (printf, 2, 0)))
@@ -52,6 +65,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 fread
+#define fread(dst, n, nmemb, fp) __fortify_fread(dst, n, nmemb, fp)
 #undef vsprintf
 #define vsprintf(s, fmt, ap) __fortify_vsprintf(s, fmt, ap)
 #undef vsnprintf