Line data    Source code

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