[1] | 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"))); |
---|