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 int mem_hook_dispatch(struct guest_info * info, addr_t mem_addr, pf_error_t access_info, struct vmm_mem_hook * hook) {
75 if (access_info.write == 1) {
78 PrintDebug("Memory hook write\n");
81 if (hook->write(mem_addr, src, length, hook->priv_data) != length) {
85 PrintDebug("Memory hook read\n");
93 int handle_special_page_fault(struct guest_info * info, addr_t mem_addr, pf_error_t access_info) {
94 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), mem_addr);
96 switch (reg->host_type) {
97 case HOST_REGION_HOOK:
98 return mem_hook_dispatch(info, mem_addr, access_info, (struct vmm_mem_hook *)(reg->host_addr));
109 void init_shadow_map(struct guest_info * info) {
110 struct shadow_map * map = &(info->mem_map);
112 map->num_regions = 0;
118 void free_shadow_map(struct shadow_map * map) {
119 struct shadow_region * cursor = map->head;
120 struct shadow_region * tmp = NULL;
124 cursor = cursor->next;
134 int add_shadow_region(struct shadow_map * map,
135 struct shadow_region * region)
137 struct shadow_region * cursor = map->head;
139 PrintDebug("Adding Shadow Region: (0x%x-0x%x)\n", region->guest_start, region->guest_end);
141 if ((!cursor) || (cursor->guest_start >= region->guest_end)) {
143 region->next = cursor;
150 // Check if it overlaps with the current cursor
151 if ((cursor->guest_end > region->guest_start) && (cursor->guest_start < region->guest_start)) {
152 // overlaps not allowed
156 if (!(cursor->next)) {
157 // add to the end of the list
158 cursor->next = region;
159 region->prev = cursor;
163 } else if (cursor->next->guest_start >= region->guest_end) {
165 region->next = cursor->next;
166 region->prev = cursor;
168 cursor->next->prev = region;
169 cursor->next = region;
174 } else if (cursor->next->guest_end <= region->guest_start) {
175 cursor = cursor->next;
177 PrintDebug("WTF?\n");
178 // This cannot happen!
179 // we should panic here
184 // This cannot happen
185 // We should panic here
190 int delete_shadow_region(struct shadow_map * map,
198 struct shadow_region *get_shadow_region_by_index(struct shadow_map * map,
200 struct shadow_region * reg = map->head;
214 struct shadow_region * get_shadow_region_by_addr(struct shadow_map * map,
216 struct shadow_region * reg = map->head;
219 if ((reg->guest_start <= addr) && (reg->guest_end > addr)) {
221 } else if (reg->guest_start > addr) {
231 host_region_type_t get_shadow_addr_type(struct guest_info * info, addr_t guest_addr) {
232 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
235 return HOST_REGION_INVALID;
237 return reg->host_type;
241 addr_t get_shadow_addr(struct guest_info * info, addr_t guest_addr) {
242 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
247 return (guest_addr - reg->guest_start) + reg->host_addr;
252 host_region_type_t lookup_shadow_map_addr(struct shadow_map * map, addr_t guest_addr, addr_t * host_addr) {
253 struct shadow_region * reg = get_shadow_region_by_addr(map, guest_addr);
257 return HOST_REGION_INVALID;
259 switch (reg->host_type) {
260 case HOST_REGION_PHYSICAL_MEMORY:
261 *host_addr = (guest_addr - reg->guest_start) + reg->host_addr;
262 return reg->host_type;
263 case HOST_REGION_MEMORY_MAPPED_DEVICE:
264 case HOST_REGION_UNALLOCATED:
268 return reg->host_type;
274 void print_shadow_map(struct shadow_map * map) {
275 struct shadow_region * cur = map->head;
278 PrintDebug("Memory Layout (regions: %d) \n", map->num_regions);
281 PrintDebug("%d: 0x%x - 0x%x (%s) -> ", i, cur->guest_start, cur->guest_end - 1,
282 cur->guest_type == GUEST_REGION_PHYSICAL_MEMORY ? "GUEST_REGION_PHYSICAL_MEMORY" :
283 cur->guest_type == GUEST_REGION_NOTHING ? "GUEST_REGION_NOTHING" :
284 cur->guest_type == GUEST_REGION_MEMORY_MAPPED_DEVICE ? "GUEST_REGION_MEMORY_MAPPED_DEVICE" :
286 if (cur->host_type == HOST_REGION_PHYSICAL_MEMORY ||
287 cur->host_type == HOST_REGION_UNALLOCATED ||
288 cur->host_type == HOST_REGION_MEMORY_MAPPED_DEVICE) {
289 PrintDebug("0x%x", cur->host_addr);
292 cur->host_type == HOST_REGION_PHYSICAL_MEMORY ? "HOST_REGION_PHYSICAL_MEMORY" :
293 cur->host_type == HOST_REGION_UNALLOCATED ? "HOST_REGION_UNALLOACTED" :
294 cur->host_type == HOST_REGION_HOOK ? "HOST_REGION_HOOK" :
295 cur->host_type == HOST_REGION_MEMORY_MAPPED_DEVICE ? "HOST_REGION_MEMORY_MAPPED_DEVICE" :
296 cur->host_type == HOST_REGION_REMOTE ? "HOST_REGION_REMOTE" :
297 cur->host_type == HOST_REGION_SWAPPED ? "HOST_REGION_SWAPPED" :
324 struct vmm_os_hooks * os_hooks;
326 void * TestMalloc(uint_t size) {
330 void * TestAllocatePages(int size) {
331 return malloc(4096 * size);
335 void TestPrint(const char * fmt, ...) {
343 int mem_list_add_test_1( vmm_mem_list_t * list) {
347 PrintDebug("\n\nTesting Memory List\n");
351 offset = PAGE_SIZE * 6;
352 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 10));
353 add_mem_list_pages(list, offset, 10);
354 print_mem_list(list);
358 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + PAGE_SIZE * 4);
359 add_mem_list_pages(list, offset, 4);
360 print_mem_list(list);
362 offset = PAGE_SIZE * 20;
363 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 1));
364 add_mem_list_pages(list, offset, 1);
365 print_mem_list(list);
367 offset = PAGE_SIZE * 21;
368 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 3));
369 add_mem_list_pages(list, offset, 3);
370 print_mem_list(list);
373 offset = PAGE_SIZE * 10;
374 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 30));
375 add_mem_list_pages(list, offset, 30);
376 print_mem_list(list);
379 offset = PAGE_SIZE * 5;
380 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 1));
381 add_mem_list_pages(list, offset, 1);
382 print_mem_list(list);
390 int mem_layout_add_test_1(vmm_mem_layout_t * layout) {
396 PrintDebug("\n\nTesting Memory Layout\n");
398 init_mem_layout(layout);
402 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
403 add_guest_mem_range(layout, start, end);
404 print_mem_layout(layout);
409 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
410 add_guest_mem_range(layout, start, end);
411 print_mem_layout(layout);
415 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
416 add_guest_mem_range(layout, start, end);
417 print_mem_layout(layout);
421 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
422 add_guest_mem_range(layout, start, end);
423 print_mem_layout(layout);
428 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
429 add_guest_mem_range(layout, start, end);
430 print_mem_layout(layout);
440 int main(int argc, char ** argv) {
441 struct vmm_os_hooks dummy_hooks;
442 os_hooks = &dummy_hooks;
444 vmm_mem_layout_t layout;
447 os_hooks->malloc = &TestMalloc;
448 os_hooks->free = &free;
449 os_hooks->print_debug = &TestPrint;
450 os_hooks->allocate_pages = &TestAllocatePages;
454 printf("mem_list_add_test_1: %d\n", mem_list_add_test_1(&list));
455 printf("layout_add_test_1: %d\n", mem_layout_add_test_1(&layout));