1 files changed, 40 insertions, 0 deletions
diff --git a/src/main/pe_map_raw_image.c b/src/main/pe_map_raw_image.c
new file mode 100644
index 0000000..8ae95d9
--- /dev/null
+++ b/src/main/pe_map_raw_image.c
@@ -0,0 +1,40 @@
+#include <stdint.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <limits.h>
+#include <errno.h>
+#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+
+#include <perk/perk.h>
+
+int pe_map_raw_image (int fd, const char * fname, struct pe_raw_image * map)
+{
+	struct stat stat;
+	int nfd, ret;
+
+	if ((nfd = !fd))
+		fd  = open(fname,O_RDONLY | O_CLOEXEC);
+
+	if ((fd < 0) || (fstat(fd,&stat) < 0))
+		return errno;
+
+	map->size = stat.st_size;
+	map->addr = (char *)mmap(0,map->size,PROT_READ,MAP_PRIVATE,fd,0);
+
+	if (map->addr == MAP_FAILED) {
+		map->addr = 0;
+		ret = PERK_MAP_ERROR;
+	} else
+		ret = 0;
+
+	if (nfd) close(fd);
+
+	return ret;
+}
+
+int pe_unmap_raw_image (struct pe_raw_image * map)
+{
+	return munmap(map->addr, map->size);
+};