1 files changed, 66 insertions, 0 deletions
diff --git a/src/logic/pe_image_raw_data.c b/src/logic/pe_image_raw_data.c
new file mode 100644
index 0000000..7ab6da6
--- /dev/null
+++ b/src/logic/pe_image_raw_data.c
@@ -0,0 +1,66 @@
+/***************************************************************/
+/*  perk: PE Resource Kit                                      */
+/*  Copyright (C) 2015--2021  SysDeer Technologies, LLC        */
+/*  Released under GPLv2 and GPLv3; see COPYING.PERK.          */
+/***************************************************************/
+
+#include <stdint.h>
+#include <stdbool.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+
+#include <perk/perk.h>
+#include "perk_driver_impl.h"
+#include "perk_errinfo_impl.h"
+
+int pe_raw_map_raw_image(
+	const struct pe_driver_ctx *	dctx,
+	int				fd,
+	const char *			path,
+	int				prot,
+	struct pe_raw_image *		map)
+{
+	int		ret;
+	struct stat	st;
+	bool		fnew;
+	int		fdcwd;
+
+	fdcwd = pe_driver_fdcwd(dctx);
+
+	if ((fnew = (fd < 0)))
+		fd  = openat(fdcwd,path,O_RDONLY | O_CLOEXEC);
+
+	if (fd < 0)
+		return PERK_SYSTEM_ERROR(dctx);
+
+	if ((ret = fstat(fd,&st) < 0) && fnew)
+		close(fd);
+
+	else if ((st.st_size == 0) && fnew)
+		close(fd);
+
+	if (ret < 0)
+		return PERK_SYSTEM_ERROR(dctx);
+
+	else if (st.st_size == 0)
+		return PERK_CUSTOM_ERROR(
+			dctx,PERK_ERR_IMAGE_SIZE_ZERO);
+
+	map->map_size = st.st_size;
+	map->map_addr = mmap(0,map->map_size,prot,MAP_PRIVATE,fd,0);
+
+	if (fnew)
+		close(fd);
+
+	return (map->map_addr == MAP_FAILED)
+		? PERK_SYSTEM_ERROR(dctx)
+		: 0;
+}
+
+int pe_raw_unmap_raw_image(struct pe_raw_image * map)
+{
+	return munmap(map->map_addr,map->map_size);
+}