X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_module%2Fmcheck.c;h=8b3c5b244b60ecabf3e2ac970f7d9470cf849c76;hb=d240aa16811174593dc6e6ca9c796db05948c353;hp=9cdc04b78f5d6077b94779f50697248424267389;hpb=acaadd79c597c8d5180fbfbec79c01fef3dff003;p=palacios.git

diff --git a/linux_module/mcheck.c b/linux_module/mcheck.c
index 9cdc04b..8b3c5b2 100644
--- a/linux_module/mcheck.c
+++ b/linux_module/mcheck.c
@@ -47,6 +47,13 @@ static int guest_init(struct v3_guest * guest, void ** vm_data) {
 
 static int guest_deinit(struct v3_guest * guest, void * vm_data) {
     
+    remove_guest_ctrl(guest, V3_VM_INJECT_SCRUBBER_MCE);
+    return 0;
+}
+
+static int all_deinit(void)
+{
+    // nothing to do
     return 0;
 }
 
@@ -54,7 +61,7 @@ static int guest_deinit(struct v3_guest * guest, void * vm_data) {
 struct linux_ext mcheck_ext = {
     .name = "MACHINE CHECK",
     .init = NULL,
-    .deinit = NULL,
+    .deinit = all_deinit,
     .guest_init = guest_init, 
     .guest_deinit = guest_deinit
 };