src/basic/io-util.h

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 #pragma once
   3
   4 /***
   5   This file is part of systemd.
   6
   7   Copyright 2010 Lennart Poettering
   8 ***/
   9
  10 #include <stdbool.h>
  11 #include <stddef.h>
  12 #include <stdint.h>
  13 #include <sys/types.h>
  14 #include <sys/uio.h>
  15
  16 #include "macro.h"
  17 #include "time-util.h"
  18
  19 int flush_fd(int fd);
  20
  21 ssize_t loop_read(int fd, void *buf, size_t nbytes, bool do_poll);
  22 int loop_read_exact(int fd, void *buf, size_t nbytes, bool do_poll);
  23 int loop_write(int fd, const void *buf, size_t nbytes, bool do_poll);
  24
  25 int pipe_eof(int fd);
  26
  27 int fd_wait_for_event(int fd, int event, usec_t timeout);
  28
  29 ssize_t sparse_write(int fd, const void *p, size_t sz, size_t run_length);
  30
  31 static inline size_t IOVEC_TOTAL_SIZE(const struct iovec *i, unsigned n) {
  32         unsigned j;
  33         size_t r = 0;
  34
  35         for (j = 0; j < n; j++)
  36                 r += i[j].iov_len;
  37
  38         return r;
  39 }
  40
  41 static inline size_t IOVEC_INCREMENT(struct iovec *i, unsigned n, size_t k) {
  42         unsigned j;
  43
  44         for (j = 0; j < n; j++) {
  45                 size_t sub;
  46
  47                 if (_unlikely_(k <= 0))
  48                         break;
  49
  50                 sub = MIN(i[j].iov_len, k);
  51                 i[j].iov_len -= sub;
  52                 i[j].iov_base = (uint8_t*) i[j].iov_base + sub;
  53                 k -= sub;
  54         }
  55
  56         return k;
  57 }
  58
  59 static inline bool FILE_SIZE_VALID(uint64_t l) {
  60         /* ftruncate() and friends take an unsigned file size, but actually cannot deal with file sizes larger than
  61          * 2^63 since the kernel internally handles it as signed value. This call allows checking for this early. */
  62
  63         return (l >> 63) == 0;
  64 }
  65
  66 static inline bool FILE_SIZE_VALID_OR_INFINITY(uint64_t l) {
  67
  68         /* Same as above, but allows one extra value: -1 as indication for infinity. */
  69
  70         if (l == (uint64_t) -1)
  71                 return true;
  72
  73         return FILE_SIZE_VALID(l);
  74
  75 }
  76
  77 #define IOVEC_INIT(base, len) { .iov_base = (base), .iov_len = (len) }
  78 #define IOVEC_MAKE(base, len) (struct iovec) IOVEC_INIT(base, len)
  79 #define IOVEC_INIT_STRING(string) IOVEC_INIT((char*) string, strlen(string))
  80 #define IOVEC_MAKE_STRING(string) (struct iovec) IOVEC_INIT_STRING(string)