[BACK]Return to fvwrite.c CVS log [TXT][DIR] Up to [local] / prex-old / usr / lib / libc / stdio

Annotation of prex-old/usr/lib/libc/stdio/fvwrite.c, Revision 1.1.1.1

1.1       nbrk        1: /*-
                      2:  * Copyright (c) 1990, 1993
                      3:  *     The Regents of the University of California.  All rights reserved.
                      4:  *
                      5:  * This code is derived from software contributed to Berkeley by
                      6:  * Chris Torek.
                      7:  *
                      8:  * Redistribution and use in source and binary forms, with or without
                      9:  * modification, are permitted provided that the following conditions
                     10:  * are met:
                     11:  * 1. Redistributions of source code must retain the above copyright
                     12:  *    notice, this list of conditions and the following disclaimer.
                     13:  * 2. Redistributions in binary form must reproduce the above copyright
                     14:  *    notice, this list of conditions and the following disclaimer in the
                     15:  *    documentation and/or other materials provided with the distribution.
                     16:  * 3. Neither the name of the University nor the names of its contributors
                     17:  *    may be used to endorse or promote products derived from this software
                     18:  *    without specific prior written permission.
                     19:  *
                     20:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     21:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     22:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     23:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     24:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     25:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     26:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     27:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     28:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     29:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     30:  * SUCH DAMAGE.
                     31:  */
                     32:
                     33: #include <stdio.h>
                     34: #include <string.h>
                     35: #include "local.h"
                     36: #include "fvwrite.h"
                     37:
                     38: /*
                     39:  * Write some memory regions.  Return zero on success, EOF on error.
                     40:  *
                     41:  * This routine is large and unsightly, but most of the ugliness due
                     42:  * to the three different kinds of output buffering is handled here.
                     43:  */
                     44: int
                     45: __sfvwrite(fp, uio)
                     46:        FILE *fp;
                     47:        struct __suio *uio;
                     48: {
                     49:        size_t len;
                     50:        char *p;
                     51:        struct __siov *iov;
                     52:        int w, s;
                     53:        char *nl;
                     54:        int nlknown, nldist;
                     55:
                     56:        if ((len = uio->uio_resid) == 0)
                     57:                return (0);
                     58:        /* make sure we can write */
                     59:        if (cantwrite(fp))
                     60:                return (EOF);
                     61:
                     62: #define        MIN(a, b) ((a) < (b) ? (a) : (b))
                     63: #define        COPY(n)   (void)memcpy((void *)fp->_p, (void *)p, (size_t)(n))
                     64:
                     65:        iov = uio->uio_iov;
                     66:        p = iov->iov_base;
                     67:        len = iov->iov_len;
                     68:        iov++;
                     69: #define GETIOV(extra_work) \
                     70:        while (len == 0) { \
                     71:                extra_work; \
                     72:                p = iov->iov_base; \
                     73:                len = iov->iov_len; \
                     74:                iov++; \
                     75:        }
                     76:        if (fp->_flags & __SNBF) {
                     77:                /*
                     78:                 * Unbuffered: write up to BUFSIZ bytes at a time.
                     79:                 */
                     80:                do {
                     81:                        GETIOV(;);
                     82:                        w = __swrite(fp, p, MIN(len, BUFSIZ));
                     83:                        if (w <= 0)
                     84:                                goto err;
                     85:                        p += w;
                     86:                        len -= w;
                     87:                } while ((uio->uio_resid -= w) != 0);
                     88:        } else if ((fp->_flags & __SLBF) == 0) {
                     89:                /*
                     90:                 * Fully buffered: fill partially full buffer, if any,
                     91:                 * and then flush.  If there is no partial buffer, write
                     92:                 * one _bf._size byte chunk directly (without copying).
                     93:                 *
                     94:                 * String output is a special case: write as many bytes
                     95:                 * as fit, but pretend we wrote everything.  This makes
                     96:                 * snprintf() return the number of bytes needed, rather
                     97:                 * than the number used, and avoids its write function
                     98:                 * (so that the write function can be invalid).
                     99:                 */
                    100:                do {
                    101:                        GETIOV(;);
                    102:                        w = fp->_w;
                    103:                        if (fp->_flags & __SSTR) {
                    104:                                if (len < w)
                    105:                                        w = len;
                    106:                                COPY(w);        /* copy MIN(fp->_w,len), */
                    107:                                fp->_w -= w;
                    108:                                fp->_p += w;
                    109:                                w = len;        /* but pretend copied all */
                    110:                        } else if (fp->_p > fp->_bf._base && len > w) {
                    111:                                /* fill and flush */
                    112:                                COPY(w);
                    113:                                /* fp->_w -= w; */ /* unneeded */
                    114:                                fp->_p += w;
                    115:                                if (fflush(fp))
                    116:                                        goto err;
                    117:                        } else if (len >= (w = fp->_bf._size)) {
                    118:                                /* write directly */
                    119:                                w = __swrite(fp, p, w);
                    120:                                if (w <= 0)
                    121:                                        goto err;
                    122:                        } else {
                    123:                                /* fill and done */
                    124:                                w = len;
                    125:                                COPY(w);
                    126:                                fp->_w -= w;
                    127:                                fp->_p += w;
                    128:                        }
                    129:                        p += w;
                    130:                        len -= w;
                    131:                } while ((uio->uio_resid -= w) != 0);
                    132:        } else {
                    133:                /*
                    134:                 * Line buffered: like fully buffered, but we
                    135:                 * must check for newlines.  Compute the distance
                    136:                 * to the first newline (including the newline),
                    137:                 * or `infinity' if there is none, then pretend
                    138:                 * that the amount to write is MIN(len,nldist).
                    139:                 */
                    140:                nlknown = 0;
                    141:                nldist = 0;     /* XXX just to keep gcc happy */
                    142:                do {
                    143:                        GETIOV(nlknown = 0);
                    144:                        if (!nlknown) {
                    145:                                nl = memchr((void *)p, '\n', len);
                    146:                                nldist = nl ? nl + 1 - p : len + 1;
                    147:                                nlknown = 1;
                    148:                        }
                    149:                        s = MIN(len, nldist);
                    150:                        w = fp->_w + fp->_bf._size;
                    151:                        if (fp->_p > fp->_bf._base && s > w) {
                    152:                                COPY(w);
                    153:                                /* fp->_w -= w; */
                    154:                                fp->_p += w;
                    155:                                if (fflush(fp))
                    156:                                        goto err;
                    157:                        } else if (s >= (w = fp->_bf._size)) {
                    158:                                w = __swrite(fp, p, w);
                    159:                                if (w <= 0)
                    160:                                        goto err;
                    161:                        } else {
                    162:                                w = s;
                    163:                                COPY(w);
                    164:                                fp->_w -= w;
                    165:                                fp->_p += w;
                    166:                        }
                    167:                        if ((nldist -= w) == 0) {
                    168:                                /* copied the newline: flush and forget */
                    169:                                if (fflush(fp))
                    170:                                        goto err;
                    171:                                nlknown = 0;
                    172:                        }
                    173:                        p += w;
                    174:                        len -= w;
                    175:                } while ((uio->uio_resid -= w) != 0);
                    176:        }
                    177:        return (0);
                    178:
                    179: err:
                    180:        fp->_flags |= __SERR;
                    181:        return (EOF);
                    182: }

CVSweb