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 }