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 shadow_map * map) {
110 map->num_regions = 0;
116 void free_shadow_map(struct shadow_map * map) {
117 struct shadow_region * cursor = map->head;
118 struct shadow_region * tmp = NULL;
122 cursor = cursor->next;
132 int add_shadow_region(struct shadow_map * map,
133 struct shadow_region * region)
135 struct shadow_region * cursor = map->head;
137 PrintDebug("Adding Shadow Region: (0x%x-0x%x)\n", region->guest_start, region->guest_end);
139 if ((!cursor) || (cursor->guest_start >= region->guest_end)) {
141 region->next = cursor;
148 // Check if it overlaps with the current cursor
149 if ((cursor->guest_end > region->guest_start) && (cursor->guest_start < region->guest_start)) {
150 // overlaps not allowed
154 if (!(cursor->next)) {
155 // add to the end of the list
156 cursor->next = region;
157 region->prev = cursor;
161 } else if (cursor->next->guest_start >= region->guest_end) {
163 region->next = cursor->next;
164 region->prev = cursor;
166 cursor->next->prev = region;
167 cursor->next = region;
172 } else if (cursor->next->guest_end <= region->guest_start) {
173 cursor = cursor->next;
175 PrintDebug("WTF?\n");
176 // This cannot happen!
177 // we should panic here
182 // This cannot happen
183 // We should panic here
188 int delete_shadow_region(struct shadow_map * map,
196 struct shadow_region *get_shadow_region_by_index(struct shadow_map * map,
198 struct shadow_region * reg = map->head;
212 struct shadow_region * get_shadow_region_by_addr(struct shadow_map * map,
214 struct shadow_region * reg = map->head;
217 if ((reg->guest_start <= addr) && (reg->guest_end > addr)) {
219 } else if (reg->guest_start > addr) {
229 host_region_type_t get_shadow_addr_type(struct guest_info * info, addr_t guest_addr) {
230 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
233 return HOST_REGION_INVALID;
235 return reg->host_type;
239 addr_t get_shadow_addr(struct guest_info * info, addr_t guest_addr) {
240 struct shadow_region * reg = get_shadow_region_by_addr(&(info->mem_map), guest_addr);
245 return (guest_addr - reg->guest_start) + reg->host_addr;
250 host_region_type_t lookup_shadow_map_addr(struct shadow_map * map, addr_t guest_addr, addr_t * host_addr) {
251 struct shadow_region * reg = get_shadow_region_by_addr(map, guest_addr);
255 return HOST_REGION_INVALID;
257 switch (reg->host_type) {
258 case HOST_REGION_PHYSICAL_MEMORY:
259 *host_addr = (guest_addr - reg->guest_start) + reg->host_addr;
260 return reg->host_type;
261 case HOST_REGION_MEMORY_MAPPED_DEVICE:
262 case HOST_REGION_UNALLOCATED:
266 return reg->host_type;
272 void print_shadow_map(struct shadow_map * map) {
273 struct shadow_region * cur = map->head;
276 PrintDebug("Memory Layout (regions: %d) \n", map->num_regions);
279 PrintDebug("%d: 0x%x - 0x%x (%s) -> ", i, cur->guest_start, cur->guest_end - 1,
280 cur->guest_type == GUEST_REGION_PHYSICAL_MEMORY ? "GUEST_REGION_PHYSICAL_MEMORY" :
281 cur->guest_type == GUEST_REGION_NOTHING ? "GUEST_REGION_NOTHING" :
282 cur->guest_type == GUEST_REGION_MEMORY_MAPPED_DEVICE ? "GUEST_REGION_MEMORY_MAPPED_DEVICE" :
284 if (cur->host_type == HOST_REGION_PHYSICAL_MEMORY ||
285 cur->host_type == HOST_REGION_UNALLOCATED ||
286 cur->host_type == HOST_REGION_MEMORY_MAPPED_DEVICE) {
287 PrintDebug("0x%x", cur->host_addr);
290 cur->host_type == HOST_REGION_PHYSICAL_MEMORY ? "HOST_REGION_PHYSICAL_MEMORY" :
291 cur->host_type == HOST_REGION_UNALLOCATED ? "HOST_REGION_UNALLOACTED" :
292 cur->host_type == HOST_REGION_HOOK ? "HOST_REGION_HOOK" :
293 cur->host_type == HOST_REGION_MEMORY_MAPPED_DEVICE ? "HOST_REGION_MEMORY_MAPPED_DEVICE" :
294 cur->host_type == HOST_REGION_REMOTE ? "HOST_REGION_REMOTE" :
295 cur->host_type == HOST_REGION_SWAPPED ? "HOST_REGION_SWAPPED" :
322 struct vmm_os_hooks * os_hooks;
324 void * TestMalloc(uint_t size) {
328 void * TestAllocatePages(int size) {
329 return malloc(4096 * size);
333 void TestPrint(const char * fmt, ...) {
341 int mem_list_add_test_1( vmm_mem_list_t * list) {
345 PrintDebug("\n\nTesting Memory List\n");
349 offset = PAGE_SIZE * 6;
350 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 10));
351 add_mem_list_pages(list, offset, 10);
352 print_mem_list(list);
356 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + PAGE_SIZE * 4);
357 add_mem_list_pages(list, offset, 4);
358 print_mem_list(list);
360 offset = PAGE_SIZE * 20;
361 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 1));
362 add_mem_list_pages(list, offset, 1);
363 print_mem_list(list);
365 offset = PAGE_SIZE * 21;
366 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 3));
367 add_mem_list_pages(list, offset, 3);
368 print_mem_list(list);
371 offset = PAGE_SIZE * 10;
372 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 30));
373 add_mem_list_pages(list, offset, 30);
374 print_mem_list(list);
377 offset = PAGE_SIZE * 5;
378 PrintDebug("Adding 0x%x - 0x%x\n", offset, offset + (PAGE_SIZE * 1));
379 add_mem_list_pages(list, offset, 1);
380 print_mem_list(list);
388 int mem_layout_add_test_1(vmm_mem_layout_t * layout) {
394 PrintDebug("\n\nTesting Memory Layout\n");
396 init_mem_layout(layout);
400 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
401 add_guest_mem_range(layout, start, end);
402 print_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);
413 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
414 add_guest_mem_range(layout, start, end);
415 print_mem_layout(layout);
419 PrintDebug("Adding 0x%x - 0x%x\n", start, end);
420 add_guest_mem_range(layout, start, end);
421 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);
438 int main(int argc, char ** argv) {
439 struct vmm_os_hooks dummy_hooks;
440 os_hooks = &dummy_hooks;
442 vmm_mem_layout_t layout;
445 os_hooks->malloc = &TestMalloc;
446 os_hooks->free = &free;
447 os_hooks->print_debug = &TestPrint;
448 os_hooks->allocate_pages = &TestAllocatePages;
452 printf("mem_list_add_test_1: %d\n", mem_list_add_test_1(&list));
453 printf("layout_add_test_1: %d\n", mem_layout_add_test_1(&layout));