You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1038 lines
37 KiB
1038 lines
37 KiB
commit 0e6ebf06e43e879d9efa2a1c79b2a55f81653663 |
|
Author: Siddhesh Poyarekar <siddhesh@sourceware.org> |
|
Date: Wed Oct 20 18:12:41 2021 +0530 |
|
|
|
Make sure that the fortified function conditionals are constant |
|
|
|
In _FORTIFY_SOURCE=3, the size expression may be non-constant, |
|
resulting in branches in the inline functions remaining intact and |
|
causing a tiny overhead. Clang (and in future, gcc) make sure that |
|
the -1 case is always safe, i.e. any comparison of the generated |
|
expression with (size_t)-1 is always false so that bit is taken care |
|
of. The rest is avoidable since we want the _chk variant whenever we |
|
have a size expression and it's not -1. |
|
|
|
Rework the conditionals in a uniform way to clearly indicate two |
|
conditions at compile time: |
|
|
|
- Either the size is unknown (-1) or we know at compile time that the |
|
operation length is less than the object size. We can call the |
|
original function in this case. It could be that either the length, |
|
object size or both are non-constant, but the compiler, through |
|
range analysis, is able to fold the *comparison* to a constant. |
|
|
|
- The size and length are known and the compiler can see at compile |
|
time that operation length > object size. This is valid grounds for |
|
a warning at compile time, followed by emitting the _chk variant. |
|
|
|
For everything else, emit the _chk variant. |
|
|
|
This simplifies most of the fortified function implementations and at |
|
the same time, ensures that only one call from _chk or the regular |
|
function is emitted. |
|
|
|
Signed-off-by: Siddhesh Poyarekar <siddhesh@sourceware.org> |
|
Reviewed-by: Adhemerval Zanella <adhemerval.zanella@linaro.org> |
|
(cherry picked from commit a643f60c53876be0d57b4b7373770e6cb356fd13) |
|
|
|
diff --git a/io/bits/poll2.h b/io/bits/poll2.h |
|
index be74d020f2e0e434..91cdcaf66a427cea 100644 |
|
--- a/io/bits/poll2.h |
|
+++ b/io/bits/poll2.h |
|
@@ -36,16 +36,9 @@ extern int __REDIRECT (__poll_chk_warn, (struct pollfd *__fds, nfds_t __nfds, |
|
__fortify_function __fortified_attr_access (__write_only__, 1, 2) int |
|
poll (struct pollfd *__fds, nfds_t __nfds, int __timeout) |
|
{ |
|
- if (__glibc_objsize (__fds) != (__SIZE_TYPE__) -1) |
|
- { |
|
- if (! __builtin_constant_p (__nfds)) |
|
- return __poll_chk (__fds, __nfds, __timeout, __glibc_objsize (__fds)); |
|
- else if (__glibc_objsize (__fds) / sizeof (*__fds) < __nfds) |
|
- return __poll_chk_warn (__fds, __nfds, __timeout, |
|
- __glibc_objsize (__fds)); |
|
- } |
|
- |
|
- return __poll_alias (__fds, __nfds, __timeout); |
|
+ return __glibc_fortify (poll, __nfds, sizeof (*__fds), |
|
+ __glibc_objsize (__fds), |
|
+ __fds, __nfds, __timeout); |
|
} |
|
|
|
|
|
@@ -68,17 +61,9 @@ __fortify_function __fortified_attr_access (__write_only__, 1, 2) int |
|
ppoll (struct pollfd *__fds, nfds_t __nfds, const struct timespec *__timeout, |
|
const __sigset_t *__ss) |
|
{ |
|
- if (__glibc_objsize (__fds) != (__SIZE_TYPE__) -1) |
|
- { |
|
- if (! __builtin_constant_p (__nfds)) |
|
- return __ppoll_chk (__fds, __nfds, __timeout, __ss, |
|
- __glibc_objsize (__fds)); |
|
- else if (__glibc_objsize (__fds) / sizeof (*__fds) < __nfds) |
|
- return __ppoll_chk_warn (__fds, __nfds, __timeout, __ss, |
|
- __glibc_objsize (__fds)); |
|
- } |
|
- |
|
- return __ppoll_alias (__fds, __nfds, __timeout, __ss); |
|
+ return __glibc_fortify (ppoll, __nfds, sizeof (*__fds), |
|
+ __glibc_objsize (__fds), |
|
+ __fds, __nfds, __timeout, __ss); |
|
} |
|
#endif |
|
|
|
diff --git a/libio/bits/stdio2.h b/libio/bits/stdio2.h |
|
index 4f016a563857a137..40ff16b01b4f4876 100644 |
|
--- a/libio/bits/stdio2.h |
|
+++ b/libio/bits/stdio2.h |
|
@@ -261,15 +261,12 @@ extern char *__REDIRECT (__fgets_chk_warn, |
|
__fortify_function __wur __fortified_attr_access (__write_only__, 1, 2) char * |
|
fgets (char *__restrict __s, int __n, FILE *__restrict __stream) |
|
{ |
|
- if (__glibc_objsize (__s) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n) || __n <= 0) |
|
- return __fgets_chk (__s, __glibc_objsize (__s), __n, __stream); |
|
- |
|
- if ((size_t) __n > __glibc_objsize (__s)) |
|
- return __fgets_chk_warn (__s, __glibc_objsize (__s), __n, __stream); |
|
- } |
|
- return __fgets_alias (__s, __n, __stream); |
|
+ size_t sz = __glibc_objsize (__s); |
|
+ if (__glibc_safe_or_unknown_len (__n, sizeof (char), sz)) |
|
+ return __fgets_alias (__s, __n, __stream); |
|
+ if (__glibc_unsafe_len (__n, sizeof (char), sz)) |
|
+ return __fgets_chk_warn (__s, sz, __n, __stream); |
|
+ return __fgets_chk (__s, sz, __n, __stream); |
|
} |
|
|
|
extern size_t __fread_chk (void *__restrict __ptr, size_t __ptrlen, |
|
@@ -291,19 +288,12 @@ __fortify_function __wur size_t |
|
fread (void *__restrict __ptr, size_t __size, size_t __n, |
|
FILE *__restrict __stream) |
|
{ |
|
- if (__glibc_objsize0 (__ptr) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__size) |
|
- || !__builtin_constant_p (__n) |
|
- || (__size | __n) >= (((size_t) 1) << (8 * sizeof (size_t) / 2))) |
|
- return __fread_chk (__ptr, __glibc_objsize0 (__ptr), __size, __n, |
|
- __stream); |
|
- |
|
- if (__size * __n > __glibc_objsize0 (__ptr)) |
|
- return __fread_chk_warn (__ptr, __glibc_objsize0 (__ptr), __size, __n, |
|
- __stream); |
|
- } |
|
- return __fread_alias (__ptr, __size, __n, __stream); |
|
+ size_t sz = __glibc_objsize0 (__ptr); |
|
+ if (__glibc_safe_or_unknown_len (__n, __size, sz)) |
|
+ return __fread_alias (__ptr, __size, __n, __stream); |
|
+ if (__glibc_unsafe_len (__n, __size, sz)) |
|
+ return __fread_chk_warn (__ptr, sz, __size, __n, __stream); |
|
+ return __fread_chk (__ptr, sz, __size, __n, __stream); |
|
} |
|
|
|
#ifdef __USE_GNU |
|
@@ -323,17 +313,12 @@ extern char *__REDIRECT (__fgets_unlocked_chk_warn, |
|
__fortify_function __wur __fortified_attr_access (__write_only__, 1, 2) char * |
|
fgets_unlocked (char *__restrict __s, int __n, FILE *__restrict __stream) |
|
{ |
|
- if (__glibc_objsize (__s) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n) || __n <= 0) |
|
- return __fgets_unlocked_chk (__s, __glibc_objsize (__s), __n, |
|
- __stream); |
|
- |
|
- if ((size_t) __n > __glibc_objsize (__s)) |
|
- return __fgets_unlocked_chk_warn (__s, __glibc_objsize (__s), __n, |
|
- __stream); |
|
- } |
|
- return __fgets_unlocked_alias (__s, __n, __stream); |
|
+ size_t sz = __glibc_objsize (__s); |
|
+ if (__glibc_safe_or_unknown_len (__n, sizeof (char), sz)) |
|
+ return __fgets_unlocked_alias (__s, __n, __stream); |
|
+ if (__glibc_unsafe_len (__n, sizeof (char), sz)) |
|
+ return __fgets_unlocked_chk_warn (__s, sz, __n, __stream); |
|
+ return __fgets_unlocked_chk (__s, sz, __n, __stream); |
|
} |
|
#endif |
|
|
|
@@ -358,41 +343,36 @@ __fortify_function __wur size_t |
|
fread_unlocked (void *__restrict __ptr, size_t __size, size_t __n, |
|
FILE *__restrict __stream) |
|
{ |
|
- if (__glibc_objsize0 (__ptr) != (size_t) -1) |
|
+ size_t sz = __glibc_objsize0 (__ptr); |
|
+ if (__glibc_safe_or_unknown_len (__n, __size, sz)) |
|
{ |
|
- if (!__builtin_constant_p (__size) |
|
- || !__builtin_constant_p (__n) |
|
- || (__size | __n) >= (((size_t) 1) << (8 * sizeof (size_t) / 2))) |
|
- return __fread_unlocked_chk (__ptr, __glibc_objsize0 (__ptr), __size, |
|
- __n, __stream); |
|
- |
|
- if (__size * __n > __glibc_objsize0 (__ptr)) |
|
- return __fread_unlocked_chk_warn (__ptr, __glibc_objsize0 (__ptr), |
|
- __size, __n, __stream); |
|
- } |
|
- |
|
# ifdef __USE_EXTERN_INLINES |
|
- if (__builtin_constant_p (__size) |
|
- && __builtin_constant_p (__n) |
|
- && (__size | __n) < (((size_t) 1) << (8 * sizeof (size_t) / 2)) |
|
- && __size * __n <= 8) |
|
- { |
|
- size_t __cnt = __size * __n; |
|
- char *__cptr = (char *) __ptr; |
|
- if (__cnt == 0) |
|
- return 0; |
|
- |
|
- for (; __cnt > 0; --__cnt) |
|
+ if (__builtin_constant_p (__size) |
|
+ && __builtin_constant_p (__n) |
|
+ && (__size | __n) < (((size_t) 1) << (8 * sizeof (size_t) / 2)) |
|
+ && __size * __n <= 8) |
|
{ |
|
- int __c = getc_unlocked (__stream); |
|
- if (__c == EOF) |
|
- break; |
|
- *__cptr++ = __c; |
|
+ size_t __cnt = __size * __n; |
|
+ char *__cptr = (char *) __ptr; |
|
+ if (__cnt == 0) |
|
+ return 0; |
|
+ |
|
+ for (; __cnt > 0; --__cnt) |
|
+ { |
|
+ int __c = getc_unlocked (__stream); |
|
+ if (__c == EOF) |
|
+ break; |
|
+ *__cptr++ = __c; |
|
+ } |
|
+ return (__cptr - (char *) __ptr) / __size; |
|
} |
|
- return (__cptr - (char *) __ptr) / __size; |
|
- } |
|
# endif |
|
- return __fread_unlocked_alias (__ptr, __size, __n, __stream); |
|
+ return __fread_unlocked_alias (__ptr, __size, __n, __stream); |
|
+ } |
|
+ if (__glibc_unsafe_len (__n, __size, sz)) |
|
+ return __fread_unlocked_chk_warn (__ptr, sz, __size, __n, __stream); |
|
+ return __fread_unlocked_chk (__ptr, sz, __size, __n, __stream); |
|
+ |
|
} |
|
#endif |
|
|
|
diff --git a/misc/sys/cdefs.h b/misc/sys/cdefs.h |
|
index cd836441a9807d6a..4825ff0351c1e5d4 100644 |
|
--- a/misc/sys/cdefs.h |
|
+++ b/misc/sys/cdefs.h |
|
@@ -150,6 +150,53 @@ |
|
# define __glibc_objsize(__o) __bos (__o) |
|
#endif |
|
|
|
+/* Compile time conditions to choose between the regular, _chk and _chk_warn |
|
+ variants. These conditions should get evaluated to constant and optimized |
|
+ away. */ |
|
+ |
|
+#define __glibc_safe_len_cond(__l, __s, __osz) ((__l) <= (__osz) / (__s)) |
|
+#define __glibc_unsigned_or_positive(__l) \ |
|
+ ((__typeof (__l)) 0 < (__typeof (__l)) -1 \ |
|
+ || (__builtin_constant_p (__l) && (__l) > 0)) |
|
+ |
|
+/* Length is known to be safe at compile time if the __L * __S <= __OBJSZ |
|
+ condition can be folded to a constant and if it is true. The -1 check is |
|
+ redundant because since it implies that __glibc_safe_len_cond is true. */ |
|
+#define __glibc_safe_or_unknown_len(__l, __s, __osz) \ |
|
+ (__glibc_unsigned_or_positive (__l) \ |
|
+ && __builtin_constant_p (__glibc_safe_len_cond ((__SIZE_TYPE__) (__l), \ |
|
+ __s, __osz)) \ |
|
+ && __glibc_safe_len_cond ((__SIZE_TYPE__) (__l), __s, __osz)) |
|
+ |
|
+/* Conversely, we know at compile time that the length is safe if the |
|
+ __L * __S <= __OBJSZ condition can be folded to a constant and if it is |
|
+ false. */ |
|
+#define __glibc_unsafe_len(__l, __s, __osz) \ |
|
+ (__glibc_unsigned_or_positive (__l) \ |
|
+ && __builtin_constant_p (__glibc_safe_len_cond ((__SIZE_TYPE__) (__l), \ |
|
+ __s, __osz)) \ |
|
+ && !__glibc_safe_len_cond ((__SIZE_TYPE__) (__l), __s, __osz)) |
|
+ |
|
+/* Fortify function f. __f_alias, __f_chk and __f_chk_warn must be |
|
+ declared. */ |
|
+ |
|
+#define __glibc_fortify(f, __l, __s, __osz, ...) \ |
|
+ (__glibc_safe_or_unknown_len (__l, __s, __osz) \ |
|
+ ? __ ## f ## _alias (__VA_ARGS__) \ |
|
+ : (__glibc_unsafe_len (__l, __s, __osz) \ |
|
+ ? __ ## f ## _chk_warn (__VA_ARGS__, __osz) \ |
|
+ : __ ## f ## _chk (__VA_ARGS__, __osz))) \ |
|
+ |
|
+/* Fortify function f, where object size argument passed to f is the number of |
|
+ elements and not total size. */ |
|
+ |
|
+#define __glibc_fortify_n(f, __l, __s, __osz, ...) \ |
|
+ (__glibc_safe_or_unknown_len (__l, __s, __osz) \ |
|
+ ? __ ## f ## _alias (__VA_ARGS__) \ |
|
+ : (__glibc_unsafe_len (__l, __s, __osz) \ |
|
+ ? __ ## f ## _chk_warn (__VA_ARGS__, (__osz) / (__s)) \ |
|
+ : __ ## f ## _chk (__VA_ARGS__, (__osz) / (__s)))) \ |
|
+ |
|
#if __GNUC_PREREQ (4,3) |
|
# define __warnattr(msg) __attribute__((__warning__ (msg))) |
|
# define __errordecl(name, msg) \ |
|
diff --git a/posix/bits/unistd.h b/posix/bits/unistd.h |
|
index 622adeb2b28ed298..697dcbbf7b4b26f6 100644 |
|
--- a/posix/bits/unistd.h |
|
+++ b/posix/bits/unistd.h |
|
@@ -35,16 +35,9 @@ extern ssize_t __REDIRECT (__read_chk_warn, |
|
__fortify_function __wur ssize_t |
|
read (int __fd, void *__buf, size_t __nbytes) |
|
{ |
|
- if (__glibc_objsize0 (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__nbytes)) |
|
- return __read_chk (__fd, __buf, __nbytes, __glibc_objsize0 (__buf)); |
|
- |
|
- if (__nbytes > __glibc_objsize0 (__buf)) |
|
- return __read_chk_warn (__fd, __buf, __nbytes, |
|
- __glibc_objsize0 (__buf)); |
|
- } |
|
- return __read_alias (__fd, __buf, __nbytes); |
|
+ return __glibc_fortify (read, __nbytes, sizeof (char), |
|
+ __glibc_objsize0 (__buf), |
|
+ __fd, __buf, __nbytes); |
|
} |
|
|
|
#ifdef __USE_UNIX98 |
|
@@ -78,34 +71,17 @@ extern ssize_t __REDIRECT (__pread64_chk_warn, |
|
__fortify_function __wur ssize_t |
|
pread (int __fd, void *__buf, size_t __nbytes, __off_t __offset) |
|
{ |
|
- if (__glibc_objsize0 (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__nbytes)) |
|
- return __pread_chk (__fd, __buf, __nbytes, __offset, |
|
- __glibc_objsize0 (__buf)); |
|
- |
|
- if ( __nbytes > __glibc_objsize0 (__buf)) |
|
- return __pread_chk_warn (__fd, __buf, __nbytes, __offset, |
|
- __glibc_objsize0 (__buf)); |
|
- } |
|
- return __pread_alias (__fd, __buf, __nbytes, __offset); |
|
+ return __glibc_fortify (pread, __nbytes, sizeof (char), |
|
+ __glibc_objsize0 (__buf), |
|
+ __fd, __buf, __nbytes, __offset); |
|
} |
|
# else |
|
__fortify_function __wur ssize_t |
|
pread (int __fd, void *__buf, size_t __nbytes, __off64_t __offset) |
|
{ |
|
- if (__glibc_objsize0 (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__nbytes)) |
|
- return __pread64_chk (__fd, __buf, __nbytes, __offset, |
|
- __glibc_objsize0 (__buf)); |
|
- |
|
- if ( __nbytes > __glibc_objsize0 (__buf)) |
|
- return __pread64_chk_warn (__fd, __buf, __nbytes, __offset, |
|
- __glibc_objsize0 (__buf)); |
|
- } |
|
- |
|
- return __pread64_alias (__fd, __buf, __nbytes, __offset); |
|
+ return __glibc_fortify (pread64, __nbytes, sizeof (char), |
|
+ __glibc_objsize0 (__buf), |
|
+ __fd, __buf, __nbytes, __offset); |
|
} |
|
# endif |
|
|
|
@@ -113,18 +89,9 @@ pread (int __fd, void *__buf, size_t __nbytes, __off64_t __offset) |
|
__fortify_function __wur ssize_t |
|
pread64 (int __fd, void *__buf, size_t __nbytes, __off64_t __offset) |
|
{ |
|
- if (__glibc_objsize0 (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__nbytes)) |
|
- return __pread64_chk (__fd, __buf, __nbytes, __offset, |
|
- __glibc_objsize0 (__buf)); |
|
- |
|
- if ( __nbytes > __glibc_objsize0 (__buf)) |
|
- return __pread64_chk_warn (__fd, __buf, __nbytes, __offset, |
|
- __glibc_objsize0 (__buf)); |
|
- } |
|
- |
|
- return __pread64_alias (__fd, __buf, __nbytes, __offset); |
|
+ return __glibc_fortify (pread64, __nbytes, sizeof (char), |
|
+ __glibc_objsize0 (__buf), |
|
+ __fd, __buf, __nbytes, __offset); |
|
} |
|
# endif |
|
#endif |
|
@@ -149,16 +116,9 @@ __fortify_function __nonnull ((1, 2)) __wur ssize_t |
|
__NTH (readlink (const char *__restrict __path, char *__restrict __buf, |
|
size_t __len)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __readlink_chk (__path, __buf, __len, __glibc_objsize (__buf)); |
|
- |
|
- if ( __len > __glibc_objsize (__buf)) |
|
- return __readlink_chk_warn (__path, __buf, __len, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __readlink_alias (__path, __buf, __len); |
|
+ return __glibc_fortify (readlink, __len, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __path, __buf, __len); |
|
} |
|
#endif |
|
|
|
@@ -184,17 +144,9 @@ __fortify_function __nonnull ((2, 3)) __wur ssize_t |
|
__NTH (readlinkat (int __fd, const char *__restrict __path, |
|
char *__restrict __buf, size_t __len)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __readlinkat_chk (__fd, __path, __buf, __len, |
|
- __glibc_objsize (__buf)); |
|
- |
|
- if (__len > __glibc_objsize (__buf)) |
|
- return __readlinkat_chk_warn (__fd, __path, __buf, __len, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __readlinkat_alias (__fd, __path, __buf, __len); |
|
+ return __glibc_fortify (readlinkat, __len, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __fd, __path, __buf, __len); |
|
} |
|
#endif |
|
|
|
@@ -211,15 +163,9 @@ extern char *__REDIRECT_NTH (__getcwd_chk_warn, |
|
__fortify_function __wur char * |
|
__NTH (getcwd (char *__buf, size_t __size)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__size)) |
|
- return __getcwd_chk (__buf, __size, __glibc_objsize (__buf)); |
|
- |
|
- if (__size > __glibc_objsize (__buf)) |
|
- return __getcwd_chk_warn (__buf, __size, __glibc_objsize (__buf)); |
|
- } |
|
- return __getcwd_alias (__buf, __size); |
|
+ return __glibc_fortify (getcwd, __size, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __buf, __size); |
|
} |
|
|
|
#if defined __USE_MISC || defined __USE_XOPEN_EXTENDED |
|
@@ -253,16 +199,9 @@ extern size_t __REDIRECT_NTH (__confstr_chk_warn, |
|
__fortify_function size_t |
|
__NTH (confstr (int __name, char *__buf, size_t __len)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __confstr_chk (__name, __buf, __len, __glibc_objsize (__buf)); |
|
- |
|
- if (__glibc_objsize (__buf) < __len) |
|
- return __confstr_chk_warn (__name, __buf, __len, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __confstr_alias (__name, __buf, __len); |
|
+ return __glibc_fortify (confstr, __len, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __name, __buf, __len); |
|
} |
|
|
|
|
|
@@ -279,15 +218,9 @@ extern int __REDIRECT_NTH (__getgroups_chk_warn, |
|
__fortify_function int |
|
__NTH (getgroups (int __size, __gid_t __list[])) |
|
{ |
|
- if (__glibc_objsize (__list) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__size) || __size < 0) |
|
- return __getgroups_chk (__size, __list, __glibc_objsize (__list)); |
|
- |
|
- if (__size * sizeof (__gid_t) > __glibc_objsize (__list)) |
|
- return __getgroups_chk_warn (__size, __list, __glibc_objsize (__list)); |
|
- } |
|
- return __getgroups_alias (__size, __list); |
|
+ return __glibc_fortify (getgroups, __size, sizeof (__gid_t), |
|
+ __glibc_objsize (__list), |
|
+ __size, __list); |
|
} |
|
|
|
|
|
@@ -306,17 +239,9 @@ extern int __REDIRECT_NTH (__ttyname_r_chk_warn, |
|
__fortify_function int |
|
__NTH (ttyname_r (int __fd, char *__buf, size_t __buflen)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__buflen)) |
|
- return __ttyname_r_chk (__fd, __buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- |
|
- if (__buflen > __glibc_objsize (__buf)) |
|
- return __ttyname_r_chk_warn (__fd, __buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __ttyname_r_alias (__fd, __buf, __buflen); |
|
+ return __glibc_fortify (ttyname_r, __buflen, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __fd, __buf, __buflen); |
|
} |
|
|
|
|
|
@@ -334,16 +259,9 @@ extern int __REDIRECT (__getlogin_r_chk_warn, |
|
__fortify_function int |
|
getlogin_r (char *__buf, size_t __buflen) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__buflen)) |
|
- return __getlogin_r_chk (__buf, __buflen, __glibc_objsize (__buf)); |
|
- |
|
- if (__buflen > __glibc_objsize (__buf)) |
|
- return __getlogin_r_chk_warn (__buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __getlogin_r_alias (__buf, __buflen); |
|
+ return __glibc_fortify (getlogin_r, __buflen, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __buf, __buflen); |
|
} |
|
#endif |
|
|
|
@@ -363,16 +281,9 @@ extern int __REDIRECT_NTH (__gethostname_chk_warn, |
|
__fortify_function int |
|
__NTH (gethostname (char *__buf, size_t __buflen)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__buflen)) |
|
- return __gethostname_chk (__buf, __buflen, __glibc_objsize (__buf)); |
|
- |
|
- if (__buflen > __glibc_objsize (__buf)) |
|
- return __gethostname_chk_warn (__buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __gethostname_alias (__buf, __buflen); |
|
+ return __glibc_fortify (gethostname, __buflen, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __buf, __buflen); |
|
} |
|
#endif |
|
|
|
@@ -394,15 +305,8 @@ extern int __REDIRECT_NTH (__getdomainname_chk_warn, |
|
__fortify_function int |
|
__NTH (getdomainname (char *__buf, size_t __buflen)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__buflen)) |
|
- return __getdomainname_chk (__buf, __buflen, __glibc_objsize (__buf)); |
|
- |
|
- if (__buflen > __glibc_objsize (__buf)) |
|
- return __getdomainname_chk_warn (__buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __getdomainname_alias (__buf, __buflen); |
|
+ return __glibc_fortify (getdomainname, __buflen, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __buf, __buflen); |
|
} |
|
#endif |
|
diff --git a/socket/bits/socket2.h b/socket/bits/socket2.h |
|
index 9c8ac69624ea4f78..b28cde55f3fd9c16 100644 |
|
--- a/socket/bits/socket2.h |
|
+++ b/socket/bits/socket2.h |
|
@@ -33,17 +33,12 @@ extern ssize_t __REDIRECT (__recv_chk_warn, |
|
__fortify_function ssize_t |
|
recv (int __fd, void *__buf, size_t __n, int __flags) |
|
{ |
|
- if (__glibc_objsize0 (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __recv_chk (__fd, __buf, __n, __glibc_objsize0 (__buf), |
|
- __flags); |
|
- |
|
- if (__n > __glibc_objsize0 (__buf)) |
|
- return __recv_chk_warn (__fd, __buf, __n, __glibc_objsize0 (__buf), |
|
- __flags); |
|
- } |
|
- return __recv_alias (__fd, __buf, __n, __flags); |
|
+ size_t sz = __glibc_objsize0 (__buf); |
|
+ if (__glibc_safe_or_unknown_len (__n, sizeof (char), sz)) |
|
+ return __recv_alias (__fd, __buf, __n, __flags); |
|
+ if (__glibc_unsafe_len (__n, sizeof (char), sz)) |
|
+ return __recv_chk_warn (__fd, __buf, __n, sz, __flags); |
|
+ return __recv_chk (__fd, __buf, __n, sz, __flags); |
|
} |
|
|
|
extern ssize_t __recvfrom_chk (int __fd, void *__restrict __buf, size_t __n, |
|
@@ -66,14 +61,11 @@ __fortify_function ssize_t |
|
recvfrom (int __fd, void *__restrict __buf, size_t __n, int __flags, |
|
__SOCKADDR_ARG __addr, socklen_t *__restrict __addr_len) |
|
{ |
|
- if (__glibc_objsize0 (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __recvfrom_chk (__fd, __buf, __n, __glibc_objsize0 (__buf), |
|
- __flags, __addr, __addr_len); |
|
- if (__n > __glibc_objsize0 (__buf)) |
|
- return __recvfrom_chk_warn (__fd, __buf, __n, __glibc_objsize0 (__buf), |
|
- __flags, __addr, __addr_len); |
|
- } |
|
- return __recvfrom_alias (__fd, __buf, __n, __flags, __addr, __addr_len); |
|
+ size_t sz = __glibc_objsize0 (__buf); |
|
+ if (__glibc_safe_or_unknown_len (__n, sizeof (char), sz)) |
|
+ return __recvfrom_alias (__fd, __buf, __n, __flags, __addr, __addr_len); |
|
+ if (__glibc_unsafe_len (__n, sizeof (char), sz)) |
|
+ return __recvfrom_chk_warn (__fd, __buf, __n, sz, __flags, __addr, |
|
+ __addr_len); |
|
+ return __recvfrom_chk (__fd, __buf, __n, sz, __flags, __addr, __addr_len); |
|
} |
|
diff --git a/stdlib/bits/stdlib.h b/stdlib/bits/stdlib.h |
|
index eae31b38f0475c2e..067115eeca123c6d 100644 |
|
--- a/stdlib/bits/stdlib.h |
|
+++ b/stdlib/bits/stdlib.h |
|
@@ -36,17 +36,16 @@ extern char *__REDIRECT_NTH (__realpath_chk_warn, |
|
__fortify_function __wur char * |
|
__NTH (realpath (const char *__restrict __name, char *__restrict __resolved)) |
|
{ |
|
- if (__glibc_objsize (__resolved) != (size_t) -1) |
|
- { |
|
+ size_t sz = __glibc_objsize (__resolved); |
|
+ |
|
+ if (sz == (size_t) -1) |
|
+ return __realpath_alias (__name, __resolved); |
|
+ |
|
#if defined _LIBC_LIMITS_H_ && defined PATH_MAX |
|
- if (__glibc_objsize (__resolved) < PATH_MAX) |
|
- return __realpath_chk_warn (__name, __resolved, |
|
- __glibc_objsize (__resolved)); |
|
+ if (__glibc_unsafe_len (sz, sizeof (char), PATH_MAX)) |
|
+ return __realpath_chk_warn (__name, __resolved, sz); |
|
#endif |
|
- return __realpath_chk (__name, __resolved, __glibc_objsize (__resolved)); |
|
- } |
|
- |
|
- return __realpath_alias (__name, __resolved); |
|
+ return __realpath_chk (__name, __resolved, sz); |
|
} |
|
|
|
|
|
@@ -65,16 +64,9 @@ extern int __REDIRECT_NTH (__ptsname_r_chk_warn, |
|
__fortify_function int |
|
__NTH (ptsname_r (int __fd, char *__buf, size_t __buflen)) |
|
{ |
|
- if (__glibc_objsize (__buf) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__buflen)) |
|
- return __ptsname_r_chk (__fd, __buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- if (__buflen > __glibc_objsize (__buf)) |
|
- return __ptsname_r_chk_warn (__fd, __buf, __buflen, |
|
- __glibc_objsize (__buf)); |
|
- } |
|
- return __ptsname_r_alias (__fd, __buf, __buflen); |
|
+ return __glibc_fortify (ptsname_r, __buflen, sizeof (char), |
|
+ __glibc_objsize (__buf), |
|
+ __fd, __buf, __buflen); |
|
} |
|
|
|
|
|
@@ -120,18 +112,9 @@ __fortify_function size_t |
|
__NTH (mbstowcs (wchar_t *__restrict __dst, const char *__restrict __src, |
|
size_t __len)) |
|
{ |
|
- if (__glibc_objsize (__dst) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __mbstowcs_chk (__dst, __src, __len, |
|
- __glibc_objsize (__dst) / sizeof (wchar_t)); |
|
- |
|
- if (__len > __glibc_objsize (__dst) / sizeof (wchar_t)) |
|
- return __mbstowcs_chk_warn (__dst, __src, __len, |
|
- (__glibc_objsize (__dst) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __mbstowcs_alias (__dst, __src, __len); |
|
+ return __glibc_fortify_n (mbstowcs, __len, sizeof (wchar_t), |
|
+ __glibc_objsize (__dst), |
|
+ __dst, __src, __len); |
|
} |
|
|
|
|
|
@@ -154,13 +137,7 @@ __fortify_function size_t |
|
__NTH (wcstombs (char *__restrict __dst, const wchar_t *__restrict __src, |
|
size_t __len)) |
|
{ |
|
- if (__glibc_objsize (__dst) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __wcstombs_chk (__dst, __src, __len, __glibc_objsize (__dst)); |
|
- if (__len > __glibc_objsize (__dst)) |
|
- return __wcstombs_chk_warn (__dst, __src, __len, |
|
- __glibc_objsize (__dst)); |
|
- } |
|
- return __wcstombs_alias (__dst, __src, __len); |
|
+ return __glibc_fortify (wcstombs, __len, sizeof (char), |
|
+ __glibc_objsize (__dst), |
|
+ __dst, __src, __len); |
|
} |
|
diff --git a/wcsmbs/bits/wchar2.h b/wcsmbs/bits/wchar2.h |
|
index ea2518dc726aae52..26012ef9366c0b88 100644 |
|
--- a/wcsmbs/bits/wchar2.h |
|
+++ b/wcsmbs/bits/wchar2.h |
|
@@ -39,17 +39,9 @@ __fortify_function wchar_t * |
|
__NTH (wmemcpy (wchar_t *__restrict __s1, const wchar_t *__restrict __s2, |
|
size_t __n)) |
|
{ |
|
- if (__glibc_objsize0 (__s1) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __wmemcpy_chk (__s1, __s2, __n, |
|
- __glibc_objsize0 (__s1) / sizeof (wchar_t)); |
|
- |
|
- if (__n > __glibc_objsize0 (__s1) / sizeof (wchar_t)) |
|
- return __wmemcpy_chk_warn (__s1, __s2, __n, |
|
- __glibc_objsize0 (__s1) / sizeof (wchar_t)); |
|
- } |
|
- return __wmemcpy_alias (__s1, __s2, __n); |
|
+ return __glibc_fortify_n (wmemcpy, __n, sizeof (wchar_t), |
|
+ __glibc_objsize0 (__s1), |
|
+ __s1, __s2, __n); |
|
} |
|
|
|
|
|
@@ -67,18 +59,9 @@ extern wchar_t *__REDIRECT_NTH (__wmemmove_chk_warn, |
|
__fortify_function wchar_t * |
|
__NTH (wmemmove (wchar_t *__s1, const wchar_t *__s2, size_t __n)) |
|
{ |
|
- if (__glibc_objsize0 (__s1) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __wmemmove_chk (__s1, __s2, __n, |
|
- __glibc_objsize0 (__s1) / sizeof (wchar_t)); |
|
- |
|
- if (__n > __glibc_objsize0 (__s1) / sizeof (wchar_t)) |
|
- return __wmemmove_chk_warn (__s1, __s2, __n, |
|
- (__glibc_objsize0 (__s1) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __wmemmove_alias (__s1, __s2, __n); |
|
+ return __glibc_fortify_n (wmemmove, __n, sizeof (wchar_t), |
|
+ __glibc_objsize0 (__s1), |
|
+ __s1, __s2, __n); |
|
} |
|
|
|
|
|
@@ -101,18 +84,9 @@ __fortify_function wchar_t * |
|
__NTH (wmempcpy (wchar_t *__restrict __s1, const wchar_t *__restrict __s2, |
|
size_t __n)) |
|
{ |
|
- if (__glibc_objsize0 (__s1) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __wmempcpy_chk (__s1, __s2, __n, |
|
- __glibc_objsize0 (__s1) / sizeof (wchar_t)); |
|
- |
|
- if (__n > __glibc_objsize0 (__s1) / sizeof (wchar_t)) |
|
- return __wmempcpy_chk_warn (__s1, __s2, __n, |
|
- (__glibc_objsize0 (__s1) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __wmempcpy_alias (__s1, __s2, __n); |
|
+ return __glibc_fortify_n (wmempcpy, __n, sizeof (wchar_t), |
|
+ __glibc_objsize0 (__s1), |
|
+ __s1, __s2, __n); |
|
} |
|
#endif |
|
|
|
@@ -130,17 +104,9 @@ extern wchar_t *__REDIRECT_NTH (__wmemset_chk_warn, |
|
__fortify_function wchar_t * |
|
__NTH (wmemset (wchar_t *__s, wchar_t __c, size_t __n)) |
|
{ |
|
- if (__glibc_objsize0 (__s) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __wmemset_chk (__s, __c, __n, |
|
- __glibc_objsize0 (__s) / sizeof (wchar_t)); |
|
- |
|
- if (__n > __glibc_objsize0 (__s) / sizeof (wchar_t)) |
|
- return __wmemset_chk_warn (__s, __c, __n, |
|
- __glibc_objsize0 (__s) / sizeof (wchar_t)); |
|
- } |
|
- return __wmemset_alias (__s, __c, __n); |
|
+ return __glibc_fortify_n (wmemset, __n, sizeof (wchar_t), |
|
+ __glibc_objsize0 (__s), |
|
+ __s, __c, __n); |
|
} |
|
|
|
|
|
@@ -154,9 +120,9 @@ extern wchar_t *__REDIRECT_NTH (__wcscpy_alias, |
|
__fortify_function wchar_t * |
|
__NTH (wcscpy (wchar_t *__restrict __dest, const wchar_t *__restrict __src)) |
|
{ |
|
- if (__glibc_objsize (__dest) != (size_t) -1) |
|
- return __wcscpy_chk (__dest, __src, |
|
- __glibc_objsize (__dest) / sizeof (wchar_t)); |
|
+ size_t sz = __glibc_objsize (__dest); |
|
+ if (sz != (size_t) -1) |
|
+ return __wcscpy_chk (__dest, __src, sz / sizeof (wchar_t)); |
|
return __wcscpy_alias (__dest, __src); |
|
} |
|
|
|
@@ -171,9 +137,9 @@ extern wchar_t *__REDIRECT_NTH (__wcpcpy_alias, |
|
__fortify_function wchar_t * |
|
__NTH (wcpcpy (wchar_t *__restrict __dest, const wchar_t *__restrict __src)) |
|
{ |
|
- if (__glibc_objsize (__dest) != (size_t) -1) |
|
- return __wcpcpy_chk (__dest, __src, |
|
- __glibc_objsize (__dest) / sizeof (wchar_t)); |
|
+ size_t sz = __glibc_objsize (__dest); |
|
+ if (sz != (size_t) -1) |
|
+ return __wcpcpy_chk (__dest, __src, sz / sizeof (wchar_t)); |
|
return __wcpcpy_alias (__dest, __src); |
|
} |
|
|
|
@@ -196,17 +162,9 @@ __fortify_function wchar_t * |
|
__NTH (wcsncpy (wchar_t *__restrict __dest, const wchar_t *__restrict __src, |
|
size_t __n)) |
|
{ |
|
- if (__glibc_objsize (__dest) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __wcsncpy_chk (__dest, __src, __n, |
|
- __glibc_objsize (__dest) / sizeof (wchar_t)); |
|
- if (__n > __glibc_objsize (__dest) / sizeof (wchar_t)) |
|
- return __wcsncpy_chk_warn (__dest, __src, __n, |
|
- (__glibc_objsize (__dest) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __wcsncpy_alias (__dest, __src, __n); |
|
+ return __glibc_fortify_n (wcsncpy, __n, sizeof (wchar_t), |
|
+ __glibc_objsize (__dest), |
|
+ __dest, __src, __n); |
|
} |
|
|
|
|
|
@@ -228,17 +186,9 @@ __fortify_function wchar_t * |
|
__NTH (wcpncpy (wchar_t *__restrict __dest, const wchar_t *__restrict __src, |
|
size_t __n)) |
|
{ |
|
- if (__glibc_objsize (__dest) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n)) |
|
- return __wcpncpy_chk (__dest, __src, __n, |
|
- __glibc_objsize (__dest) / sizeof (wchar_t)); |
|
- if (__n > __glibc_objsize (__dest) / sizeof (wchar_t)) |
|
- return __wcpncpy_chk_warn (__dest, __src, __n, |
|
- (__glibc_objsize (__dest) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __wcpncpy_alias (__dest, __src, __n); |
|
+ return __glibc_fortify_n (wcpncpy, __n, sizeof (wchar_t), |
|
+ __glibc_objsize (__dest), |
|
+ __dest, __src, __n); |
|
} |
|
|
|
|
|
@@ -252,9 +202,9 @@ extern wchar_t *__REDIRECT_NTH (__wcscat_alias, |
|
__fortify_function wchar_t * |
|
__NTH (wcscat (wchar_t *__restrict __dest, const wchar_t *__restrict __src)) |
|
{ |
|
- if (__glibc_objsize (__dest) != (size_t) -1) |
|
- return __wcscat_chk (__dest, __src, |
|
- __glibc_objsize (__dest) / sizeof (wchar_t)); |
|
+ size_t sz = __glibc_objsize (__dest); |
|
+ if (sz != (size_t) -1) |
|
+ return __wcscat_chk (__dest, __src, sz / sizeof (wchar_t)); |
|
return __wcscat_alias (__dest, __src); |
|
} |
|
|
|
@@ -271,9 +221,9 @@ __fortify_function wchar_t * |
|
__NTH (wcsncat (wchar_t *__restrict __dest, const wchar_t *__restrict __src, |
|
size_t __n)) |
|
{ |
|
- if (__glibc_objsize (__dest) != (size_t) -1) |
|
- return __wcsncat_chk (__dest, __src, __n, |
|
- __glibc_objsize (__dest) / sizeof (wchar_t)); |
|
+ size_t sz = __glibc_objsize (__dest); |
|
+ if (sz != (size_t) -1) |
|
+ return __wcsncat_chk (__dest, __src, __n, sz / sizeof (wchar_t)); |
|
return __wcsncat_alias (__dest, __src, __n); |
|
} |
|
|
|
@@ -293,10 +243,10 @@ __fortify_function int |
|
__NTH (swprintf (wchar_t *__restrict __s, size_t __n, |
|
const wchar_t *__restrict __fmt, ...)) |
|
{ |
|
- if (__glibc_objsize (__s) != (size_t) -1 || __USE_FORTIFY_LEVEL > 1) |
|
+ size_t sz = __glibc_objsize (__s); |
|
+ if (sz != (size_t) -1 || __USE_FORTIFY_LEVEL > 1) |
|
return __swprintf_chk (__s, __n, __USE_FORTIFY_LEVEL - 1, |
|
- __glibc_objsize (__s) / sizeof (wchar_t), |
|
- __fmt, __va_arg_pack ()); |
|
+ sz / sizeof (wchar_t), __fmt, __va_arg_pack ()); |
|
return __swprintf_alias (__s, __n, __fmt, __va_arg_pack ()); |
|
} |
|
#elif !defined __cplusplus |
|
@@ -323,10 +273,10 @@ __fortify_function int |
|
__NTH (vswprintf (wchar_t *__restrict __s, size_t __n, |
|
const wchar_t *__restrict __fmt, __gnuc_va_list __ap)) |
|
{ |
|
- if (__glibc_objsize (__s) != (size_t) -1 || __USE_FORTIFY_LEVEL > 1) |
|
+ size_t sz = __glibc_objsize (__s); |
|
+ if (sz != (size_t) -1 || __USE_FORTIFY_LEVEL > 1) |
|
return __vswprintf_chk (__s, __n, __USE_FORTIFY_LEVEL - 1, |
|
- __glibc_objsize (__s) / sizeof (wchar_t), __fmt, |
|
- __ap); |
|
+ sz / sizeof (wchar_t), __fmt, __ap); |
|
return __vswprintf_alias (__s, __n, __fmt, __ap); |
|
} |
|
|
|
@@ -392,18 +342,12 @@ extern wchar_t *__REDIRECT (__fgetws_chk_warn, |
|
__fortify_function __wur wchar_t * |
|
fgetws (wchar_t *__restrict __s, int __n, __FILE *__restrict __stream) |
|
{ |
|
- if (__glibc_objsize (__s) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n) || __n <= 0) |
|
- return __fgetws_chk (__s, __glibc_objsize (__s) / sizeof (wchar_t), |
|
- __n, __stream); |
|
- |
|
- if ((size_t) __n > __glibc_objsize (__s) / sizeof (wchar_t)) |
|
- return __fgetws_chk_warn (__s, |
|
- __glibc_objsize (__s) / sizeof (wchar_t), |
|
- __n, __stream); |
|
- } |
|
- return __fgetws_alias (__s, __n, __stream); |
|
+ size_t sz = __glibc_objsize (__s); |
|
+ if (__glibc_safe_or_unknown_len (__n, sizeof (wchar_t), sz)) |
|
+ return __fgetws_alias (__s, __n, __stream); |
|
+ if (__glibc_unsafe_len (__n, sizeof (wchar_t), sz)) |
|
+ return __fgetws_chk_warn (__s, sz / sizeof (wchar_t), __n, __stream); |
|
+ return __fgetws_chk (__s, sz / sizeof (wchar_t), __n, __stream); |
|
} |
|
|
|
#ifdef __USE_GNU |
|
@@ -424,20 +368,13 @@ extern wchar_t *__REDIRECT (__fgetws_unlocked_chk_warn, |
|
__fortify_function __wur wchar_t * |
|
fgetws_unlocked (wchar_t *__restrict __s, int __n, __FILE *__restrict __stream) |
|
{ |
|
- if (__glibc_objsize (__s) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__n) || __n <= 0) |
|
- return __fgetws_unlocked_chk (__s, |
|
- __glibc_objsize (__s) / sizeof (wchar_t), |
|
- __n, __stream); |
|
- |
|
- if ((size_t) __n > __glibc_objsize (__s) / sizeof (wchar_t)) |
|
- return __fgetws_unlocked_chk_warn (__s, |
|
- (__glibc_objsize (__s) |
|
- / sizeof (wchar_t)), |
|
- __n, __stream); |
|
- } |
|
- return __fgetws_unlocked_alias (__s, __n, __stream); |
|
+ size_t sz = __glibc_objsize (__s); |
|
+ if (__glibc_safe_or_unknown_len (__n, sizeof (wchar_t), sz)) |
|
+ return __fgetws_unlocked_alias (__s, __n, __stream); |
|
+ if (__glibc_unsafe_len (__n, sizeof (wchar_t), sz)) |
|
+ return __fgetws_unlocked_chk_warn (__s, sz / sizeof (wchar_t), __n, |
|
+ __stream); |
|
+ return __fgetws_unlocked_chk (__s, sz / sizeof (wchar_t), __n, __stream); |
|
} |
|
#endif |
|
|
|
@@ -488,18 +425,9 @@ __fortify_function size_t |
|
__NTH (mbsrtowcs (wchar_t *__restrict __dst, const char **__restrict __src, |
|
size_t __len, mbstate_t *__restrict __ps)) |
|
{ |
|
- if (__glibc_objsize (__dst) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __mbsrtowcs_chk (__dst, __src, __len, __ps, |
|
- __glibc_objsize (__dst) / sizeof (wchar_t)); |
|
- |
|
- if (__len > __glibc_objsize (__dst) / sizeof (wchar_t)) |
|
- return __mbsrtowcs_chk_warn (__dst, __src, __len, __ps, |
|
- (__glibc_objsize (__dst) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __mbsrtowcs_alias (__dst, __src, __len, __ps); |
|
+ return __glibc_fortify_n (mbsrtowcs, __len, sizeof (wchar_t), |
|
+ __glibc_objsize (__dst), |
|
+ __dst, __src, __len, __ps); |
|
} |
|
|
|
|
|
@@ -523,17 +451,9 @@ __fortify_function size_t |
|
__NTH (wcsrtombs (char *__restrict __dst, const wchar_t **__restrict __src, |
|
size_t __len, mbstate_t *__restrict __ps)) |
|
{ |
|
- if (__glibc_objsize (__dst) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __wcsrtombs_chk (__dst, __src, __len, __ps, |
|
- __glibc_objsize (__dst)); |
|
- |
|
- if (__len > __glibc_objsize (__dst)) |
|
- return __wcsrtombs_chk_warn (__dst, __src, __len, __ps, |
|
- __glibc_objsize (__dst)); |
|
- } |
|
- return __wcsrtombs_alias (__dst, __src, __len, __ps); |
|
+ return __glibc_fortify (wcsrtombs, __len, sizeof (char), |
|
+ __glibc_objsize (__dst), |
|
+ __dst, __src, __len, __ps); |
|
} |
|
|
|
|
|
@@ -559,18 +479,9 @@ __fortify_function size_t |
|
__NTH (mbsnrtowcs (wchar_t *__restrict __dst, const char **__restrict __src, |
|
size_t __nmc, size_t __len, mbstate_t *__restrict __ps)) |
|
{ |
|
- if (__glibc_objsize (__dst) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __mbsnrtowcs_chk (__dst, __src, __nmc, __len, __ps, |
|
- __glibc_objsize (__dst) / sizeof (wchar_t)); |
|
- |
|
- if (__len > __glibc_objsize (__dst) / sizeof (wchar_t)) |
|
- return __mbsnrtowcs_chk_warn (__dst, __src, __nmc, __len, __ps, |
|
- (__glibc_objsize (__dst) |
|
- / sizeof (wchar_t))); |
|
- } |
|
- return __mbsnrtowcs_alias (__dst, __src, __nmc, __len, __ps); |
|
+ return __glibc_fortify_n (mbsnrtowcs, __len, sizeof (wchar_t), |
|
+ __glibc_objsize (__dst), |
|
+ __dst, __src, __nmc, __len, __ps); |
|
} |
|
|
|
|
|
@@ -596,16 +507,8 @@ __fortify_function size_t |
|
__NTH (wcsnrtombs (char *__restrict __dst, const wchar_t **__restrict __src, |
|
size_t __nwc, size_t __len, mbstate_t *__restrict __ps)) |
|
{ |
|
- if (__glibc_objsize (__dst) != (size_t) -1) |
|
- { |
|
- if (!__builtin_constant_p (__len)) |
|
- return __wcsnrtombs_chk (__dst, __src, __nwc, __len, __ps, |
|
- __glibc_objsize (__dst)); |
|
- |
|
- if (__len > __glibc_objsize (__dst)) |
|
- return __wcsnrtombs_chk_warn (__dst, __src, __nwc, __len, __ps, |
|
- __glibc_objsize (__dst)); |
|
- } |
|
- return __wcsnrtombs_alias (__dst, __src, __nwc, __len, __ps); |
|
+ return __glibc_fortify (wcsnrtombs, __len, sizeof (char), |
|
+ __glibc_objsize (__dst), |
|
+ __dst, __src, __nwc, __len, __ps); |
|
} |
|
#endif
|
|
|