1 #include <palacios/vmm_mem.h>
2 #include <palacios/vmm.h>
3 #include <palacios/vmm_util.h>
4 #include <palacios/vmm_decoder.h>
8 void init_shadow_region(struct shadow_region * entry,
9 addr_t guest_addr_start,
10 addr_t guest_addr_end,
11 guest_region_type_t guest_region_type,
12 host_region_type_t host_region_type)
14 entry->guest_type = guest_region_type;
15 entry->guest_start = guest_addr_start;
16 entry->guest_end = guest_addr_end;
17 entry->host_type = host_region_type;
19 entry->next=entry->prev = NULL;
22 int add_shadow_region_passthrough( struct guest_info * guest_info,
23 addr_t guest_addr_start,
24 addr_t guest_addr_end,
27 struct shadow_region * entry = (struct shadow_region *)V3_Malloc(sizeof(struct shadow_region));
29 init_shadow_region(entry, guest_addr_start, guest_addr_end,
30 GUEST_REGION_PHYSICAL_MEMORY, HOST_REGION_PHYSICAL_MEMORY);
31 entry->host_addr = host_addr;
33 return add_shadow_region(&(guest_info->mem_map), entry);
36 int hook_guest_mem(struct guest_info * info, addr_t guest_addr_start, addr_t guest_addr_end,
37 int (*read)(addr_t guest_addr, void * dst, uint_t length, void * priv_data),
38 int (*write)(addr_t guest_addr, void * src, uint_t length, void * priv_data),
41 struct shadow_region * entry = (struct shadow_region *)V3_Malloc(sizeof(struct shadow_region));
42 struct vmm_mem_hook * hook = (struct vmm_mem_hook *)V3_Malloc(sizeof(struct vmm_mem_hook));
44 memset(hook, 0, sizeof(struct vmm_mem_hook));
49 hook->priv_data = priv_data;
52 init_shadow_region(entry, guest_addr_start, guest_addr_end,
53 GUEST_REGION_PHYSICAL_MEMORY, HOST_REGION_HOOK);
55 entry->host_addr = (addr_t)hook;
57 return add_shadow_region(&(info->mem_map), entry);
61 struct vmm_mem_hook * get_mem_hook(struct guest_info * info, addr_t guest_addr) {
62 struct shadow_region * region = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
65 PrintDebug("Could not find shadow region for addr: %x\n", guest_addr);
69 return (struct vmm_mem_hook *)(region->host_addr);
73 /* mem_addr is the guest physical memory address */
74 static int mem_hook_dispatch(struct guest_info * info, addr_t fault_addr, addr_t guest_phys_page, pf_error_t access_info, struct vmm_mem_hook * hook) {
76 // emulate and then dispatch
77 // or dispatch and emulate
80 if (access_info.write == 1) {
84 PrintDebug("Memory hook write\n");
87 if (hook->write(fault_addr, src, length, hook->priv_data) != length) {
91 PrintDebug("Memory hook read\n");
99 int handle_special_page_fault(struct guest_info * info, addr_t fault_addr, addr_t guest_phys_page, pf_error_t access_info) {
100 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_phys_page);
102 switch (reg->host_type) {
103 case HOST_REGION_HOOK:
104 return mem_hook_dispatch(info, fault_addr, guest_phys_page, access_info, (struct vmm_mem_hook *)(reg->host_addr));
115 void init_shadow_map(struct guest_info * info) {
116 struct shadow_map * map = &(info->mem_map);
118 map->num_regions = 0;
124 void free_shadow_map(struct shadow_map * map) {
125 struct shadow_region * cursor = map->head;
126 struct shadow_region * tmp = NULL;
130 cursor = cursor->next;
140 int add_shadow_region(struct shadow_map * map,
141 struct shadow_region * region)
143 struct shadow_region * cursor = map->head;
145 PrintDebug("Adding Shadow Region: (0x%x-0x%x)\n", region->guest_start, region->guest_end);
147 if ((!cursor) || (cursor->guest_start >= region->guest_end)) {
149 region->next = cursor;
156 // Check if it overlaps with the current cursor
157 if ((cursor->guest_end > region->guest_start) && (cursor->guest_start < region->guest_start)) {
158 // overlaps not allowed
162 if (!(cursor->next)) {
163 // add to the end of the list
164 cursor->next = region;
165 region->prev = cursor;
169 } else if (cursor->next->guest_start >= region->guest_end) {
171 region->next = cursor->next;
172 region->prev = cursor;
174 cursor->next->prev = region;
175 cursor->next = region;
180 } else if (cursor->next->guest_end <= region->guest_start) {
181 cursor = cursor->next;
183 // This cannot happen!
184 // we should panic here
189 // This cannot happen
190 // We should panic here
195 int delete_shadow_region(struct shadow_map * map,
203 struct shadow_region *get_shadow_region_by_index(struct shadow_map * map,
205 struct shadow_region * reg = map->head;
219 struct shadow_region * get_shadow_region_by_addr(struct shadow_map * map,
221 struct shadow_region * reg = map->head;
224 if ((reg->guest_start <= addr) && (reg->guest_end > addr)) {
226 } else if (reg->guest_start > addr) {
236 host_region_type_t get_shadow_addr_type(struct guest_info * info, addr_t guest_addr) {
237 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
240 return HOST_REGION_INVALID;
242 return reg->host_type;
246 addr_t get_shadow_addr(struct guest_info * info, addr_t guest_addr) {
247 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
252 return (guest_addr - reg->guest_start) + reg->host_addr;
257 host_region_type_t lookup_shadow_map_addr(struct shadow_map * map, addr_t guest_addr, addr_t * host_addr) {
258 struct shadow_region * reg = get_shadow_region_by_addr(map, guest_addr);
262 return HOST_REGION_INVALID;
264 switch (reg->host_type) {
265 case HOST_REGION_PHYSICAL_MEMORY:
266 *host_addr = (guest_addr - reg->guest_start) + reg->host_addr;
267 return reg->host_type;
268 case HOST_REGION_MEMORY_MAPPED_DEVICE:
269 case HOST_REGION_UNALLOCATED:
273 return reg->host_type;
279 void print_shadow_map(struct shadow_map * map) {
280 struct shadow_region * cur = map->head;
283 PrintDebug("Memory Layout (regions: %d) \n", map->num_regions);
286 PrintDebug("%d: 0x%x - 0x%x (%s) -> ", i, cur->guest_start, cur->guest_end - 1,
287 cur->guest_type == GUEST_REGION_PHYSICAL_MEMORY ? "GUEST_REGION_PHYSICAL_MEMORY" :
288 cur->guest_type == GUEST_REGION_NOTHING ? "GUEST_REGION_NOTHING" :
289 cur->guest_type == GUEST_REGION_MEMORY_MAPPED_DEVICE ? "GUEST_REGION_MEMORY_MAPPED_DEVICE" :
291 if (cur->host_type == HOST_REGION_PHYSICAL_MEMORY ||
292 cur->host_type == HOST_REGION_UNALLOCATED ||
293 cur->host_type == HOST_REGION_MEMORY_MAPPED_DEVICE) {
294 PrintDebug("0x%x", cur->host_addr);
297 cur->host_type == HOST_REGION_PHYSICAL_MEMORY ? "HOST_REGION_PHYSICAL_MEMORY" :
298 cur->host_type == HOST_REGION_UNALLOCATED ? "HOST_REGION_UNALLOACTED" :
299 cur->host_type == HOST_REGION_HOOK ? "HOST_REGION_HOOK" :
300 cur->host_type == HOST_REGION_MEMORY_MAPPED_DEVICE ? "HOST_REGION_MEMORY_MAPPED_DEVICE" :
301 cur->host_type == HOST_REGION_REMOTE ? "HOST_REGION_REMOTE" :
302 cur->host_type == HOST_REGION_SWAPPED ? "HOST_REGION_SWAPPED" :
329 struct vmm_os_hooks * os_hooks;
331 void * TestMalloc(uint_t size) {
335 void * TestAllocatePages(int size) {
336 return malloc(4096 * size);
340 void TestPrint(const char * fmt, ...) {
348 int mem_list_add_test_1( vmm_mem_list_t * list) {
352 PrintDebug("\n\nTesting Memory List\n");
356 offset = PAGE_SIZE * 6;
357 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 10));
358 add_mem_list_pages(list, offset, 10);
359 print_mem_list(list);
363 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + PAGE_SIZE * 4);
364 add_mem_list_pages(list, offset, 4);
365 print_mem_list(list);
367 offset = PAGE_SIZE * 20;
368 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 1));
369 add_mem_list_pages(list, offset, 1);
370 print_mem_list(list);
372 offset = PAGE_SIZE * 21;
373 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 3));
374 add_mem_list_pages(list, offset, 3);
375 print_mem_list(list);
378 offset = PAGE_SIZE * 10;
379 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 30));
380 add_mem_list_pages(list, offset, 30);
381 print_mem_list(list);
384 offset = PAGE_SIZE * 5;
385 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 1));
386 add_mem_list_pages(list, offset, 1);
387 print_mem_list(list);
395 int mem_layout_add_test_1(vmm_mem_layout_t * layout) {
401 PrintDebug("\n\nTesting Memory Layout\n");
403 init_mem_layout(layout);
407 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
408 add_guest_mem_range(layout, start, end);
409 print_mem_layout(layout);
414 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
415 add_guest_mem_range(layout, start, end);
416 print_mem_layout(layout);
420 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
421 add_guest_mem_range(layout, start, end);
422 print_mem_layout(layout);
426 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
427 add_guest_mem_range(layout, start, end);
428 print_mem_layout(layout);
433 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
434 add_guest_mem_range(layout, start, end);
435 print_mem_layout(layout);
445 int main(int argc, char ** argv) {
446 struct vmm_os_hooks dummy_hooks;
447 os_hooks = &dummy_hooks;
449 vmm_mem_layout_t layout;
452 os_hooks->malloc = &TestMalloc;
453 os_hooks->free = &free;
454 os_hooks->print_debug = &TestPrint;
455 os_hooks->allocate_pages = &TestAllocatePages;
459 printf("mem_list_add_test_1: %d\n", mem_list_add_test_1(&list));
460 printf("layout_add_test_1: %d\n", mem_layout_add_test_1(&layout));