1*8326Smckusick /* @(#)wbuf.c 4.3 (Berkeley) 10/05/82 */ 22003Swnj #include <stdio.h> 3*8326Smckusick #include <sys/types.h> 4*8326Smckusick #include <sys/stat.h> 52003Swnj 62003Swnj char *malloc(); 72003Swnj 82003Swnj _flsbuf(c, iop) 92003Swnj register FILE *iop; 102003Swnj { 112003Swnj register char *base; 122003Swnj register n, rn; 132003Swnj char c1; 14*8326Smckusick int size; 15*8326Smckusick struct stat stbuf; 162003Swnj extern char _sobuf[]; 172003Swnj 183163Stoy if (iop->_flag & _IORW) { 193163Stoy iop->_flag |= _IOWRT; 203163Stoy iop->_flag &= ~_IOEOF; 213163Stoy } 223163Stoy 232003Swnj if ((iop->_flag&_IOWRT)==0) 242003Swnj return(EOF); 252003Swnj tryagain: 262003Swnj if (iop->_flag&_IOLBF) { 272003Swnj base = iop->_base; 282003Swnj *iop->_ptr++ = c; 29*8326Smckusick if (iop->_ptr >= base+iop->_bufsiz || c == '\n') { 302003Swnj n = write(fileno(iop), base, rn = iop->_ptr - base); 312003Swnj iop->_ptr = base; 322003Swnj } else 332003Swnj rn = n = 0; 342003Swnj iop->_cnt = 0; 352003Swnj } else if (iop->_flag&_IONBF) { 362003Swnj c1 = c; 372003Swnj rn = 1; 382003Swnj n = write(fileno(iop), &c1, rn); 392003Swnj iop->_cnt = 0; 402003Swnj } else { 412003Swnj if ((base=iop->_base)==NULL) { 42*8326Smckusick if (fstat(fileno(iop), &stbuf) < 0 || 43*8326Smckusick stbuf.st_blksize <= NULL) 44*8326Smckusick size = BUFSIZ; 45*8326Smckusick else 46*8326Smckusick size = stbuf.st_blksize; 472003Swnj if (iop==stdout) { 482003Swnj if (isatty(fileno(stdout))) 492003Swnj iop->_flag |= _IOLBF; 502003Swnj iop->_base = _sobuf; 512003Swnj iop->_ptr = _sobuf; 52*8326Smckusick iop->_bufsiz = size; 532003Swnj goto tryagain; 542003Swnj } 55*8326Smckusick if ((iop->_base=base=malloc(size)) == NULL) { 562003Swnj iop->_flag |= _IONBF; 572003Swnj goto tryagain; 582003Swnj } 592003Swnj iop->_flag |= _IOMYBUF; 60*8326Smckusick iop->_bufsiz = size; 612003Swnj rn = n = 0; 622003Swnj } else if ((rn = n = iop->_ptr - base) > 0) { 632003Swnj iop->_ptr = base; 642003Swnj n = write(fileno(iop), base, n); 652003Swnj } 66*8326Smckusick iop->_cnt = iop->_bufsiz-1; 672003Swnj *base++ = c; 682003Swnj iop->_ptr = base; 692003Swnj } 702003Swnj if (rn != n) { 712003Swnj iop->_flag |= _IOERR; 722003Swnj return(EOF); 732003Swnj } 742003Swnj return(c); 752003Swnj } 762003Swnj 772003Swnj fflush(iop) 782003Swnj register struct _iobuf *iop; 792003Swnj { 802003Swnj register char *base; 812003Swnj register n; 822003Swnj 832003Swnj if ((iop->_flag&(_IONBF|_IOWRT))==_IOWRT 842003Swnj && (base=iop->_base)!=NULL && (n=iop->_ptr-base)>0) { 852003Swnj iop->_ptr = base; 86*8326Smckusick iop->_cnt = (iop->_flag&(_IOLBF|_IONBF)) ? 0 : iop->_bufsiz; 872003Swnj if (write(fileno(iop), base, n)!=n) { 882003Swnj iop->_flag |= _IOERR; 892003Swnj return(EOF); 902003Swnj } 912003Swnj } 922003Swnj return(0); 932003Swnj } 942003Swnj 952003Swnj /* 962003Swnj * Flush buffers on exit 972003Swnj */ 982003Swnj 992003Swnj _cleanup() 1002003Swnj { 1012003Swnj register struct _iobuf *iop; 1022003Swnj extern struct _iobuf *_lastbuf; 1032003Swnj 1042003Swnj for (iop = _iob; iop < _lastbuf; iop++) 1052003Swnj fclose(iop); 1062003Swnj } 1072003Swnj 1082003Swnj fclose(iop) 1092003Swnj register struct _iobuf *iop; 1102003Swnj { 1112003Swnj register r; 1122003Swnj 1132003Swnj r = EOF; 1143163Stoy if (iop->_flag&(_IOREAD|_IOWRT|_IORW) && (iop->_flag&_IOSTRG)==0) { 1152003Swnj r = fflush(iop); 1162003Swnj if (close(fileno(iop)) < 0) 1172003Swnj r = EOF; 1182003Swnj if (iop->_flag&_IOMYBUF) 1192003Swnj free(iop->_base); 120*8326Smckusick if (iop->_flag&(_IOMYBUF|_IONBF|_IOLBF)) { 1212003Swnj iop->_base = NULL; 122*8326Smckusick iop->_bufsiz = 0; 123*8326Smckusick } 1242003Swnj } 1253163Stoy iop->_flag &= ~(_IOREAD|_IOWRT|_IOLBF|_IONBF|_IOMYBUF|_IOERR|_IOEOF|_IOSTRG|_IORW); 1262003Swnj iop->_cnt = 0; 1272003Swnj return(r); 1282003Swnj } 129