fvwrite.c
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:5k
源码类别:

MySQL数据库

开发平台:

Visual C++

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