yeahuh / rpms / qemu-kvm

Forked from rpms/qemu-kvm 2 years ago
Clone
9ae3a8
From 90fad0c40a1947e229fdf82a30219e33a3e1dad8 Mon Sep 17 00:00:00 2001
9ae3a8
From: Max Reitz <mreitz@redhat.com>
9ae3a8
Date: Sat, 13 Jun 2015 16:21:57 +0200
9ae3a8
Subject: [PATCH 03/42] block: Introduce qemu_try_blockalign()
9ae3a8
9ae3a8
Message-id: <1434212556-3927-4-git-send-email-mreitz@redhat.com>
9ae3a8
Patchwork-id: 66022
9ae3a8
O-Subject: [RHEL-7.2 qemu-kvm PATCH 03/42] block: Introduce qemu_try_blockalign()
9ae3a8
Bugzilla: 1129893
9ae3a8
RH-Acked-by: Jeffrey Cody <jcody@redhat.com>
9ae3a8
RH-Acked-by: Fam Zheng <famz@redhat.com>
9ae3a8
RH-Acked-by: Stefan Hajnoczi <stefanha@redhat.com>
9ae3a8
9ae3a8
From: Kevin Wolf <kwolf@redhat.com>
9ae3a8
9ae3a8
BZ: 1129893
9ae3a8
9ae3a8
This function returns NULL instead of aborting when an allocation fails.
9ae3a8
9ae3a8
Signed-off-by: Kevin Wolf <kwolf@redhat.com>
9ae3a8
Reviewed-by: Benoit Canet <benoit@irqsave.net>
9ae3a8
(cherry picked from commit 7d2a35cc921ea4832083a7e8598461868bb538ce)
9ae3a8
9ae3a8
Signed-off-by: Max Reitz <mreitz@redhat.com>
9ae3a8
Signed-off-by: Miroslav Rezanina <mrezanin@redhat.com>
9ae3a8
---
9ae3a8
 block.c               | 13 +++++++++++++
9ae3a8
 include/block/block.h |  1 +
9ae3a8
 include/qemu/osdep.h  |  1 +
9ae3a8
 util/oslib-posix.c    | 16 ++++++++++------
9ae3a8
 util/oslib-win32.c    |  9 +++++++--
9ae3a8
 5 files changed, 32 insertions(+), 8 deletions(-)
9ae3a8
9ae3a8
diff --git a/block.c b/block.c
9ae3a8
index f2caf20..caea2ea 100644
9ae3a8
--- a/block.c
9ae3a8
+++ b/block.c
9ae3a8
@@ -5175,6 +5175,19 @@ void *qemu_blockalign(BlockDriverState *bs, size_t size)
9ae3a8
     return qemu_memalign(bdrv_opt_mem_align(bs), size);
9ae3a8
 }
9ae3a8
 
9ae3a8
+void *qemu_try_blockalign(BlockDriverState *bs, size_t size)
9ae3a8
+{
9ae3a8
+    size_t align = bdrv_opt_mem_align(bs);
9ae3a8
+
9ae3a8
+    /* Ensure that NULL is never returned on success */
9ae3a8
+    assert(align > 0);
9ae3a8
+    if (size == 0) {
9ae3a8
+        size = align;
9ae3a8
+    }
9ae3a8
+
9ae3a8
+    return qemu_try_memalign(align, size);
9ae3a8
+}
9ae3a8
+
9ae3a8
 /*
9ae3a8
  * Check if all memory in this vector is sector aligned.
9ae3a8
  */
9ae3a8
diff --git a/include/block/block.h b/include/block/block.h
9ae3a8
index 3170cbc..7b538b7 100644
9ae3a8
--- a/include/block/block.h
9ae3a8
+++ b/include/block/block.h
9ae3a8
@@ -417,6 +417,7 @@ void bdrv_img_create(const char *filename, const char *fmt,
9ae3a8
 size_t bdrv_opt_mem_align(BlockDriverState *bs);
9ae3a8
 void bdrv_set_guest_block_size(BlockDriverState *bs, int align);
9ae3a8
 void *qemu_blockalign(BlockDriverState *bs, size_t size);
9ae3a8
+void *qemu_try_blockalign(BlockDriverState *bs, size_t size);
9ae3a8
 bool bdrv_qiov_is_aligned(BlockDriverState *bs, QEMUIOVector *qiov);
9ae3a8
 
9ae3a8
 struct HBitmapIter;
9ae3a8
diff --git a/include/qemu/osdep.h b/include/qemu/osdep.h
9ae3a8
index 7a5ae28..8984da0 100644
9ae3a8
--- a/include/qemu/osdep.h
9ae3a8
+++ b/include/qemu/osdep.h
9ae3a8
@@ -95,6 +95,7 @@ typedef signed int              int_fast16_t;
9ae3a8
 #define qemu_printf printf
9ae3a8
 
9ae3a8
 int qemu_daemon(int nochdir, int noclose);
9ae3a8
+void *qemu_try_memalign(size_t alignment, size_t size);
9ae3a8
 void *qemu_memalign(size_t alignment, size_t size);
9ae3a8
 void *qemu_anon_ram_alloc(size_t size);
9ae3a8
 void qemu_vfree(void *ptr);
9ae3a8
diff --git a/util/oslib-posix.c b/util/oslib-posix.c
9ae3a8
index fef840a..f0fb297 100644
9ae3a8
--- a/util/oslib-posix.c
9ae3a8
+++ b/util/oslib-posix.c
9ae3a8
@@ -83,7 +83,7 @@ void *qemu_oom_check(void *ptr)
9ae3a8
     return ptr;
9ae3a8
 }
9ae3a8
 
9ae3a8
-void *qemu_memalign(size_t alignment, size_t size)
9ae3a8
+void *qemu_try_memalign(size_t alignment, size_t size)
9ae3a8
 {
9ae3a8
     void *ptr;
9ae3a8
 
9ae3a8
@@ -95,19 +95,23 @@ void *qemu_memalign(size_t alignment, size_t size)
9ae3a8
     int ret;
9ae3a8
     ret = posix_memalign(&ptr, alignment, size);
9ae3a8
     if (ret != 0) {
9ae3a8
-        fprintf(stderr, "Failed to allocate %zu B: %s\n",
9ae3a8
-                size, strerror(ret));
9ae3a8
-        abort();
9ae3a8
+        errno = ret;
9ae3a8
+        ptr = NULL;
9ae3a8
     }
9ae3a8
 #elif defined(CONFIG_BSD)
9ae3a8
-    ptr = qemu_oom_check(valloc(size));
9ae3a8
+    ptr = valloc(size);
9ae3a8
 #else
9ae3a8
-    ptr = qemu_oom_check(memalign(alignment, size));
9ae3a8
+    ptr = memalign(alignment, size);
9ae3a8
 #endif
9ae3a8
     trace_qemu_memalign(alignment, size, ptr);
9ae3a8
     return ptr;
9ae3a8
 }
9ae3a8
 
9ae3a8
+void *qemu_memalign(size_t alignment, size_t size)
9ae3a8
+{
9ae3a8
+    return qemu_oom_check(qemu_try_memalign(alignment, size));
9ae3a8
+}
9ae3a8
+
9ae3a8
 /* alloc shared memory pages */
9ae3a8
 void *qemu_anon_ram_alloc(size_t size)
9ae3a8
 {
9ae3a8
diff --git a/util/oslib-win32.c b/util/oslib-win32.c
9ae3a8
index 332e743..0fb78ef 100644
9ae3a8
--- a/util/oslib-win32.c
9ae3a8
+++ b/util/oslib-win32.c
9ae3a8
@@ -46,18 +46,23 @@ void *qemu_oom_check(void *ptr)
9ae3a8
     return ptr;
9ae3a8
 }
9ae3a8
 
9ae3a8
-void *qemu_memalign(size_t alignment, size_t size)
9ae3a8
+void *qemu_try_memalign(size_t alignment, size_t size)
9ae3a8
 {
9ae3a8
     void *ptr;
9ae3a8
 
9ae3a8
     if (!size) {
9ae3a8
         abort();
9ae3a8
     }
9ae3a8
-    ptr = qemu_oom_check(VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE));
9ae3a8
+    ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE);
9ae3a8
     trace_qemu_memalign(alignment, size, ptr);
9ae3a8
     return ptr;
9ae3a8
 }
9ae3a8
 
9ae3a8
+void *qemu_memalign(size_t alignment, size_t size)
9ae3a8
+{
9ae3a8
+    return qemu_oom_check(qemu_try_memalign(alignment, size));
9ae3a8
+}
9ae3a8
+
9ae3a8
 void *qemu_anon_ram_alloc(size_t size)
9ae3a8
 {
9ae3a8
     void *ptr;
9ae3a8
-- 
9ae3a8
1.8.3.1
9ae3a8