added inspection framework

[palacios.git] / linux_module / palacios-dev.c

diff --git a/linux_module/palacios-dev.c b/linux_module/palacios-dev.c
index 8b30352..1dfafc2 100644 (file)
--- a/linux_module/palacios-dev.c
+++ b/linux_module/palacios-dev.c
@@ -24,9 +24,15 @@
 #include "palacios-stream.h"
 #include "palacios-file.h"
 #include "palacios-serial.h"
+#include "palacios-socket.h"
+#include "palacios-vnet.h"
+#include "palacios-packet.h"
 
 MODULE_LICENSE("GPL");
 
+int mod_allocs = 0;
+int mod_frees = 0;
+
 
 static int v3_major_num = 0;
 
@@ -76,9 +82,9 @@ static long v3_dev_ioctl(struct file * filp,
 
     switch (ioctl) {
        case V3_START_GUEST:{
+           int vm_minor = 0;
            struct v3_guest_img user_image;
            struct v3_guest * guest = kmalloc(sizeof(struct v3_guest), GFP_KERNEL);
-           int vm_minor = 0;
 
            if (IS_ERR(guest)) {
                printk("Error allocating Kernel guest_image\n");
@@ -162,6 +168,37 @@ static long v3_dev_ioctl(struct file * filp,
 
            break;
        }
+
+       case V3_START_NETWORK: {
+            struct v3_network net;
+            memset(&net, 0, sizeof(struct v3_network));
+   
+            if(copy_from_user(&net, argp, sizeof(struct v3_network))){
+                printk("copy from user error getting network service requests ... \n");
+                return -EFAULT;
+            }
+ 
+        #ifdef CONFIG_PALACIOS_SOCKET
+            if(net.socket == 1){
+                palacios_socket_init();
+               printk("Started Palacios Socket\n");
+            }
+        #endif
+        #ifdef CONFIG_PALACIOS_PACKET
+            if(net.packet == 1){
+                palacios_init_packet(NULL);
+               printk("Started Palacios Direct Network Bridge\n");
+            }
+        #endif
+        #ifdef CONFIG_PALACIOS_VNET
+            if(net.vnet == 1){
+                palacios_init_vnet();
+               printk("Started Palacios VNET Service\n");
+            }
+        #endif
+ 
+            break;
+        }
        default: 
            printk("\tUnhandled\n");
            return -EINVAL;
@@ -251,6 +288,11 @@ static void __exit v3_exit(void) {
     extern u32 mallocs;
     extern u32 frees;
 
+
+    // should probably try to stop any guests
+
+
+
     dev_t dev = MKDEV(v3_major_num, MAX_VMS + 1);
 
     printk("Removing V3 Control device\n");
@@ -267,9 +309,34 @@ static void __exit v3_exit(void) {
 
     device_destroy(v3_class, dev);
     class_destroy(v3_class);
+
+
+    palacios_file_deinit();
+    palacios_deinit_stream();
+
+    palacios_deinit_mm();
+
+    printk("Palacios Module Mallocs = %d, Frees = %d\n", mod_allocs, mod_frees);
 }
 
 
 
 module_init(v3_init);
 module_exit(v3_exit);
+
+
+
+void * trace_malloc(size_t size, gfp_t flags) {
+    void * addr = NULL;
+
+    mod_allocs++;
+    addr = kmalloc(size, flags);
+
+    return addr;
+}
+
+
+void trace_free(const void * objp) {
+    mod_frees++;
+    kfree(objp);
+}