palacios/src/palacios/vmm_scheduler.c

   1 /*
   2  * This file is part of the Palacios Virtual Machine Monitor developed
   3  * by the V3VEE Project with funding from the United States National
   4  * Science Foundation and the Department of Energy.
   5  *
   6  * The V3VEE Project is a joint project between Northwestern University
   7  * and the University of New Mexico.  You can find out more at
   8  * http://www.v3vee.org
   9  *
  10  * Copyright (c) 2013, Oscar Mondragon <omondrag@cs.unm.edu>
  11  * Copyright (c) 2013, Patrick G. Bridges <bridges@cs.unm.edu>
  12  * Copyright (c) 2013, The V3VEE Project <http://www.v3vee.org>
  13  * All rights reserved.
  14  *
  15  * Author: Oscar Mondragon <omondrag@cs.unm.edu>
  16  *         Patrick G. Bridges <bridges@cs.unm.edu>
  17  *
  18  * This is free software.  You are permitted to use,
  19  * redistribute, and modify it as specified in the file "V3VEE_LICENSE".
  20  */
  21
  22 #include <palacios/vmm.h>
  23 #include <palacios/vm_guest.h>
  24 #include <palacios/vmm_scheduler.h>
  25 #include <palacios/vmm_hashtable.h>
  26
  27 #ifndef V3_CONFIG_DEBUG_SCHEDULER
  28 #undef PrintDebug
  29 #define PrintDebug(fmt, args...)
  30 #endif
  31
  32 static char default_strategy[] = "host";
  33 static struct hashtable * master_scheduler_table = NULL;
  34 static int create_host_scheduler();
  35 static int destroy_host_scheduler();
  36
  37 static struct vm_scheduler_impl *scheduler = NULL;
  38
  39 static uint_t scheduler_hash_fn(addr_t key) {
  40     char * name = (char *)key;
  41     return v3_hash_buffer((uint8_t *)name, strlen(name));
  42 }
  43
  44 static int scheduler_eq_fn(addr_t key1, addr_t key2) {
  45     char * name1 = (char *)key1;
  46     char * name2 = (char *)key2;
  47
  48     return (strcmp(name1, name2) == 0);
  49 }
  50
  51 int V3_init_scheduling() {
  52
  53      PrintDebug(VM_NONE, VCORE_NONE,"Initializing scheduler");
  54
  55     master_scheduler_table = v3_create_htable(0, scheduler_hash_fn, scheduler_eq_fn);
  56     return create_host_scheduler();
  57 }
  58
  59 int V3_deinit_scheduling()
  60 {
  61     destroy_host_scheduler();
  62     v3_free_htable(master_scheduler_table,1,1);
  63     return 0;
  64 }
  65
  66
  67 int v3_register_scheduler(struct vm_scheduler_impl *s) {
  68
  69     PrintDebug(VM_NONE, VCORE_NONE,"Registering Scheduler (%s)\n", s->name);
  70
  71     if (v3_htable_search(master_scheduler_table, (addr_t)(s->name))) {
  72         PrintError(VM_NONE, VCORE_NONE, "Multiple instances of scheduler (%s)\n", s->name);
  73         return -1;
  74     }
  75
  76     if (v3_htable_insert(master_scheduler_table,
  77                          (addr_t)(s->name),
  78                          (addr_t)(s)) == 0) {
  79         PrintError(VM_NONE, VCORE_NONE, "Could not register scheduler (%s)\n", s->name);
  80         return -1;
  81     }
  82
  83     return 0;
  84 }
  85
  86
  87 struct vm_scheduler_impl *v3_unregister_scheduler(char *name) {
  88
  89     PrintDebug(VM_NONE, VCORE_NONE,"Unregistering Scheduler (%s)\n",name);
  90
  91     struct vm_scheduler_impl *f = (struct vm_scheduler_impl *) v3_htable_remove(master_scheduler_table,(addr_t)(name),0);
  92
  93     if (!f) {
  94         PrintError(VM_NONE,VCORE_NONE,"Could not find Scheduler (%s)\n",name);
  95         return NULL;
  96     } else {
  97         return f;
  98     }
  99 }
 100
 101
 102
 103
 104 struct vm_scheduler_impl *v3_scheduler_lookup(char *name)
 105 {
 106     return (struct vm_scheduler_impl *)v3_htable_search(master_scheduler_table, (addr_t)(name));
 107 }
 108
 109 int V3_enable_scheduler() {
 110     char *sched_name;
 111
 112     scheduler = NULL;
 113     sched_name = v3_lookup_option("scheduler");
 114
 115     if (sched_name) {
 116         scheduler = v3_scheduler_lookup(sched_name);
 117     }
 118
 119     if (!scheduler) {
 120         scheduler = v3_scheduler_lookup(default_strategy);
 121     }
 122
 123     if (!scheduler) {
 124         PrintError(VM_NONE, VCORE_NONE,"Specified Palacios scheduler \"%s\" not found.\n", default_strategy);
 125         return -1;
 126     }
 127
 128     PrintDebug(VM_NONE, VCORE_NONE,"Scheduler %s found",scheduler->name);
 129
 130     if (scheduler->init) {
 131         return scheduler->init();
 132     } else {
 133         return 0;
 134     }
 135 }
 136
 137 int V3_disable_scheduler()
 138 {
 139     if (scheduler->deinit) {
 140         return scheduler->deinit();
 141     } else {
 142         return 0;
 143     }
 144 }
 145
 146 int v3_scheduler_register_vm(struct v3_vm_info *vm) {
 147     if (scheduler->vm_init) {
 148         return scheduler->vm_init(vm);
 149     } else {
 150         return 0;
 151     }
 152 }
 153 int v3_scheduler_register_core(struct guest_info *core) {
 154     if (scheduler->core_init) {
 155         return scheduler->core_init(core);
 156     } else {
 157         return 0;
 158     }
 159 }
 160 int v3_scheduler_admit_vm(struct v3_vm_info *vm) {
 161     if (scheduler->admit) {
 162         return scheduler->admit(vm);
 163     } else {
 164         return 0;
 165     }
 166 }
 167 int v3_scheduler_notify_remap(struct v3_vm_info *vm) {
 168     if (scheduler->remap) {
 169         return scheduler->remap(vm);
 170     } else {
 171         return 0;
 172     }
 173 }
 174 int v3_scheduler_notify_dvfs(struct v3_vm_info *vm) {
 175     if (scheduler->dvfs) {
 176         return scheduler->dvfs(vm);
 177     } else {
 178         return 0;
 179     }
 180 }
 181 void v3_schedule(struct guest_info *core) {
 182     if (scheduler->schedule) {
 183         scheduler->schedule(core);
 184     }
 185     return;
 186 }
 187 void v3_yield(struct guest_info *core, int usec) {
 188     if (scheduler->yield) {
 189         scheduler->yield(core, usec);
 190     }
 191     return;
 192 }
 193
 194 int host_sched_vm_init(struct v3_vm_info *vm)
 195 {
 196
 197     PrintDebug(vm, VCORE_NONE,"Sched. host_sched_init\n");
 198
 199     char * schedule_hz_str = v3_cfg_val(vm->cfg_data->cfg, "schedule_hz");
 200     uint32_t sched_hz = 100;
 201
 202
 203     if (schedule_hz_str) {
 204         sched_hz = atoi(schedule_hz_str);
 205     }
 206
 207     PrintDebug(vm, VCORE_NONE,"CPU_KHZ = %d, schedule_freq=%p\n", V3_CPU_KHZ(),
 208                (void *)(addr_t)sched_hz);
 209
 210     uint64_t yield_cycle_period = (V3_CPU_KHZ() * 1000) / sched_hz;
 211     vm->sched_priv_data = (void *)yield_cycle_period;
 212
 213     return 0;
 214 }
 215
 216 int host_sched_core_init(struct guest_info *core)
 217 {
 218     PrintDebug(core->vm_info, core,"Sched. host_sched_core_init\n");
 219
 220     uint64_t t = v3_get_host_time(&core->time_state);
 221     core->sched_priv_data = (void *)t;
 222
 223     return 0;
 224 }
 225
 226 void host_sched_schedule(struct guest_info *core)
 227 {
 228     uint64_t cur_cycle;
 229     cur_cycle = v3_get_host_time(&core->time_state);
 230
 231     if (cur_cycle > ( (uint64_t)core->sched_priv_data + (uint64_t)core->vm_info->sched_priv_data)) {
 232
 233         V3_Yield();
 234
 235         core->sched_priv_data = (void*)v3_get_host_time(&(core->time_state));
 236
 237     }
 238 }
 239
 240 /*
 241  * unconditional cpu yield
 242  * if the yielding thread is a guest context, the guest quantum is reset on resumption
 243  * Non guest context threads should call this function with a NULL argument
 244  *
 245  * usec <0  => the non-timed yield is used
 246  * usec >=0 => the timed yield is used, which also usually implies interruptible
 247  */
 248 void host_sched_yield(struct guest_info * core, int usec) {
 249
 250     if (usec < 0) {
 251         V3_Yield();
 252     } else {
 253         V3_Sleep(usec);
 254     }
 255     if (core){
 256         core->sched_priv_data = (void*)v3_get_host_time(&(core->time_state));
 257     }
 258 }
 259
 260
 261 int host_sched_admit(struct v3_vm_info *vm){
 262     return 0;
 263 }
 264
 265 static struct vm_scheduler_impl host_sched_impl = {
 266     .name = "host",
 267     .init = NULL,
 268     .deinit = NULL,
 269     .vm_init = host_sched_vm_init,
 270     .vm_deinit = NULL,
 271     .core_init = host_sched_core_init,
 272     .core_deinit = NULL,
 273     .schedule = host_sched_schedule,
 274     .yield = host_sched_yield,
 275     .admit = host_sched_admit,
 276     .remap = NULL,
 277     .dvfs=NULL
 278 };
 279
 280 static int create_host_scheduler()
 281 {
 282         v3_register_scheduler(&host_sched_impl);
 283         return 0;
 284 }
 285
 286 static int destroy_host_scheduler()
 287 {
 288         v3_unregister_scheduler(host_sched_impl.name);
 289         return 0;
 290 }