yeahuh / rpms / qemu-kvm

Forked from rpms/qemu-kvm 2 years ago
Clone
9ae3a8
From b75b57e4a5b75f03c5d302ced1287f47de6c9ccc Mon Sep 17 00:00:00 2001
9ae3a8
From: Orit Wasserman <owasserm@redhat.com>
9ae3a8
Date: Tue, 11 Feb 2014 15:32:37 +0100
9ae3a8
Subject: [PATCH 16/28] Don't abort on memory allocation error
9ae3a8
9ae3a8
RH-Author: Orit Wasserman <owasserm@redhat.com>
9ae3a8
Message-id: <1392132757-18587-4-git-send-email-owasserm@redhat.com>
9ae3a8
Patchwork-id: 57211
9ae3a8
O-Subject: [RHEL7 qemu-kvm PATCH 3/3] Don't abort on memory allocation error
9ae3a8
Bugzilla: 1047448
9ae3a8
RH-Acked-by: Dr. David Alan Gilbert (git) <dgilbert@redhat.com>
9ae3a8
RH-Acked-by: Juan Quintela <quintela@redhat.com>
9ae3a8
RH-Acked-by: Miroslav Rezanina <mrezanin@redhat.com>
9ae3a8
9ae3a8
It is better to fail migration in case of failure to
9ae3a8
allocate new cache item
9ae3a8
9ae3a8
Signed-off-by: Orit Wasserman <owasserm@redhat.com>
9ae3a8
Reviewed-by: Dr. David Alan Gilbert <dgilbert@redhat.com>
9ae3a8
Signed-off-by: Juan Quintela <quintela@redhat.com>
9ae3a8
(cherry picked from commit 89db9987c07977bdb78d5d4b41d65e7acb9a5a2c)
9ae3a8
---
9ae3a8
 arch_init.c                    |  4 +++-
9ae3a8
 include/migration/page_cache.h |  4 +++-
9ae3a8
 page_cache.c                   | 16 +++++++++++-----
9ae3a8
 3 files changed, 17 insertions(+), 7 deletions(-)
9ae3a8
9ae3a8
Signed-off-by: Miroslav Rezanina <mrezanin@redhat.com>
9ae3a8
---
9ae3a8
 arch_init.c                    |    4 +++-
9ae3a8
 include/migration/page_cache.h |    4 +++-
9ae3a8
 page_cache.c                   |   16 +++++++++++-----
9ae3a8
 3 files changed, 17 insertions(+), 7 deletions(-)
9ae3a8
9ae3a8
diff --git a/arch_init.c b/arch_init.c
9ae3a8
index fc0f569..0aacdac 100644
9ae3a8
--- a/arch_init.c
9ae3a8
+++ b/arch_init.c
9ae3a8
@@ -286,7 +286,9 @@ static int save_xbzrle_page(QEMUFile *f, uint8_t *current_data,
9ae3a8
 
9ae3a8
     if (!cache_is_cached(XBZRLE.cache, current_addr)) {
9ae3a8
         if (!last_stage) {
9ae3a8
-            cache_insert(XBZRLE.cache, current_addr, current_data);
9ae3a8
+            if (cache_insert(XBZRLE.cache, current_addr, current_data) == -1) {
9ae3a8
+                return -1;
9ae3a8
+            }
9ae3a8
         }
9ae3a8
         acct_info.xbzrle_cache_miss++;
9ae3a8
         return -1;
9ae3a8
diff --git a/include/migration/page_cache.h b/include/migration/page_cache.h
9ae3a8
index 87894fe..d156f0d 100644
9ae3a8
--- a/include/migration/page_cache.h
9ae3a8
+++ b/include/migration/page_cache.h
9ae3a8
@@ -60,11 +60,13 @@ uint8_t *get_cached_data(const PageCache *cache, uint64_t addr);
9ae3a8
  * cache_insert: insert the page into the cache. the page cache
9ae3a8
  * will dup the data on insert. the previous value will be overwritten
9ae3a8
  *
9ae3a8
+ * Returns -1 on error
9ae3a8
+ *
9ae3a8
  * @cache pointer to the PageCache struct
9ae3a8
  * @addr: page address
9ae3a8
  * @pdata: pointer to the page
9ae3a8
  */
9ae3a8
-void cache_insert(PageCache *cache, uint64_t addr, uint8_t *pdata);
9ae3a8
+int cache_insert(PageCache *cache, uint64_t addr, uint8_t *pdata);
9ae3a8
 
9ae3a8
 /**
9ae3a8
  * cache_resize: resize the page cache. In case of size reduction the extra
9ae3a8
diff --git a/page_cache.c b/page_cache.c
9ae3a8
index 2920123..250772d 100644
9ae3a8
--- a/page_cache.c
9ae3a8
+++ b/page_cache.c
9ae3a8
@@ -151,7 +151,7 @@ uint8_t *get_cached_data(const PageCache *cache, uint64_t addr)
9ae3a8
     return cache_get_by_addr(cache, addr)->it_data;
9ae3a8
 }
9ae3a8
 
9ae3a8
-void cache_insert(PageCache *cache, uint64_t addr, uint8_t *pdata)
9ae3a8
+int cache_insert(PageCache *cache, uint64_t addr, uint8_t *pdata)
9ae3a8
 {
9ae3a8
 
9ae3a8
     CacheItem *it = NULL;
9ae3a8
@@ -162,16 +162,22 @@ void cache_insert(PageCache *cache, uint64_t addr, uint8_t *pdata)
9ae3a8
     /* actual update of entry */
9ae3a8
     it = cache_get_by_addr(cache, addr);
9ae3a8
 
9ae3a8
-    /* free old cached data if any */
9ae3a8
-    g_free(it->it_data);
9ae3a8
-
9ae3a8
+    /* allocate page */
9ae3a8
     if (!it->it_data) {
9ae3a8
+        it->it_data = g_try_malloc(cache->page_size);
9ae3a8
+        if (!it->it_data) {
9ae3a8
+            DPRINTF("Error allocating page\n");
9ae3a8
+            return -1;
9ae3a8
+        }
9ae3a8
         cache->num_items++;
9ae3a8
     }
9ae3a8
 
9ae3a8
-    it->it_data = g_memdup(pdata, cache->page_size);
9ae3a8
+    memcpy(it->it_data, pdata, cache->page_size);
9ae3a8
+
9ae3a8
     it->it_age = ++cache->max_item_age;
9ae3a8
     it->it_addr = addr;
9ae3a8
+
9ae3a8
+    return 0;
9ae3a8
 }
9ae3a8
 
9ae3a8
 int64_t cache_resize(PageCache *cache, int64_t new_num_pages)
9ae3a8
-- 
9ae3a8
1.7.1
9ae3a8