From 013d07a11c796651404a841db9179e3fda086b5f Mon Sep 17 00:00:00 2001
From: sin
Date: Wed, 28 Jan 2015 16:21:38 +0000
Subject: Add stpncpy()

---
 include/string.h | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'include')

diff --git a/include/string.h b/include/string.h
index f9fcf51..cee84d4 100644
--- a/include/string.h
+++ b/include/string.h
@@ -53,6 +53,18 @@ char *__fortify_stpcpy(char *__restrict dest, const char *__restrict src)
 	return stpcpy(dest, src);
 }
 
+static inline __attribute__ ((always_inline))
+char *__fortify_stpncpy(char *__restrict dest, const char *__restrict src, size_t n)
+{
+	size_t bos = __builtin_object_size(dest, 0);
+
+	if (bos == (size_t)-1)
+		return stpcpy(dest, src);
+	if (__builtin_constant_p(n) && n > bos)
+		__builtin_trap();
+	return stpncpy(dest, src, n);
+}
+
 static inline __attribute__ ((always_inline))
 char *__fortify_strcat(char *__restrict dest, const char *__restrict src)
 {
@@ -142,6 +154,8 @@ size_t __fortify_strlcpy(char *__restrict dest, const char *__restrict src, size
 #define memset(dest, src, n) __fortify_memset(dest, src, n)
 #undef stpcpy
 #define stpcpy(dest, src) __fortify_stpcpy(dest, src);
+#undef stpncpy
+#define stpncpy(dest, src, n) __fortify_stpncpy(dest, src, n);
 #undef strcat
 #define strcat(dest, src) __fortify_strcat(dest, src)
 #undef strcpy
-- 
cgit v1.3