bsd-user/mmap.c

   1 /*
   2  *  mmap support for qemu
   3  *
   4  *  Copyright (c) 2003 - 2008 Fabrice Bellard
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 2 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  */
  19 #include "qemu/osdep.h"
  20
  21 #include "qemu.h"
  22 #include "qemu-common.h"
  23 #include "bsd-mman.h"
  24
  25 //#define DEBUG_MMAP
  26
  27 static pthread_mutex_t mmap_mutex = PTHREAD_MUTEX_INITIALIZER;
  28 static __thread int mmap_lock_count;
  29
  30 void mmap_lock(void)
  31 {
  32     if (mmap_lock_count++ == 0) {
  33         pthread_mutex_lock(&mmap_mutex);
  34     }
  35 }
  36
  37 void mmap_unlock(void)
  38 {
  39     if (--mmap_lock_count == 0) {
  40         pthread_mutex_unlock(&mmap_mutex);
  41     }
  42 }
  43
  44 bool have_mmap_lock(void)
  45 {
  46     return mmap_lock_count > 0 ? true : false;
  47 }
  48
  49 /* Grab lock to make sure things are in a consistent state after fork().  */
  50 void mmap_fork_start(void)
  51 {
  52     if (mmap_lock_count)
  53         abort();
  54     pthread_mutex_lock(&mmap_mutex);
  55 }
  56
  57 void mmap_fork_end(int child)
  58 {
  59     if (child)
  60         pthread_mutex_init(&mmap_mutex, NULL);
  61     else
  62         pthread_mutex_unlock(&mmap_mutex);
  63 }
  64
  65 /* NOTE: all the constants are the HOST ones, but addresses are target. */
  66 int target_mprotect(abi_ulong start, abi_ulong len, int prot)
  67 {
  68     abi_ulong end, host_start, host_end, addr;
  69     int prot1, ret;
  70
  71 #ifdef DEBUG_MMAP
  72     printf("mprotect: start=0x" TARGET_FMT_lx
  73            " len=0x" TARGET_FMT_lx " prot=%c%c%c\n", start, len,
  74            prot & PROT_READ ? 'r' : '-',
  75            prot & PROT_WRITE ? 'w' : '-',
  76            prot & PROT_EXEC ? 'x' : '-');
  77 #endif
  78
  79     if ((start & ~TARGET_PAGE_MASK) != 0)
  80         return -EINVAL;
  81     len = TARGET_PAGE_ALIGN(len);
  82     end = start + len;
  83     if (end < start)
  84         return -EINVAL;
  85     prot &= PROT_READ | PROT_WRITE | PROT_EXEC;
  86     if (len == 0)
  87         return 0;
  88
  89     mmap_lock();
  90     host_start = start & qemu_host_page_mask;
  91     host_end = HOST_PAGE_ALIGN(end);
  92     if (start > host_start) {
  93         /* handle host page containing start */
  94         prot1 = prot;
  95         for(addr = host_start; addr < start; addr += TARGET_PAGE_SIZE) {
  96             prot1 |= page_get_flags(addr);
  97         }
  98         if (host_end == host_start + qemu_host_page_size) {
  99             for(addr = end; addr < host_end; addr += TARGET_PAGE_SIZE) {
 100                 prot1 |= page_get_flags(addr);
 101             }
 102             end = host_end;
 103         }
 104         ret = mprotect(g2h(host_start), qemu_host_page_size, prot1 & PAGE_BITS);
 105         if (ret != 0)
 106             goto error;
 107         host_start += qemu_host_page_size;
 108     }
 109     if (end < host_end) {
 110         prot1 = prot;
 111         for(addr = end; addr < host_end; addr += TARGET_PAGE_SIZE) {
 112             prot1 |= page_get_flags(addr);
 113         }
 114         ret = mprotect(g2h(host_end - qemu_host_page_size), qemu_host_page_size,
 115                        prot1 & PAGE_BITS);
 116         if (ret != 0)
 117             goto error;
 118         host_end -= qemu_host_page_size;
 119     }
 120
 121     /* handle the pages in the middle */
 122     if (host_start < host_end) {
 123         ret = mprotect(g2h(host_start), host_end - host_start, prot);
 124         if (ret != 0)
 125             goto error;
 126     }
 127     page_set_flags(start, start + len, prot | PAGE_VALID);
 128     mmap_unlock();
 129     return 0;
 130 error:
 131     mmap_unlock();
 132     return ret;
 133 }
 134
 135 /* map an incomplete host page */
 136 static int mmap_frag(abi_ulong real_start,
 137                      abi_ulong start, abi_ulong end,
 138                      int prot, int flags, int fd, abi_ulong offset)
 139 {
 140     abi_ulong real_end, addr;
 141     void *host_start;
 142     int prot1, prot_new;
 143
 144     real_end = real_start + qemu_host_page_size;
 145     host_start = g2h(real_start);
 146
 147     /* get the protection of the target pages outside the mapping */
 148     prot1 = 0;
 149     for(addr = real_start; addr < real_end; addr++) {
 150         if (addr < start || addr >= end)
 151             prot1 |= page_get_flags(addr);
 152     }
 153
 154     if (prot1 == 0) {
 155         /* no page was there, so we allocate one */
 156         void *p = mmap(host_start, qemu_host_page_size, prot,
 157                        flags | MAP_ANON, -1, 0);
 158         if (p == MAP_FAILED)
 159             return -1;
 160         prot1 = prot;
 161     }
 162     prot1 &= PAGE_BITS;
 163
 164     prot_new = prot | prot1;
 165     if (!(flags & MAP_ANON)) {
 166         /* msync() won't work here, so we return an error if write is
 167            possible while it is a shared mapping */
 168         if ((flags & TARGET_BSD_MAP_FLAGMASK) == MAP_SHARED &&
 169             (prot & PROT_WRITE))
 170             return -1;
 171
 172         /* adjust protection to be able to read */
 173         if (!(prot1 & PROT_WRITE))
 174             mprotect(host_start, qemu_host_page_size, prot1 | PROT_WRITE);
 175
 176         /* read the corresponding file data */
 177         pread(fd, g2h(start), end - start, offset);
 178
 179         /* put final protection */
 180         if (prot_new != (prot1 | PROT_WRITE))
 181             mprotect(host_start, qemu_host_page_size, prot_new);
 182     } else {
 183         /* just update the protection */
 184         if (prot_new != prot1) {
 185             mprotect(host_start, qemu_host_page_size, prot_new);
 186         }
 187     }
 188     return 0;
 189 }
 190
 191 static abi_ulong mmap_next_start = 0x40000000;
 192
 193 unsigned long last_brk;
 194
 195 /* find a free memory area of size 'size'. The search starts at
 196    'start'. If 'start' == 0, then a default start address is used.
 197    Return -1 if error.
 198 */
 199 /* page_init() marks pages used by the host as reserved to be sure not
 200    to use them. */
 201 static abi_ulong mmap_find_vma(abi_ulong start, abi_ulong size)
 202 {
 203     abi_ulong addr, addr1, addr_start;
 204     int prot;
 205     unsigned long new_brk;
 206
 207     new_brk = (unsigned long)sbrk(0);
 208     if (last_brk && last_brk < new_brk && last_brk == (target_ulong)last_brk) {
 209         /* This is a hack to catch the host allocating memory with brk().
 210            If it uses mmap then we loose.
 211            FIXME: We really want to avoid the host allocating memory in
 212            the first place, and maybe leave some slack to avoid switching
 213            to mmap.  */
 214         page_set_flags(last_brk & TARGET_PAGE_MASK,
 215                        TARGET_PAGE_ALIGN(new_brk),
 216                        PAGE_RESERVED);
 217     }
 218     last_brk = new_brk;
 219
 220     size = HOST_PAGE_ALIGN(size);
 221     start = start & qemu_host_page_mask;
 222     addr = start;
 223     if (addr == 0)
 224         addr = mmap_next_start;
 225     addr_start = addr;
 226     for(;;) {
 227         prot = 0;
 228         for(addr1 = addr; addr1 < (addr + size); addr1 += TARGET_PAGE_SIZE) {
 229             prot |= page_get_flags(addr1);
 230         }
 231         if (prot == 0)
 232             break;
 233         addr += qemu_host_page_size;
 234         /* we found nothing */
 235         if (addr == addr_start)
 236             return (abi_ulong)-1;
 237     }
 238     if (start == 0)
 239         mmap_next_start = addr + size;
 240     return addr;
 241 }
 242
 243 /* NOTE: all the constants are the HOST ones */
 244 abi_long target_mmap(abi_ulong start, abi_ulong len, int prot,
 245                      int flags, int fd, abi_ulong offset)
 246 {
 247     abi_ulong ret, end, real_start, real_end, retaddr, host_offset, host_len;
 248     unsigned long host_start;
 249
 250     mmap_lock();
 251 #ifdef DEBUG_MMAP
 252     {
 253         printf("mmap: start=0x" TARGET_FMT_lx
 254                " len=0x" TARGET_FMT_lx " prot=%c%c%c flags=",
 255                start, len,
 256                prot & PROT_READ ? 'r' : '-',
 257                prot & PROT_WRITE ? 'w' : '-',
 258                prot & PROT_EXEC ? 'x' : '-');
 259         if (flags & MAP_FIXED)
 260             printf("MAP_FIXED ");
 261         if (flags & MAP_ANON)
 262             printf("MAP_ANON ");
 263         switch(flags & TARGET_BSD_MAP_FLAGMASK) {
 264         case MAP_PRIVATE:
 265             printf("MAP_PRIVATE ");
 266             break;
 267         case MAP_SHARED:
 268             printf("MAP_SHARED ");
 269             break;
 270         default:
 271             printf("[MAP_FLAGMASK=0x%x] ", flags & TARGET_BSD_MAP_FLAGMASK);
 272             break;
 273         }
 274         printf("fd=%d offset=" TARGET_FMT_lx "\n", fd, offset);
 275     }
 276 #endif
 277
 278     if (offset & ~TARGET_PAGE_MASK) {
 279         errno = EINVAL;
 280         goto fail;
 281     }
 282
 283     len = TARGET_PAGE_ALIGN(len);
 284     if (len == 0)
 285         goto the_end;
 286     real_start = start & qemu_host_page_mask;
 287
 288     if (!(flags & MAP_FIXED)) {
 289         abi_ulong mmap_start;
 290         void *p;
 291         host_offset = offset & qemu_host_page_mask;
 292         host_len = len + offset - host_offset;
 293         host_len = HOST_PAGE_ALIGN(host_len);
 294         mmap_start = mmap_find_vma(real_start, host_len);
 295         if (mmap_start == (abi_ulong)-1) {
 296             errno = ENOMEM;
 297             goto fail;
 298         }
 299         /* Note: we prefer to control the mapping address. It is
 300            especially important if qemu_host_page_size >
 301            qemu_real_host_page_size */
 302         p = mmap(g2h(mmap_start),
 303                  host_len, prot, flags | MAP_FIXED, fd, host_offset);
 304         if (p == MAP_FAILED)
 305             goto fail;
 306         /* update start so that it points to the file position at 'offset' */
 307         host_start = (unsigned long)p;
 308         if (!(flags & MAP_ANON))
 309             host_start += offset - host_offset;
 310         start = h2g(host_start);
 311     } else {
 312         int flg;
 313         target_ulong addr;
 314
 315         if (start & ~TARGET_PAGE_MASK) {
 316             errno = EINVAL;
 317             goto fail;
 318         }
 319         end = start + len;
 320         real_end = HOST_PAGE_ALIGN(end);
 321
 322         for(addr = real_start; addr < real_end; addr += TARGET_PAGE_SIZE) {
 323             flg = page_get_flags(addr);
 324             if (flg & PAGE_RESERVED) {
 325                 errno = ENXIO;
 326                 goto fail;
 327             }
 328         }
 329
 330         /* worst case: we cannot map the file because the offset is not
 331            aligned, so we read it */
 332         if (!(flags & MAP_ANON) &&
 333             (offset & ~qemu_host_page_mask) != (start & ~qemu_host_page_mask)) {
 334             /* msync() won't work here, so we return an error if write is
 335                possible while it is a shared mapping */
 336             if ((flags & TARGET_BSD_MAP_FLAGMASK) == MAP_SHARED &&
 337                 (prot & PROT_WRITE)) {
 338                 errno = EINVAL;
 339                 goto fail;
 340             }
 341             retaddr = target_mmap(start, len, prot | PROT_WRITE,
 342                                   MAP_FIXED | MAP_PRIVATE | MAP_ANON,
 343                                   -1, 0);
 344             if (retaddr == -1)
 345                 goto fail;
 346             pread(fd, g2h(start), len, offset);
 347             if (!(prot & PROT_WRITE)) {
 348                 ret = target_mprotect(start, len, prot);
 349                 if (ret != 0) {
 350                     start = ret;
 351                     goto the_end;
 352                 }
 353             }
 354             goto the_end;
 355         }
 356
 357         /* handle the start of the mapping */
 358         if (start > real_start) {
 359             if (real_end == real_start + qemu_host_page_size) {
 360                 /* one single host page */
 361                 ret = mmap_frag(real_start, start, end,
 362                                 prot, flags, fd, offset);
 363                 if (ret == -1)
 364                     goto fail;
 365                 goto the_end1;
 366             }
 367             ret = mmap_frag(real_start, start, real_start + qemu_host_page_size,
 368                             prot, flags, fd, offset);
 369             if (ret == -1)
 370                 goto fail;
 371             real_start += qemu_host_page_size;
 372         }
 373         /* handle the end of the mapping */
 374         if (end < real_end) {
 375             ret = mmap_frag(real_end - qemu_host_page_size,
 376                             real_end - qemu_host_page_size, real_end,
 377                             prot, flags, fd,
 378                             offset + real_end - qemu_host_page_size - start);
 379             if (ret == -1)
 380                 goto fail;
 381             real_end -= qemu_host_page_size;
 382         }
 383
 384         /* map the middle (easier) */
 385         if (real_start < real_end) {
 386             void *p;
 387             unsigned long offset1;
 388             if (flags & MAP_ANON)
 389                 offset1 = 0;
 390             else
 391                 offset1 = offset + real_start - start;
 392             p = mmap(g2h(real_start), real_end - real_start,
 393                      prot, flags, fd, offset1);
 394             if (p == MAP_FAILED)
 395                 goto fail;
 396         }
 397     }
 398  the_end1:
 399     page_set_flags(start, start + len, prot | PAGE_VALID);
 400  the_end:
 401 #ifdef DEBUG_MMAP
 402     printf("ret=0x" TARGET_FMT_lx "\n", start);
 403     page_dump(stdout);
 404     printf("\n");
 405 #endif
 406     mmap_unlock();
 407     return start;
 408 fail:
 409     mmap_unlock();
 410     return -1;
 411 }
 412
 413 int target_munmap(abi_ulong start, abi_ulong len)
 414 {
 415     abi_ulong end, real_start, real_end, addr;
 416     int prot, ret;
 417
 418 #ifdef DEBUG_MMAP
 419     printf("munmap: start=0x%lx len=0x%lx\n", start, len);
 420 #endif
 421     if (start & ~TARGET_PAGE_MASK)
 422         return -EINVAL;
 423     len = TARGET_PAGE_ALIGN(len);
 424     if (len == 0)
 425         return -EINVAL;
 426     mmap_lock();
 427     end = start + len;
 428     real_start = start & qemu_host_page_mask;
 429     real_end = HOST_PAGE_ALIGN(end);
 430
 431     if (start > real_start) {
 432         /* handle host page containing start */
 433         prot = 0;
 434         for(addr = real_start; addr < start; addr += TARGET_PAGE_SIZE) {
 435             prot |= page_get_flags(addr);
 436         }
 437         if (real_end == real_start + qemu_host_page_size) {
 438             for(addr = end; addr < real_end; addr += TARGET_PAGE_SIZE) {
 439                 prot |= page_get_flags(addr);
 440             }
 441             end = real_end;
 442         }
 443         if (prot != 0)
 444             real_start += qemu_host_page_size;
 445     }
 446     if (end < real_end) {
 447         prot = 0;
 448         for(addr = end; addr < real_end; addr += TARGET_PAGE_SIZE) {
 449             prot |= page_get_flags(addr);
 450         }
 451         if (prot != 0)
 452             real_end -= qemu_host_page_size;
 453     }
 454
 455     ret = 0;
 456     /* unmap what we can */
 457     if (real_start < real_end) {
 458         ret = munmap(g2h(real_start), real_end - real_start);
 459     }
 460
 461     if (ret == 0)
 462         page_set_flags(start, start + len, 0);
 463     mmap_unlock();
 464     return ret;
 465 }
 466
 467 int target_msync(abi_ulong start, abi_ulong len, int flags)
 468 {
 469     abi_ulong end;
 470
 471     if (start & ~TARGET_PAGE_MASK)
 472         return -EINVAL;
 473     len = TARGET_PAGE_ALIGN(len);
 474     end = start + len;
 475     if (end < start)
 476         return -EINVAL;
 477     if (end == start)
 478         return 0;
 479
 480     start &= qemu_host_page_mask;
 481     return msync(g2h(start), end - start, flags);
 482 }