1*38fd1498Szrj /* Checking gets.
2*38fd1498Szrj Copyright (C) 2005-2018 Free Software Foundation, Inc.
3*38fd1498Szrj
4*38fd1498Szrj This file is part of GCC.
5*38fd1498Szrj
6*38fd1498Szrj GCC is free software; you can redistribute it and/or modify it under
7*38fd1498Szrj the terms of the GNU General Public License as published by the Free
8*38fd1498Szrj Software Foundation; either version 3, or (at your option) any later
9*38fd1498Szrj version.
10*38fd1498Szrj
11*38fd1498Szrj In addition to the permissions in the GNU General Public License, the
12*38fd1498Szrj Free Software Foundation gives you unlimited permission to link the
13*38fd1498Szrj compiled version of this file into combinations with other programs,
14*38fd1498Szrj and to distribute those combinations without any restriction coming
15*38fd1498Szrj from the use of this file. (The General Public License restrictions
16*38fd1498Szrj do apply in other respects; for example, they cover modification of
17*38fd1498Szrj the file, and distribution when not linked into a combine
18*38fd1498Szrj executable.)
19*38fd1498Szrj
20*38fd1498Szrj GCC is distributed in the hope that it will be useful, but WITHOUT ANY
21*38fd1498Szrj WARRANTY; without even the implied warranty of MERCHANTABILITY or
22*38fd1498Szrj FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
23*38fd1498Szrj for more details.
24*38fd1498Szrj
25*38fd1498Szrj Under Section 7 of GPL version 3, you are granted additional
26*38fd1498Szrj permissions described in the GCC Runtime Library Exception, version
27*38fd1498Szrj 3.1, as published by the Free Software Foundation.
28*38fd1498Szrj
29*38fd1498Szrj You should have received a copy of the GNU General Public License and
30*38fd1498Szrj a copy of the GCC Runtime Library Exception along with this program;
31*38fd1498Szrj see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
32*38fd1498Szrj <http://www.gnu.org/licenses/>. */
33*38fd1498Szrj
34*38fd1498Szrj
35*38fd1498Szrj #include "config.h"
36*38fd1498Szrj #include <ssp/ssp.h>
37*38fd1498Szrj #include <stdarg.h>
38*38fd1498Szrj #ifdef HAVE_STDLIB_H
39*38fd1498Szrj # include <stdlib.h>
40*38fd1498Szrj #endif
41*38fd1498Szrj #ifdef HAVE_ALLOCA_H
42*38fd1498Szrj # include <alloca.h>
43*38fd1498Szrj #endif
44*38fd1498Szrj #ifdef HAVE_LIMITS_H
45*38fd1498Szrj # include <limits.h>
46*38fd1498Szrj #endif
47*38fd1498Szrj #ifdef HAVE_STDIO_H
48*38fd1498Szrj # include <stdio.h>
49*38fd1498Szrj #endif
50*38fd1498Szrj #ifdef HAVE_STRING_H
51*38fd1498Szrj # include <string.h>
52*38fd1498Szrj #endif
53*38fd1498Szrj
54*38fd1498Szrj #if !(!defined __USE_ISOC11 \
55*38fd1498Szrj || (defined __cplusplus && __cplusplus <= 201103L))
56*38fd1498Szrj extern char *gets (char *);
57*38fd1498Szrj #endif
58*38fd1498Szrj
59*38fd1498Szrj extern void __chk_fail (void) __attribute__((__noreturn__));
60*38fd1498Szrj
61*38fd1498Szrj char *
__gets_chk(char * s,size_t slen)62*38fd1498Szrj __gets_chk (char *s, size_t slen)
63*38fd1498Szrj {
64*38fd1498Szrj char *ret, *buf;
65*38fd1498Szrj
66*38fd1498Szrj if (slen >= (size_t) INT_MAX)
67*38fd1498Szrj return gets (s);
68*38fd1498Szrj
69*38fd1498Szrj if (slen <= 8192)
70*38fd1498Szrj buf = alloca (slen + 1);
71*38fd1498Szrj else
72*38fd1498Szrj buf = malloc (slen + 1);
73*38fd1498Szrj if (buf == NULL)
74*38fd1498Szrj return gets (s);
75*38fd1498Szrj
76*38fd1498Szrj ret = fgets (buf, (int) (slen + 1), stdin);
77*38fd1498Szrj if (ret != NULL)
78*38fd1498Szrj {
79*38fd1498Szrj size_t len = strlen (buf);
80*38fd1498Szrj if (len > 0 && buf[len - 1] == '\n')
81*38fd1498Szrj --len;
82*38fd1498Szrj if (len == slen)
83*38fd1498Szrj __chk_fail ();
84*38fd1498Szrj memcpy (s, buf, len);
85*38fd1498Szrj s[len] = '\0';
86*38fd1498Szrj ret = s;
87*38fd1498Szrj }
88*38fd1498Szrj
89*38fd1498Szrj if (slen > 8192)
90*38fd1498Szrj free (buf);
91*38fd1498Szrj return ret;
92*38fd1498Szrj }
93