1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
/*
* No copyright is claimed. This code is in the public domain; do with
* it what you wish.
*
* Written by Karel Zak <kzak@redhat.com>
* Petr Uzel <petr.uzel@suse.cz>
*/
#ifndef UTIL_LINUX_ALL_IO_H
#define UTIL_LINUX_ALL_IO_H
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#ifdef HAVE_SYS_SENDFILE_H
# include <sys/sendfile.h>
#endif
#include "c.h"
static inline int write_all(int fd, const void *buf, size_t count)
{
while (count) {
ssize_t tmp;
errno = 0;
tmp = write(fd, buf, count);
if (tmp > 0) {
count -= tmp;
if (count)
buf = (const void *) ((const char *) buf + tmp);
} else if (errno != EINTR && errno != EAGAIN)
return -1;
if (errno == EAGAIN) /* Try later, *sigh* */
xusleep(250000);
}
return 0;
}
static inline int fwrite_all(const void *ptr, size_t size,
size_t nmemb, FILE *stream)
{
while (nmemb) {
size_t tmp;
errno = 0;
tmp = fwrite(ptr, size, nmemb, stream);
if (tmp > 0) {
nmemb -= tmp;
if (nmemb)
ptr = (const void *) ((const char *) ptr + (tmp * size));
} else if (errno != EINTR && errno != EAGAIN)
return -1;
if (errno == EAGAIN) /* Try later, *sigh* */
xusleep(250000);
}
return 0;
}
static inline ssize_t read_all(int fd, char *buf, size_t count)
{
ssize_t ret;
ssize_t c = 0;
int tries = 0;
memset(buf, 0, count);
while (count > 0) {
ret = read(fd, buf, count);
if (ret < 0) {
if ((errno == EAGAIN || errno == EINTR) && (tries++ < 5)) {
xusleep(250000);
continue;
}
return c ? c : -1;
}
if (ret == 0)
return c;
tries = 0;
count -= ret;
buf += ret;
c += ret;
}
return c;
}
static inline ssize_t read_all_alloc(int fd, char **buf)
{
size_t size = 1024, c = 0;
ssize_t ret;
*buf = malloc(size);
if (!*buf)
return -1;
while (1) {
ret = read_all(fd, *buf + c, size - c);
if (ret < 0) {
free(*buf);
*buf = NULL;
return -1;
}
if (ret == 0)
return c;
c += ret;
if (c == size) {
size *= 2;
*buf = realloc(*buf, size);
if (!*buf)
return -1;
}
}
}
static inline ssize_t sendfile_all(int out, int in, off_t *off, size_t count)
{
#if defined(HAVE_SENDFILE) && defined(__linux__)
ssize_t ret;
ssize_t c = 0;
int tries = 0;
while (count) {
ret = sendfile(out, in, off, count);
if (ret < 0) {
if ((errno == EAGAIN || errno == EINTR) && (tries++ < 5)) {
xusleep(250000);
continue;
}
return c ? c : -1;
}
if (ret == 0)
return c;
tries = 0;
count -= ret;
c += ret;
}
return c;
#else
errno = ENOSYS;
return -1;
#endif
}
#endif /* UTIL_LINUX_ALL_IO_H */
|
