Palacios Public Git Repository

To checkout Palacios execute

  git clone http://v3vee.org/palacios/palacios.web/palacios.git
This will give you the master branch. You probably want the devel branch or one of the release branches. To switch to the devel branch, simply execute
  cd palacios
  git checkout --track -b devel origin/devel
The other branches are similar.


Extensive, Pedantic Error Checking in Linux module, especially for memory
[palacios.git] / linux_module / main.c
index b24df34..e37f207 100644 (file)
@@ -19,6 +19,8 @@
 #include <linux/spinlock.h>
 #include <linux/kthread.h>
 
+#include <linux/proc_fs.h>
+
 #include "palacios.h"
 #include "mm.h"
 #include "vm.h"
@@ -42,6 +44,7 @@ int mod_frees = 0;
 static int v3_major_num = 0;
 
 static struct v3_guest * guest_map[MAX_VMS] = {[0 ... MAX_VMS - 1] = 0};
+static struct proc_dir_entry *dir = 0;
 
 struct class * v3_class = NULL;
 static struct cdev ctrl_dev;
@@ -64,50 +67,54 @@ static int register_vm(struct v3_guest * guest) {
 static long v3_dev_ioctl(struct file * filp,
                         unsigned int ioctl, unsigned long arg) {
     void __user * argp = (void __user *)arg;
-    printk("V3 IOCTL %d\n", ioctl);
+    DEBUG("V3 IOCTL %d\n", ioctl);
 
 
     switch (ioctl) {
        case V3_CREATE_GUEST:{
            int vm_minor = 0;
            struct v3_guest_img user_image;
-           struct v3_guest * guest = kmalloc(sizeof(struct v3_guest), GFP_KERNEL);
+           struct v3_guest * guest = palacios_alloc(sizeof(struct v3_guest));
 
            if (IS_ERR(guest)) {
-               printk("Palacios: Error allocating Kernel guest_image\n");
+               ERROR("Palacios: Error allocating Kernel guest_image\n");
                return -EFAULT;
            }
 
            memset(guest, 0, sizeof(struct v3_guest));
 
-           printk("Palacios: Creating V3 Guest...\n");
+           INFO("Palacios: Creating V3 Guest...\n");
 
            vm_minor = register_vm(guest);
 
            if (vm_minor == -1) {
-               printk("Palacios Error: Too many VMs are currently running\n");
+               ERROR("Palacios Error: Too many VMs are currently running\n");
+               palacios_free(guest);
                return -EFAULT;
            }
 
            guest->vm_dev = MKDEV(v3_major_num, vm_minor);
 
            if (copy_from_user(&user_image, argp, sizeof(struct v3_guest_img))) {
-               printk("Palacios Error: copy from user error getting guest image...\n");
+               ERROR("Palacios Error: copy from user error getting guest image...\n");
+               palacios_free(guest);
                return -EFAULT;
            }
 
            guest->img_size = user_image.size;
 
-           printk("Palacios: Allocating kernel memory for guest image (%llu bytes)\n", user_image.size);
+           DEBUG("Palacios: Allocating kernel memory for guest image (%llu bytes)\n", user_image.size);
            guest->img = vmalloc(guest->img_size);
 
            if (IS_ERR(guest->img)) {
-               printk("Palacios Error: Could not allocate space for guest image\n");
+               ERROR("Palacios Error: Could not allocate space for guest image\n");
+               palacios_free(guest);
                return -EFAULT;
            }
 
            if (copy_from_user(guest->img, user_image.guest_data, guest->img_size)) {
-               printk("Palacios: Error loading guest data\n");
+               ERROR("Palacios: Error loading guest data\n");
+               palacios_free(guest);
                return -EFAULT;
            }      
 
@@ -116,7 +123,9 @@ static long v3_dev_ioctl(struct file * filp,
            INIT_LIST_HEAD(&(guest->exts));
 
            if (create_palacios_vm(guest) == -1) {
-               printk("Palacios: Error creating guest\n");
+               ERROR("Palacios: Error creating guest\n");
+               palacios_free(guest->img);
+               palacios_free(guest);
                return -EFAULT;
            }
 
@@ -127,7 +136,7 @@ static long v3_dev_ioctl(struct file * filp,
            unsigned long vm_idx = arg;
            struct v3_guest * guest = guest_map[vm_idx];
 
-           printk("Freeing VM (%s) (%p)\n", guest->name, guest);
+           INFO("Freeing VM (%s) (%p)\n", guest->name, guest);
 
            free_palacios_vm(guest);
            guest_map[vm_idx] = NULL;
@@ -139,14 +148,14 @@ static long v3_dev_ioctl(struct file * filp,
            memset(&mem, 0, sizeof(struct v3_mem_region));
            
            if (copy_from_user(&mem, argp, sizeof(struct v3_mem_region))) {
-               printk("copy from user error getting mem_region...\n");
+               ERROR("copy from user error getting mem_region...\n");
                return -EFAULT;
            }
 
-           printk("Adding %llu pages to Palacios memory\n", mem.num_pages);
+           DEBUG("Adding %llu pages to Palacios memory\n", mem.num_pages);
 
            if (add_palacios_memory(mem.base_addr, mem.num_pages) == -1) {
-               printk("Error adding memory to Palacios\n");
+               ERROR("Error adding memory to Palacios\n");
                return -EFAULT;
            }
 
@@ -154,7 +163,7 @@ static long v3_dev_ioctl(struct file * filp,
        }
 
        default: 
-           printk("\tUnhandled\n");
+           ERROR("\tUnhandled\n");
            return -EINVAL;
     }
 
@@ -171,6 +180,42 @@ static struct file_operations v3_ctrl_fops = {
 
 
 
+struct proc_dir_entry *palacios_get_procdir(void) 
+{
+    return dir;
+}
+
+static int read_guests(char * buf, char ** start, off_t off, int count,
+                      int * eof, void * data)
+{
+    int len = 0;
+    unsigned int i = 0;
+    
+    for(i = 0; i < MAX_VMS; i++) {
+       if (guest_map[i] != NULL) {
+           if (len<count) { 
+               len += snprintf(buf+len, count-len,
+                               "%s\t/dev/v3-vm%d\n", 
+                               guest_map[i]->name, i);
+           }
+       }
+    }
+    
+    return len;
+}
+
+static int show_mem(char * buf, char ** start, off_t off, int count,
+                   int * eof, void * data)
+{
+    int len = 0;
+    
+    len = snprintf(buf,count, "%p\n", (void *)get_palacios_base_addr());
+    len += snprintf(buf+len,count-len, "%lld\n", get_palacios_num_pages());
+    
+    return len;
+}
+
+
 static int __init v3_init(void) {
     dev_t dev = MKDEV(0, 0); // We dynamicallly assign the major number
     int ret = 0;
@@ -188,16 +233,16 @@ static int __init v3_init(void) {
 
     v3_class = class_create(THIS_MODULE, "vms");
     if (IS_ERR(v3_class)) {
-       printk("Failed to register V3 VM device class\n");
+       ERROR("Failed to register V3 VM device class\n");
        return PTR_ERR(v3_class);
     }
 
-    printk("intializing V3 Control device\n");
+    INFO("intializing V3 Control device\n");
 
     ret = alloc_chrdev_region(&dev, 0, MAX_VMS + 1, "v3vee");
 
     if (ret < 0) {
-       printk("Error registering device region for V3 devices\n");
+       ERROR("Error registering device region for V3 devices\n");
        goto failure2;
     }
 
@@ -206,7 +251,7 @@ static int __init v3_init(void) {
     dev = MKDEV(v3_major_num, MAX_VMS + 1);
 
     
-    printk("Creating V3 Control device: Major %d, Minor %d\n", v3_major_num, MINOR(dev));
+    DEBUG("Creating V3 Control device: Major %d, Minor %d\n", v3_major_num, MINOR(dev));
     cdev_init(&ctrl_dev, &v3_ctrl_fops);
     ctrl_dev.owner = THIS_MODULE;
     ctrl_dev.ops = &v3_ctrl_fops;
@@ -215,12 +260,36 @@ static int __init v3_init(void) {
     device_create(v3_class, NULL, dev, NULL, "v3vee");
 
     if (ret != 0) {
-       printk("Error adding v3 control device\n");
+       ERROR("Error adding v3 control device\n");
        goto failure1;
     }
 
-
-
+    dir = proc_mkdir("v3vee", NULL);
+    if(dir) {
+       struct proc_dir_entry *entry;
+
+       entry = create_proc_read_entry("v3-guests", 0444, dir, 
+                                      read_guests, NULL);
+        if (entry) {
+           INFO("/proc/v3vee/v3-guests successfully created\n");
+       } else {
+           ERROR("Could not create proc entry\n");
+           goto failure1;
+       }
+       
+       entry = create_proc_read_entry("v3-mem", 0444, dir,
+                                      show_mem, NULL);
+       if (entry) {
+           INFO("/proc/v3vee/v3-mem successfully added\n");
+       } else {
+           ERROR("Could not create proc entry\n");
+           goto failure1;
+       }
+    } else {
+       ERROR("Could not create proc entry\n");
+       goto failure1;
+    }
+       
     return 0;
 
  failure1:
@@ -245,13 +314,13 @@ static void __exit v3_exit(void) {
 
     dev_t dev = MKDEV(v3_major_num, MAX_VMS + 1);
 
-    printk("Removing V3 Control device\n");
+    INFO("Removing V3 Control device\n");
 
 
     palacios_vmm_exit();
 
-    printk("Palacios Mallocs = %d, Frees = %d\n", mallocs, frees);
-    printk("Palacios Page Allocs = %d, Page Frees = %d\n", pg_allocs, pg_frees);
+    DEBUG("Palacios Mallocs = %d, Frees = %d\n", mallocs, frees);
+    DEBUG("Palacios Page Allocs = %d, Page Frees = %d\n", pg_allocs, pg_frees);
 
     unregister_chrdev_region(MKDEV(v3_major_num, 0), MAX_VMS + 1);
 
@@ -265,7 +334,11 @@ static void __exit v3_exit(void) {
 
     palacios_deinit_mm();
 
-    printk("Palacios Module Mallocs = %d, Frees = %d\n", mod_allocs, mod_frees);
+    remove_proc_entry("v3-guests", dir);
+    remove_proc_entry("v3-mem", dir);
+    remove_proc_entry("v3vee", NULL);
+
+    DEBUG("Palacios Module Mallocs = %d, Frees = %d\n", mod_allocs, mod_frees);
 }