2e9612c3205c6299f4c43a66e385e2036eb08a57
[tpg/acess2.git] / AcessNative / ld-acess_src / memory.c
1 /*
2  */
3 #include "common.h"
4 #include <stdio.h>
5 #include <stdlib.h>
6 #if __WIN32__
7 # include <windows.h>
8 #else
9 # include <sys/mman.h>
10 # include <errno.h>
11 #endif
12
13 // === PROTOTYPES ===
14  int    AllocateMemory(uintptr_t VirtAddr, size_t ByteCount);
15 uintptr_t       FindFreeRange(size_t ByteCount, int MaxBits);
16
17 // === CODE ===
18 int AllocateMemory(uintptr_t VirtAddr, size_t ByteCount)
19 {
20         uintptr_t       base = (VirtAddr >> 12) << 12;
21         size_t  size = (VirtAddr & 0xFFF) + ByteCount;
22         void    *tmp;
23         #if __WIN32__
24         tmp = VirtualAlloc(base, size, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
25         if( tmp == NULL ) {
26                 printf("ERROR: Unable to allocate memory (%i)\n", GetLastError());
27                 return -1;
28         }
29         #else
30 //      printf("AllocateMemory: mmap(%p, 0x%lx, ...)\n", (void*)base, ByteCount);
31         tmp = mmap((void*)base, size, PROT_READ|PROT_WRITE|PROT_EXEC, MAP_PRIVATE|MAP_FIXED|MAP_ANONYMOUS, -1, 0);
32         if( tmp == MAP_FAILED ) {
33                 printf("ERROR: Unable to allocate memory\n");
34                 perror("AllocateMemory");
35                 return -1;
36         }
37 //      printf("AllocateMemory: RETURN 0\n");
38         #endif
39         return 0;
40 }
41
42 uintptr_t FindFreeRange(size_t ByteCount, int MaxBits)
43 {
44         #if __WIN32__
45         # error "Windows FindFreeRange() unimplemented"
46         #else
47         uintptr_t       base, ofs, size;
48         uintptr_t       end = -1;
49         static const int        PAGE_SIZE = 0x1000;
50         
51         size = (ByteCount + PAGE_SIZE - 1) / PAGE_SIZE;
52         size *= PAGE_SIZE;
53
54         end <<= (sizeof(intptr_t)*8-MaxBits);
55         end >>= (sizeof(intptr_t)*8-MaxBits);
56 //      printf("end = %p\n", (void*)end);
57         
58 //      for( base = 0; base < end - size; base -= PAGE_SIZE )
59         for( base = end - size + 1; base > 0; base -= PAGE_SIZE )
60         {
61                 for( ofs = 0; ofs < size; ofs += PAGE_SIZE ) {
62                         if( msync( (void*)(base+ofs), 1, 0) == 0 )
63                                 break;
64                         if( errno != ENOMEM )
65                                 perror("FindFreeRange, msync");
66                 }
67                 if( ofs >= size ) {
68                         return base;
69                 }
70         }
71         return 0;
72         #endif
73 }

UCC git Repository :: git.ucc.asn.au