mirror of
git://sourceware.org/git/glibc.git
synced 2024-12-21 04:31:04 +08:00
76 lines
2.4 KiB
C
76 lines
2.4 KiB
C
/* Copyright (C) 1993-2014 Free Software Foundation, Inc.
|
|
This file is part of the GNU C Library.
|
|
|
|
The GNU C Library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
The GNU C Library 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
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with the GNU C Library; if not, see
|
|
<http://www.gnu.org/licenses/>.
|
|
|
|
As a special exception, if you link the code in this file with
|
|
files compiled with a GNU compiler to produce an executable,
|
|
that does not cause the resulting executable to be covered by
|
|
the GNU Lesser General Public License. This exception does not
|
|
however invalidate any other reasons why the executable file
|
|
might be covered by the GNU Lesser General Public License.
|
|
This exception applies to code released by its copyright holders
|
|
in files containing the exception. */
|
|
|
|
#include "../libio/libioP.h"
|
|
#include <limits.h>
|
|
|
|
char *
|
|
__gets_chk (char *buf, size_t size)
|
|
{
|
|
_IO_size_t count;
|
|
int ch;
|
|
char *retval;
|
|
|
|
if (size == 0)
|
|
__chk_fail ();
|
|
|
|
_IO_acquire_lock (_IO_stdin);
|
|
ch = _IO_getc_unlocked (_IO_stdin);
|
|
if (ch == EOF)
|
|
{
|
|
retval = NULL;
|
|
goto unlock_return;
|
|
}
|
|
if (ch == '\n')
|
|
count = 0;
|
|
else
|
|
{
|
|
/* This is very tricky since a file descriptor may be in the
|
|
non-blocking mode. The error flag doesn't mean much in this
|
|
case. We return an error only when there is a new error. */
|
|
int old_error = _IO_stdin->_IO_file_flags & _IO_ERR_SEEN;
|
|
_IO_stdin->_IO_file_flags &= ~_IO_ERR_SEEN;
|
|
buf[0] = (char) ch;
|
|
count = _IO_getline (_IO_stdin, buf + 1, size - 1, '\n', 0) + 1;
|
|
if (_IO_stdin->_IO_file_flags & _IO_ERR_SEEN)
|
|
{
|
|
retval = NULL;
|
|
goto unlock_return;
|
|
}
|
|
else
|
|
_IO_stdin->_IO_file_flags |= old_error;
|
|
}
|
|
if (count >= size)
|
|
__chk_fail ();
|
|
buf[count] = 0;
|
|
retval = buf;
|
|
unlock_return:
|
|
_IO_release_lock (_IO_stdin);
|
|
return retval;
|
|
}
|
|
|
|
link_warning (__gets_chk, "the `gets' function is dangerous and should not be used.")
|