xen-mapcache.c

   1 /*
   2  * Copyright (C) 2011       Citrix Ltd.
   3  *
   4  * This work is licensed under the terms of the GNU GPL, version 2.  See
   5  * the COPYING file in the top-level directory.
   6  *
   7  * Contributions after 2012-01-13 are licensed under the terms of the
   8  * GNU GPL, version 2 or (at your option) any later version.
   9  */
  10
  11 #include "config.h"
  12
  13 #include <sys/resource.h>
  14
  15 #include "hw/xen_backend.h"
  16 #include "blockdev.h"
  17 #include "bitmap.h"
  18
  19 #include <xen/hvm/params.h>
  20 #include <sys/mman.h>
  21
  22 #include "xen-mapcache.h"
  23 #include "trace.h"
  24
  25
  26 //#define MAPCACHE_DEBUG
  27
  28 #ifdef MAPCACHE_DEBUG
  29 #  define DPRINTF(fmt, ...) do { \
  30     fprintf(stderr, "xen_mapcache: " fmt, ## __VA_ARGS__); \
  31 } while (0)
  32 #else
  33 #  define DPRINTF(fmt, ...) do { } while (0)
  34 #endif
  35
  36 #if defined(__i386__)
  37 #  define MCACHE_BUCKET_SHIFT 16
  38 #  define MCACHE_MAX_SIZE     (1UL<<31) /* 2GB Cap */
  39 #elif defined(__x86_64__)
  40 #  define MCACHE_BUCKET_SHIFT 20
  41 #  define MCACHE_MAX_SIZE     (1UL<<35) /* 32GB Cap */
  42 #endif
  43 #define MCACHE_BUCKET_SIZE (1UL << MCACHE_BUCKET_SHIFT)
  44
  45 /* This is the size of the virtual address space reserve to QEMU that will not
  46  * be use by MapCache.
  47  * From empirical tests I observed that qemu use 75MB more than the
  48  * max_mcache_size.
  49  */
  50 #define NON_MCACHE_MEMORY_SIZE (80 * 1024 * 1024)
  51
  52 #define mapcache_lock()   ((void)0)
  53 #define mapcache_unlock() ((void)0)
  54
  55 typedef struct MapCacheEntry {
  56     target_phys_addr_t paddr_index;
  57     uint8_t *vaddr_base;
  58     unsigned long *valid_mapping;
  59     uint8_t lock;
  60     target_phys_addr_t size;
  61     struct MapCacheEntry *next;
  62 } MapCacheEntry;
  63
  64 typedef struct MapCacheRev {
  65     uint8_t *vaddr_req;
  66     target_phys_addr_t paddr_index;
  67     target_phys_addr_t size;
  68     QTAILQ_ENTRY(MapCacheRev) next;
  69 } MapCacheRev;
  70
  71 typedef struct MapCache {
  72     MapCacheEntry *entry;
  73     unsigned long nr_buckets;
  74     QTAILQ_HEAD(map_cache_head, MapCacheRev) locked_entries;
  75
  76     /* For most cases (>99.9%), the page address is the same. */
  77     target_phys_addr_t last_address_index;
  78     uint8_t *last_address_vaddr;
  79     unsigned long max_mcache_size;
  80     unsigned int mcache_bucket_shift;
  81
  82     phys_offset_to_gaddr_t phys_offset_to_gaddr;
  83     void *opaque;
  84 } MapCache;
  85
  86 static MapCache *mapcache;
  87
  88 static inline int test_bits(int nr, int size, const unsigned long *addr)
  89 {
  90     unsigned long res = find_next_zero_bit(addr, size + nr, nr);
  91     if (res >= nr + size)
  92         return 1;
  93     else
  94         return 0;
  95 }
  96
  97 void xen_map_cache_init(phys_offset_to_gaddr_t f, void *opaque)
  98 {
  99     unsigned long size;
 100     struct rlimit rlimit_as;
 101
 102     mapcache = g_malloc0(sizeof (MapCache));
 103
 104     mapcache->phys_offset_to_gaddr = f;
 105     mapcache->opaque = opaque;
 106
 107     QTAILQ_INIT(&mapcache->locked_entries);
 108     mapcache->last_address_index = -1;
 109
 110     if (geteuid() == 0) {
 111         rlimit_as.rlim_cur = RLIM_INFINITY;
 112         rlimit_as.rlim_max = RLIM_INFINITY;
 113         mapcache->max_mcache_size = MCACHE_MAX_SIZE;
 114     } else {
 115         getrlimit(RLIMIT_AS, &rlimit_as);
 116         rlimit_as.rlim_cur = rlimit_as.rlim_max;
 117
 118         if (rlimit_as.rlim_max != RLIM_INFINITY) {
 119             fprintf(stderr, "Warning: QEMU's maximum size of virtual"
 120                     " memory is not infinity.\n");
 121         }
 122         if (rlimit_as.rlim_max < MCACHE_MAX_SIZE + NON_MCACHE_MEMORY_SIZE) {
 123             mapcache->max_mcache_size = rlimit_as.rlim_max -
 124                 NON_MCACHE_MEMORY_SIZE;
 125         } else {
 126             mapcache->max_mcache_size = MCACHE_MAX_SIZE;
 127         }
 128     }
 129
 130     setrlimit(RLIMIT_AS, &rlimit_as);
 131
 132     mapcache->nr_buckets =
 133         (((mapcache->max_mcache_size >> XC_PAGE_SHIFT) +
 134           (1UL << (MCACHE_BUCKET_SHIFT - XC_PAGE_SHIFT)) - 1) >>
 135          (MCACHE_BUCKET_SHIFT - XC_PAGE_SHIFT));
 136
 137     size = mapcache->nr_buckets * sizeof (MapCacheEntry);
 138     size = (size + XC_PAGE_SIZE - 1) & ~(XC_PAGE_SIZE - 1);
 139     DPRINTF("%s, nr_buckets = %lx size %lu\n", __func__,
 140             mapcache->nr_buckets, size);
 141     mapcache->entry = g_malloc0(size);
 142 }
 143
 144 static void xen_remap_bucket(MapCacheEntry *entry,
 145                              target_phys_addr_t size,
 146                              target_phys_addr_t address_index)
 147 {
 148     uint8_t *vaddr_base;
 149     xen_pfn_t *pfns;
 150     int *err;
 151     unsigned int i;
 152     target_phys_addr_t nb_pfn = size >> XC_PAGE_SHIFT;
 153
 154     trace_xen_remap_bucket(address_index);
 155
 156     pfns = g_malloc0(nb_pfn * sizeof (xen_pfn_t));
 157     err = g_malloc0(nb_pfn * sizeof (int));
 158
 159     if (entry->vaddr_base != NULL) {
 160         if (munmap(entry->vaddr_base, entry->size) != 0) {
 161             perror("unmap fails");
 162             exit(-1);
 163         }
 164     }
 165     if (entry->valid_mapping != NULL) {
 166         g_free(entry->valid_mapping);
 167         entry->valid_mapping = NULL;
 168     }
 169
 170     for (i = 0; i < nb_pfn; i++) {
 171         pfns[i] = (address_index << (MCACHE_BUCKET_SHIFT-XC_PAGE_SHIFT)) + i;
 172     }
 173
 174     vaddr_base = xc_map_foreign_bulk(xen_xc, xen_domid, PROT_READ|PROT_WRITE,
 175                                      pfns, err, nb_pfn);
 176     if (vaddr_base == NULL) {
 177         perror("xc_map_foreign_bulk");
 178         exit(-1);
 179     }
 180
 181     entry->vaddr_base = vaddr_base;
 182     entry->paddr_index = address_index;
 183     entry->size = size;
 184     entry->valid_mapping = (unsigned long *) g_malloc0(sizeof(unsigned long) *
 185             BITS_TO_LONGS(size >> XC_PAGE_SHIFT));
 186
 187     bitmap_zero(entry->valid_mapping, nb_pfn);
 188     for (i = 0; i < nb_pfn; i++) {
 189         if (!err[i]) {
 190             bitmap_set(entry->valid_mapping, i, 1);
 191         }
 192     }
 193
 194     g_free(pfns);
 195     g_free(err);
 196 }
 197
 198 uint8_t *xen_map_cache(target_phys_addr_t phys_addr, target_phys_addr_t size,
 199                        uint8_t lock)
 200 {
 201     MapCacheEntry *entry, *pentry = NULL;
 202     target_phys_addr_t address_index;
 203     target_phys_addr_t address_offset;
 204     target_phys_addr_t __size = size;
 205     bool translated = false;
 206
 207 tryagain:
 208     address_index  = phys_addr >> MCACHE_BUCKET_SHIFT;
 209     address_offset = phys_addr & (MCACHE_BUCKET_SIZE - 1);
 210
 211     trace_xen_map_cache(phys_addr);
 212
 213     if (address_index == mapcache->last_address_index && !lock && !__size) {
 214         trace_xen_map_cache_return(mapcache->last_address_vaddr + address_offset);
 215         return mapcache->last_address_vaddr + address_offset;
 216     }
 217
 218     /* size is always a multiple of MCACHE_BUCKET_SIZE */
 219     if ((address_offset + (__size % MCACHE_BUCKET_SIZE)) > MCACHE_BUCKET_SIZE)
 220         __size += MCACHE_BUCKET_SIZE;
 221     if (__size % MCACHE_BUCKET_SIZE)
 222         __size += MCACHE_BUCKET_SIZE - (__size % MCACHE_BUCKET_SIZE);
 223     if (!__size)
 224         __size = MCACHE_BUCKET_SIZE;
 225
 226     entry = &mapcache->entry[address_index % mapcache->nr_buckets];
 227
 228     while (entry && entry->lock && entry->vaddr_base &&
 229             (entry->paddr_index != address_index || entry->size != __size ||
 230              !test_bits(address_offset >> XC_PAGE_SHIFT, size >> XC_PAGE_SHIFT,
 231                  entry->valid_mapping))) {
 232         pentry = entry;
 233         entry = entry->next;
 234     }
 235     if (!entry) {
 236         entry = g_malloc0(sizeof (MapCacheEntry));
 237         pentry->next = entry;
 238         xen_remap_bucket(entry, __size, address_index);
 239     } else if (!entry->lock) {
 240         if (!entry->vaddr_base || entry->paddr_index != address_index ||
 241                 entry->size != __size ||
 242                 !test_bits(address_offset >> XC_PAGE_SHIFT, size >> XC_PAGE_SHIFT,
 243                     entry->valid_mapping)) {
 244             xen_remap_bucket(entry, __size, address_index);
 245         }
 246     }
 247
 248     if(!test_bits(address_offset >> XC_PAGE_SHIFT, size >> XC_PAGE_SHIFT,
 249                 entry->valid_mapping)) {
 250         mapcache->last_address_index = -1;
 251         if (!translated && mapcache->phys_offset_to_gaddr) {
 252             phys_addr = mapcache->phys_offset_to_gaddr(phys_addr, size, mapcache->opaque);
 253             translated = true;
 254             goto tryagain;
 255         }
 256         trace_xen_map_cache_return(NULL);
 257         return NULL;
 258     }
 259
 260     mapcache->last_address_index = address_index;
 261     mapcache->last_address_vaddr = entry->vaddr_base;
 262     if (lock) {
 263         MapCacheRev *reventry = g_malloc0(sizeof(MapCacheRev));
 264         entry->lock++;
 265         reventry->vaddr_req = mapcache->last_address_vaddr + address_offset;
 266         reventry->paddr_index = mapcache->last_address_index;
 267         reventry->size = entry->size;
 268         QTAILQ_INSERT_HEAD(&mapcache->locked_entries, reventry, next);
 269     }
 270
 271     trace_xen_map_cache_return(mapcache->last_address_vaddr + address_offset);
 272     return mapcache->last_address_vaddr + address_offset;
 273 }
 274
 275 ram_addr_t xen_ram_addr_from_mapcache(void *ptr)
 276 {
 277     MapCacheEntry *entry = NULL;
 278     MapCacheRev *reventry;
 279     target_phys_addr_t paddr_index;
 280     target_phys_addr_t size;
 281     int found = 0;
 282
 283     QTAILQ_FOREACH(reventry, &mapcache->locked_entries, next) {
 284         if (reventry->vaddr_req == ptr) {
 285             paddr_index = reventry->paddr_index;
 286             size = reventry->size;
 287             found = 1;
 288             break;
 289         }
 290     }
 291     if (!found) {
 292         fprintf(stderr, "%s, could not find %p\n", __func__, ptr);
 293         QTAILQ_FOREACH(reventry, &mapcache->locked_entries, next) {
 294             DPRINTF("   "TARGET_FMT_plx" -> %p is present\n", reventry->paddr_index,
 295                     reventry->vaddr_req);
 296         }
 297         abort();
 298         return 0;
 299     }
 300
 301     entry = &mapcache->entry[paddr_index % mapcache->nr_buckets];
 302     while (entry && (entry->paddr_index != paddr_index || entry->size != size)) {
 303         entry = entry->next;
 304     }
 305     if (!entry) {
 306         DPRINTF("Trying to find address %p that is not in the mapcache!\n", ptr);
 307         return 0;
 308     }
 309     return (reventry->paddr_index << MCACHE_BUCKET_SHIFT) +
 310         ((unsigned long) ptr - (unsigned long) entry->vaddr_base);
 311 }
 312
 313 void xen_invalidate_map_cache_entry(uint8_t *buffer)
 314 {
 315     MapCacheEntry *entry = NULL, *pentry = NULL;
 316     MapCacheRev *reventry;
 317     target_phys_addr_t paddr_index;
 318     target_phys_addr_t size;
 319     int found = 0;
 320
 321     if (mapcache->last_address_vaddr == buffer) {
 322         mapcache->last_address_index = -1;
 323     }
 324
 325     QTAILQ_FOREACH(reventry, &mapcache->locked_entries, next) {
 326         if (reventry->vaddr_req == buffer) {
 327             paddr_index = reventry->paddr_index;
 328             size = reventry->size;
 329             found = 1;
 330             break;
 331         }
 332     }
 333     if (!found) {
 334         DPRINTF("%s, could not find %p\n", __func__, buffer);
 335         QTAILQ_FOREACH(reventry, &mapcache->locked_entries, next) {
 336             DPRINTF("   "TARGET_FMT_plx" -> %p is present\n", reventry->paddr_index, reventry->vaddr_req);
 337         }
 338         return;
 339     }
 340     QTAILQ_REMOVE(&mapcache->locked_entries, reventry, next);
 341     g_free(reventry);
 342
 343     entry = &mapcache->entry[paddr_index % mapcache->nr_buckets];
 344     while (entry && (entry->paddr_index != paddr_index || entry->size != size)) {
 345         pentry = entry;
 346         entry = entry->next;
 347     }
 348     if (!entry) {
 349         DPRINTF("Trying to unmap address %p that is not in the mapcache!\n", buffer);
 350         return;
 351     }
 352     entry->lock--;
 353     if (entry->lock > 0 || pentry == NULL) {
 354         return;
 355     }
 356
 357     pentry->next = entry->next;
 358     if (munmap(entry->vaddr_base, entry->size) != 0) {
 359         perror("unmap fails");
 360         exit(-1);
 361     }
 362     g_free(entry->valid_mapping);
 363     g_free(entry);
 364 }
 365
 366 void xen_invalidate_map_cache(void)
 367 {
 368     unsigned long i;
 369     MapCacheRev *reventry;
 370
 371     /* Flush pending AIO before destroying the mapcache */
 372     bdrv_drain_all();
 373
 374     QTAILQ_FOREACH(reventry, &mapcache->locked_entries, next) {
 375         DPRINTF("There should be no locked mappings at this time, "
 376                 "but "TARGET_FMT_plx" -> %p is present\n",
 377                 reventry->paddr_index, reventry->vaddr_req);
 378     }
 379
 380     mapcache_lock();
 381
 382     for (i = 0; i < mapcache->nr_buckets; i++) {
 383         MapCacheEntry *entry = &mapcache->entry[i];
 384
 385         if (entry->vaddr_base == NULL) {
 386             continue;
 387         }
 388
 389         if (munmap(entry->vaddr_base, entry->size) != 0) {
 390             perror("unmap fails");
 391             exit(-1);
 392         }
 393
 394         entry->paddr_index = 0;
 395         entry->vaddr_base = NULL;
 396         entry->size = 0;
 397         g_free(entry->valid_mapping);
 398         entry->valid_mapping = NULL;
 399     }
 400
 401     mapcache->last_address_index = -1;
 402     mapcache->last_address_vaddr = NULL;
 403
 404     mapcache_unlock();
 405 }