Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 1 | //===- Win32/Memory.cpp - Win32 Memory Implementation -----------*- C++ -*-===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame^] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file provides the Win32 specific implementation of various Memory |
| 11 | // management utilities |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 15 | #include "Win32.h" |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 16 | #include "llvm/System/Process.h" |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 17 | |
| 18 | namespace llvm { |
| 19 | using namespace sys; |
| 20 | |
| 21 | //===----------------------------------------------------------------------===// |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 22 | //=== WARNING: Implementation here must contain only Win32 specific code |
| 23 | //=== and must not be UNIX code |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 24 | //===----------------------------------------------------------------------===// |
| 25 | |
Chris Lattner | bed22d8 | 2006-07-07 17:32:37 +0000 | [diff] [blame] | 26 | MemoryBlock Memory::AllocateRWX(unsigned NumBytes, |
| 27 | const MemoryBlock *NearBlock, |
| 28 | std::string *ErrMsg) { |
Reid Spencer | 33189e7 | 2004-09-13 22:38:11 +0000 | [diff] [blame] | 29 | if (NumBytes == 0) return MemoryBlock(); |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 30 | |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 31 | static const long pageSize = Process::GetPageSize(); |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 32 | unsigned NumPages = (NumBytes+pageSize-1)/pageSize; |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 33 | |
Andrew Lenharth | a00269b | 2005-07-29 23:40:16 +0000 | [diff] [blame] | 34 | //FIXME: support NearBlock if ever needed on Win64. |
| 35 | |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 36 | void *pa = VirtualAlloc(NULL, NumPages*pageSize, MEM_COMMIT, |
| 37 | PAGE_EXECUTE_READWRITE); |
| 38 | if (pa == NULL) { |
Reid Spencer | 0554575 | 2006-08-25 21:37:17 +0000 | [diff] [blame] | 39 | MakeErrMsg(ErrMsg, "Can't allocate RWX Memory: "); |
Chris Lattner | bed22d8 | 2006-07-07 17:32:37 +0000 | [diff] [blame] | 40 | return MemoryBlock(); |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 41 | } |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 42 | |
Reid Spencer | 33189e7 | 2004-09-13 22:38:11 +0000 | [diff] [blame] | 43 | MemoryBlock result; |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 44 | result.Address = pa; |
| 45 | result.Size = NumPages*pageSize; |
Reid Spencer | 33189e7 | 2004-09-13 22:38:11 +0000 | [diff] [blame] | 46 | return result; |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 47 | } |
| 48 | |
Chris Lattner | bed22d8 | 2006-07-07 17:32:37 +0000 | [diff] [blame] | 49 | bool Memory::ReleaseRWX(MemoryBlock &M, std::string *ErrMsg) { |
Chris Lattner | fab30f2 | 2006-07-26 20:37:11 +0000 | [diff] [blame] | 50 | if (M.Address == 0 || M.Size == 0) return false; |
Chris Lattner | bed22d8 | 2006-07-07 17:32:37 +0000 | [diff] [blame] | 51 | if (!VirtualFree(M.Address, 0, MEM_RELEASE)) |
Reid Spencer | 0554575 | 2006-08-25 21:37:17 +0000 | [diff] [blame] | 52 | return MakeErrMsg(ErrMsg, "Can't release RWX Memory: "); |
Chris Lattner | fab30f2 | 2006-07-26 20:37:11 +0000 | [diff] [blame] | 53 | return false; |
Reid Spencer | cbad701 | 2004-09-11 04:59:30 +0000 | [diff] [blame] | 54 | } |
Reid Spencer | b016a37 | 2004-09-15 05:49:50 +0000 | [diff] [blame] | 55 | |
| 56 | } |
| 57 | |