*/
#include <palacios/vmm.h>
+#include <palacios/vmm_dev_mgr.h>
+#include <devices/lnx_virtio_blk.h>
+#include <palacios/vmm_sym_swap.h>
-#include <devices/sym_swap.h>
-#include <devices/lnx_virtio.h>
+#define SWAP_CAPACITY (150 * 1024 * 1024)
+
+
+
+/* This is the first page that linux writes to the swap area */
+/* Taken from Linux */
+union swap_header {
+ struct {
+ char reserved[PAGE_SIZE - 10];
+ char magic[10]; /* SWAP-SPACE or SWAPSPACE2 */
+ } magic;
+ struct {
+ char bootbits[1024]; /* Space for disklabel etc. */
+ uint32_t version;
+ uint32_t last_page;
+ uint32_t nr_badpages;
+ unsigned char sws_uuid[16];
+ unsigned char sws_volume[16];
+ uint32_t type; // The index into the swap_map
+ uint32_t padding[116];
+ // uint32_t padding[117];
+ uint32_t badpages[1];
+ } info;
+};
struct swap_state {
-
+ int active;
+
struct vm_device * blk_dev;
+ uint_t swapped_pages;
+ uint_t unswapped_pages;
+
+
+ uint64_t capacity;
+ uint8_t * swap_space;
+ addr_t swap_base_addr;
+
+ uint8_t usage_map[0]; // This must be the last structure member
};
-static int swap_init(struct vm_device * dev) {
- return -1;
+
+
+static inline void set_index_usage(struct swap_state * swap, uint32_t index, int used) {
+ int major = index / 8;
+ int minor = index % 8;
+
+ if (used) {
+ swap->usage_map[major] |= (1 << minor);
+ } else {
+ swap->usage_map[major] &= ~(1 << minor);
+ }
+}
+
+static inline int get_index_usage(struct swap_state * swap, uint32_t index) {
+ int major = index / 8;
+ int minor = index % 8;
+ return swap->usage_map[major] & (1 << minor);
+}
+
+
+static inline uint32_t get_swap_index_from_offset(uint32_t offset) {
+ // CAREFUL: The index might be offset by 1, because the first 4K is the header
+ return (offset / 4096);
+}
+
+
+/*
+ static inline uint32_t get_swap_index(uint32_t offset) {
+ // CAREFUL: The index might be offset by 1, because the first 4K is the header
+ return (swap_addr - swap->swap_space) / 4096;
+ }
+*/
+
+
+static inline void * get_swap_entry(uint32_t pg_index, void * private_data) {
+ struct swap_state * swap = (struct swap_state *)private_data;
+ void * pg_addr = NULL;
+
+ if (get_index_usage(swap, pg_index)) {
+ // CAREFUL: The index might be offset by 1, because the first 4K is the header
+ pg_addr = (void *)(swap->swap_space + (pg_index * 4096));
+ }
+
+ return pg_addr;
+}
+
+
+
+static uint64_t swap_get_capacity(void * private_data) {
+ struct vm_device * dev = (struct vm_device *)private_data;
+ struct swap_state * swap = (struct swap_state *)(dev->private_data);
+
+ PrintDebug("SymSwap: Getting Capacity %d\n", (uint32_t)(swap->capacity));
+
+ return swap->capacity / HD_SECTOR_SIZE;
}
-static int swap_deinit(struct vm_device * dev) {
+static struct v3_swap_ops swap_ops = {
+ .get_swap_entry = get_swap_entry,
+};
+
+
+
+static int swap_read(uint8_t * buf, int sector_count, uint64_t lba, void * private_data) {
+ struct vm_device * dev = (struct vm_device *)private_data;
+ struct swap_state * swap = (struct swap_state *)(dev->private_data);
+ uint32_t offset = lba * HD_SECTOR_SIZE;
+ uint32_t length = sector_count * HD_SECTOR_SIZE;
+
+
+ PrintDebug("SymSwap: Reading %d bytes to %p from %p\n", length,
+ buf, (void *)(swap->swap_space + offset));
+
+ if (length % 4096) {
+ PrintError("Swapping in length that is not a page multiple\n");
+ }
+
+ memcpy(buf, swap->swap_space + offset, length);
+
+ swap->unswapped_pages += (length / 4096);
+
+ set_index_usage(swap, get_swap_index_from_offset(offset), 0);
+
+
+ // Notify the shadow paging layer
+
+ PrintDebug("Swapped in %d pages\n", length / 4096);
+
+ return 0;
+}
+
+
+
+
+static int swap_write(uint8_t * buf, int sector_count, uint64_t lba, void * private_data) {
+ struct vm_device * dev = (struct vm_device *)private_data;
+ struct swap_state * swap = (struct swap_state *)(dev->private_data);
+ uint32_t offset = lba * HD_SECTOR_SIZE;
+ uint32_t length = sector_count * HD_SECTOR_SIZE;
+
+ /*
+ PrintDebug("SymSwap: Writing %d bytes to %p from %p\n", length,
+ (void *)(swap->swap_space + offset), buf);
+ */
+
+ if (length % 4096) {
+ PrintError("Swapping out length that is not a page multiple\n");
+ }
+
+ if ((swap->active == 0) && (offset == 0)) {
+ // This is the swap header page
+ union swap_header * hdr = (union swap_header *)buf;
+
+ if (length != 4096) {
+ PrintError("Initializing Swap space by not writing page multiples. This sucks...\n");
+ return -1;
+ }
+
+ swap->active = 1;
+
+ PrintDebug("Swap Type=%d (magic=%s)\n", hdr->info.type, hdr->magic.magic);
+
+ if (v3_register_swap_disk(dev->vm, hdr->info.type, &swap_ops, dev) == -1) {
+ PrintError("Error registering symbiotic swap disk\n");
+ return -1;
+ }
+
+ }
+
+ memcpy(swap->swap_space + offset, buf, length);
+
+ swap->swapped_pages += (length / 4096);
+
+ set_index_usage(swap, get_swap_index_from_offset(offset), 1);
+
+ PrintDebug("Swapped out %d pages\n", length / 4096);
+
+ return 0;
+}
+
+
+static int swap_free(struct vm_device * dev) {
return -1;
}
+static struct v3_hd_ops hd_ops = {
+ .read = swap_read,
+ .write = swap_write,
+ .get_capacity = swap_get_capacity,
+};
+
+
-static struct vm_device_ops dev_ops = {
- .init = swap_init,
- .deinit = swap_deinit,
+static struct v3_device_ops dev_ops = {
+ .free = swap_free,
.reset = NULL,
.start = NULL,
.stop = NULL,
-struct vm_device * v3_create_swap(struct vm_device * virtio_blk) {
+
+
+
+static int swap_init(struct guest_info * vm, void * cfg_data) {
struct swap_state * swap = NULL;
+ struct vm_device * virtio_blk = v3_find_dev(vm, (char *)cfg_data);
+
+ if (!virtio_blk) {
+ PrintError("could not find Virtio backend\n");
+ return -1;
+ }
PrintDebug("Creating Swap Device\n");
if (virtio_blk == NULL) {
PrintError("Swap device requires a virtio block device\n");
- return NULL;
+ return -1;
}
- swap = (struct swap_state *)V3_Malloc(sizeof(struct swap_state));
+ swap = (struct swap_state *)V3_Malloc(sizeof(struct swap_state) + ((SWAP_CAPACITY / 4096) / 8));
swap->blk_dev = virtio_blk;
+ swap->capacity = SWAP_CAPACITY;
+
+ swap->swapped_pages = 0;
+ swap->unswapped_pages = 0;
- return v3_create_device("SYM_SWAP", &dev_ops, swap);
+ swap->active = 0;
+
+ swap->swap_base_addr = (addr_t)V3_AllocPages(swap->capacity / 4096);
+ swap->swap_space = (uint8_t *)V3_VAddr((void *)(swap->swap_base_addr));
+ memset(swap->swap_space, 0, SWAP_CAPACITY);
+
+ memset(swap->usage_map, 0, ((SWAP_CAPACITY / 4096) / 8));
+
+ struct vm_device * dev = v3_allocate_device("SYM_SWAP", &dev_ops, swap);
+
+ if (v3_attach_device(vm, dev) == -1) {
+ PrintError("Could not attach device %s\n", "SYM_SWAP");
+ return -1;
+ }
+
+
+ v3_virtio_register_harddisk(virtio_blk, &hd_ops, dev);
+
+ return 0;
}
+
+
+
+device_register("SYM_SWAP", swap_init)
