nsz Git - musl/blob - src/stdio/__stdio_read.c

   1 #include "stdio_impl.h"
   2 #include <sys/uio.h>
   3 #include <pthread.h>
   4
   5 static void cleanup(void *p)
   6 {
   7         FILE *f = p;
   8         if (!f->lockcount) __unlockfile(f);
   9 }
  10
  11 size_t __stdio_read(FILE *f, unsigned char *buf, size_t len)
  12 {
  13         struct iovec iov[2] = {
  14                 { .iov_base = buf, .iov_len = len - !!f->buf_size },
  15                 { .iov_base = f->buf, .iov_len = f->buf_size }
  16         };
  17         ssize_t cnt;
  18
  19         if (libc.main_thread) {
  20                 pthread_cleanup_push(cleanup, f);
  21                 cnt = syscall_cp(SYS_readv, f->fd, iov, 2);
  22                 pthread_cleanup_pop(0);
  23         } else {
  24                 cnt = syscall(SYS_readv, f->fd, iov, 2);
  25         }
  26         if (cnt <= 0) {
  27                 f->flags |= F_EOF ^ ((F_ERR^F_EOF) & cnt);
  28                 f->rpos = f->rend = 0;
  29                 return cnt;
  30         }
  31         if (cnt <= iov[0].iov_len) return cnt;
  32         cnt -= iov[0].iov_len;
  33         f->rpos = f->buf;
  34         f->rend = f->buf + cnt;
  35         if (f->buf_size) buf[len-1] = *f->rpos++;
  36         return len;
  37 }