X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_module%2Fiface-guard-mods.c;h=18ee4c323629067b14377081f141141107d7e212;hb=564dade4aee58eb392b960dd33823b5e030869ac;hp=67917d778f1739412c7ef245b877a0b6cc7f9307;hpb=324f9e9a6e83b8079926902a26e3f6bb5e624a06;p=palacios.git

diff --git a/linux_module/iface-guard-mods.c b/linux_module/iface-guard-mods.c
index 67917d7..18ee4c3 100644
--- a/linux_module/iface-guard-mods.c
+++ b/linux_module/iface-guard-mods.c
@@ -6,7 +6,6 @@
  */
 
 #include <linux/uaccess.h>
-#include <linux/vmalloc.h>
 #include <linux/module.h>
 #include <linux/types.h>
 #include <linux/list.h>
@@ -160,13 +159,13 @@ vm_register_mod (struct v3_guest * guest,
         return -1;
     }
 
-    kfree(gm->name);
-    kfree(gm->content_hash);
+    palacios_free(gm->name);
+    palacios_free(gm->content_hash);
     for (i = 0; i < gm->num_entries; i++) {
-        kfree(gm->entry_points[i].name);
+        palacios_free(gm->entry_points[i].name);
     }
-    kfree(gm->entry_points);
-    kfree(gm);
+    palacios_free(gm->entry_points);
+    palacios_free(gm);
     return 0;
 }
 
@@ -196,6 +195,7 @@ guest_init_guard_mods (struct v3_guest * guest, void ** vm_data)
 static int 
 guest_deinit_guard_mods (struct v3_guest * guest, void * vm_data) 
 {
+    remove_guest_ctrl(guest, V3_VM_REGISTER_MOD);
     return 0;
 }