AcessKernel - cleaning up debug messages
[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((void*)base, size, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
25         if( tmp == NULL ) {
26                 printf("ERROR: Unable to allocate memory (0x%x)\n", (int)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         uintptr_t       base, ofs, size;
45         uintptr_t       end = -1;
46         static const int        PAGE_SIZE = 0x1000;
47         
48         size = (ByteCount + PAGE_SIZE - 1) / PAGE_SIZE;
49         size *= PAGE_SIZE;
50
51         end <<= (sizeof(intptr_t)*8-MaxBits);
52         end >>= (sizeof(intptr_t)*8-MaxBits);
53 //      printf("end = %p\n", (void*)end);
54         
55 //      for( base = 0; base < end - size; base -= PAGE_SIZE )
56         for( base = end - size + 1; base > 0; base -= PAGE_SIZE )
57         {
58                 for( ofs = 0; ofs < size; ofs += PAGE_SIZE ) {
59                         #if __WIN32__
60                         MEMORY_BASIC_INFORMATION        info;
61                         VirtualQuery( (void*)(base + ofs), &info, sizeof(info) );
62                         if( info.State != MEM_FREE )
63                                 break;
64                         #else
65                         if( msync( (void*)(base+ofs), 1, 0) == 0 )
66                                 break;
67                         if( errno != ENOMEM )
68                                 perror("FindFreeRange, msync");
69                         #endif
70                 }
71                 if( ofs >= size ) {
72                         return base;
73                 }
74         }
75         return 0;
76 }

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