#ifndef FORTIFY_WCHAR_H_ #define FORTIFY_WCHAR_H_ #include_next #include_next #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); } static inline __attribute__ ((always_inline)) size_t __fortify_mbsnrtowcs(wchar_t *d, const char **s, size_t n, size_t wn, mbstate_t *st) { size_t bos = __builtin_object_size(d, 0); size_t actual; if (wn > (size_t)-1 / sizeof(wchar_t)) __builtin_trap(); actual = wn * sizeof(wchar_t); if (actual > n) actual = n; if (actual > bos) __builtin_trap(); return mbsnrtowcs(d, s, n, wn, st); } static inline __attribute__ ((always_inline)) size_t __fortify_mbsrtowcs(wchar_t *d, const char **s, size_t wn, mbstate_t *st) { size_t bos = __builtin_object_size(d, 0); size_t r; bos /= sizeof(wchar_t); r = mbsrtowcs(d, s, wn > bos ? bos : wn, st); if (bos < wn && d && *s && r != (size_t)-1) __builtin_trap(); return r; } static inline __attribute__ ((always_inline)) size_t __fortify_mbstowcs(wchar_t *ws, const char *s, size_t wn) { size_t bos = __builtin_object_size(ws, 0); if (wn > bos / sizeof(wchar_t)) __builtin_trap(); return mbstowcs(ws, s, wn); } static inline __attribute__ ((always_inline)) size_t __fortify_wcrtomb(char *s, wchar_t wc, mbstate_t *st) { size_t bos = __builtin_object_size(s, 0); if (MB_CUR_MAX > bos) __builtin_trap(); return wcrtomb(s, wc, st); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wcscat(wchar_t *d, const wchar_t *s) { size_t bos = __builtin_object_size(d, 0); if (wcslen(s) + wcslen(d) + 1 > bos / sizeof(wchar_t)) __builtin_trap(); return wcscat(d, s); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wcscpy(wchar_t *d, const wchar_t *s) { size_t bos = __builtin_object_size(d, 0); if (wcslen(s) + 1 > bos / sizeof(wchar_t)) __builtin_trap(); return wcscpy(d, s); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wcsncat(wchar_t *d, const wchar_t *s, size_t n) { size_t bos = __builtin_object_size(d, 0); size_t slen, dlen; if (n > bos / sizeof(wchar_t)) { slen = wcslen(s); dlen = wcslen(d); if (slen > n) slen = n; if (slen + dlen + 1 > bos / sizeof(wchar_t)) __builtin_trap(); } return wcsncat(d, s, n); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wcsncpy(wchar_t *d, const wchar_t *s, size_t n) { size_t bos = __builtin_object_size(d, 0); if (n > bos / sizeof(wchar_t)) __builtin_trap(); return wcsncpy(d, s, n); } static inline __attribute__ ((always_inline)) size_t __fortify_wcsnrtombs(char *d, const wchar_t **s, size_t wn, size_t n, mbstate_t *st) { size_t bos = __builtin_object_size(d, 0); size_t actual; if (wn > (size_t)-1 / sizeof(wchar_t)) __builtin_trap(); actual = wn * sizeof(wchar_t); if (actual > n) actual = n; if (actual > bos) __builtin_trap(); return wcsnrtombs(d, s, wn, n, st); } static inline __attribute__ ((always_inline)) size_t __fortify_wcsrtombs(char *d, const wchar_t **s, size_t n, mbstate_t *st) { size_t bos = __builtin_object_size(d, 0); size_t r; r = wcsrtombs(d, s, n > bos ? bos : n, st); if (bos < n && d && *s && r != (size_t)-1) __builtin_trap(); return r; } static inline __attribute__ ((always_inline)) size_t __fortify_wcstombs(char *s, const wchar_t *ws, size_t n) { size_t bos = __builtin_object_size(s, 0); if (n > bos) __builtin_trap(); return wcstombs(s, ws, n); } static inline __attribute__ ((always_inline)) int __fortify_wctomb(char *s, wchar_t wc) { size_t bos = __builtin_object_size(s, 0); if (MB_CUR_MAX > bos) __builtin_trap(); return wctomb(s, wc); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wmemcpy(wchar_t *d, const wchar_t *s, size_t n) { size_t bos = __builtin_object_size(d, 0); if (n > bos / sizeof(wchar_t)) __builtin_trap(); return wmemcpy(d, s, n); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wmemmove(wchar_t *d, const wchar_t *s, size_t n) { size_t bos = __builtin_object_size(d, 0); if (n > bos / sizeof(wchar_t)) __builtin_trap(); return wmemmove(d, s, n); } static inline __attribute__ ((always_inline)) wchar_t * __fortify_wmemset(wchar_t *s, wchar_t c, size_t n) { size_t bos = __builtin_object_size(s, 0); if (n > bos / sizeof(wchar_t)) __builtin_trap(); return wmemset(s, c, n); } #undef fgetws #define fgetws(s, n, fp) __fortify_fgetws(s, n, fp) #undef mbsnrtowcs #define mbsnrtowcs(d, s, n, wn, st) __fortify_mbsnrtowcs(d, s, n, wn, st) #undef mbsrtowcs #define mbsrtowcs(d, s, wn, st) __fortify_mbsrtowcs(d, s, wn, st) #undef mbstowcs #define mbstowcs(ws, s, wn) __fortify_mbstowcs(ws, s, wn) #undef wcrtomb #define wcrtomb(s, wc, st) __fortify_wcrtomb(s, wc, st) #undef wcscat #define wcscat(d, s) __fortify_wcscat(d, s) #undef wcscpy #define wcscpy(d, s) __fortify_wcscpy(d, s) #undef wcsncat #define wcsncat(d, s, n) __fortify_wcsncat(d, s, n) #undef wcsncpy #define wcsncpy(d, s, n) __fortify_wcsncpy(d, s, n) #undef wcsnrtombs #define wcsnrtombs(d, s, wn, n, st) __fortify_wcsnrtombs(d, s, wn, n, st) #undef wcsrtombs #define wcsrtombs(d, s, n, st) __fortify_wcsrtombs(d, s, n, st) #undef wcstombs #define wcstombs(s, ws, n) __fortify_wcstombs(s, ws, n) #undef wctomb #define wctomb(s, wc) __fortify_wctomb(s, wc) #undef wmemcpy #define wmemcpy(d, s, n) __fortify_wmemcpy(d, s, n) #undef wmemmove #define wmemmove(d, s, n) __fortify_wmemmove(d, s, n) #undef wmemset #define wmemset(s, c, n) __fortify_wmemset(s, c, n) #endif #endif