git.oblomov.eu Git - wine/blob - dlls/ntdll/virtual.c

   1 /*
   2  * Win32 virtual memory functions
   3  *
   4  * Copyright 1997, 2002 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library 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 GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <assert.h>
  25 #include <errno.h>
  26 #include <fcntl.h>
  27 #ifdef HAVE_UNISTD_H
  28 # include <unistd.h>
  29 #endif
  30 #include <stdarg.h>
  31 #include <stdlib.h>
  32 #include <stdio.h>
  33 #include <string.h>
  34 #include <sys/types.h>
  35 #ifdef HAVE_SYS_STAT_H
  36 # include <sys/stat.h>
  37 #endif
  38 #ifdef HAVE_SYS_MMAN_H
  39 # include <sys/mman.h>
  40 #endif
  41 #ifdef HAVE_VALGRIND_VALGRIND_H
  42 # include <valgrind/valgrind.h>
  43 #endif
  44
  45 #define NONAMELESSUNION
  46 #define NONAMELESSSTRUCT
  47 #include "ntstatus.h"
  48 #define WIN32_NO_STATUS
  49 #include "windef.h"
  50 #include "winternl.h"
  51 #include "wine/library.h"
  52 #include "wine/server.h"
  53 #include "wine/exception.h"
  54 #include "wine/list.h"
  55 #include "wine/debug.h"
  56 #include "ntdll_misc.h"
  57
  58 WINE_DEFAULT_DEBUG_CHANNEL(virtual);
  59 WINE_DECLARE_DEBUG_CHANNEL(module);
  60
  61 #ifndef MS_SYNC
  62 #define MS_SYNC 0
  63 #endif
  64
  65 #ifndef MAP_NORESERVE
  66 #define MAP_NORESERVE 0
  67 #endif
  68
  69 /* File view */
  70 struct file_view
  71 {
  72     struct list   entry;       /* Entry in global view list */
  73     void         *base;        /* Base address */
  74     size_t        size;        /* Size in bytes */
  75     HANDLE        mapping;     /* Handle to the file mapping */
  76     unsigned int  map_protect; /* Mapping protection */
  77     unsigned int  protect;     /* Protection for all pages at allocation time */
  78     BYTE          prot[1];     /* Protection byte for each page */
  79 };
  80
  81
  82 /* Conversion from VPROT_* to Win32 flags */
  83 static const BYTE VIRTUAL_Win32Flags[16] =
  84 {
  85     PAGE_NOACCESS,              /* 0 */
  86     PAGE_READONLY,              /* READ */
  87     PAGE_READWRITE,             /* WRITE */
  88     PAGE_READWRITE,             /* READ | WRITE */
  89     PAGE_EXECUTE,               /* EXEC */
  90     PAGE_EXECUTE_READ,          /* READ | EXEC */
  91     PAGE_EXECUTE_READWRITE,     /* WRITE | EXEC */
  92     PAGE_EXECUTE_READWRITE,     /* READ | WRITE | EXEC */
  93     PAGE_WRITECOPY,             /* WRITECOPY */
  94     PAGE_WRITECOPY,             /* READ | WRITECOPY */
  95     PAGE_WRITECOPY,             /* WRITE | WRITECOPY */
  96     PAGE_WRITECOPY,             /* READ | WRITE | WRITECOPY */
  97     PAGE_EXECUTE_WRITECOPY,     /* EXEC | WRITECOPY */
  98     PAGE_EXECUTE_WRITECOPY,     /* READ | EXEC | WRITECOPY */
  99     PAGE_EXECUTE_WRITECOPY,     /* WRITE | EXEC | WRITECOPY */
 100     PAGE_EXECUTE_WRITECOPY      /* READ | WRITE | EXEC | WRITECOPY */
 101 };
 102
 103 static struct list views_list = LIST_INIT(views_list);
 104
 105 static RTL_CRITICAL_SECTION csVirtual;
 106 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
 107 {
 108     0, 0, &csVirtual,
 109     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
 110       0, 0, { (DWORD_PTR)(__FILE__ ": csVirtual") }
 111 };
 112 static RTL_CRITICAL_SECTION csVirtual = { &critsect_debug, -1, 0, 0, 0, 0 };
 113
 114 #ifdef __i386__
 115 static const UINT page_shift = 12;
 116 static const UINT_PTR page_mask = 0xfff;
 117 /* Note: these are Windows limits, you cannot change them. */
 118 static void *address_space_limit = (void *)0xc0000000;  /* top of the total available address space */
 119 static void *user_space_limit    = (void *)0x7fff0000;  /* top of the user address space */
 120 static void *working_set_limit   = (void *)0x7fff0000;  /* top of the current working set */
 121 static void *address_space_start = (void *)0x110000;    /* keep DOS area clear */
 122 #elif defined(__x86_64__)
 123 static const UINT page_shift = 12;
 124 static const UINT_PTR page_mask = 0xfff;
 125 static void *address_space_limit = (void *)0x7fffffff0000;
 126 static void *user_space_limit    = (void *)0x7fffffff0000;
 127 static void *working_set_limit   = (void *)0x7fffffff0000;
 128 static void *address_space_start = (void *)0x10000;
 129 #else
 130 UINT_PTR page_size = 0;
 131 static UINT page_shift;
 132 static UINT_PTR page_mask;
 133 static void *address_space_limit;
 134 static void *user_space_limit;
 135 static void *working_set_limit;
 136 static void *address_space_start = (void *)0x10000;
 137 #endif  /* __i386__ */
 138 static const int is_win64 = (sizeof(void *) > sizeof(int));
 139
 140 #define ROUND_ADDR(addr,mask) \
 141    ((void *)((UINT_PTR)(addr) & ~(UINT_PTR)(mask)))
 142
 143 #define ROUND_SIZE(addr,size) \
 144    (((SIZE_T)(size) + ((UINT_PTR)(addr) & page_mask) + page_mask) & ~page_mask)
 145
 146 #define VIRTUAL_DEBUG_DUMP_VIEW(view) \
 147     do { if (TRACE_ON(virtual)) VIRTUAL_DumpView(view); } while (0)
 148
 149 #define VIRTUAL_HEAP_SIZE (4*1024*1024)
 150
 151 static HANDLE virtual_heap;
 152 static void *preload_reserve_start;
 153 static void *preload_reserve_end;
 154 static int use_locks;
 155 static int force_exec_prot;  /* whether to force PROT_EXEC on all PROT_READ mmaps */
 156
 157
 158 /***********************************************************************
 159  *           VIRTUAL_GetProtStr
 160  */
 161 static const char *VIRTUAL_GetProtStr( BYTE prot )
 162 {
 163     static char buffer[6];
 164     buffer[0] = (prot & VPROT_COMMITTED) ? 'c' : '-';
 165     buffer[1] = (prot & VPROT_GUARD) ? 'g' : ((prot & VPROT_WRITEWATCH) ? 'H' : '-');
 166     buffer[2] = (prot & VPROT_READ) ? 'r' : '-';
 167     buffer[3] = (prot & VPROT_WRITECOPY) ? 'W' : ((prot & VPROT_WRITE) ? 'w' : '-');
 168     buffer[4] = (prot & VPROT_EXEC) ? 'x' : '-';
 169     buffer[5] = 0;
 170     return buffer;
 171 }
 172
 173
 174 /***********************************************************************
 175  *           VIRTUAL_GetUnixProt
 176  *
 177  * Convert page protections to protection for mmap/mprotect.
 178  */
 179 static int VIRTUAL_GetUnixProt( BYTE vprot )
 180 {
 181     int prot = 0;
 182     if ((vprot & VPROT_COMMITTED) && !(vprot & VPROT_GUARD))
 183     {
 184         if (vprot & VPROT_READ) prot |= PROT_READ;
 185         if (vprot & VPROT_WRITE) prot |= PROT_WRITE | PROT_READ;
 186         if (vprot & VPROT_WRITECOPY) prot |= PROT_WRITE | PROT_READ;
 187         if (vprot & VPROT_EXEC) prot |= PROT_EXEC | PROT_READ;
 188         if (vprot & VPROT_WRITEWATCH) prot &= ~PROT_WRITE;
 189     }
 190     if (!prot) prot = PROT_NONE;
 191     return prot;
 192 }
 193
 194
 195 /***********************************************************************
 196  *           VIRTUAL_DumpView
 197  */
 198 static void VIRTUAL_DumpView( struct file_view *view )
 199 {
 200     UINT i, count;
 201     char *addr = view->base;
 202     BYTE prot = view->prot[0];
 203
 204     TRACE( "View: %p - %p", addr, addr + view->size - 1 );
 205     if (view->protect & VPROT_SYSTEM)
 206         TRACE( " (system)\n" );
 207     else if (view->protect & VPROT_VALLOC)
 208         TRACE( " (valloc)\n" );
 209     else if (view->mapping)
 210         TRACE( " %p\n", view->mapping );
 211     else
 212         TRACE( " (anonymous)\n");
 213
 214     for (count = i = 1; i < view->size >> page_shift; i++, count++)
 215     {
 216         if (view->prot[i] == prot) continue;
 217         TRACE( "      %p - %p %s\n",
 218                  addr, addr + (count << page_shift) - 1, VIRTUAL_GetProtStr(prot) );
 219         addr += (count << page_shift);
 220         prot = view->prot[i];
 221         count = 0;
 222     }
 223     if (count)
 224         TRACE( "      %p - %p %s\n",
 225                  addr, addr + (count << page_shift) - 1, VIRTUAL_GetProtStr(prot) );
 226 }
 227
 228
 229 /***********************************************************************
 230  *           VIRTUAL_Dump
 231  */
 232 #ifdef WINE_VM_DEBUG
 233 static void VIRTUAL_Dump(void)
 234 {
 235     sigset_t sigset;
 236     struct file_view *view;
 237
 238     TRACE( "Dump of all virtual memory views:\n" );
 239     server_enter_uninterrupted_section( &csVirtual, &sigset );
 240     LIST_FOR_EACH_ENTRY( view, &views_list, struct file_view, entry )
 241     {
 242         VIRTUAL_DumpView( view );
 243     }
 244     server_leave_uninterrupted_section( &csVirtual, &sigset );
 245 }
 246 #endif
 247
 248
 249 /***********************************************************************
 250  *           VIRTUAL_FindView
 251  *
 252  * Find the view containing a given address. The csVirtual section must be held by caller.
 253  *
 254  * PARAMS
 255  *      addr  [I] Address
 256  *
 257  * RETURNS
 258  *      View: Success
 259  *      NULL: Failure
 260  */
 261 static struct file_view *VIRTUAL_FindView( const void *addr, size_t size )
 262 {
 263     struct file_view *view;
 264
 265     LIST_FOR_EACH_ENTRY( view, &views_list, struct file_view, entry )
 266     {
 267         if (view->base > addr) break;  /* no matching view */
 268         if ((const char *)view->base + view->size <= (const char *)addr) continue;
 269         if ((const char *)view->base + view->size < (const char *)addr + size) break;  /* size too large */
 270         if ((const char *)addr + size < (const char *)addr) break; /* overflow */
 271         return view;
 272     }
 273     return NULL;
 274 }
 275
 276
 277 /***********************************************************************
 278  *           get_mask
 279  */
 280 static inline UINT_PTR get_mask( ULONG zero_bits )
 281 {
 282     if (!zero_bits) return 0xffff;  /* allocations are aligned to 64K by default */
 283     if (zero_bits < page_shift) zero_bits = page_shift;
 284     return (1 << zero_bits) - 1;
 285 }
 286
 287
 288 /***********************************************************************
 289  *           find_view_range
 290  *
 291  * Find the first view overlapping at least part of the specified range.
 292  * The csVirtual section must be held by caller.
 293  */
 294 static struct file_view *find_view_range( const void *addr, size_t size )
 295 {
 296     struct file_view *view;
 297
 298     LIST_FOR_EACH_ENTRY( view, &views_list, struct file_view, entry )
 299     {
 300         if ((const char *)view->base >= (const char *)addr + size) break;
 301         if ((const char *)view->base + view->size > (const char *)addr) return view;
 302     }
 303     return NULL;
 304 }
 305
 306
 307 /***********************************************************************
 308  *           find_free_area
 309  *
 310  * Find a free area between views inside the specified range.
 311  * The csVirtual section must be held by caller.
 312  */
 313 static void *find_free_area( void *base, void *end, size_t size, size_t mask, int top_down )
 314 {
 315     struct list *ptr;
 316     void *start;
 317
 318     if (top_down)
 319     {
 320         start = ROUND_ADDR( (char *)end - size, mask );
 321         if (start >= end || start < base) return NULL;
 322
 323         for (ptr = views_list.prev; ptr != &views_list; ptr = ptr->prev)
 324         {
 325             struct file_view *view = LIST_ENTRY( ptr, struct file_view, entry );
 326
 327             if ((char *)view->base + view->size <= (char *)start) break;
 328             if ((char *)view->base >= (char *)start + size) continue;
 329             start = ROUND_ADDR( (char *)view->base - size, mask );
 330             /* stop if remaining space is not large enough */
 331             if (!start || start >= end || start < base) return NULL;
 332         }
 333     }
 334     else
 335     {
 336         start = ROUND_ADDR( (char *)base + mask, mask );
 337         if (start >= end || (char *)end - (char *)start < size) return NULL;
 338
 339         for (ptr = views_list.next; ptr != &views_list; ptr = ptr->next)
 340         {
 341             struct file_view *view = LIST_ENTRY( ptr, struct file_view, entry );
 342
 343             if ((char *)view->base >= (char *)start + size) break;
 344             if ((char *)view->base + view->size <= (char *)start) continue;
 345             start = ROUND_ADDR( (char *)view->base + view->size + mask, mask );
 346             /* stop if remaining space is not large enough */
 347             if (!start || start >= end || (char *)end - (char *)start < size) return NULL;
 348         }
 349     }
 350     return start;
 351 }
 352
 353
 354 /***********************************************************************
 355  *           add_reserved_area
 356  *
 357  * Add a reserved area to the list maintained by libwine.
 358  * The csVirtual section must be held by caller.
 359  */
 360 static void add_reserved_area( void *addr, size_t size )
 361 {
 362     TRACE( "adding %p-%p\n", addr, (char *)addr + size );
 363
 364     if (addr < user_space_limit)
 365     {
 366         /* unmap the part of the area that is below the limit */
 367         assert( (char *)addr + size > (char *)user_space_limit );
 368         munmap( addr, (char *)user_space_limit - (char *)addr );
 369         size -= (char *)user_space_limit - (char *)addr;
 370         addr = user_space_limit;
 371     }
 372     /* blow away existing mappings */
 373     wine_anon_mmap( addr, size, PROT_NONE, MAP_NORESERVE | MAP_FIXED );
 374     wine_mmap_add_reserved_area( addr, size );
 375 }
 376
 377
 378 /***********************************************************************
 379  *           remove_reserved_area
 380  *
 381  * Remove a reserved area from the list maintained by libwine.
 382  * The csVirtual section must be held by caller.
 383  */
 384 static void remove_reserved_area( void *addr, size_t size )
 385 {
 386     struct file_view *view;
 387
 388     TRACE( "removing %p-%p\n", addr, (char *)addr + size );
 389     wine_mmap_remove_reserved_area( addr, size, 0 );
 390
 391     /* unmap areas not covered by an existing view */
 392     LIST_FOR_EACH_ENTRY( view, &views_list, struct file_view, entry )
 393     {
 394         if ((char *)view->base >= (char *)addr + size)
 395         {
 396             munmap( addr, size );
 397             break;
 398         }
 399         if ((char *)view->base + view->size <= (char *)addr) continue;
 400         if (view->base > addr) munmap( addr, (char *)view->base - (char *)addr );
 401         if ((char *)view->base + view->size > (char *)addr + size) break;
 402         size = (char *)addr + size - ((char *)view->base + view->size);
 403         addr = (char *)view->base + view->size;
 404     }
 405 }
 406
 407
 408 /***********************************************************************
 409  *           is_beyond_limit
 410  *
 411  * Check if an address range goes beyond a given limit.
 412  */
 413 static inline int is_beyond_limit( const void *addr, size_t size, const void *limit )
 414 {
 415     return (addr >= limit || (const char *)addr + size > (const char *)limit);
 416 }
 417
 418
 419 /***********************************************************************
 420  *           unmap_area
 421  *
 422  * Unmap an area, or simply replace it by an empty mapping if it is
 423  * in a reserved area. The csVirtual section must be held by caller.
 424  */
 425 static inline void unmap_area( void *addr, size_t size )
 426 {
 427     if (wine_mmap_is_in_reserved_area( addr, size ))
 428         wine_anon_mmap( addr, size, PROT_NONE, MAP_NORESERVE | MAP_FIXED );
 429     else if (is_beyond_limit( addr, size, user_space_limit ))
 430         add_reserved_area( addr, size );
 431     else
 432         munmap( addr, size );
 433 }
 434
 435
 436 /***********************************************************************
 437  *           delete_view
 438  *
 439  * Deletes a view. The csVirtual section must be held by caller.
 440  */
 441 static void delete_view( struct file_view *view ) /* [in] View */
 442 {
 443     if (!(view->protect & VPROT_SYSTEM)) unmap_area( view->base, view->size );
 444     list_remove( &view->entry );
 445     if (view->mapping) close_handle( view->mapping );
 446     RtlFreeHeap( virtual_heap, 0, view );
 447 }
 448
 449
 450 /***********************************************************************
 451  *           create_view
 452  *
 453  * Create a view. The csVirtual section must be held by caller.
 454  */
 455 static NTSTATUS create_view( struct file_view **view_ret, void *base, size_t size, unsigned int vprot )
 456 {
 457     struct file_view *view;
 458     struct list *ptr;
 459     int unix_prot = VIRTUAL_GetUnixProt( vprot );
 460
 461     assert( !((UINT_PTR)base & page_mask) );
 462     assert( !(size & page_mask) );
 463
 464     /* Create the view structure */
 465
 466     if (!(view = RtlAllocateHeap( virtual_heap, 0, sizeof(*view) + (size >> page_shift) - 1 )))
 467     {
 468         FIXME( "out of memory in virtual heap for %p-%p\n", base, (char *)base + size );
 469         return STATUS_NO_MEMORY;
 470     }
 471
 472     view->base    = base;
 473     view->size    = size;
 474     view->mapping = 0;
 475     view->map_protect = 0;
 476     view->protect = vprot;
 477     memset( view->prot, vprot, size >> page_shift );
 478
 479     /* Insert it in the linked list */
 480
 481     LIST_FOR_EACH( ptr, &views_list )
 482     {
 483         struct file_view *next = LIST_ENTRY( ptr, struct file_view, entry );
 484         if (next->base > base) break;
 485     }
 486     list_add_before( ptr, &view->entry );
 487
 488     /* Check for overlapping views. This can happen if the previous view
 489      * was a system view that got unmapped behind our back. In that case
 490      * we recover by simply deleting it. */
 491
 492     if ((ptr = list_prev( &views_list, &view->entry )) != NULL)
 493     {
 494         struct file_view *prev = LIST_ENTRY( ptr, struct file_view, entry );
 495         if ((char *)prev->base + prev->size > (char *)base)
 496         {
 497             TRACE( "overlapping prev view %p-%p for %p-%p\n",
 498                    prev->base, (char *)prev->base + prev->size,
 499                    base, (char *)base + view->size );
 500             assert( prev->protect & VPROT_SYSTEM );
 501             delete_view( prev );
 502         }
 503     }
 504     if ((ptr = list_next( &views_list, &view->entry )) != NULL)
 505     {
 506         struct file_view *next = LIST_ENTRY( ptr, struct file_view, entry );
 507         if ((char *)base + view->size > (char *)next->base)
 508         {
 509             TRACE( "overlapping next view %p-%p for %p-%p\n",
 510                    next->base, (char *)next->base + next->size,
 511                    base, (char *)base + view->size );
 512             assert( next->protect & VPROT_SYSTEM );
 513             delete_view( next );
 514         }
 515     }
 516
 517     *view_ret = view;
 518     VIRTUAL_DEBUG_DUMP_VIEW( view );
 519
 520     if (force_exec_prot && !(vprot & VPROT_NOEXEC) && (unix_prot & PROT_READ) && !(unix_prot & PROT_EXEC))
 521     {
 522         TRACE( "forcing exec permission on %p-%p\n", base, (char *)base + size - 1 );
 523         mprotect( base, size, unix_prot | PROT_EXEC );
 524     }
 525     return STATUS_SUCCESS;
 526 }
 527
 528
 529 /***********************************************************************
 530  *           VIRTUAL_GetWin32Prot
 531  *
 532  * Convert page protections to Win32 flags.
 533  */
 534 static DWORD VIRTUAL_GetWin32Prot( BYTE vprot )
 535 {
 536     DWORD ret = VIRTUAL_Win32Flags[vprot & 0x0f];
 537     if (vprot & VPROT_NOCACHE) ret |= PAGE_NOCACHE;
 538     if (vprot & VPROT_GUARD) ret |= PAGE_GUARD;
 539     return ret;
 540 }
 541
 542
 543 /***********************************************************************
 544  *           get_vprot_flags
 545  *
 546  * Build page protections from Win32 flags.
 547  *
 548  * PARAMS
 549  *      protect [I] Win32 protection flags
 550  *
 551  * RETURNS
 552  *      Value of page protection flags
 553  */
 554 static NTSTATUS get_vprot_flags( DWORD protect, unsigned int *vprot, BOOL image )
 555 {
 556     switch(protect & 0xff)
 557     {
 558     case PAGE_READONLY:
 559         *vprot = VPROT_READ;
 560         break;
 561     case PAGE_READWRITE:
 562         if (image)
 563             *vprot = VPROT_READ | VPROT_WRITECOPY;
 564         else
 565             *vprot = VPROT_READ | VPROT_WRITE;
 566         break;
 567     case PAGE_WRITECOPY:
 568         *vprot = VPROT_READ | VPROT_WRITECOPY;
 569         break;
 570     case PAGE_EXECUTE:
 571         *vprot = VPROT_EXEC;
 572         break;
 573     case PAGE_EXECUTE_READ:
 574         *vprot = VPROT_EXEC | VPROT_READ;
 575         break;
 576     case PAGE_EXECUTE_READWRITE:
 577         if (image)
 578             *vprot = VPROT_EXEC | VPROT_READ | VPROT_WRITECOPY;
 579         else
 580             *vprot = VPROT_EXEC | VPROT_READ | VPROT_WRITE;
 581         break;
 582     case PAGE_EXECUTE_WRITECOPY:
 583         *vprot = VPROT_EXEC | VPROT_READ | VPROT_WRITECOPY;
 584         break;
 585     case PAGE_NOACCESS:
 586         *vprot = 0;
 587         break;
 588     default:
 589         return STATUS_INVALID_PAGE_PROTECTION;
 590     }
 591     if (protect & PAGE_GUARD) *vprot |= VPROT_GUARD;
 592     if (protect & PAGE_NOCACHE) *vprot |= VPROT_NOCACHE;
 593     return STATUS_SUCCESS;
 594 }
 595
 596
 597 /***********************************************************************
 598  *           VIRTUAL_SetProt
 599  *
 600  * Change the protection of a range of pages.
 601  *
 602  * RETURNS
 603  *      TRUE: Success
 604  *      FALSE: Failure
 605  */
 606 static BOOL VIRTUAL_SetProt( struct file_view *view, /* [in] Pointer to view */
 607                              void *base,      /* [in] Starting address */
 608                              size_t size,     /* [in] Size in bytes */
 609                              BYTE vprot )     /* [in] Protections to use */
 610 {
 611     int unix_prot = VIRTUAL_GetUnixProt(vprot);
 612     BYTE *p = view->prot + (((char *)base - (char *)view->base) >> page_shift);
 613
 614     TRACE("%p-%p %s\n",
 615           base, (char *)base + size - 1, VIRTUAL_GetProtStr( vprot ) );
 616
 617     if (view->protect & VPROT_WRITEWATCH)
 618     {
 619         /* each page may need different protections depending on write watch flag */
 620         UINT i, count;
 621         char *addr = base;
 622         int prot;
 623
 624         p[0] = vprot | (p[0] & VPROT_WRITEWATCH);
 625         unix_prot = VIRTUAL_GetUnixProt( p[0] );
 626         for (count = i = 1; i < size >> page_shift; i++, count++)
 627         {
 628             p[i] = vprot | (p[i] & VPROT_WRITEWATCH);
 629             prot = VIRTUAL_GetUnixProt( p[i] );
 630             if (prot == unix_prot) continue;
 631             mprotect( addr, count << page_shift, unix_prot );
 632             addr += count << page_shift;
 633             unix_prot = prot;
 634             count = 0;
 635         }
 636         if (count) mprotect( addr, count << page_shift, unix_prot );
 637         VIRTUAL_DEBUG_DUMP_VIEW( view );
 638         return TRUE;
 639     }
 640
 641     /* if setting stack guard pages, store the permissions first, as the guard may be
 642      * triggered at any point after mprotect and change the permissions again */
 643     if ((vprot & VPROT_GUARD) &&
 644         (base >= NtCurrentTeb()->DeallocationStack) &&
 645         (base < NtCurrentTeb()->Tib.StackBase))
 646     {
 647         memset( p, vprot, size >> page_shift );
 648         mprotect( base, size, unix_prot );
 649         VIRTUAL_DEBUG_DUMP_VIEW( view );
 650         return TRUE;
 651     }
 652
 653     if (force_exec_prot && !(view->protect & VPROT_NOEXEC) &&
 654         (unix_prot & PROT_READ) && !(unix_prot & PROT_EXEC))
 655     {
 656         TRACE( "forcing exec permission on %p-%p\n", base, (char *)base + size - 1 );
 657         if (!mprotect( base, size, unix_prot | PROT_EXEC )) goto done;
 658         /* exec + write may legitimately fail, in that case fall back to write only */
 659         if (!(unix_prot & PROT_WRITE)) return FALSE;
 660     }
 661
 662     if (mprotect( base, size, unix_prot )) return FALSE;  /* FIXME: last error */
 663
 664 done:
 665     memset( p, vprot, size >> page_shift );
 666     VIRTUAL_DEBUG_DUMP_VIEW( view );
 667     return TRUE;
 668 }
 669
 670
 671 /***********************************************************************
 672  *           reset_write_watches
 673  *
 674  * Reset write watches in a memory range.
 675  */
 676 static void reset_write_watches( struct file_view *view, void *base, SIZE_T size )
 677 {
 678     SIZE_T i, count;
 679     int prot, unix_prot;
 680     char *addr = base;
 681     BYTE *p = view->prot + ((addr - (char *)view->base) >> page_shift);
 682
 683     p[0] |= VPROT_WRITEWATCH;
 684     unix_prot = VIRTUAL_GetUnixProt( p[0] );
 685     for (count = i = 1; i < size >> page_shift; i++, count++)
 686     {
 687         p[i] |= VPROT_WRITEWATCH;
 688         prot = VIRTUAL_GetUnixProt( p[i] );
 689         if (prot == unix_prot) continue;
 690         mprotect( addr, count << page_shift, unix_prot );
 691         addr += count << page_shift;
 692         unix_prot = prot;
 693         count = 0;
 694     }
 695     if (count) mprotect( addr, count << page_shift, unix_prot );
 696 }
 697
 698
 699 /***********************************************************************
 700  *           unmap_extra_space
 701  *
 702  * Release the extra memory while keeping the range starting on the granularity boundary.
 703  */
 704 static inline void *unmap_extra_space( void *ptr, size_t total_size, size_t wanted_size, size_t mask )
 705 {
 706     if ((ULONG_PTR)ptr & mask)
 707     {
 708         size_t extra = mask + 1 - ((ULONG_PTR)ptr & mask);
 709         munmap( ptr, extra );
 710         ptr = (char *)ptr + extra;
 711         total_size -= extra;
 712     }
 713     if (total_size > wanted_size)
 714         munmap( (char *)ptr + wanted_size, total_size - wanted_size );
 715     return ptr;
 716 }
 717
 718
 719 struct alloc_area
 720 {
 721     size_t size;
 722     size_t mask;
 723     int    top_down;
 724     void  *limit;
 725     void  *result;
 726 };
 727
 728 /***********************************************************************
 729  *           alloc_reserved_area_callback
 730  *
 731  * Try to map some space inside a reserved area. Callback for wine_mmap_enum_reserved_areas.
 732  */
 733 static int alloc_reserved_area_callback( void *start, size_t size, void *arg )
 734 {
 735     struct alloc_area *alloc = arg;
 736     void *end = (char *)start + size;
 737
 738     if (start < address_space_start) start = address_space_start;
 739     if (is_beyond_limit( start, size, alloc->limit )) end = alloc->limit;
 740     if (start >= end) return 0;
 741
 742     /* make sure we don't touch the preloader reserved range */
 743     if (preload_reserve_end >= start)
 744     {
 745         if (preload_reserve_end >= end)
 746         {
 747             if (preload_reserve_start <= start) return 0;  /* no space in that area */
 748             if (preload_reserve_start < end) end = preload_reserve_start;
 749         }
 750         else if (preload_reserve_start <= start) start = preload_reserve_end;
 751         else
 752         {
 753             /* range is split in two by the preloader reservation, try first part */
 754             if ((alloc->result = find_free_area( start, preload_reserve_start, alloc->size,
 755                                                  alloc->mask, alloc->top_down )))
 756                 return 1;
 757             /* then fall through to try second part */
 758             start = preload_reserve_end;
 759         }
 760     }
 761     if ((alloc->result = find_free_area( start, end, alloc->size, alloc->mask, alloc->top_down )))
 762         return 1;
 763
 764     return 0;
 765 }
 766
 767
 768 /***********************************************************************
 769  *           map_view
 770  *
 771  * Create a view and mmap the corresponding memory area.
 772  * The csVirtual section must be held by caller.
 773  */
 774 static NTSTATUS map_view( struct file_view **view_ret, void *base, size_t size, size_t mask,
 775                           int top_down, unsigned int vprot )
 776 {
 777     void *ptr;
 778     NTSTATUS status;
 779
 780     if (base)
 781     {
 782         if (is_beyond_limit( base, size, address_space_limit ))
 783             return STATUS_WORKING_SET_LIMIT_RANGE;
 784
 785         switch (wine_mmap_is_in_reserved_area( base, size ))
 786         {
 787         case -1: /* partially in a reserved area */
 788             return STATUS_CONFLICTING_ADDRESSES;
 789
 790         case 0:  /* not in a reserved area, do a normal allocation */
 791             if ((ptr = wine_anon_mmap( base, size, VIRTUAL_GetUnixProt(vprot), 0 )) == (void *)-1)
 792             {
 793                 if (errno == ENOMEM) return STATUS_NO_MEMORY;
 794                 return STATUS_INVALID_PARAMETER;
 795             }
 796             if (ptr != base)
 797             {
 798                 /* We couldn't get the address we wanted */
 799                 if (is_beyond_limit( ptr, size, user_space_limit )) add_reserved_area( ptr, size );
 800                 else munmap( ptr, size );
 801                 return STATUS_CONFLICTING_ADDRESSES;
 802             }
 803             break;
 804
 805         default:
 806         case 1:  /* in a reserved area, make sure the address is available */
 807             if (find_view_range( base, size )) return STATUS_CONFLICTING_ADDRESSES;
 808             /* replace the reserved area by our mapping */
 809             if ((ptr = wine_anon_mmap( base, size, VIRTUAL_GetUnixProt(vprot), MAP_FIXED )) != base)
 810                 return STATUS_INVALID_PARAMETER;
 811             break;
 812         }
 813         if (is_beyond_limit( ptr, size, working_set_limit )) working_set_limit = address_space_limit;
 814     }
 815     else
 816     {
 817         size_t view_size = size + mask + 1;
 818         struct alloc_area alloc;
 819
 820         alloc.size = size;
 821         alloc.mask = mask;
 822         alloc.top_down = top_down;
 823         alloc.limit = user_space_limit;
 824         if (wine_mmap_enum_reserved_areas( alloc_reserved_area_callback, &alloc, top_down ))
 825         {
 826             ptr = alloc.result;
 827             TRACE( "got mem in reserved area %p-%p\n", ptr, (char *)ptr + size );
 828             if (wine_anon_mmap( ptr, size, VIRTUAL_GetUnixProt(vprot), MAP_FIXED ) != ptr)
 829                 return STATUS_INVALID_PARAMETER;
 830             goto done;
 831         }
 832
 833         for (;;)
 834         {
 835             if ((ptr = wine_anon_mmap( NULL, view_size, VIRTUAL_GetUnixProt(vprot), 0 )) == (void *)-1)
 836             {
 837                 if (errno == ENOMEM) return STATUS_NO_MEMORY;
 838                 return STATUS_INVALID_PARAMETER;
 839             }
 840             TRACE( "got mem with anon mmap %p-%p\n", ptr, (char *)ptr + size );
 841             /* if we got something beyond the user limit, unmap it and retry */
 842             if (is_beyond_limit( ptr, view_size, user_space_limit )) add_reserved_area( ptr, view_size );
 843             else break;
 844         }
 845         ptr = unmap_extra_space( ptr, view_size, size, mask );
 846     }
 847 done:
 848     status = create_view( view_ret, ptr, size, vprot );
 849     if (status != STATUS_SUCCESS) unmap_area( ptr, size );
 850     return status;
 851 }
 852
 853
 854 /***********************************************************************
 855  *           map_file_into_view
 856  *
 857  * Wrapper for mmap() to map a file into a view, falling back to read if mmap fails.
 858  * The csVirtual section must be held by caller.
 859  */
 860 static NTSTATUS map_file_into_view( struct file_view *view, int fd, size_t start, size_t size,
 861                                     off_t offset, unsigned int vprot, BOOL removable )
 862 {
 863     void *ptr;
 864     int prot = VIRTUAL_GetUnixProt( vprot | VPROT_COMMITTED /* make sure it is accessible */ );
 865     BOOL shared_write = (vprot & VPROT_WRITE) != 0;
 866
 867     assert( start < view->size );
 868     assert( start + size <= view->size );
 869
 870     if (force_exec_prot && !(vprot & VPROT_NOEXEC) && (vprot & VPROT_READ))
 871     {
 872         TRACE( "forcing exec permission on mapping %p-%p\n",
 873                (char *)view->base + start, (char *)view->base + start + size - 1 );
 874         prot |= PROT_EXEC;
 875     }
 876
 877     /* only try mmap if media is not removable (or if we require write access) */
 878     if (!removable || shared_write)
 879     {
 880         int flags = MAP_FIXED | (shared_write ? MAP_SHARED : MAP_PRIVATE);
 881
 882         if (mmap( (char *)view->base + start, size, prot, flags, fd, offset ) != (void *)-1)
 883             goto done;
 884
 885         if ((errno == EPERM) && (prot & PROT_EXEC))
 886             ERR( "failed to set %08x protection on file map, noexec filesystem?\n", prot );
 887
 888         /* mmap() failed; if this is because the file offset is not    */
 889         /* page-aligned (EINVAL), or because the underlying filesystem */
 890         /* does not support mmap() (ENOEXEC,ENODEV), we do it by hand. */
 891         if ((errno != ENOEXEC) && (errno != EINVAL) && (errno != ENODEV)) return FILE_GetNtStatus();
 892         if (shared_write)  /* we cannot fake shared write mappings */
 893         {
 894             if (errno == EINVAL) return STATUS_INVALID_PARAMETER;
 895             ERR( "shared writable mmap not supported, broken filesystem?\n" );
 896             return STATUS_NOT_SUPPORTED;
 897         }
 898     }
 899
 900     /* Reserve the memory with an anonymous mmap */
 901     ptr = wine_anon_mmap( (char *)view->base + start, size, PROT_READ | PROT_WRITE, MAP_FIXED );
 902     if (ptr == (void *)-1) return FILE_GetNtStatus();
 903     /* Now read in the file */
 904     pread( fd, ptr, size, offset );
 905     if (prot != (PROT_READ|PROT_WRITE)) mprotect( ptr, size, prot );  /* Set the right protection */
 906 done:
 907     memset( view->prot + (start >> page_shift), vprot, ROUND_SIZE(start,size) >> page_shift );
 908     return STATUS_SUCCESS;
 909 }
 910
 911
 912 /***********************************************************************
 913  *           get_committed_size
 914  *
 915  * Get the size of the committed range starting at base.
 916  * Also return the protections for the first page.
 917  */
 918 static SIZE_T get_committed_size( struct file_view *view, void *base, BYTE *vprot )
 919 {
 920     SIZE_T i, start;
 921
 922     start = ((char *)base - (char *)view->base) >> page_shift;
 923     *vprot = view->prot[start];
 924
 925     if (view->mapping && !(view->protect & VPROT_COMMITTED))
 926     {
 927         SIZE_T ret = 0;
 928         SERVER_START_REQ( get_mapping_committed_range )
 929         {
 930             req->handle = wine_server_obj_handle( view->mapping );
 931             req->offset = start << page_shift;
 932             if (!wine_server_call( req ))
 933             {
 934                 ret = reply->size;
 935                 if (reply->committed)
 936                 {
 937                     *vprot |= VPROT_COMMITTED;
 938                     for (i = 0; i < ret >> page_shift; i++) view->prot[start+i] |= VPROT_COMMITTED;
 939                 }
 940             }
 941         }
 942         SERVER_END_REQ;
 943         return ret;
 944     }
 945     for (i = start + 1; i < view->size >> page_shift; i++)
 946         if ((*vprot ^ view->prot[i]) & VPROT_COMMITTED) break;
 947     return (i - start) << page_shift;
 948 }
 949
 950
 951 /***********************************************************************
 952  *           decommit_view
 953  *
 954  * Decommit some pages of a given view.
 955  * The csVirtual section must be held by caller.
 956  */
 957 static NTSTATUS decommit_pages( struct file_view *view, size_t start, size_t size )
 958 {
 959     if (wine_anon_mmap( (char *)view->base + start, size, PROT_NONE, MAP_FIXED ) != (void *)-1)
 960     {
 961         BYTE *p = view->prot + (start >> page_shift);
 962         size >>= page_shift;
 963         while (size--) *p++ &= ~VPROT_COMMITTED;
 964         return STATUS_SUCCESS;
 965     }
 966     return FILE_GetNtStatus();
 967 }
 968
 969
 970 /***********************************************************************
 971  *           allocate_dos_memory
 972  *
 973  * Allocate the DOS memory range.
 974  */
 975 static NTSTATUS allocate_dos_memory( struct file_view **view, unsigned int vprot )
 976 {
 977     size_t size;
 978     void *addr = NULL;
 979     void * const low_64k = (void *)0x10000;
 980     const size_t dosmem_size = 0x110000;
 981     int unix_prot = VIRTUAL_GetUnixProt( vprot );
 982     struct list *ptr;
 983
 984     /* check for existing view */
 985
 986     if ((ptr = list_head( &views_list )))
 987     {
 988         struct file_view *first_view = LIST_ENTRY( ptr, struct file_view, entry );
 989         if (first_view->base < (void *)dosmem_size) return STATUS_CONFLICTING_ADDRESSES;
 990     }
 991
 992     /* check without the first 64K */
 993
 994     if (wine_mmap_is_in_reserved_area( low_64k, dosmem_size - 0x10000 ) != 1)
 995     {
 996         addr = wine_anon_mmap( low_64k, dosmem_size - 0x10000, unix_prot, 0 );
 997         if (addr != low_64k)
 998         {
 999             if (addr != (void *)-1) munmap( addr, dosmem_size - 0x10000 );
1000             return map_view( view, NULL, dosmem_size, 0xffff, 0, vprot );
1001         }
1002     }
1003
1004     /* now try to allocate the low 64K too */
1005
1006     if (wine_mmap_is_in_reserved_area( NULL, 0x10000 ) != 1)
1007     {
1008         addr = wine_anon_mmap( (void *)page_size, 0x10000 - page_size, unix_prot, 0 );
1009         if (addr == (void *)page_size)
1010         {
1011             if (!wine_anon_mmap( NULL, page_size, unix_prot, MAP_FIXED ))
1012             {
1013                 addr = NULL;
1014                 TRACE( "successfully mapped low 64K range\n" );
1015             }
1016             else TRACE( "failed to map page 0\n" );
1017         }
1018         else
1019         {
1020             if (addr != (void *)-1) munmap( addr, 0x10000 - page_size );
1021             addr = low_64k;
1022             TRACE( "failed to map low 64K range\n" );
1023         }
1024     }
1025
1026     /* now reserve the whole range */
1027
1028     size = (char *)dosmem_size - (char *)addr;
1029     wine_anon_mmap( addr, size, unix_prot, MAP_FIXED );
1030     return create_view( view, addr, size, vprot );
1031 }
1032
1033
1034 /***********************************************************************
1035  *           check_architecture
1036  *
1037  * Check the architecture of a PE binary.
1038  */
1039 static NTSTATUS check_architecture( const IMAGE_NT_HEADERS *nt )
1040 {
1041     static const char *arch;
1042
1043 #ifdef __i386__
1044     if (nt->FileHeader.Machine == IMAGE_FILE_MACHINE_I386) return STATUS_SUCCESS;
1045     if (nt->FileHeader.Machine == IMAGE_FILE_MACHINE_AMD64)
1046     {
1047         if (nt->FileHeader.Characteristics & IMAGE_FILE_DLL)  /* don't warn for a 64-bit exe */
1048             WARN( "loading amd64 dll in 32-bit mode will fail\n" );
1049         return STATUS_INVALID_IMAGE_FORMAT;
1050     }
1051 #elif defined(__x86_64__)
1052     if (nt->FileHeader.Machine == IMAGE_FILE_MACHINE_AMD64) return STATUS_SUCCESS;
1053     if (nt->FileHeader.Machine == IMAGE_FILE_MACHINE_I386)
1054     {
1055         if (nt->FileHeader.Characteristics & IMAGE_FILE_DLL)  /* don't warn for a 32-bit exe */
1056             WARN( "loading 32-bit dll in 64-bit mode will fail\n" );
1057         return STATUS_INVALID_IMAGE_FORMAT;
1058     }
1059 #elif defined(__arm__) && !defined(__ARMEB__)
1060     if (nt->FileHeader.Machine == IMAGE_FILE_MACHINE_ARM ||
1061         nt->FileHeader.Machine == IMAGE_FILE_MACHINE_THUMB)
1062         return STATUS_SUCCESS;
1063     if (nt->FileHeader.Machine == IMAGE_FILE_MACHINE_ARMNT)
1064     {
1065         SYSTEM_CPU_INFORMATION sci;
1066         if (SUCCEEDED(NtQuerySystemInformation( SystemCpuInformation, &sci, sizeof(sci), NULL )) &&
1067             sci.Architecture == PROCESSOR_ARCHITECTURE_ARM && sci.Level >= 7)
1068             return STATUS_SUCCESS;
1069     }
1070 #endif
1071
1072     switch (nt->FileHeader.Machine)
1073     {
1074         case IMAGE_FILE_MACHINE_UNKNOWN: arch = "Unknown"; break;
1075         case IMAGE_FILE_MACHINE_I860:    arch = "I860"; break;
1076         case IMAGE_FILE_MACHINE_I386:    arch = "I386"; break;
1077         case IMAGE_FILE_MACHINE_R3000:   arch = "R3000"; break;
1078         case IMAGE_FILE_MACHINE_R4000:   arch = "R4000"; break;
1079         case IMAGE_FILE_MACHINE_R10000:  arch = "R10000"; break;
1080         case IMAGE_FILE_MACHINE_ALPHA:   arch = "Alpha"; break;
1081         case IMAGE_FILE_MACHINE_POWERPC: arch = "PowerPC"; break;
1082         case IMAGE_FILE_MACHINE_IA64:    arch = "IA-64"; break;
1083         case IMAGE_FILE_MACHINE_ALPHA64: arch = "Alpha-64"; break;
1084         case IMAGE_FILE_MACHINE_AMD64:   arch = "AMD-64"; break;
1085         case IMAGE_FILE_MACHINE_ARM:     arch = "ARM"; break;
1086         case IMAGE_FILE_MACHINE_ARMNT:   arch = "ARMNT"; break;
1087         case IMAGE_FILE_MACHINE_THUMB:   arch = "ARM Thumb"; break;
1088         case IMAGE_FILE_MACHINE_SPARC:   arch = "SPARC"; break;
1089         default: arch = wine_dbg_sprintf( "Unknown-%04x", nt->FileHeader.Machine ); break;
1090     }
1091     ERR( "Trying to load PE image for unsupported architecture %s\n", arch );
1092     return STATUS_INVALID_IMAGE_FORMAT;
1093 }
1094
1095
1096 /***********************************************************************
1097  *           stat_mapping_file
1098  *
1099  * Stat the underlying file for a memory view.
1100  */
1101 static NTSTATUS stat_mapping_file( struct file_view *view, struct stat *st )
1102 {
1103     NTSTATUS status;
1104     int unix_fd, needs_close;
1105
1106     if (!view->mapping) return STATUS_NOT_MAPPED_VIEW;
1107     if (!(status = server_get_unix_fd( view->mapping, 0, &unix_fd, &needs_close, NULL, NULL )))
1108     {
1109         if (fstat( unix_fd, st ) == -1) status = FILE_GetNtStatus();
1110         if (needs_close) close( unix_fd );
1111     }
1112     return status;
1113 }
1114
1115
1116 /***********************************************************************
1117  *           map_image
1118  *
1119  * Map an executable (PE format) image into memory.
1120  */
1121 static NTSTATUS map_image( HANDLE hmapping, int fd, char *base, SIZE_T total_size, SIZE_T mask,
1122                            SIZE_T header_size, int shared_fd, HANDLE dup_mapping, unsigned int map_vprot, PVOID *addr_ptr )
1123 {
1124     IMAGE_DOS_HEADER *dos;
1125     IMAGE_NT_HEADERS *nt;
1126     IMAGE_SECTION_HEADER sections[96];
1127     IMAGE_SECTION_HEADER *sec;
1128     IMAGE_DATA_DIRECTORY *imports;
1129     NTSTATUS status = STATUS_CONFLICTING_ADDRESSES;
1130     int i;
1131     off_t pos;
1132     sigset_t sigset;
1133     struct stat st;
1134     struct file_view *view = NULL;
1135     char *ptr, *header_end, *header_start;
1136     INT_PTR delta = 0;
1137
1138     /* zero-map the whole range */
1139
1140     server_enter_uninterrupted_section( &csVirtual, &sigset );
1141
1142     if (base >= (char *)address_space_start)  /* make sure the DOS area remains free */
1143         status = map_view( &view, base, total_size, mask, FALSE,
1144                            VPROT_COMMITTED | VPROT_READ | VPROT_EXEC | VPROT_WRITECOPY | VPROT_IMAGE );
1145
1146     if (status != STATUS_SUCCESS)
1147         status = map_view( &view, NULL, total_size, mask, FALSE,
1148                            VPROT_COMMITTED | VPROT_READ | VPROT_EXEC | VPROT_WRITECOPY | VPROT_IMAGE );
1149
1150     if (status != STATUS_SUCCESS) goto error;
1151
1152     ptr = view->base;
1153     TRACE_(module)( "mapped PE file at %p-%p\n", ptr, ptr + total_size );
1154
1155     /* map the header */
1156
1157     if (fstat( fd, &st ) == -1)
1158     {
1159         status = FILE_GetNtStatus();
1160         goto error;
1161     }
1162     status = STATUS_INVALID_IMAGE_FORMAT;  /* generic error */
1163     if (!st.st_size) goto error;
1164     header_size = min( header_size, st.st_size );
1165     if (map_file_into_view( view, fd, 0, header_size, 0, VPROT_COMMITTED | VPROT_READ | VPROT_WRITECOPY,
1166                             !dup_mapping ) != STATUS_SUCCESS) goto error;
1167     dos = (IMAGE_DOS_HEADER *)ptr;
1168     nt = (IMAGE_NT_HEADERS *)(ptr + dos->e_lfanew);
1169     header_end = ptr + ROUND_SIZE( 0, header_size );
1170     memset( ptr + header_size, 0, header_end - (ptr + header_size) );
1171     if ((char *)(nt + 1) > header_end) goto error;
1172     header_start = (char*)&nt->OptionalHeader+nt->FileHeader.SizeOfOptionalHeader;
1173     if (nt->FileHeader.NumberOfSections > sizeof(sections)/sizeof(*sections)) goto error;
1174     if (header_start + sizeof(*sections) * nt->FileHeader.NumberOfSections > header_end) goto error;
1175     /* Some applications (e.g. the Steam version of Borderlands) map over the top of the section headers,
1176      * copying the headers into local memory is necessary to properly load such applications. */
1177     memcpy(sections, header_start, sizeof(*sections) * nt->FileHeader.NumberOfSections);
1178     sec = sections;
1179
1180     imports = nt->OptionalHeader.DataDirectory + IMAGE_DIRECTORY_ENTRY_IMPORT;
1181     if (!imports->Size || !imports->VirtualAddress) imports = NULL;
1182
1183     if (check_architecture( nt )) goto error;
1184
1185     /* check for non page-aligned binary */
1186
1187     if (nt->OptionalHeader.SectionAlignment <= page_mask)
1188     {
1189         /* unaligned sections, this happens for native subsystem binaries */
1190         /* in that case Windows simply maps in the whole file */
1191
1192         if (map_file_into_view( view, fd, 0, total_size, 0, VPROT_COMMITTED | VPROT_READ,
1193                                 !dup_mapping ) != STATUS_SUCCESS) goto error;
1194
1195         /* check that all sections are loaded at the right offset */
1196         if (nt->OptionalHeader.FileAlignment != nt->OptionalHeader.SectionAlignment) goto error;
1197         for (i = 0; i < nt->FileHeader.NumberOfSections; i++)
1198         {
1199             if (sec[i].VirtualAddress != sec[i].PointerToRawData)
1200                 goto error;  /* Windows refuses to load in that case too */
1201         }
1202
1203         /* set the image protections */
1204         VIRTUAL_SetProt( view, ptr, total_size,
1205                          VPROT_COMMITTED | VPROT_READ | VPROT_WRITECOPY | VPROT_EXEC );
1206
1207         /* no relocations are performed on non page-aligned binaries */
1208         goto done;
1209     }
1210
1211
1212     /* map all the sections */
1213
1214     for (i = pos = 0; i < nt->FileHeader.NumberOfSections; i++, sec++)
1215     {
1216         static const SIZE_T sector_align = 0x1ff;
1217         SIZE_T map_size, file_start, file_size, end;
1218
1219         if (!sec->Misc.VirtualSize)
1220             map_size = ROUND_SIZE( 0, sec->SizeOfRawData );
1221         else
1222             map_size = ROUND_SIZE( 0, sec->Misc.VirtualSize );
1223
1224         /* file positions are rounded to sector boundaries regardless of OptionalHeader.FileAlignment */
1225         file_start = sec->PointerToRawData & ~sector_align;
1226         file_size = (sec->SizeOfRawData + (sec->PointerToRawData & sector_align) + sector_align) & ~sector_align;
1227         if (file_size > map_size) file_size = map_size;
1228
1229         /* a few sanity checks */
1230         end = sec->VirtualAddress + ROUND_SIZE( sec->VirtualAddress, map_size );
1231         if (sec->VirtualAddress > total_size || end > total_size || end < sec->VirtualAddress)
1232         {
1233             WARN_(module)( "Section %.8s too large (%x+%lx/%lx)\n",
1234                            sec->Name, sec->VirtualAddress, map_size, total_size );
1235             goto error;
1236         }
1237
1238         if ((sec->Characteristics & IMAGE_SCN_MEM_SHARED) &&
1239             (sec->Characteristics & IMAGE_SCN_MEM_WRITE))
1240         {
1241             TRACE_(module)( "mapping shared section %.8s at %p off %x (%x) size %lx (%lx) flags %x\n",
1242                             sec->Name, ptr + sec->VirtualAddress,
1243                             sec->PointerToRawData, (int)pos, file_size, map_size,
1244                             sec->Characteristics );
1245             if (map_file_into_view( view, shared_fd, sec->VirtualAddress, map_size, pos,
1246                                     VPROT_COMMITTED | VPROT_READ | VPROT_WRITE,
1247                                     FALSE ) != STATUS_SUCCESS)
1248             {
1249                 ERR_(module)( "Could not map shared section %.8s\n", sec->Name );
1250                 goto error;
1251             }
1252
1253             /* check if the import directory falls inside this section */
1254             if (imports && imports->VirtualAddress >= sec->VirtualAddress &&
1255                 imports->VirtualAddress < sec->VirtualAddress + map_size)
1256             {
1257                 UINT_PTR base = imports->VirtualAddress & ~page_mask;
1258                 UINT_PTR end = base + ROUND_SIZE( imports->VirtualAddress, imports->Size );
1259                 if (end > sec->VirtualAddress + map_size) end = sec->VirtualAddress + map_size;
1260                 if (end > base)
1261                     map_file_into_view( view, shared_fd, base, end - base,
1262                                         pos + (base - sec->VirtualAddress),
1263                                         VPROT_COMMITTED | VPROT_READ | VPROT_WRITECOPY,
1264                                         FALSE );
1265             }
1266             pos += map_size;
1267             continue;
1268         }
1269
1270         TRACE_(module)( "mapping section %.8s at %p off %x size %x virt %x flags %x\n",
1271                         sec->Name, ptr + sec->VirtualAddress,
1272                         sec->PointerToRawData, sec->SizeOfRawData,
1273                         sec->Misc.VirtualSize, sec->Characteristics );
1274
1275         if (!sec->PointerToRawData || !file_size) continue;
1276
1277         /* Note: if the section is not aligned properly map_file_into_view will magically
1278          *       fall back to read(), so we don't need to check anything here.
1279          */
1280         end = file_start + file_size;
1281         if (sec->PointerToRawData >= st.st_size ||
1282             end > ((st.st_size + sector_align) & ~sector_align) ||
1283             end < file_start ||
1284             map_file_into_view( view, fd, sec->VirtualAddress, file_size, file_start,
1285                                 VPROT_COMMITTED | VPROT_READ | VPROT_WRITECOPY,
1286                                 !dup_mapping ) != STATUS_SUCCESS)
1287         {
1288             ERR_(module)( "Could not map section %.8s, file probably truncated\n", sec->Name );
1289             goto error;
1290         }
1291
1292         if (file_size & page_mask)
1293         {
1294             end = ROUND_SIZE( 0, file_size );
1295             if (end > map_size) end = map_size;
1296             TRACE_(module)("clearing %p - %p\n",
1297                            ptr + sec->VirtualAddress + file_size,
1298                            ptr + sec->VirtualAddress + end );
1299             memset( ptr + sec->VirtualAddress + file_size, 0, end - file_size );
1300         }
1301     }
1302
1303
1304     /* perform base relocation, if necessary */
1305
1306     if (ptr != base &&
1307         ((nt->FileHeader.Characteristics & IMAGE_FILE_DLL) ||
1308           !NtCurrentTeb()->Peb->ImageBaseAddress) )
1309     {
1310         IMAGE_BASE_RELOCATION *rel, *end;
1311         const IMAGE_DATA_DIRECTORY *relocs;
1312
1313         if (nt->FileHeader.Characteristics & IMAGE_FILE_RELOCS_STRIPPED)
1314         {
1315             WARN_(module)( "Need to relocate module from %p to %p, but there are no relocation records\n",
1316                            base, ptr );
1317             status = STATUS_CONFLICTING_ADDRESSES;
1318             goto error;
1319         }
1320
1321         TRACE_(module)( "relocating from %p-%p to %p-%p\n",
1322                         base, base + total_size, ptr, ptr + total_size );
1323
1324         relocs = &nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC];
1325         rel = (IMAGE_BASE_RELOCATION *)(ptr + relocs->VirtualAddress);
1326         end = (IMAGE_BASE_RELOCATION *)(ptr + relocs->VirtualAddress + relocs->Size);
1327         delta = ptr - base;
1328
1329         while (rel < end - 1 && rel->SizeOfBlock)
1330         {
1331             if (rel->VirtualAddress >= total_size)
1332             {
1333                 WARN_(module)( "invalid address %p in relocation %p\n", ptr + rel->VirtualAddress, rel );
1334                 status = STATUS_ACCESS_VIOLATION;
1335                 goto error;
1336             }
1337             rel = LdrProcessRelocationBlock( ptr + rel->VirtualAddress,
1338                                              (rel->SizeOfBlock - sizeof(*rel)) / sizeof(USHORT),
1339                                              (USHORT *)(rel + 1), delta );
1340             if (!rel) goto error;
1341         }
1342     }
1343
1344     /* set the image protections */
1345
1346     VIRTUAL_SetProt( view, ptr, ROUND_SIZE( 0, header_size ), VPROT_COMMITTED | VPROT_READ );
1347
1348     sec = sections;
1349     for (i = 0; i < nt->FileHeader.NumberOfSections; i++, sec++)
1350     {
1351         SIZE_T size;
1352         BYTE vprot = VPROT_COMMITTED;
1353
1354         if (sec->Misc.VirtualSize)
1355             size = ROUND_SIZE( sec->VirtualAddress, sec->Misc.VirtualSize );
1356         else
1357             size = ROUND_SIZE( sec->VirtualAddress, sec->SizeOfRawData );
1358
1359         if (sec->Characteristics & IMAGE_SCN_MEM_READ)    vprot |= VPROT_READ;
1360         if (sec->Characteristics & IMAGE_SCN_MEM_WRITE)   vprot |= VPROT_WRITECOPY;
1361         if (sec->Characteristics & IMAGE_SCN_MEM_EXECUTE) vprot |= VPROT_EXEC;
1362
1363         /* Dumb game crack lets the AOEP point into a data section. Adjust. */
1364         if ((nt->OptionalHeader.AddressOfEntryPoint >= sec->VirtualAddress) &&
1365             (nt->OptionalHeader.AddressOfEntryPoint < sec->VirtualAddress + size))
1366             vprot |= VPROT_EXEC;
1367
1368         if (!VIRTUAL_SetProt( view, ptr + sec->VirtualAddress, size, vprot ) && (vprot & VPROT_EXEC))
1369             ERR( "failed to set %08x protection on section %.8s, noexec filesystem?\n",
1370                  sec->Characteristics, sec->Name );
1371     }
1372
1373  done:
1374     view->mapping = dup_mapping;
1375     view->map_protect = map_vprot;
1376     server_leave_uninterrupted_section( &csVirtual, &sigset );
1377
1378     *addr_ptr = ptr;
1379 #ifdef VALGRIND_LOAD_PDB_DEBUGINFO
1380     VALGRIND_LOAD_PDB_DEBUGINFO(fd, ptr, total_size, delta);
1381 #endif
1382     if (ptr != base) return STATUS_IMAGE_NOT_AT_BASE;
1383     return STATUS_SUCCESS;
1384
1385  error:
1386     if (view) delete_view( view );
1387     server_leave_uninterrupted_section( &csVirtual, &sigset );
1388     if (dup_mapping) NtClose( dup_mapping );
1389     return status;
1390 }
1391
1392
1393 /* callback for wine_mmap_enum_reserved_areas to allocate space for the virtual heap */
1394 static int alloc_virtual_heap( void *base, size_t size, void *arg )
1395 {
1396     void **heap_base = arg;
1397
1398     if (is_beyond_limit( base, size, address_space_limit )) address_space_limit = (char *)base + size;
1399     if (size < VIRTUAL_HEAP_SIZE) return 0;
1400     if (is_win64 && base < (void *)0x80000000) return 0;
1401     *heap_base = wine_anon_mmap( (char *)base + size - VIRTUAL_HEAP_SIZE,
1402                                  VIRTUAL_HEAP_SIZE, PROT_READ|PROT_WRITE, MAP_FIXED );
1403     return (*heap_base != (void *)-1);
1404 }
1405
1406 /***********************************************************************
1407  *           virtual_init
1408  */
1409 void virtual_init(void)
1410 {
1411     const char *preload;
1412     void *heap_base;
1413     size_t size;
1414     struct file_view *heap_view;
1415
1416 #if !defined(__i386__) && !defined(__x86_64__)
1417     page_size = sysconf( _SC_PAGESIZE );
1418     page_mask = page_size - 1;
1419     /* Make sure we have a power of 2 */
1420     assert( !(page_size & page_mask) );
1421     page_shift = 0;
1422     while ((1 << page_shift) != page_size) page_shift++;
1423     user_space_limit = working_set_limit = address_space_limit = (void *)~page_mask;
1424 #endif  /* page_mask */
1425     if ((preload = getenv("WINEPRELOADRESERVE")))
1426     {
1427         unsigned long start, end;
1428         if (sscanf( preload, "%lx-%lx", &start, &end ) == 2)
1429         {
1430             preload_reserve_start = (void *)start;
1431             preload_reserve_end = (void *)end;
1432         }
1433     }
1434
1435     /* try to find space in a reserved area for the virtual heap */
1436     if (!wine_mmap_enum_reserved_areas( alloc_virtual_heap, &heap_base, 1 ))
1437         heap_base = wine_anon_mmap( NULL, VIRTUAL_HEAP_SIZE, PROT_READ|PROT_WRITE, 0 );
1438
1439     assert( heap_base != (void *)-1 );
1440     virtual_heap = RtlCreateHeap( HEAP_NO_SERIALIZE, heap_base, VIRTUAL_HEAP_SIZE,
1441                                   VIRTUAL_HEAP_SIZE, NULL, NULL );
1442     create_view( &heap_view, heap_base, VIRTUAL_HEAP_SIZE, VPROT_COMMITTED | VPROT_READ | VPROT_WRITE );
1443
1444     /* make the DOS area accessible (except the low 64K) to hide bugs in broken apps like Excel 2003 */
1445     size = (char *)address_space_start - (char *)0x10000;
1446     if (size && wine_mmap_is_in_reserved_area( (void*)0x10000, size ) == 1)
1447         wine_anon_mmap( (void *)0x10000, size, PROT_READ | PROT_WRITE, MAP_FIXED );
1448 }
1449
1450
1451 /***********************************************************************
1452  *           virtual_init_threading
1453  */
1454 void virtual_init_threading(void)
1455 {
1456     use_locks = 1;
1457 }
1458
1459
1460 /***********************************************************************
1461  *           virtual_get_system_info
1462  */
1463 void virtual_get_system_info( SYSTEM_BASIC_INFORMATION *info )
1464 {
1465     info->unknown                 = 0;
1466     info->KeMaximumIncrement      = 0;  /* FIXME */
1467     info->PageSize                = page_size;
1468     info->MmLowestPhysicalPage    = 1;
1469     info->MmHighestPhysicalPage   = 0x7fffffff / page_size;
1470     info->MmNumberOfPhysicalPages = info->MmHighestPhysicalPage - info->MmLowestPhysicalPage;
1471     info->AllocationGranularity   = get_mask(0) + 1;
1472     info->LowestUserAddress       = (void *)0x10000;
1473     info->HighestUserAddress      = (char *)user_space_limit - 1;
1474     info->ActiveProcessorsAffinityMask = (1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
1475     info->NumberOfProcessors      = NtCurrentTeb()->Peb->NumberOfProcessors;
1476 }
1477
1478
1479 /***********************************************************************
1480  *           virtual_create_builtin_view
1481  */
1482 NTSTATUS virtual_create_builtin_view( void *module )
1483 {
1484     NTSTATUS status;
1485     sigset_t sigset;
1486     IMAGE_NT_HEADERS *nt = RtlImageNtHeader( module );
1487     SIZE_T size = nt->OptionalHeader.SizeOfImage;
1488     IMAGE_SECTION_HEADER *sec;
1489     struct file_view *view;
1490     void *base;
1491     int i;
1492
1493     size = ROUND_SIZE( module, size );
1494     base = ROUND_ADDR( module, page_mask );
1495     server_enter_uninterrupted_section( &csVirtual, &sigset );
1496     status = create_view( &view, base, size, VPROT_SYSTEM | VPROT_IMAGE |
1497                           VPROT_COMMITTED | VPROT_READ | VPROT_WRITECOPY | VPROT_EXEC );
1498     if (!status) TRACE( "created %p-%p\n", base, (char *)base + size );
1499     server_leave_uninterrupted_section( &csVirtual, &sigset );
1500
1501     if (status) return status;
1502
1503     /* The PE header is always read-only, no write, no execute. */
1504     view->prot[0] = VPROT_COMMITTED | VPROT_READ;
1505
1506     sec = (IMAGE_SECTION_HEADER *)((char *)&nt->OptionalHeader + nt->FileHeader.SizeOfOptionalHeader);
1507     for (i = 0; i < nt->FileHeader.NumberOfSections; i++)
1508     {
1509         BYTE flags = VPROT_COMMITTED;
1510
1511         if (sec[i].Characteristics & IMAGE_SCN_MEM_EXECUTE) flags |= VPROT_EXEC;
1512         if (sec[i].Characteristics & IMAGE_SCN_MEM_READ) flags |= VPROT_READ;
1513         if (sec[i].Characteristics & IMAGE_SCN_MEM_WRITE) flags |= VPROT_WRITE;
1514         memset (view->prot + (sec[i].VirtualAddress >> page_shift), flags,
1515                 ROUND_SIZE( sec[i].VirtualAddress, sec[i].Misc.VirtualSize ) >> page_shift );
1516     }
1517
1518     return status;
1519 }
1520
1521
1522 /***********************************************************************
1523  *           virtual_alloc_thread_stack
1524  */
1525 NTSTATUS virtual_alloc_thread_stack( TEB *teb, SIZE_T reserve_size, SIZE_T commit_size )
1526 {
1527     struct file_view *view;
1528     NTSTATUS status;
1529     sigset_t sigset;
1530     SIZE_T size;
1531
1532     if (!reserve_size || !commit_size)
1533     {
1534         IMAGE_NT_HEADERS *nt = RtlImageNtHeader( NtCurrentTeb()->Peb->ImageBaseAddress );
1535         if (!reserve_size) reserve_size = nt->OptionalHeader.SizeOfStackReserve;
1536         if (!commit_size) commit_size = nt->OptionalHeader.SizeOfStackCommit;
1537     }
1538
1539     size = max( reserve_size, commit_size );
1540     if (size < 1024 * 1024) size = 1024 * 1024;  /* Xlib needs a large stack */
1541     size = (size + 0xffff) & ~0xffff;  /* round to 64K boundary */
1542
1543     server_enter_uninterrupted_section( &csVirtual, &sigset );
1544
1545     if ((status = map_view( &view, NULL, size, 0xffff, 0,
1546                             VPROT_READ | VPROT_WRITE | VPROT_COMMITTED | VPROT_VALLOC )) != STATUS_SUCCESS)
1547         goto done;
1548
1549 #ifdef VALGRIND_STACK_REGISTER
1550     VALGRIND_STACK_REGISTER( view->base, (char *)view->base + view->size );
1551 #endif
1552
1553     /* setup no access guard page */
1554     VIRTUAL_SetProt( view, view->base, page_size, VPROT_COMMITTED );
1555     VIRTUAL_SetProt( view, (char *)view->base + page_size, page_size,
1556                      VPROT_READ | VPROT_WRITE | VPROT_COMMITTED | VPROT_GUARD );
1557
1558     /* note: limit is lower than base since the stack grows down */
1559     teb->DeallocationStack = view->base;
1560     teb->Tib.StackBase     = (char *)view->base + view->size;
1561     teb->Tib.StackLimit    = (char *)view->base + 2 * page_size;
1562 done:
1563     server_leave_uninterrupted_section( &csVirtual, &sigset );
1564     return status;
1565 }
1566
1567
1568 /***********************************************************************
1569  *           virtual_clear_thread_stack
1570  *
1571  * Clear the stack contents before calling the main entry point, some broken apps need that.
1572  */
1573 void virtual_clear_thread_stack(void)
1574 {
1575     void *stack = NtCurrentTeb()->Tib.StackLimit;
1576     size_t size = (char *)NtCurrentTeb()->Tib.StackBase - (char *)NtCurrentTeb()->Tib.StackLimit;
1577
1578     wine_anon_mmap( stack, size, PROT_READ | PROT_WRITE, MAP_FIXED );
1579     if (force_exec_prot) mprotect( stack, size, PROT_READ | PROT_WRITE | PROT_EXEC );
1580 }
1581
1582
1583 /***********************************************************************
1584  *           virtual_handle_fault
1585  */
1586 NTSTATUS virtual_handle_fault( LPCVOID addr, DWORD err )
1587 {
1588     struct file_view *view;
1589     NTSTATUS ret = STATUS_ACCESS_VIOLATION;
1590     sigset_t sigset;
1591
1592     server_enter_uninterrupted_section( &csVirtual, &sigset );
1593     if ((view = VIRTUAL_FindView( addr, 0 )))
1594     {
1595         void *page = ROUND_ADDR( addr, page_mask );
1596         BYTE *vprot = &view->prot[((const char *)page - (const char *)view->base) >> page_shift];
1597         if (*vprot & VPROT_GUARD)
1598         {
1599             VIRTUAL_SetProt( view, page, page_size, *vprot & ~VPROT_GUARD );
1600             ret = STATUS_GUARD_PAGE_VIOLATION;
1601         }
1602         if ((err & EXCEPTION_WRITE_FAULT) && (view->protect & VPROT_WRITEWATCH))
1603         {
1604             if (*vprot & VPROT_WRITEWATCH)
1605             {
1606                 *vprot &= ~VPROT_WRITEWATCH;
1607                 VIRTUAL_SetProt( view, page, page_size, *vprot );
1608             }
1609             /* ignore fault if page is writable now */
1610             if (VIRTUAL_GetUnixProt( *vprot ) & PROT_WRITE) ret = STATUS_SUCCESS;
1611         }
1612     }
1613     server_leave_uninterrupted_section( &csVirtual, &sigset );
1614     return ret;
1615 }
1616
1617
1618
1619 /***********************************************************************
1620  *           virtual_handle_stack_fault
1621  *
1622  * Handle an access fault inside the current thread stack.
1623  * Called from inside a signal handler.
1624  */
1625 BOOL virtual_handle_stack_fault( void *addr )
1626 {
1627     struct file_view *view;
1628     BOOL ret = FALSE;
1629
1630     RtlEnterCriticalSection( &csVirtual );  /* no need for signal masking inside signal handler */
1631     if ((view = VIRTUAL_FindView( addr, 0 )))
1632     {
1633         void *page = ROUND_ADDR( addr, page_mask );
1634         BYTE vprot = view->prot[((const char *)page - (const char *)view->base) >> page_shift];
1635         if (vprot & VPROT_GUARD)
1636         {
1637             VIRTUAL_SetProt( view, page, page_size, vprot & ~VPROT_GUARD );
1638             NtCurrentTeb()->Tib.StackLimit = page;
1639             if ((char *)page >= (char *)NtCurrentTeb()->DeallocationStack + 2*page_size)
1640             {
1641                 vprot = view->prot[((char *)page - page_size - (char *)view->base) >> page_shift];
1642                 VIRTUAL_SetProt( view, (char *)page - page_size, page_size, vprot | VPROT_GUARD );
1643             }
1644             ret = TRUE;
1645         }
1646     }
1647     RtlLeaveCriticalSection( &csVirtual );
1648     return ret;
1649 }
1650
1651
1652 /***********************************************************************
1653  *           virtual_check_buffer_for_read
1654  *
1655  * Check if a memory buffer can be read, triggering page faults if needed for DIB section access.
1656  */
1657 BOOL virtual_check_buffer_for_read( const void *ptr, SIZE_T size )
1658 {
1659     if (!size) return TRUE;
1660     if (!ptr) return FALSE;
1661
1662     __TRY
1663     {
1664         volatile const char *p = ptr;
1665         char dummy __attribute__((unused));
1666         SIZE_T count = size;
1667
1668         while (count > page_size)
1669         {
1670             dummy = *p;
1671             p += page_size;
1672             count -= page_size;
1673         }
1674         dummy = p[0];
1675         dummy = p[count - 1];
1676     }
1677     __EXCEPT_PAGE_FAULT
1678     {
1679         return FALSE;
1680     }
1681     __ENDTRY
1682     return TRUE;
1683 }
1684
1685
1686 /***********************************************************************
1687  *           virtual_check_buffer_for_write
1688  *
1689  * Check if a memory buffer can be written to, triggering page faults if needed for write watches.
1690  */
1691 BOOL virtual_check_buffer_for_write( void *ptr, SIZE_T size )
1692 {
1693     if (!size) return TRUE;
1694     if (!ptr) return FALSE;
1695
1696     __TRY
1697     {
1698         volatile char *p = ptr;
1699         SIZE_T count = size;
1700
1701         while (count > page_size)
1702         {
1703             *p |= 0;
1704             p += page_size;
1705             count -= page_size;
1706         }
1707         p[0] |= 0;
1708         p[count - 1] |= 0;
1709     }
1710     __EXCEPT_PAGE_FAULT
1711     {
1712         return FALSE;
1713     }
1714     __ENDTRY
1715     return TRUE;
1716 }
1717
1718
1719 /***********************************************************************
1720  *           VIRTUAL_SetForceExec
1721  *
1722  * Whether to force exec prot on all views.
1723  */
1724 void VIRTUAL_SetForceExec( BOOL enable )
1725 {
1726     struct file_view *view;
1727     sigset_t sigset;
1728
1729     server_enter_uninterrupted_section( &csVirtual, &sigset );
1730     if (!force_exec_prot != !enable)  /* change all existing views */
1731     {
1732         force_exec_prot = enable;
1733
1734         LIST_FOR_EACH_ENTRY( view, &views_list, struct file_view, entry )
1735         {
1736             UINT i, count;
1737             char *addr = view->base;
1738             BYTE commit = view->mapping ? VPROT_COMMITTED : 0;  /* file mappings are always accessible */
1739             int unix_prot = VIRTUAL_GetUnixProt( view->prot[0] | commit );
1740
1741             if (view->protect & VPROT_NOEXEC) continue;
1742             for (count = i = 1; i < view->size >> page_shift; i++, count++)
1743             {
1744                 int prot = VIRTUAL_GetUnixProt( view->prot[i] | commit );
1745                 if (prot == unix_prot) continue;
1746                 if ((unix_prot & PROT_READ) && !(unix_prot & PROT_EXEC))
1747                 {
1748                     TRACE( "%s exec prot for %p-%p\n",
1749                            force_exec_prot ? "enabling" : "disabling",
1750                            addr, addr + (count << page_shift) - 1 );
1751                     mprotect( addr, count << page_shift,
1752                               unix_prot | (force_exec_prot ? PROT_EXEC : 0) );
1753                 }
1754                 addr += (count << page_shift);
1755                 unix_prot = prot;
1756                 count = 0;
1757             }
1758             if (count)
1759             {
1760                 if ((unix_prot & PROT_READ) && !(unix_prot & PROT_EXEC))
1761                 {
1762                     TRACE( "%s exec prot for %p-%p\n",
1763                            force_exec_prot ? "enabling" : "disabling",
1764                            addr, addr + (count << page_shift) - 1 );
1765                     mprotect( addr, count << page_shift,
1766                               unix_prot | (force_exec_prot ? PROT_EXEC : 0) );
1767                 }
1768             }
1769         }
1770     }
1771     server_leave_uninterrupted_section( &csVirtual, &sigset );
1772 }
1773
1774 struct free_range
1775 {
1776     char *base;
1777     char *limit;
1778 };
1779
1780 /* free reserved areas above the limit; callback for wine_mmap_enum_reserved_areas */
1781 static int free_reserved_memory( void *base, size_t size, void *arg )
1782 {
1783     struct free_range *range = arg;
1784
1785     if ((char *)base >= range->limit) return 0;
1786     if ((char *)base + size <= range->base) return 0;
1787     if ((char *)base < range->base)
1788     {
1789         size -= range->base - (char *)base;
1790         base = range->base;
1791     }
1792     if ((char *)base + size > range->limit) size = range->limit - (char *)base;
1793     remove_reserved_area( base, size );
1794     return 1;  /* stop enumeration since the list has changed */
1795 }
1796
1797 /***********************************************************************
1798  *           virtual_release_address_space
1799  *
1800  * Release some address space once we have loaded and initialized the app.
1801  */
1802 void virtual_release_address_space( BOOL free_high_mem )
1803 {
1804     struct free_range range;
1805     sigset_t sigset;
1806
1807     if (user_space_limit == address_space_limit) return;  /* no need to free anything */
1808
1809     server_enter_uninterrupted_section( &csVirtual, &sigset );
1810
1811     /* no large address space on win9x */
1812     if (free_high_mem && NtCurrentTeb()->Peb->OSPlatformId == VER_PLATFORM_WIN32_NT)
1813     {
1814         range.base  = (char *)0x82000000;
1815         range.limit = address_space_limit;
1816         while (wine_mmap_enum_reserved_areas( free_reserved_memory, &range, 1 )) /* nothing */;
1817         user_space_limit = working_set_limit = address_space_limit;
1818     }
1819     else
1820     {
1821 #ifndef __APPLE__  /* dyld doesn't support parts of the WINE_DOS segment being unmapped */
1822         range.base  = (char *)0x20000000;
1823         range.limit = (char *)0x7f000000;
1824         while (wine_mmap_enum_reserved_areas( free_reserved_memory, &range, 0 )) /* nothing */;
1825 #endif
1826     }
1827
1828     server_leave_uninterrupted_section( &csVirtual, &sigset );
1829 }
1830
1831
1832 /***********************************************************************
1833  *             NtAllocateVirtualMemory   (NTDLL.@)
1834  *             ZwAllocateVirtualMemory   (NTDLL.@)
1835  */
1836 NTSTATUS WINAPI NtAllocateVirtualMemory( HANDLE process, PVOID *ret, ULONG zero_bits,
1837                                          SIZE_T *size_ptr, ULONG type, ULONG protect )
1838 {
1839     void *base;
1840     unsigned int vprot;
1841     SIZE_T size = *size_ptr;
1842     SIZE_T mask = get_mask( zero_bits );
1843     NTSTATUS status = STATUS_SUCCESS;
1844     struct file_view *view;
1845     sigset_t sigset;
1846
1847     TRACE("%p %p %08lx %x %08x\n", process, *ret, size, type, protect );
1848
1849     if (!size) return STATUS_INVALID_PARAMETER;
1850
1851     if (process != NtCurrentProcess())
1852     {
1853         apc_call_t call;
1854         apc_result_t result;
1855
1856         memset( &call, 0, sizeof(call) );
1857
1858         call.virtual_alloc.type      = APC_VIRTUAL_ALLOC;
1859         call.virtual_alloc.addr      = wine_server_client_ptr( *ret );
1860         call.virtual_alloc.size      = *size_ptr;
1861         call.virtual_alloc.zero_bits = zero_bits;
1862         call.virtual_alloc.op_type   = type;
1863         call.virtual_alloc.prot      = protect;
1864         status = NTDLL_queue_process_apc( process, &call, &result );
1865         if (status != STATUS_SUCCESS) return status;
1866
1867         if (result.virtual_alloc.status == STATUS_SUCCESS)
1868         {
1869             *ret      = wine_server_get_ptr( result.virtual_alloc.addr );
1870             *size_ptr = result.virtual_alloc.size;
1871         }
1872         return result.virtual_alloc.status;
1873     }
1874
1875     /* Round parameters to a page boundary */
1876
1877     if (is_beyond_limit( 0, size, working_set_limit )) return STATUS_WORKING_SET_LIMIT_RANGE;
1878
1879     if ((status = get_vprot_flags( protect, &vprot, FALSE ))) return status;
1880     if (vprot & VPROT_WRITECOPY) return STATUS_INVALID_PAGE_PROTECTION;
1881     vprot |= VPROT_VALLOC;
1882     if (type & MEM_COMMIT) vprot |= VPROT_COMMITTED;
1883
1884     if (*ret)
1885     {
1886         if (type & MEM_RESERVE) /* Round down to 64k boundary */
1887             base = ROUND_ADDR( *ret, mask );
1888         else
1889             base = ROUND_ADDR( *ret, page_mask );
1890         size = (((UINT_PTR)*ret + size + page_mask) & ~page_mask) - (UINT_PTR)base;
1891
1892         /* address 1 is magic to mean DOS area */
1893         if (!base && *ret == (void *)1 && size == 0x110000)
1894         {
1895             server_enter_uninterrupted_section( &csVirtual, &sigset );
1896             status = allocate_dos_memory( &view, vprot );
1897             if (status == STATUS_SUCCESS)
1898             {
1899                 *ret = view->base;
1900                 *size_ptr = view->size;
1901             }
1902             server_leave_uninterrupted_section( &csVirtual, &sigset );
1903             return status;
1904         }
1905
1906         /* disallow low 64k, wrap-around and kernel space */
1907         if (((char *)base < (char *)0x10000) ||
1908             ((char *)base + size < (char *)base) ||
1909             is_beyond_limit( base, size, address_space_limit ))
1910             return STATUS_INVALID_PARAMETER;
1911     }
1912     else
1913     {
1914         base = NULL;
1915         size = (size + page_mask) & ~page_mask;
1916     }
1917
1918     /* Compute the alloc type flags */
1919
1920     if (!(type & (MEM_COMMIT | MEM_RESERVE | MEM_RESET)) ||
1921         (type & ~(MEM_COMMIT | MEM_RESERVE | MEM_TOP_DOWN | MEM_WRITE_WATCH | MEM_RESET)))
1922     {
1923         WARN("called with wrong alloc type flags (%08x) !\n", type);
1924         return STATUS_INVALID_PARAMETER;
1925     }
1926
1927     /* Reserve the memory */
1928
1929     if (use_locks) server_enter_uninterrupted_section( &csVirtual, &sigset );
1930
1931     if ((type & MEM_RESERVE) || !base)
1932     {
1933         if (type & MEM_WRITE_WATCH) vprot |= VPROT_WRITEWATCH;
1934         status = map_view( &view, base, size, mask, type & MEM_TOP_DOWN, vprot );
1935         if (status == STATUS_SUCCESS) base = view->base;
1936     }
1937     else if (type & MEM_RESET)
1938     {
1939         if (!(view = VIRTUAL_FindView( base, size ))) status = STATUS_NOT_MAPPED_VIEW;
1940         else madvise( base, size, MADV_DONTNEED );
1941     }
1942     else  /* commit the pages */
1943     {
1944         if (!(view = VIRTUAL_FindView( base, size ))) status = STATUS_NOT_MAPPED_VIEW;
1945         else if (!VIRTUAL_SetProt( view, base, size, vprot )) status = STATUS_ACCESS_DENIED;
1946         else if (view->mapping && !(view->protect & VPROT_COMMITTED))
1947         {
1948             SERVER_START_REQ( add_mapping_committed_range )
1949             {
1950                 req->handle = wine_server_obj_handle( view->mapping );
1951                 req->offset = (char *)base - (char *)view->base;
1952                 req->size   = size;
1953                 wine_server_call( req );
1954             }
1955             SERVER_END_REQ;
1956         }
1957     }
1958
1959     if (use_locks) server_leave_uninterrupted_section( &csVirtual, &sigset );
1960
1961     if (status == STATUS_SUCCESS)
1962     {
1963         *ret = base;
1964         *size_ptr = size;
1965     }
1966     return status;
1967 }
1968
1969
1970 /***********************************************************************
1971  *             NtFreeVirtualMemory   (NTDLL.@)
1972  *             ZwFreeVirtualMemory   (NTDLL.@)
1973  */
1974 NTSTATUS WINAPI NtFreeVirtualMemory( HANDLE process, PVOID *addr_ptr, SIZE_T *size_ptr, ULONG type )
1975 {
1976     struct file_view *view;
1977     char *base;
1978     sigset_t sigset;
1979     NTSTATUS status = STATUS_SUCCESS;
1980     LPVOID addr = *addr_ptr;
1981     SIZE_T size = *size_ptr;
1982
1983     TRACE("%p %p %08lx %x\n", process, addr, size, type );
1984
1985     if (process != NtCurrentProcess())
1986     {
1987         apc_call_t call;
1988         apc_result_t result;
1989
1990         memset( &call, 0, sizeof(call) );
1991
1992         call.virtual_free.type      = APC_VIRTUAL_FREE;
1993         call.virtual_free.addr      = wine_server_client_ptr( addr );
1994         call.virtual_free.size      = size;
1995         call.virtual_free.op_type   = type;
1996         status = NTDLL_queue_process_apc( process, &call, &result );
1997         if (status != STATUS_SUCCESS) return status;
1998
1999         if (result.virtual_free.status == STATUS_SUCCESS)
2000         {
2001             *addr_ptr = wine_server_get_ptr( result.virtual_free.addr );
2002             *size_ptr = result.virtual_free.size;
2003         }
2004         return result.virtual_free.status;
2005     }
2006
2007     /* Fix the parameters */
2008
2009     size = ROUND_SIZE( addr, size );
2010     base = ROUND_ADDR( addr, page_mask );
2011
2012     /* avoid freeing the DOS area when a broken app passes a NULL pointer */
2013     if (!base) return STATUS_INVALID_PARAMETER;
2014
2015     server_enter_uninterrupted_section( &csVirtual, &sigset );
2016
2017     if (!(view = VIRTUAL_FindView( base, size )) || !(view->protect & VPROT_VALLOC))
2018     {
2019         status = STATUS_INVALID_PARAMETER;
2020     }
2021     else if (type == MEM_RELEASE)
2022     {
2023         /* Free the pages */
2024
2025         if (size || (base != view->base)) status = STATUS_INVALID_PARAMETER;
2026         else
2027         {
2028             delete_view( view );
2029             *addr_ptr = base;
2030             *size_ptr = size;
2031         }
2032     }
2033     else if (type == MEM_DECOMMIT)
2034     {
2035         status = decommit_pages( view, base - (char *)view->base, size );
2036         if (status == STATUS_SUCCESS)
2037         {
2038             *addr_ptr = base;
2039             *size_ptr = size;
2040         }
2041     }
2042     else
2043     {
2044         WARN("called with wrong free type flags (%08x) !\n", type);
2045         status = STATUS_INVALID_PARAMETER;
2046     }
2047
2048     server_leave_uninterrupted_section( &csVirtual, &sigset );
2049     return status;
2050 }
2051
2052 static ULONG map_protection_to_access( ULONG vprot )
2053 {
2054     vprot &= VPROT_READ | VPROT_WRITE | VPROT_EXEC | VPROT_WRITECOPY;
2055     if (vprot & VPROT_EXEC)
2056     {
2057         if (vprot & VPROT_WRITE) vprot |= VPROT_WRITECOPY;
2058     }
2059     else vprot &= ~VPROT_WRITECOPY;
2060     return vprot;
2061 }
2062
2063 static BOOL is_compatible_protection( const struct file_view *view, ULONG new_prot )
2064 {
2065     ULONG view_prot, map_prot;
2066
2067     view_prot = map_protection_to_access( view->protect );
2068     new_prot = map_protection_to_access( new_prot );
2069
2070     if (view_prot == new_prot) return TRUE;
2071     if (!view_prot) return FALSE;
2072
2073     if ((view_prot & new_prot) != new_prot) return FALSE;
2074
2075     map_prot = map_protection_to_access( view->map_protect );
2076     if ((map_prot & new_prot) == new_prot) return TRUE;
2077
2078     return FALSE;
2079 }
2080
2081 /***********************************************************************
2082  *             NtProtectVirtualMemory   (NTDLL.@)
2083  *             ZwProtectVirtualMemory   (NTDLL.@)
2084  */
2085 NTSTATUS WINAPI NtProtectVirtualMemory( HANDLE process, PVOID *addr_ptr, SIZE_T *size_ptr,
2086                                         ULONG new_prot, ULONG *old_prot )
2087 {
2088     struct file_view *view;
2089     sigset_t sigset;
2090     NTSTATUS status = STATUS_SUCCESS;
2091     char *base;
2092     BYTE vprot;
2093     unsigned int new_vprot;
2094     SIZE_T size = *size_ptr;
2095     LPVOID addr = *addr_ptr;
2096
2097     TRACE("%p %p %08lx %08x\n", process, addr, size, new_prot );
2098
2099     if (process != NtCurrentProcess())
2100     {
2101         apc_call_t call;
2102         apc_result_t result;
2103
2104         memset( &call, 0, sizeof(call) );
2105
2106         call.virtual_protect.type = APC_VIRTUAL_PROTECT;
2107         call.virtual_protect.addr = wine_server_client_ptr( addr );
2108         call.virtual_protect.size = size;
2109         call.virtual_protect.prot = new_prot;
2110         status = NTDLL_queue_process_apc( process, &call, &result );
2111         if (status != STATUS_SUCCESS) return status;
2112
2113         if (result.virtual_protect.status == STATUS_SUCCESS)
2114         {
2115             *addr_ptr = wine_server_get_ptr( result.virtual_protect.addr );
2116             *size_ptr = result.virtual_protect.size;
2117             if (old_prot) *old_prot = result.virtual_protect.prot;
2118         }
2119         return result.virtual_protect.status;
2120     }
2121
2122     /* Fix the parameters */
2123
2124     size = ROUND_SIZE( addr, size );
2125     base = ROUND_ADDR( addr, page_mask );
2126
2127     server_enter_uninterrupted_section( &csVirtual, &sigset );
2128
2129     if ((view = VIRTUAL_FindView( base, size )))
2130     {
2131         /* Make sure all the pages are committed */
2132         if (get_committed_size( view, base, &vprot ) >= size && (vprot & VPROT_COMMITTED))
2133         {
2134             if (!(status = get_vprot_flags( new_prot, &new_vprot, view->protect & VPROT_IMAGE )))
2135             {
2136                 if ((new_vprot & VPROT_WRITECOPY) && (view->protect & VPROT_VALLOC))
2137                     status = STATUS_INVALID_PAGE_PROTECTION;
2138                 else
2139                 {
2140                     if (!view->mapping || is_compatible_protection( view, new_vprot ))
2141                     {
2142                         new_vprot |= VPROT_COMMITTED;
2143                         if (old_prot) *old_prot = VIRTUAL_GetWin32Prot( vprot );
2144                         if (!VIRTUAL_SetProt( view, base, size, new_vprot )) status = STATUS_ACCESS_DENIED;
2145                     }
2146                     else status = STATUS_INVALID_PAGE_PROTECTION;
2147                 }
2148             }
2149         }
2150         else status = STATUS_NOT_COMMITTED;
2151     }
2152     else status = STATUS_INVALID_PARAMETER;
2153
2154     server_leave_uninterrupted_section( &csVirtual, &sigset );
2155
2156     if (status == STATUS_SUCCESS)
2157     {
2158         *addr_ptr = base;
2159         *size_ptr = size;
2160     }
2161     return status;
2162 }
2163
2164
2165 /* retrieve state for a free memory area; callback for wine_mmap_enum_reserved_areas */
2166 static int get_free_mem_state_callback( void *start, size_t size, void *arg )
2167 {
2168     MEMORY_BASIC_INFORMATION *info = arg;
2169     void *end = (char *)start + size;
2170
2171     if ((char *)info->BaseAddress + info->RegionSize < (char *)start) return 0;
2172
2173     if (info->BaseAddress >= end)
2174     {
2175         if (info->AllocationBase < end) info->AllocationBase = end;
2176         return 0;
2177     }
2178
2179     if (info->BaseAddress >= start || start <= address_space_start)
2180     {
2181         /* it's a real free area */
2182         info->State             = MEM_FREE;
2183         info->Protect           = PAGE_NOACCESS;
2184         info->AllocationBase    = 0;
2185         info->AllocationProtect = 0;
2186         info->Type              = 0;
2187         if ((char *)info->BaseAddress + info->RegionSize > (char *)end)
2188             info->RegionSize = (char *)end - (char *)info->BaseAddress;
2189     }
2190     else /* outside of the reserved area, pretend it's allocated */
2191     {
2192         info->RegionSize        = (char *)start - (char *)info->BaseAddress;
2193         info->State             = MEM_RESERVE;
2194         info->Protect           = PAGE_NOACCESS;
2195         info->AllocationProtect = PAGE_NOACCESS;
2196         info->Type              = MEM_PRIVATE;
2197     }
2198     return 1;
2199 }
2200
2201 #define UNIMPLEMENTED_INFO_CLASS(c) \
2202     case c: \
2203         FIXME("(process=%p,addr=%p) Unimplemented information class: " #c "\n", process, addr); \
2204         return STATUS_INVALID_INFO_CLASS
2205
2206 /***********************************************************************
2207  *             NtQueryVirtualMemory   (NTDLL.@)
2208  *             ZwQueryVirtualMemory   (NTDLL.@)
2209  */
2210 NTSTATUS WINAPI NtQueryVirtualMemory( HANDLE process, LPCVOID addr,
2211                                       MEMORY_INFORMATION_CLASS info_class, PVOID buffer,
2212                                       SIZE_T len, SIZE_T *res_len )
2213 {
2214     struct file_view *view;
2215     char *base, *alloc_base = 0;
2216     struct list *ptr;
2217     SIZE_T size = 0;
2218     MEMORY_BASIC_INFORMATION *info = buffer;
2219     sigset_t sigset;
2220
2221     if (info_class != MemoryBasicInformation)
2222     {
2223         switch(info_class)
2224         {
2225             UNIMPLEMENTED_INFO_CLASS(MemoryWorkingSetList);
2226             UNIMPLEMENTED_INFO_CLASS(MemorySectionName);
2227             UNIMPLEMENTED_INFO_CLASS(MemoryBasicVlmInformation);
2228
2229             default:
2230                 FIXME("(%p,%p,info_class=%d,%p,%ld,%p) Unknown information class\n",
2231                       process, addr, info_class, buffer, len, res_len);
2232                 return STATUS_INVALID_INFO_CLASS;
2233         }
2234     }
2235
2236     if (process != NtCurrentProcess())
2237     {
2238         NTSTATUS status;
2239         apc_call_t call;
2240         apc_result_t result;
2241
2242         memset( &call, 0, sizeof(call) );
2243
2244         call.virtual_query.type = APC_VIRTUAL_QUERY;
2245         call.virtual_query.addr = wine_server_client_ptr( addr );
2246         status = NTDLL_queue_process_apc( process, &call, &result );
2247         if (status != STATUS_SUCCESS) return status;
2248
2249         if (result.virtual_query.status == STATUS_SUCCESS)
2250         {
2251             info->BaseAddress       = wine_server_get_ptr( result.virtual_query.base );
2252             info->AllocationBase    = wine_server_get_ptr( result.virtual_query.alloc_base );
2253             info->RegionSize        = result.virtual_query.size;
2254             info->Protect           = result.virtual_query.prot;
2255             info->AllocationProtect = result.virtual_query.alloc_prot;
2256             info->State             = (DWORD)result.virtual_query.state << 12;
2257             info->Type              = (DWORD)result.virtual_query.alloc_type << 16;
2258             if (info->RegionSize != result.virtual_query.size)  /* truncated */
2259                 return STATUS_INVALID_PARAMETER;  /* FIXME */
2260             if (res_len) *res_len = sizeof(*info);
2261         }
2262         return result.virtual_query.status;
2263     }
2264
2265     base = ROUND_ADDR( addr, page_mask );
2266
2267     if (is_beyond_limit( base, 1, working_set_limit )) return STATUS_WORKING_SET_LIMIT_RANGE;
2268
2269     /* Find the view containing the address */
2270
2271     server_enter_uninterrupted_section( &csVirtual, &sigset );
2272     ptr = list_head( &views_list );
2273     for (;;)
2274     {
2275         if (!ptr)
2276         {
2277             size = (char *)working_set_limit - alloc_base;
2278             view = NULL;
2279             break;
2280         }
2281         view = LIST_ENTRY( ptr, struct file_view, entry );
2282         if ((char *)view->base > base)
2283         {
2284             size = (char *)view->base - alloc_base;
2285             view = NULL;
2286             break;
2287         }
2288         if ((char *)view->base + view->size > base)
2289         {
2290             alloc_base = view->base;
2291             size = view->size;
2292             break;
2293         }
2294         alloc_base = (char *)view->base + view->size;
2295         ptr = list_next( &views_list, ptr );
2296     }
2297
2298     /* Fill the info structure */
2299
2300     info->AllocationBase = alloc_base;
2301     info->BaseAddress    = base;
2302     info->RegionSize     = size - (base - alloc_base);
2303
2304     if (!view)
2305     {
2306         if (!wine_mmap_enum_reserved_areas( get_free_mem_state_callback, info, 0 ))
2307         {
2308             /* not in a reserved area at all, pretend it's allocated */
2309 #ifdef __i386__
2310             if (base >= (char *)address_space_start)
2311             {
2312                 info->State             = MEM_RESERVE;
2313                 info->Protect           = PAGE_NOACCESS;
2314                 info->AllocationProtect = PAGE_NOACCESS;
2315                 info->Type              = MEM_PRIVATE;
2316             }
2317             else
2318 #endif
2319             {
2320                 info->State             = MEM_FREE;
2321                 info->Protect           = PAGE_NOACCESS;
2322                 info->AllocationBase    = 0;
2323                 info->AllocationProtect = 0;
2324                 info->Type              = 0;
2325             }
2326         }
2327     }
2328     else
2329     {
2330         BYTE vprot;
2331         SIZE_T range_size = get_committed_size( view, base, &vprot );
2332
2333         info->State = (vprot & VPROT_COMMITTED) ? MEM_COMMIT : MEM_RESERVE;
2334         info->Protect = (vprot & VPROT_COMMITTED) ? VIRTUAL_GetWin32Prot( vprot ) : 0;
2335         info->AllocationBase = alloc_base;
2336         info->AllocationProtect = VIRTUAL_GetWin32Prot( view->protect );
2337         if (view->protect & VPROT_IMAGE) info->Type = MEM_IMAGE;
2338         else if (view->protect & VPROT_VALLOC) info->Type = MEM_PRIVATE;
2339         else info->Type = MEM_MAPPED;
2340         for (size = base - alloc_base; size < base + range_size - alloc_base; size += page_size)
2341             if ((view->prot[size >> page_shift] ^ vprot) & ~VPROT_WRITEWATCH) break;
2342         info->RegionSize = size - (base - alloc_base);
2343     }
2344     server_leave_uninterrupted_section( &csVirtual, &sigset );
2345
2346     if (res_len) *res_len = sizeof(*info);
2347     return STATUS_SUCCESS;
2348 }
2349
2350
2351 /***********************************************************************
2352  *             NtLockVirtualMemory   (NTDLL.@)
2353  *             ZwLockVirtualMemory   (NTDLL.@)
2354  */
2355 NTSTATUS WINAPI NtLockVirtualMemory( HANDLE process, PVOID *addr, SIZE_T *size, ULONG unknown )
2356 {
2357     NTSTATUS status = STATUS_SUCCESS;
2358
2359     if (process != NtCurrentProcess())
2360     {
2361         apc_call_t call;
2362         apc_result_t result;
2363
2364         memset( &call, 0, sizeof(call) );
2365
2366         call.virtual_lock.type = APC_VIRTUAL_LOCK;
2367         call.virtual_lock.addr = wine_server_client_ptr( *addr );
2368         call.virtual_lock.size = *size;
2369         status = NTDLL_queue_process_apc( process, &call, &result );
2370         if (status != STATUS_SUCCESS) return status;
2371
2372         if (result.virtual_lock.status == STATUS_SUCCESS)
2373         {
2374             *addr = wine_server_get_ptr( result.virtual_lock.addr );
2375             *size = result.virtual_lock.size;
2376         }
2377         return result.virtual_lock.status;
2378     }
2379
2380     *size = ROUND_SIZE( *addr, *size );
2381     *addr = ROUND_ADDR( *addr, page_mask );
2382
2383     if (mlock( *addr, *size )) status = STATUS_ACCESS_DENIED;
2384     return status;
2385 }
2386
2387
2388 /***********************************************************************
2389  *             NtUnlockVirtualMemory   (NTDLL.@)
2390  *             ZwUnlockVirtualMemory   (NTDLL.@)
2391  */
2392 NTSTATUS WINAPI NtUnlockVirtualMemory( HANDLE process, PVOID *addr, SIZE_T *size, ULONG unknown )
2393 {
2394     NTSTATUS status = STATUS_SUCCESS;
2395
2396     if (process != NtCurrentProcess())
2397     {
2398         apc_call_t call;
2399         apc_result_t result;
2400
2401         memset( &call, 0, sizeof(call) );
2402
2403         call.virtual_unlock.type = APC_VIRTUAL_UNLOCK;
2404         call.virtual_unlock.addr = wine_server_client_ptr( *addr );
2405         call.virtual_unlock.size = *size;
2406         status = NTDLL_queue_process_apc( process, &call, &result );
2407         if (status != STATUS_SUCCESS) return status;
2408
2409         if (result.virtual_unlock.status == STATUS_SUCCESS)
2410         {
2411             *addr = wine_server_get_ptr( result.virtual_unlock.addr );
2412             *size = result.virtual_unlock.size;
2413         }
2414         return result.virtual_unlock.status;
2415     }
2416
2417     *size = ROUND_SIZE( *addr, *size );
2418     *addr = ROUND_ADDR( *addr, page_mask );
2419
2420     if (munlock( *addr, *size )) status = STATUS_ACCESS_DENIED;
2421     return status;
2422 }
2423
2424
2425 /***********************************************************************
2426  *             NtCreateSection   (NTDLL.@)
2427  *             ZwCreateSection   (NTDLL.@)
2428  */
2429 NTSTATUS WINAPI NtCreateSection( HANDLE *handle, ACCESS_MASK access, const OBJECT_ATTRIBUTES *attr,
2430                                  const LARGE_INTEGER *size, ULONG protect,
2431                                  ULONG sec_flags, HANDLE file )
2432 {
2433     NTSTATUS ret;
2434     unsigned int vprot;
2435     DWORD len = (attr && attr->ObjectName) ? attr->ObjectName->Length : 0;
2436     struct security_descriptor *sd = NULL;
2437     struct object_attributes objattr;
2438
2439     /* Check parameters */
2440
2441     if (len > MAX_PATH*sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
2442
2443     if ((ret = get_vprot_flags( protect, &vprot, sec_flags & SEC_IMAGE ))) return ret;
2444
2445     objattr.rootdir = wine_server_obj_handle( attr ? attr->RootDirectory : 0 );
2446     objattr.sd_len = 0;
2447     objattr.name_len = len;
2448     if (attr)
2449     {
2450         ret = NTDLL_create_struct_sd( attr->SecurityDescriptor, &sd, &objattr.sd_len );
2451         if (ret != STATUS_SUCCESS) return ret;
2452     }
2453
2454     if (!(sec_flags & SEC_RESERVE)) vprot |= VPROT_COMMITTED;
2455     if (sec_flags & SEC_NOCACHE) vprot |= VPROT_NOCACHE;
2456     if (sec_flags & SEC_IMAGE) vprot |= VPROT_IMAGE;
2457
2458     /* Create the server object */
2459
2460     SERVER_START_REQ( create_mapping )
2461     {
2462         req->access      = access;
2463         req->attributes  = (attr) ? attr->Attributes : 0;
2464         req->file_handle = wine_server_obj_handle( file );
2465         req->size        = size ? size->QuadPart : 0;
2466         req->protect     = vprot;
2467         wine_server_add_data( req, &objattr, sizeof(objattr) );
2468         if (objattr.sd_len) wine_server_add_data( req, sd, objattr.sd_len );
2469         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
2470         ret = wine_server_call( req );
2471         *handle = wine_server_ptr_handle( reply->handle );
2472     }
2473     SERVER_END_REQ;
2474
2475     NTDLL_free_struct_sd( sd );
2476
2477     return ret;
2478 }
2479
2480
2481 /***********************************************************************
2482  *             NtOpenSection   (NTDLL.@)
2483  *             ZwOpenSection   (NTDLL.@)
2484  */
2485 NTSTATUS WINAPI NtOpenSection( HANDLE *handle, ACCESS_MASK access, const OBJECT_ATTRIBUTES *attr )
2486 {
2487     NTSTATUS ret;
2488     DWORD len = attr->ObjectName->Length;
2489
2490     if (len > MAX_PATH*sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
2491
2492     SERVER_START_REQ( open_mapping )
2493     {
2494         req->access  = access;
2495         req->attributes = attr->Attributes;
2496         req->rootdir = wine_server_obj_handle( attr->RootDirectory );
2497         wine_server_add_data( req, attr->ObjectName->Buffer, len );
2498         if (!(ret = wine_server_call( req ))) *handle = wine_server_ptr_handle( reply->handle );
2499     }
2500     SERVER_END_REQ;
2501     return ret;
2502 }
2503
2504
2505 /***********************************************************************
2506  *             NtMapViewOfSection   (NTDLL.@)
2507  *             ZwMapViewOfSection   (NTDLL.@)
2508  */
2509 NTSTATUS WINAPI NtMapViewOfSection( HANDLE handle, HANDLE process, PVOID *addr_ptr, ULONG zero_bits,
2510                                     SIZE_T commit_size, const LARGE_INTEGER *offset_ptr, SIZE_T *size_ptr,
2511                                     SECTION_INHERIT inherit, ULONG alloc_type, ULONG protect )
2512 {
2513     NTSTATUS res;
2514     mem_size_t full_size;
2515     ACCESS_MASK access;
2516     SIZE_T size, mask = get_mask( zero_bits );
2517     int unix_handle = -1, needs_close;
2518     unsigned int map_vprot, vprot;
2519     void *base;
2520     struct file_view *view;
2521     DWORD header_size;
2522     HANDLE dup_mapping, shared_file;
2523     LARGE_INTEGER offset;
2524     sigset_t sigset;
2525
2526     offset.QuadPart = offset_ptr ? offset_ptr->QuadPart : 0;
2527
2528     TRACE("handle=%p process=%p addr=%p off=%x%08x size=%lx access=%x\n",
2529           handle, process, *addr_ptr, offset.u.HighPart, offset.u.LowPart, *size_ptr, protect );
2530
2531     /* Check parameters */
2532
2533     if ((offset.u.LowPart & mask) || (*addr_ptr && ((UINT_PTR)*addr_ptr & mask)))
2534         return STATUS_INVALID_PARAMETER;
2535
2536     switch(protect)
2537     {
2538     case PAGE_NOACCESS:
2539         access = SECTION_MAP_READ;
2540         break;
2541     case PAGE_READWRITE:
2542     case PAGE_EXECUTE_READWRITE:
2543         access = SECTION_MAP_WRITE;
2544         break;
2545     case PAGE_READONLY:
2546     case PAGE_WRITECOPY:
2547     case PAGE_EXECUTE:
2548     case PAGE_EXECUTE_READ:
2549     case PAGE_EXECUTE_WRITECOPY:
2550         access = SECTION_MAP_READ;
2551         break;
2552     default:
2553         return STATUS_INVALID_PAGE_PROTECTION;
2554     }
2555
2556     if (process != NtCurrentProcess())
2557     {
2558         apc_call_t call;
2559         apc_result_t result;
2560
2561         memset( &call, 0, sizeof(call) );
2562
2563         call.map_view.type        = APC_MAP_VIEW;
2564         call.map_view.handle      = wine_server_obj_handle( handle );
2565         call.map_view.addr        = wine_server_client_ptr( *addr_ptr );
2566         call.map_view.size        = *size_ptr;
2567         call.map_view.offset      = offset.QuadPart;
2568         call.map_view.zero_bits   = zero_bits;
2569         call.map_view.alloc_type  = alloc_type;
2570         call.map_view.prot        = protect;
2571         res = NTDLL_queue_process_apc( process, &call, &result );
2572         if (res != STATUS_SUCCESS) return res;
2573
2574         if ((NTSTATUS)result.map_view.status >= 0)
2575         {
2576             *addr_ptr = wine_server_get_ptr( result.map_view.addr );
2577             *size_ptr = result.map_view.size;
2578         }
2579         return result.map_view.status;
2580     }
2581
2582     SERVER_START_REQ( get_mapping_info )
2583     {
2584         req->handle = wine_server_obj_handle( handle );
2585         req->access = access;
2586         res = wine_server_call( req );
2587         map_vprot   = reply->protect;
2588         base        = wine_server_get_ptr( reply->base );
2589         full_size   = reply->size;
2590         header_size = reply->header_size;
2591         dup_mapping = wine_server_ptr_handle( reply->mapping );
2592         shared_file = wine_server_ptr_handle( reply->shared_file );
2593         if ((ULONG_PTR)base != reply->base) base = NULL;
2594     }
2595     SERVER_END_REQ;
2596     if (res) return res;
2597
2598     if ((res = server_get_unix_fd( handle, 0, &unix_handle, &needs_close, NULL, NULL ))) goto done;
2599
2600     if (map_vprot & VPROT_IMAGE)
2601     {
2602         size = full_size;
2603         if (size != full_size)  /* truncated */
2604         {
2605             WARN( "Modules larger than 4Gb (%s) not supported\n", wine_dbgstr_longlong(full_size) );
2606             res = STATUS_INVALID_PARAMETER;
2607             goto done;
2608         }
2609         if (shared_file)
2610         {
2611             int shared_fd, shared_needs_close;
2612
2613             if ((res = server_get_unix_fd( shared_file, FILE_READ_DATA|FILE_WRITE_DATA,
2614                                            &shared_fd, &shared_needs_close, NULL, NULL ))) goto done;
2615             res = map_image( handle, unix_handle, base, size, mask, header_size,
2616                              shared_fd, dup_mapping, map_vprot, addr_ptr );
2617             if (shared_needs_close) close( shared_fd );
2618             NtClose( shared_file );
2619         }
2620         else
2621         {
2622             res = map_image( handle, unix_handle, base, size, mask, header_size,
2623                              -1, dup_mapping, map_vprot, addr_ptr );
2624         }
2625         if (needs_close) close( unix_handle );
2626         if (res >= 0) *size_ptr = size;
2627         return res;
2628     }
2629
2630     res = STATUS_INVALID_PARAMETER;
2631     if (offset.QuadPart >= full_size) goto done;
2632     if (*size_ptr)
2633     {
2634         if (*size_ptr > full_size - offset.QuadPart) goto done;
2635         size = ROUND_SIZE( offset.u.LowPart, *size_ptr );
2636         if (size < *size_ptr) goto done;  /* wrap-around */
2637     }
2638     else
2639     {
2640         size = full_size - offset.QuadPart;
2641         if (size != full_size - offset.QuadPart)  /* truncated */
2642         {
2643             WARN( "Files larger than 4Gb (%s) not supported on this platform\n",
2644                   wine_dbgstr_longlong(full_size) );
2645             goto done;
2646         }
2647     }
2648
2649     /* Reserve a properly aligned area */
2650
2651     server_enter_uninterrupted_section( &csVirtual, &sigset );
2652
2653     get_vprot_flags( protect, &vprot, map_vprot & VPROT_IMAGE );
2654     vprot |= (map_vprot & VPROT_COMMITTED);
2655     res = map_view( &view, *addr_ptr, size, mask, FALSE, vprot );
2656     if (res)
2657     {
2658         server_leave_uninterrupted_section( &csVirtual, &sigset );
2659         goto done;
2660     }
2661
2662     /* Map the file */
2663
2664     TRACE("handle=%p size=%lx offset=%x%08x\n",
2665           handle, size, offset.u.HighPart, offset.u.LowPart );
2666
2667     res = map_file_into_view( view, unix_handle, 0, size, offset.QuadPart, vprot, !dup_mapping );
2668     if (res == STATUS_SUCCESS)
2669     {
2670         *addr_ptr = view->base;
2671         *size_ptr = size;
2672         view->mapping = dup_mapping;
2673         view->map_protect = map_vprot;
2674         dup_mapping = 0;  /* don't close it */
2675     }
2676     else
2677     {
2678         ERR( "map_file_into_view %p %lx %x%08x failed\n",
2679              view->base, size, offset.u.HighPart, offset.u.LowPart );
2680         delete_view( view );
2681     }
2682
2683     server_leave_uninterrupted_section( &csVirtual, &sigset );
2684
2685 done:
2686     if (dup_mapping) NtClose( dup_mapping );
2687     if (needs_close) close( unix_handle );
2688     return res;
2689 }
2690
2691
2692 /***********************************************************************
2693  *             NtUnmapViewOfSection   (NTDLL.@)
2694  *             ZwUnmapViewOfSection   (NTDLL.@)
2695  */
2696 NTSTATUS WINAPI NtUnmapViewOfSection( HANDLE process, PVOID addr )
2697 {
2698     struct file_view *view;
2699     NTSTATUS status = STATUS_NOT_MAPPED_VIEW;
2700     sigset_t sigset;
2701     void *base = ROUND_ADDR( addr, page_mask );
2702
2703     if (process != NtCurrentProcess())
2704     {
2705         apc_call_t call;
2706         apc_result_t result;
2707
2708         memset( &call, 0, sizeof(call) );
2709
2710         call.unmap_view.type = APC_UNMAP_VIEW;
2711         call.unmap_view.addr = wine_server_client_ptr( addr );
2712         status = NTDLL_queue_process_apc( process, &call, &result );
2713         if (status == STATUS_SUCCESS) status = result.unmap_view.status;
2714         return status;
2715     }
2716
2717     server_enter_uninterrupted_section( &csVirtual, &sigset );
2718     if ((view = VIRTUAL_FindView( base, 0 )) && (base == view->base) && !(view->protect & VPROT_VALLOC))
2719     {
2720         delete_view( view );
2721         status = STATUS_SUCCESS;
2722     }
2723     server_leave_uninterrupted_section( &csVirtual, &sigset );
2724     return status;
2725 }
2726
2727
2728 /***********************************************************************
2729  *             NtFlushVirtualMemory   (NTDLL.@)
2730  *             ZwFlushVirtualMemory   (NTDLL.@)
2731  */
2732 NTSTATUS WINAPI NtFlushVirtualMemory( HANDLE process, LPCVOID *addr_ptr,
2733                                       SIZE_T *size_ptr, ULONG unknown )
2734 {
2735     struct file_view *view;
2736     NTSTATUS status = STATUS_SUCCESS;
2737     sigset_t sigset;
2738     void *addr = ROUND_ADDR( *addr_ptr, page_mask );
2739
2740     if (process != NtCurrentProcess())
2741     {
2742         apc_call_t call;
2743         apc_result_t result;
2744
2745         memset( &call, 0, sizeof(call) );
2746
2747         call.virtual_flush.type = APC_VIRTUAL_FLUSH;
2748         call.virtual_flush.addr = wine_server_client_ptr( addr );
2749         call.virtual_flush.size = *size_ptr;
2750         status = NTDLL_queue_process_apc( process, &call, &result );
2751         if (status != STATUS_SUCCESS) return status;
2752
2753         if (result.virtual_flush.status == STATUS_SUCCESS)
2754         {
2755             *addr_ptr = wine_server_get_ptr( result.virtual_flush.addr );
2756             *size_ptr = result.virtual_flush.size;
2757         }
2758         return result.virtual_flush.status;
2759     }
2760
2761     server_enter_uninterrupted_section( &csVirtual, &sigset );
2762     if (!(view = VIRTUAL_FindView( addr, *size_ptr ))) status = STATUS_INVALID_PARAMETER;
2763     else
2764     {
2765         if (!*size_ptr) *size_ptr = view->size;
2766         *addr_ptr = addr;
2767         if (msync( addr, *size_ptr, MS_SYNC )) status = STATUS_NOT_MAPPED_DATA;
2768     }
2769     server_leave_uninterrupted_section( &csVirtual, &sigset );
2770     return status;
2771 }
2772
2773
2774 /***********************************************************************
2775  *             NtGetWriteWatch   (NTDLL.@)
2776  *             ZwGetWriteWatch   (NTDLL.@)
2777  */
2778 NTSTATUS WINAPI NtGetWriteWatch( HANDLE process, ULONG flags, PVOID base, SIZE_T size, PVOID *addresses,
2779                                  ULONG_PTR *count, ULONG *granularity )
2780 {
2781     struct file_view *view;
2782     NTSTATUS status = STATUS_SUCCESS;
2783     sigset_t sigset;
2784
2785     size = ROUND_SIZE( base, size );
2786     base = ROUND_ADDR( base, page_mask );
2787
2788     if (!count || !granularity) return STATUS_ACCESS_VIOLATION;
2789     if (!*count || !size) return STATUS_INVALID_PARAMETER;
2790     if (flags & ~WRITE_WATCH_FLAG_RESET) return STATUS_INVALID_PARAMETER;
2791
2792     if (!addresses) return STATUS_ACCESS_VIOLATION;
2793
2794     TRACE( "%p %x %p-%p %p %lu\n", process, flags, base, (char *)base + size,
2795            addresses, *count );
2796
2797     server_enter_uninterrupted_section( &csVirtual, &sigset );
2798
2799     if ((view = VIRTUAL_FindView( base, size )) && (view->protect & VPROT_WRITEWATCH))
2800     {
2801         ULONG_PTR pos = 0;
2802         char *addr = base;
2803         char *end = addr + size;
2804
2805         while (pos < *count && addr < end)
2806         {
2807             BYTE prot = view->prot[(addr - (char *)view->base) >> page_shift];
2808             if (!(prot & VPROT_WRITEWATCH)) addresses[pos++] = addr;
2809             addr += page_size;
2810         }
2811         if (flags & WRITE_WATCH_FLAG_RESET) reset_write_watches( view, base, addr - (char *)base );
2812         *count = pos;
2813         *granularity = page_size;
2814     }
2815     else status = STATUS_INVALID_PARAMETER;
2816
2817     server_leave_uninterrupted_section( &csVirtual, &sigset );
2818     return status;
2819 }
2820
2821
2822 /***********************************************************************
2823  *             NtResetWriteWatch   (NTDLL.@)
2824  *             ZwResetWriteWatch   (NTDLL.@)
2825  */
2826 NTSTATUS WINAPI NtResetWriteWatch( HANDLE process, PVOID base, SIZE_T size )
2827 {
2828     struct file_view *view;
2829     NTSTATUS status = STATUS_SUCCESS;
2830     sigset_t sigset;
2831
2832     size = ROUND_SIZE( base, size );
2833     base = ROUND_ADDR( base, page_mask );
2834
2835     TRACE( "%p %p-%p\n", process, base, (char *)base + size );
2836
2837     if (!size) return STATUS_INVALID_PARAMETER;
2838
2839     server_enter_uninterrupted_section( &csVirtual, &sigset );
2840
2841     if ((view = VIRTUAL_FindView( base, size )) && (view->protect & VPROT_WRITEWATCH))
2842         reset_write_watches( view, base, size );
2843     else
2844         status = STATUS_INVALID_PARAMETER;
2845
2846     server_leave_uninterrupted_section( &csVirtual, &sigset );
2847     return status;
2848 }
2849
2850
2851 /***********************************************************************
2852  *             NtReadVirtualMemory   (NTDLL.@)
2853  *             ZwReadVirtualMemory   (NTDLL.@)
2854  */
2855 NTSTATUS WINAPI NtReadVirtualMemory( HANDLE process, const void *addr, void *buffer,
2856                                      SIZE_T size, SIZE_T *bytes_read )
2857 {
2858     NTSTATUS status;
2859
2860     if (virtual_check_buffer_for_write( buffer, size ))
2861     {
2862         SERVER_START_REQ( read_process_memory )
2863         {
2864             req->handle = wine_server_obj_handle( process );
2865             req->addr   = wine_server_client_ptr( addr );
2866             wine_server_set_reply( req, buffer, size );
2867             if ((status = wine_server_call( req ))) size = 0;
2868         }
2869         SERVER_END_REQ;
2870     }
2871     else
2872     {
2873         status = STATUS_ACCESS_VIOLATION;
2874         size = 0;
2875     }
2876     if (bytes_read) *bytes_read = size;
2877     return status;
2878 }
2879
2880
2881 /***********************************************************************
2882  *             NtWriteVirtualMemory   (NTDLL.@)
2883  *             ZwWriteVirtualMemory   (NTDLL.@)
2884  */
2885 NTSTATUS WINAPI NtWriteVirtualMemory( HANDLE process, void *addr, const void *buffer,
2886                                       SIZE_T size, SIZE_T *bytes_written )
2887 {
2888     NTSTATUS status;
2889
2890     if (virtual_check_buffer_for_read( buffer, size ))
2891     {
2892         SERVER_START_REQ( write_process_memory )
2893         {
2894             req->handle     = wine_server_obj_handle( process );
2895             req->addr       = wine_server_client_ptr( addr );
2896             wine_server_add_data( req, buffer, size );
2897             if ((status = wine_server_call( req ))) size = 0;
2898         }
2899         SERVER_END_REQ;
2900     }
2901     else
2902     {
2903         status = STATUS_PARTIAL_COPY;
2904         size = 0;
2905     }
2906     if (bytes_written) *bytes_written = size;
2907     return status;
2908 }
2909
2910
2911 /***********************************************************************
2912  *             NtAreMappedFilesTheSame   (NTDLL.@)
2913  *             ZwAreMappedFilesTheSame   (NTDLL.@)
2914  */
2915 NTSTATUS WINAPI NtAreMappedFilesTheSame(PVOID addr1, PVOID addr2)
2916 {
2917     struct file_view *view1, *view2;
2918     struct stat st1, st2;
2919     NTSTATUS status;
2920     sigset_t sigset;
2921
2922     TRACE("%p %p\n", addr1, addr2);
2923
2924     server_enter_uninterrupted_section( &csVirtual, &sigset );
2925
2926     view1 = VIRTUAL_FindView( addr1, 0 );
2927     view2 = VIRTUAL_FindView( addr2, 0 );
2928
2929     if (!view1 || !view2)
2930         status = STATUS_INVALID_ADDRESS;
2931     else if ((view1->protect & VPROT_VALLOC) || (view2->protect & VPROT_VALLOC))
2932         status = STATUS_CONFLICTING_ADDRESSES;
2933     else if (!(view1->protect & VPROT_IMAGE) || !(view2->protect & VPROT_IMAGE))
2934         status = STATUS_NOT_SAME_DEVICE;
2935     else if (view1 == view2)
2936         status = STATUS_SUCCESS;
2937     else if (!stat_mapping_file( view1, &st1 ) && !stat_mapping_file( view2, &st2 ) &&
2938              st1.st_dev == st2.st_dev && st1.st_ino == st2.st_ino)
2939         status = STATUS_SUCCESS;
2940     else
2941         status = STATUS_NOT_SAME_DEVICE;
2942
2943     server_leave_uninterrupted_section( &csVirtual, &sigset );
2944     return status;
2945 }