Add copyright

[unix-history] / usr / src / lib / libc / stdio / wbuf.c

/*
* Copyright (c) 1980 Regents of the University of California.
* All rights reserved.  The Berkeley software License Agreement
* specifies the terms and conditions for redistribution.
*/

#ifndef lint
static char sccsid[] = "@(#)wbuf.c      5.1 (Berkeley) %G%";
#endif not lint

#include        <stdio.h>
#include        <sys/types.h>
#include        <sys/stat.h>

char    *malloc();

_flsbuf(c, iop)
unsigned char c;
register FILE *iop;
{
       register char *base;
       register n, rn;
       char c1;
       int size;
       struct stat stbuf;

       if (iop->_flag & _IORW) {
               iop->_flag |= _IOWRT;
               iop->_flag &= ~(_IOEOF|_IOREAD);
       }

       if ((iop->_flag&_IOWRT)==0)
               return(EOF);
tryagain:
       if (iop->_flag&_IOLBF) {
               base = iop->_base;
               *iop->_ptr++ = c;
               if (iop->_ptr >= base+iop->_bufsiz || c == '\n') {
                       n = write(fileno(iop), base, rn = iop->_ptr - base);
                       iop->_ptr = base;
                       iop->_cnt = 0;
               } else
                       rn = n = 0;
       } else if (iop->_flag&_IONBF) {
               c1 = c;
               rn = 1;
               n = write(fileno(iop), &c1, rn);
               iop->_cnt = 0;
       } else {
               if ((base=iop->_base)==NULL) {
                       if (fstat(fileno(iop), &stbuf) < 0 ||
                           stbuf.st_blksize <= NULL)
                               size = BUFSIZ;
                       else
                               size = stbuf.st_blksize;
                       if ((iop->_base=base=malloc(size)) == NULL) {
                               iop->_flag |= _IONBF;
                               goto tryagain;
                       }
                       iop->_flag |= _IOMYBUF;
                       iop->_bufsiz = size;
                       if (iop==stdout && isatty(fileno(stdout))) {
                               iop->_flag |= _IOLBF;
                               iop->_ptr = base;
                               goto tryagain;
                       }
                       rn = n = 0;
               } else if ((rn = n = iop->_ptr - base) > 0) {
                       iop->_ptr = base;
                       n = write(fileno(iop), base, n);
               }
               iop->_cnt = iop->_bufsiz-1;
               *base++ = c;
               iop->_ptr = base;
       }
       if (rn != n) {
               iop->_flag |= _IOERR;
               return(EOF);
       }
       return(c);
}

fflush(iop)
register FILE *iop;
{
       register char *base;
       register n;

       if ((iop->_flag&(_IONBF|_IOWRT))==_IOWRT
        && (base=iop->_base)!=NULL && (n=iop->_ptr-base)>0) {
               iop->_ptr = base;
               iop->_cnt = (iop->_flag&(_IOLBF|_IONBF)) ? 0 : iop->_bufsiz;
               if (write(fileno(iop), base, n)!=n) {
                       iop->_flag |= _IOERR;
                       return(EOF);
               }
       }
       return(0);
}

fclose(iop)
       register FILE *iop;
{
       register int r;

       r = EOF;
       if (iop->_flag&(_IOREAD|_IOWRT|_IORW) && (iop->_flag&_IOSTRG)==0) {
               r = fflush(iop);
               if (close(fileno(iop)) < 0)
                       r = EOF;
               if (iop->_flag&_IOMYBUF)
                       free(iop->_base);
       }
       iop->_cnt = 0;
       iop->_base = (char *)NULL;
       iop->_ptr = (char *)NULL;
       iop->_bufsiz = 0;
       iop->_flag = 0;
       iop->_file = 0;
       return(r);
}