X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_module%2Fiface-code-inject.c;h=1262d772d1db80139dcdb43305efc0a708d2ec95;hb=2cb41f7db5b9f89113432d6b3daff4807ba8e5f2;hp=d0b549576fd5d9dd9d86a16f6e23615f32fc02e9;hpb=791ea2f3e21cfbc9c47341efbb98995c33d86fcb;p=palacios.git

diff --git a/linux_module/iface-code-inject.c b/linux_module/iface-code-inject.c
index d0b5495..1262d77 100644
--- a/linux_module/iface-code-inject.c
+++ b/linux_module/iface-code-inject.c
@@ -7,7 +7,6 @@
 
 #include <linux/elf.h>
 #include <linux/uaccess.h>
-#include <linux/vmalloc.h>
 
 #include <linux/module.h>
 
@@ -43,8 +42,8 @@ static void free_inject_data (void) {
 
     for(i = 0; i < MAX_INJ; i++) {
         if (top_map[i]) {
-            kfree(top_map[i]->elf_data);
-            kfree(top_map[i]);
+            palacios_free(top_map[i]->elf_data);
+            palacios_free(top_map[i]);
         }
     }
 }
@@ -55,8 +54,8 @@ static int vm_tophalf_inject (struct v3_guest * guest, unsigned int cmd, unsigne
     struct top_half_data top_arg;
     struct top_half_data * top;
 
-    top = kmalloc(sizeof(struct top_half_data), GFP_KERNEL);
-    if (IS_ERR(top)) {
+    top = palacios_alloc(sizeof(struct top_half_data));
+    if (!top) {
         ERROR("Palacios Error: could not allocate space for top half data\n");
         return -EFAULT;
     }
@@ -65,6 +64,7 @@ static int vm_tophalf_inject (struct v3_guest * guest, unsigned int cmd, unsigne
     INFO("Palacios: Loading ELF data...\n");
     if (copy_from_user(&top_arg, (void __user *)arg, sizeof(struct top_half_data))) {
         ERROR("palacios: error copying ELF from userspace\n");
+	palacios_free(top);
         return -EFAULT;
     }
 
@@ -80,9 +80,10 @@ static int vm_tophalf_inject (struct v3_guest * guest, unsigned int cmd, unsigne
     } 
 
     DEBUG("Palacios: Allocating %lu B of kernel memory for ELF binary data...\n", top->elf_size);
-    top->elf_data = kmalloc(top->elf_size, GFP_KERNEL);
-    if (IS_ERR(top->elf_data)) {
+    top->elf_data = palacios_alloc(top->elf_size);
+    if (!(top->elf_data)) {
         ERROR("Palacios Error: could not allocate space for binary image\n");
+	palacios_free(top);
         return -EFAULT;
     }
     memset(top->elf_data, 0, top->elf_size);
@@ -90,16 +91,24 @@ static int vm_tophalf_inject (struct v3_guest * guest, unsigned int cmd, unsigne
     INFO("Palacios: Copying ELF image into kernel module...\n");
     if (copy_from_user(top->elf_data, (void __user *)top_arg.elf_data, top->elf_size)) {
         ERROR("Palacios: Error loading elf data\n");
+	palacios_free(top->elf_data);
+	palacios_free(top);
         return -EFAULT;
     }
 
-    if (register_top(top) < 0) 
+    if (register_top(top) < 0) {
+	ERROR("Cannot register top half\n");
+	palacios_free(top->elf_data);
+	palacios_free(top);
         return -1;
-    
+    }
+
     INFO("Palacios: setting up inject code...\n");
     if (v3_insert_code_inject(guest->v3_ctx, top->elf_data, top->elf_size, 
                          top->bin_file, top->is_dyn, top->is_exec_hooked, top->func_offset) < 0) {
         ERROR("Palacios Error: error setting up inject code\n");
+	palacios_free(top->elf_data);
+	palacios_free(top);
         return -1;
     }
 
@@ -126,6 +135,7 @@ static int guest_init_code_inject (struct v3_guest * guest, void ** vm_data) {
 
 static int guest_deinit_code_inject (struct v3_guest * guest, void * vm_data) {
     free_inject_data();
+    remove_guest_ctrl(guest, V3_VM_TOPHALF_INJECT);
     return 0;
 }