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