218461c6fe20852aaeacd2027a12c51c6b5407c2
[kopensolaris-gnu/glibc.git] / 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 }