Formerly ../stdio/fgets.c.~17~
[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   /* Leave space for the null.  */
66   --n;
67
68   while (n > 0)
69     {
70       size_t i;
71       char *found;
72
73       i = stream->__get_limit - stream->__bufp; 
74       if (i == 0)
75         {
76           /* Refill the buffer.  */
77           int c = __fillbf(stream);
78           if (c == EOF)
79             break; 
80           *p++ = c;
81           if (c == '\n')
82             {
83               *p = '\0';
84               return s;
85             }
86         }
87
88       if (i > n)
89         i = n;
90
91       found = (char *) __memccpy((PTR) p, stream->__bufp, '\n', i);
92
93       if (found != NULL)
94         {
95           stream->__bufp += found - p;
96           p = found;
97           break;
98         }
99
100       stream->__bufp += i;
101       n -= i;
102       p += i;
103     }
104
105   *p = '\0';
106   return (p == s || ferror(stream)) ? NULL : s;
107 }