]> oss.titaniummirror.com Git - msp430-gcc.git/blobdiff - libssp/gets-chk.c
Imported gcc-4.4.3
[msp430-gcc.git] / libssp / gets-chk.c
diff --git a/libssp/gets-chk.c b/libssp/gets-chk.c
new file mode 100644 (file)
index 0000000..053c446
--- /dev/null
@@ -0,0 +1,87 @@
+/* Checking gets.
+   Copyright (C) 2005, 2009 Free Software Foundation, Inc.
+
+This file is part of GCC.
+
+GCC is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 3, or (at your option) any later
+version.
+
+In addition to the permissions in the GNU General Public License, the
+Free Software Foundation gives you unlimited permission to link the
+compiled version of this file into combinations with other programs,
+and to distribute those combinations without any restriction coming
+from the use of this file.  (The General Public License restrictions
+do apply in other respects; for example, they cover modification of
+the file, and distribution when not linked into a combine
+executable.)
+
+GCC is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+for more details.
+
+Under Section 7 of GPL version 3, you are granted additional
+permissions described in the GCC Runtime Library Exception, version
+3.1, as published by the Free Software Foundation.
+
+You should have received a copy of the GNU General Public License and
+a copy of the GCC Runtime Library Exception along with this program;
+see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
+<http://www.gnu.org/licenses/>.  */
+
+
+#include "config.h"
+#include <ssp/ssp.h>
+#include <stdarg.h>
+#ifdef HAVE_STDLIB_H
+# include <stdlib.h>
+#endif
+#ifdef HAVE_ALLOCA_H
+# include <alloca.h>
+#endif
+#ifdef HAVE_LIMITS_H
+# include <limits.h>
+#endif
+#ifdef HAVE_STDIO_H
+# include <stdio.h>
+#endif
+#ifdef HAVE_STRING_H
+# include <string.h>
+#endif
+
+extern void __chk_fail (void) __attribute__((__noreturn__));
+
+char *
+__gets_chk (char *s, size_t slen)
+{
+  char *ret, *buf;
+
+  if (slen >= (size_t) INT_MAX)
+    return gets (s);
+
+  if (slen <= 8192)
+    buf = alloca (slen + 1);
+  else
+    buf = malloc (slen + 1);
+  if (buf == NULL)
+    return gets (s);
+
+  ret = fgets (buf, (int) (slen + 1), stdin);
+  if (ret != NULL)
+    {
+      size_t len = strlen (buf);
+      if (len > 0 && buf[len - 1] == '\n')
+        --len;
+      if (len == slen)
+        __chk_fail ();
+      memcpy (s, buf, len);
+      s[len] = '\0';
+      ret = s;
+    }
+
+  if (slen > 8192)
+    free (buf);
+  return ret;
+}