palacios_alloc() patch

diff --git a/linux_module/palacios-stubs.c b/linux_module/palacios-stubs.c
index be06001..0310a40 100644 (file)
--- a/linux_module/palacios-stubs.c
+++ b/linux_module/palacios-stubs.c
@@ -85,7 +85,11 @@ static void *
 palacios_alloc(unsigned int size) {
     void * addr = NULL;
 
-    addr = kmalloc(size, GFP_KERNEL);
+    if (irqs_disabled()) {
+       addr = kmalloc(size, GFP_ATOMIC);
+    } else {
+       addr = kmalloc(size, GFP_KERNEL);
+    }
     mallocs++;
  
     return addr;