4 * Copyright (c) 2003-2008 Fabrice Bellard
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
24 #include "qemu-common.h"
26 #include "qemu/sockets.h"
27 #include "block/coroutine.h"
28 #include "migration/qemu-file.h"
29 #include "migration/qemu-file-internal.h"
31 typedef struct QEMUFileSocket
{
36 static ssize_t
socket_writev_buffer(void *opaque
, struct iovec
*iov
, int iovcnt
,
39 QEMUFileSocket
*s
= opaque
;
41 ssize_t size
= iov_size(iov
, iovcnt
);
43 len
= iov_send(s
->fd
, iov
, iovcnt
, 0, size
);
45 len
= -socket_error();
50 static int socket_get_fd(void *opaque
)
52 QEMUFileSocket
*s
= opaque
;
57 static ssize_t
socket_get_buffer(void *opaque
, uint8_t *buf
, int64_t pos
,
60 QEMUFileSocket
*s
= opaque
;
64 len
= qemu_recv(s
->fd
, buf
, size
, 0);
68 if (socket_error() == EAGAIN
) {
69 yield_until_fd_readable(s
->fd
);
70 } else if (socket_error() != EINTR
) {
76 len
= -socket_error();
81 static int socket_close(void *opaque
)
83 QEMUFileSocket
*s
= opaque
;
89 static int socket_shutdown(void *opaque
, bool rd
, bool wr
)
91 QEMUFileSocket
*s
= opaque
;
93 if (shutdown(s
->fd
, rd
? (wr
? SHUT_RDWR
: SHUT_RD
) : SHUT_WR
)) {
100 static ssize_t
unix_writev_buffer(void *opaque
, struct iovec
*iov
, int iovcnt
,
103 QEMUFileSocket
*s
= opaque
;
105 ssize_t size
= iov_size(iov
, iovcnt
);
111 /* Find the next start position; skip all full-sized vector elements */
112 while (offset
>= iov
[0].iov_len
) {
113 offset
-= iov
[0].iov_len
;
117 /* skip `offset' bytes from the (now) first element, undo it on exit */
119 iov
[0].iov_base
+= offset
;
120 iov
[0].iov_len
-= offset
;
123 len
= writev(s
->fd
, iov
, iovcnt
);
124 } while (len
== -1 && errno
== EINTR
);
129 /* Undo the changes above */
130 iov
[0].iov_base
-= offset
;
131 iov
[0].iov_len
+= offset
;
133 /* Prepare for the next iteration */
142 static ssize_t
unix_get_buffer(void *opaque
, uint8_t *buf
, int64_t pos
,
145 QEMUFileSocket
*s
= opaque
;
149 len
= read(s
->fd
, buf
, size
);
153 if (errno
== EAGAIN
) {
154 yield_until_fd_readable(s
->fd
);
155 } else if (errno
!= EINTR
) {
166 static int unix_close(void *opaque
)
168 QEMUFileSocket
*s
= opaque
;
174 static const QEMUFileOps unix_read_ops
= {
175 .get_fd
= socket_get_fd
,
176 .get_buffer
= unix_get_buffer
,
180 static const QEMUFileOps unix_write_ops
= {
181 .get_fd
= socket_get_fd
,
182 .writev_buffer
= unix_writev_buffer
,
186 QEMUFile
*qemu_fdopen(int fd
, const char *mode
)
191 (mode
[0] != 'r' && mode
[0] != 'w') ||
192 mode
[1] != 'b' || mode
[2] != 0) {
193 fprintf(stderr
, "qemu_fdopen: Argument validity check failed\n");
197 s
= g_new0(QEMUFileSocket
, 1);
200 if (mode
[0] == 'r') {
201 s
->file
= qemu_fopen_ops(s
, &unix_read_ops
);
203 s
->file
= qemu_fopen_ops(s
, &unix_write_ops
);
208 static const QEMUFileOps socket_read_ops
= {
209 .get_fd
= socket_get_fd
,
210 .get_buffer
= socket_get_buffer
,
211 .close
= socket_close
,
212 .shut_down
= socket_shutdown
216 static const QEMUFileOps socket_write_ops
= {
217 .get_fd
= socket_get_fd
,
218 .writev_buffer
= socket_writev_buffer
,
219 .close
= socket_close
,
220 .shut_down
= socket_shutdown
223 QEMUFile
*qemu_fopen_socket(int fd
, const char *mode
)
227 if (qemu_file_mode_is_not_valid(mode
)) {
231 s
= g_new0(QEMUFileSocket
, 1);
233 if (mode
[0] == 'w') {
234 qemu_set_block(s
->fd
);
235 s
->file
= qemu_fopen_ops(s
, &socket_write_ops
);
237 s
->file
= qemu_fopen_ops(s
, &socket_read_ops
);