94 lines
2.0 KiB
C
94 lines
2.0 KiB
C
#ifdef __CYGWIN__
|
|
|
|
|
|
#include <sys/mman.h>
|
|
#include <stdio.h>
|
|
#include <strings.h>
|
|
#include <errno.h>
|
|
#include <assert.h>
|
|
|
|
#define pagesize 4096
|
|
#define pageshift 12
|
|
#define segment_size (16*pagesize)
|
|
#define segment_shift (4+pageshift)
|
|
#include "winmmap.h"
|
|
|
|
/* vim:syntax=c */
|
|
|
|
static unsigned short* table = 0;
|
|
static char* ap = 0;
|
|
static size_t as = 0;
|
|
|
|
static void*
|
|
do_mmap(size_t n){
|
|
void* x = mmap(0, n, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0);
|
|
if(x == (void*)-1){
|
|
fprintf(stderr, "failed to mmap: %s\n", strerror(errno));
|
|
exit(-1);
|
|
}
|
|
assert((((unsigned int)x) & (segment_size-1)) == 0);
|
|
return x;
|
|
}
|
|
|
|
static void
|
|
do_munmap(void* addr, size_t size){
|
|
int err = munmap(addr, size);
|
|
if(err){
|
|
fprintf(stderr, "failed to unmap\n");
|
|
exit(-1);
|
|
}
|
|
}
|
|
|
|
|
|
static void
|
|
init_table(){
|
|
assert(sizeof(unsigned short) == 2);
|
|
table = do_mmap(32*pagesize);
|
|
bzero(table, 32*pagesize);
|
|
}
|
|
|
|
void
|
|
win_munmap(char* addr, size_t size){
|
|
while(size){
|
|
unsigned int page = (((unsigned int) addr) >> pageshift);
|
|
unsigned int segment_index = page / 16;
|
|
unsigned int page_index = page & 15;
|
|
unsigned short alloc_bits = table[segment_index];
|
|
assert((alloc_bits & (1<<page_index)) == (1<<page_index));
|
|
unsigned int new_bits = alloc_bits & ~ (1 << page_index);
|
|
table[segment_index] = new_bits;
|
|
if(new_bits == 0){
|
|
do_munmap((void*)(segment_index*segment_size), segment_size);
|
|
}
|
|
size -= pagesize;
|
|
addr += pagesize;
|
|
}
|
|
}
|
|
|
|
|
|
char*
|
|
win_mmap(size_t size){
|
|
if(size <= as){
|
|
char* x = ap;
|
|
ap += size;
|
|
as -= size;
|
|
return x;
|
|
}
|
|
if(table == 0) init_table();
|
|
win_munmap(ap, as);
|
|
size_t segments = ((size+segment_size-1) >> segment_shift);
|
|
size_t aligned_size = segments << segment_shift;
|
|
char* addr = do_mmap(aligned_size);
|
|
unsigned int page = (((unsigned int) addr) >> pageshift);
|
|
unsigned int segment_index = page / 16;
|
|
int i;
|
|
for(i=0; i<segments; i++){
|
|
table[segment_index+i] = -1;
|
|
}
|
|
ap = addr + size;
|
|
as = aligned_size - size;
|
|
return addr;
|
|
}
|
|
|
|
#endif
|