From 0c985218e9043289f4e72d1492cdb83f07fe600a Mon Sep 17 00:00:00 2001
From: sin
Date: Sun, 1 Mar 2015 13:12:09 +0000
Subject: Add fread() check

---
 include/stdio.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'include/stdio.h')

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
-- 
cgit v1.3