]>
Commit | Line | Data |
---|---|---|
77008252 JJ |
1 | /* Checking gets. |
2 | Copyright (C) 2005 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 2, 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 | You should have received a copy of the GNU General Public License | |
26 | along with GCC; see the file COPYING. If not, write to the Free | |
cf398b1e KC |
27 | Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA |
28 | 02110-1301, USA. */ | |
77008252 JJ |
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 | #include "config.h" | |
37 | #include <ssp/ssp.h> | |
38 | #include <stdarg.h> | |
39 | #ifdef HAVE_STDLIB_H | |
40 | # include <stdlib.h> | |
41 | #endif | |
42 | #ifdef HAVE_ALLOCA_H | |
43 | # include <alloca.h> | |
44 | #endif | |
45 | #ifdef HAVE_LIMITS_H | |
46 | # include <limits.h> | |
47 | #endif | |
48 | #ifdef HAVE_STDIO_H | |
49 | # include <stdio.h> | |
50 | #endif | |
51 | #ifdef HAVE_STRING_H | |
52 | # include <string.h> | |
53 | #endif | |
54 | ||
55 | extern void __chk_fail (void) __attribute__((__noreturn__)); | |
56 | ||
57 | char * | |
58 | __gets_chk (char *s, size_t slen) | |
59 | { | |
60 | char *ret, *buf; | |
61 | ||
62 | if (slen >= (size_t) INT_MAX) | |
63 | return gets (s); | |
64 | ||
65 | if (slen <= 8192) | |
66 | buf = alloca (slen + 1); | |
67 | else | |
68 | buf = malloc (slen + 1); | |
69 | if (buf == NULL) | |
70 | return gets (s); | |
71 | ||
72 | ret = fgets (buf, (int) (slen + 1), stdin); | |
73 | if (ret != NULL) | |
74 | { | |
75 | size_t len = strlen (buf); | |
76 | if (len > 0 && buf[len - 1] == '\n') | |
77 | --len; | |
78 | if (len == slen) | |
79 | __chk_fail (); | |
80 | memcpy (s, buf, len); | |
81 | s[len] = '\0'; | |
82 | ret = s; | |
83 | } | |
84 | ||
85 | if (slen > 8192) | |
86 | free (buf); | |
87 | return ret; | |
88 | } |