Subversion Repositories openrisc

/*

 * Copyright (c) 1990 The Regents of the University of California.

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms are permitted

 * provided that the above copyright notice and this paragraph are

 * duplicated in all such forms and that any documentation,

 * advertising materials, and other materials related to such

 * distribution and use acknowledge that the software was developed

 * by the University of California, Berkeley.  The name of the

 * University may not be used to endorse or promote products derived

 * from this software without specific prior written permission.

 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED

 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.

 */

/* No user fns here.  Pesch 15apr92. */

#include <_ansi.h>

#include <stdio.h>

#include <string.h>

#include <stdlib.h>

#include <errno.h>

#include "local.h"

#include "fvwrite.h"

#define MIN(a, b) ((a) < (b) ? (a) : (b))

#define COPY(n)   _CAST_VOID memmove ((_PTR) fp->_p, (_PTR) p, (size_t) (n))

#define GETIOV(extra_work) \

  while (len == 0) \

    { \

      extra_work; \

      p = iov->iov_base; \

      len = iov->iov_len; \

      iov++; \

    }

/*

 * Write some memory regions.  Return zero on success, EOF on error.

 *

 * This routine is large and unsightly, but most of the ugliness due

 * to the three different kinds of output buffering is handled here.

 */

int

_DEFUN(__sfvwrite_r, (ptr, fp, uio),

       struct _reent *ptr _AND

       register FILE *fp _AND

       register struct __suio *uio)

{

  register size_t len;

  register _CONST char *p = NULL;

  register struct __siov *iov;

  register int w, s;

  char *nl;

  int nlknown, nldist;

  if ((len = uio->uio_resid) == 0)

    return 0;

  /* make sure we can write */

  if (cantwrite (ptr, fp))

    {

      fp->_flags |= __SERR;

      ptr->_errno = EBADF;

      return EOF;

    }

  iov = uio->uio_iov;

  len = 0;

#ifdef __SCLE

  if (fp->_flags & __SCLE) /* text mode */

    {

      do

        {

          GETIOV (;);

          while (len > 0)

            {

              if (putc (*p, fp) == EOF)

                return EOF;

              p++;

              len--;

              uio->uio_resid--;

            }

        }

      while (uio->uio_resid > 0);

      return 0;

    }

#endif

  if (fp->_flags & __SNBF)

    {

      /*

       * Unbuffered: write up to BUFSIZ bytes at a time.

       */

      do

        {

          GETIOV (;);

          w = fp->_write (ptr, fp->_cookie, p, MIN (len, BUFSIZ));

          if (w <= 0)

            goto err;

          p += w;

          len -= w;

        }

      while ((uio->uio_resid -= w) != 0);

    }

  else if ((fp->_flags & __SLBF) == 0)

    {

      /*

       * Fully buffered: fill partially full buffer, if any,

       * and then flush.  If there is no partial buffer, write

       * one _bf._size byte chunk directly (without copying).

       *

       * String output is a special case: write as many bytes

       * as fit, but pretend we wrote everything.  This makes

       * snprintf() return the number of bytes needed, rather

       * than the number used, and avoids its write function

       * (so that the write function can be invalid).  If

       * we are dealing with the asprintf routines, we will

       * dynamically increase the buffer size as needed.

       */

      do

        {

          GETIOV (;);

          w = fp->_w;

          if (fp->_flags & __SSTR)

            {

              if (len >= w && fp->_flags & (__SMBF | __SOPT))

                { /* must be asprintf family */

                  unsigned char *str;

                  int curpos = (fp->_p - fp->_bf._base);

                  /* Choose a geometric growth factor to avoid

                     quadratic realloc behavior, but use a rate less

                     than (1+sqrt(5))/2 to accomodate malloc

                     overhead. asprintf EXPECTS us to overallocate, so

                     that it can add a trailing \0 without

                     reallocating.  The new allocation should thus be

                     max(prev_size*1.5, curpos+len+1). */

                  int newsize = fp->_bf._size * 3 / 2;

                  if (newsize < curpos + len + 1)

                    newsize = curpos + len + 1;

                  if (fp->_flags & __SOPT)

                    {

                      /* asnprintf leaves original buffer alone.  */

                      str = (unsigned char *)_malloc_r (ptr, newsize);

                      if (!str)

                        {

                          ptr->_errno = ENOMEM;

                          goto err;

                        }

                      memcpy (str, fp->_bf._base, curpos);

                      fp->_flags = (fp->_flags & ~__SOPT) | __SMBF;

                    }

                  else

                    {

                      str = (unsigned char *)_realloc_r (ptr, fp->_bf._base,

                                                         newsize);

                      if (!str)

                        {

                          /* Free buffer which is no longer used.  */

                          _free_r (ptr, fp->_bf._base);

                          /* Ensure correct errno, even if free changed it.  */

                          ptr->_errno = ENOMEM;

                          goto err;

                        }

                    }

                  fp->_bf._base = str;

                  fp->_p = str + curpos;

                  fp->_bf._size = newsize;

                  w = len;

                  fp->_w = newsize - curpos;

                }

              if (len < w)

                w = len;

              COPY (w);         /* copy MIN(fp->_w,len), */

              fp->_w -= w;

              fp->_p += w;

              w = len;          /* but pretend copied all */

            }

          else if (fp->_p > fp->_bf._base && len > w)

            {

              /* fill and flush */

              COPY (w);

              /* fp->_w -= w; *//* unneeded */

              fp->_p += w;

              if (_fflush_r (ptr, fp))

                goto err;

            }

          else if (len >= (w = fp->_bf._size))

            {

              /* write directly */

              w = fp->_write (ptr, fp->_cookie, p, w);

              if (w <= 0)

                goto err;

            }

          else

            {

              /* fill and done */

              w = len;

              COPY (w);

              fp->_w -= w;

              fp->_p += w;

            }

          p += w;

          len -= w;

        }

      while ((uio->uio_resid -= w) != 0);

    }

  else

    {

      /*

       * Line buffered: like fully buffered, but we

       * must check for newlines.  Compute the distance

       * to the first newline (including the newline),

       * or `infinity' if there is none, then pretend

       * that the amount to write is MIN(len,nldist).

       */

      nlknown = 0;

      nldist = 0;

      do

        {

          GETIOV (nlknown = 0);

          if (!nlknown)

            {

              nl = memchr ((_PTR) p, '\n', len);

              nldist = nl ? nl + 1 - p : len + 1;

              nlknown = 1;

            }

          s = MIN (len, nldist);

          w = fp->_w + fp->_bf._size;

          if (fp->_p > fp->_bf._base && s > w)

            {

              COPY (w);

              /* fp->_w -= w; */

              fp->_p += w;

              if (_fflush_r (ptr, fp))

                goto err;

            }

          else if (s >= (w = fp->_bf._size))

            {

              w = fp->_write (ptr, fp->_cookie, p, w);

              if (w <= 0)

                goto err;

            }

          else

            {

              w = s;

              COPY (w);

              fp->_w -= w;

              fp->_p += w;

            }

          if ((nldist -= w) == 0)

            {

              /* copied the newline: flush and forget */

              if (_fflush_r (ptr, fp))

                goto err;

              nlknown = 0;

            }

          p += w;

          len -= w;

        }

      while ((uio->uio_resid -= w) != 0);

    }

  return 0;

err:

  fp->_flags |= __SERR;

  return EOF;

}