stdio/fgets.c

   1 /* Copyright (C) 1991 Free Software Foundation, Inc.
   2 This file is part of the GNU C Library.
   3
   4 The GNU C Library is free software; you can redistribute it and/or
   5 modify it under the terms of the GNU Library General Public License as
   6 published by the Free Software Foundation; either version 2 of the
   7 License, or (at your option) any later version.
   8
   9 The GNU C Library is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 Library General Public License for more details.
  13
  14 You should have received a copy of the GNU Library General Public
  15 License along with the GNU C Library; see the file COPYING.LIB.  If
  16 not, write to the Free Software Foundation, Inc., 675 Mass Ave,
  17 Cambridge, MA 02139, USA.  */
  18
  19 #include <ansidecl.h>
  20 #include <errno.h>
  21 #include <stdio.h>
  22 #include <string.h>
  23
  24 /* Get a newline-terminated string of not more than
  25    N - 1 characters from `stream' and put it in S.  */
  26 char *
  27 DEFUN(fgets, (s, n, stream), char *s AND size_t n AND register FILE *stream)
  28 {
  29   register char *p = s;
  30
  31   if (!__validfp(stream))
  32     {
  33       errno = EINVAL;
  34       return NULL;
  35     }
  36
  37   if (s == NULL || n == 0 || ferror(stream))
  38     return NULL;
  39
  40   if (stream->__buffer == NULL && stream->__userbuf)
  41     {
  42       /* Unbuffered stream.  Not much optimization to do.  */
  43       register int c;
  44       while (--n > 0 && (c = getc(stream)) != EOF)
  45         if ((*p++ = c) == '\n')
  46           break;
  47       *p = '\0';
  48       return s;
  49     }
  50
  51   if (!stream->__seen || stream->__buffer == NULL || stream->__pushed_back)
  52     {
  53       /* Do one with getc to allocate a buffer.  */
  54       int c = getc (stream);
  55       if (c == EOF)
  56         return NULL;
  57       *p++ = c;
  58       if (c == '\n')
  59         {
  60           *p = '\0';
  61           return s;
  62         }
  63     }
  64
  65   while (n > 0)
  66     {
  67       size_t i;
  68       char *found;
  69
  70       i = stream->__get_limit - stream->__bufp;
  71       if (i == 0)
  72         {
  73           /* Refill the buffer.  */
  74           int c = __fillbf(stream);
  75           if (c == EOF)
  76             break;
  77           *p++ = c;
  78           if (c == '\n')
  79             {
  80               *p = '\0';
  81               return s;
  82             }
  83         }
  84
  85       if (i > n)
  86         i = n;
  87
  88       found = (char *) __memccpy((PTR) p, stream->__bufp, '\n', i);
  89
  90       if (found != NULL)
  91         {
  92           stream->__bufp += found - p;
  93           p = found;
  94           break;
  95         }
  96
  97       stream->__bufp += i;
  98       n -= i;
  99       p += i;
 100     }
 101
 102   *p = '\0';
 103   return (p == s || ferror(stream)) ? NULL : s;
 104 }