Mercurial > hg > CbC > CbC_gcc
comparison libssp/gets-chk.c @ 0:a06113de4d67
first commit
author | kent <kent@cr.ie.u-ryukyu.ac.jp> |
---|---|
date | Fri, 17 Jul 2009 14:47:48 +0900 |
parents | |
children | 04ced10e8804 |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:a06113de4d67 |
---|---|
1 /* Checking gets. | |
2 Copyright (C) 2005, 2009 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 #include "config.h" | |
36 #include <ssp/ssp.h> | |
37 #include <stdarg.h> | |
38 #ifdef HAVE_STDLIB_H | |
39 # include <stdlib.h> | |
40 #endif | |
41 #ifdef HAVE_ALLOCA_H | |
42 # include <alloca.h> | |
43 #endif | |
44 #ifdef HAVE_LIMITS_H | |
45 # include <limits.h> | |
46 #endif | |
47 #ifdef HAVE_STDIO_H | |
48 # include <stdio.h> | |
49 #endif | |
50 #ifdef HAVE_STRING_H | |
51 # include <string.h> | |
52 #endif | |
53 | |
54 extern void __chk_fail (void) __attribute__((__noreturn__)); | |
55 | |
56 char * | |
57 __gets_chk (char *s, size_t slen) | |
58 { | |
59 char *ret, *buf; | |
60 | |
61 if (slen >= (size_t) INT_MAX) | |
62 return gets (s); | |
63 | |
64 if (slen <= 8192) | |
65 buf = alloca (slen + 1); | |
66 else | |
67 buf = malloc (slen + 1); | |
68 if (buf == NULL) | |
69 return gets (s); | |
70 | |
71 ret = fgets (buf, (int) (slen + 1), stdin); | |
72 if (ret != NULL) | |
73 { | |
74 size_t len = strlen (buf); | |
75 if (len > 0 && buf[len - 1] == '\n') | |
76 --len; | |
77 if (len == slen) | |
78 __chk_fail (); | |
79 memcpy (s, buf, len); | |
80 s[len] = '\0'; | |
81 ret = s; | |
82 } | |
83 | |
84 if (slen > 8192) | |
85 free (buf); | |
86 return ret; | |
87 } |