VMM: init and cleanup take the proc *
authorBarret Rhoden <brho@cs.berkeley.edu>
Thu, 26 Mar 2015 20:22:46 +0000 (16:22 -0400)
committerBarret Rhoden <brho@cs.berkeley.edu>
Tue, 7 Apr 2015 19:06:59 +0000 (15:06 -0400)
And do some minor connections between the guest pcore and its proc.

Might need to think a bit about the weak vs strong refs.  I'm pretty
sure the cur_proc ref will always be active whenever we have a
guest_pcore (vmx_vcpu).  We'll see.

kern/arch/x86/vmm/intel/vmx.c
kern/arch/x86/vmm/intel/vmx.h
kern/arch/x86/vmm/vmm.c
kern/arch/x86/vmm/vmm.h
kern/src/process.c
kern/src/syscall.c
tests/vmmcp.c

index 6a1fb7c..7af1198 100644 (file)
@@ -1207,7 +1207,7 @@ static void vmx_free_vpid(struct vmx_vcpu *vmx)
  *
  * Returns: A new VCPU structure
  */
-struct vmx_vcpu *vmx_create_vcpu(void)
+struct vmx_vcpu *vmx_create_vcpu(struct proc *p)
 {
        struct vmx_vcpu *vcpu = kmalloc(sizeof(struct vmx_vcpu), KMALLOC_WAIT);
        if (!vcpu) {
@@ -1216,6 +1216,7 @@ struct vmx_vcpu *vmx_create_vcpu(void)
 
        memset(vcpu, 0, sizeof(*vcpu));
 
+       vcpu->proc = p; /* uncounted (weak) reference */
        vcpu->vmcs = vmx_alloc_vmcs();
        printd("%d: vcpu->vmcs is %p\n", core_id(), vcpu->vmcs);
        if (!vcpu->vmcs)
index 26d5800..2b233bd 100644 (file)
@@ -625,7 +625,7 @@ struct vmx_vcpu {
 
        int shutdown;
        int ret_code;
-       struct dune_guest *guest;
+       struct proc *proc;
 
        struct msr_autoload {
                unsigned nr;
index 4af9853..d19593a 100644 (file)
@@ -71,8 +71,9 @@ int vm_run(uint64_t rip, uint64_t rsp, uint64_t cr3)
 
 /* Initializes a process to run virtual machine contexts, returning the number
  * initialized, optionally setting errno */
-int vmm_struct_init(struct vmm *vmm, unsigned int nr_guest_pcores)
+int vmm_struct_init(struct proc *p, unsigned int nr_guest_pcores)
 {
+       struct vmm *vmm = &p->vmm;
        unsigned int i;
        qlock(&vmm->qlock);
        if (vmm->vmmcp) {
@@ -86,7 +87,7 @@ int vmm_struct_init(struct vmm *vmm, unsigned int nr_guest_pcores)
        vmm->amd = 0;
        vmm->guest_pcores = kzmalloc(sizeof(void*) * nr_guest_pcores, KMALLOC_WAIT);
        for (i = 0; i < nr_guest_pcores; i++) {
-               vmm->guest_pcores[i] = vmx_create_vcpu();
+               vmm->guest_pcores[i] = vmx_create_vcpu(p);
                /* If we failed, we'll clean it up when the process dies */
                if (!vmm->guest_pcores[i]) {
                        set_errno(ENOMEM);
@@ -101,8 +102,9 @@ int vmm_struct_init(struct vmm *vmm, unsigned int nr_guest_pcores)
 /* Has no concurrency protection - only call this when you know you have the
  * only ref to vmm.  For instance, from __proc_free, where there is only one ref
  * to the proc (and thus proc.vmm). */
-void __vmm_struct_cleanup(struct vmm *vmm)
+void __vmm_struct_cleanup(struct proc *p)
 {
+       struct vmm *vmm = &p->vmm;
        if (!vmm->vmmcp)
                return;
        for (int i = 0; i < vmm->nr_guest_pcores; i++) {
index 5945df5..7e604bc 100644 (file)
@@ -40,14 +40,14 @@ struct vmm {
 void vmm_init(void);
 void vmm_pcpu_init(void);
 
-int vmm_struct_init(struct vmm *vmm, unsigned int nr_guest_pcores);
-void __vmm_struct_cleanup(struct vmm *vmm);
+int vmm_struct_init(struct proc *p, unsigned int nr_guest_pcores);
+void __vmm_struct_cleanup(struct proc *p);
 
 int vm_run(uint64_t,uint64_t, uint64_t);
 int intel_vmx_start(int id);
 int intel_vmx_setup(int nvmcs);
 
-struct vmx_vcpu *vmx_create_vcpu(void);
+struct vmx_vcpu *vmx_create_vcpu(struct proc *p);
 void vmx_destroy_vcpu(struct vmx_vcpu *vcpu);
 
 #endif /* _VMM_H_ */
index 3b05c08..cac3977 100644 (file)
@@ -458,7 +458,7 @@ static void __proc_free(struct kref *kref)
        assert(kref_refcnt(&p->p_kref) == 0);
        assert(TAILQ_EMPTY(&p->alarmset.list));
 
-       __vmm_struct_cleanup(&p->vmm);
+       __vmm_struct_cleanup(p);
        p->progname[0] = 0;
        cclose(p->dot);
        cclose(p->slash);
index 8e91d58..7d2e0d5 100644 (file)
@@ -1151,7 +1151,7 @@ static int sys_change_to_m(struct proc *p)
  * initialized, optionally setting errno */
 static int sys_setup_vmm(struct proc *p, unsigned int nr_guest_pcores)
 {
-       return vmm_struct_init(&p->vmm, nr_guest_pcores);
+       return vmm_struct_init(p, nr_guest_pcores);
 }
 
 /* Pokes the ksched for the given resource for target_pid.  If the target pid
index d083baa..886c06a 100644 (file)
@@ -53,6 +53,7 @@ int nr_threads = 2;
 
 int main(int argc, char **argv)
 {
+       int nr_gpcs = 1;
        int fd = open("#c/sysctl", O_RDWR), ret;
        void * x;
        static char cmd[512];
@@ -60,6 +61,10 @@ int main(int argc, char **argv)
                perror("#c/sysctl");
                exit(1);
        }
+       if (ros_syscall(SYS_setup_vmm, nr_gpcs, 0, 0, 0, 0, 0) != nr_gpcs) {
+               perror("Guest pcore setup failed");
+               exit(1);
+       }
 
        mcp = 1; //argc - 1;
        if (mcp) {