[PATCH v9 7/7] multifd: Implement zero copy write in multifd migration (multifd-zero-copy)
Implement zero copy send on nocomp_send_write(), by making use of QIOChannel writev + flags & flush interface. Change multifd_send_sync_main() so flush_zero_copy() can be called after each iteration in order to make sure all dirty pages are sent before a new iteration is started. It will also flush at the beginning and at the end of migration. Also make it return -1 if flush_zero_copy() fails, in order to cancel the migration process, and avoid resuming the guest in the target host without receiving all current RAM. This will work fine on RAM migration because the RAM pages are not usually freed, and there is no problem on changing the pages content between writev_zero_copy() and the actual sending of the buffer, because this change will dirty the page and cause it to be re-sent on a next iteration anyway. A lot of locked memory may be needed in order to use multifd migration with zero-copy enabled, so disabling the feature should be necessary for low-privileged users trying to perform multifd migrations. Signed-off-by: Leonardo Bras --- migration/multifd.h | 2 ++ migration/migration.c | 11 ++- migration/multifd.c | 34 ++ migration/socket.c| 5 +++-- 4 files changed, 45 insertions(+), 7 deletions(-) diff --git a/migration/multifd.h b/migration/multifd.h index bcf5992945..4d8d89e5e5 100644 --- a/migration/multifd.h +++ b/migration/multifd.h @@ -92,6 +92,8 @@ typedef struct { uint32_t packet_len; /* pointer to the packet */ MultiFDPacket_t *packet; +/* multifd flags for sending ram */ +int write_flags; /* multifd flags for each packet */ uint32_t flags; /* size of the next packet that contains pages */ diff --git a/migration/migration.c b/migration/migration.c index 4b6df2eb5e..31739b2af9 100644 --- a/migration/migration.c +++ b/migration/migration.c @@ -1497,7 +1497,16 @@ static bool migrate_params_check(MigrationParameters *params, Error **errp) error_prepend(errp, "Invalid mapping given for block-bitmap-mapping: "); return false; } - +#ifdef CONFIG_LINUX +if (params->zero_copy_send && +(!migrate_use_multifd() || + params->multifd_compression != MULTIFD_COMPRESSION_NONE || + (params->tls_creds && *params->tls_creds))) { +error_setg(errp, + "Zero copy only available for non-compressed non-TLS multifd migration"); +return false; +} +#endif return true; } diff --git a/migration/multifd.c b/migration/multifd.c index 6c940aaa98..e37cc6e0d9 100644 --- a/migration/multifd.c +++ b/migration/multifd.c @@ -569,6 +569,7 @@ void multifd_save_cleanup(void) int multifd_send_sync_main(QEMUFile *f) { int i; +bool flush_zero_copy; if (!migrate_use_multifd()) { return 0; @@ -579,6 +580,14 @@ int multifd_send_sync_main(QEMUFile *f) return -1; } } + +/* + * When using zero-copy, it's necessary to flush after each iteration to + * make sure pages from earlier iterations don't end up replacing newer + * pages. + */ +flush_zero_copy = migrate_use_zero_copy_send(); + for (i = 0; i < migrate_multifd_channels(); i++) { MultiFDSendParams *p = &multifd_send_state->params[i]; @@ -600,6 +609,17 @@ int multifd_send_sync_main(QEMUFile *f) ram_counters.transferred += p->packet_len; qemu_mutex_unlock(&p->mutex); qemu_sem_post(&p->sem); + +if (flush_zero_copy && p->c) { +int ret; +Error *err = NULL; + +ret = qio_channel_flush(p->c, &err); +if (ret < 0) { +error_report_err(err); +return -1; +} +} } for (i = 0; i < migrate_multifd_channels(); i++) { MultiFDSendParams *p = &multifd_send_state->params[i]; @@ -688,10 +708,9 @@ static void *multifd_send_thread(void *opaque) p->iov[0].iov_base = p->packet; } -ret = qio_channel_writev_all(p->c, p->iov + iov_offset, - p->iovs_num - iov_offset, - &local_err); - +ret = qio_channel_writev_full_all(p->c, p->iov + iov_offset, + p->iovs_num - iov_offset, NULL, + 0, p->write_flags, &local_err); if (ret != 0) { break; } @@ -920,6 +939,13 @@ int multifd_save_setup(Error **errp) /* We need one extra place for the packet header */ p->iov = g_new0(struct iovec, page_count + 1); p->normal = g_new0(ram_addr_t, page_count); + +if (migrate_use_zero_copy_send()) { +p->write_flags = QIO_CHANNEL_WRITE_FLAG_ZERO_COPY; +} else { +p->write_flags = 0; +} + socket_send_channel_create(multifd_new_send_channel_async, p); } diff --g
Re: [PATCH v9 7/7] multifd: Implement zero copy write in multifd migration (multifd-zero-copy)
Leo, This patch looks mostly good to me, a few nitpicks below. On Mon, Apr 25, 2022 at 06:50:56PM -0300, Leonardo Bras wrote: > Implement zero copy send on nocomp_send_write(), by making use of QIOChannel > writev + flags & flush interface. > > Change multifd_send_sync_main() so flush_zero_copy() can be called > after each iteration in order to make sure all dirty pages are sent before > a new iteration is started. It will also flush at the beginning and at the > end of migration. > > Also make it return -1 if flush_zero_copy() fails, in order to cancel > the migration process, and avoid resuming the guest in the target host > without receiving all current RAM. > > This will work fine on RAM migration because the RAM pages are not usually > freed, > and there is no problem on changing the pages content between > writev_zero_copy() and > the actual sending of the buffer, because this change will dirty the page and > cause it to be re-sent on a next iteration anyway. > > A lot of locked memory may be needed in order to use multifd migration > with zero-copy enabled, so disabling the feature should be necessary for > low-privileged users trying to perform multifd migrations. > > Signed-off-by: Leonardo Bras > --- > migration/multifd.h | 2 ++ > migration/migration.c | 11 ++- > migration/multifd.c | 34 ++ > migration/socket.c| 5 +++-- > 4 files changed, 45 insertions(+), 7 deletions(-) > > diff --git a/migration/multifd.h b/migration/multifd.h > index bcf5992945..4d8d89e5e5 100644 > --- a/migration/multifd.h > +++ b/migration/multifd.h > @@ -92,6 +92,8 @@ typedef struct { > uint32_t packet_len; > /* pointer to the packet */ > MultiFDPacket_t *packet; > +/* multifd flags for sending ram */ > +int write_flags; > /* multifd flags for each packet */ > uint32_t flags; > /* size of the next packet that contains pages */ > diff --git a/migration/migration.c b/migration/migration.c > index 4b6df2eb5e..31739b2af9 100644 > --- a/migration/migration.c > +++ b/migration/migration.c > @@ -1497,7 +1497,16 @@ static bool migrate_params_check(MigrationParameters > *params, Error **errp) > error_prepend(errp, "Invalid mapping given for block-bitmap-mapping: > "); > return false; > } > - > +#ifdef CONFIG_LINUX > +if (params->zero_copy_send && > +(!migrate_use_multifd() || > + params->multifd_compression != MULTIFD_COMPRESSION_NONE || > + (params->tls_creds && *params->tls_creds))) { > +error_setg(errp, > + "Zero copy only available for non-compressed non-TLS > multifd migration"); > +return false; > +} > +#endif > return true; > } > > diff --git a/migration/multifd.c b/migration/multifd.c > index 6c940aaa98..e37cc6e0d9 100644 > --- a/migration/multifd.c > +++ b/migration/multifd.c > @@ -569,6 +569,7 @@ void multifd_save_cleanup(void) > int multifd_send_sync_main(QEMUFile *f) > { > int i; > +bool flush_zero_copy; > > if (!migrate_use_multifd()) { > return 0; > @@ -579,6 +580,14 @@ int multifd_send_sync_main(QEMUFile *f) > return -1; > } > } > + > +/* > + * When using zero-copy, it's necessary to flush after each iteration to > + * make sure pages from earlier iterations don't end up replacing newer > + * pages. > + */ > +flush_zero_copy = migrate_use_zero_copy_send(); Would you mind inline it if it's only used once? It's great to have that comment, but IMHO it could be more explicit, even marking a TODO showing that maybe we could do better in the future: /* * When using zero-copy, it's necessary to flush the pages before any of * the pages can be sent again, so we'll make sure the new version of the * pages will always arrive _later_ than the old pages. * * Currently we achieve this by flushing the zero-page requested writes * per ram iteration, but in the future we could potentially optimize it * to be less frequent, e.g. only after we finished one whole scanning of * all the dirty bitmaps. */ > + > for (i = 0; i < migrate_multifd_channels(); i++) { > MultiFDSendParams *p = &multifd_send_state->params[i]; > > @@ -600,6 +609,17 @@ int multifd_send_sync_main(QEMUFile *f) > ram_counters.transferred += p->packet_len; > qemu_mutex_unlock(&p->mutex); > qemu_sem_post(&p->sem); > + > +if (flush_zero_copy && p->c) { > +int ret; > +Error *err = NULL; > + > +ret = qio_channel_flush(p->c, &err); > +if (ret < 0) { > +error_report_err(err); > +return -1; > +} > +} > } > for (i = 0; i < migrate_multifd_channels(); i++) { > MultiFDSendParams *p = &multifd_send_state->params[i]; > @@ -688,10 +708,9 @@ static void *multifd_send_thread(void *opaque) > p-
Re: [PATCH v9 7/7] multifd: Implement zero copy write in multifd migration (multifd-zero-copy)
Hello Peter, thanks for helping! On Tue, Apr 26, 2022 at 1:02 PM Peter Xu wrote: > > Leo, > > This patch looks mostly good to me, a few nitpicks below. > > On Mon, Apr 25, 2022 at 06:50:56PM -0300, Leonardo Bras wrote: [...] > > } > > + > > +/* > > + * When using zero-copy, it's necessary to flush after each iteration > > to > > + * make sure pages from earlier iterations don't end up replacing newer > > + * pages. > > + */ > > +flush_zero_copy = migrate_use_zero_copy_send(); > > Would you mind inline it if it's only used once? It's not obvious in the diff, but this is used in a loop bellow, so I inserted the variable to avoid calling migrate_use_zero_copy_send() for each multifd channel. > > It's great to have that comment, but IMHO it could be more explicit, even > marking a TODO showing that maybe we could do better in the future: > > /* >* When using zero-copy, it's necessary to flush the pages before any of >* the pages can be sent again, so we'll make sure the new version of the >* pages will always arrive _later_ than the old pages. >* >* Currently we achieve this by flushing the zero-page requested writes >* per ram iteration, but in the future we could potentially optimize it >* to be less frequent, e.g. only after we finished one whole scanning of >* all the dirty bitmaps. >*/ > Thanks! I will insert that in the next version. The thing here is that I was under the impression an iteration was equivalent to a whole scanning of all the dirty bitmaps. I see now that it may not be the case. [...] > > @@ -688,10 +708,9 @@ static void *multifd_send_thread(void *opaque) > > p->iov[0].iov_base = p->packet; > > } > > > > -ret = qio_channel_writev_all(p->c, p->iov + iov_offset, > > - p->iovs_num - iov_offset, > > - &local_err); > > - > > +ret = qio_channel_writev_full_all(p->c, p->iov + iov_offset, > > + p->iovs_num - iov_offset, > > NULL, > > + 0, p->write_flags, > > &local_err); > > I kind of agree with Dan in previous patch - this iov_offset is confusing, > better drop it. Sure, fixed for v10. > [...] > -- > Peter Xu > Best regards, Leo