From 2ced6e28c30c98428a85a3c3fac7758d1ddf12b6 Mon Sep 17 00:00:00 2001
From: sin <sin@2f30.org>
Date: Tue, 3 Mar 2015 14:34:35 +0000
Subject: [PATCH] Add fgetws() check

---
 include/wchar.h | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 include/wchar.h

diff --git a/include/wchar.h b/include/wchar.h
new file mode 100644
index 0000000..fe2bcfc
--- /dev/null
+++ b/include/wchar.h
@@ -0,0 +1,24 @@
+#ifndef FORTIFY_WCHAR_H_
+#define FORTIFY_WCHAR_H_
+
+#include_next <wchar.h>
+
+#if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0
+
+static inline __attribute__ ((always_inline))
+wchar_t *
+__fortify_fgetws(wchar_t *s, int n, FILE *fp)
+{
+	size_t bos = __builtin_object_size(s, 0);
+
+	if ((size_t)n > bos / sizeof(wchar_t))
+		__builtin_trap();
+	return fgetws(s, n, fp);
+}
+
+#undef fgetws
+#define fgetws(s, n, fp) __fortify_fgetws(s, n, fp)
+
+#endif
+
+#endif