io.c

   1 /*
   2  * I/O utilities
   3  */
   4
   5 #include "shared.h"
   6
   7 int io_poll(int fd, int events, int msec)
   8 {
   9         struct pollfd pfd;
  10
  11         pfd.fd = fd;
  12         pfd.events = events;
  13
  14         return poll(&pfd, 1, msec);
  15 }
  16
  17 int io_send_all(int fd, const void *buf, size_t len)
  18 {
  19         int poll_ret, sent, loops = 0;
  20         size_t total = 0;
  21
  22         if (!buf || !len)
  23                 return 0;
  24
  25         poll_ret = io_poll(fd, POLLOUT, 0);
  26         if (poll_ret < 0)
  27                 lerr("io_poll(%d, POLLOUT, 0) returned %d: %s", fd, poll_ret, strerror(errno));
  28
  29         do {
  30                 loops++;
  31                 sent = send(fd, buf + total, len - total, MSG_DONTWAIT);
  32                 if (poll_ret > 0 && sent + total == 0) {
  33                         /* disconnected peer? */
  34                         return 0;
  35                 }
  36                 if (sent < 0) {
  37                         if (errno == EAGAIN || errno == EWOULDBLOCK) {
  38                                 sent = io_write_ok(fd, 100);
  39                                 continue;
  40                         }
  41
  42                         lerr("send(%d, (buf + total), %zu, MSG_DONTWAIT) returned %d (%s)",
  43                                  fd, len - total, sent, strerror(errno));
  44                         continue;
  45                 }
  46
  47                 total += sent;
  48         } while (total < len && sent > 0 && loops < 15);
  49
  50         if (sent < 0)
  51                 return sent;
  52
  53         return total;
  54 }