| 1 | #include "dietstdio.h" |
|---|
| 2 | #include <unistd.h> |
|---|
| 3 | #include <errno.h> |
|---|
| 4 | #include <string.h> |
|---|
| 5 | |
|---|
| 6 | size_t fwrite_unlocked(const void *ptr, size_t size, size_t nmemb, FILE *stream) { |
|---|
| 7 | ssize_t res; |
|---|
| 8 | size_t len=size*nmemb; |
|---|
| 9 | size_t i,done; |
|---|
| 10 | |
|---|
| 11 | //log_msg("fwrite_unlocked started, streem's fd %d, size %d, nmemb %d\n", stream->fd, size, nmemb); |
|---|
| 12 | if (!(stream->flags&CANWRITE) || __fflush4(stream,0)) { |
|---|
| 13 | kaputt: |
|---|
| 14 | //log_msg("fwrite_unlocked errnor at line %d, streem's fd %d, can write ? %d\n", __LINE__, stream->fd, |
|---|
| 15 | // stream->flags & CANWRITE); |
|---|
| 16 | stream->flags|=ERRORINDICATOR; |
|---|
| 17 | return 0; |
|---|
| 18 | } |
|---|
| 19 | |
|---|
| 20 | if (!nmemb || len/nmemb!=size) return 0; /* check for integer overflow */ |
|---|
| 21 | |
|---|
| 22 | if (len>stream->buflen || (stream->flags&NOBUF)) { |
|---|
| 23 | if (fflush_unlocked(stream)) return 0; |
|---|
| 24 | i=2; |
|---|
| 25 | do { |
|---|
| 26 | res= write(stream->fd,ptr,len); |
|---|
| 27 | } while ((res==-1) && (i--));// && errno==EINTR); |
|---|
| 28 | } else { |
|---|
| 29 | /* try to make the common case fast */ |
|---|
| 30 | size_t todo=stream->buflen-stream->bm; |
|---|
| 31 | if (todo>len) todo=len; |
|---|
| 32 | |
|---|
| 33 | if (todo) { |
|---|
| 34 | if (stream->flags&BUFLINEWISE) { |
|---|
| 35 | for (i=0; i<todo; ++i) { |
|---|
| 36 | if ((stream->buf[stream->bm++]=((char*)ptr)[i])=='\n') { |
|---|
| 37 | if (fflush_unlocked(stream)) goto kaputt; |
|---|
| 38 | } |
|---|
| 39 | } |
|---|
| 40 | } else { |
|---|
| 41 | memcpy(stream->buf+stream->bm,ptr,todo); |
|---|
| 42 | stream->bm+=todo; |
|---|
| 43 | } |
|---|
| 44 | done=todo; |
|---|
| 45 | } else |
|---|
| 46 | done=0; |
|---|
| 47 | for (i=done; i<len; ++i) |
|---|
| 48 | if (fputc_unlocked(((char*)ptr)[i],stream)) { |
|---|
| 49 | res=len-i; |
|---|
| 50 | goto abort; |
|---|
| 51 | } |
|---|
| 52 | res=len; |
|---|
| 53 | } |
|---|
| 54 | if (res<0) { |
|---|
| 55 | stream->flags|=ERRORINDICATOR; |
|---|
| 56 | return 0; |
|---|
| 57 | } |
|---|
| 58 | abort: |
|---|
| 59 | return size?res/size:0; |
|---|
| 60 | } |
|---|
| 61 | |
|---|
| 62 | size_t fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream) __attribute__((weak,alias("fwrite_unlocked"))); |
|---|