23
# include <winsock2.h>
25
# include <sys/types.h>
26
# include <sys/socket.h>
20
29
size_t iov_from_buf(struct iovec *iov, unsigned int iov_cnt,
21
const void *buf, size_t iov_off, size_t size)
30
size_t offset, const void *buf, size_t bytes)
23
size_t iovec_off, buf_off;
28
for (i = 0; i < iov_cnt && size; i++) {
29
if (iov_off < (iovec_off + iov[i].iov_len)) {
30
size_t len = MIN((iovec_off + iov[i].iov_len) - iov_off, size);
32
memcpy(iov[i].iov_base + (iov_off - iovec_off), buf + buf_off, len);
34
for (i = 0, done = 0; (offset || done < bytes) && i < iov_cnt; i++) {
35
if (offset < iov[i].iov_len) {
36
size_t len = MIN(iov[i].iov_len - offset, bytes - done);
37
memcpy(iov[i].iov_base + offset, buf + done, len);
41
offset -= iov[i].iov_len;
38
iovec_off += iov[i].iov_len;
43
48
size_t iov_to_buf(const struct iovec *iov, const unsigned int iov_cnt,
44
void *buf, size_t iov_off, size_t size)
49
size_t offset, void *buf, size_t bytes)
47
size_t iovec_off, buf_off;
53
for (i = 0; i < iov_cnt && size; i++) {
54
if (iov_off < (iovec_off + iov[i].iov_len)) {
55
size_t len = MIN((iovec_off + iov[i].iov_len) - iov_off , size);
57
memcpy(ptr + buf_off, iov[i].iov_base + (iov_off - iovec_off), len);
53
for (i = 0, done = 0; (offset || done < bytes) && i < iov_cnt; i++) {
54
if (offset < iov[i].iov_len) {
55
size_t len = MIN(iov[i].iov_len - offset, bytes - done);
56
memcpy(buf + done, iov[i].iov_base + offset, len);
60
offset -= iov[i].iov_len;
63
iovec_off += iov[i].iov_len;
68
size_t iov_clear(const struct iovec *iov, const unsigned int iov_cnt,
69
size_t iov_off, size_t size)
67
size_t iov_memset(const struct iovec *iov, const unsigned int iov_cnt,
68
size_t offset, int fillc, size_t bytes)
71
size_t iovec_off, buf_off;
76
for (i = 0; i < iov_cnt && size; i++) {
77
if (iov_off < (iovec_off + iov[i].iov_len)) {
78
size_t len = MIN((iovec_off + iov[i].iov_len) - iov_off , size);
80
memset(iov[i].iov_base + (iov_off - iovec_off), 0, len);
72
for (i = 0, done = 0; (offset || done < bytes) && i < iov_cnt; i++) {
73
if (offset < iov[i].iov_len) {
74
size_t len = MIN(iov[i].iov_len - offset, bytes - done);
75
memset(iov[i].iov_base + offset, fillc, len);
79
offset -= iov[i].iov_len;
86
iovec_off += iov[i].iov_len;
91
86
size_t iov_size(const struct iovec *iov, const unsigned int iov_cnt)
98
/* helper function for iov_send_recv() */
100
do_send_recv(int sockfd, struct iovec *iov, unsigned iov_cnt, bool do_send)
102
#if defined CONFIG_IOVEC && defined CONFIG_POSIX
105
memset(&msg, 0, sizeof(msg));
107
msg.msg_iovlen = iov_cnt;
110
? sendmsg(sockfd, &msg, 0)
111
: recvmsg(sockfd, &msg, 0);
112
} while (ret < 0 && errno == EINTR);
115
/* else send piece-by-piece */
116
/*XXX Note: windows has WSASend() and WSARecv() */
119
while (i < iov_cnt) {
121
? send(sockfd, iov[i].iov_base, iov[i].iov_len, 0)
122
: recv(sockfd, iov[i].iov_base, iov[i].iov_len, 0);
127
} else if (errno == EINTR) {
130
/* else it is some "other" error,
131
* only return if there was no data processed. */
143
ssize_t iov_send_recv(int sockfd, struct iovec *iov, unsigned iov_cnt,
144
size_t offset, size_t bytes,
148
unsigned si, ei; /* start and end indexes */
150
/* Find the start position, skipping `offset' bytes:
151
* first, skip all full-sized vector elements, */
152
for (si = 0; si < iov_cnt && offset >= iov[si].iov_len; ++si) {
153
offset -= iov[si].iov_len;
156
assert(si < iov_cnt);
157
/* second, skip `offset' bytes from the (now) first element,
159
iov[si].iov_base += offset;
160
iov[si].iov_len -= offset;
162
/* Find the end position skipping `bytes' bytes: */
163
/* first, skip all full-sized elements */
164
for (ei = si; ei < iov_cnt && iov[ei].iov_len <= bytes; ++ei) {
165
bytes -= iov[ei].iov_len;
168
/* second, fixup the last element, and remember
169
* the length we've cut from the end of it in `bytes' */
171
assert(ei < iov_cnt);
172
assert(iov[ei].iov_len > bytes);
173
tail = iov[ei].iov_len - bytes;
174
iov[ei].iov_len = bytes;
175
bytes = tail; /* bytes is now equal to the tail size */
179
ret = do_send_recv(sockfd, iov + si, ei - si, do_send);
181
/* Undo the changes above */
183
iov[si].iov_base -= offset;
184
iov[si].iov_len += offset;
187
iov[ei-1].iov_len += bytes;
103
194
void iov_hexdump(const struct iovec *iov, const unsigned int iov_cnt,
104
195
FILE *fp, const char *prefix, size_t limit)