motorhead/qemu
1
0
Fork 0
mirror of https://github.com/Motorhead1991/qemu.git synced 2025-08-06 09:13:55 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

multifd: Create page_size fields into both MultiFD{Recv,Send}Params

Browse source 
We were calling qemu_target_page_size() left and right.

Signed-off-by: Juan Quintela <quintela@redhat.com>
Reviewed-by: Leonardo Bras <leobras@redhat.com>
This commit is contained in:
Juan Quintela 2022-05-02 16:45:35 +02:00
parent 5204b499a6
commit ddec20f858
4 changed files with 23 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

14
migration/multifd-zlib.c
View file

@ -116,7 +116,6 @@ static void zlib_send_cleanup(MultiFDSendParams *p, Error **errp)
static int zlib_send_prepare(MultiFDSendParams *p, Error **errp) static int zlib_send_prepare(MultiFDSendParams *p, Error **errp)
{ {
struct zlib_data *z = p->data; struct zlib_data *z = p->data;
size_t page_size = qemu_target_page_size();
z_stream *zs = &z->zs; z_stream *zs = &z->zs;
uint32_t out_size = 0; uint32_t out_size = 0;
int ret; int ret;
@ -135,8 +134,8 @@ static int zlib_send_prepare(MultiFDSendParams *p, Error **errp)
* with compression. zlib does not guarantee that this is safe, * with compression. zlib does not guarantee that this is safe,
* therefore copy the page before calling deflate(). * therefore copy the page before calling deflate().
*/ */
memcpy(z->buf, p->pages->block->host + p->normal[i], page_size); memcpy(z->buf, p->pages->block->host + p->normal[i], p->page_size);
zs->avail_in = page_size; zs->avail_in = p->page_size;
zs->next_in = z->buf; zs->next_in = z->buf;
zs->avail_out = available; zs->avail_out = available;
@ -242,12 +241,11 @@ static void zlib_recv_cleanup(MultiFDRecvParams *p)
static int zlib_recv_pages(MultiFDRecvParams *p, Error **errp) static int zlib_recv_pages(MultiFDRecvParams *p, Error **errp)
{ {
struct zlib_data *z = p->data; struct zlib_data *z = p->data;
size_t page_size = qemu_target_page_size();
z_stream *zs = &z->zs; z_stream *zs = &z->zs;
uint32_t in_size = p->next_packet_size; uint32_t in_size = p->next_packet_size;
/* we measure the change of total_out */ /* we measure the change of total_out */
uint32_t out_size = zs->total_out; uint32_t out_size = zs->total_out;
uint32_t expected_size = p->normal_num * page_size; uint32_t expected_size = p->normal_num * p->page_size;
uint32_t flags = p->flags & MULTIFD_FLAG_COMPRESSION_MASK; uint32_t flags = p->flags & MULTIFD_FLAG_COMPRESSION_MASK;
int ret; int ret;
int i; int i;
@ -274,7 +272,7 @@ static int zlib_recv_pages(MultiFDRecvParams *p, Error **errp)
flush = Z_SYNC_FLUSH; flush = Z_SYNC_FLUSH;
} }
zs->avail_out = page_size; zs->avail_out = p->page_size;
zs->next_out = p->host + p->normal[i]; zs->next_out = p->host + p->normal[i];
/* /*
@ -288,8 +286,8 @@ static int zlib_recv_pages(MultiFDRecvParams *p, Error **errp)
do { do {
ret = inflate(zs, flush); ret = inflate(zs, flush);
} while (ret == Z_OK && zs->avail_in } while (ret == Z_OK && zs->avail_in
&& (zs->total_out - start) < page_size); && (zs->total_out - start) < p->page_size);
if (ret == Z_OK && (zs->total_out - start) < page_size) { if (ret == Z_OK && (zs->total_out - start) < p->page_size) {
error_setg(errp, "multifd %u: inflate generated too few output", error_setg(errp, "multifd %u: inflate generated too few output",
p->id); p->id);
return -1; return -1;

12
migration/multifd-zstd.c
View file

@ -113,7 +113,6 @@ static void zstd_send_cleanup(MultiFDSendParams *p, Error **errp)
static int zstd_send_prepare(MultiFDSendParams *p, Error **errp) static int zstd_send_prepare(MultiFDSendParams *p, Error **errp)
{ {
struct zstd_data *z = p->data; struct zstd_data *z = p->data;
size_t page_size = qemu_target_page_size();
int ret; int ret;
uint32_t i; uint32_t i;
@ -128,7 +127,7 @@ static int zstd_send_prepare(MultiFDSendParams *p, Error **errp)
flush = ZSTD_e_flush; flush = ZSTD_e_flush;
} }
z->in.src = p->pages->block->host + p->normal[i]; z->in.src = p->pages->block->host + p->normal[i];
z->in.size = page_size; z->in.size = p->page_size;
z->in.pos = 0; z->in.pos = 0;
/* /*
@ -241,8 +240,7 @@ static int zstd_recv_pages(MultiFDRecvParams *p, Error **errp)
{ {
uint32_t in_size = p->next_packet_size; uint32_t in_size = p->next_packet_size;
uint32_t out_size = 0; uint32_t out_size = 0;
size_t page_size = qemu_target_page_size(); uint32_t expected_size = p->normal_num * p->page_size;
uint32_t expected_size = p->normal_num * page_size;
uint32_t flags = p->flags & MULTIFD_FLAG_COMPRESSION_MASK; uint32_t flags = p->flags & MULTIFD_FLAG_COMPRESSION_MASK;
struct zstd_data *z = p->data; struct zstd_data *z = p->data;
int ret; int ret;
@ -265,7 +263,7 @@ static int zstd_recv_pages(MultiFDRecvParams *p, Error **errp)
for (i = 0; i < p->normal_num; i++) { for (i = 0; i < p->normal_num; i++) {
z->out.dst = p->host + p->normal[i]; z->out.dst = p->host + p->normal[i];
z->out.size = page_size; z->out.size = p->page_size;
z->out.pos = 0; z->out.pos = 0;
/* /*
@ -279,8 +277,8 @@ static int zstd_recv_pages(MultiFDRecvParams *p, Error **errp)
do { do {
ret = ZSTD_decompressStream(z->zds, &z->out, &z->in); ret = ZSTD_decompressStream(z->zds, &z->out, &z->in);
} while (ret > 0 && (z->in.size - z->in.pos > 0) } while (ret > 0 && (z->in.size - z->in.pos > 0)
&& (z->out.pos < page_size)); && (z->out.pos < p->page_size));
if (ret > 0 && (z->out.pos < page_size)) { if (ret > 0 && (z->out.pos < p->page_size)) {
error_setg(errp, "multifd %u: decompressStream buffer too small", error_setg(errp, "multifd %u: decompressStream buffer too small",
p->id); p->id);
return -1; return -1;

18
migration/multifd.c
View file

@ -87,15 +87,14 @@ static void nocomp_send_cleanup(MultiFDSendParams *p, Error **errp)
static int nocomp_send_prepare(MultiFDSendParams *p, Error **errp) static int nocomp_send_prepare(MultiFDSendParams *p, Error **errp)
{ {
MultiFDPages_t *pages = p->pages; MultiFDPages_t *pages = p->pages;
size_t page_size = qemu_target_page_size();
for (int i = 0; i < p->normal_num; i++) { for (int i = 0; i < p->normal_num; i++) {
p->iov[p->iovs_num].iov_base = pages->block->host + p->normal[i]; p->iov[p->iovs_num].iov_base = pages->block->host + p->normal[i];
p->iov[p->iovs_num].iov_len = page_size; p->iov[p->iovs_num].iov_len = p->page_size;
p->iovs_num++; p->iovs_num++;
} }
p->next_packet_size = p->normal_num * page_size; p->next_packet_size = p->normal_num * p->page_size;
p->flags |= MULTIFD_FLAG_NOCOMP; p->flags |= MULTIFD_FLAG_NOCOMP;
return 0; return 0;
} }
@ -139,7 +138,6 @@ static void nocomp_recv_cleanup(MultiFDRecvParams *p)
static int nocomp_recv_pages(MultiFDRecvParams *p, Error **errp) static int nocomp_recv_pages(MultiFDRecvParams *p, Error **errp)
{ {
uint32_t flags = p->flags & MULTIFD_FLAG_COMPRESSION_MASK; uint32_t flags = p->flags & MULTIFD_FLAG_COMPRESSION_MASK;
size_t page_size = qemu_target_page_size();
if (flags != MULTIFD_FLAG_NOCOMP) { if (flags != MULTIFD_FLAG_NOCOMP) {
error_setg(errp, "multifd %u: flags received %x flags expected %x", error_setg(errp, "multifd %u: flags received %x flags expected %x",
@ -148,7 +146,7 @@ static int nocomp_recv_pages(MultiFDRecvParams *p, Error **errp)
} }
for (int i = 0; i < p->normal_num; i++) { for (int i = 0; i < p->normal_num; i++) {
p->iov[i].iov_base = p->host + p->normal[i]; p->iov[i].iov_base = p->host + p->normal[i];
p->iov[i].iov_len = page_size; p->iov[i].iov_len = p->page_size;
} }
return qio_channel_readv_all(p->c, p->iov, p->normal_num, errp); return qio_channel_readv_all(p->c, p->iov, p->normal_num, errp);
} }
@ -281,8 +279,7 @@ static void multifd_send_fill_packet(MultiFDSendParams *p)
static int multifd_recv_unfill_packet(MultiFDRecvParams *p, Error **errp) static int multifd_recv_unfill_packet(MultiFDRecvParams *p, Error **errp)
{ {
MultiFDPacket_t *packet = p->packet; MultiFDPacket_t *packet = p->packet;
size_t page_size = qemu_target_page_size(); uint32_t page_count = MULTIFD_PACKET_SIZE / p->page_size;
uint32_t page_count = MULTIFD_PACKET_SIZE / page_size;
RAMBlock *block; RAMBlock *block;
int i; int i;
@ -344,7 +341,7 @@ static int multifd_recv_unfill_packet(MultiFDRecvParams *p, Error **errp)
for (i = 0; i < p->normal_num; i++) { for (i = 0; i < p->normal_num; i++) {
uint64_t offset = be64_to_cpu(packet->offset[i]); uint64_t offset = be64_to_cpu(packet->offset[i]);
if (offset > (block->used_length - page_size)) { if (offset > (block->used_length - p->page_size)) {
error_setg(errp, "multifd: offset too long %" PRIu64 error_setg(errp, "multifd: offset too long %" PRIu64
" (max " RAM_ADDR_FMT ")", " (max " RAM_ADDR_FMT ")",
offset, block->used_length); offset, block->used_length);
@ -433,8 +430,7 @@ static int multifd_send_pages(QEMUFile *f)
p->packet_num = multifd_send_state->packet_num++; p->packet_num = multifd_send_state->packet_num++;
multifd_send_state->pages = p->pages; multifd_send_state->pages = p->pages;
p->pages = pages; p->pages = pages;
transferred = ((uint64_t) pages->num) * qemu_target_page_size() transferred = ((uint64_t) pages->num) * p->page_size + p->packet_len;
+ p->packet_len;
qemu_file_acct_rate_limit(f, transferred); qemu_file_acct_rate_limit(f, transferred);
ram_counters.multifd_bytes += transferred; ram_counters.multifd_bytes += transferred;
ram_counters.transferred += transferred; ram_counters.transferred += transferred;
@ -947,6 +943,7 @@ int multifd_save_setup(Error **errp)
/* We need one extra place for the packet header */ /* We need one extra place for the packet header */
p->iov = g_new0(struct iovec, page_count + 1); p->iov = g_new0(struct iovec, page_count + 1);
p->normal = g_new0(ram_addr_t, page_count); p->normal = g_new0(ram_addr_t, page_count);
p->page_size = qemu_target_page_size();
if (migrate_use_zero_copy_send()) { if (migrate_use_zero_copy_send()) {
p->write_flags = QIO_CHANNEL_WRITE_FLAG_ZERO_COPY; p->write_flags = QIO_CHANNEL_WRITE_FLAG_ZERO_COPY;
@ -1194,6 +1191,7 @@ int multifd_load_setup(Error **errp)
p->name = g_strdup_printf("multifdrecv_%d", i); p->name = g_strdup_printf("multifdrecv_%d", i);
p->iov = g_new0(struct iovec, page_count); p->iov = g_new0(struct iovec, page_count);
p->normal = g_new0(ram_addr_t, page_count); p->normal = g_new0(ram_addr_t, page_count);
p->page_size = qemu_target_page_size();
} }
for (i = 0; i < thread_count; i++) { for (i = 0; i < thread_count; i++) {

4
migration/multifd.h
View file

@ -80,6 +80,8 @@ typedef struct {
bool registered_yank; bool registered_yank;
/* packet allocated len */ /* packet allocated len */
uint32_t packet_len; uint32_t packet_len;
/* guest page size */
uint32_t page_size;
/* multifd flags for sending ram */ /* multifd flags for sending ram */
int write_flags; int write_flags;
@ -143,6 +145,8 @@ typedef struct {
QIOChannel *c; QIOChannel *c;
/* packet allocated len */ /* packet allocated len */
uint32_t packet_len; uint32_t packet_len;
/* guest page size */
uint32_t page_size;
/* syncs main thread and channels */ /* syncs main thread and channels */
QemuSemaphore sem_sync; QemuSemaphore sem_sync;