2 * Copyright (c) 2021-2023 Oracle and/or its affiliates.
4 * This work is licensed under the terms of the GNU GPL, version 2 or later.
5 * See the COPYING file in the top-level directory.
8 #include "qemu/osdep.h"
9 #include "exec/ramblock.h"
10 #include "qemu/cutils.h"
11 #include "qemu/error-report.h"
12 #include "qapi/error.h"
16 #include "migration.h"
17 #include "io/channel-file.h"
18 #include "io/channel-util.h"
22 #define OFFSET_OPTION ",offset="
24 static struct FileOutgoingArgs
{
28 /* Remove the offset option from @filespec and return it in @offsetp. */
30 int file_parse_offset(char *filespec
, uint64_t *offsetp
, Error
**errp
)
32 char *option
= strstr(filespec
, OFFSET_OPTION
);
37 option
+= sizeof(OFFSET_OPTION
) - 1;
38 ret
= qemu_strtosz(option
, NULL
, offsetp
);
40 error_setg_errno(errp
, -ret
, "file URI has bad offset %s", option
);
47 void file_cleanup_outgoing_migration(void)
49 g_free(outgoing_args
.fname
);
50 outgoing_args
.fname
= NULL
;
53 bool file_send_channel_create(gpointer opaque
, Error
**errp
)
58 int fd
= fd_args_get_fd();
61 ioc
= qio_channel_file_new_fd(dup(fd
));
63 ioc
= qio_channel_file_new_path(outgoing_args
.fname
, flags
, 0, errp
);
69 multifd_channel_connect(opaque
, QIO_CHANNEL(ioc
));
74 * File channel creation is synchronous. However posting this
75 * semaphore here is simpler than adding a special case.
77 multifd_send_channel_created();
82 void file_start_outgoing_migration(MigrationState
*s
,
83 FileMigrationArgs
*file_args
, Error
**errp
)
85 g_autoptr(QIOChannelFile
) fioc
= NULL
;
86 g_autofree
char *filename
= g_strdup(file_args
->filename
);
87 uint64_t offset
= file_args
->offset
;
90 trace_migration_file_outgoing(filename
);
92 fioc
= qio_channel_file_new_path(filename
, O_CREAT
| O_WRONLY
| O_TRUNC
,
98 outgoing_args
.fname
= g_strdup(filename
);
100 ioc
= QIO_CHANNEL(fioc
);
101 if (offset
&& qio_channel_io_seek(ioc
, offset
, SEEK_SET
, errp
) < 0) {
104 qio_channel_set_name(ioc
, "migration-file-outgoing");
105 migration_channel_connect(s
, ioc
, NULL
, NULL
);
108 static gboolean
file_accept_incoming_migration(QIOChannel
*ioc
,
109 GIOCondition condition
,
112 migration_channel_process_incoming(ioc
);
113 object_unref(OBJECT(ioc
));
114 return G_SOURCE_REMOVE
;
117 void file_start_incoming_migration(FileMigrationArgs
*file_args
, Error
**errp
)
119 g_autofree
char *filename
= g_strdup(file_args
->filename
);
120 QIOChannelFile
*fioc
= NULL
;
121 uint64_t offset
= file_args
->offset
;
125 trace_migration_file_incoming(filename
);
127 fioc
= qio_channel_file_new_path(filename
, O_RDONLY
, 0, errp
);
133 qio_channel_io_seek(QIO_CHANNEL(fioc
), offset
, SEEK_SET
, errp
) < 0) {
137 if (migrate_multifd()) {
138 channels
+= migrate_multifd_channels();
142 QIOChannel
*ioc
= QIO_CHANNEL(fioc
);
144 qio_channel_set_name(ioc
, "migration-file-incoming");
145 qio_channel_add_watch_full(ioc
, G_IO_IN
,
146 file_accept_incoming_migration
,
148 g_main_context_get_thread_default());
150 fioc
= qio_channel_file_new_fd(dup(fioc
->fd
));
152 if (!fioc
|| fioc
->fd
== -1) {
153 error_setg(errp
, "Error creating migration incoming channel");
156 } while (++i
< channels
);
159 int file_write_ramblock_iov(QIOChannel
*ioc
, const struct iovec
*iov
,
160 int niov
, RAMBlock
*block
, Error
**errp
)
163 int i
, slice_idx
, slice_num
;
164 uintptr_t base
, next
, offset
;
171 * If the iov array doesn't have contiguous elements, we need to
172 * split it in slices because we only have one file offset for the
173 * whole iov. Do this here so callers don't need to break the iov
176 for (i
= 0; i
< niov
; i
++, slice_num
++) {
177 base
= (uintptr_t) iov
[i
].iov_base
;
180 len
= iov
[i
].iov_len
;
181 next
= (uintptr_t) iov
[i
+ 1].iov_base
;
183 if (base
+ len
== next
) {
189 * Use the offset of the first element of the segment that
192 offset
= (uintptr_t) iov
[slice_idx
].iov_base
- (uintptr_t) block
->host
;
193 if (offset
>= block
->used_length
) {
194 error_setg(errp
, "offset %" PRIxPTR
195 "outside of ramblock %s range", offset
, block
->idstr
);
200 ret
= qio_channel_pwritev(ioc
, &iov
[slice_idx
], slice_num
,
201 block
->pages_offset
+ offset
, errp
);
206 slice_idx
+= slice_num
;
210 return (ret
< 0) ? ret
: 0;
213 int multifd_file_recv_data(MultiFDRecvParams
*p
, Error
**errp
)
215 MultiFDRecvData
*data
= p
->data
;
218 ret
= qio_channel_pread(p
->c
, (char *) data
->opaque
,
219 data
->size
, data
->file_offset
, errp
);
220 if (ret
!= data
->size
) {
222 "multifd recv (%u): read 0x%zx, expected 0x%zx",
223 p
->id
, ret
, data
->size
);