forked from SLFO-pool/xen
94 lines
2.8 KiB
Diff
94 lines
2.8 KiB
Diff
From: Olaf Hering <olaf@aepfle.de>
|
|
Date: Fri, 23 Oct 2020 14:39:31 +0200
|
|
Subject: libxc sr restore types
|
|
|
|
tools: restore: preallocate types array
|
|
|
|
Remove repeated allocation from migration loop. There will never be
|
|
more than MAX_BATCH_SIZE pages to process in an incoming batch.
|
|
Allocate the space once.
|
|
|
|
Signed-off-by: Olaf Hering <olaf@aepfle.de>
|
|
---
|
|
tools/libs/guest/xg_sr_common.h | 1 +
|
|
tools/libs/guest/xg_sr_restore.c | 22 +++++++---------------
|
|
2 files changed, 8 insertions(+), 15 deletions(-)
|
|
|
|
--- a/tools/libs/guest/xg_sr_common.h
|
|
+++ b/tools/libs/guest/xg_sr_common.h
|
|
@@ -257,6 +257,7 @@ struct xc_sr_context
|
|
struct xc_sr_restore_ops ops;
|
|
struct restore_callbacks *callbacks;
|
|
xen_pfn_t *pfns;
|
|
+ uint32_t *types;
|
|
|
|
int send_back_fd;
|
|
unsigned long p2m_size;
|
|
--- a/tools/libs/guest/xg_sr_restore.c
|
|
+++ b/tools/libs/guest/xg_sr_restore.c
|
|
@@ -315,7 +315,7 @@ static int handle_page_data(struct xc_sr
|
|
int rc = -1;
|
|
|
|
xen_pfn_t pfn;
|
|
- uint32_t *types = NULL, type;
|
|
+ uint32_t type;
|
|
|
|
/*
|
|
* v2 compatibility only exists for x86 streams. This is a bit of a
|
|
@@ -362,14 +362,6 @@ static int handle_page_data(struct xc_sr
|
|
goto err;
|
|
}
|
|
|
|
- types = malloc(pages->count * sizeof(*types));
|
|
- if ( !types )
|
|
- {
|
|
- ERROR("Unable to allocate enough memory for %u pfns",
|
|
- pages->count);
|
|
- goto err;
|
|
- }
|
|
-
|
|
for ( i = 0; i < pages->count; ++i )
|
|
{
|
|
pfn = pages->pfn[i] & PAGE_DATA_PFN_MASK;
|
|
@@ -393,7 +385,7 @@ static int handle_page_data(struct xc_sr
|
|
pages_of_data++;
|
|
|
|
ctx->restore.pfns[i] = pfn;
|
|
- types[i] = type;
|
|
+ ctx->restore.types[i] = type;
|
|
}
|
|
|
|
if ( rec->length != (sizeof(*pages) +
|
|
@@ -406,11 +398,9 @@ static int handle_page_data(struct xc_sr
|
|
goto err;
|
|
}
|
|
|
|
- rc = process_page_data(ctx, pages->count, ctx->restore.pfns, types,
|
|
- &pages->pfn[pages->count]);
|
|
+ rc = process_page_data(ctx, pages->count, ctx->restore.pfns,
|
|
+ ctx->restore.types, &pages->pfn[pages->count]);
|
|
err:
|
|
- free(types);
|
|
-
|
|
return rc;
|
|
}
|
|
|
|
@@ -714,7 +704,8 @@ static int setup(struct xc_sr_context *c
|
|
}
|
|
|
|
ctx->restore.pfns = malloc(MAX_BATCH_SIZE * sizeof(*ctx->restore.pfns));
|
|
- if ( !ctx->restore.pfns )
|
|
+ ctx->restore.types = malloc(MAX_BATCH_SIZE * sizeof(*ctx->restore.types));
|
|
+ if ( !ctx->restore.pfns || !ctx->restore.types )
|
|
{
|
|
ERROR("Unable to allocate memory");
|
|
rc = -1;
|
|
@@ -751,6 +742,7 @@ static void cleanup(struct xc_sr_context
|
|
|
|
free(ctx->restore.buffered_records);
|
|
free(ctx->restore.populated_pfns);
|
|
+ free(ctx->restore.types);
|
|
free(ctx->restore.pfns);
|
|
|
|
if ( ctx->restore.ops.cleanup(ctx) )
|