You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.5 KiB
64 lines
1.5 KiB
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <fcntl.h>
|
|
#include <sys/mman.h>
|
|
|
|
#include <macros.h>
|
|
#include <map.h>
|
|
|
|
// Set all bytes to 0xff (ints i.e. fd will be -1)
|
|
static inline void _map_init(map_t *pOUT map)
|
|
{
|
|
memset(map, 0xff, sizeof(map_t));
|
|
}
|
|
|
|
static inline bool _map_is_file(const map_t* map)
|
|
{
|
|
return map->file.fd > 0;
|
|
}
|
|
|
|
map_result_t map_fd(int fd, bool write, usize size, off_t offset, map_t *pOUT map)
|
|
{
|
|
void* ptr = mmap(NULL, size, PROT_READ | (write ? PROT_WRITE : 0), MAP_SHARED, fd, offset);
|
|
if(ptr == MAP_FAILED) return MAP_ERR_MMAP;
|
|
map->file = (struct mm_file){
|
|
.origin = ptr,
|
|
.len = size,
|
|
.fd = fd,
|
|
.fd_offset = offset,
|
|
};
|
|
return MAP_SUCCESS;
|
|
}
|
|
|
|
map_result_t map_file(const char* file, bool write, usize size, off_t offset, map_t *pOUT map)
|
|
{
|
|
int fd = open(file, write ? O_RDWR : O_RDONLY);
|
|
if(fd < 0) return MAP_ERR_OPEN;
|
|
return map_fd(fd, write, size, offset, map);
|
|
}
|
|
|
|
map_result_t map_anon(void* ptr, usize len, map_t *pOUT map)
|
|
{
|
|
ptr = mmap(ptr, len, PROT_READ|PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | (ptr ? MAP_FIXED : 0), -1, 0);
|
|
if(ptr==MAP_FAILED) return MAP_ERR_MMAP;
|
|
|
|
// Make sure .file.fd is -1, so we can tell this is anon in `map_free()`.
|
|
_map_init(map);
|
|
map->anon = (struct mm_anon) {
|
|
.origin = ptr,
|
|
.len = len,
|
|
};
|
|
return MAP_SUCCESS;
|
|
}
|
|
|
|
map_result_t map_free(map_t map)
|
|
{
|
|
if( munmap(map.anon.origin, map.anon.len) != 0) return MAP_ERR_UNMAP;
|
|
if( _map_is_file(&map) ) {
|
|
if( close(map.file.fd) != 0 ) return MAP_ERR_CLOSE;
|
|
}
|
|
return MAP_SUCCESS;
|
|
}
|