Add fortify_fn() helper in fortify-headers.h

1 files changed, 25 insertions, 48 deletions

diff --git a/include/string.h b/include/string.h
index eca7c63..42c43a2 100644
--- a/include/string.h
+++ b/include/string.h
@@ -4,6 +4,7 @@
 #include_next <string.h>
 
 #if defined(_FORTIFY_SOURCE) && _FORTIFY_SOURCE > 0 && defined(__OPTIMIZE__) && __OPTIMIZE__ > 0
+#include "fortify-headers.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -17,9 +18,7 @@ extern "C" {
 #undef strncat
 #undef strncpy
 
-__typeof__(memcpy) __memcpy_orig __asm__(__USER_LABEL_PREFIX__ "memcpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-void *memcpy(void *dst, const void *src, size_t n)
+fortify_fn(memcpy) void *memcpy(void *dst, const void *src, size_t n)
 {
         size_t bos_dst = __builtin_object_size(dst, 0);
         size_t bos_src = __builtin_object_size(src, 0);
@@ -33,85 +32,71 @@ void *memcpy(void *dst, const void *src, size_t n)
                 __builtin_trap();
         if (n > bos_dst || n > bos_src)
                 __builtin_trap();
-        return __memcpy_orig(dst, src, n);
+        return __orig_memcpy(dst, src, n);
 }
 
-__typeof__(memmove) __memmove_orig __asm__(__USER_LABEL_PREFIX__ "memmove");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-void *memmove(void *dst, const void *src, size_t n)
+fortify_fn(memmove) void *memmove(void *dst, const void *src, size_t n)
 {
         size_t bos_dst = __builtin_object_size(dst, 0);
         size_t bos_src = __builtin_object_size(src, 0);
 
         if (n > bos_dst || n > bos_src)
                 __builtin_trap();
-        return __memmove_orig(dst, src, n);
+        return __orig_memmove(dst, src, n);
 }
 
-__typeof__(memset) __memset_orig __asm__(__USER_LABEL_PREFIX__ "memset");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-void *memset(void *dst, int c, size_t n)
+fortify_fn(memset) void *memset(void *dst, int c, size_t n)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (n > bos)
                 __builtin_trap();
-        return __memset_orig(dst, c, n);
+        return __orig_memset(dst, c, n);
 }
 
 #if defined(_POSIX_SOURCE) || defined(_POSIX_C_SOURCE) \
  || defined(_XOPEN_SOURCE) || defined(_GNU_SOURCE) \
  || defined(_BSD_SOURCE)
 #undef stpcpy
-__typeof__(stpcpy) __stpcpy_orig __asm__(__USER_LABEL_PREFIX__ "stpcpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-char *stpcpy(char *dst, const char *src)
+fortify_fn(stpcpy) char *stpcpy(char *dst, const char *src)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (strlen(src) + 1 > bos)
                 __builtin_trap();
-        return __stpcpy_orig(dst, src);
+        return __orig_stpcpy(dst, src);
 }
 
 #undef stpncpy
-__typeof__(stpncpy) __stpncpy_orig __asm__(__USER_LABEL_PREFIX__ "stpncpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-char *stpncpy(char *dst, const char *src, size_t n)
+fortify_fn(stpncpy) char *stpncpy(char *dst, const char *src, size_t n)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (n > bos)
                 __builtin_trap();
-        return __stpncpy_orig(dst, src, n);
+        return __orig_stpncpy(dst, src, n);
 }
 #endif
 
-__typeof__(strcat) __strcat_orig __asm__(__USER_LABEL_PREFIX__ "strcat");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-char *strcat(char *dst, const char *src)
+fortify_fn(strcat) char *strcat(char *dst, const char *src)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (strlen(src) + strlen(dst) + 1 > bos)
                 __builtin_trap();
-        return __strcat_orig(dst, src);
+        return __orig_strcat(dst, src);
 }
 
-__typeof__(strcpy) __strcpy_orig __asm__(__USER_LABEL_PREFIX__ "strcpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-char *strcpy(char *dst, const char *src)
+fortify_fn(strcpy) char *strcpy(char *dst, const char *src)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (strlen(src) + 1 > bos)
                 __builtin_trap();
-        return __strcpy_orig(dst, src);
+        return __orig_strcpy(dst, src);
 }
 
-__typeof__(strncat) __strncat_orig __asm__(__USER_LABEL_PREFIX__ "strncat");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-char *strncat(char *dst, const char *src, size_t n)
+fortify_fn(strncat) char *strncat(char *dst, const char *src, size_t n)
 {
         size_t bos = __builtin_object_size(dst, 0);
         size_t slen, dlen;
@@ -124,58 +109,50 @@ char *strncat(char *dst, const char *src, size_t n)
                 if (slen + dlen + 1 > bos)
                         __builtin_trap();
         }
-        return __strncat_orig(dst, src, n);
+        return __orig_strncat(dst, src, n);
 }
 
-__typeof__(strncpy) __strncpy_orig __asm__(__USER_LABEL_PREFIX__ "strncpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-char *strncpy(char *dst, const char *src, size_t n)
+fortify_fn(strncpy) char *strncpy(char *dst, const char *src, size_t n)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (n > bos)
                 __builtin_trap();
-        return __strncpy_orig(dst, src, n);
+        return __orig_strncpy(dst, src, n);
 }
 
 #ifdef _GNU_SOURCE
 #undef mempcpy
-__typeof__(mempcpy) __mempcpy_orig __asm__(__USER_LABEL_PREFIX__ "mempcpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-void *mempcpy(void *dst, const void *src, size_t n)
+fortify_fn(mempcpy) void *mempcpy(void *dst, const void *src, size_t n)
 {
         size_t bos_dst = __builtin_object_size(dst, 0);
         size_t bos_src = __builtin_object_size(src, 0);
 
         if (n > bos_dst || n > bos_src)
                 __builtin_trap();
-        return __mempcpy_orig(dst, src, n);
+        return __orig_mempcpy(dst, src, n);
 }
 #endif
 
 #if defined(_GNU_SOURCE) || defined(_BSD_SOURCE)
 #undef strlcat
 #undef strlcpy
-__typeof__(strlcat) __strlcat_orig __asm__(__USER_LABEL_PREFIX__ "strlcat");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-size_t strlcat(char *dst, const char *src, size_t n)
+fortify_fn(strlcat) size_t strlcat(char *dst, const char *src, size_t n)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (n > bos)
                 __builtin_trap();
-        return __strlcat_orig(dst, src, n);
+        return __orig_strlcat(dst, src, n);
 }
 
-__typeof__(strlcpy) __strlcpy_orig __asm__(__USER_LABEL_PREFIX__ "strlcpy");
-extern __inline __attribute__((__always_inline__,__gnu_inline__,__artificial__))
-size_t strlcpy(char *dst, const char *src, size_t n)
+fortify_fn(strlcpy) size_t strlcpy(char *dst, const char *src, size_t n)
 {
         size_t bos = __builtin_object_size(dst, 0);
 
         if (n > bos)
                 __builtin_trap();
-        return __strlcpy_orig(dst, src, n);
+        return __orig_strlcpy(dst, src, n);
 }
 #endif