libssp/ssp/stdio.h

   1 /* Checking macros for stdio functions.
   2    Copyright (C) 2004-2018 Free Software Foundation, Inc.
   3
   4 This file is part of GCC.
   5
   6 GCC is free software; you can redistribute it and/or modify it under
   7 the terms of the GNU General Public License as published by the Free
   8 Software Foundation; either version 3, or (at your option) any later
   9 version.
  10
  11 In addition to the permissions in the GNU General Public License, the
  12 Free Software Foundation gives you unlimited permission to link the
  13 compiled version of this file into combinations with other programs,
  14 and to distribute those combinations without any restriction coming
  15 from the use of this file.  (The General Public License restrictions
  16 do apply in other respects; for example, they cover modification of
  17 the file, and distribution when not linked into a combine
  18 executable.)
  19
  20 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
  21 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  22 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  23 for more details.
  24
  25 Under Section 7 of GPL version 3, you are granted additional
  26 permissions described in the GCC Runtime Library Exception, version
  27 3.1, as published by the Free Software Foundation.
  28
  29 You should have received a copy of the GNU General Public License and
  30 a copy of the GCC Runtime Library Exception along with this program;
  31 see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
  32 <http://www.gnu.org/licenses/>.  */
  33
  34
  35 #ifndef _SSP_STDIO_H
  36 #define _SSP_STDIO_H 1
  37
  38 #include <ssp.h>
  39 #include_next <stdio.h>
  40
  41 #if __SSP_FORTIFY_LEVEL > 0
  42
  43 #include <stdarg.h>
  44
  45 #undef sprintf
  46 #undef vsprintf
  47 #undef snprintf
  48 #undef vsnprintf
  49 #undef gets
  50 #undef fgets
  51
  52 extern int __sprintf_chk (char *__restrict__ __s, int __flag, size_t __slen,
  53                           __const char *__restrict__ __format, ...);
  54 extern int __vsprintf_chk (char *__restrict__ __s, int __flag, size_t __slen,
  55                            __const char *__restrict__ __format,
  56                            va_list __ap);
  57
  58 #define sprintf(str, ...) \
  59   __builtin___sprintf_chk (str, 0, __ssp_bos (str), \
  60                            __VA_ARGS__)
  61 #define vsprintf(str, fmt, ap) \
  62   __builtin___vsprintf_chk (str, 0, __ssp_bos (str), fmt, ap)
  63
  64 extern int __snprintf_chk (char *__restrict__ __s, size_t __n, int __flag,
  65                            size_t __slen, __const char *__restrict__ __format,
  66                            ...);
  67 extern int __vsnprintf_chk (char *__restrict__ __s, size_t __n, int __flag,
  68                             size_t __slen, __const char *__restrict__ __format,
  69                             va_list __ap);
  70
  71 #define snprintf(str, len, ...) \
  72   __builtin___snprintf_chk (str, len, 0, __ssp_bos (str), __VA_ARGS__)
  73 #define vsnprintf(str, len, fmt, ap) \
  74   __builtin___vsnprintf_chk (str, len, 0, __ssp_bos (str), fmt, ap)
  75
  76 extern char *__gets_chk (char *__str, size_t);
  77 extern char *__SSP_REDIRECT (__gets_alias, (char *__str), gets);
  78
  79 extern inline __attribute__((__always_inline__)) char *
  80 gets (char *__str)
  81 {
  82   if (__ssp_bos (__str) != (size_t) -1)
  83     return __gets_chk (__str, __ssp_bos (__str));
  84   return __gets_alias (__str);
  85 }
  86
  87 extern char *__SSP_REDIRECT (__fgets_alias,
  88                              (char *__restrict__ __s, int __n,
  89                               FILE *__restrict__ __stream), fgets);
  90
  91 extern inline __attribute__((__always_inline__)) char *
  92 fgets (char *__restrict__ __s, int __n, FILE *__restrict__ __stream)
  93 {
  94   if (__ssp_bos (__s) != (size_t) -1 && (size_t) __n > __ssp_bos (__s))
  95     __chk_fail ();
  96   return __fgets_alias (__s, __n, __stream);
  97 }
  98
  99 #endif /* __SSP_FORTIFY_LEVEL > 0 */
 100 #endif /* _SSP_STDIO_H */