fix potential overflow condition in VMX assist

[palacios.git] / palacios / src / extensions / ext_mtrr.c

diff --git a/palacios/src/extensions/ext_mtrr.c b/palacios/src/extensions/ext_mtrr.c
index 9232bb7..94cb6b9 100644 (file)
--- a/palacios/src/extensions/ext_mtrr.c
+++ b/palacios/src/extensions/ext_mtrr.c
@@ -621,8 +621,9 @@ static int init_mtrrs(struct v3_vm_info * vm, v3_cfg_tree_t * cfg, void ** priv_
 
 static struct v3_extension_impl mtrr_impl = {
     .name = "MTRRS",
-    .init = init_mtrrs,
-    .deinit = deinit_mtrrs,
+    .init = NULL,
+    .vm_init = init_mtrrs,
+    .vm_deinit = deinit_mtrrs,
     .core_init = NULL,
     .core_deinit = NULL,
     .on_entry = NULL,