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 }