stdio/common/fgets.c

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 1989 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*      Copyright (c) 1984 AT&T */
/*        All Rights Reserved   */

#pragma ident   "%Z%%M% %I% %E% SMI"  /* from S5R2 3.3 */

/*LINTLIBRARY*/
/*
 * This version reads directly from the buffer rather than looping on getc.
 * Ptr args aren't checked for NULL because the program would be a
 * catastrophic mess anyway.  Better to abort than just to return NULL.
 */
#include <stdio.h>
#include "stdiom.h"

#define MIN(x, y)   (x < y ? x : y)

extern int _filbuf();
extern char *memccpy();

char *
fgets(ptr, size, iop)
char *ptr;
register int size;
register FILE *iop;
{
    char *p, *ptr0 = ptr;
    register int n;

    if ( !(iop->_flag & (_IOREAD|_IORW)) ) {
        iop->_flag |= _IOERR;
        return (NULL);
    }

    for (size--; size > 0; size -= n) {
        if (iop->_cnt <= 0) { /* empty buffer */
            if (_filbuf(iop) == EOF) {
                if (ptr0 == ptr)
                    return (NULL);
                break; /* no more data */
            }
            iop->_ptr--;
            iop->_cnt++;
        }
        n = MIN(size, iop->_cnt);
        if ((p = memccpy(ptr, (char *) iop->_ptr, '\n', n)) != NULL)
            n = p - ptr;
        ptr += n;
        iop->_cnt -= n;
        iop->_ptr += n;
        _BUFSYNC(iop);
        if (p != NULL)
            break; /* found '\n' in buffer */
    }
    *ptr = '\0';
    return (ptr0);
}